Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 1 | /******************************************************************** |
| 2 | |
Guido van Rossum | eaf9b6c | 1996-08-22 00:06:59 +0000 | [diff] [blame] | 3 | import_nt.c |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 4 | |
| 5 | Win32 specific import code. |
| 6 | |
| 7 | */ |
| 8 | |
Guido van Rossum | 7688bba | 1997-05-05 21:45:44 +0000 | [diff] [blame] | 9 | #include "Python.h" |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 10 | #include "osdefs.h" |
| 11 | #include <windows.h> |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 12 | #include "importdl.h" |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 13 | #include "malloc.h" // for alloca |
| 14 | |
| 15 | extern const char *PyWin_DLLVersionString; // a string loaded from the DLL at startup. |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 16 | |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 17 | FILE *PyWin_FindRegisteredModule( const char *moduleName, struct filedescr **ppFileDesc, char *pathBuf, int pathLen) |
| 18 | { |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 19 | char *moduleKey; |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 20 | const char keyPrefix[] = "Software\\Python\\PythonCore\\"; |
| 21 | const char keySuffix[] = "\\Modules\\"; |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 22 | #ifdef _DEBUG |
| 23 | // In debugging builds, we _must_ have the debug version registered. |
| 24 | const char debugString[] = "\\Debug"; |
| 25 | #else |
| 26 | const char debugString[] = ""; |
| 27 | #endif |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 28 | struct filedescr *fdp = NULL; |
| 29 | FILE *fp; |
Guido van Rossum | 4f3cc35 | 2000-04-04 22:48:55 +0000 | [diff] [blame] | 30 | HKEY keyBase = HKEY_LOCAL_MACHINE; |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 31 | int modNameSize; |
Guido van Rossum | 76ec53c | 1998-06-15 18:01:34 +0000 | [diff] [blame] | 32 | long regStat; |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 33 | |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 34 | // Calculate the size for the sprintf buffer. |
| 35 | // Get the size of the chars only, plus 1 NULL. |
Guido van Rossum | 1c44e28 | 2000-06-28 22:20:06 +0000 | [diff] [blame^] | 36 | size_t bufSize = sizeof(keyPrefix)-1 + strlen(PyWin_DLLVersionString) + sizeof(keySuffix) + strlen(moduleName) + sizeof(debugString) - 1; |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 37 | // alloca == no free required, but memory only local to fn, also no heap fragmentation! |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 38 | moduleKey = alloca(bufSize); |
| 39 | sprintf(moduleKey, "Software\\Python\\PythonCore\\%s\\Modules\\%s%s", PyWin_DLLVersionString, moduleName, debugString); |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 40 | |
Guido van Rossum | 9455547 | 1999-01-14 22:40:30 +0000 | [diff] [blame] | 41 | modNameSize = pathLen; |
Guido van Rossum | 76ec53c | 1998-06-15 18:01:34 +0000 | [diff] [blame] | 42 | regStat = RegQueryValue(keyBase, moduleKey, pathBuf, &modNameSize); |
| 43 | if (regStat!=ERROR_SUCCESS) |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 44 | return NULL; |
| 45 | // use the file extension to locate the type entry. |
Guido van Rossum | 22822d7 | 1997-05-08 23:43:52 +0000 | [diff] [blame] | 46 | for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) { |
Guido van Rossum | 1c44e28 | 2000-06-28 22:20:06 +0000 | [diff] [blame^] | 47 | size_t extLen = strlen(fdp->suffix); |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 48 | if (modNameSize>extLen && strnicmp(pathBuf+(modNameSize-extLen-1),fdp->suffix,extLen)==0) |
| 49 | break; |
| 50 | } |
| 51 | if (fdp->suffix==NULL) |
| 52 | return NULL; |
| 53 | fp = fopen(pathBuf, fdp->mode); |
| 54 | if (fp != NULL) |
| 55 | *ppFileDesc = fdp; |
| 56 | return fp; |
| 57 | } |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 58 | |