blob: 0d6d5a0f9ff6e9c03e9e8f95009d2ac8dcee0b62 [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
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000397#ifdef HAVE_DLOPEN
398static PyObject *
399sys_setdlopenflags(PyObject *self, PyObject *args)
400{
401 int new_val;
402 PyThreadState *tstate = PyThreadState_Get();
403 if (!PyArg_ParseTuple(args, "i:setdlopenflags", &new_val))
404 return NULL;
405 if (!tstate)
406 return NULL;
407 tstate->interp->dlopenflags = new_val;
408 Py_INCREF(Py_None);
409 return Py_None;
410}
411
412static char setdlopenflags_doc[] =
413"setdlopenflags(n) -> None\n\
414\n\
415Set the flags that will be used for dlopen() calls. Among other\n\
416things, this will enable a lazy resolving of symbols when imporing\n\
417a module, if called as sys.setdlopenflags(0)\n\
418To share symols across extension modules, call as\n\
419sys.setdlopenflags(dl.RTLD_NOW|dl.RTLD_GLOBAL)";
420
421static PyObject *
422sys_getdlopenflags(PyObject *self, PyObject *args)
423{
424 PyThreadState *tstate = PyThreadState_Get();
425 if (!PyArg_ParseTuple(args, ":getdlopenflags"))
426 return NULL;
427 if (!tstate)
428 return NULL;
429 return PyInt_FromLong(tstate->interp->dlopenflags);
430}
431
432static char getdlopenflags_doc[] =
433"getdlopenflags() -> int\n\
434\n\
435Return the current value of the flags that are used for dlopen()\n\
436calls. The flag constants are defined in the dl module.";
437#endif
438
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000439#ifdef USE_MALLOPT
440/* Link with -lmalloc (or -lmpc) on an SGI */
441#include <malloc.h>
442
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000443static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000444sys_mdebug(PyObject *self, PyObject *args)
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000445{
446 int flag;
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000447 if (!PyArg_ParseTuple(args, "i:mdebug", &flag))
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000448 return NULL;
449 mallopt(M_DEBUG, flag);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000450 Py_INCREF(Py_None);
451 return Py_None;
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000452}
453#endif /* USE_MALLOPT */
454
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000455static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000456sys_getrefcount(PyObject *self, PyObject *args)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000457{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000458 PyObject *arg;
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000459 if (!PyArg_ParseTuple(args, "O:getrefcount", &arg))
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000460 return NULL;
Mark Hammond440d8982000-06-20 08:12:48 +0000461 return PyInt_FromLong(arg->ob_refcnt);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000462}
463
Mark Hammond440d8982000-06-20 08:12:48 +0000464#ifdef Py_TRACE_REFS
465static PyObject *
466sys_gettotalrefcount(PyObject *self, PyObject *args)
467{
468 extern long _Py_RefTotal;
469 if (!PyArg_ParseTuple(args, ":gettotalrefcount"))
470 return NULL;
471 return PyInt_FromLong(_Py_RefTotal);
472}
473
474#endif /* Py_TRACE_REFS */
475
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000476static char getrefcount_doc[] =
477"getrefcount(object) -> integer\n\
478\n\
479Return the current reference count for the object. This includes the\n\
480temporary reference in the argument list, so it is at least 2.";
481
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000482#ifdef COUNT_ALLOCS
483static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000484sys_getcounts(PyObject *self, PyObject *args)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000485{
Tim Petersdbd9ba62000-07-09 03:09:57 +0000486 extern PyObject *get_counts(void);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000487
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000488 if (!PyArg_ParseTuple(args, ":getcounts"))
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000489 return NULL;
490 return get_counts();
491}
492#endif
493
Barry Warsawb6a54d22000-12-06 21:47:46 +0000494static char getframe_doc[] =
495"_getframe([depth]) -> frameobject\n\
496\n\
497Return a frame object from the call stack. If optional integer depth is\n\
498given, return the frame object that many calls below the top of the stack.\n\
499If that is deeper than the call stack, ValueError is raised. The default\n\
500for depth is zero, returning the frame at the top of the call stack.\n\
501\n\
502This function should be used for internal and specialized\n\
503purposes only.";
504
505static PyObject *
506sys_getframe(PyObject *self, PyObject *args)
507{
508 PyFrameObject *f = PyThreadState_Get()->frame;
509 int depth = -1;
510
511 if (!PyArg_ParseTuple(args, "|i:_getframe", &depth))
512 return NULL;
513
514 while (depth > 0 && f != NULL) {
515 f = f->f_back;
516 --depth;
517 }
518 if (f == NULL) {
519 PyErr_SetString(PyExc_ValueError,
520 "call stack is not deep enough");
521 return NULL;
522 }
523 Py_INCREF(f);
524 return (PyObject*)f;
525}
526
527
Guido van Rossum7f3f2c11996-05-23 22:45:41 +0000528#ifdef Py_TRACE_REFS
Guido van Rossumded690f1996-05-24 20:48:31 +0000529/* Defined in objects.c because it uses static globals if that file */
Tim Petersdbd9ba62000-07-09 03:09:57 +0000530extern PyObject *_Py_GetObjects(PyObject *, PyObject *);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000531#endif
Guido van Rossumded690f1996-05-24 20:48:31 +0000532
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000533#ifdef DYNAMIC_EXECUTION_PROFILE
534/* Defined in ceval.c because it uses static globals if that file */
Tim Petersdbd9ba62000-07-09 03:09:57 +0000535extern PyObject *_Py_GetDXProfile(PyObject *, PyObject *);
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000536#endif
537
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000538static PyMethodDef sys_methods[] = {
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000539 /* Might as well keep this in alphabetic order */
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000540 {"displayhook", sys_displayhook, 1, displayhook_doc},
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000541 {"exc_info", sys_exc_info, 1, exc_info_doc},
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000542 {"excepthook", sys_excepthook, 1, excepthook_doc},
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000543 {"exit", sys_exit, 0, exit_doc},
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000544 {"getdefaultencoding", sys_getdefaultencoding, 1,
545 getdefaultencoding_doc},
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000546#ifdef HAVE_DLOPEN
547 {"getdlopenflags", sys_getdlopenflags, 1,
548 getdlopenflags_doc},
549#endif
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000550#ifdef COUNT_ALLOCS
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000551 {"getcounts", sys_getcounts, 1},
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000552#endif
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000553#ifdef DYNAMIC_EXECUTION_PROFILE
554 {"getdxp", _Py_GetDXProfile, 1},
555#endif
Guido van Rossum7f3f2c11996-05-23 22:45:41 +0000556#ifdef Py_TRACE_REFS
Guido van Rossumded690f1996-05-24 20:48:31 +0000557 {"getobjects", _Py_GetObjects, 1},
Mark Hammond440d8982000-06-20 08:12:48 +0000558 {"gettotalrefcount", sys_gettotalrefcount, 1},
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000559#endif
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000560 {"getrefcount", sys_getrefcount, 1, getrefcount_doc},
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000561 {"getrecursionlimit", sys_getrecursionlimit, 1,
562 getrecursionlimit_doc},
Barry Warsawb6a54d22000-12-06 21:47:46 +0000563 {"_getframe", sys_getframe, 1, getframe_doc},
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000564#ifdef USE_MALLOPT
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000565 {"mdebug", sys_mdebug, 1},
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000566#endif
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000567 {"setdefaultencoding", sys_setdefaultencoding, 1,
568 setdefaultencoding_doc},
569 {"setcheckinterval", sys_setcheckinterval, 1,
570 setcheckinterval_doc},
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000571#ifdef HAVE_DLOPEN
572 {"setdlopenflags", sys_setdlopenflags, 1,
573 setdlopenflags_doc},
574#endif
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000575 {"setprofile", sys_setprofile, 0, setprofile_doc},
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000576 {"setrecursionlimit", sys_setrecursionlimit, 1,
577 setrecursionlimit_doc},
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000578 {"settrace", sys_settrace, 0, settrace_doc},
Guido van Rossum3f5da241990-12-20 15:06:42 +0000579 {NULL, NULL} /* sentinel */
580};
581
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000582static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000583list_builtin_module_names(void)
Guido van Rossum34679b71993-01-26 13:33:44 +0000584{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000585 PyObject *list = PyList_New(0);
Guido van Rossum34679b71993-01-26 13:33:44 +0000586 int i;
587 if (list == NULL)
588 return NULL;
Guido van Rossum25c649f1997-11-04 17:04:34 +0000589 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
Guido van Rossuma027efa1997-05-05 20:56:21 +0000590 PyObject *name = PyString_FromString(
Guido van Rossum25c649f1997-11-04 17:04:34 +0000591 PyImport_Inittab[i].name);
Guido van Rossum34679b71993-01-26 13:33:44 +0000592 if (name == NULL)
593 break;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000594 PyList_Append(list, name);
595 Py_DECREF(name);
Guido van Rossum34679b71993-01-26 13:33:44 +0000596 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000597 if (PyList_Sort(list) != 0) {
598 Py_DECREF(list);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000599 list = NULL;
600 }
Guido van Rossum8f49e121997-01-06 22:55:54 +0000601 if (list) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000602 PyObject *v = PyList_AsTuple(list);
603 Py_DECREF(list);
Guido van Rossum8f49e121997-01-06 22:55:54 +0000604 list = v;
605 }
Guido van Rossum34679b71993-01-26 13:33:44 +0000606 return list;
607}
608
Guido van Rossum23fff912000-12-15 22:02:05 +0000609static PyObject *warnoptions = NULL;
610
611void
612PySys_ResetWarnOptions(void)
613{
614 if (warnoptions == NULL || !PyList_Check(warnoptions))
615 return;
616 PyList_SetSlice(warnoptions, 0, PyList_GET_SIZE(warnoptions), NULL);
617}
618
619void
620PySys_AddWarnOption(char *s)
621{
622 PyObject *str;
623
624 if (warnoptions == NULL || !PyList_Check(warnoptions)) {
625 Py_XDECREF(warnoptions);
626 warnoptions = PyList_New(0);
627 if (warnoptions == NULL)
628 return;
629 }
630 str = PyString_FromString(s);
631 if (str != NULL) {
632 PyList_Append(warnoptions, str);
633 Py_DECREF(str);
634 }
635}
636
Guido van Rossum40552d01998-08-06 03:34:39 +0000637/* XXX This doc string is too long to be a single string literal in VC++ 5.0.
638 Two literals concatenated works just fine. If you have a K&R compiler
639 or other abomination that however *does* understand longer strings,
640 get rid of the !!! comment in the middle and the quotes that surround it. */
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000641static char sys_doc[] =
642"This module provides access to some objects used or maintained by the\n\
643interpreter and to functions that interact strongly with the interpreter.\n\
644\n\
645Dynamic objects:\n\
646\n\
647argv -- command line arguments; argv[0] is the script pathname if known\n\
648path -- module search path; path[0] is the script directory, else ''\n\
649modules -- dictionary of loaded modules\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000650\n\
651displayhook -- called to show results in an interactive session\n\
652excepthook -- called to handle any uncaught exception other than SystemExit\n\
653 To customize printing in an interactive session or to install a custom\n\
654 top-level exception handler, assign other functions to replace these.\n\
655\n\
656exitfunc -- if sys.exitfunc exists, this routine is called when Python exits\n\
657 Assigning to sys.exitfunc is deprecated; use the atexit module instead.\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000658\n\
659stdin -- standard input file object; used by raw_input() and input()\n\
660stdout -- standard output file object; used by the print statement\n\
661stderr -- standard error object; used for error messages\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000662 By assigning other file objects (or objects that behave like files)\n\
663 to these, it is possible to redirect all of the interpreter's I/O.\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000664\n\
665last_type -- type of last uncaught exception\n\
666last_value -- value of last uncaught exception\n\
667last_traceback -- traceback of last uncaught exception\n\
668 These three are only available in an interactive session after a\n\
669 traceback has been printed.\n\
670\n\
671exc_type -- type of exception currently being handled\n\
672exc_value -- value of exception currently being handled\n\
673exc_traceback -- traceback of exception currently being handled\n\
674 The function exc_info() should be used instead of these three,\n\
675 because it is thread-safe.\n\
Guido van Rossuma71b5f41999-01-14 19:07:00 +0000676"
677#ifndef MS_WIN16
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000678/* concatenating string here */
Guido van Rossuma71b5f41999-01-14 19:07:00 +0000679"\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000680Static objects:\n\
681\n\
682maxint -- the largest supported integer (the smallest is -maxint-1)\n\
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000683maxunicode -- the largest supported character\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000684builtin_module_names -- tuple of module names built into this intepreter\n\
Fred Drake801c08d2000-04-13 15:29:10 +0000685version -- the version of this interpreter as a string\n\
686version_info -- version information as a tuple\n\
687hexversion -- version information encoded as a single integer\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000688copyright -- copyright notice pertaining to this interpreter\n\
689platform -- platform identifier\n\
690executable -- pathname of this Python interpreter\n\
691prefix -- prefix used to find the Python library\n\
692exec_prefix -- prefix used to find the machine-specific Python library\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000693"
694#ifdef MS_WINDOWS
695/* concatenating string here */
696"dllhandle -- [Windows only] integer handle of the Python DLL\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000697winver -- [Windows only] version number of the Python DLL\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000698"
699#endif /* MS_WINDOWS */
700"__stdin__ -- the original stdin; don't touch!\n\
701__stdout__ -- the original stdout; don't touch!\n\
702__stderr__ -- the original stderr; don't touch!\n\
703__displayhook__ -- the original displayhook; don't touch!\n\
704__excepthook__ -- the original excepthook; don't touch!\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000705\n\
706Functions:\n\
707\n\
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000708displayhook() -- print an object to the screen, and save it in __builtin__._\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000709excepthook() -- print an exception and its traceback to sys.stderr\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000710exc_info() -- return thread-safe information about the current exception\n\
711exit() -- exit the interpreter by raising SystemExit\n\
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000712getdlopenflags() -- returns flags to be used for dlopen() calls\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000713getrefcount() -- return the reference count for an object (plus one :-)\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000714getrecursionlimit() -- return the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000715setcheckinterval() -- control how often the interpreter checks for events\n\
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000716setdlopenflags() -- set the flags to be used for dlopen() calls\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000717setprofile() -- set the global profiling function\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000718setrecursionlimit() -- set the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000719settrace() -- set the global debug tracing function\n\
Fred Drakeccede592000-08-14 20:59:57 +0000720"
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000721#endif /* MS_WIN16 */
Fred Drakeccede592000-08-14 20:59:57 +0000722/* end of sys_doc */ ;
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000723
Guido van Rossum25ce5661997-08-02 03:10:38 +0000724PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000725_PySys_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000726{
Guido van Rossum25ce5661997-08-02 03:10:38 +0000727 PyObject *m, *v, *sysdict;
728 PyObject *sysin, *sysout, *syserr;
Fred Drake6d27c1e2000-04-13 20:03:20 +0000729 char *s;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000730
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000731 m = Py_InitModule3("sys", sys_methods, sys_doc);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000732 sysdict = PyModule_GetDict(m);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000733
734 sysin = PyFile_FromFile(stdin, "<stdin>", "r", NULL);
735 sysout = PyFile_FromFile(stdout, "<stdout>", "w", NULL);
736 syserr = PyFile_FromFile(stderr, "<stderr>", "w", NULL);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000737 if (PyErr_Occurred())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000738 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000739 PyDict_SetItemString(sysdict, "stdin", sysin);
740 PyDict_SetItemString(sysdict, "stdout", sysout);
741 PyDict_SetItemString(sysdict, "stderr", syserr);
Guido van Rossumbd36dba1998-02-19 20:53:06 +0000742 /* Make backup copies for cleanup */
743 PyDict_SetItemString(sysdict, "__stdin__", sysin);
744 PyDict_SetItemString(sysdict, "__stdout__", sysout);
745 PyDict_SetItemString(sysdict, "__stderr__", syserr);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000746 PyDict_SetItemString(sysdict, "__displayhook__",
747 PyDict_GetItemString(sysdict, "displayhook"));
748 PyDict_SetItemString(sysdict, "__excepthook__",
749 PyDict_GetItemString(sysdict, "excepthook"));
Guido van Rossum25ce5661997-08-02 03:10:38 +0000750 Py_XDECREF(sysin);
751 Py_XDECREF(sysout);
752 Py_XDECREF(syserr);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000753 PyDict_SetItemString(sysdict, "version",
754 v = PyString_FromString(Py_GetVersion()));
Barry Warsaw54892c41999-01-27 16:33:19 +0000755 Py_XDECREF(v);
Guido van Rossume0d7dae1999-01-03 12:55:39 +0000756 PyDict_SetItemString(sysdict, "hexversion",
757 v = PyInt_FromLong(PY_VERSION_HEX));
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000758 Py_XDECREF(v);
Fred Drake93a20bf2000-04-13 17:44:51 +0000759 /*
760 * These release level checks are mutually exclusive and cover
761 * the field, so don't get too fancy with the pre-processor!
762 */
763#if PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_ALPHA
Fred Drake6d27c1e2000-04-13 20:03:20 +0000764 s = "alpha";
Fred Drake592f2d62000-08-31 15:21:11 +0000765#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_BETA
Fred Drake6d27c1e2000-04-13 20:03:20 +0000766 s = "beta";
Fred Drake592f2d62000-08-31 15:21:11 +0000767#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_GAMMA
Fred Drake6d27c1e2000-04-13 20:03:20 +0000768 s = "candidate";
Fred Drake592f2d62000-08-31 15:21:11 +0000769#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_FINAL
Fred Drake6d27c1e2000-04-13 20:03:20 +0000770 s = "final";
Fred Drake93a20bf2000-04-13 17:44:51 +0000771#endif
Fred Drake801c08d2000-04-13 15:29:10 +0000772 PyDict_SetItemString(sysdict, "version_info",
Fred Drake6d27c1e2000-04-13 20:03:20 +0000773 v = Py_BuildValue("iiisi", PY_MAJOR_VERSION,
Fred Drake801c08d2000-04-13 15:29:10 +0000774 PY_MINOR_VERSION,
Fred Drake6d27c1e2000-04-13 20:03:20 +0000775 PY_MICRO_VERSION, s,
Fred Drake93a20bf2000-04-13 17:44:51 +0000776 PY_RELEASE_SERIAL));
Fred Drake801c08d2000-04-13 15:29:10 +0000777 Py_XDECREF(v);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000778 PyDict_SetItemString(sysdict, "copyright",
779 v = PyString_FromString(Py_GetCopyright()));
780 Py_XDECREF(v);
781 PyDict_SetItemString(sysdict, "platform",
782 v = PyString_FromString(Py_GetPlatform()));
783 Py_XDECREF(v);
Guido van Rossumb2c8ec41997-05-22 20:41:20 +0000784 PyDict_SetItemString(sysdict, "executable",
785 v = PyString_FromString(Py_GetProgramFullPath()));
786 Py_XDECREF(v);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000787 PyDict_SetItemString(sysdict, "prefix",
788 v = PyString_FromString(Py_GetPrefix()));
789 Py_XDECREF(v);
790 PyDict_SetItemString(sysdict, "exec_prefix",
791 v = PyString_FromString(Py_GetExecPrefix()));
792 Py_XDECREF(v);
793 PyDict_SetItemString(sysdict, "maxint",
794 v = PyInt_FromLong(PyInt_GetMax()));
795 Py_XDECREF(v);
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000796 PyDict_SetItemString(sysdict, "maxunicode",
797 v = PyInt_FromLong(PyUnicode_GetMax()));
798 Py_XDECREF(v);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000799 PyDict_SetItemString(sysdict, "builtin_module_names",
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000800 v = list_builtin_module_names());
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000801 Py_XDECREF(v);
Fred Drake099325e2000-08-14 15:47:03 +0000802 {
803 /* Assumes that longs are at least 2 bytes long.
804 Should be safe! */
805 unsigned long number = 1;
Fred Drakea2b6ad62000-08-15 04:24:43 +0000806 char *value;
Fred Drake099325e2000-08-14 15:47:03 +0000807
808 s = (char *) &number;
809 if (s[0] == 0)
Fred Drakea2b6ad62000-08-15 04:24:43 +0000810 value = "big";
Fred Drake099325e2000-08-14 15:47:03 +0000811 else
Fred Drakea2b6ad62000-08-15 04:24:43 +0000812 value = "little";
813 PyDict_SetItemString(sysdict, "byteorder",
Barry Warsawf2581c92000-08-16 23:03:57 +0000814 v = PyString_FromString(value));
815 Py_XDECREF(v);
Fred Drake099325e2000-08-14 15:47:03 +0000816 }
Guido van Rossum8b9ea871996-08-23 18:14:47 +0000817#ifdef MS_COREDLL
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000818 PyDict_SetItemString(sysdict, "dllhandle",
Guido van Rossum582acec2000-06-28 22:07:35 +0000819 v = PyLong_FromVoidPtr(PyWin_DLLhModule));
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000820 Py_XDECREF(v);
821 PyDict_SetItemString(sysdict, "winver",
Guido van Rossum6c1e5f21997-09-29 23:34:23 +0000822 v = PyString_FromString(PyWin_DLLVersionString));
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000823 Py_XDECREF(v);
Guido van Rossumc606fe11996-04-09 02:37:57 +0000824#endif
Guido van Rossum23fff912000-12-15 22:02:05 +0000825 if (warnoptions == NULL) {
826 warnoptions = PyList_New(0);
827 }
828 else {
829 Py_INCREF(warnoptions);
830 }
831 if (warnoptions != NULL) {
Guido van Rossum03df3b32001-01-13 22:06:05 +0000832 PyDict_SetItemString(sysdict, "warnoptions", warnoptions);
Guido van Rossum23fff912000-12-15 22:02:05 +0000833 }
834
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000835 if (PyErr_Occurred())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000836 return NULL;
837 return m;
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000838}
839
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000840static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000841makepathobject(char *path, int delim)
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000842{
Guido van Rossum3f5da241990-12-20 15:06:42 +0000843 int i, n;
844 char *p;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000845 PyObject *v, *w;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000846
847 n = 1;
848 p = path;
849 while ((p = strchr(p, delim)) != NULL) {
850 n++;
851 p++;
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000852 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000853 v = PyList_New(n);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000854 if (v == NULL)
855 return NULL;
856 for (i = 0; ; i++) {
857 p = strchr(path, delim);
858 if (p == NULL)
859 p = strchr(path, '\0'); /* End of string */
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000860 w = PyString_FromStringAndSize(path, (int) (p - path));
Guido van Rossum3f5da241990-12-20 15:06:42 +0000861 if (w == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000862 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000863 return NULL;
864 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000865 PyList_SetItem(v, i, w);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000866 if (*p == '\0')
867 break;
868 path = p+1;
869 }
870 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000871}
872
873void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000874PySys_SetPath(char *path)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000875{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000876 PyObject *v;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000877 if ((v = makepathobject(path, DELIM)) == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000878 Py_FatalError("can't create sys.path");
879 if (PySys_SetObject("path", v) != 0)
880 Py_FatalError("can't assign sys.path");
881 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000882}
883
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000884static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000885makeargvobject(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000886{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000887 PyObject *av;
Guido van Rossumee3a2991992-01-14 18:42:53 +0000888 if (argc <= 0 || argv == NULL) {
889 /* Ensure at least one (empty) argument is seen */
890 static char *empty_argv[1] = {""};
891 argv = empty_argv;
892 argc = 1;
893 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000894 av = PyList_New(argc);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000895 if (av != NULL) {
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000896 int i;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000897 for (i = 0; i < argc; i++) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000898 PyObject *v = PyString_FromString(argv[i]);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000899 if (v == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000900 Py_DECREF(av);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000901 av = NULL;
902 break;
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000903 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000904 PyList_SetItem(av, i, v);
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000905 }
Guido van Rossum5b3138b1990-11-18 17:41:40 +0000906 }
Guido van Rossum3f5da241990-12-20 15:06:42 +0000907 return av;
908}
909
910void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000911PySys_SetArgv(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000912{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000913 PyObject *av = makeargvobject(argc, argv);
914 PyObject *path = PySys_GetObject("path");
Guido van Rossum3f5da241990-12-20 15:06:42 +0000915 if (av == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000916 Py_FatalError("no mem for sys.argv");
917 if (PySys_SetObject("argv", av) != 0)
918 Py_FatalError("can't assign sys.argv");
Guido van Rossum94a96671996-07-30 20:35:50 +0000919 if (path != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +0000920 char *argv0 = argv[0];
Guido van Rossum94a96671996-07-30 20:35:50 +0000921 char *p = NULL;
Guido van Rossumcc883411996-09-10 14:44:21 +0000922 int n = 0;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000923 PyObject *a;
Guido van Rossumc474dea1997-04-25 15:38:31 +0000924#ifdef HAVE_READLINK
925 char link[MAXPATHLEN+1];
926 char argv0copy[2*MAXPATHLEN+1];
927 int nr = 0;
928 if (argc > 0 && argv0 != NULL)
929 nr = readlink(argv0, link, MAXPATHLEN);
930 if (nr > 0) {
931 /* It's a symlink */
932 link[nr] = '\0';
933 if (link[0] == SEP)
934 argv0 = link; /* Link to absolute path */
935 else if (strchr(link, SEP) == NULL)
936 ; /* Link without path */
937 else {
938 /* Must join(dirname(argv0), link) */
939 char *q = strrchr(argv0, SEP);
940 if (q == NULL)
941 argv0 = link; /* argv0 without path */
942 else {
943 /* Must make a copy */
944 strcpy(argv0copy, argv0);
945 q = strrchr(argv0copy, SEP);
946 strcpy(q+1, link);
947 argv0 = argv0copy;
948 }
949 }
950 }
951#endif /* HAVE_READLINK */
Guido van Rossumcc883411996-09-10 14:44:21 +0000952#if SEP == '\\' /* Special case for MS filename syntax */
Guido van Rossumc474dea1997-04-25 15:38:31 +0000953 if (argc > 0 && argv0 != NULL) {
Guido van Rossumcc883411996-09-10 14:44:21 +0000954 char *q;
Guido van Rossumc474dea1997-04-25 15:38:31 +0000955 p = strrchr(argv0, SEP);
Guido van Rossumcc883411996-09-10 14:44:21 +0000956 /* Test for alternate separator */
Guido van Rossumc474dea1997-04-25 15:38:31 +0000957 q = strrchr(p ? p : argv0, '/');
Guido van Rossumcc883411996-09-10 14:44:21 +0000958 if (q != NULL)
959 p = q;
960 if (p != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +0000961 n = p + 1 - argv0;
Guido van Rossumcc883411996-09-10 14:44:21 +0000962 if (n > 1 && p[-1] != ':')
963 n--; /* Drop trailing separator */
964 }
965 }
966#else /* All other filename syntaxes */
Guido van Rossumc474dea1997-04-25 15:38:31 +0000967 if (argc > 0 && argv0 != NULL)
968 p = strrchr(argv0, SEP);
Guido van Rossumcc883411996-09-10 14:44:21 +0000969 if (p != NULL) {
Guido van Rossumbceccf52001-04-10 22:07:43 +0000970#ifndef RISCOS
Guido van Rossumc474dea1997-04-25 15:38:31 +0000971 n = p + 1 - argv0;
Guido van Rossumbceccf52001-04-10 22:07:43 +0000972#else /* don't include trailing separator */
973 n = p - argv0;
974#endif /* RISCOS */
Guido van Rossumcc883411996-09-10 14:44:21 +0000975#if SEP == '/' /* Special case for Unix filename syntax */
976 if (n > 1)
977 n--; /* Drop trailing separator */
978#endif /* Unix */
979 }
980#endif /* All others */
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000981 a = PyString_FromStringAndSize(argv0, n);
Guido van Rossum94a96671996-07-30 20:35:50 +0000982 if (a == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000983 Py_FatalError("no mem for sys.path insertion");
984 if (PyList_Insert(path, 0, a) < 0)
985 Py_FatalError("sys.path.insert(0) failed");
986 Py_DECREF(a);
Guido van Rossuma63d9f41996-07-24 01:31:37 +0000987 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000988 Py_DECREF(av);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000989}
Guido van Rossuma890e681998-05-12 14:59:24 +0000990
991
992/* APIs to write to sys.stdout or sys.stderr using a printf-like interface.
993 Adapted from code submitted by Just van Rossum.
994
995 PySys_WriteStdout(format, ...)
996 PySys_WriteStderr(format, ...)
997
998 The first function writes to sys.stdout; the second to sys.stderr. When
999 there is a problem, they write to the real (C level) stdout or stderr;
Guido van Rossum8442af31998-10-12 18:22:10 +00001000 no exceptions are raised.
Guido van Rossuma890e681998-05-12 14:59:24 +00001001
1002 Both take a printf-style format string as their first argument followed
1003 by a variable length argument list determined by the format string.
1004
1005 *** WARNING ***
1006
1007 The format should limit the total size of the formatted output string to
1008 1000 bytes. In particular, this means that no unrestricted "%s" formats
1009 should occur; these should be limited using "%.<N>s where <N> is a
1010 decimal number calculated so that <N> plus the maximum size of other
1011 formatted text does not exceed 1000 bytes. Also watch out for "%f",
1012 which can print hundreds of digits for very large numbers.
1013
1014 */
1015
1016static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001017mywrite(char *name, FILE *fp, const char *format, va_list va)
Guido van Rossuma890e681998-05-12 14:59:24 +00001018{
1019 PyObject *file;
Guido van Rossum8442af31998-10-12 18:22:10 +00001020 PyObject *error_type, *error_value, *error_traceback;
Guido van Rossuma890e681998-05-12 14:59:24 +00001021
Guido van Rossum8442af31998-10-12 18:22:10 +00001022 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +00001023 file = PySys_GetObject(name);
1024 if (file == NULL || PyFile_AsFile(file) == fp)
1025 vfprintf(fp, format, va);
1026 else {
1027 char buffer[1001];
Guido van Rossum8442af31998-10-12 18:22:10 +00001028 if (vsprintf(buffer, format, va) >= sizeof(buffer))
1029 Py_FatalError("PySys_WriteStdout/err: buffer overrun");
Guido van Rossuma890e681998-05-12 14:59:24 +00001030 if (PyFile_WriteString(buffer, file) != 0) {
1031 PyErr_Clear();
1032 fputs(buffer, fp);
1033 }
1034 }
Guido van Rossum8442af31998-10-12 18:22:10 +00001035 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +00001036}
1037
1038void
Guido van Rossuma890e681998-05-12 14:59:24 +00001039PySys_WriteStdout(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +00001040{
1041 va_list va;
1042
Guido van Rossuma890e681998-05-12 14:59:24 +00001043 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +00001044 mywrite("stdout", stdout, format, va);
1045 va_end(va);
1046}
1047
1048void
Guido van Rossuma890e681998-05-12 14:59:24 +00001049PySys_WriteStderr(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +00001050{
1051 va_list va;
1052
Guido van Rossuma890e681998-05-12 14:59:24 +00001053 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +00001054 mywrite("stderr", stderr, format, va);
1055 va_end(va);
1056}