summaryrefslogtreecommitdiffstats
path: root/PC/frozen_dllmain.c
blob: 6ca4cd47acaa224e0eb59c1585ef5719f07bc39f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
/* FreezeDLLMain.cpp

This is a DLLMain suitable for frozen applications/DLLs on
a Windows platform.

The general problem is that many Python extension modules may define
DLL main functions, but when statically linked together to form
a frozen application, this DLLMain symbol exists multiple times.

The solution is:
* Each module checks for a frozen build, and if so, defines its DLLMain
  function as "__declspec(dllexport) DllMain%module%" 
  (eg, DllMainpythoncom, or DllMainpywintypes)

* The frozen .EXE/.DLL links against this module, which provides
  the single DllMain.

* This DllMain attempts to locate and call the DllMain for each
  of the extension modules.

* This code also has hooks to "simulate" DllMain when used from
  a frozen .EXE.

At this stage, there is a static table of "possibly embedded modules".
This should change to something better, but it will work OK for now.

Note that this scheme does not handle dependencies in the order
of DllMain calls - except it does call pywintypes first :-)

As an example of how an extension module with a DllMain should be
changed, here is a snippet from the pythoncom extension module.

  // end of example code from pythoncom's DllMain.cpp
  #ifndef BUILD_FREEZE
  #define DLLMAIN DllMain
  #define DLLMAIN_DECL
  #else
  #define DLLMAIN DllMainpythoncom
  #define DLLMAIN_DECL __declspec(dllexport)
  #endif

  extern "C" DLLMAIN_DECL
  BOOL WINAPI DLLMAIN(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
  // end of example code from pythoncom's DllMain.cpp

***************************************************************************/
#include "windows.h"

static char *possibleModules[] = {
	"pywintypes",
	"pythoncom",
	"win32ui",
	NULL,
};

BOOL CallModuleDllMain(char *modName, DWORD dwReason);


/*
  Called by a frozen .EXE only, so that built-in extension
  modules are initialized correctly
*/
void PyWinFreeze_ExeInit()
{
	char **modName;
	for (modName = possibleModules;*modName;*modName++) {
		printf("Initialising '%s'\n", *modName);
		CallModuleDllMain(*modName, DLL_PROCESS_ATTACH);
	}
}

/*
  Called by a frozen .EXE only, so that built-in extension
  modules are cleaned up 
*/
void PyWinFreeze_ExeTerm()
{
	// Must go backwards
	char **modName;
	for (modName = possibleModules+(sizeof(possibleModules) / sizeof(char *))-2;
	     modName >= possibleModules;
	     *modName--) {
		printf("Terminating '%s'\n", *modName);
		CallModuleDllMain(*modName, DLL_PROCESS_DETACH);
	}
}

BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
{
	BOOL ret = TRUE;
	switch (dwReason) {
		case DLL_PROCESS_ATTACH: 
		{
			char **modName;
			for (modName = possibleModules;*modName;*modName++) {
				BOOL ok = CallModuleDllMain(*modName, dwReason);
				if (!ok)
					ret = FALSE;
			}
			break;
		}
		case DLL_PROCESS_DETACH: 
		{
			// Must go backwards
			char **modName;
			for (modName = possibleModules+(sizeof(possibleModules) / sizeof(char *))-2;
			     modName >= possibleModules;
			     *modName--)
				CallModuleDllMain(*modName, DLL_PROCESS_DETACH);
			break;
		}
	}
	return ret;
}

BOOL CallModuleDllMain(char *modName, DWORD dwReason)
{
	BOOL (WINAPI * pfndllmain)(HINSTANCE, DWORD, LPVOID);

	char funcName[255];
	HMODULE hmod = GetModuleHandle(NULL);
	strcpy(funcName, "_DllMain");
	strcat(funcName, modName);
	strcat(funcName, "@12"); // stdcall convention.
	pfndllmain = (BOOL (WINAPI *)(HINSTANCE, DWORD, LPVOID))GetProcAddress(hmod, funcName);
	if (pfndllmain==NULL) {
		/* No function by that name exported - then that module does
 		   not appear in our frozen program - return OK
                */
		return TRUE;
	}
	return (*pfndllmain)(hmod, dwReason, NULL);
}