fix: actually verify payload digest when needed. (#46)

CorentinB · Sep 14, 2024 · 7c0b7a6 · 7c0b7a6
1 parent fd03901
commit 7c0b7a6
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/cmd/verify.go b/cmd/verify.go
@@ -218,6 +218,13 @@ func verifyPayloadDigest(record *warc.Record, filepath string) (errorsCount int,
 			errorsCount++
 			return errorsCount, valid
 		}
+
+		if payloadDigest != payloadDigestSplitted[1] {
+			logger.Error("payload digests do not match", "file", filepath, "recordID", record.Header.Get("WARC-Record-ID"), "expected", payloadDigestSplitted[1], "got", payloadDigest)
+			valid = false
+			errorsCount++
+			return errorsCount, valid
+		}
 	} else if payloadDigestSplitted[0] == "sha256" {
 		payloadDigest := warc.GetSHA256Base16(resp.Body)
 		if payloadDigest == "ERROR" {
@@ -226,6 +233,13 @@ func verifyPayloadDigest(record *warc.Record, filepath string) (errorsCount int,
 			errorsCount++
 			return errorsCount, valid
 		}
+
+		if payloadDigest != payloadDigestSplitted[1] {
+			logger.Error("payload digests do not match", "file", filepath, "recordID", record.Header.Get("WARC-Record-ID"), "expected", payloadDigestSplitted[1], "got", payloadDigest)
+			valid = false
+			errorsCount++
+			return errorsCount, valid
+		}
 	} else {
 		logger.Error("WARC-Payload-Digest is not SHA1 or SHA256", "file", filepath, "recordID", record.Header.Get("WARC-Record-ID"))
 		valid = false