package com.mbridge.msdk.thrid.okio; import java.io.IOException; import java.security.InvalidKeyException; import java.security.MessageDigest; import java.security.NoSuchAlgorithmException; import javax.crypto.Mac; import javax.crypto.spec.SecretKeySpec; public final class HashingSink extends ForwardingSink { private final Mac mac; private final MessageDigest messageDigest; public static HashingSink md5(Sink sink) { return new HashingSink(sink, "MD5"); } public static HashingSink sha1(Sink sink) { return new HashingSink(sink, "SHA-1"); } public static HashingSink sha256(Sink sink) { return new HashingSink(sink, "SHA-256"); } public static HashingSink sha512(Sink sink) { return new HashingSink(sink, "SHA-512"); } public static HashingSink hmacSha1(Sink sink, ByteString byteString) { return new HashingSink(sink, byteString, "HmacSHA1"); } public static HashingSink hmacSha256(Sink sink, ByteString byteString) { return new HashingSink(sink, byteString, "HmacSHA256"); } public static HashingSink hmacSha512(Sink sink, ByteString byteString) { return new HashingSink(sink, byteString, "HmacSHA512"); } private HashingSink(Sink sink, String str) { super(sink); try { this.messageDigest = MessageDigest.getInstance(str); this.mac = null; } catch (NoSuchAlgorithmException unused) { throw new AssertionError(); } } private HashingSink(Sink sink, ByteString byteString, String str) { super(sink); try { Mac instance = Mac.getInstance(str); this.mac = instance; instance.init(new SecretKeySpec(byteString.toByteArray(), str)); this.messageDigest = null; } catch (NoSuchAlgorithmException unused) { throw new AssertionError(); } catch (InvalidKeyException e) { throw new IllegalArgumentException(e); } } @Override // com.mbridge.msdk.thrid.okio.Sink, com.mbridge.msdk.thrid.okio.ForwardingSink public void write(Buffer buffer, long j) throws IOException { Util.checkOffsetAndCount(buffer.size, 0, j); Segment segment = buffer.head; long j2 = 0; while (j2 < j) { int min = (int) Math.min(j - j2, (long) (segment.limit - segment.pos)); MessageDigest messageDigest2 = this.messageDigest; if (messageDigest2 != null) { messageDigest2.update(segment.data, segment.pos, min); } else { this.mac.update(segment.data, segment.pos, min); } j2 += (long) min; segment = segment.next; } super.write(buffer, j); } public final ByteString hash() { MessageDigest messageDigest2 = this.messageDigest; return ByteString.of(messageDigest2 != null ? messageDigest2.digest() : this.mac.doFinal()); } }