Fix D3D9 tracer build.

diff --git a/d3d9.py b/d3d9.py
index f2675d904399174ea59d90e275749533963bebd5..057fad6906671780132176cd4f0929f06b8c00a6 100644 (file)
--- a/d3d9.py
+++ b/d3d9.py
@@ -419,6 +419,7 @@ if __name__ == '__main__':
     print '#include <d3d9.h>'
     print
     print '#include "trace_write.hpp"'
+    print '#include "os.hpp"'
     print
     tracer = D3D9Tracer('d3d9.dll')
     tracer.trace_api(d3d9)

diff --git a/trace.py b/trace.py
index 5711fa0ae508cabaea359e6d16a2ca53f695c218..7b2ba73941fa60c91aab68326b4838221b050a02 100644 (file)
--- 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)
+