codeql/java/ql/src/Performance/InefficientOutputStreamBad.java at main · github/codeql · GitHub

Repository navigation

27 lines (23 loc) · 726 Bytes

public class DigestCheckingFileOutputStream extends OutputStream {

private DigestOutputStream digest;

private byte[] expectedMD5;

public DigestCheckingFileOutputStream(File file, byte[] expectedMD5)

throws IOException, NoSuchAlgorithmException {

this.expectedMD5 = expectedMD5;

digest = new DigestOutputStream(new FileOutputStream(file),

MessageDigest.getInstance("MD5"));

}

@Override

public void write(int b) throws IOException {

digest.write(b);

}

@Override

public void close() throws IOException {

super.close();

digest.close();

byte[] md5 = digest.getMessageDigest().digest();

if (expectedMD5 != null && !Arrays.equals(expectedMD5, md5)) {

throw new InternalError();

}