Mark Hammond's main program for a frozen application.

1 files changed, 134 insertions, 0 deletions

diff --git a/PC/frozen_dllmain.c b/PC/frozen_dllmain.c
new file mode 100644
index 0000000..6ca4cd4
--- /dev/null
+++ b/PC/frozen_dllmain.c
@@ -0,0 +1,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);
+}
+