mirror of
				https://github.com/python/cpython.git
				synced 2025-10-30 01:47:38 +00:00 
			
		
		
		
	 76ec53c64d
			
		
	
	
		76ec53c64d
		
	
	
	
	
		
			
			From: Dan Pierson <dan@remote.control.com>
To: "Mark Hammond (E-mail)" <MHammond@skippinet.com.au>,
    "Guido van Rossum (E-mail)" <guido@cnri.reston.va.us>
Date: Mon, 8 Jun 1998 17:25:07 -0400
RegistryQueryValue requires that its fourth argument be initialized to
the length of the buffer being passed in, this wasn't being done.  I
also split the call and status test into two lines with a local variable
so that I could look at the status in the debugger.
		
	
			
		
			
				
	
	
		
			74 lines
		
	
	
	
		
			2.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			74 lines
		
	
	
	
		
			2.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /********************************************************************
 | |
| 
 | |
|  import_nt.c 
 | |
| 
 | |
|   Win32 specific import code.
 | |
| 
 | |
| */
 | |
| 
 | |
| #include "Python.h"
 | |
| #include "osdefs.h"
 | |
| #include <windows.h>
 | |
| #include "importdl.h"
 | |
| #include "malloc.h" // for alloca
 | |
| 
 | |
| extern const char *PyWin_DLLVersionString; // a string loaded from the DLL at startup.
 | |
| 
 | |
| /* Return whether this is Win32s, i.e., Win32 API on Win 3.1(1).
 | |
|    This function is exported! */
 | |
| 
 | |
| BOOL PyWin_IsWin32s()
 | |
| {
 | |
| 	static BOOL bIsWin32s = -1; /* flag as "not yet looked" */
 | |
| 
 | |
| 	if (bIsWin32s == -1) {
 | |
| 		OSVERSIONINFO ver;
 | |
| 		ver.dwOSVersionInfoSize = sizeof(ver);
 | |
| 		GetVersionEx(&ver);
 | |
| 		bIsWin32s = ver.dwPlatformId == VER_PLATFORM_WIN32s;
 | |
| 	}
 | |
| 	return bIsWin32s;
 | |
| }
 | |
| 
 | |
| FILE *PyWin_FindRegisteredModule( const char *moduleName, struct filedescr **ppFileDesc, char *pathBuf, int pathLen)
 | |
| {
 | |
| 	char *moduleKey;
 | |
| 	const char keyPrefix[] = "Software\\Python\\PythonCore\\";
 | |
| 	const char keySuffix[] = "\\Modules\\";
 | |
| #ifdef _DEBUG
 | |
| 	// In debugging builds, we _must_ have the debug version registered.
 | |
| 	const char debugString[] = "\\Debug";
 | |
| #else
 | |
| 	const char debugString[] = "";
 | |
| #endif
 | |
| 	struct filedescr *fdp = NULL;
 | |
| 	FILE *fp;
 | |
| 	HKEY keyBase = PyWin_IsWin32s() ? HKEY_CLASSES_ROOT : HKEY_LOCAL_MACHINE;
 | |
| 	int modNameSize;
 | |
| 	long regStat;
 | |
| 
 | |
| 	// Calculate the size for the sprintf buffer.
 | |
| 	// Get the size of the chars only, plus 1 NULL.
 | |
| 	int bufSize = sizeof(keyPrefix)-1 + strlen(PyWin_DLLVersionString) + sizeof(keySuffix) + strlen(moduleName) + sizeof(debugString) - 1;
 | |
| 	// alloca == no free required, but memory only local to fn, also no heap fragmentation!
 | |
| 	moduleKey = alloca(bufSize); 
 | |
| 	sprintf(moduleKey, "Software\\Python\\PythonCore\\%s\\Modules\\%s%s", PyWin_DLLVersionString, moduleName, debugString);
 | |
| 
 | |
| 	modNameSize = bufSize;
 | |
| 	regStat = RegQueryValue(keyBase, moduleKey, pathBuf, &modNameSize);
 | |
| 	if (regStat!=ERROR_SUCCESS)
 | |
| 		return NULL;
 | |
| 	// use the file extension to locate the type entry.
 | |
| 	for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) {
 | |
| 		int extLen=strlen(fdp->suffix);
 | |
| 		if (modNameSize>extLen && strnicmp(pathBuf+(modNameSize-extLen-1),fdp->suffix,extLen)==0)
 | |
| 			break;
 | |
| 	}
 | |
| 	if (fdp->suffix==NULL)
 | |
| 		return NULL;
 | |
| 	fp = fopen(pathBuf, fdp->mode);
 | |
| 	if (fp != NULL)
 | |
| 		*ppFileDesc = fdp;
 | |
| 	return fp;
 | |
| }
 | |
| 
 |