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 HashingSource extends ForwardingSource { private final Mac mac; private final MessageDigest messageDigest; public static HashingSource md5(Source source) { return new HashingSource(source, "MD5"); } public static HashingSource sha1(Source source) { return new HashingSource(source, "SHA-1"); } public static HashingSource sha256(Source source) { return new HashingSource(source, "SHA-256"); } public static HashingSource hmacSha1(Source source, ByteString byteString) { return new HashingSource(source, byteString, "HmacSHA1"); } public static HashingSource hmacSha256(Source source, ByteString byteString) { return new HashingSource(source, byteString, "HmacSHA256"); } private HashingSource(Source source, String str) { super(source); try { this.messageDigest = MessageDigest.getInstance(str); this.mac = null; } catch (NoSuchAlgorithmException unused) { throw new AssertionError(); } } private HashingSource(Source source, ByteString byteString, String str) { super(source); 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.ForwardingSource, com.mbridge.msdk.thrid.okio.Source public long read(Buffer buffer, long j) throws IOException { long read = super.read(buffer, j); if (read != -1) { long j2 = buffer.size - read; long j3 = buffer.size; Segment segment = buffer.head; while (j3 > j2) { segment = segment.prev; j3 -= (long) (segment.limit - segment.pos); } while (j3 < buffer.size) { int i = (int) ((((long) segment.pos) + j2) - j3); MessageDigest messageDigest2 = this.messageDigest; if (messageDigest2 != null) { messageDigest2.update(segment.data, i, segment.limit - i); } else { this.mac.update(segment.data, i, segment.limit - i); } j3 += (long) (segment.limit - segment.pos); segment = segment.next; j2 = j3; } } return read; } public final ByteString hash() { MessageDigest messageDigest2 = this.messageDigest; return ByteString.of(messageDigest2 != null ? messageDigest2.digest() : this.mac.doFinal()); } }