使用DigestInputStream计算来自同一InputStream的多个校验和

时间:2013-10-10 16:00:59

标签: java md5 sha1 checksum gnupg

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

我不想重新读取流来计算校验和。

2 个答案:

答案 0 :(得分:13)

你可以递归地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();

答案 1 :(得分:2)

Javadoc非常清楚。您可以使用仅克隆来使用相同的算法计算不同的中间摘要。如果不多次读取流,则无法使用DigestInputStream计算不同的摘要算法。您必须使用常规InputStream和多个MessageDigest对象;读取数据一次,将每个缓冲区传递给所有MessageDigest个对象,以获得具有不同算法的多个摘要。

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

Pseudojava(省略错误处理)

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