#!/usr/bin/env python3 from argparse import ArgumentParser import json def needs_arg_ptr_stub(decl): res = (decl['ret']['tag'] == 'S') for arg in decl['args']: if arg['type']['tag'] == 'S': res = True return(res) def printError(str): # This prints a string with a red foreground color. # See this link for an explanation of console escape codes: https://stackabuse.com/how-to-print-colored-text-in-python/ print("\x1b[38;5;196m" + "error: " + str + "\033[0;0m") def bindgen(apiName, spec, **kwargs): guest_stubs_path = kwargs.get("guest_stubs") guest_include = kwargs.get("guest_include") wasm3_bindings_path = kwargs.get("wasm3_bindings") if guest_stubs_path == None: guest_stubs_path = 'bindgen_' + apiName + '_guest_stubs.c' if wasm3_bindings_path == None: wasm3_bindings_path = 'bindgen_' + apiName + '_wasm3_bindings.c' host_bindings = open(wasm3_bindings_path, 'w') guest_bindings = None specFile = open(spec, 'r') data = json.load(specFile) for decl in data: if needs_arg_ptr_stub(decl): guest_bindings = open(guest_stubs_path, 'w') if guest_include != None: s = '#include"' + guest_include + '"\n\n' print(s, file=guest_bindings) break for decl in data: name = decl['name'] cname = decl.get('cname', name) if needs_arg_ptr_stub(decl): argPtrStubName = name + '_argptr_stub' # pointer arg stub declaration s = '' if decl['ret']['tag'] == 'S': s += 'void' else: s += decl['ret']['name'] s += ' ORCA_IMPORT(' + argPtrStubName + ') (' if decl['ret']['tag'] == 'S': s += decl['ret']['name'] + '* __retArg' if len(decl['args']) > 0: s += ', ' elif len(decl['args']) == 0: s += 'void' for i, arg in enumerate(decl['args']): s += arg['type']['name'] if arg['type']['tag'] == 'S': s += '*' s += ' ' + arg['name'] if i+1 < len(decl['args']): s += ', ' s += ');\n\n' # forward function to pointer arg stub declaration s += decl['ret']['name'] + ' ' + name + '(' if len(decl['args']) == 0: s += 'void' for i, arg in enumerate(decl['args']): s += arg['type']['name'] + ' ' + arg['name'] if i+1 < len(decl['args']): s += ', ' s += ')\n' s += '{\n' s += '\t' if decl['ret']['tag'] == 'S': s += decl['ret']['name'] + ' __ret;\n\t' elif decl['ret']['tag'] != 'v': s += decl['ret']['name'] s += ' __ret = ' s += argPtrStubName + '(' if decl['ret']['tag'] == 'S': s += '&__ret' if len(decl['args']) > 0: s += ', ' for i, arg in enumerate(decl['args']): if arg['type']['tag'] == 'S': s += '&' s += arg['name'] if i+1 < len(decl['args']): s += ', ' s += ');\n' if decl['ret']['tag'] != 'v': s += '\treturn(__ret);\n' s += '}\n\n' print(s, file=guest_bindings) # host-side stub s = 'const void* ' + cname + '_stub(IM3Runtime runtime, IM3ImportContext _ctx, uint64_t* _sp, void* _mem)' gen_stub = decl.get('gen_stub', True) if gen_stub == False: s += ';\n\n' else: s += '\n{\n' # NOTE: check and cast arguments retTag = decl['ret']['tag'] firstArgIndex = 0 if retTag != 'v': firstArgIndex = 1 if retTag == 'S': retTypeName = decl['ret']['name'] retTypeCName = decl['ret'].get('cname', retTypeName) s += retTypeCName + '* __retPtr = (' + retTypeCName + '*)((char*)_mem + *(i32*)&_sp[0]);\n' s += '\t{\n' s += '\t\tOC_ASSERT_DIALOG(((char*)__retPtr >= (char*)_mem) && (((char*)__retPtr - (char*)_mem) < m3_GetMemorySize(runtime)), "return pointer is out of bounds");\n' s += '\t\tOC_ASSERT_DIALOG((char*)__retPtr + sizeof(' + retTypeCName + ') <= ((char*)_mem + m3_GetMemorySize(runtime)), "return pointer is out of bounds");\n' s += '\t}\n' for argIndex, arg in enumerate(decl['args']): argName = arg['name'] typeName = arg['type']['name'] typeCName = arg['type'].get('cname', typeName) argTag = arg['type']['tag'] s += '\t' if argTag == 'i': s += typeCName + ' ' + argName + ' = ('+typeCName+')*(i32*)&_sp[' + str(firstArgIndex + argIndex) + '];\n' elif argTag == 'I': s += typeCName + ' ' + argName + ' = ('+typeCName+')*(i64*)&_sp[' + str(firstArgIndex + argIndex) + '];\n' elif argTag == 'f': s += typeCName + ' ' + argName + ' = ('+typeCName+')*(f32*)&_sp[' + str(firstArgIndex + argIndex) + '];\n' elif argTag == 'F': s += typeCName + ' ' + argName + ' = ('+typeCName+')*(f64*)&_sp[' + str(firstArgIndex + argIndex) + '];\n' elif argTag == 'p': s += typeCName + ' ' + argName + ' = ('+ typeCName +')((char*)_mem + *(u32*)&_sp[' + str(firstArgIndex + argIndex) + ']);\n' elif argTag == 'S': s += typeCName + ' ' + argName + ' = *('+ typeCName +'*)((char*)_mem + *(u32*)&_sp[' + str(firstArgIndex + argIndex) + ']);\n' else: print('unrecognized type ' + c + ' in procedure signature\n') break # check pointer arg length for arg in decl['args']: argName = arg['name'] typeName = arg['type']['name'] typeCName = arg['type'].get('cname', typeName) argTag = arg['type']['tag'] argLen = arg.get('len') if argTag == 'p': if argLen == None: printError("binding '" + name + "' missing pointer length decoration for param '" + argName + "'") else: s += '\t{\n' s += '\t\tOC_ASSERT_DIALOG(((char*)'+ argName + ' >= (char*)_mem) && (((char*)'+ argName +' - (char*)_mem) < m3_GetMemorySize(runtime)), "parameter \''+argName+'\' is out of bounds");\n' s += '\t\tOC_ASSERT_DIALOG((char*)' + argName + ' + ' proc = argLen.get('proc') if proc != None: s += proc + '(runtime, ' lenProcArgs = argLen['args'] for i, lenProcArg in enumerate(lenProcArgs): s += lenProcArg if i < len(lenProcArgs)-1: s += ', ' s += ')' else: components = argLen.get('components') countArg = argLen.get('count') if components != None: s += str(components) if countArg != None: s += '*' if countArg != None: s += countArg if typeCName.endswith('**') or (typeCName.startswith('void') == False and typeCName.startswith('const void') == False): s += '*sizeof('+typeCName[:-1]+')' s += ' <= ((char*)_mem + m3_GetMemorySize(runtime)), "parameter \''+argName+'\' is out of bounds");\n' s += '\t}\n' s += '\t' if retTag == 'i': s += '*((i32*)&_sp[0]) = (i32)' elif retTag == 'I': s += '*((i64*)&_sp[0]) = (i64)' elif retTag == 'f': s += '*((f32*)&_sp[0]) = (f32)' elif retTag == 'F': s += '*((f64*)&_sp[0]) = (f64)' elif retTag == 'S': s += '*__retPtr = ' elif retTag == 'p': print("Warning: " + name + ": pointer return type not supported yet") s += cname + '(' for i, arg in enumerate(decl['args']): s += arg['name'] if i+1 < len(decl['args']): s += ', ' s += ');\n\treturn(0);\n}\n\n' print(s, file=host_bindings) # link function s = 'int bindgen_link_' + apiName + '_api(IM3Module module)\n{\n' s += ' M3Result res;\n' s += ' int ret = 0;\n' for decl in data: name = decl['name'] cname = decl.get('cname', name) if needs_arg_ptr_stub(decl): name = name + '_argptr_stub' m3Sig = '' if decl['ret']['tag'] == 'S': m3Sig += 'v' else: m3Sig += decl['ret']['tag'] m3Sig += '(' if decl['ret']['tag'] == 'S': m3Sig += 'i' for arg in decl['args']: tag = arg['type']['tag'] if tag == 'p' or tag == 'S': tag = 'i' m3Sig += tag m3Sig += ')' s += ' res = m3_LinkRawFunction(module, "*", "' + name + '", "' + m3Sig + '", ' + cname + '_stub);\n' s += ' if(res != m3Err_none && res != m3Err_functionLookupFailed)\n' s += ' {\n' s += ' oc_log_error("Couldn\'t link function ' + name + ' (%s)\\n", res);\n' s += ' ret = -1;\n' s += ' }\n\n' s += '\treturn(ret);\n}\n' print(s, file=host_bindings) if __name__ == "__main__": parser = ArgumentParser(prog='bindgen.py') parser.add_argument('api') parser.add_argument('spec') parser.add_argument('-g', '--guest-stubs') parser.add_argument('--guest-include') parser.add_argument('--wasm3-bindings') args = parser.parse_args() apiName = args.api spec = args.spec guest_stubs_path = args.guest_stubs if guest_stubs_path == None: guest_stubs_path = 'bindgen_' + apiName + '_guest_stubs.c' wasm3_bindings_path = args.wasm3_bindings if wasm3_bindings_path == None: wasm3_bindings_path = 'bindgen_' + apiName + '_wasm3_bindings.c' bindgen(apiName, spec, guest_stubs=guest_stubs_path, guest_include=args.guest_include, wasm3_bindings=wasm3_bindings_path, )