diff Python/dynload_win.c @ 32469:b8710e6d8d78 legacy-trunk

When importing an extension on Windows, the code reads the PE 'import table' of the dll, to make sure that the dll really was build for the correct Python version. It does this by looking for an entry 'pythonXY.dll' (X.Y is the Python version number). The code now checks the size of the dll's import table before reading entries from it. Before this patch, the code crashed trying to read the import table when the size was zero (as in Win2k's wmi.dll, for example). Look for imports of 'pythonXY_d.dll' in a debug build instead of 'pythonXY.dll'. Fixes SF 951851: Crash when reading "import table" of certain windows dlls. Already backported to the 2.3 branch.
author Thomas Heller <theller@ctypes.org>
date Fri, 02 Jul 2004 08:53:57 +0000
parents ed0b02672f09
children c1cef1d2dccf
line wrap: on
line diff
--- a/Python/dynload_win.c	Fri Jul 02 08:02:40 2004 +0000
+++ b/Python/dynload_win.c	Fri Jul 02 08:53:57 2004 +0000
@@ -116,6 +116,10 @@
 	   string constant holding the import name is located. */
 
 	if (DWORD_AT(dllbase + opt_offset + num_dict_off) >= 2) {
+		/* We have at least 2 tables - the import table is the second
+		   one.  But still it may be that the table size is zero */
+		if (0 == DWORD_AT(dllbase + opt_offset + import_off + sizeof(DWORD)))
+			return NULL;
 		import_data = dllbase + DWORD_AT(dllbase +
 						 opt_offset +
 						 import_off);
@@ -128,7 +132,11 @@
 				/* Ensure python prefix is followed only
 				   by numbers to the end of the basename */
 				pch = import_name + 6;
+#ifdef _DEBUG
+				while (*pch && pch[0] != '_' && pch[1] != 'd' && pch[2] != '.') {
+#else
 				while (*pch && *pch != '.') {
+#endif
 					if (*pch >= '0' && *pch <= '9') {
 						pch++;
 					} else {
@@ -221,7 +229,11 @@
 		} else {
 			char buffer[256];
 
+#ifdef _DEBUG
+			PyOS_snprintf(buffer, sizeof(buffer), "python%d%d_d.dll",
+#else
 			PyOS_snprintf(buffer, sizeof(buffer), "python%d%d.dll",
+#endif
 				      PY_MAJOR_VERSION,PY_MINOR_VERSION);
 			import_python = GetPythonImport(hDLL);