blob: 709087994da8496f6bd0bbceb74b950d3abe0fec [file] [log] [blame]
Guido van Rossum582646a1996-05-28 22:30:17 +00001/* Return the initial module search path. */
2
Guido van Rossum667d7041995-08-04 04:20:48 +00003#include "Python.h"
4#include "osdefs.h"
5
Guido van Rossum305e5d01997-04-11 17:18:45 +00006#include <sys/types.h>
Guido van Rossum21f84971997-06-02 22:18:31 +00007#include <string.h>
Guido van Rossum667d7041995-08-04 04:20:48 +00008
Brett Cannonf6af76d2004-06-26 04:03:06 +00009#ifdef __APPLE__
Guido van Rossum54ecc3d1999-01-27 17:53:11 +000010#include <mach-o/dyld.h>
11#endif
12
Guido van Rossum305e5d01997-04-11 17:18:45 +000013/* Search in some common locations for the associated Python libraries.
14 *
15 * Two directories must be found, the platform independent directory
Barry Warsaw90126031997-04-11 20:27:03 +000016 * (prefix), containing the common .py and .pyc files, and the platform
17 * dependent directory (exec_prefix), containing the shared library
18 * modules. Note that prefix and exec_prefix can be the same directory,
19 * but for some installations, they are different.
Guido van Rossum305e5d01997-04-11 17:18:45 +000020 *
Barry Warsaw90126031997-04-11 20:27:03 +000021 * Py_GetPath() carries out separate searches for prefix and exec_prefix.
22 * Each search tries a number of different locations until a ``landmark''
23 * file or directory is found. If no prefix or exec_prefix is found, a
24 * warning message is issued and the preprocessor defined PREFIX and
25 * EXEC_PREFIX are used (even though they will not work); python carries on
26 * as best as is possible, but most imports will fail.
Guido van Rossum305e5d01997-04-11 17:18:45 +000027 *
28 * Before any searches are done, the location of the executable is
Guido van Rossumd8faa362007-04-27 19:54:29 +000029 * determined. If argv[0] has one or more slashes in it, it is used
Barry Warsaw90126031997-04-11 20:27:03 +000030 * unchanged. Otherwise, it must have been invoked from the shell's path,
31 * so we search $PATH for the named executable and use that. If the
32 * executable was not found on $PATH (or there was no $PATH environment
33 * variable), the original argv[0] string is used.
Guido van Rossum305e5d01997-04-11 17:18:45 +000034 *
Barry Warsaw90126031997-04-11 20:27:03 +000035 * Next, the executable location is examined to see if it is a symbolic
36 * link. If so, the link is chased (correctly interpreting a relative
37 * pathname if one is found) and the directory of the link target is used.
Guido van Rossum305e5d01997-04-11 17:18:45 +000038 *
Barry Warsaw90126031997-04-11 20:27:03 +000039 * Finally, argv0_path is set to the directory containing the executable
40 * (i.e. the last component is stripped).
Guido van Rossum305e5d01997-04-11 17:18:45 +000041 *
Barry Warsaw90126031997-04-11 20:27:03 +000042 * With argv0_path in hand, we perform a number of steps. The same steps
43 * are performed for prefix and for exec_prefix, but with a different
44 * landmark.
Guido van Rossum305e5d01997-04-11 17:18:45 +000045 *
46 * Step 1. Are we running python out of the build directory? This is
47 * checked by looking for a different kind of landmark relative to
Barry Warsaw90126031997-04-11 20:27:03 +000048 * argv0_path. For prefix, the landmark's path is derived from the VPATH
49 * preprocessor variable (taking into account that its value is almost, but
50 * not quite, what we need). For exec_prefix, the landmark is
Antoine Pitroueba57b62010-08-14 12:33:18 +000051 * pybuilddir.txt. If the landmark is found, we're done.
Guido van Rossum305e5d01997-04-11 17:18:45 +000052 *
53 * For the remaining steps, the prefix landmark will always be
Jeremy Hylton847a9962000-05-26 21:49:07 +000054 * lib/python$VERSION/os.py and the exec_prefix will always be
Guido van Rossum266033e1997-10-20 23:20:32 +000055 * lib/python$VERSION/lib-dynload, where $VERSION is Python's version
Barry Warsaw90126031997-04-11 20:27:03 +000056 * number as supplied by the Makefile. Note that this means that no more
57 * build directory checking is performed; if the first step did not find
58 * the landmarks, the assumption is that python is running from an
59 * installed setup.
Guido van Rossum305e5d01997-04-11 17:18:45 +000060 *
61 * Step 2. See if the $PYTHONHOME environment variable points to the
Barry Warsaw90126031997-04-11 20:27:03 +000062 * installed location of the Python libraries. If $PYTHONHOME is set, then
63 * it points to prefix and exec_prefix. $PYTHONHOME can be a single
64 * directory, which is used for both, or the prefix and exec_prefix
65 * directories separated by a colon.
Guido van Rossum305e5d01997-04-11 17:18:45 +000066 *
67 * Step 3. Try to find prefix and exec_prefix relative to argv0_path,
Barry Warsaw90126031997-04-11 20:27:03 +000068 * backtracking up the path until it is exhausted. This is the most common
69 * step to succeed. Note that if prefix and exec_prefix are different,
70 * exec_prefix is more likely to be found; however if exec_prefix is a
71 * subdirectory of prefix, both will be found.
Guido van Rossum305e5d01997-04-11 17:18:45 +000072 *
Barry Warsaw90126031997-04-11 20:27:03 +000073 * Step 4. Search the directories pointed to by the preprocessor variables
74 * PREFIX and EXEC_PREFIX. These are supplied by the Makefile but can be
75 * passed in as options to the configure script.
Guido van Rossum305e5d01997-04-11 17:18:45 +000076 *
Barry Warsaw90126031997-04-11 20:27:03 +000077 * That's it!
78 *
79 * Well, almost. Once we have determined prefix and exec_prefix, the
Thomas Wouters7e474022000-07-16 12:04:32 +000080 * preprocessor variable PYTHONPATH is used to construct a path. Each
Barry Warsaw90126031997-04-11 20:27:03 +000081 * relative path on PYTHONPATH is prefixed with prefix. Then the directory
82 * containing the shared library modules is appended. The environment
83 * variable $PYTHONPATH is inserted in front of it all. Finally, the
84 * prefix and exec_prefix globals are tweaked so they reflect the values
85 * expected by other code, by stripping the "lib/python$VERSION/..." stuff
86 * off. If either points to the build directory, the globals are reset to
87 * the corresponding preprocessor variables (so sys.prefix will reflect the
88 * installation location, even though sys.path points into the build
89 * directory). This seems to make more sense given that currently the only
90 * known use of sys.prefix and sys.exec_prefix is for the ILU installation
91 * process to find the installed Python tree.
Antoine Pitroueba57b62010-08-14 12:33:18 +000092 *
Kristján Valur Jónsson3b69db22010-09-27 05:32:54 +000093 * An embedding application can use Py_SetPath() to override all of
94 * these authomatic path computations.
95 *
Antoine Pitroueba57b62010-08-14 12:33:18 +000096 * NOTE: Windows MSVC builds use PC/getpathp.c instead!
Barry Warsaw90126031997-04-11 20:27:03 +000097 */
Guido van Rossum305e5d01997-04-11 17:18:45 +000098
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000099#ifdef __cplusplus
100 extern "C" {
101#endif
102
103
Guido van Rossum305e5d01997-04-11 17:18:45 +0000104#ifndef VERSION
Guido van Rossumd0e74ea2001-01-23 01:57:40 +0000105#define VERSION "2.1"
Guido van Rossum305e5d01997-04-11 17:18:45 +0000106#endif
107
108#ifndef VPATH
109#define VPATH "."
Guido van Rossum667d7041995-08-04 04:20:48 +0000110#endif
111
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000112#ifndef PREFIX
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000113# ifdef __VMS
114# define PREFIX ""
115# else
116# define PREFIX "/usr/local"
117# endif
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000118#endif
119
120#ifndef EXEC_PREFIX
Guido van Rossum6e12d561996-07-30 20:36:12 +0000121#define EXEC_PREFIX PREFIX
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000122#endif
123
Guido van Rossum305e5d01997-04-11 17:18:45 +0000124#ifndef PYTHONPATH
Guido van Rossum305e5d01997-04-11 17:18:45 +0000125#define PYTHONPATH PREFIX "/lib/python" VERSION ":" \
Guido van Rossume296ced2001-09-28 20:00:29 +0000126 EXEC_PREFIX "/lib/python" VERSION "/lib-dynload"
Guido van Rossum305e5d01997-04-11 17:18:45 +0000127#endif
Guido van Rossum667d7041995-08-04 04:20:48 +0000128
Guido van Rossum305e5d01997-04-11 17:18:45 +0000129#ifndef LANDMARK
Martin v. Löwis790465f2008-04-05 20:41:37 +0000130#define LANDMARK L"os.py"
Guido van Rossum305e5d01997-04-11 17:18:45 +0000131#endif
Guido van Rossume296ced2001-09-28 20:00:29 +0000132
Martin v. Löwis790465f2008-04-05 20:41:37 +0000133static wchar_t prefix[MAXPATHLEN+1];
134static wchar_t exec_prefix[MAXPATHLEN+1];
135static wchar_t progpath[MAXPATHLEN+1];
136static wchar_t *module_search_path = NULL;
Antoine Pitroubb6eb852011-03-17 23:34:33 +0100137static int module_search_path_malloced = 0;
Andrew M. Kuchlingb2ceb3a2010-02-22 23:26:10 +0000138static wchar_t *lib_python = L"lib/python" VERSION;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000139
Guido van Rossum305e5d01997-04-11 17:18:45 +0000140static void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000141reduce(wchar_t *dir)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000142{
Martin v. Löwis790465f2008-04-05 20:41:37 +0000143 size_t i = wcslen(dir);
Fred Drakeedabdc12000-07-08 06:16:37 +0000144 while (i > 0 && dir[i] != SEP)
145 --i;
146 dir[i] = '\0';
Guido van Rossum305e5d01997-04-11 17:18:45 +0000147}
Guido van Rossumd29806c1998-01-19 22:06:22 +0000148
Guido van Rossum305e5d01997-04-11 17:18:45 +0000149static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000150isfile(wchar_t *filename) /* Is file, not directory */
Guido van Rossum305e5d01997-04-11 17:18:45 +0000151{
Fred Drakeedabdc12000-07-08 06:16:37 +0000152 struct stat buf;
Victor Stinner4e314432010-10-07 21:45:39 +0000153 if (_Py_wstat(filename, &buf) != 0)
Fred Drakeedabdc12000-07-08 06:16:37 +0000154 return 0;
155 if (!S_ISREG(buf.st_mode))
156 return 0;
157 return 1;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000158}
159
160
161static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000162ismodule(wchar_t *filename) /* Is module -- check for .pyc/.pyo too */
Guido van Rossumd29806c1998-01-19 22:06:22 +0000163{
Fred Drakeedabdc12000-07-08 06:16:37 +0000164 if (isfile(filename))
165 return 1;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000166
Fred Drakeedabdc12000-07-08 06:16:37 +0000167 /* Check for the compiled version of prefix. */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000168 if (wcslen(filename) < MAXPATHLEN) {
169 wcscat(filename, Py_OptimizeFlag ? L"o" : L"c");
Fred Drakeedabdc12000-07-08 06:16:37 +0000170 if (isfile(filename))
171 return 1;
172 }
173 return 0;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000174}
175
176
177static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000178isxfile(wchar_t *filename) /* Is executable file */
Guido van Rossumd29806c1998-01-19 22:06:22 +0000179{
Fred Drakeedabdc12000-07-08 06:16:37 +0000180 struct stat buf;
Victor Stinner4e314432010-10-07 21:45:39 +0000181 if (_Py_wstat(filename, &buf) != 0)
Fred Drakeedabdc12000-07-08 06:16:37 +0000182 return 0;
183 if (!S_ISREG(buf.st_mode))
184 return 0;
185 if ((buf.st_mode & 0111) == 0)
186 return 0;
187 return 1;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000188}
189
190
191static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000192isdir(wchar_t *filename) /* Is directory */
Guido van Rossumd29806c1998-01-19 22:06:22 +0000193{
Fred Drakeedabdc12000-07-08 06:16:37 +0000194 struct stat buf;
Victor Stinner4e314432010-10-07 21:45:39 +0000195 if (_Py_wstat(filename, &buf) != 0)
Fred Drakeedabdc12000-07-08 06:16:37 +0000196 return 0;
197 if (!S_ISDIR(buf.st_mode))
198 return 0;
199 return 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000200}
201
202
Tim Petersec8c5a92004-08-08 01:00:47 +0000203/* Add a path component, by appending stuff to buffer.
204 buffer must have at least MAXPATHLEN + 1 bytes allocated, and contain a
205 NUL-terminated string with no more than MAXPATHLEN characters (not counting
206 the trailing NUL). It's a fatal error if it contains a string longer than
207 that (callers must be careful!). If these requirements are met, it's
208 guaranteed that buffer will still be a NUL-terminated string with no more
209 than MAXPATHLEN characters at exit. If stuff is too long, only as much of
210 stuff as fits will be appended.
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000211*/
Guido van Rossum305e5d01997-04-11 17:18:45 +0000212static void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000213joinpath(wchar_t *buffer, wchar_t *stuff)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000214{
Fred Drakeedabdc12000-07-08 06:16:37 +0000215 size_t n, k;
216 if (stuff[0] == SEP)
217 n = 0;
218 else {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000219 n = wcslen(buffer);
Fred Drakeedabdc12000-07-08 06:16:37 +0000220 if (n > 0 && buffer[n-1] != SEP && n < MAXPATHLEN)
221 buffer[n++] = SEP;
222 }
Tim Petersec8c5a92004-08-08 01:00:47 +0000223 if (n > MAXPATHLEN)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000224 Py_FatalError("buffer overflow in getpath.c's joinpath()");
Martin v. Löwis790465f2008-04-05 20:41:37 +0000225 k = wcslen(stuff);
Fred Drakeedabdc12000-07-08 06:16:37 +0000226 if (n + k > MAXPATHLEN)
227 k = MAXPATHLEN - n;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000228 wcsncpy(buffer+n, stuff, k);
Fred Drakeedabdc12000-07-08 06:16:37 +0000229 buffer[n+k] = '\0';
Guido van Rossum305e5d01997-04-11 17:18:45 +0000230}
231
Guido van Rossume296ced2001-09-28 20:00:29 +0000232/* copy_absolute requires that path be allocated at least
233 MAXPATHLEN + 1 bytes and that p be no more than MAXPATHLEN bytes. */
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000234static void
Victor Stinnerf4061da2010-10-14 12:37:19 +0000235copy_absolute(wchar_t *path, wchar_t *p, size_t pathlen)
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000236{
Guido van Rossume296ced2001-09-28 20:00:29 +0000237 if (p[0] == SEP)
Martin v. Löwis790465f2008-04-05 20:41:37 +0000238 wcscpy(path, p);
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000239 else {
Victor Stinnerf4061da2010-10-14 12:37:19 +0000240 if (!_Py_wgetcwd(path, pathlen)) {
Victor Stinner4f3abb02010-10-07 23:29:18 +0000241 /* unable to get the current directory */
242 wcscpy(path, p);
243 return;
244 }
Guido van Rossume296ced2001-09-28 20:00:29 +0000245 if (p[0] == '.' && p[1] == SEP)
246 p += 2;
247 joinpath(path, p);
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000248 }
249}
Guido van Rossum305e5d01997-04-11 17:18:45 +0000250
Guido van Rossume296ced2001-09-28 20:00:29 +0000251/* absolutize() requires that path be allocated at least MAXPATHLEN+1 bytes. */
252static void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000253absolutize(wchar_t *path)
Guido van Rossume296ced2001-09-28 20:00:29 +0000254{
Victor Stinnerf4061da2010-10-14 12:37:19 +0000255 wchar_t buffer[MAXPATHLEN+1];
Guido van Rossume296ced2001-09-28 20:00:29 +0000256
257 if (path[0] == SEP)
258 return;
Victor Stinnerf4061da2010-10-14 12:37:19 +0000259 copy_absolute(buffer, path, MAXPATHLEN+1);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000260 wcscpy(path, buffer);
Guido van Rossume296ced2001-09-28 20:00:29 +0000261}
262
263/* search_for_prefix requires that argv0_path be no more than MAXPATHLEN
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000264 bytes long.
265*/
Guido van Rossum305e5d01997-04-11 17:18:45 +0000266static int
Victor Stinnerae4836d2010-11-08 23:49:47 +0000267search_for_prefix(wchar_t *argv0_path, wchar_t *home, wchar_t *_prefix)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000268{
Fred Drakeedabdc12000-07-08 06:16:37 +0000269 size_t n;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000270 wchar_t *vpath;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000271
Fred Drakeedabdc12000-07-08 06:16:37 +0000272 /* If PYTHONHOME is set, we believe it unconditionally */
273 if (home) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000274 wchar_t *delim;
275 wcsncpy(prefix, home, MAXPATHLEN);
276 delim = wcschr(prefix, DELIM);
Fred Drakeedabdc12000-07-08 06:16:37 +0000277 if (delim)
Martin v. Löwis790465f2008-04-05 20:41:37 +0000278 *delim = L'\0';
Fred Drakeedabdc12000-07-08 06:16:37 +0000279 joinpath(prefix, lib_python);
280 joinpath(prefix, LANDMARK);
281 return 1;
282 }
Jeremy Hylton847a9962000-05-26 21:49:07 +0000283
Fred Drakeedabdc12000-07-08 06:16:37 +0000284 /* Check to see if argv[0] is in the build directory */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000285 wcscpy(prefix, argv0_path);
286 joinpath(prefix, L"Modules/Setup");
Fred Drakeedabdc12000-07-08 06:16:37 +0000287 if (isfile(prefix)) {
Neil Schemenauer6cf07022001-01-24 17:13:11 +0000288 /* Check VPATH to see if argv0_path is in the build directory. */
Victor Stinner21582312010-10-23 00:13:28 +0000289 vpath = _Py_char2wchar(VPATH, NULL);
290 if (vpath != NULL) {
291 wcscpy(prefix, argv0_path);
292 joinpath(prefix, vpath);
293 PyMem_Free(vpath);
294 joinpath(prefix, L"Lib");
295 joinpath(prefix, LANDMARK);
296 if (ismodule(prefix))
297 return -1;
298 }
Fred Drakeedabdc12000-07-08 06:16:37 +0000299 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000300
Fred Drakeedabdc12000-07-08 06:16:37 +0000301 /* Search from argv0_path, until root is found */
Victor Stinnerf4061da2010-10-14 12:37:19 +0000302 copy_absolute(prefix, argv0_path, MAXPATHLEN+1);
Fred Drakeedabdc12000-07-08 06:16:37 +0000303 do {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000304 n = wcslen(prefix);
Fred Drakeedabdc12000-07-08 06:16:37 +0000305 joinpath(prefix, lib_python);
306 joinpath(prefix, LANDMARK);
307 if (ismodule(prefix))
308 return 1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000309 prefix[n] = L'\0';
Fred Drakeedabdc12000-07-08 06:16:37 +0000310 reduce(prefix);
311 } while (prefix[0]);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000312
Fred Drakeedabdc12000-07-08 06:16:37 +0000313 /* Look at configure's PREFIX */
Victor Stinnerae4836d2010-11-08 23:49:47 +0000314 wcsncpy(prefix, _prefix, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000315 joinpath(prefix, lib_python);
316 joinpath(prefix, LANDMARK);
317 if (ismodule(prefix))
318 return 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000319
Fred Drakeedabdc12000-07-08 06:16:37 +0000320 /* Fail */
321 return 0;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000322}
323
324
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000325/* search_for_exec_prefix requires that argv0_path be no more than
Guido van Rossume296ced2001-09-28 20:00:29 +0000326 MAXPATHLEN bytes long.
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000327*/
Guido van Rossum305e5d01997-04-11 17:18:45 +0000328static int
Victor Stinnerae4836d2010-11-08 23:49:47 +0000329search_for_exec_prefix(wchar_t *argv0_path, wchar_t *home, wchar_t *_exec_prefix)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000330{
Fred Drakeedabdc12000-07-08 06:16:37 +0000331 size_t n;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000332
Fred Drakeedabdc12000-07-08 06:16:37 +0000333 /* If PYTHONHOME is set, we believe it unconditionally */
334 if (home) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000335 wchar_t *delim;
336 delim = wcschr(home, DELIM);
Fred Drakeedabdc12000-07-08 06:16:37 +0000337 if (delim)
Martin v. Löwis790465f2008-04-05 20:41:37 +0000338 wcsncpy(exec_prefix, delim+1, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000339 else
Martin v. Löwis790465f2008-04-05 20:41:37 +0000340 wcsncpy(exec_prefix, home, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000341 joinpath(exec_prefix, lib_python);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000342 joinpath(exec_prefix, L"lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000343 return 1;
344 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000345
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000346 /* Check to see if argv[0] is in the build directory. "pybuilddir.txt"
347 is written by setup.py and contains the relative path to the location
348 of shared library modules. */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000349 wcscpy(exec_prefix, argv0_path);
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000350 joinpath(exec_prefix, L"pybuilddir.txt");
Fred Drakeedabdc12000-07-08 06:16:37 +0000351 if (isfile(exec_prefix)) {
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000352 FILE *f = _Py_wfopen(exec_prefix, L"rb");
353 if (f == NULL)
354 errno = 0;
355 else {
356 char buf[MAXPATHLEN+1];
357 PyObject *decoded;
358 wchar_t rel_builddir_path[MAXPATHLEN+1];
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000359 n = fread(buf, 1, MAXPATHLEN, f);
360 buf[n] = '\0';
361 fclose(f);
362 decoded = PyUnicode_DecodeUTF8(buf, n, "surrogateescape");
363 if (decoded != NULL) {
Victor Stinner47f637c2010-11-10 14:12:20 +0000364 Py_ssize_t k;
Martin v. Löwis4d0d4712010-12-03 20:14:31 +0000365 k = PyUnicode_AsWideChar(decoded,
Victor Stinnere406ef42010-08-14 00:07:14 +0000366 rel_builddir_path, MAXPATHLEN);
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000367 Py_DECREF(decoded);
Victor Stinner47f637c2010-11-10 14:12:20 +0000368 if (k >= 0) {
369 rel_builddir_path[k] = L'\0';
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000370 wcscpy(exec_prefix, argv0_path);
371 joinpath(exec_prefix, rel_builddir_path);
372 return -1;
373 }
374 }
375 }
Fred Drakeedabdc12000-07-08 06:16:37 +0000376 }
Jeremy Hylton847a9962000-05-26 21:49:07 +0000377
Fred Drakeedabdc12000-07-08 06:16:37 +0000378 /* Search from argv0_path, until root is found */
Victor Stinnerf4061da2010-10-14 12:37:19 +0000379 copy_absolute(exec_prefix, argv0_path, MAXPATHLEN+1);
Fred Drakeedabdc12000-07-08 06:16:37 +0000380 do {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000381 n = wcslen(exec_prefix);
Fred Drakeedabdc12000-07-08 06:16:37 +0000382 joinpath(exec_prefix, lib_python);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000383 joinpath(exec_prefix, L"lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000384 if (isdir(exec_prefix))
385 return 1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000386 exec_prefix[n] = L'\0';
Fred Drakeedabdc12000-07-08 06:16:37 +0000387 reduce(exec_prefix);
388 } while (exec_prefix[0]);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000389
Fred Drakeedabdc12000-07-08 06:16:37 +0000390 /* Look at configure's EXEC_PREFIX */
Victor Stinnerae4836d2010-11-08 23:49:47 +0000391 wcsncpy(exec_prefix, _exec_prefix, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000392 joinpath(exec_prefix, lib_python);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000393 joinpath(exec_prefix, L"lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000394 if (isdir(exec_prefix))
395 return 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000396
Fred Drakeedabdc12000-07-08 06:16:37 +0000397 /* Fail */
398 return 0;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000399}
400
Guido van Rossum305e5d01997-04-11 17:18:45 +0000401static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000402calculate_path(void)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000403{
Martin v. Löwis790465f2008-04-05 20:41:37 +0000404 extern wchar_t *Py_GetProgramName(void);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000405
Martin v. Löwis790465f2008-04-05 20:41:37 +0000406 static wchar_t delimiter[2] = {DELIM, '\0'};
407 static wchar_t separator[2] = {SEP, '\0'};
Martin v. Löwis790465f2008-04-05 20:41:37 +0000408 char *_rtpypath = Py_GETENV("PYTHONPATH"); /* XXX use wide version on Windows */
Victor Stinner99563b12011-06-20 14:45:54 +0200409 wchar_t *rtpypath = NULL;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000410 wchar_t *home = Py_GetPythonHome();
411 char *_path = getenv("PATH");
Victor Stinner61aad572010-10-16 23:38:07 +0000412 wchar_t *path_buffer = NULL;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000413 wchar_t *path = NULL;
414 wchar_t *prog = Py_GetProgramName();
415 wchar_t argv0_path[MAXPATHLEN+1];
416 wchar_t zip_path[MAXPATHLEN+1];
Fred Drakeedabdc12000-07-08 06:16:37 +0000417 int pfound, efound; /* 1 if found; -1 if found build directory */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000418 wchar_t *buf;
Fred Drakeedabdc12000-07-08 06:16:37 +0000419 size_t bufsz;
420 size_t prefixsz;
Victor Stinnerae4836d2010-11-08 23:49:47 +0000421 wchar_t *defpath;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000422#ifdef WITH_NEXT_FRAMEWORK
Fred Drakeedabdc12000-07-08 06:16:37 +0000423 NSModule pythonModule;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000424#endif
Jack Jansen1afd4802004-06-03 14:33:03 +0000425#ifdef __APPLE__
Skip Montanaro2b474452006-01-05 10:49:13 +0000426#if MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_4
Brett Cannonbd9c3f72005-09-16 02:43:35 +0000427 uint32_t nsexeclength = MAXPATHLEN;
Skip Montanaro2b474452006-01-05 10:49:13 +0000428#else
429 unsigned long nsexeclength = MAXPATHLEN;
430#endif
Victor Stinnerae4836d2010-11-08 23:49:47 +0000431 char execpath[MAXPATHLEN+1];
Jack Jansen1afd4802004-06-03 14:33:03 +0000432#endif
Victor Stinnerae4836d2010-11-08 23:49:47 +0000433 wchar_t *_pythonpath, *_prefix, *_exec_prefix;
434
435 _pythonpath = _Py_char2wchar(PYTHONPATH, NULL);
436 _prefix = _Py_char2wchar(PREFIX, NULL);
437 _exec_prefix = _Py_char2wchar(EXEC_PREFIX, NULL);
438
439 if (!_pythonpath || !_prefix || !_exec_prefix) {
440 Py_FatalError(
441 "Unable to decode path variables in getpath.c: "
442 "memory error");
443 }
Guido van Rossume296ced2001-09-28 20:00:29 +0000444
Victor Stinner61aad572010-10-16 23:38:07 +0000445 if (_path) {
446 path_buffer = _Py_char2wchar(_path, NULL);
447 path = path_buffer;
448 }
Martin v. Löwis790465f2008-04-05 20:41:37 +0000449
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000450 /* If there is no slash in the argv0 path, then we have to
451 * assume python is on the user's $PATH, since there's no
452 * other way to find a directory to start the search from. If
453 * $PATH isn't exported, you lose.
454 */
455 if (wcschr(prog, SEP))
456 wcsncpy(progpath, prog, MAXPATHLEN);
Jack Jansen1afd4802004-06-03 14:33:03 +0000457#ifdef __APPLE__
458 /* On Mac OS X, if a script uses an interpreter of the form
459 * "#!/opt/python2.3/bin/python", the kernel only passes "python"
460 * as argv[0], which falls through to the $PATH search below.
461 * If /opt/python2.3/bin isn't in your path, or is near the end,
462 * this algorithm may incorrectly find /usr/bin/python. To work
463 * around this, we can use _NSGetExecutablePath to get a better
464 * hint of what the intended interpreter was, although this
465 * will fail if a relative path was used. but in that case,
466 * absolutize() should help us out below
467 */
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000468 else if(0 == _NSGetExecutablePath(execpath, &nsexeclength) && execpath[0] == SEP) {
469 size_t r = mbstowcs(progpath, execpath, MAXPATHLEN+1);
470 if (r == (size_t)-1 || r > MAXPATHLEN) {
471 /* Could not convert execpath, or it's too long. */
472 progpath[0] = '\0';
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000473 }
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000474 }
Brett Cannon6cc48142004-06-24 00:48:44 +0000475#endif /* __APPLE__ */
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000476 else if (path) {
477 while (1) {
478 wchar_t *delim = wcschr(path, DELIM);
Jack Jansen55070f52001-12-02 23:56:28 +0000479
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000480 if (delim) {
481 size_t len = delim - path;
482 if (len > MAXPATHLEN)
483 len = MAXPATHLEN;
484 wcsncpy(progpath, path, len);
485 *(progpath + len) = '\0';
486 }
487 else
488 wcsncpy(progpath, path, MAXPATHLEN);
Jack Jansen55070f52001-12-02 23:56:28 +0000489
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000490 joinpath(progpath, prog);
491 if (isxfile(progpath))
492 break;
Jack Jansen55070f52001-12-02 23:56:28 +0000493
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000494 if (!delim) {
495 progpath[0] = L'\0';
496 break;
497 }
498 path = delim + 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000499 }
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000500 }
501 else
502 progpath[0] = '\0';
Victor Stinner61aad572010-10-16 23:38:07 +0000503 if (path_buffer != NULL)
504 PyMem_Free(path_buffer);
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000505 if (progpath[0] != SEP && progpath[0] != '\0')
506 absolutize(progpath);
507 wcsncpy(argv0_path, progpath, MAXPATHLEN);
508 argv0_path[MAXPATHLEN] = '\0';
Jack Jansen55070f52001-12-02 23:56:28 +0000509
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000510#ifdef WITH_NEXT_FRAMEWORK
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000511 /* On Mac OS X we have a special case if we're running from a framework.
512 ** This is because the python home should be set relative to the library,
513 ** which is in the framework, not relative to the executable, which may
514 ** be outside of the framework. Except when we're in the build directory...
515 */
Fred Drakeedabdc12000-07-08 06:16:37 +0000516 pythonModule = NSModuleForSymbol(NSLookupAndBindSymbol("_Py_Initialize"));
517 /* Use dylib functions to find out where the framework was loaded from */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000518 buf = (wchar_t *)NSLibraryNameForModule(pythonModule);
Fred Drakeedabdc12000-07-08 06:16:37 +0000519 if (buf != NULL) {
520 /* We're in a framework. */
Jack Jansene925faf2001-08-15 01:14:40 +0000521 /* See if we might be in the build directory. The framework in the
522 ** build directory is incomplete, it only has the .dylib and a few
523 ** needed symlinks, it doesn't have the Lib directories and such.
524 ** If we're running with the framework from the build directory we must
525 ** be running the interpreter in the build directory, so we use the
526 ** build-directory-specific logic to find Lib and such.
527 */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000528 wcsncpy(argv0_path, buf, MAXPATHLEN);
Jack Jansene925faf2001-08-15 01:14:40 +0000529 reduce(argv0_path);
530 joinpath(argv0_path, lib_python);
531 joinpath(argv0_path, LANDMARK);
532 if (!ismodule(argv0_path)) {
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000533 /* We are in the build directory so use the name of the
534 executable - we know that the absolute path is passed */
535 wcsncpy(argv0_path, progpath, MAXPATHLEN);
Jack Jansene925faf2001-08-15 01:14:40 +0000536 }
537 else {
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000538 /* Use the location of the library as the progpath */
539 wcsncpy(argv0_path, buf, MAXPATHLEN);
Jack Jansene925faf2001-08-15 01:14:40 +0000540 }
Fred Drakeedabdc12000-07-08 06:16:37 +0000541 }
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000542#endif
Guido van Rossume296ced2001-09-28 20:00:29 +0000543
Guido van Rossum305e5d01997-04-11 17:18:45 +0000544#if HAVE_READLINK
Fred Drakeedabdc12000-07-08 06:16:37 +0000545 {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000546 wchar_t tmpbuffer[MAXPATHLEN+1];
547 int linklen = _Py_wreadlink(progpath, tmpbuffer, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000548 while (linklen != -1) {
Fred Drakeedabdc12000-07-08 06:16:37 +0000549 if (tmpbuffer[0] == SEP)
Guido van Rossume296ced2001-09-28 20:00:29 +0000550 /* tmpbuffer should never be longer than MAXPATHLEN,
551 but extra check does not hurt */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000552 wcsncpy(argv0_path, tmpbuffer, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000553 else {
554 /* Interpret relative to progpath */
555 reduce(argv0_path);
556 joinpath(argv0_path, tmpbuffer);
557 }
Martin v. Löwis790465f2008-04-05 20:41:37 +0000558 linklen = _Py_wreadlink(argv0_path, tmpbuffer, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000559 }
560 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000561#endif /* HAVE_READLINK */
562
Fred Drakeedabdc12000-07-08 06:16:37 +0000563 reduce(argv0_path);
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000564 /* At this point, argv0_path is guaranteed to be less than
565 MAXPATHLEN bytes long.
566 */
Guido van Rossum305e5d01997-04-11 17:18:45 +0000567
Victor Stinnerae4836d2010-11-08 23:49:47 +0000568 if (!(pfound = search_for_prefix(argv0_path, home, _prefix))) {
Fred Drakeedabdc12000-07-08 06:16:37 +0000569 if (!Py_FrozenFlag)
570 fprintf(stderr,
Guido van Rossume296ced2001-09-28 20:00:29 +0000571 "Could not find platform independent libraries <prefix>\n");
Victor Stinnerae4836d2010-11-08 23:49:47 +0000572 wcsncpy(prefix, _prefix, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000573 joinpath(prefix, lib_python);
574 }
575 else
576 reduce(prefix);
Guido van Rossume296ced2001-09-28 20:00:29 +0000577
Martin v. Löwis790465f2008-04-05 20:41:37 +0000578 wcsncpy(zip_path, prefix, MAXPATHLEN);
579 zip_path[MAXPATHLEN] = L'\0';
Just van Rossum52e14d62002-12-30 22:08:05 +0000580 if (pfound > 0) { /* Use the reduced prefix returned by Py_GetPrefix() */
581 reduce(zip_path);
582 reduce(zip_path);
583 }
584 else
Victor Stinnerae4836d2010-11-08 23:49:47 +0000585 wcsncpy(zip_path, _prefix, MAXPATHLEN);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000586 joinpath(zip_path, L"lib/python00.zip");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000587 bufsz = wcslen(zip_path); /* Replace "00" with version */
Just van Rossum52e14d62002-12-30 22:08:05 +0000588 zip_path[bufsz - 6] = VERSION[0];
589 zip_path[bufsz - 5] = VERSION[2];
590
Victor Stinnerae4836d2010-11-08 23:49:47 +0000591 if (!(efound = search_for_exec_prefix(argv0_path, home, _exec_prefix))) {
Fred Drakeedabdc12000-07-08 06:16:37 +0000592 if (!Py_FrozenFlag)
593 fprintf(stderr,
Guido van Rossume296ced2001-09-28 20:00:29 +0000594 "Could not find platform dependent libraries <exec_prefix>\n");
Victor Stinnerae4836d2010-11-08 23:49:47 +0000595 wcsncpy(exec_prefix, _exec_prefix, MAXPATHLEN);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000596 joinpath(exec_prefix, L"lib/lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000597 }
598 /* If we found EXEC_PREFIX do *not* reduce it! (Yet.) */
Guido van Rossum305e5d01997-04-11 17:18:45 +0000599
Fred Drakeedabdc12000-07-08 06:16:37 +0000600 if ((!pfound || !efound) && !Py_FrozenFlag)
601 fprintf(stderr,
602 "Consider setting $PYTHONHOME to <prefix>[:<exec_prefix>]\n");
Guido van Rossum305e5d01997-04-11 17:18:45 +0000603
Fred Drakeedabdc12000-07-08 06:16:37 +0000604 /* Calculate size of return buffer.
605 */
606 bufsz = 0;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000607
Martin v. Löwis790465f2008-04-05 20:41:37 +0000608 if (_rtpypath) {
Victor Stinner99563b12011-06-20 14:45:54 +0200609 size_t rtpypath_len;
610 rtpypath = _Py_char2wchar(_rtpypath, &rtpypath_len);
611 if (rtpypath != NULL)
612 bufsz += rtpypath_len + 1;
613 else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 _rtpypath = NULL;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000615 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000616
Victor Stinnerae4836d2010-11-08 23:49:47 +0000617 defpath = _pythonpath;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000618 prefixsz = wcslen(prefix) + 1;
Fred Drakeedabdc12000-07-08 06:16:37 +0000619 while (1) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000620 wchar_t *delim = wcschr(defpath, DELIM);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000621
Fred Drakeedabdc12000-07-08 06:16:37 +0000622 if (defpath[0] != SEP)
623 /* Paths are relative to prefix */
624 bufsz += prefixsz;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000625
Fred Drakeedabdc12000-07-08 06:16:37 +0000626 if (delim)
627 bufsz += delim - defpath + 1;
628 else {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000629 bufsz += wcslen(defpath) + 1;
Fred Drakeedabdc12000-07-08 06:16:37 +0000630 break;
631 }
632 defpath = delim + 1;
633 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000634
Martin v. Löwis790465f2008-04-05 20:41:37 +0000635 bufsz += wcslen(zip_path) + 1;
636 bufsz += wcslen(exec_prefix) + 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000637
Martin v. Löwis790465f2008-04-05 20:41:37 +0000638 buf = (wchar_t *)PyMem_Malloc(bufsz*sizeof(wchar_t));
Guido van Rossum305e5d01997-04-11 17:18:45 +0000639
Fred Drakeedabdc12000-07-08 06:16:37 +0000640 if (buf == NULL) {
641 /* We can't exit, so print a warning and limp along */
642 fprintf(stderr, "Not enough memory for dynamic PYTHONPATH.\n");
643 fprintf(stderr, "Using default static PYTHONPATH.\n");
Martin v. Löwis790465f2008-04-05 20:41:37 +0000644 module_search_path = L"" PYTHONPATH;
Fred Drakeedabdc12000-07-08 06:16:37 +0000645 }
646 else {
647 /* Run-time value of $PYTHONPATH goes first */
Victor Stinner99563b12011-06-20 14:45:54 +0200648 if (rtpypath) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000649 wcscpy(buf, rtpypath);
650 wcscat(buf, delimiter);
Fred Drakeedabdc12000-07-08 06:16:37 +0000651 }
652 else
653 buf[0] = '\0';
Guido van Rossum305e5d01997-04-11 17:18:45 +0000654
Just van Rossum52e14d62002-12-30 22:08:05 +0000655 /* Next is the default zip path */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000656 wcscat(buf, zip_path);
657 wcscat(buf, delimiter);
Just van Rossum52e14d62002-12-30 22:08:05 +0000658
Fred Drakeedabdc12000-07-08 06:16:37 +0000659 /* Next goes merge of compile-time $PYTHONPATH with
660 * dynamically located prefix.
661 */
Victor Stinnerae4836d2010-11-08 23:49:47 +0000662 defpath = _pythonpath;
Fred Drakeedabdc12000-07-08 06:16:37 +0000663 while (1) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000664 wchar_t *delim = wcschr(defpath, DELIM);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000665
Fred Drakeedabdc12000-07-08 06:16:37 +0000666 if (defpath[0] != SEP) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000667 wcscat(buf, prefix);
668 wcscat(buf, separator);
Fred Drakeedabdc12000-07-08 06:16:37 +0000669 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000670
Fred Drakeedabdc12000-07-08 06:16:37 +0000671 if (delim) {
672 size_t len = delim - defpath + 1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000673 size_t end = wcslen(buf) + len;
674 wcsncat(buf, defpath, len);
Fred Drakeedabdc12000-07-08 06:16:37 +0000675 *(buf + end) = '\0';
676 }
677 else {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000678 wcscat(buf, defpath);
Fred Drakeedabdc12000-07-08 06:16:37 +0000679 break;
680 }
681 defpath = delim + 1;
682 }
Martin v. Löwis790465f2008-04-05 20:41:37 +0000683 wcscat(buf, delimiter);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000684
Fred Drakeedabdc12000-07-08 06:16:37 +0000685 /* Finally, on goes the directory for dynamic-load modules */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000686 wcscat(buf, exec_prefix);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000687
Fred Drakeedabdc12000-07-08 06:16:37 +0000688 /* And publish the results */
689 module_search_path = buf;
Antoine Pitroubb6eb852011-03-17 23:34:33 +0100690 module_search_path_malloced = 1;
Fred Drakeedabdc12000-07-08 06:16:37 +0000691 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000692
Fred Drakeedabdc12000-07-08 06:16:37 +0000693 /* Reduce prefix and exec_prefix to their essence,
694 * e.g. /usr/local/lib/python1.5 is reduced to /usr/local.
695 * If we're loading relative to the build directory,
696 * return the compiled-in defaults instead.
697 */
698 if (pfound > 0) {
699 reduce(prefix);
700 reduce(prefix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 /* The prefix is the root directory, but reduce() chopped
702 * off the "/". */
703 if (!prefix[0])
704 wcscpy(prefix, separator);
Fred Drakeedabdc12000-07-08 06:16:37 +0000705 }
706 else
Victor Stinnerae4836d2010-11-08 23:49:47 +0000707 wcsncpy(prefix, _prefix, MAXPATHLEN);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000708
Fred Drakeedabdc12000-07-08 06:16:37 +0000709 if (efound > 0) {
710 reduce(exec_prefix);
711 reduce(exec_prefix);
712 reduce(exec_prefix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000713 if (!exec_prefix[0])
714 wcscpy(exec_prefix, separator);
Fred Drakeedabdc12000-07-08 06:16:37 +0000715 }
716 else
Victor Stinnerae4836d2010-11-08 23:49:47 +0000717 wcsncpy(exec_prefix, _exec_prefix, MAXPATHLEN);
718
719 PyMem_Free(_pythonpath);
720 PyMem_Free(_prefix);
721 PyMem_Free(_exec_prefix);
Victor Stinner99563b12011-06-20 14:45:54 +0200722 if (rtpypath != NULL)
723 PyMem_Free(rtpypath);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000724}
725
726
727/* External interface */
Kristján Valur Jónsson3b69db22010-09-27 05:32:54 +0000728void
729Py_SetPath(const wchar_t *path)
730{
731 if (module_search_path != NULL) {
Antoine Pitroubb6eb852011-03-17 23:34:33 +0100732 if (module_search_path_malloced)
733 PyMem_Free(module_search_path);
Kristján Valur Jónsson3b69db22010-09-27 05:32:54 +0000734 module_search_path = NULL;
Antoine Pitroubb6eb852011-03-17 23:34:33 +0100735 module_search_path_malloced = 0;
Kristján Valur Jónsson3b69db22010-09-27 05:32:54 +0000736 }
737 if (path != NULL) {
738 extern wchar_t *Py_GetProgramName(void);
739 wchar_t *prog = Py_GetProgramName();
740 wcsncpy(progpath, prog, MAXPATHLEN);
741 exec_prefix[0] = prefix[0] = L'\0';
Antoine Pitroubb6eb852011-03-17 23:34:33 +0100742 module_search_path = PyMem_Malloc((wcslen(path) + 1) * sizeof(wchar_t));
743 module_search_path_malloced = 1;
Kristján Valur Jónsson3b69db22010-09-27 05:32:54 +0000744 if (module_search_path != NULL)
745 wcscpy(module_search_path, path);
746 }
747}
Guido van Rossum667d7041995-08-04 04:20:48 +0000748
Martin v. Löwis790465f2008-04-05 20:41:37 +0000749wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000750Py_GetPath(void)
Guido van Rossum667d7041995-08-04 04:20:48 +0000751{
Fred Drakeedabdc12000-07-08 06:16:37 +0000752 if (!module_search_path)
753 calculate_path();
754 return module_search_path;
Guido van Rossum667d7041995-08-04 04:20:48 +0000755}
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000756
Martin v. Löwis790465f2008-04-05 20:41:37 +0000757wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000758Py_GetPrefix(void)
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000759{
Fred Drakeedabdc12000-07-08 06:16:37 +0000760 if (!module_search_path)
761 calculate_path();
762 return prefix;
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000763}
764
Martin v. Löwis790465f2008-04-05 20:41:37 +0000765wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000766Py_GetExecPrefix(void)
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000767{
Fred Drakeedabdc12000-07-08 06:16:37 +0000768 if (!module_search_path)
769 calculate_path();
770 return exec_prefix;
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000771}
Guido van Rossum7929c6f1997-05-20 22:38:21 +0000772
Martin v. Löwis790465f2008-04-05 20:41:37 +0000773wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000774Py_GetProgramFullPath(void)
Guido van Rossum7929c6f1997-05-20 22:38:21 +0000775{
Fred Drakeedabdc12000-07-08 06:16:37 +0000776 if (!module_search_path)
777 calculate_path();
778 return progpath;
Guido van Rossum7929c6f1997-05-20 22:38:21 +0000779}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000780
781
782#ifdef __cplusplus
783}
784#endif
785