blob: e0db751f22d90d42663129e3cd085aad1bfbb11a [file] [log] [blame]
#!/usr/bin/env python
# Copyright 2021 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""sha256_file.py takes one or more files, computes a SHA-256 hash over it,
and writes a .h/.cc file pair containing variables with the digest bytes.
Usage:
sha256_file.py path/to/hashes file1.txt file2.pak
Which will create path/to/hashes.h and path/to/hashes.cc with two variable
declarations, one for each of the specified input file's hash.
"""
import hashlib
import os.path
import sys
def main(argv):
if len(argv) < 3:
print('Usage: {} output_path_prefix file1.pak...'.format(argv[0]),
file=sys.stderr)
return 1
output_path_prefix = argv[1]
h_guard = output_path_prefix.upper().replace('/', '_') + '_H_'
h_contents = '#ifndef {guard}\n#define {guard}\n\n'.format(guard=h_guard)
cc_contents = '#include "{}.h"\n\n'.format(
os.path.basename(output_path_prefix))
for (name, value) in _hash_files(argv[2:]):
name = 'kSha256_' + os.path.basename(name).replace('.', '_')
h_contents += 'extern const std::array<uint8_t, 32> {};\n\n'.format(
name)
cc_contents += 'const std::array<uint8_t, 32> {} = {{'.format(name)
cc_contents += ', '.join(map(hex, value))
cc_contents += '};\n\n'
h_contents += '#endif // {}'.format(h_guard)
with open(output_path_prefix + '.h', 'w') as f:
f.write(FILE_TEMPLATE.format(contents=h_contents))
with open(output_path_prefix + '.cc', 'w') as f:
f.write(FILE_TEMPLATE.format(contents=cc_contents))
def _hash_files(files):
for path in files:
with open(path, 'rb') as f:
yield path, _hash_file_contents(f)
def _hash_file_contents(f):
sha2 = hashlib.sha256()
while True:
data = f.read(4096)
if not data:
break
sha2.update(data)
return sha2.digest()
FILE_TEMPLATE = """// Generated by chrome/tools/build/sha256_file.py
// !! DO NOT EDIT !!
#include <stdint.h>
#include <array>
{contents}
"""
if __name__ == '__main__':
sys.exit(main(sys.argv))