mirror of
https://github.com/vivisect/vivisect
synced 2026-06-08 18:04:23 +00:00
James Gross
80de840881
A lot of cleanup things in prep for a python 3 transition. Getting rid of the old exception syntax, converting prints over to logging, cutting random scraps of code to be proper unit tests, cut away some older bits of code, etc. This still works in python2. It's just a lot of tidying up. There are no major functionality changes.
79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
'''
|
|
Calling convention and API definitions for various APIs/archs.
|
|
'''
|
|
import imp
|
|
import sys
|
|
import vstruct.primitives
|
|
|
|
|
|
class ImportApi:
|
|
|
|
def __init__(self):
|
|
self._api_lookup = {}
|
|
self._apitype_lookup = {}
|
|
|
|
def getImpApiType(self, tname):
|
|
return self._apitype_lookup.get(tname)
|
|
|
|
def updateApiDef(self, apidict):
|
|
self._api_lookup.update(apidict)
|
|
|
|
def getImpApi(self, funcname):
|
|
'''
|
|
An API definition consists of the following:
|
|
( rettype, retname, callconv, funcname, ( (argtype, argname), ...) )
|
|
'''
|
|
return self._api_lookup.get(funcname.lower())
|
|
|
|
def getImpApiCallConv(self, funcname):
|
|
ret = self._api_lookup.get(funcname.lower())
|
|
if ret is None:
|
|
return None
|
|
return ret[2]
|
|
|
|
def getImpApiArgs(self, funcname):
|
|
ret = self._api_lookup.get(funcname.lower())
|
|
if ret is None:
|
|
return None
|
|
return ret[4]
|
|
|
|
def getImpApiRetType(self, funcname):
|
|
ret = self._api_lookup.get(funcname.lower())
|
|
if ret is None:
|
|
return None
|
|
return ret[0]
|
|
|
|
def getImpApiRetName(self, funcname):
|
|
ret = self._api_lookup.get(funcname.lower())
|
|
if ret is None:
|
|
return None
|
|
return ret[1]
|
|
|
|
def getImpApiArgTypes(self, funcname):
|
|
ret = self._api_lookup.get(funcname.lower())
|
|
if ret is None:
|
|
return None
|
|
return [argt for (argt, argn) in ret[4]]
|
|
|
|
def getImpApiArgNames(self, funcname):
|
|
ret = self._api_lookup.get(funcname.lower())
|
|
if ret is None:
|
|
return None
|
|
return [argn for (argt, argn) in ret[4]]
|
|
|
|
def addImpApi(self, api, arch):
|
|
api = api.lower()
|
|
arch = arch.lower()
|
|
modname = 'vivisect.impapi.%s.%s' % (api, arch)
|
|
# mod = imp.load_module( modname, *imp.find_module( modname ) )
|
|
__import__(modname)
|
|
mod = sys.modules[modname]
|
|
self._api_lookup.update(mod.api)
|
|
self._apitype_lookup.update(mod.apitypes)
|
|
|
|
|
|
def getImportApi(api, arch):
|
|
impapi = ImportApi()
|
|
impapi.addImpApi(api, arch)
|
|
return impapi
|