X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=wrappers%2Fdlltrace.py;h=cea7dc7464c4717efe44828a917f65a6a4c2d3d9;hb=81301939f025407ceb284a9dcd5d5a1f05d27b8f;hp=8456d5f7a63a4052129e69aef305582ceeb240ad;hpb=2b807458f983e01e6c5c49a6c77ee28a145a1b2d;p=apitrace

diff --git a/wrappers/dlltrace.py b/wrappers/dlltrace.py
index 8456d5f..cea7dc7 100644
--- a/wrappers/dlltrace.py
+++ b/wrappers/dlltrace.py
@@ -28,6 +28,7 @@
 
 from trace import Tracer
 from dispatch import Dispatcher
+from specs.stdapi import API
 
 
 class DllTracer(Tracer):
@@ -62,8 +63,13 @@ _getPublicProcAddress(LPCSTR lpProcName)
 
 ''' % self.dllname
 
-        dispatcher = Dispatcher()
-        dispatcher.dispatchApi(api)
+        for module in api.modules:
+            dispatcher = Dispatcher()
+            dispatcher.dispatchModule(module)
 
         Tracer.header(self, api)
 
+    def traceModule(self, module):
+        api = API()
+        api.addModule(module)
+        self.traceApi(api)