orca/scripts/bindgen.py

#!/usr/bin/env python3
import sys

def bindgen(apiName, cdir):
	inPath = cdir + '/bindgen_' + apiName + '_api.txt'
	outPath = cdir + '/bindgen_' + apiName + '_api.c'

	inFile = open(inPath, 'r')
	outFile = open(outPath, 'w')

	stubs = []
	links = []

	def gen_stub(name, sig, native_name):
		if native_name == None:
			native_name = name

		src = 'const void* ' + name + '_stub(IM3Runtime runtime, IM3ImportContext _ctx, uint64_t * _sp, void * _mem)\n'
		src += '{\n'
		spIndex = 0

		parsingRet = True
		retCount = 0
		argCount = 0
		retString = ''
		argString = ''

		for index, c in enumerate(sig):
			if parsingRet:
				if retCount > 1:
					print('unsupported multiple return types\n')
					break
				if c == '(':
					parsingRet = False
					continue
				elif c == ')':
					print('unexpected ) while parsing return type\n')
					break;
				elif c == 'v':
					continue
				elif c == 'i':
					retString = '*((i32*)&_sp[0]) = '
				elif c == 'I':
					retString = '*((i64*)&_sp[0]) = '
				elif c == 'f':
					retString = '*((f32*)&_sp[0]) = '
				elif c == 'F':
					retString = '*((f64*)&_sp[0]) = '
				elif c == 'p':
					print('returning pointers is not supported yet\n')
					break
				else:
					print('unrecognized type ' + c + ' in procedure return\n')
					break
				retCount += 1
			else:
				argIndex = argCount + retCount
				if c == ')':
					break
				elif c == 'v':
					break
				elif c == 'i':
					argString += '*(i32*)&_sp[' + str(argIndex) + ']'
				elif c == 'I':
					argString += '*(i64*)&_sp[' + str(argIndex) + ']'
				elif c == 'f':
					argString += '*(f32*)&_sp[' + str(argIndex) + ']'
				elif c == 'F':
					argString += '*(f64*)&_sp[' + str(argIndex) + ']'
				elif c == 'p':
					argString += '(void*)((char*)_mem + *(i32*)&_sp[' + str(argIndex) + '])'
				else:
					print('unrecognized type ' + c + ' in procedure signature\n')
					break

				if index+2 < len(sig):
					argString += ', '
				argCount += 1

		src += '\t' + retString + native_name + '(' + argString + ');\n'
		src += '\treturn(0);\n'
		src += '}\n'
		stubs.append(src)

	def gen_link(name, sig):
		m3_Sig = ''
		for c in sig:
			if c == 'p':
				m3_Sig += 'i'
			else:
				m3_Sig += c

		src = '\tres = m3_LinkRawFunction(module, "*", "' + name + '", "' + m3_Sig + '", ' + name + '_stub);\n'
		src += '\tif(res != m3Err_none && res != m3Err_functionLookupFailed) { log_error("error: %s\\n", res); return(-1); }\n\n'
		links.append(src)

	for line in inFile:
		if line.isspace():
			continue
		desc = line.split()

		gen_stub(desc[0], desc[1], desc[2] if len(desc) > 2 else None)
		gen_link(desc[0], desc[1])

	linkProc = 'int bindgen_link_' + apiName + '_api(IM3Module module)\n'
	linkProc += '{\n'
	linkProc += '\tM3Result res;\n'

	for link in links:
		linkProc += link

	linkProc += '\treturn(0);\n'
	linkProc += '}\n'

	for stub in stubs:
		outFile.write(stub)

	outFile.write('\n')
	outFile.write(linkProc)

	inFile.close()
	outFile.close()

if __name__ == "__main__":
	if len(sys.argv) < 2:
		print("bindgen require an api name\n")
		exit(-1)

	apiName = sys.argv[1]
	cdir = ''

	if len(sys.argv) > 2:
		cdir = sys.argv[2]
	
	bindgen(apiName, cdir)