import_nt.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 */

/* a string loaded from the DLL at startup */
externconstchar*PyWin_DLLVersionString;

FILE*PyWin_FindRegisteredModule(constchar*moduleName,
structfiledescr**ppFileDesc,
char*pathBuf,
Py_ssize_tpathLen)
{
char*moduleKey;
constcharkeyPrefix[] ="Software\\Python\\PythonCore\\";
constcharkeySuffix[] ="\\Modules\\";
#ifdef_DEBUG
/* In debugging builds, we _must_ have the debug version
 * registered.
 */
constchardebugString[] ="\\Debug";
#else
constchardebugString[] ="";
#endif
structfiledescr*fdp=NULL;
FILE*fp;
HKEYkeyBase=HKEY_CURRENT_USER;
intmodNameSize;
longregStat;

/* Calculate the size for the sprintf buffer.
 * Get the size of the chars only, plus 1 NULL.
 */
size_tbufSize=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);
PyOS_snprintf(moduleKey, bufSize,
"Software\\Python\\PythonCore\\%s\\Modules\\%s%s",
PyWin_DLLVersionString, moduleName, debugString);

assert(pathLen<INT_MAX);
modNameSize= (int)pathLen;
regStat=RegQueryValue(keyBase, moduleKey, pathBuf, &modNameSize);
if (regStat!=ERROR_SUCCESS) {
/* No user setting - lookup in machine settings */
keyBase=HKEY_LOCAL_MACHINE;
/* be anal - failure may have reset size param */
modNameSize= (int)pathLen;
regStat=RegQueryValue(keyBase, moduleKey,
pathBuf, &modNameSize);

if (regStat!=ERROR_SUCCESS)
returnNULL;
 }
/* use the file extension to locate the type entry. */
for (fdp=_PyImport_Filetab; fdp->suffix!=NULL; fdp++) {
size_textLen=strlen(fdp->suffix);
assert(modNameSize >= 0); /* else cast to size_t is wrong */
if ((size_t)modNameSize>extLen&&
strnicmp(pathBuf+ ((size_t)modNameSize-extLen-1),
fdp->suffix,
extLen) ==0)
break;
 }
if (fdp->suffix==NULL)
returnNULL;
fp=fopen(pathBuf, fdp->mode);
if (fp!=NULL)
*ppFileDesc=fdp;
returnfp;
}