1 #!/usr/bin/python3 2 3 import os 4 import re 5 6 HERE = os.path.dirname(os.path.abspath(__file__)) 7 BLAKE2 = os.path.join(HERE, 'impl') 8 9 PUBLIC_SEARCH = re.compile(r'\ int (blake2[bs]p?[a-z_]*)\(') 10 11 12 def getfiles(): 13 for name in os.listdir(BLAKE2): 14 name = os.path.join(BLAKE2, name) 15 if os.path.isfile(name): 16 yield name 17 18 19 def find_public(): 20 public_funcs = set() 21 for name in getfiles(): 22 with open(name) as f: 23 for line in f: 24 # find public functions 25 mo = PUBLIC_SEARCH.search(line) 26 if mo: 27 public_funcs.add(mo.group(1)) 28 29 for f in sorted(public_funcs): 30 print('#define {0:<18} PyBlake2_{0}'.format(f)) 31 32 return public_funcs 33 34 35 def main(): 36 lines = [] 37 with open(os.path.join(HERE, 'blake2b_impl.c')) as f: 38 for line in f: 39 line = line.replace('blake2b', 'blake2s') 40 line = line.replace('BLAKE2b', 'BLAKE2s') 41 line = line.replace('BLAKE2B', 'BLAKE2S') 42 lines.append(line) 43 with open(os.path.join(HERE, 'blake2s_impl.c'), 'w') as f: 44 f.write(''.join(lines)) 45 # find_public() 46 47 48 if __name__ == '__main__': 49 main() 50
