fwupd/libfwupdplugin/fu-hash.py
2021-06-16 06:41:46 -05:00

34 lines
889 B
Python

#!/usr/bin/python3
# SPDX-License-Identifier: LGPL-2.1+
""" Builds a header for the plugins to include """
# pylint: disable=invalid-name,wrong-import-position,pointless-string-statement
import sys
import hashlib
def usage(return_code):
"""print usage and exit with the supplied return code"""
if return_code == 0:
out = sys.stdout
else:
out = sys.stderr
out.write("usage: fu-hash.py <HEADER> <SRC1> <SRC2>...")
sys.exit(return_code)
if __name__ == "__main__":
if {"-?", "--help", "--usage"}.intersection(set(sys.argv)):
usage(0)
if len(sys.argv) < 3:
usage(1)
m = hashlib.sha256()
for argv in sys.argv[2:]:
with open(argv, "rb") as f:
m.update(f.read())
with open(sys.argv[1], "w") as f2:
f2.write("#pragma once\n")
f2.write('#define FU_BUILD_HASH "%s"\n' % m.hexdigest())