使用 DigestInputStream 从同一个 InputStream 计算多个校验和

Question

我试图弄清楚如何使用DigestInputStream 基于相同的InputStream 读取多个摘要（md5、sha1、gpg）。从我在文档中检查的内容来看，似乎可以通过克隆摘要来实现。有人可以说明一下吗？

我不想为了计算校验和而重新读取流。

Answer 1

您可以将DigestInputStream 包裹在DigestInputStream 周围，以此类推：

DigestInputStream shaStream = new DigestInputStream(
    inStream, MessageDigest.getInstance("SHA-1"));
DigestInputStream md5Stream = new DigestInputStream(
    shaStream, MessageDigest.getInstance("MD5"));
// VERY IMPORTANT: read from final stream since it's FilterInputStream
byte[] shaDigest = shaStream.getMessageDigest().digest();
byte[] md5Digest = md5Stream.getMessageDigest().digest();

Answer 2

Javadoc 非常清晰。您只能使用克隆来使用相同的算法计算不同的中间摘要。您不能使用DigestInputStream 来计算不同的摘要算法，而无需多次读取流。您必须使用常规的InputStream 和多个MessageDigest 对象；读取一次数据，将每个缓冲区传递给所有MessageDigest 对象，以使用不同算法获取多个摘要。

您可以轻松地将其封装在您自己的DigestInputStream 变体中，例如MultipleDigestInputStream，它遵循相同的通用方法，但接受MessageDigest 对象或算法名称的集合。

伪java（省略错误处理）

MessageDigest sha = MessageDigest.getInstance("SHA-1");
MessageDigest md5 = MessageDigest.getInstance("MD5");
InputStream input = ...;
byte[] buffer = new byte[BUFFER_SIZE];
int len;
while((len = input.read(buffer)) >= 0)
{
    sha.update(buffer,0,len);
    md5.update(buffer,0,len);
    ...
}
byte[] shaDigest = sha.digest();
byte[] md5Digest = md5.digest();