2 /* Return the initial module search path. */
3 /* Used by DOS, OS/2, Windows 3.1, Windows 95/98, Windows NT. */
5 /* ----------------------------------------------------------------
6 PATH RULES FOR WINDOWS:
7 This describes how sys.path is formed on Windows. It describes the
8 functionality, not the implementation (ie, the order in which these
9 are actually fetched is different)
11 * Python always adds an empty entry at the start, which corresponds
12 to the current directory.
14 * If the PYTHONPATH env. var. exists, it's entries are added next.
16 * We look in the registry for "application paths" - that is, sub-keys
17 under the main PythonPath registry key. These are added next (the
18 order of sub-key processing is undefined).
19 HKEY_CURRENT_USER is searched and added first.
20 HKEY_LOCAL_MACHINE is searched and added next.
21 (Note that all known installers only use HKLM, so HKCU is typically
24 * We attempt to locate the "Python Home" - if the PYTHONHOME env var
25 is set, we believe it. Otherwise, we use the path of our host .EXE's
26 to try and locate our "landmark" (lib\\os.py) and deduce our home.
27 - If we DO have a Python Home: The relevant sub-directories (Lib,
28 plat-win, lib-tk, etc) are based on the Python Home
29 - If we DO NOT have a Python Home, the core Python Path is
30 loaded from the registry. This is the main PythonPath key,
31 and both HKLM and HKCU are combined to form the path)
33 * Iff - we can not locate the Python Home, have not had a PYTHONPATH
34 specified, and can't locate any Registry entries (ie, we have _nothing_
35 we can assume is a good path), a default path with relative entries is
36 used (eg. .\Lib;.\plat-win, etc)
39 The end result of all this is:
40 * When running python.exe, or any other .exe in the main Python directory
41 (either an installed version, or directly from the PCbuild directory),
42 the core path is deduced, and the core paths in the registry are
43 ignored. Other "application paths" in the registry are always read.
45 * When Python is hosted in another exe (different directory, embedded via
46 COM, etc), the Python Home will not be deduced, so the core path from
47 the registry is used. Other "application paths" in the registry are
50 * If Python can't find its home and there is no registry (eg, frozen
51 exe, some very strange installation setup) you get a path with
52 some default, but relative, paths.
54 ---------------------------------------------------------------- */
65 #include <sys/types.h>
69 /* Search in some common locations for the associated Python libraries.
71 * Py_GetPath() tries to return a sensible Python module search path.
73 * The approach is an adaptation for Windows of the strategy used in
74 * ../Modules/getpath.c; it uses the Windows Registry as one of its
75 * information sources.
79 #define LANDMARK "lib\\os.py"
82 static char prefix
[MAXPATHLEN
+1];
83 static char progpath
[MAXPATHLEN
+1];
84 static char dllpath
[MAXPATHLEN
+1];
85 static char *module_search_path
= NULL
;
89 is_sep(char ch
) /* determine if "ch" is a separator character */
92 return ch
== SEP
|| ch
== ALTSEP
;
98 /* assumes 'dir' null terminated in bounds. Never writes
99 beyond existing terminator.
104 size_t i
= strlen(dir
);
105 while (i
> 0 && !is_sep(dir
[i
]))
112 exists(char *filename
)
115 return stat(filename
, &buf
) == 0;
118 /* Assumes 'filename' MAXPATHLEN+1 bytes long -
119 may extend 'filename' by one character.
122 ismodule(char *filename
) /* Is module -- check for .pyc/.pyo too */
124 if (exists(filename
))
127 /* Check for the compiled version of prefix. */
128 if (strlen(filename
) < MAXPATHLEN
) {
129 strcat(filename
, Py_OptimizeFlag
? "o" : "c");
130 if (exists(filename
))
136 /* guarantees buffer will never overflow MAXPATHLEN+1 bytes */
138 join(char *buffer
, char *stuff
)
141 if (is_sep(stuff
[0]))
145 if (n
> 0 && !is_sep(buffer
[n
-1]) && n
< MAXPATHLEN
)
149 if (n
+ k
> MAXPATHLEN
)
151 strncpy(buffer
+n
, stuff
, k
);
155 /* gotlandmark only called by search_for_prefix, which ensures
156 'prefix' is null terminated in bounds. join() ensures
157 'landmark' can not overflow prefix if too long.
160 gotlandmark(char *landmark
)
165 join(prefix
, landmark
);
166 ok
= ismodule(prefix
);
171 /* assumes argv0_path is MAXPATHLEN+1 bytes long, already \0 term'd.
172 assumption provided by only caller, calculate_path() */
174 search_for_prefix(char *argv0_path
, char *landmark
)
176 /* Search from argv0_path, until landmark is found */
177 strcpy(prefix
, argv0_path
);
179 if (gotlandmark(landmark
))
188 /* a string loaded from the DLL at startup.*/
189 extern const char *PyWin_DLLVersionString
;
192 /* Load a PYTHONPATH value from the registry.
193 Load from either HKEY_LOCAL_MACHINE or HKEY_CURRENT_USER.
195 Works in both Unicode and 8bit environments. Only uses the
196 Ex family of functions so it also works with Windows CE.
198 Returns NULL, or a pointer that should be freed.
200 XXX - this code is pretty strange, as it used to also
201 work on Win16, where the buffer sizes werent available
202 in advance. It could be simplied now Win16/Win32s is dead!
206 getpythonregpath(HKEY keyBase
, int skipcore
)
213 TCHAR
*dataBuf
= NULL
;
214 static const TCHAR keyPrefix
[] = _T("Software\\Python\\PythonCore\\");
215 static const TCHAR keySuffix
[] = _T("\\PythonPath");
218 TCHAR
*keyBuf
= NULL
;
220 TCHAR
**ppPaths
= NULL
;
222 /* Tried to use sysget("winver") but here is too early :-( */
223 versionLen
= _tcslen(PyWin_DLLVersionString
);
224 /* Space for all the chars, plus one \0 */
225 keyBuf
= keyBufPtr
= malloc(sizeof(keyPrefix
) +
226 sizeof(TCHAR
)*(versionLen
-1) +
228 if (keyBuf
==NULL
) goto done
;
230 memcpy(keyBufPtr
, keyPrefix
, sizeof(keyPrefix
)-sizeof(TCHAR
));
231 keyBufPtr
+= sizeof(keyPrefix
)/sizeof(TCHAR
) - 1;
232 memcpy(keyBufPtr
, PyWin_DLLVersionString
, versionLen
* sizeof(TCHAR
));
233 keyBufPtr
+= versionLen
;
234 /* NULL comes with this one! */
235 memcpy(keyBufPtr
, keySuffix
, sizeof(keySuffix
));
236 /* Open the root Python key */
237 rc
=RegOpenKeyEx(keyBase
,
242 if (rc
!=ERROR_SUCCESS
) goto done
;
243 /* Find out how big our core buffer is, and how many subkeys we have */
244 rc
= RegQueryInfoKey(newKey
, NULL
, NULL
, NULL
, &numKeys
, NULL
, NULL
,
245 NULL
, NULL
, &dataSize
, NULL
, NULL
);
246 if (rc
!=ERROR_SUCCESS
) goto done
;
247 if (skipcore
) dataSize
= 0; /* Only count core ones if we want them! */
248 /* Allocate a temp array of char buffers, so we only need to loop
249 reading the registry once
251 ppPaths
= malloc( sizeof(TCHAR
*) * numKeys
);
252 if (ppPaths
==NULL
) goto done
;
253 memset(ppPaths
, 0, sizeof(TCHAR
*) * numKeys
);
254 /* Loop over all subkeys, allocating a temp sub-buffer. */
255 for(index
=0;index
<numKeys
;index
++) {
256 TCHAR keyBuf
[MAX_PATH
+1];
258 DWORD reqdSize
= MAX_PATH
+1;
259 /* Get the sub-key name */
260 DWORD rc
= RegEnumKeyEx(newKey
, index
, keyBuf
, &reqdSize
,
261 NULL
, NULL
, NULL
, NULL
);
262 if (rc
!=ERROR_SUCCESS
) goto done
;
263 /* Open the sub-key */
264 rc
=RegOpenKeyEx(newKey
,
269 if (rc
!=ERROR_SUCCESS
) goto done
;
270 /* Find the value of the buffer size, malloc, then read it */
271 RegQueryValueEx(subKey
, NULL
, 0, NULL
, NULL
, &reqdSize
);
273 ppPaths
[index
] = malloc(reqdSize
);
274 if (ppPaths
[index
]) {
275 RegQueryValueEx(subKey
, NULL
, 0, NULL
,
276 (LPBYTE
)ppPaths
[index
],
278 dataSize
+= reqdSize
+ 1; /* 1 for the ";" */
283 /* original datasize from RegQueryInfo doesn't include the \0 */
284 dataBuf
= malloc((dataSize
+1) * sizeof(TCHAR
));
286 TCHAR
*szCur
= dataBuf
;
287 DWORD reqdSize
= dataSize
;
288 /* Copy our collected strings */
289 for (index
=0;index
<numKeys
;index
++) {
291 *(szCur
++) = _T(';');
294 if (ppPaths
[index
]) {
295 int len
= _tcslen(ppPaths
[index
]);
296 _tcsncpy(szCur
, ppPaths
[index
], len
);
304 /* If we have no values, we dont need a ';' */
306 *(szCur
++) = _T(';');
309 /* Now append the core path entries -
310 this will include the NULL
312 rc
= RegQueryValueEx(newKey
, NULL
, 0, NULL
,
313 (LPBYTE
)szCur
, &dataSize
);
315 /* And set the result - caller must free
316 If MBCS, it is fine as is. If Unicode, allocate new
320 retval
= (char *)malloc(reqdSize
+1);
322 WideCharToMultiByte(CP_ACP
, 0,
323 dataBuf
, -1, /* source */
324 retval
, reqdSize
+1, /* dest */
332 /* Loop freeing my temp buffers */
334 for(index
=0;index
<numKeys
;index
++)
335 if (ppPaths
[index
]) free(ppPaths
[index
]);
344 #endif /* MS_WINDOWS */
349 extern char *Py_GetProgramName(void);
350 char *path
= getenv("PATH");
351 char *prog
= Py_GetProgramName();
354 extern HANDLE PyWin_DLLhModule
;
356 WCHAR wprogpath
[MAXPATHLEN
+1];
357 /* Windows documents that GetModuleFileName() will "truncate",
358 but makes no mention of the null terminator. Play it safe.
359 PLUS Windows itself defines MAX_PATH as the same, but anyway...
361 wprogpath
[MAXPATHLEN
]=_T('\0');
362 if (PyWin_DLLhModule
&&
363 GetModuleFileName(PyWin_DLLhModule
, wprogpath
, MAXPATHLEN
)) {
364 WideCharToMultiByte(CP_ACP
, 0,
366 dllpath
, MAXPATHLEN
+1,
369 wprogpath
[MAXPATHLEN
]=_T('\0');
370 if (GetModuleFileName(NULL
, wprogpath
, MAXPATHLEN
)) {
371 WideCharToMultiByte(CP_ACP
, 0,
373 progpath
, MAXPATHLEN
+1,
378 /* static init of progpath ensures final char remains \0 */
379 if (PyWin_DLLhModule
)
380 if (!GetModuleFileName(PyWin_DLLhModule
, dllpath
, MAXPATHLEN
))
382 if (GetModuleFileName(NULL
, progpath
, MAXPATHLEN
))
386 if (prog
== NULL
|| *prog
== '\0')
389 /* If there is no slash in the argv0 path, then we have to
390 * assume python is on the user's $PATH, since there's no
391 * other way to find a directory to start the search from. If
392 * $PATH isn't exported, you lose.
395 if (strchr(prog
, SEP
) || strchr(prog
, ALTSEP
))
397 if (strchr(prog
, SEP
))
399 strncpy(progpath
, prog
, MAXPATHLEN
);
402 char *delim
= strchr(path
, DELIM
);
405 size_t len
= delim
- path
;
406 /* ensure we can't overwrite buffer */
407 len
= min(MAXPATHLEN
,len
);
408 strncpy(progpath
, path
, len
);
409 *(progpath
+ len
) = '\0';
412 strncpy(progpath
, path
, MAXPATHLEN
);
414 /* join() is safe for MAXPATHLEN+1 size buffer */
415 join(progpath
, prog
);
416 if (exists(progpath
))
433 char argv0_path
[MAXPATHLEN
+1];
436 char *pythonhome
= Py_GetPythonHome();
437 char *envpath
= Py_GETENV("PYTHONPATH");
440 int skiphome
, skipdefault
;
441 char *machinepath
= NULL
;
442 char *userpath
= NULL
;
443 char zip_path
[MAXPATHLEN
+1];
448 /* progpath guaranteed \0 terminated in MAXPATH+1 bytes. */
449 strcpy(argv0_path
, progpath
);
451 if (pythonhome
== NULL
|| *pythonhome
== '\0') {
452 if (search_for_prefix(argv0_path
, LANDMARK
))
458 strncpy(prefix
, pythonhome
, MAXPATHLEN
);
460 if (envpath
&& *envpath
== '\0')
465 /* Calculate zip archive path */
466 if (dllpath
[0]) /* use name of python DLL */
467 strncpy(zip_path
, dllpath
, MAXPATHLEN
);
468 else /* use name of executable program */
469 strncpy(zip_path
, progpath
, MAXPATHLEN
);
470 zip_path
[MAXPATHLEN
] = '\0';
471 len
= strlen(zip_path
);
473 zip_path
[len
-3] = 'z'; /* change ending to "zip" */
474 zip_path
[len
-2] = 'i';
475 zip_path
[len
-1] = 'p';
481 skiphome
= pythonhome
==NULL
? 0 : 1;
482 machinepath
= getpythonregpath(HKEY_LOCAL_MACHINE
, skiphome
);
483 userpath
= getpythonregpath(HKEY_CURRENT_USER
, skiphome
);
484 /* We only use the default relative PYTHONPATH if we havent
485 anything better to use! */
486 skipdefault
= envpath
!=NULL
|| pythonhome
!=NULL
|| \
487 machinepath
!=NULL
|| userpath
!=NULL
;
490 /* We need to construct a path from the following parts.
491 (1) the PYTHONPATH environment variable, if set;
492 (2) for Win32, the zip archive file path;
493 (3) for Win32, the machinepath and userpath, if set;
494 (4) the PYTHONPATH config macro, with the leading "."
495 of each component replaced with pythonhome, if set;
496 (5) the directory containing the executable (argv0_path).
497 The length calculation calculates #4 first.
499 - If PYTHONHOME is set (in any way) item (3) is ignored.
500 - If registry values are used, (4) and (5) are ignored.
503 /* Calculate size of return buffer */
504 if (pythonhome
!= NULL
) {
507 for (p
= PYTHONPATH
; *p
; p
++) {
509 bufsz
++; /* number of DELIM plus one */
511 bufsz
*= strlen(pythonhome
);
515 bufsz
+= strlen(PYTHONPATH
) + 1;
516 bufsz
+= strlen(argv0_path
) + 1;
519 bufsz
+= strlen(userpath
) + 1;
521 bufsz
+= strlen(machinepath
) + 1;
522 bufsz
+= strlen(zip_path
) + 1;
525 bufsz
+= strlen(envpath
) + 1;
527 module_search_path
= buf
= malloc(bufsz
);
529 /* We can't exit, so print a warning and limp along */
530 fprintf(stderr
, "Can't malloc dynamic PYTHONPATH.\n");
532 fprintf(stderr
, "Using environment $PYTHONPATH.\n");
533 module_search_path
= envpath
;
536 fprintf(stderr
, "Using default static path.\n");
537 module_search_path
= PYTHONPATH
;
544 #endif /* MS_WINDOWS */
549 strcpy(buf
, envpath
);
550 buf
= strchr(buf
, '\0');
555 strcpy(buf
, zip_path
);
556 buf
= strchr(buf
, '\0');
560 strcpy(buf
, userpath
);
561 buf
= strchr(buf
, '\0');
566 strcpy(buf
, machinepath
);
567 buf
= strchr(buf
, '\0');
571 if (pythonhome
== NULL
) {
573 strcpy(buf
, PYTHONPATH
);
574 buf
= strchr(buf
, '\0');
578 if (pythonhome
== NULL
) {
579 strcpy(buf
, PYTHONPATH
);
580 buf
= strchr(buf
, '\0');
582 #endif /* MS_WINDOWS */
584 char *p
= PYTHONPATH
;
588 q
= strchr(p
, DELIM
);
593 if (p
[0] == '.' && is_sep(p
[1])) {
594 strcpy(buf
, pythonhome
);
595 buf
= strchr(buf
, '\0');
609 strcpy(buf
, argv0_path
);
610 buf
= strchr(buf
, '\0');
613 /* Now to pull one last hack/trick. If sys.prefix is
614 empty, then try and find it somewhere on the paths
615 we calculated. We scan backwards, as our general policy
616 is that Python core directories are at the *end* of
617 sys.path. We assume that our "lib" directory is
618 on the path, and that our 'prefix' directory is
622 char lookBuf
[MAXPATHLEN
+1];
623 char *look
= buf
- 1; /* 'buf' is at the end of the buffer */
626 char *lookEnd
= look
;
627 /* 'look' will end up one character before the
628 start of the path in question - even if this
629 is one character before the start of the buffer
631 while (*look
!= DELIM
&& look
>= module_search_path
)
633 nchars
= lookEnd
-look
;
634 strncpy(lookBuf
, look
+1, nchars
);
635 lookBuf
[nchars
] = '\0';
636 /* Up one level to the parent */
638 if (search_for_prefix(lookBuf
, LANDMARK
)) {
641 /* If we are out of paths to search - give up */
642 if (look
< module_search_path
)
650 /* External interface */
655 if (!module_search_path
)
657 return module_search_path
;
663 if (!module_search_path
)
669 Py_GetExecPrefix(void)
671 return Py_GetPrefix();
675 Py_GetProgramFullPath(void)
677 if (!module_search_path
)