|
|
|
import sys, re, argparse
|
|
|
|
|
|
|
|
#RE_PROTOTYPE = re.compile(r'MCLBN_DLL_API\s\w\s\w\([^)]*\);')
|
|
|
|
RE_PROTOTYPE = re.compile(r'\w*\s(\w*)\s(\w*)\(([^)]*)\);')
|
|
|
|
def export_functions(modName, fileNames, reToAddUnderscore):
|
|
|
|
if not reToAddUnderscore:
|
|
|
|
reToAddUnderscore = r'(mclBn_init|setStr|getStr|[sS]erialize|setLittleEndian|setHashOf|hashAndMapTo|DecStr|HexStr|HashTo|blsSign|blsVerify|GetCurveOrder|GetFieldOrder|KeyShare|KeyRecover|blsSignatureRecover|blsInit)'
|
|
|
|
reSpecialFunctionName = re.compile(reToAddUnderscore)
|
|
|
|
if modName:
|
|
|
|
print 'function define_exported_' + modName + '(mod) {'
|
|
|
|
comma = ''
|
|
|
|
for fileName in fileNames:
|
|
|
|
with open(fileName, 'rb') as f:
|
|
|
|
for line in f.readlines():
|
|
|
|
p = RE_PROTOTYPE.search(line)
|
|
|
|
if p:
|
|
|
|
ret = p.group(1)
|
|
|
|
name = p.group(2)
|
|
|
|
arg = p.group(3)
|
|
|
|
if modName:
|
|
|
|
retType = 'null' if ret == 'void' else 'number'
|
|
|
|
if arg == '' or arg == 'void':
|
|
|
|
paramType = '[]'
|
|
|
|
else:
|
|
|
|
paramType = '[' + ("'number', " * len(arg.split(','))) + ']'
|
|
|
|
if reSpecialFunctionName.search(name):
|
|
|
|
exportName = '_' + name # to wrap function
|
|
|
|
else:
|
|
|
|
exportName = name
|
|
|
|
print "{0} = mod.cwrap('{1}', '{2}', {3})".format(exportName, name, retType, paramType)
|
|
|
|
else:
|
|
|
|
print comma + "'_" + name + "'",
|
|
|
|
if comma == '':
|
|
|
|
comma = ','
|
|
|
|
if modName:
|
|
|
|
print '}'
|
|
|
|
|
|
|
|
def main():
|
|
|
|
p = argparse.ArgumentParser('export_functions')
|
|
|
|
p.add_argument('header', type=str, nargs='+', help='headers')
|
|
|
|
p.add_argument('-js', type=str, nargs='?', help='module name')
|
|
|
|
p.add_argument('-re', type=str, nargs='?', help='regular expression file to add underscore to function name')
|
|
|
|
args = p.parse_args()
|
|
|
|
|
|
|
|
reToAddUnderscore = ''
|
|
|
|
if args.re:
|
|
|
|
reToAddUnderscore = open(args.re).read().strip()
|
|
|
|
export_functions(args.js, args.header, reToAddUnderscore)
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
main()
|
|
|
|
|