Lubuntu/ubuntu-dev-tools
3
0
Fork 0
mirror of https://git.launchpad.net/ubuntu-dev-tools synced 2025-03-13 08:01:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

misc: add verify_file_checksums() function

Browse Source 
Signed-off-by: Dan Streetman <ddstreet@canonical.com>
This commit is contained in:
Dan Streetman 2020-07-21 15:00:40 -04:00
parent 359cb18d8d
commit e5a42b1ba1
1 changed files with 49 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
ubuntutools/misc.py
View File

@ -201,7 +201,20 @@ def codename_to_distribution(codename):
return distro
def verify_file_checksum(pathname, alg, checksum, size=0):
def verify_file_checksums(pathname, checksums={}, size=0):
""" verify checksums of file
Any failure will log an error.
pathname: str
full path to file
checksums: dict
keys are alg name, values are expected checksum
size: int
size of file, if known
Returns True if all checks pass, False otherwise
"""
if not os.path.isfile(pathname):
Logger.error('File not found: %s', pathname)
return False
@ -210,6 +223,8 @@ def verify_file_checksum(pathname, alg, checksum, size=0):
Logger.error('File %s incorrect size, got %s expected %s',
filename, os.path.getsize(pathname), size)
return False
for (alg, checksum) in checksums.items():
h = hashlib.new(alg)
with open(pathname, 'rb') as f:
while True:
@ -224,7 +239,25 @@ def verify_file_checksum(pathname, alg, checksum, size=0):
else:
Logger.error('File %s checksum (%s) mismatch: got %s expected %s',
filename, alg, h.hexdigest(), checksum)
return match
return False
return True
def verify_file_checksum(pathname, alg, checksum, size=0):
""" verify checksum of file
pathname: str
full path to file
alg: str
name of checksum alg
checksum: str
expected checksum
size: int
size of file, if known
Returns True if all checks pass, False otherwise
"""
return verify_file_checksums(pathname, {alg: checksum}, size)
def download(src, dst, size=0):