mpd/python/build/verify.py

import hashlib

def feed_file(h, f):
    """Feed data read from an open file into the hashlib instance."""

    while True:
        data = f.read(65536)
        if len(data) == 0:
            # end of file
            break
        h.update(data)

def feed_file_path(h, path):
    """Feed data read from a file (to be opened by this function) into the hashlib instance."""

    with open(path, 'rb') as f:
        feed_file(h, f)

def file_digest(algorithm, path):
    """Calculate the digest of a file and return it in hexadecimal notation."""

    h = algorithm()
    feed_file_path(h, path)
    return h.hexdigest()

def guess_digest_algorithm(digest):
    l = len(digest)
    if l == 32:
        return hashlib.md5
    elif l == 40:
        return hashlib.sha1
    elif l == 64:
        return hashlib.sha256
    else:
        return None

def verify_file_digest(path, expected_digest):
    """Verify the digest of a file, and return True if the digest matches with the given expected digest."""

    algorithm = guess_digest_algorithm(expected_digest)
    assert(algorithm is not None)
    return file_digest(algorithm, path) == expected_digest