blob: 76d40bfd0badcd293f919960996014babb986a5a [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* System module */
3
4/*
5Various bits of information used by the interpreter are collected in
6module 'sys'.
Guido van Rossum3f5da241990-12-20 15:06:42 +00007Function member:
Guido van Rossumcc8914f1995-03-20 15:09:40 +00008- exit(sts): raise SystemExit
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00009Data members:
10- stdin, stdout, stderr: standard file objects
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000011- modules: the table of modules (dictionary)
Guido van Rossum3f5da241990-12-20 15:06:42 +000012- path: module search path (list of strings)
13- argv: script arguments (list of strings)
14- ps1, ps2: optional primary and secondary prompts (strings)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000015*/
16
Guido van Rossum65bf9f21997-04-29 18:33:38 +000017#include "Python.h"
Barry Warsawb6a54d22000-12-06 21:47:46 +000018#include "compile.h"
19#include "frameobject.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000020
Guido van Rossume2437a11992-03-23 18:20:18 +000021#include "osdefs.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +000022
Guido van Rossum1254d791997-05-20 15:57:25 +000023#ifdef HAVE_UNISTD_H
Guido van Rossumc474dea1997-04-25 15:38:31 +000024#include <unistd.h>
25#endif
26
Guido van Rossum9b38a141996-09-11 23:12:24 +000027#ifdef MS_COREDLL
Guido van Rossumc606fe11996-04-09 02:37:57 +000028extern void *PyWin_DLLhModule;
Guido van Rossum6c1e5f21997-09-29 23:34:23 +000029/* A string loaded from the DLL at startup: */
30extern const char *PyWin_DLLVersionString;
Guido van Rossumc606fe11996-04-09 02:37:57 +000031#endif
32
Guido van Rossum65bf9f21997-04-29 18:33:38 +000033PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000034PySys_GetObject(char *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000035{
Guido van Rossum25ce5661997-08-02 03:10:38 +000036 PyThreadState *tstate = PyThreadState_Get();
37 PyObject *sd = tstate->interp->sysdict;
Guido van Rossumbe203361999-10-05 22:17:41 +000038 if (sd == NULL)
39 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +000040 return PyDict_GetItemString(sd, name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000041}
42
43FILE *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000044PySys_GetFile(char *name, FILE *def)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000045{
46 FILE *fp = NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +000047 PyObject *v = PySys_GetObject(name);
48 if (v != NULL && PyFile_Check(v))
49 fp = PyFile_AsFile(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000050 if (fp == NULL)
51 fp = def;
52 return fp;
53}
54
55int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000056PySys_SetObject(char *name, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000057{
Guido van Rossum25ce5661997-08-02 03:10:38 +000058 PyThreadState *tstate = PyThreadState_Get();
59 PyObject *sd = tstate->interp->sysdict;
Guido van Rossum5ad58c61992-01-26 18:15:48 +000060 if (v == NULL) {
Guido van Rossum25ce5661997-08-02 03:10:38 +000061 if (PyDict_GetItemString(sd, name) == NULL)
Guido van Rossum5ad58c61992-01-26 18:15:48 +000062 return 0;
63 else
Guido van Rossum25ce5661997-08-02 03:10:38 +000064 return PyDict_DelItemString(sd, name);
Guido van Rossum5ad58c61992-01-26 18:15:48 +000065 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000066 else
Guido van Rossum25ce5661997-08-02 03:10:38 +000067 return PyDict_SetItemString(sd, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000068}
69
Guido van Rossum65bf9f21997-04-29 18:33:38 +000070static PyObject *
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000071sys_displayhook(PyObject *self, PyObject *args)
72{
Greg Steinceb9b7c2001-01-11 09:27:34 +000073 PyObject *o, *outf;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000074 PyInterpreterState *interp = PyThreadState_Get()->interp;
75 PyObject *modules = interp->modules;
76 PyObject *builtins = PyDict_GetItemString(modules, "__builtin__");
77
78 /* parse arguments */
79 if (!PyArg_ParseTuple(args, "O:displayhook", &o))
80 return NULL;
81
82 /* Print value except if None */
83 /* After printing, also assign to '_' */
84 /* Before, set '_' to None to avoid recursion */
85 if (o == Py_None) {
86 Py_INCREF(Py_None);
87 return Py_None;
88 }
89 if (PyObject_SetAttrString(builtins, "_", Py_None) != 0)
90 return NULL;
91 if (Py_FlushLine() != 0)
92 return NULL;
Greg Steinceb9b7c2001-01-11 09:27:34 +000093 outf = PySys_GetObject("stdout");
94 if (outf == NULL) {
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000095 PyErr_SetString(PyExc_RuntimeError, "lost sys.stdout");
96 return NULL;
97 }
Greg Steinceb9b7c2001-01-11 09:27:34 +000098 if (PyFile_WriteObject(o, outf, 0) != 0)
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000099 return NULL;
Greg Steinceb9b7c2001-01-11 09:27:34 +0000100 PyFile_SoftSpace(outf, 1);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000101 if (Py_FlushLine() != 0)
102 return NULL;
103 if (PyObject_SetAttrString(builtins, "_", o) != 0)
104 return NULL;
105 Py_INCREF(Py_None);
106 return Py_None;
107}
108
109static char displayhook_doc[] =
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000110"displayhook(object) -> None\n"
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000111"\n"
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000112"Print an object to sys.stdout and also save it in __builtin__._\n";
113
114static PyObject *
115sys_excepthook(PyObject* self, PyObject* args)
116{
117 PyObject *exc, *value, *tb;
118 if (!PyArg_ParseTuple(args, "OOO:excepthook", &exc, &value, &tb))
119 return NULL;
120 PyErr_Display(exc, value, tb);
121 Py_INCREF(Py_None);
122 return Py_None;
123}
124
125static char excepthook_doc[] =
126"excepthook(exctype, value, traceback) -> None\n"
127"\n"
128"Handle an exception by displaying it with a traceback on sys.stderr.\n";
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000129
130static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000131sys_exc_info(PyObject *self, PyObject *args)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000132{
133 PyThreadState *tstate;
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000134 if (!PyArg_ParseTuple(args, ":exc_info"))
Guido van Rossuma027efa1997-05-05 20:56:21 +0000135 return NULL;
136 tstate = PyThreadState_Get();
Guido van Rossuma027efa1997-05-05 20:56:21 +0000137 return Py_BuildValue(
138 "(OOO)",
139 tstate->exc_type != NULL ? tstate->exc_type : Py_None,
140 tstate->exc_value != NULL ? tstate->exc_value : Py_None,
141 tstate->exc_traceback != NULL ?
142 tstate->exc_traceback : Py_None);
143}
144
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000145static char exc_info_doc[] =
146"exc_info() -> (type, value, traceback)\n\
147\n\
148Return information about the exception that is currently being handled.\n\
149This should be called from inside an except clause only.";
150
Guido van Rossuma027efa1997-05-05 20:56:21 +0000151static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000152sys_exit(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000153{
Guido van Rossum6a468bf1991-12-31 13:15:35 +0000154 /* Raise SystemExit so callers may catch it or clean up. */
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000155 PyErr_SetObject(PyExc_SystemExit, args);
Guido van Rossum6a468bf1991-12-31 13:15:35 +0000156 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000157}
158
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000159static char exit_doc[] =
160"exit([status])\n\
161\n\
162Exit the interpreter by raising SystemExit(status).\n\
163If the status is omitted or None, it defaults to zero (i.e., success).\n\
164If the status numeric, it will be used as the system exit status.\n\
165If it is another kind of object, it will be printed and the system\n\
166exit status will be one (i.e., failure).";
167
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000168static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000169sys_getdefaultencoding(PyObject *self, PyObject *args)
Fred Drake8b4d01d2000-05-09 19:57:01 +0000170{
Marc-André Lemburg99964b82000-06-07 09:13:41 +0000171 if (!PyArg_ParseTuple(args, ":getdefaultencoding"))
Fred Drake8b4d01d2000-05-09 19:57:01 +0000172 return NULL;
173 return PyString_FromString(PyUnicode_GetDefaultEncoding());
174}
175
Marc-André Lemburg99964b82000-06-07 09:13:41 +0000176static char getdefaultencoding_doc[] =
177"getdefaultencoding() -> string\n\
Fred Drake8b4d01d2000-05-09 19:57:01 +0000178\n\
179Return the current default string encoding used by the Unicode \n\
180implementation.";
181
182static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000183sys_setdefaultencoding(PyObject *self, PyObject *args)
Fred Drake8b4d01d2000-05-09 19:57:01 +0000184{
185 char *encoding;
Marc-André Lemburg99964b82000-06-07 09:13:41 +0000186 if (!PyArg_ParseTuple(args, "s:setdefaultencoding", &encoding))
Fred Drake8b4d01d2000-05-09 19:57:01 +0000187 return NULL;
188 if (PyUnicode_SetDefaultEncoding(encoding))
189 return NULL;
190 Py_INCREF(Py_None);
191 return Py_None;
192}
193
Marc-André Lemburg99964b82000-06-07 09:13:41 +0000194static char setdefaultencoding_doc[] =
195"setdefaultencoding(encoding)\n\
Fred Drake8b4d01d2000-05-09 19:57:01 +0000196\n\
197Set the current default string encoding used by the Unicode implementation.";
198
Fred Drake5755ce62001-06-27 19:19:46 +0000199/*
200 * Cached interned string objects used for calling the profile and
201 * trace functions. Initialized by trace_init().
202 */
203static PyObject *whatstrings[4] = {NULL, NULL, NULL, NULL};
204
205static int
206trace_init(void)
207{
208 static char *whatnames[4] = {"call", "exception", "line", "return"};
209 PyObject *name;
210 int i;
211 for (i = 0; i < 4; ++i) {
212 if (whatstrings[i] == NULL) {
213 name = PyString_InternFromString(whatnames[i]);
214 if (name == NULL)
215 return -1;
216 whatstrings[i] = name;
217 }
218 }
219 return 0;
220}
221
222
223static PyObject *
224call_trampoline(PyThreadState *tstate, PyObject* callback,
225 PyFrameObject *frame, int what, PyObject *arg)
226{
227 PyObject *args = PyTuple_New(3);
228 PyObject *whatstr;
229 PyObject *result;
230
231 if (args == NULL)
232 return NULL;
233 Py_INCREF(frame);
234 whatstr = whatstrings[what];
235 Py_INCREF(whatstr);
236 if (arg == NULL)
237 arg = Py_None;
238 Py_INCREF(arg);
239 PyTuple_SET_ITEM(args, 0, (PyObject *)frame);
240 PyTuple_SET_ITEM(args, 1, whatstr);
241 PyTuple_SET_ITEM(args, 2, arg);
242
243 /* call the Python-level function */
244 PyFrame_FastToLocals(frame);
245 result = PyEval_CallObject(callback, args);
246 PyFrame_LocalsToFast(frame, 1);
247 if (result == NULL)
248 PyTraceBack_Here(frame);
249
250 /* cleanup */
251 Py_DECREF(args);
252 return result;
253}
254
255static int
256profile_trampoline(PyObject *self, PyFrameObject *frame,
257 int what, PyObject *arg)
258{
259 PyThreadState *tstate = frame->f_tstate;
260 PyObject *result;
261
262 result = call_trampoline(tstate, self, frame, what, arg);
263 if (result == NULL) {
264 PyEval_SetProfile(NULL, NULL);
265 return -1;
266 }
267 Py_DECREF(result);
268 return 0;
269}
270
271static int
272trace_trampoline(PyObject *self, PyFrameObject *frame,
273 int what, PyObject *arg)
274{
275 PyThreadState *tstate = frame->f_tstate;
276 PyObject *callback;
277 PyObject *result;
278
279 if (what == PyTrace_CALL)
280 callback = self;
281 else
282 callback = frame->f_trace;
283 if (callback == NULL)
284 return 0;
285 result = call_trampoline(tstate, callback, frame, what, arg);
286 if (result == NULL) {
287 PyEval_SetTrace(NULL, NULL);
288 Py_XDECREF(frame->f_trace);
289 frame->f_trace = NULL;
290 return -1;
291 }
292 if (result != Py_None) {
293 PyObject *temp = frame->f_trace;
294 frame->f_trace = NULL;
295 Py_XDECREF(temp);
296 frame->f_trace = result;
297 }
298 else {
299 Py_DECREF(result);
300 }
301 return 0;
302}
Fred Draked0838392001-06-16 21:02:31 +0000303
Fred Drake8b4d01d2000-05-09 19:57:01 +0000304static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000305sys_settrace(PyObject *self, PyObject *args)
Guido van Rossume2437a11992-03-23 18:20:18 +0000306{
Fred Drake5755ce62001-06-27 19:19:46 +0000307 if (trace_init() == -1)
Fred Draked0838392001-06-16 21:02:31 +0000308 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000309 if (args == Py_None)
Fred Drake5755ce62001-06-27 19:19:46 +0000310 PyEval_SetTrace(NULL, NULL);
Guido van Rossume765f7d1992-04-05 14:17:55 +0000311 else
Fred Drake5755ce62001-06-27 19:19:46 +0000312 PyEval_SetTrace(trace_trampoline, args);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000313 Py_INCREF(Py_None);
314 return Py_None;
Guido van Rossume2437a11992-03-23 18:20:18 +0000315}
316
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000317static char settrace_doc[] =
318"settrace(function)\n\
319\n\
320Set the global debug tracing function. It will be called on each\n\
321function call. See the debugger chapter in the library manual.";
322
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000323static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000324sys_setprofile(PyObject *self, PyObject *args)
Guido van Rossume2437a11992-03-23 18:20:18 +0000325{
Fred Drake5755ce62001-06-27 19:19:46 +0000326 if (trace_init() == -1)
Fred Draked0838392001-06-16 21:02:31 +0000327 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000328 if (args == Py_None)
Fred Drake5755ce62001-06-27 19:19:46 +0000329 PyEval_SetProfile(NULL, NULL);
Guido van Rossume765f7d1992-04-05 14:17:55 +0000330 else
Fred Drake5755ce62001-06-27 19:19:46 +0000331 PyEval_SetProfile(profile_trampoline, args);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000332 Py_INCREF(Py_None);
333 return Py_None;
Guido van Rossume2437a11992-03-23 18:20:18 +0000334}
335
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000336static char setprofile_doc[] =
337"setprofile(function)\n\
338\n\
339Set the profiling function. It will be called on each function call\n\
340and return. See the profiler chapter in the library manual.";
341
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000342static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000343sys_setcheckinterval(PyObject *self, PyObject *args)
Guido van Rossuma0d7a231995-01-09 17:46:13 +0000344{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000345 PyThreadState *tstate = PyThreadState_Get();
Guido van Rossum43713e52000-02-29 13:59:29 +0000346 if (!PyArg_ParseTuple(args, "i:setcheckinterval", &tstate->interp->checkinterval))
Guido van Rossuma0d7a231995-01-09 17:46:13 +0000347 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000348 Py_INCREF(Py_None);
349 return Py_None;
Guido van Rossuma0d7a231995-01-09 17:46:13 +0000350}
351
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000352static char setcheckinterval_doc[] =
353"setcheckinterval(n)\n\
354\n\
355Tell the Python interpreter to check for asynchronous events every\n\
356n instructions. This also affects how often thread switches occur.";
357
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000358static PyObject *
359sys_setrecursionlimit(PyObject *self, PyObject *args)
360{
361 int new_limit;
362 if (!PyArg_ParseTuple(args, "i:setrecursionlimit", &new_limit))
363 return NULL;
364 if (new_limit <= 0) {
365 PyErr_SetString(PyExc_ValueError,
366 "recursion limit must be positive");
367 return NULL;
368 }
369 Py_SetRecursionLimit(new_limit);
370 Py_INCREF(Py_None);
371 return Py_None;
372}
373
374static char setrecursionlimit_doc[] =
375"setrecursionlimit(n)\n\
376\n\
377Set the maximum depth of the Python interpreter stack to n. This\n\
378limit prevents infinite recursion from causing an overflow of the C\n\
379stack and crashing Python. The highest possible limit is platform-\n\
380dependent.";
381
382static PyObject *
383sys_getrecursionlimit(PyObject *self, PyObject *args)
384{
385 if (!PyArg_ParseTuple(args, ":getrecursionlimit"))
386 return NULL;
387 return PyInt_FromLong(Py_GetRecursionLimit());
388}
389
390static char getrecursionlimit_doc[] =
391"getrecursionlimit()\n\
392\n\
393Return the current value of the recursion limit, the maximum depth\n\
394of the Python interpreter stack. This limit prevents infinite\n\
395recursion from causing an overflow of the C stack and crashing Python.";
396
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000397#ifdef USE_MALLOPT
398/* Link with -lmalloc (or -lmpc) on an SGI */
399#include <malloc.h>
400
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000401static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000402sys_mdebug(PyObject *self, PyObject *args)
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000403{
404 int flag;
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000405 if (!PyArg_ParseTuple(args, "i:mdebug", &flag))
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000406 return NULL;
407 mallopt(M_DEBUG, flag);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000408 Py_INCREF(Py_None);
409 return Py_None;
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000410}
411#endif /* USE_MALLOPT */
412
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000413static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000414sys_getrefcount(PyObject *self, PyObject *args)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000415{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000416 PyObject *arg;
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000417 if (!PyArg_ParseTuple(args, "O:getrefcount", &arg))
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000418 return NULL;
Mark Hammond440d8982000-06-20 08:12:48 +0000419 return PyInt_FromLong(arg->ob_refcnt);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000420}
421
Mark Hammond440d8982000-06-20 08:12:48 +0000422#ifdef Py_TRACE_REFS
423static PyObject *
424sys_gettotalrefcount(PyObject *self, PyObject *args)
425{
426 extern long _Py_RefTotal;
427 if (!PyArg_ParseTuple(args, ":gettotalrefcount"))
428 return NULL;
429 return PyInt_FromLong(_Py_RefTotal);
430}
431
432#endif /* Py_TRACE_REFS */
433
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000434static char getrefcount_doc[] =
435"getrefcount(object) -> integer\n\
436\n\
437Return the current reference count for the object. This includes the\n\
438temporary reference in the argument list, so it is at least 2.";
439
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000440#ifdef COUNT_ALLOCS
441static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000442sys_getcounts(PyObject *self, PyObject *args)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000443{
Tim Petersdbd9ba62000-07-09 03:09:57 +0000444 extern PyObject *get_counts(void);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000445
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000446 if (!PyArg_ParseTuple(args, ":getcounts"))
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000447 return NULL;
448 return get_counts();
449}
450#endif
451
Barry Warsawb6a54d22000-12-06 21:47:46 +0000452static char getframe_doc[] =
453"_getframe([depth]) -> frameobject\n\
454\n\
455Return a frame object from the call stack. If optional integer depth is\n\
456given, return the frame object that many calls below the top of the stack.\n\
457If that is deeper than the call stack, ValueError is raised. The default\n\
458for depth is zero, returning the frame at the top of the call stack.\n\
459\n\
460This function should be used for internal and specialized\n\
461purposes only.";
462
463static PyObject *
464sys_getframe(PyObject *self, PyObject *args)
465{
466 PyFrameObject *f = PyThreadState_Get()->frame;
467 int depth = -1;
468
469 if (!PyArg_ParseTuple(args, "|i:_getframe", &depth))
470 return NULL;
471
472 while (depth > 0 && f != NULL) {
473 f = f->f_back;
474 --depth;
475 }
476 if (f == NULL) {
477 PyErr_SetString(PyExc_ValueError,
478 "call stack is not deep enough");
479 return NULL;
480 }
481 Py_INCREF(f);
482 return (PyObject*)f;
483}
484
485
Guido van Rossum7f3f2c11996-05-23 22:45:41 +0000486#ifdef Py_TRACE_REFS
Guido van Rossumded690f1996-05-24 20:48:31 +0000487/* Defined in objects.c because it uses static globals if that file */
Tim Petersdbd9ba62000-07-09 03:09:57 +0000488extern PyObject *_Py_GetObjects(PyObject *, PyObject *);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000489#endif
Guido van Rossumded690f1996-05-24 20:48:31 +0000490
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000491#ifdef DYNAMIC_EXECUTION_PROFILE
492/* Defined in ceval.c because it uses static globals if that file */
Tim Petersdbd9ba62000-07-09 03:09:57 +0000493extern PyObject *_Py_GetDXProfile(PyObject *, PyObject *);
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000494#endif
495
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000496static PyMethodDef sys_methods[] = {
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000497 /* Might as well keep this in alphabetic order */
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000498 {"displayhook", sys_displayhook, 1, displayhook_doc},
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000499 {"exc_info", sys_exc_info, 1, exc_info_doc},
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000500 {"excepthook", sys_excepthook, 1, excepthook_doc},
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000501 {"exit", sys_exit, 0, exit_doc},
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000502 {"getdefaultencoding", sys_getdefaultencoding, 1,
503 getdefaultencoding_doc},
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000504#ifdef COUNT_ALLOCS
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000505 {"getcounts", sys_getcounts, 1},
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000506#endif
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000507#ifdef DYNAMIC_EXECUTION_PROFILE
508 {"getdxp", _Py_GetDXProfile, 1},
509#endif
Guido van Rossum7f3f2c11996-05-23 22:45:41 +0000510#ifdef Py_TRACE_REFS
Guido van Rossumded690f1996-05-24 20:48:31 +0000511 {"getobjects", _Py_GetObjects, 1},
Mark Hammond440d8982000-06-20 08:12:48 +0000512 {"gettotalrefcount", sys_gettotalrefcount, 1},
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000513#endif
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000514 {"getrefcount", sys_getrefcount, 1, getrefcount_doc},
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000515 {"getrecursionlimit", sys_getrecursionlimit, 1,
516 getrecursionlimit_doc},
Barry Warsawb6a54d22000-12-06 21:47:46 +0000517 {"_getframe", sys_getframe, 1, getframe_doc},
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000518#ifdef USE_MALLOPT
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000519 {"mdebug", sys_mdebug, 1},
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000520#endif
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000521 {"setdefaultencoding", sys_setdefaultencoding, 1,
522 setdefaultencoding_doc},
523 {"setcheckinterval", sys_setcheckinterval, 1,
524 setcheckinterval_doc},
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000525 {"setprofile", sys_setprofile, 0, setprofile_doc},
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000526 {"setrecursionlimit", sys_setrecursionlimit, 1,
527 setrecursionlimit_doc},
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000528 {"settrace", sys_settrace, 0, settrace_doc},
Guido van Rossum3f5da241990-12-20 15:06:42 +0000529 {NULL, NULL} /* sentinel */
530};
531
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000532static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000533list_builtin_module_names(void)
Guido van Rossum34679b71993-01-26 13:33:44 +0000534{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000535 PyObject *list = PyList_New(0);
Guido van Rossum34679b71993-01-26 13:33:44 +0000536 int i;
537 if (list == NULL)
538 return NULL;
Guido van Rossum25c649f1997-11-04 17:04:34 +0000539 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
Guido van Rossuma027efa1997-05-05 20:56:21 +0000540 PyObject *name = PyString_FromString(
Guido van Rossum25c649f1997-11-04 17:04:34 +0000541 PyImport_Inittab[i].name);
Guido van Rossum34679b71993-01-26 13:33:44 +0000542 if (name == NULL)
543 break;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000544 PyList_Append(list, name);
545 Py_DECREF(name);
Guido van Rossum34679b71993-01-26 13:33:44 +0000546 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000547 if (PyList_Sort(list) != 0) {
548 Py_DECREF(list);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000549 list = NULL;
550 }
Guido van Rossum8f49e121997-01-06 22:55:54 +0000551 if (list) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000552 PyObject *v = PyList_AsTuple(list);
553 Py_DECREF(list);
Guido van Rossum8f49e121997-01-06 22:55:54 +0000554 list = v;
555 }
Guido van Rossum34679b71993-01-26 13:33:44 +0000556 return list;
557}
558
Guido van Rossum23fff912000-12-15 22:02:05 +0000559static PyObject *warnoptions = NULL;
560
561void
562PySys_ResetWarnOptions(void)
563{
564 if (warnoptions == NULL || !PyList_Check(warnoptions))
565 return;
566 PyList_SetSlice(warnoptions, 0, PyList_GET_SIZE(warnoptions), NULL);
567}
568
569void
570PySys_AddWarnOption(char *s)
571{
572 PyObject *str;
573
574 if (warnoptions == NULL || !PyList_Check(warnoptions)) {
575 Py_XDECREF(warnoptions);
576 warnoptions = PyList_New(0);
577 if (warnoptions == NULL)
578 return;
579 }
580 str = PyString_FromString(s);
581 if (str != NULL) {
582 PyList_Append(warnoptions, str);
583 Py_DECREF(str);
584 }
585}
586
Guido van Rossum40552d01998-08-06 03:34:39 +0000587/* XXX This doc string is too long to be a single string literal in VC++ 5.0.
588 Two literals concatenated works just fine. If you have a K&R compiler
589 or other abomination that however *does* understand longer strings,
590 get rid of the !!! comment in the middle and the quotes that surround it. */
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000591static char sys_doc[] =
592"This module provides access to some objects used or maintained by the\n\
593interpreter and to functions that interact strongly with the interpreter.\n\
594\n\
595Dynamic objects:\n\
596\n\
597argv -- command line arguments; argv[0] is the script pathname if known\n\
598path -- module search path; path[0] is the script directory, else ''\n\
599modules -- dictionary of loaded modules\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000600\n\
601displayhook -- called to show results in an interactive session\n\
602excepthook -- called to handle any uncaught exception other than SystemExit\n\
603 To customize printing in an interactive session or to install a custom\n\
604 top-level exception handler, assign other functions to replace these.\n\
605\n\
606exitfunc -- if sys.exitfunc exists, this routine is called when Python exits\n\
607 Assigning to sys.exitfunc is deprecated; use the atexit module instead.\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000608\n\
609stdin -- standard input file object; used by raw_input() and input()\n\
610stdout -- standard output file object; used by the print statement\n\
611stderr -- standard error object; used for error messages\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000612 By assigning other file objects (or objects that behave like files)\n\
613 to these, it is possible to redirect all of the interpreter's I/O.\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000614\n\
615last_type -- type of last uncaught exception\n\
616last_value -- value of last uncaught exception\n\
617last_traceback -- traceback of last uncaught exception\n\
618 These three are only available in an interactive session after a\n\
619 traceback has been printed.\n\
620\n\
621exc_type -- type of exception currently being handled\n\
622exc_value -- value of exception currently being handled\n\
623exc_traceback -- traceback of exception currently being handled\n\
624 The function exc_info() should be used instead of these three,\n\
625 because it is thread-safe.\n\
Guido van Rossuma71b5f41999-01-14 19:07:00 +0000626"
627#ifndef MS_WIN16
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000628/* concatenating string here */
Guido van Rossuma71b5f41999-01-14 19:07:00 +0000629"\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000630Static objects:\n\
631\n\
632maxint -- the largest supported integer (the smallest is -maxint-1)\n\
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000633maxunicode -- the largest supported character\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000634builtin_module_names -- tuple of module names built into this intepreter\n\
Fred Drake801c08d2000-04-13 15:29:10 +0000635version -- the version of this interpreter as a string\n\
636version_info -- version information as a tuple\n\
637hexversion -- version information encoded as a single integer\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000638copyright -- copyright notice pertaining to this interpreter\n\
639platform -- platform identifier\n\
640executable -- pathname of this Python interpreter\n\
641prefix -- prefix used to find the Python library\n\
642exec_prefix -- prefix used to find the machine-specific Python library\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000643"
644#ifdef MS_WINDOWS
645/* concatenating string here */
646"dllhandle -- [Windows only] integer handle of the Python DLL\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000647winver -- [Windows only] version number of the Python DLL\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000648"
649#endif /* MS_WINDOWS */
650"__stdin__ -- the original stdin; don't touch!\n\
651__stdout__ -- the original stdout; don't touch!\n\
652__stderr__ -- the original stderr; don't touch!\n\
653__displayhook__ -- the original displayhook; don't touch!\n\
654__excepthook__ -- the original excepthook; don't touch!\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000655\n\
656Functions:\n\
657\n\
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000658displayhook() -- print an object to the screen, and save it in __builtin__._\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000659excepthook() -- print an exception and its traceback to sys.stderr\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000660exc_info() -- return thread-safe information about the current exception\n\
661exit() -- exit the interpreter by raising SystemExit\n\
662getrefcount() -- return the reference count for an object (plus one :-)\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000663getrecursionlimit() -- return the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000664setcheckinterval() -- control how often the interpreter checks for events\n\
665setprofile() -- set the global profiling function\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000666setrecursionlimit() -- set the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000667settrace() -- set the global debug tracing function\n\
Fred Drakeccede592000-08-14 20:59:57 +0000668"
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000669#endif /* MS_WIN16 */
Fred Drakeccede592000-08-14 20:59:57 +0000670/* end of sys_doc */ ;
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000671
Guido van Rossum25ce5661997-08-02 03:10:38 +0000672PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000673_PySys_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000674{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000675 PyObject *m, *v, *sysdict;
676 PyObject *sysin, *sysout, *syserr;
Fred Drake6d27c1e2000-04-13 20:03:20 +0000677 char *s;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000678
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000679 m = Py_InitModule3("sys", sys_methods, sys_doc);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000680 sysdict = PyModule_GetDict(m);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000681
682 sysin = PyFile_FromFile(stdin, "<stdin>", "r", NULL);
683 sysout = PyFile_FromFile(stdout, "<stdout>", "w", NULL);
684 syserr = PyFile_FromFile(stderr, "<stderr>", "w", NULL);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000685 if (PyErr_Occurred())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000686 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000687 PyDict_SetItemString(sysdict, "stdin", sysin);
688 PyDict_SetItemString(sysdict, "stdout", sysout);
689 PyDict_SetItemString(sysdict, "stderr", syserr);
Guido van Rossumbd36dba1998-02-19 20:53:06 +0000690 /* Make backup copies for cleanup */
691 PyDict_SetItemString(sysdict, "__stdin__", sysin);
692 PyDict_SetItemString(sysdict, "__stdout__", sysout);
693 PyDict_SetItemString(sysdict, "__stderr__", syserr);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000694 PyDict_SetItemString(sysdict, "__displayhook__",
695 PyDict_GetItemString(sysdict, "displayhook"));
696 PyDict_SetItemString(sysdict, "__excepthook__",
697 PyDict_GetItemString(sysdict, "excepthook"));
Guido van Rossum25ce5661997-08-02 03:10:38 +0000698 Py_XDECREF(sysin);
699 Py_XDECREF(sysout);
700 Py_XDECREF(syserr);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000701 PyDict_SetItemString(sysdict, "version",
702 v = PyString_FromString(Py_GetVersion()));
Barry Warsaw54892c41999-01-27 16:33:19 +0000703 Py_XDECREF(v);
Guido van Rossume0d7dae1999-01-03 12:55:39 +0000704 PyDict_SetItemString(sysdict, "hexversion",
705 v = PyInt_FromLong(PY_VERSION_HEX));
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000706 Py_XDECREF(v);
Fred Drake93a20bf2000-04-13 17:44:51 +0000707 /*
708 * These release level checks are mutually exclusive and cover
709 * the field, so don't get too fancy with the pre-processor!
710 */
711#if PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_ALPHA
Fred Drake6d27c1e2000-04-13 20:03:20 +0000712 s = "alpha";
Fred Drake592f2d62000-08-31 15:21:11 +0000713#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_BETA
Fred Drake6d27c1e2000-04-13 20:03:20 +0000714 s = "beta";
Fred Drake592f2d62000-08-31 15:21:11 +0000715#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_GAMMA
Fred Drake6d27c1e2000-04-13 20:03:20 +0000716 s = "candidate";
Fred Drake592f2d62000-08-31 15:21:11 +0000717#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_FINAL
Fred Drake6d27c1e2000-04-13 20:03:20 +0000718 s = "final";
Fred Drake93a20bf2000-04-13 17:44:51 +0000719#endif
Fred Drake801c08d2000-04-13 15:29:10 +0000720 PyDict_SetItemString(sysdict, "version_info",
Fred Drake6d27c1e2000-04-13 20:03:20 +0000721 v = Py_BuildValue("iiisi", PY_MAJOR_VERSION,
Fred Drake801c08d2000-04-13 15:29:10 +0000722 PY_MINOR_VERSION,
Fred Drake6d27c1e2000-04-13 20:03:20 +0000723 PY_MICRO_VERSION, s,
Fred Drake93a20bf2000-04-13 17:44:51 +0000724 PY_RELEASE_SERIAL));
Fred Drake801c08d2000-04-13 15:29:10 +0000725 Py_XDECREF(v);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000726 PyDict_SetItemString(sysdict, "copyright",
727 v = PyString_FromString(Py_GetCopyright()));
728 Py_XDECREF(v);
729 PyDict_SetItemString(sysdict, "platform",
730 v = PyString_FromString(Py_GetPlatform()));
731 Py_XDECREF(v);
Guido van Rossumb2c8ec41997-05-22 20:41:20 +0000732 PyDict_SetItemString(sysdict, "executable",
733 v = PyString_FromString(Py_GetProgramFullPath()));
734 Py_XDECREF(v);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000735 PyDict_SetItemString(sysdict, "prefix",
736 v = PyString_FromString(Py_GetPrefix()));
737 Py_XDECREF(v);
738 PyDict_SetItemString(sysdict, "exec_prefix",
739 v = PyString_FromString(Py_GetExecPrefix()));
740 Py_XDECREF(v);
741 PyDict_SetItemString(sysdict, "maxint",
742 v = PyInt_FromLong(PyInt_GetMax()));
743 Py_XDECREF(v);
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000744 PyDict_SetItemString(sysdict, "maxunicode",
745 v = PyInt_FromLong(PyUnicode_GetMax()));
746 Py_XDECREF(v);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000747 PyDict_SetItemString(sysdict, "builtin_module_names",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000748 v = list_builtin_module_names());
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000749 Py_XDECREF(v);
Fred Drake099325e2000-08-14 15:47:03 +0000750 {
751 /* Assumes that longs are at least 2 bytes long.
752 Should be safe! */
753 unsigned long number = 1;
Fred Drakea2b6ad62000-08-15 04:24:43 +0000754 char *value;
Fred Drake099325e2000-08-14 15:47:03 +0000755
756 s = (char *) &number;
757 if (s[0] == 0)
Fred Drakea2b6ad62000-08-15 04:24:43 +0000758 value = "big";
Fred Drake099325e2000-08-14 15:47:03 +0000759 else
Fred Drakea2b6ad62000-08-15 04:24:43 +0000760 value = "little";
761 PyDict_SetItemString(sysdict, "byteorder",
Barry Warsawf2581c92000-08-16 23:03:57 +0000762 v = PyString_FromString(value));
763 Py_XDECREF(v);
Fred Drake099325e2000-08-14 15:47:03 +0000764 }
Guido van Rossum8b9ea871996-08-23 18:14:47 +0000765#ifdef MS_COREDLL
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000766 PyDict_SetItemString(sysdict, "dllhandle",
Guido van Rossum582acec2000-06-28 22:07:35 +0000767 v = PyLong_FromVoidPtr(PyWin_DLLhModule));
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000768 Py_XDECREF(v);
769 PyDict_SetItemString(sysdict, "winver",
Guido van Rossum6c1e5f21997-09-29 23:34:23 +0000770 v = PyString_FromString(PyWin_DLLVersionString));
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000771 Py_XDECREF(v);
Guido van Rossumc606fe11996-04-09 02:37:57 +0000772#endif
Guido van Rossum23fff912000-12-15 22:02:05 +0000773 if (warnoptions == NULL) {
774 warnoptions = PyList_New(0);
775 }
776 else {
777 Py_INCREF(warnoptions);
778 }
779 if (warnoptions != NULL) {
Guido van Rossum03df3b32001-01-13 22:06:05 +0000780 PyDict_SetItemString(sysdict, "warnoptions", warnoptions);
Guido van Rossum23fff912000-12-15 22:02:05 +0000781 }
782
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000783 if (PyErr_Occurred())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000784 return NULL;
785 return m;
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000786}
787
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000788static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000789makepathobject(char *path, int delim)
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000790{
Guido van Rossum3f5da241990-12-20 15:06:42 +0000791 int i, n;
792 char *p;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000793 PyObject *v, *w;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000794
795 n = 1;
796 p = path;
797 while ((p = strchr(p, delim)) != NULL) {
798 n++;
799 p++;
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000800 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000801 v = PyList_New(n);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000802 if (v == NULL)
803 return NULL;
804 for (i = 0; ; i++) {
805 p = strchr(path, delim);
806 if (p == NULL)
807 p = strchr(path, '\0'); /* End of string */
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000808 w = PyString_FromStringAndSize(path, (int) (p - path));
Guido van Rossum3f5da241990-12-20 15:06:42 +0000809 if (w == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000810 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000811 return NULL;
812 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000813 PyList_SetItem(v, i, w);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000814 if (*p == '\0')
815 break;
816 path = p+1;
817 }
818 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000819}
820
821void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000822PySys_SetPath(char *path)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000823{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000824 PyObject *v;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000825 if ((v = makepathobject(path, DELIM)) == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000826 Py_FatalError("can't create sys.path");
827 if (PySys_SetObject("path", v) != 0)
828 Py_FatalError("can't assign sys.path");
829 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000830}
831
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000832static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000833makeargvobject(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000834{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000835 PyObject *av;
Guido van Rossumee3a2991992-01-14 18:42:53 +0000836 if (argc <= 0 || argv == NULL) {
837 /* Ensure at least one (empty) argument is seen */
838 static char *empty_argv[1] = {""};
839 argv = empty_argv;
840 argc = 1;
841 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000842 av = PyList_New(argc);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000843 if (av != NULL) {
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000844 int i;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000845 for (i = 0; i < argc; i++) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000846 PyObject *v = PyString_FromString(argv[i]);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000847 if (v == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000848 Py_DECREF(av);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000849 av = NULL;
850 break;
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000851 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000852 PyList_SetItem(av, i, v);
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000853 }
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000854 }
Guido van Rossum3f5da241990-12-20 15:06:42 +0000855 return av;
856}
857
858void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000859PySys_SetArgv(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000860{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000861 PyObject *av = makeargvobject(argc, argv);
862 PyObject *path = PySys_GetObject("path");
Guido van Rossum3f5da241990-12-20 15:06:42 +0000863 if (av == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000864 Py_FatalError("no mem for sys.argv");
865 if (PySys_SetObject("argv", av) != 0)
866 Py_FatalError("can't assign sys.argv");
Guido van Rossum94a96671996-07-30 20:35:50 +0000867 if (path != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +0000868 char *argv0 = argv[0];
Guido van Rossum94a96671996-07-30 20:35:50 +0000869 char *p = NULL;
Guido van Rossumcc883411996-09-10 14:44:21 +0000870 int n = 0;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000871 PyObject *a;
Guido van Rossumc474dea1997-04-25 15:38:31 +0000872#ifdef HAVE_READLINK
873 char link[MAXPATHLEN+1];
874 char argv0copy[2*MAXPATHLEN+1];
875 int nr = 0;
876 if (argc > 0 && argv0 != NULL)
877 nr = readlink(argv0, link, MAXPATHLEN);
878 if (nr > 0) {
879 /* It's a symlink */
880 link[nr] = '\0';
881 if (link[0] == SEP)
882 argv0 = link; /* Link to absolute path */
883 else if (strchr(link, SEP) == NULL)
884 ; /* Link without path */
885 else {
886 /* Must join(dirname(argv0), link) */
887 char *q = strrchr(argv0, SEP);
888 if (q == NULL)
889 argv0 = link; /* argv0 without path */
890 else {
891 /* Must make a copy */
892 strcpy(argv0copy, argv0);
893 q = strrchr(argv0copy, SEP);
894 strcpy(q+1, link);
895 argv0 = argv0copy;
896 }
897 }
898 }
899#endif /* HAVE_READLINK */
Guido van Rossumcc883411996-09-10 14:44:21 +0000900#if SEP == '\\' /* Special case for MS filename syntax */
Guido van Rossumc474dea1997-04-25 15:38:31 +0000901 if (argc > 0 && argv0 != NULL) {
Guido van Rossumcc883411996-09-10 14:44:21 +0000902 char *q;
Guido van Rossumc474dea1997-04-25 15:38:31 +0000903 p = strrchr(argv0, SEP);
Guido van Rossumcc883411996-09-10 14:44:21 +0000904 /* Test for alternate separator */
Guido van Rossumc474dea1997-04-25 15:38:31 +0000905 q = strrchr(p ? p : argv0, '/');
Guido van Rossumcc883411996-09-10 14:44:21 +0000906 if (q != NULL)
907 p = q;
908 if (p != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +0000909 n = p + 1 - argv0;
Guido van Rossumcc883411996-09-10 14:44:21 +0000910 if (n > 1 && p[-1] != ':')
911 n--; /* Drop trailing separator */
912 }
913 }
914#else /* All other filename syntaxes */
Guido van Rossumc474dea1997-04-25 15:38:31 +0000915 if (argc > 0 && argv0 != NULL)
916 p = strrchr(argv0, SEP);
Guido van Rossumcc883411996-09-10 14:44:21 +0000917 if (p != NULL) {
Guido van Rossumbceccf52001-04-10 22:07:43 +0000918#ifndef RISCOS
Guido van Rossumc474dea1997-04-25 15:38:31 +0000919 n = p + 1 - argv0;
Guido van Rossumbceccf52001-04-10 22:07:43 +0000920#else /* don't include trailing separator */
921 n = p - argv0;
922#endif /* RISCOS */
Guido van Rossumcc883411996-09-10 14:44:21 +0000923#if SEP == '/' /* Special case for Unix filename syntax */
924 if (n > 1)
925 n--; /* Drop trailing separator */
926#endif /* Unix */
927 }
928#endif /* All others */
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000929 a = PyString_FromStringAndSize(argv0, n);
Guido van Rossum94a96671996-07-30 20:35:50 +0000930 if (a == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000931 Py_FatalError("no mem for sys.path insertion");
932 if (PyList_Insert(path, 0, a) < 0)
933 Py_FatalError("sys.path.insert(0) failed");
934 Py_DECREF(a);
Guido van Rossuma63d9f41996-07-24 01:31:37 +0000935 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000936 Py_DECREF(av);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000937}
Guido van Rossuma890e681998-05-12 14:59:24 +0000938
939
940/* APIs to write to sys.stdout or sys.stderr using a printf-like interface.
941 Adapted from code submitted by Just van Rossum.
942
943 PySys_WriteStdout(format, ...)
944 PySys_WriteStderr(format, ...)
945
946 The first function writes to sys.stdout; the second to sys.stderr. When
947 there is a problem, they write to the real (C level) stdout or stderr;
Guido van Rossum8442af31998-10-12 18:22:10 +0000948 no exceptions are raised.
Guido van Rossuma890e681998-05-12 14:59:24 +0000949
950 Both take a printf-style format string as their first argument followed
951 by a variable length argument list determined by the format string.
952
953 *** WARNING ***
954
955 The format should limit the total size of the formatted output string to
956 1000 bytes. In particular, this means that no unrestricted "%s" formats
957 should occur; these should be limited using "%.<N>s where <N> is a
958 decimal number calculated so that <N> plus the maximum size of other
959 formatted text does not exceed 1000 bytes. Also watch out for "%f",
960 which can print hundreds of digits for very large numbers.
961
962 */
963
964static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000965mywrite(char *name, FILE *fp, const char *format, va_list va)
Guido van Rossuma890e681998-05-12 14:59:24 +0000966{
967 PyObject *file;
Guido van Rossum8442af31998-10-12 18:22:10 +0000968 PyObject *error_type, *error_value, *error_traceback;
Guido van Rossuma890e681998-05-12 14:59:24 +0000969
Guido van Rossum8442af31998-10-12 18:22:10 +0000970 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +0000971 file = PySys_GetObject(name);
972 if (file == NULL || PyFile_AsFile(file) == fp)
973 vfprintf(fp, format, va);
974 else {
975 char buffer[1001];
Guido van Rossum8442af31998-10-12 18:22:10 +0000976 if (vsprintf(buffer, format, va) >= sizeof(buffer))
977 Py_FatalError("PySys_WriteStdout/err: buffer overrun");
Guido van Rossuma890e681998-05-12 14:59:24 +0000978 if (PyFile_WriteString(buffer, file) != 0) {
979 PyErr_Clear();
980 fputs(buffer, fp);
981 }
982 }
Guido van Rossum8442af31998-10-12 18:22:10 +0000983 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +0000984}
985
986void
Guido van Rossuma890e681998-05-12 14:59:24 +0000987PySys_WriteStdout(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +0000988{
989 va_list va;
990
Guido van Rossuma890e681998-05-12 14:59:24 +0000991 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +0000992 mywrite("stdout", stdout, format, va);
993 va_end(va);
994}
995
996void
Guido van Rossuma890e681998-05-12 14:59:24 +0000997PySys_WriteStderr(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +0000998{
999 va_list va;
1000
Guido van Rossuma890e681998-05-12 14:59:24 +00001001 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +00001002 mywrite("stderr", stderr, format, va);
1003 va_end(va);
1004}