Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 1 | /******************************************************************** |
| 2 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +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" |
Tim Peters | db63829 | 2000-07-03 23:51:17 +0000 | [diff] [blame] | 13 | #include "malloc.h" /* for alloca */ |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 14 | |
Tim Peters | db63829 | 2000-07-03 23:51:17 +0000 | [diff] [blame] | 15 | /* a string loaded from the DLL at startup */ |
| 16 | extern const char *PyWin_DLLVersionString; |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 17 | |
Victor Stinner | 9b99b44 | 2011-02-22 23:12:28 +0000 | [diff] [blame] | 18 | FILE *_PyWin_FindRegisteredModule(const char *moduleName, |
| 19 | struct filedescr **ppFileDesc, |
| 20 | char *pathBuf, |
| 21 | Py_ssize_t pathLen) |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 22 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 23 | char *moduleKey; |
| 24 | const char keyPrefix[] = "Software\\Python\\PythonCore\\"; |
| 25 | const char keySuffix[] = "\\Modules\\"; |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 26 | #ifdef _DEBUG |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 27 | /* In debugging builds, we _must_ have the debug version |
| 28 | * registered. |
| 29 | */ |
| 30 | const char debugString[] = "\\Debug"; |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 31 | #else |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 32 | const char debugString[] = ""; |
Guido van Rossum | 1c096b7 | 1998-05-26 13:53:23 +0000 | [diff] [blame] | 33 | #endif |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 34 | struct filedescr *fdp = NULL; |
| 35 | FILE *fp; |
| 36 | HKEY keyBase = HKEY_CURRENT_USER; |
| 37 | int modNameSize; |
| 38 | long regStat; |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 39 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 40 | /* Calculate the size for the sprintf buffer. |
| 41 | * Get the size of the chars only, plus 1 NULL. |
| 42 | */ |
| 43 | size_t bufSize = sizeof(keyPrefix)-1 + |
| 44 | strlen(PyWin_DLLVersionString) + |
| 45 | sizeof(keySuffix) + |
| 46 | strlen(moduleName) + |
| 47 | sizeof(debugString) - 1; |
| 48 | /* alloca == no free required, but memory only local to fn, |
| 49 | * also no heap fragmentation! |
| 50 | */ |
| 51 | moduleKey = alloca(bufSize); |
| 52 | PyOS_snprintf(moduleKey, bufSize, |
| 53 | "Software\\Python\\PythonCore\\%s\\Modules\\%s%s", |
| 54 | PyWin_DLLVersionString, moduleName, debugString); |
Guido van Rossum | 271f977 | 1997-09-29 23:39:31 +0000 | [diff] [blame] | 55 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 56 | assert(pathLen < INT_MAX); |
| 57 | modNameSize = (int)pathLen; |
| 58 | regStat = RegQueryValue(keyBase, moduleKey, pathBuf, &modNameSize); |
| 59 | if (regStat != ERROR_SUCCESS) { |
| 60 | /* No user setting - lookup in machine settings */ |
| 61 | keyBase = HKEY_LOCAL_MACHINE; |
| 62 | /* be anal - failure may have reset size param */ |
| 63 | modNameSize = (int)pathLen; |
| 64 | regStat = RegQueryValue(keyBase, moduleKey, |
| 65 | pathBuf, &modNameSize); |
Mark Hammond | 48b3eee | 2000-08-22 11:20:21 +0000 | [diff] [blame] | 66 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 67 | if (regStat != ERROR_SUCCESS) |
| 68 | return NULL; |
| 69 | } |
| 70 | /* use the file extension to locate the type entry. */ |
| 71 | for (fdp = _PyImport_Filetab; fdp->suffix != NULL; fdp++) { |
| 72 | size_t extLen = strlen(fdp->suffix); |
| 73 | assert(modNameSize >= 0); /* else cast to size_t is wrong */ |
| 74 | if ((size_t)modNameSize > extLen && |
| 75 | strnicmp(pathBuf + ((size_t)modNameSize-extLen-1), |
| 76 | fdp->suffix, |
| 77 | extLen) == 0) |
| 78 | break; |
| 79 | } |
| 80 | if (fdp->suffix == NULL) |
| 81 | return NULL; |
| 82 | fp = fopen(pathBuf, fdp->mode); |
| 83 | if (fp != NULL) |
| 84 | *ppFileDesc = fdp; |
| 85 | return fp; |
Guido van Rossum | 6dbd190 | 1996-08-21 15:03:37 +0000 | [diff] [blame] | 86 | } |