blob: c55b3d74be129975a66ba7d55580d55e669d22b3 [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 *
93 * NOTE: Windows MSVC builds use PC/getpathp.c instead!
Barry Warsaw90126031997-04-11 20:27:03 +000094 */
Guido van Rossum305e5d01997-04-11 17:18:45 +000095
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000096#ifdef __cplusplus
97 extern "C" {
98#endif
99
100
Guido van Rossum305e5d01997-04-11 17:18:45 +0000101#ifndef VERSION
Guido van Rossumd0e74ea2001-01-23 01:57:40 +0000102#define VERSION "2.1"
Guido van Rossum305e5d01997-04-11 17:18:45 +0000103#endif
104
105#ifndef VPATH
106#define VPATH "."
Guido van Rossum667d7041995-08-04 04:20:48 +0000107#endif
108
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000109#ifndef PREFIX
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000110# ifdef __VMS
111# define PREFIX ""
112# else
113# define PREFIX "/usr/local"
114# endif
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000115#endif
116
117#ifndef EXEC_PREFIX
Guido van Rossum6e12d561996-07-30 20:36:12 +0000118#define EXEC_PREFIX PREFIX
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000119#endif
120
Guido van Rossum305e5d01997-04-11 17:18:45 +0000121#ifndef PYTHONPATH
Guido van Rossum305e5d01997-04-11 17:18:45 +0000122#define PYTHONPATH PREFIX "/lib/python" VERSION ":" \
Guido van Rossume296ced2001-09-28 20:00:29 +0000123 EXEC_PREFIX "/lib/python" VERSION "/lib-dynload"
Guido van Rossum305e5d01997-04-11 17:18:45 +0000124#endif
Guido van Rossum667d7041995-08-04 04:20:48 +0000125
Guido van Rossum305e5d01997-04-11 17:18:45 +0000126#ifndef LANDMARK
Martin v. Löwis790465f2008-04-05 20:41:37 +0000127#define LANDMARK L"os.py"
Guido van Rossum305e5d01997-04-11 17:18:45 +0000128#endif
Guido van Rossume296ced2001-09-28 20:00:29 +0000129
Martin v. Löwis790465f2008-04-05 20:41:37 +0000130static wchar_t prefix[MAXPATHLEN+1];
131static wchar_t exec_prefix[MAXPATHLEN+1];
132static wchar_t progpath[MAXPATHLEN+1];
133static wchar_t *module_search_path = NULL;
Andrew M. Kuchlingb2ceb3a2010-02-22 23:26:10 +0000134static wchar_t *lib_python = L"lib/python" VERSION;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000135
136/* In principle, this should use HAVE__WSTAT, and _wstat
137 should be detected by autoconf. However, no current
138 POSIX system provides that function, so testing for
Antoine Pitroueba57b62010-08-14 12:33:18 +0000139 it is pointless.
140 Not sure whether the MS_WINDOWS guards are necessary:
141 perhaps for cygwin/mingw builds?
142*/
Martin v. Löwis790465f2008-04-05 20:41:37 +0000143#ifndef MS_WINDOWS
144static int
145_wstat(const wchar_t* path, struct stat *buf)
146{
Victor Stinnerf2e08b32010-08-13 23:29:08 +0000147 int err;
148 char *fname;
149 fname = _Py_wchar2char(path);
150 if (fname == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000151 errno = EINVAL;
152 return -1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000153 }
Victor Stinnerf2e08b32010-08-13 23:29:08 +0000154 err = stat(fname, buf);
155 PyMem_Free(fname);
156 return err;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000157}
158#endif
159
160#ifndef MS_WINDOWS
161static wchar_t*
162_wgetcwd(wchar_t *buf, size_t size)
163{
164 char fname[PATH_MAX];
165 if (getcwd(fname, PATH_MAX) == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000166 return NULL;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000167 if (mbstowcs(buf, fname, size) >= size) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000168 errno = ERANGE;
169 return NULL;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000170 }
171 return buf;
172}
173#endif
174
175#ifdef HAVE_READLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000176int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000177_Py_wreadlink(const wchar_t *path, wchar_t *buf, size_t bufsiz)
178{
179 char cbuf[PATH_MAX];
180 char cpath[PATH_MAX];
181 int res;
182 size_t r1 = wcstombs(cpath, path, PATH_MAX);
183 if (r1 == (size_t)-1 || r1 >= PATH_MAX) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000184 errno = EINVAL;
185 return -1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000186 }
187 res = (int)readlink(cpath, cbuf, PATH_MAX);
188 if (res == -1)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000189 return -1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000190 if (res == PATH_MAX) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000191 errno = EINVAL;
192 return -1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000193 }
Hirokazu Yamamotof8247792008-09-09 13:56:11 +0000194 cbuf[res] = '\0'; /* buf will be null terminated */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000195 r1 = mbstowcs(buf, cbuf, bufsiz);
196 if (r1 == -1) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000197 errno = EINVAL;
198 return -1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000199 }
200 return (int)r1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000201
Martin v. Löwis790465f2008-04-05 20:41:37 +0000202}
203#endif
Guido van Rossum305e5d01997-04-11 17:18:45 +0000204
205static void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000206reduce(wchar_t *dir)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000207{
Martin v. Löwis790465f2008-04-05 20:41:37 +0000208 size_t i = wcslen(dir);
Fred Drakeedabdc12000-07-08 06:16:37 +0000209 while (i > 0 && dir[i] != SEP)
210 --i;
211 dir[i] = '\0';
Guido van Rossum305e5d01997-04-11 17:18:45 +0000212}
Guido van Rossumd29806c1998-01-19 22:06:22 +0000213
214
Guido van Rossum305e5d01997-04-11 17:18:45 +0000215static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000216isfile(wchar_t *filename) /* Is file, not directory */
Guido van Rossum305e5d01997-04-11 17:18:45 +0000217{
Fred Drakeedabdc12000-07-08 06:16:37 +0000218 struct stat buf;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000219 if (_wstat(filename, &buf) != 0)
Fred Drakeedabdc12000-07-08 06:16:37 +0000220 return 0;
221 if (!S_ISREG(buf.st_mode))
222 return 0;
223 return 1;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000224}
225
226
227static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000228ismodule(wchar_t *filename) /* Is module -- check for .pyc/.pyo too */
Guido van Rossumd29806c1998-01-19 22:06:22 +0000229{
Fred Drakeedabdc12000-07-08 06:16:37 +0000230 if (isfile(filename))
231 return 1;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000232
Fred Drakeedabdc12000-07-08 06:16:37 +0000233 /* Check for the compiled version of prefix. */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000234 if (wcslen(filename) < MAXPATHLEN) {
235 wcscat(filename, Py_OptimizeFlag ? L"o" : L"c");
Fred Drakeedabdc12000-07-08 06:16:37 +0000236 if (isfile(filename))
237 return 1;
238 }
239 return 0;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000240}
241
242
243static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000244isxfile(wchar_t *filename) /* Is executable file */
Guido van Rossumd29806c1998-01-19 22:06:22 +0000245{
Fred Drakeedabdc12000-07-08 06:16:37 +0000246 struct stat buf;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000247 if (_wstat(filename, &buf) != 0)
Fred Drakeedabdc12000-07-08 06:16:37 +0000248 return 0;
249 if (!S_ISREG(buf.st_mode))
250 return 0;
251 if ((buf.st_mode & 0111) == 0)
252 return 0;
253 return 1;
Guido van Rossumd29806c1998-01-19 22:06:22 +0000254}
255
256
257static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000258isdir(wchar_t *filename) /* Is directory */
Guido van Rossumd29806c1998-01-19 22:06:22 +0000259{
Fred Drakeedabdc12000-07-08 06:16:37 +0000260 struct stat buf;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000261 if (_wstat(filename, &buf) != 0)
Fred Drakeedabdc12000-07-08 06:16:37 +0000262 return 0;
263 if (!S_ISDIR(buf.st_mode))
264 return 0;
265 return 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000266}
267
268
Tim Petersec8c5a92004-08-08 01:00:47 +0000269/* Add a path component, by appending stuff to buffer.
270 buffer must have at least MAXPATHLEN + 1 bytes allocated, and contain a
271 NUL-terminated string with no more than MAXPATHLEN characters (not counting
272 the trailing NUL). It's a fatal error if it contains a string longer than
273 that (callers must be careful!). If these requirements are met, it's
274 guaranteed that buffer will still be a NUL-terminated string with no more
275 than MAXPATHLEN characters at exit. If stuff is too long, only as much of
276 stuff as fits will be appended.
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000277*/
Guido van Rossum305e5d01997-04-11 17:18:45 +0000278static void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000279joinpath(wchar_t *buffer, wchar_t *stuff)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000280{
Fred Drakeedabdc12000-07-08 06:16:37 +0000281 size_t n, k;
282 if (stuff[0] == SEP)
283 n = 0;
284 else {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000285 n = wcslen(buffer);
Fred Drakeedabdc12000-07-08 06:16:37 +0000286 if (n > 0 && buffer[n-1] != SEP && n < MAXPATHLEN)
287 buffer[n++] = SEP;
288 }
Tim Petersec8c5a92004-08-08 01:00:47 +0000289 if (n > MAXPATHLEN)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000290 Py_FatalError("buffer overflow in getpath.c's joinpath()");
Martin v. Löwis790465f2008-04-05 20:41:37 +0000291 k = wcslen(stuff);
Fred Drakeedabdc12000-07-08 06:16:37 +0000292 if (n + k > MAXPATHLEN)
293 k = MAXPATHLEN - n;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000294 wcsncpy(buffer+n, stuff, k);
Fred Drakeedabdc12000-07-08 06:16:37 +0000295 buffer[n+k] = '\0';
Guido van Rossum305e5d01997-04-11 17:18:45 +0000296}
297
Guido van Rossume296ced2001-09-28 20:00:29 +0000298/* copy_absolute requires that path be allocated at least
299 MAXPATHLEN + 1 bytes and that p be no more than MAXPATHLEN bytes. */
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000300static void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000301copy_absolute(wchar_t *path, wchar_t *p)
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000302{
Guido van Rossume296ced2001-09-28 20:00:29 +0000303 if (p[0] == SEP)
Martin v. Löwis790465f2008-04-05 20:41:37 +0000304 wcscpy(path, p);
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000305 else {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000306 _wgetcwd(path, MAXPATHLEN);
Guido van Rossume296ced2001-09-28 20:00:29 +0000307 if (p[0] == '.' && p[1] == SEP)
308 p += 2;
309 joinpath(path, p);
Jeremy Hylton7198ba92000-09-25 17:00:24 +0000310 }
311}
Guido van Rossum305e5d01997-04-11 17:18:45 +0000312
Guido van Rossume296ced2001-09-28 20:00:29 +0000313/* absolutize() requires that path be allocated at least MAXPATHLEN+1 bytes. */
314static void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000315absolutize(wchar_t *path)
Guido van Rossume296ced2001-09-28 20:00:29 +0000316{
Martin v. Löwis790465f2008-04-05 20:41:37 +0000317 wchar_t buffer[MAXPATHLEN + 1];
Guido van Rossume296ced2001-09-28 20:00:29 +0000318
319 if (path[0] == SEP)
320 return;
321 copy_absolute(buffer, path);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000322 wcscpy(path, buffer);
Guido van Rossume296ced2001-09-28 20:00:29 +0000323}
324
325/* search_for_prefix requires that argv0_path be no more than MAXPATHLEN
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000326 bytes long.
327*/
Guido van Rossum305e5d01997-04-11 17:18:45 +0000328static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000329search_for_prefix(wchar_t *argv0_path, wchar_t *home)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000330{
Fred Drakeedabdc12000-07-08 06:16:37 +0000331 size_t n;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000332 wchar_t *vpath;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000333
Fred Drakeedabdc12000-07-08 06:16:37 +0000334 /* If PYTHONHOME is set, we believe it unconditionally */
335 if (home) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000336 wchar_t *delim;
337 wcsncpy(prefix, home, MAXPATHLEN);
338 delim = wcschr(prefix, DELIM);
Fred Drakeedabdc12000-07-08 06:16:37 +0000339 if (delim)
Martin v. Löwis790465f2008-04-05 20:41:37 +0000340 *delim = L'\0';
Fred Drakeedabdc12000-07-08 06:16:37 +0000341 joinpath(prefix, lib_python);
342 joinpath(prefix, LANDMARK);
343 return 1;
344 }
Jeremy Hylton847a9962000-05-26 21:49:07 +0000345
Fred Drakeedabdc12000-07-08 06:16:37 +0000346 /* Check to see if argv[0] is in the build directory */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000347 wcscpy(prefix, argv0_path);
348 joinpath(prefix, L"Modules/Setup");
Fred Drakeedabdc12000-07-08 06:16:37 +0000349 if (isfile(prefix)) {
Neil Schemenauer6cf07022001-01-24 17:13:11 +0000350 /* Check VPATH to see if argv0_path is in the build directory. */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000351 vpath = L"" VPATH;
352 wcscpy(prefix, argv0_path);
Fred Drakeedabdc12000-07-08 06:16:37 +0000353 joinpath(prefix, vpath);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000354 joinpath(prefix, L"Lib");
Fred Drakeedabdc12000-07-08 06:16:37 +0000355 joinpath(prefix, LANDMARK);
356 if (ismodule(prefix))
357 return -1;
358 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000359
Fred Drakeedabdc12000-07-08 06:16:37 +0000360 /* Search from argv0_path, until root is found */
Guido van Rossume296ced2001-09-28 20:00:29 +0000361 copy_absolute(prefix, argv0_path);
Fred Drakeedabdc12000-07-08 06:16:37 +0000362 do {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000363 n = wcslen(prefix);
Fred Drakeedabdc12000-07-08 06:16:37 +0000364 joinpath(prefix, lib_python);
365 joinpath(prefix, LANDMARK);
366 if (ismodule(prefix))
367 return 1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000368 prefix[n] = L'\0';
Fred Drakeedabdc12000-07-08 06:16:37 +0000369 reduce(prefix);
370 } while (prefix[0]);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000371
Fred Drakeedabdc12000-07-08 06:16:37 +0000372 /* Look at configure's PREFIX */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000373 wcsncpy(prefix, L"" PREFIX, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000374 joinpath(prefix, lib_python);
375 joinpath(prefix, LANDMARK);
376 if (ismodule(prefix))
377 return 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000378
Fred Drakeedabdc12000-07-08 06:16:37 +0000379 /* Fail */
380 return 0;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000381}
382
383
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000384/* search_for_exec_prefix requires that argv0_path be no more than
Guido van Rossume296ced2001-09-28 20:00:29 +0000385 MAXPATHLEN bytes long.
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000386*/
Guido van Rossum305e5d01997-04-11 17:18:45 +0000387static int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000388search_for_exec_prefix(wchar_t *argv0_path, wchar_t *home)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000389{
Fred Drakeedabdc12000-07-08 06:16:37 +0000390 size_t n;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000391
Fred Drakeedabdc12000-07-08 06:16:37 +0000392 /* If PYTHONHOME is set, we believe it unconditionally */
393 if (home) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000394 wchar_t *delim;
395 delim = wcschr(home, DELIM);
Fred Drakeedabdc12000-07-08 06:16:37 +0000396 if (delim)
Martin v. Löwis790465f2008-04-05 20:41:37 +0000397 wcsncpy(exec_prefix, delim+1, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000398 else
Martin v. Löwis790465f2008-04-05 20:41:37 +0000399 wcsncpy(exec_prefix, home, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000400 joinpath(exec_prefix, lib_python);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000401 joinpath(exec_prefix, L"lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000402 return 1;
403 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000404
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000405 /* Check to see if argv[0] is in the build directory. "pybuilddir.txt"
406 is written by setup.py and contains the relative path to the location
407 of shared library modules. */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000408 wcscpy(exec_prefix, argv0_path);
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000409 joinpath(exec_prefix, L"pybuilddir.txt");
Fred Drakeedabdc12000-07-08 06:16:37 +0000410 if (isfile(exec_prefix)) {
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000411 FILE *f = _Py_wfopen(exec_prefix, L"rb");
412 if (f == NULL)
413 errno = 0;
414 else {
415 char buf[MAXPATHLEN+1];
416 PyObject *decoded;
417 wchar_t rel_builddir_path[MAXPATHLEN+1];
418 size_t n;
419 n = fread(buf, 1, MAXPATHLEN, f);
420 buf[n] = '\0';
421 fclose(f);
422 decoded = PyUnicode_DecodeUTF8(buf, n, "surrogateescape");
423 if (decoded != NULL) {
Victor Stinnere406ef42010-08-14 00:07:14 +0000424 n = PyUnicode_AsWideChar((PyUnicodeObject*)decoded,
425 rel_builddir_path, MAXPATHLEN);
Antoine Pitroue9b428f2010-08-13 22:25:01 +0000426 Py_DECREF(decoded);
427 if (n >= 0) {
428 rel_builddir_path[n] = L'\0';
429 wcscpy(exec_prefix, argv0_path);
430 joinpath(exec_prefix, rel_builddir_path);
431 return -1;
432 }
433 }
434 }
Fred Drakeedabdc12000-07-08 06:16:37 +0000435 }
Jeremy Hylton847a9962000-05-26 21:49:07 +0000436
Fred Drakeedabdc12000-07-08 06:16:37 +0000437 /* Search from argv0_path, until root is found */
Guido van Rossume296ced2001-09-28 20:00:29 +0000438 copy_absolute(exec_prefix, argv0_path);
Fred Drakeedabdc12000-07-08 06:16:37 +0000439 do {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000440 n = wcslen(exec_prefix);
Fred Drakeedabdc12000-07-08 06:16:37 +0000441 joinpath(exec_prefix, lib_python);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000442 joinpath(exec_prefix, L"lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000443 if (isdir(exec_prefix))
444 return 1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000445 exec_prefix[n] = L'\0';
Fred Drakeedabdc12000-07-08 06:16:37 +0000446 reduce(exec_prefix);
447 } while (exec_prefix[0]);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000448
Fred Drakeedabdc12000-07-08 06:16:37 +0000449 /* Look at configure's EXEC_PREFIX */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000450 wcsncpy(exec_prefix, L"" EXEC_PREFIX, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000451 joinpath(exec_prefix, lib_python);
Martin v. Löwis790465f2008-04-05 20:41:37 +0000452 joinpath(exec_prefix, L"lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000453 if (isdir(exec_prefix))
454 return 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000455
Fred Drakeedabdc12000-07-08 06:16:37 +0000456 /* Fail */
457 return 0;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000458}
459
460
461static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000462calculate_path(void)
Guido van Rossum305e5d01997-04-11 17:18:45 +0000463{
Martin v. Löwis790465f2008-04-05 20:41:37 +0000464 extern wchar_t *Py_GetProgramName(void);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000465
Martin v. Löwis790465f2008-04-05 20:41:37 +0000466 static wchar_t delimiter[2] = {DELIM, '\0'};
467 static wchar_t separator[2] = {SEP, '\0'};
468 wchar_t *pythonpath = L"" PYTHONPATH;
469 char *_rtpypath = Py_GETENV("PYTHONPATH"); /* XXX use wide version on Windows */
470 wchar_t rtpypath[MAXPATHLEN+1];
471 wchar_t *home = Py_GetPythonHome();
472 char *_path = getenv("PATH");
473 wchar_t wpath[MAXPATHLEN+1];
474 wchar_t *path = NULL;
475 wchar_t *prog = Py_GetProgramName();
476 wchar_t argv0_path[MAXPATHLEN+1];
477 wchar_t zip_path[MAXPATHLEN+1];
Fred Drakeedabdc12000-07-08 06:16:37 +0000478 int pfound, efound; /* 1 if found; -1 if found build directory */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000479 wchar_t *buf;
Fred Drakeedabdc12000-07-08 06:16:37 +0000480 size_t bufsz;
481 size_t prefixsz;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000482 wchar_t *defpath = pythonpath;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000483#ifdef WITH_NEXT_FRAMEWORK
Fred Drakeedabdc12000-07-08 06:16:37 +0000484 NSModule pythonModule;
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000485#endif
Jack Jansen1afd4802004-06-03 14:33:03 +0000486#ifdef __APPLE__
Skip Montanaro2b474452006-01-05 10:49:13 +0000487#if MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_4
Brett Cannonbd9c3f72005-09-16 02:43:35 +0000488 uint32_t nsexeclength = MAXPATHLEN;
Skip Montanaro2b474452006-01-05 10:49:13 +0000489#else
490 unsigned long nsexeclength = MAXPATHLEN;
491#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000492 char execpath[MAXPATHLEN+1];
Jack Jansen1afd4802004-06-03 14:33:03 +0000493#endif
Guido van Rossume296ced2001-09-28 20:00:29 +0000494
Martin v. Löwis790465f2008-04-05 20:41:37 +0000495 if (_path) {
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000496 size_t r = mbstowcs(wpath, _path, MAXPATHLEN+1);
497 path = wpath;
498 if (r == (size_t)-1 || r > MAXPATHLEN) {
499 /* Could not convert PATH, or it's too long. */
500 path = NULL;
501 }
Martin v. Löwis790465f2008-04-05 20:41:37 +0000502 }
503
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000504 /* If there is no slash in the argv0 path, then we have to
505 * assume python is on the user's $PATH, since there's no
506 * other way to find a directory to start the search from. If
507 * $PATH isn't exported, you lose.
508 */
509 if (wcschr(prog, SEP))
510 wcsncpy(progpath, prog, MAXPATHLEN);
Jack Jansen1afd4802004-06-03 14:33:03 +0000511#ifdef __APPLE__
512 /* On Mac OS X, if a script uses an interpreter of the form
513 * "#!/opt/python2.3/bin/python", the kernel only passes "python"
514 * as argv[0], which falls through to the $PATH search below.
515 * If /opt/python2.3/bin isn't in your path, or is near the end,
516 * this algorithm may incorrectly find /usr/bin/python. To work
517 * around this, we can use _NSGetExecutablePath to get a better
518 * hint of what the intended interpreter was, although this
519 * will fail if a relative path was used. but in that case,
520 * absolutize() should help us out below
521 */
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000522 else if(0 == _NSGetExecutablePath(execpath, &nsexeclength) && execpath[0] == SEP) {
523 size_t r = mbstowcs(progpath, execpath, MAXPATHLEN+1);
524 if (r == (size_t)-1 || r > MAXPATHLEN) {
525 /* Could not convert execpath, or it's too long. */
526 progpath[0] = '\0';
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000527 }
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000528 }
Brett Cannon6cc48142004-06-24 00:48:44 +0000529#endif /* __APPLE__ */
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000530 else if (path) {
531 while (1) {
532 wchar_t *delim = wcschr(path, DELIM);
Jack Jansen55070f52001-12-02 23:56:28 +0000533
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000534 if (delim) {
535 size_t len = delim - path;
536 if (len > MAXPATHLEN)
537 len = MAXPATHLEN;
538 wcsncpy(progpath, path, len);
539 *(progpath + len) = '\0';
540 }
541 else
542 wcsncpy(progpath, path, MAXPATHLEN);
Jack Jansen55070f52001-12-02 23:56:28 +0000543
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000544 joinpath(progpath, prog);
545 if (isxfile(progpath))
546 break;
Jack Jansen55070f52001-12-02 23:56:28 +0000547
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000548 if (!delim) {
549 progpath[0] = L'\0';
550 break;
551 }
552 path = delim + 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000553 }
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000554 }
555 else
556 progpath[0] = '\0';
557 if (progpath[0] != SEP && progpath[0] != '\0')
558 absolutize(progpath);
559 wcsncpy(argv0_path, progpath, MAXPATHLEN);
560 argv0_path[MAXPATHLEN] = '\0';
Jack Jansen55070f52001-12-02 23:56:28 +0000561
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000562#ifdef WITH_NEXT_FRAMEWORK
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000563 /* On Mac OS X we have a special case if we're running from a framework.
564 ** This is because the python home should be set relative to the library,
565 ** which is in the framework, not relative to the executable, which may
566 ** be outside of the framework. Except when we're in the build directory...
567 */
Fred Drakeedabdc12000-07-08 06:16:37 +0000568 pythonModule = NSModuleForSymbol(NSLookupAndBindSymbol("_Py_Initialize"));
569 /* Use dylib functions to find out where the framework was loaded from */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000570 buf = (wchar_t *)NSLibraryNameForModule(pythonModule);
Fred Drakeedabdc12000-07-08 06:16:37 +0000571 if (buf != NULL) {
572 /* We're in a framework. */
Jack Jansene925faf2001-08-15 01:14:40 +0000573 /* See if we might be in the build directory. The framework in the
574 ** build directory is incomplete, it only has the .dylib and a few
575 ** needed symlinks, it doesn't have the Lib directories and such.
576 ** If we're running with the framework from the build directory we must
577 ** be running the interpreter in the build directory, so we use the
578 ** build-directory-specific logic to find Lib and such.
579 */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000580 wcsncpy(argv0_path, buf, MAXPATHLEN);
Jack Jansene925faf2001-08-15 01:14:40 +0000581 reduce(argv0_path);
582 joinpath(argv0_path, lib_python);
583 joinpath(argv0_path, LANDMARK);
584 if (!ismodule(argv0_path)) {
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000585 /* We are in the build directory so use the name of the
586 executable - we know that the absolute path is passed */
587 wcsncpy(argv0_path, progpath, MAXPATHLEN);
Jack Jansene925faf2001-08-15 01:14:40 +0000588 }
589 else {
Antoine Pitrou99773ac2010-08-14 12:34:41 +0000590 /* Use the location of the library as the progpath */
591 wcsncpy(argv0_path, buf, MAXPATHLEN);
Jack Jansene925faf2001-08-15 01:14:40 +0000592 }
Fred Drakeedabdc12000-07-08 06:16:37 +0000593 }
Guido van Rossum54ecc3d1999-01-27 17:53:11 +0000594#endif
Guido van Rossume296ced2001-09-28 20:00:29 +0000595
Guido van Rossum305e5d01997-04-11 17:18:45 +0000596#if HAVE_READLINK
Fred Drakeedabdc12000-07-08 06:16:37 +0000597 {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000598 wchar_t tmpbuffer[MAXPATHLEN+1];
599 int linklen = _Py_wreadlink(progpath, tmpbuffer, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000600 while (linklen != -1) {
Fred Drakeedabdc12000-07-08 06:16:37 +0000601 if (tmpbuffer[0] == SEP)
Guido van Rossume296ced2001-09-28 20:00:29 +0000602 /* tmpbuffer should never be longer than MAXPATHLEN,
603 but extra check does not hurt */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000604 wcsncpy(argv0_path, tmpbuffer, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000605 else {
606 /* Interpret relative to progpath */
607 reduce(argv0_path);
608 joinpath(argv0_path, tmpbuffer);
609 }
Martin v. Löwis790465f2008-04-05 20:41:37 +0000610 linklen = _Py_wreadlink(argv0_path, tmpbuffer, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000611 }
612 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000613#endif /* HAVE_READLINK */
614
Fred Drakeedabdc12000-07-08 06:16:37 +0000615 reduce(argv0_path);
Jeremy Hylton6372fe12000-09-27 20:51:17 +0000616 /* At this point, argv0_path is guaranteed to be less than
617 MAXPATHLEN bytes long.
618 */
Guido van Rossum305e5d01997-04-11 17:18:45 +0000619
Fred Drakeedabdc12000-07-08 06:16:37 +0000620 if (!(pfound = search_for_prefix(argv0_path, home))) {
621 if (!Py_FrozenFlag)
622 fprintf(stderr,
Guido van Rossume296ced2001-09-28 20:00:29 +0000623 "Could not find platform independent libraries <prefix>\n");
Martin v. Löwis790465f2008-04-05 20:41:37 +0000624 wcsncpy(prefix, L"" PREFIX, MAXPATHLEN);
Fred Drakeedabdc12000-07-08 06:16:37 +0000625 joinpath(prefix, lib_python);
626 }
627 else
628 reduce(prefix);
Guido van Rossume296ced2001-09-28 20:00:29 +0000629
Martin v. Löwis790465f2008-04-05 20:41:37 +0000630 wcsncpy(zip_path, prefix, MAXPATHLEN);
631 zip_path[MAXPATHLEN] = L'\0';
Just van Rossum52e14d62002-12-30 22:08:05 +0000632 if (pfound > 0) { /* Use the reduced prefix returned by Py_GetPrefix() */
633 reduce(zip_path);
634 reduce(zip_path);
635 }
636 else
Martin v. Löwis790465f2008-04-05 20:41:37 +0000637 wcsncpy(zip_path, L"" PREFIX, MAXPATHLEN);
638 joinpath(zip_path, L"lib/python00.zip");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000639 bufsz = wcslen(zip_path); /* Replace "00" with version */
Just van Rossum52e14d62002-12-30 22:08:05 +0000640 zip_path[bufsz - 6] = VERSION[0];
641 zip_path[bufsz - 5] = VERSION[2];
642
Fred Drakeedabdc12000-07-08 06:16:37 +0000643 if (!(efound = search_for_exec_prefix(argv0_path, home))) {
644 if (!Py_FrozenFlag)
645 fprintf(stderr,
Guido van Rossume296ced2001-09-28 20:00:29 +0000646 "Could not find platform dependent libraries <exec_prefix>\n");
Martin v. Löwis790465f2008-04-05 20:41:37 +0000647 wcsncpy(exec_prefix, L"" EXEC_PREFIX, MAXPATHLEN);
648 joinpath(exec_prefix, L"lib/lib-dynload");
Fred Drakeedabdc12000-07-08 06:16:37 +0000649 }
650 /* If we found EXEC_PREFIX do *not* reduce it! (Yet.) */
Guido van Rossum305e5d01997-04-11 17:18:45 +0000651
Fred Drakeedabdc12000-07-08 06:16:37 +0000652 if ((!pfound || !efound) && !Py_FrozenFlag)
653 fprintf(stderr,
654 "Consider setting $PYTHONHOME to <prefix>[:<exec_prefix>]\n");
Guido van Rossum305e5d01997-04-11 17:18:45 +0000655
Fred Drakeedabdc12000-07-08 06:16:37 +0000656 /* Calculate size of return buffer.
657 */
658 bufsz = 0;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000659
Martin v. Löwis790465f2008-04-05 20:41:37 +0000660 if (_rtpypath) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000661 size_t s = mbstowcs(rtpypath, _rtpypath, sizeof(rtpypath)/sizeof(wchar_t));
662 if (s == (size_t)-1 || s >=sizeof(rtpypath))
663 /* XXX deal with errors more gracefully */
664 _rtpypath = NULL;
665 if (_rtpypath)
666 bufsz += wcslen(rtpypath) + 1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000667 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000668
Martin v. Löwis790465f2008-04-05 20:41:37 +0000669 prefixsz = wcslen(prefix) + 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000670
Fred Drakeedabdc12000-07-08 06:16:37 +0000671 while (1) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000672 wchar_t *delim = wcschr(defpath, DELIM);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000673
Fred Drakeedabdc12000-07-08 06:16:37 +0000674 if (defpath[0] != SEP)
675 /* Paths are relative to prefix */
676 bufsz += prefixsz;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000677
Fred Drakeedabdc12000-07-08 06:16:37 +0000678 if (delim)
679 bufsz += delim - defpath + 1;
680 else {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000681 bufsz += wcslen(defpath) + 1;
Fred Drakeedabdc12000-07-08 06:16:37 +0000682 break;
683 }
684 defpath = delim + 1;
685 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000686
Martin v. Löwis790465f2008-04-05 20:41:37 +0000687 bufsz += wcslen(zip_path) + 1;
688 bufsz += wcslen(exec_prefix) + 1;
Guido van Rossum305e5d01997-04-11 17:18:45 +0000689
Fred Drakeedabdc12000-07-08 06:16:37 +0000690 /* This is the only malloc call in this file */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000691 buf = (wchar_t *)PyMem_Malloc(bufsz*sizeof(wchar_t));
Guido van Rossum305e5d01997-04-11 17:18:45 +0000692
Fred Drakeedabdc12000-07-08 06:16:37 +0000693 if (buf == NULL) {
694 /* We can't exit, so print a warning and limp along */
695 fprintf(stderr, "Not enough memory for dynamic PYTHONPATH.\n");
696 fprintf(stderr, "Using default static PYTHONPATH.\n");
Martin v. Löwis790465f2008-04-05 20:41:37 +0000697 module_search_path = L"" PYTHONPATH;
Fred Drakeedabdc12000-07-08 06:16:37 +0000698 }
699 else {
700 /* Run-time value of $PYTHONPATH goes first */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000701 if (_rtpypath) {
702 wcscpy(buf, rtpypath);
703 wcscat(buf, delimiter);
Fred Drakeedabdc12000-07-08 06:16:37 +0000704 }
705 else
706 buf[0] = '\0';
Guido van Rossum305e5d01997-04-11 17:18:45 +0000707
Just van Rossum52e14d62002-12-30 22:08:05 +0000708 /* Next is the default zip path */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000709 wcscat(buf, zip_path);
710 wcscat(buf, delimiter);
Just van Rossum52e14d62002-12-30 22:08:05 +0000711
Fred Drakeedabdc12000-07-08 06:16:37 +0000712 /* Next goes merge of compile-time $PYTHONPATH with
713 * dynamically located prefix.
714 */
715 defpath = pythonpath;
716 while (1) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000717 wchar_t *delim = wcschr(defpath, DELIM);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000718
Fred Drakeedabdc12000-07-08 06:16:37 +0000719 if (defpath[0] != SEP) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000720 wcscat(buf, prefix);
721 wcscat(buf, separator);
Fred Drakeedabdc12000-07-08 06:16:37 +0000722 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000723
Fred Drakeedabdc12000-07-08 06:16:37 +0000724 if (delim) {
725 size_t len = delim - defpath + 1;
Martin v. Löwis790465f2008-04-05 20:41:37 +0000726 size_t end = wcslen(buf) + len;
727 wcsncat(buf, defpath, len);
Fred Drakeedabdc12000-07-08 06:16:37 +0000728 *(buf + end) = '\0';
729 }
730 else {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000731 wcscat(buf, defpath);
Fred Drakeedabdc12000-07-08 06:16:37 +0000732 break;
733 }
734 defpath = delim + 1;
735 }
Martin v. Löwis790465f2008-04-05 20:41:37 +0000736 wcscat(buf, delimiter);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000737
Fred Drakeedabdc12000-07-08 06:16:37 +0000738 /* Finally, on goes the directory for dynamic-load modules */
Martin v. Löwis790465f2008-04-05 20:41:37 +0000739 wcscat(buf, exec_prefix);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000740
Fred Drakeedabdc12000-07-08 06:16:37 +0000741 /* And publish the results */
742 module_search_path = buf;
743 }
Guido van Rossum305e5d01997-04-11 17:18:45 +0000744
Fred Drakeedabdc12000-07-08 06:16:37 +0000745 /* Reduce prefix and exec_prefix to their essence,
746 * e.g. /usr/local/lib/python1.5 is reduced to /usr/local.
747 * If we're loading relative to the build directory,
748 * return the compiled-in defaults instead.
749 */
750 if (pfound > 0) {
751 reduce(prefix);
752 reduce(prefix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000753 /* The prefix is the root directory, but reduce() chopped
754 * off the "/". */
755 if (!prefix[0])
756 wcscpy(prefix, separator);
Fred Drakeedabdc12000-07-08 06:16:37 +0000757 }
758 else
Martin v. Löwis790465f2008-04-05 20:41:37 +0000759 wcsncpy(prefix, L"" PREFIX, MAXPATHLEN);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000760
Fred Drakeedabdc12000-07-08 06:16:37 +0000761 if (efound > 0) {
762 reduce(exec_prefix);
763 reduce(exec_prefix);
764 reduce(exec_prefix);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000765 if (!exec_prefix[0])
766 wcscpy(exec_prefix, separator);
Fred Drakeedabdc12000-07-08 06:16:37 +0000767 }
768 else
Martin v. Löwis790465f2008-04-05 20:41:37 +0000769 wcsncpy(exec_prefix, L"" EXEC_PREFIX, MAXPATHLEN);
Guido van Rossum305e5d01997-04-11 17:18:45 +0000770}
771
772
773/* External interface */
Guido van Rossum667d7041995-08-04 04:20:48 +0000774
Martin v. Löwis790465f2008-04-05 20:41:37 +0000775wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000776Py_GetPath(void)
Guido van Rossum667d7041995-08-04 04:20:48 +0000777{
Fred Drakeedabdc12000-07-08 06:16:37 +0000778 if (!module_search_path)
779 calculate_path();
780 return module_search_path;
Guido van Rossum667d7041995-08-04 04:20:48 +0000781}
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000782
Martin v. Löwis790465f2008-04-05 20:41:37 +0000783wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000784Py_GetPrefix(void)
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000785{
Fred Drakeedabdc12000-07-08 06:16:37 +0000786 if (!module_search_path)
787 calculate_path();
788 return prefix;
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000789}
790
Martin v. Löwis790465f2008-04-05 20:41:37 +0000791wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000792Py_GetExecPrefix(void)
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000793{
Fred Drakeedabdc12000-07-08 06:16:37 +0000794 if (!module_search_path)
795 calculate_path();
796 return exec_prefix;
Guido van Rossumc34c9a51996-06-12 04:20:27 +0000797}
Guido van Rossum7929c6f1997-05-20 22:38:21 +0000798
Martin v. Löwis790465f2008-04-05 20:41:37 +0000799wchar_t *
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000800Py_GetProgramFullPath(void)
Guido van Rossum7929c6f1997-05-20 22:38:21 +0000801{
Fred Drakeedabdc12000-07-08 06:16:37 +0000802 if (!module_search_path)
803 calculate_path();
804 return progpath;
Guido van Rossum7929c6f1997-05-20 22:38:21 +0000805}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000806
807
808#ifdef __cplusplus
809}
810#endif
811