#!/usr/bin/env python # # Copyright (C) 2011-2012 The Android Open Source Project # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # import os import sys try: import hashlib sha1 = hashlib.sha1 except ImportError, e: import sha sha1 = sha.sha def compute_sha1(h, path): f = open(path, 'rb') while True: buf = f.read(1024) h.update(buf) if len(buf) < 1024: break f.close() """The result is a list of pair of file path and its SHA-1 digest""" def compute_sha1_list(path_list): result = [] for path in path_list: h = sha1() compute_sha1(h, path) result.append((path, h.digest())) return result """For each path like /xxx/libfoo.so, generate a symbol named libfoo_so_SHA1""" def get_symbol_name(path): return os.path.basename(path).replace('.', '_') + '_SHA1'; """Print out header for assembly file.""" def print_asm_header(symbols): sys.stdout.write(""" /*\ * The mid-2007 version of gcc that ships with Macs requires a\n\ * comma on the .section line, but the rest of the world thinks\n\ * that's a syntax error. It also wants globals to be explicitly\n\ * prefixed with \"_\" as opposed to modern gccs that do the\n\ * prefixing for you.\n\ */\n\ """) for sym in symbols: sys.stdout.write(""" #ifdef __APPLE_CC__ .globl _%s\n\ #else\n\ .globl %s\n\ #endif\ """ % (sym, sym)) sys.stdout.write(""" #ifdef __APPLE_CC__ .section .rodata,\n\ #else\n\ .section .rodata\n\ #endif\ """ ) def print_asm_data(data, size): col = 0 for i in xrange(size): c = data[i] if col == 0: sys.stdout.write(".byte ") elif col % 4 == 0: sys.stdout.write(", ") else: sys.stdout.write(",") sys.stdout.write("0x%02x" % ord(c)) col += 1 if col == 8: sys.stdout.write("\n") col = 0 if col != 0: sys.stdout.write("\n") def print_asm_symbol_data(sym, h): sys.stdout.write(""" .align 8 #ifdef __APPLE_CC__ _%s:\n\ #else\n\ %s:\n\ #endif\n\ """ % (sym, sym)) print_asm_data(h, 20) def print_asm(x): symbols = [get_symbol_name(item[0]) for item in x] print_asm_header(symbols) for (symbol, y) in zip(symbols, x): print_asm_symbol_data(symbol, y[1]) def main(): if len(sys.argv) < 2: print 'USAGE:', sys.argv[0], '[OUTPUT] [INPUTs]' sys.exit(1) result = compute_sha1_list(sys.argv[1:]) print_asm(result) if __name__ == '__main__': main()