X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=trace.py;h=7b2ba73941fa60c91aab68326b4838221b050a02;hb=ead6aad30fdde3921e0224895aa066c907d76c85;hp=5711fa0ae508cabaea359e6d16a2ca53f695c218;hpb=9922183f368ba2bfbe3bce18320684212cef8c2e;p=apitrace

diff --git a/trace.py b/trace.py
index 5711fa0..7b2ba73 100644
--- a/trace.py
+++ b/trace.py
@@ -27,6 +27,7 @@
 
 
 import stdapi
+from dispatch import Dispatcher
 
 
 def interface_wrap_name(interface):
@@ -437,13 +438,11 @@ class DllTracer(Tracer):
         return '__%s' % (function.name,)
 
     def header(self, api):
-        Tracer.header(self, api)
-
         print '''
 static HINSTANCE g_hDll = NULL;
 
 static PROC
-__GetProcAddress(LPCSTR lpProcName)
+__getPublicProcAddress(LPCSTR lpProcName)
 {
     if (!g_hDll) {
         char szDll[MAX_PATH] = {0};
@@ -465,3 +464,8 @@ __GetProcAddress(LPCSTR lpProcName)
 
     ''' % self.dllname
 
+        dispatcher = Dispatcher()
+        dispatcher.dispatch_api(api)
+
+        Tracer.header(self, api)
+