|
1 /* FreezeDLLMain.cpp |
|
2 |
|
3 This is a DLLMain suitable for frozen applications/DLLs on |
|
4 a Windows platform. |
|
5 |
|
6 The general problem is that many Python extension modules may define |
|
7 DLL main functions, but when statically linked together to form |
|
8 a frozen application, this DLLMain symbol exists multiple times. |
|
9 |
|
10 The solution is: |
|
11 * Each module checks for a frozen build, and if so, defines its DLLMain |
|
12 function as "__declspec(dllexport) DllMain%module%" |
|
13 (eg, DllMainpythoncom, or DllMainpywintypes) |
|
14 |
|
15 * The frozen .EXE/.DLL links against this module, which provides |
|
16 the single DllMain. |
|
17 |
|
18 * This DllMain attempts to locate and call the DllMain for each |
|
19 of the extension modules. |
|
20 |
|
21 * This code also has hooks to "simulate" DllMain when used from |
|
22 a frozen .EXE. |
|
23 |
|
24 At this stage, there is a static table of "possibly embedded modules". |
|
25 This should change to something better, but it will work OK for now. |
|
26 |
|
27 Note that this scheme does not handle dependencies in the order |
|
28 of DllMain calls - except it does call pywintypes first :-) |
|
29 |
|
30 As an example of how an extension module with a DllMain should be |
|
31 changed, here is a snippet from the pythoncom extension module. |
|
32 |
|
33 // end of example code from pythoncom's DllMain.cpp |
|
34 #ifndef BUILD_FREEZE |
|
35 #define DLLMAIN DllMain |
|
36 #define DLLMAIN_DECL |
|
37 #else |
|
38 #define DLLMAIN DllMainpythoncom |
|
39 #define DLLMAIN_DECL __declspec(dllexport) |
|
40 #endif |
|
41 |
|
42 extern "C" DLLMAIN_DECL |
|
43 BOOL WINAPI DLLMAIN(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved) |
|
44 // end of example code from pythoncom's DllMain.cpp |
|
45 |
|
46 ***************************************************************************/ |
|
47 #include "windows.h" |
|
48 |
|
49 static char *possibleModules[] = { |
|
50 "pywintypes", |
|
51 "pythoncom", |
|
52 "win32ui", |
|
53 NULL, |
|
54 }; |
|
55 |
|
56 BOOL CallModuleDllMain(char *modName, DWORD dwReason); |
|
57 |
|
58 |
|
59 /* |
|
60 Called by a frozen .EXE only, so that built-in extension |
|
61 modules are initialized correctly |
|
62 */ |
|
63 void PyWinFreeze_ExeInit(void) |
|
64 { |
|
65 char **modName; |
|
66 for (modName = possibleModules;*modName;*modName++) { |
|
67 /* printf("Initialising '%s'\n", *modName); */ |
|
68 CallModuleDllMain(*modName, DLL_PROCESS_ATTACH); |
|
69 } |
|
70 } |
|
71 |
|
72 /* |
|
73 Called by a frozen .EXE only, so that built-in extension |
|
74 modules are cleaned up |
|
75 */ |
|
76 void PyWinFreeze_ExeTerm(void) |
|
77 { |
|
78 // Must go backwards |
|
79 char **modName; |
|
80 for (modName = possibleModules+(sizeof(possibleModules) / sizeof(char *))-2; |
|
81 modName >= possibleModules; |
|
82 *modName--) { |
|
83 /* printf("Terminating '%s'\n", *modName);*/ |
|
84 CallModuleDllMain(*modName, DLL_PROCESS_DETACH); |
|
85 } |
|
86 } |
|
87 |
|
88 BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved) |
|
89 { |
|
90 BOOL ret = TRUE; |
|
91 switch (dwReason) { |
|
92 case DLL_PROCESS_ATTACH: |
|
93 { |
|
94 char **modName; |
|
95 for (modName = possibleModules;*modName;*modName++) { |
|
96 BOOL ok = CallModuleDllMain(*modName, dwReason); |
|
97 if (!ok) |
|
98 ret = FALSE; |
|
99 } |
|
100 break; |
|
101 } |
|
102 case DLL_PROCESS_DETACH: |
|
103 { |
|
104 // Must go backwards |
|
105 char **modName; |
|
106 for (modName = possibleModules+(sizeof(possibleModules) / sizeof(char *))-2; |
|
107 modName >= possibleModules; |
|
108 *modName--) |
|
109 CallModuleDllMain(*modName, DLL_PROCESS_DETACH); |
|
110 break; |
|
111 } |
|
112 } |
|
113 return ret; |
|
114 } |
|
115 |
|
116 BOOL CallModuleDllMain(char *modName, DWORD dwReason) |
|
117 { |
|
118 BOOL (WINAPI * pfndllmain)(HINSTANCE, DWORD, LPVOID); |
|
119 |
|
120 char funcName[255]; |
|
121 HMODULE hmod = GetModuleHandle(NULL); |
|
122 strcpy(funcName, "_DllMain"); |
|
123 strcat(funcName, modName); |
|
124 strcat(funcName, "@12"); // stdcall convention. |
|
125 pfndllmain = (BOOL (WINAPI *)(HINSTANCE, DWORD, LPVOID))GetProcAddress(hmod, funcName); |
|
126 if (pfndllmain==NULL) { |
|
127 /* No function by that name exported - then that module does |
|
128 not appear in our frozen program - return OK |
|
129 */ |
|
130 return TRUE; |
|
131 } |
|
132 return (*pfndllmain)(hmod, dwReason, NULL); |
|
133 } |
|
134 |