X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=retrace%2Fdllretrace.py;h=ef1483f0b83f82f46a2c18da58de5a20e5af3767;hb=8512a4b2ec582407bbfe743772af506c4133e34f;hp=926bacd250ec7017a97929aa86311e16a4a6d53a;hpb=3c683135551c596282eb5718e72a5cd087da5c37;p=apitrace

diff --git a/retrace/dllretrace.py b/retrace/dllretrace.py
index 926bacd..ef1483f 100644
--- a/retrace/dllretrace.py
+++ b/retrace/dllretrace.py
@@ -37,7 +37,7 @@ class DllDispatcher(Dispatcher):
         print r'HMODULE g_h%sModule = NULL;' % (tag,)
         print r''
         print r'static PROC'
-        print r'_getPublicProcAddress(LPCSTR lpProcName) {'
+        print r'_get%sProcAddress(LPCSTR lpProcName) {' % tag
         print r'    if (!g_h%sModule) {' % tag
         print r'        if (g_sz%sDllName) {' % tag
         print r'            g_h%sModule = LoadLibraryA(g_sz%sDllName);' % (tag, tag)
@@ -59,6 +59,10 @@ class DllDispatcher(Dispatcher):
 
         Dispatcher.dispatchModule(self, module)
 
+    def getProcAddressName(self, module, function):
+        assert self.isFunctionPublic(module, function)
+        return '_get%sProcAddress' % (module.name.upper())
+
 
 class DllRetracer(Retracer):