From 0715cc5f94438d58e2fc59c065a0afbd3dbb525a Mon Sep 17 00:00:00 2001
From: Helmut Grohne <helmut@subdivi.de>
Date: Sat, 16 Apr 2016 11:22:18 +0200
Subject: add a validate method to HashedStream

---
 dedup/hashing.py | 10 ++++++++++
 importpkg.py     |  5 +----
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/dedup/hashing.py b/dedup/hashing.py
index 1419811..2a83929 100644
--- a/dedup/hashing.py
+++ b/dedup/hashing.py
@@ -183,3 +183,13 @@ class HashedStream(object):
 
     def hexdigest(self):
         return self.hashobj.hexdigest()
+
+    def validate(self, hexdigest):
+        """Soak up any remaining input and validate the read data using the
+        given hexdigest.
+        @raises ValueError: when the hash does not match
+        """
+        while self.read(65536):
+            pass
+        if self.hexdigest() != hexdigest:
+            raise ValueError("hash sum mismatch")
diff --git a/importpkg.py b/importpkg.py
index 2cdefc4..f72cf03 100755
--- a/importpkg.py
+++ b/importpkg.py
@@ -107,10 +107,7 @@ def main():
         stdin = HashedStream(stdin, hashlib.sha256())
     process_package(stdin, hash_functions, dumper.represent)
     if options.hash:
-        while stdin.read(4096):
-            pass
-        if stdin.hexdigest() != options.hash:
-            raise ValueError("hash sum mismatch")
+        stdin.validate(options.hash)
     dumper.represent("commit")
     dumper.close()
 
-- 
cgit v1.2.3