java - 哈希整个对象而不先转换为 byte[]

Question

我想获取特定 java 对象的 sha2 哈希值。我不希望它是 int，我想要 byte[] 或至少是 String。我有以下代码来创建 sha2:

static byte[] sha2(byte[] message) {
    if (message == null || message.length == 0) {
        throw new IllegalArgumentException("message is null");
    }
    try {
        MessageDigest sha256 = MessageDigest.getInstance(SHA_256);
        return sha256.digest(message);
    } catch (NoSuchAlgorithmException e) {
        throw new IllegalArgumentException(e);
    }
}

我可以将我的对象转换为 byte[]，但我认为将大数组存储在内存中只是为了创建 32 字节数组不是一个好主意。那么如何计算对象的 sha2(或者另一个加密哈希函数)呢？

Answer 1

您不必将整个对象加载到内存中，您可以将它的一部分加载到临时缓冲区中。

使用 FileOutputStream/BufferedOutputStream 将对象转储到临时文件中，这将确保序列化对象不会污染 JVM 内存。

使用 FileInputStream/BufferedInputStream 从临时文件加载序列化对象，并将其提供给循环中的 MessageDigest#update(buf) 方法。

最后调用MessageDigest#digest()完成工作:

int[] buf = new int[1024];
while (/* has more data */) {
    int readBytes = readIntoBuf(buf);
    sha256.update(buf, 0, readBytes);
}
return sha256.digest();

如果您有能力在内存中存储整个序列化对象，请使用 ByteArrayOutputStream 并将结果 byte[] 传递给 MessageDigest#digest(buf):

try (ByteArrayOutputStream baos = new ByteArrayOutputStream();
        ObjectOutputStream oos = new ObjectOuputStream(baos)) {
    oos.writeObject(obj);

    MessageDigest sha256 = MessageDigest.getInstance(SHA_256);
    return sha256.digest(baos.toByteArray());
}