Cope with NULL functions.

diff --git a/opengl32.py b/opengl32.py
index e12ec872856a93f70832ad35b601021a53176f7b..53f4d8dc002b27e0bb54bdd140c284c1124db0df 100644 (file)
--- a/opengl32.py
+++ b/opengl32.py
@@ -580,13 +580,15 @@ class WglGetProcAddressFunction(DllFunction):
         DllFunction.wrap_impl(self)
 
     def post_call_impl(self):
+        print '    if(result) {'
         for function in self.functions:
             ptype = function.pointer_type()
             pvalue = function.pointer_value()
-            print '    if(!strcmp("%s", lpszProc)) {' % function.name
-            print '        %s = (%s)result;' % (pvalue, ptype)
-            print '        result = (PROC)&%s;' % function.name;
-            print '    }'
+            print '        if(!strcmp("%s", lpszProc)) {' % function.name
+            print '            %s = (%s)result;' % (pvalue, ptype)
+            print '            result = (PROC)&%s;' % function.name;
+            print '        }'
+        print '    }'
 
 
 wglgetprocaddress = WglGetProcAddressFunction(PROC, "wglGetProcAddress", [(LPCSTR, "lpszProc")])