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

Question

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

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

Answer 1

你可以递归地DigestInputStream环绕 aDigestInputStream等等：

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();