Add ArchiveArtifacts

ArchiveArtifacts check a downloaded archive against a provided checksum.
The Archive is downloaded to cache/<name>/<name>.<format>.
The only suported format is tar.
ArchiveArtifacts can be used as sources of an Allocator.

1 files changed, 11 insertions, 0 deletions

diff --git a/src/util.py b/src/util.py
index 027cbcb..8b9e585 100644
--- a/src/util.py
+++ b/src/util.py
@@ -17,6 +17,7 @@
 
 """Helper functions for allocbench"""
 
+import hashlib
 import os
 import subprocess
 import sys
@@ -173,3 +174,13 @@ def print_version_and_exit():
     
     print(f"{commit}{dirty}")
     exit(0)
+
+def sha1sum(filename):
+    """Return sha1sum of a file"""
+    sha1  = hashlib.sha1()
+    barray  = bytearray(64*1024)
+    view = memoryview(barray)
+    with open(filename, 'rb', buffering=0) as f:
+        for n in iter(lambda : f.readinto(view), 0):
+            sha1.update(view[:n])
+    return sha1.hexdigest()