blob: 35834ae93e07617ab5ea61dc42bd7e8c045232e3 [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"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000018#include "code.h"
Barry Warsawb6a54d22000-12-06 21:47:46 +000019#include "frameobject.h"
Guido van Rossuma12fe4e2003-04-09 19:06:21 +000020#include "eval.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000021
Guido van Rossume2437a11992-03-23 18:20:18 +000022#include "osdefs.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +000023
Mark Hammond8696ebc2002-10-08 02:44:31 +000024#ifdef MS_WINDOWS
25#define WIN32_LEAN_AND_MEAN
Amaury Forgeot d'Arc06cfe952007-11-10 13:55:44 +000026#include <windows.h>
Mark Hammond8696ebc2002-10-08 02:44:31 +000027#endif /* MS_WINDOWS */
28
Guido van Rossum9b38a141996-09-11 23:12:24 +000029#ifdef MS_COREDLL
Guido van Rossumc606fe11996-04-09 02:37:57 +000030extern void *PyWin_DLLhModule;
Guido van Rossum6c1e5f21997-09-29 23:34:23 +000031/* A string loaded from the DLL at startup: */
32extern const char *PyWin_DLLVersionString;
Guido van Rossumc606fe11996-04-09 02:37:57 +000033#endif
34
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +000035#ifdef __VMS
36#include <unixlib.h>
37#endif
38
Martin v. Löwis5467d4c2003-05-10 07:10:12 +000039#ifdef HAVE_LANGINFO_H
40#include <locale.h>
41#include <langinfo.h>
42#endif
43
Guido van Rossum65bf9f21997-04-29 18:33:38 +000044PyObject *
Neal Norwitzf3081322007-08-25 00:32:45 +000045PySys_GetObject(const char *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000046{
Nicholas Bastine5662ae2004-03-24 22:22:12 +000047 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +000048 PyObject *sd = tstate->interp->sysdict;
Guido van Rossumbe203361999-10-05 22:17:41 +000049 if (sd == NULL)
50 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +000051 return PyDict_GetItemString(sd, name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000052}
53
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000054int
Neal Norwitzf3081322007-08-25 00:32:45 +000055PySys_SetObject(const char *name, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000056{
Nicholas Bastine5662ae2004-03-24 22:22:12 +000057 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +000058 PyObject *sd = tstate->interp->sysdict;
Guido van Rossum5ad58c61992-01-26 18:15:48 +000059 if (v == NULL) {
Guido van Rossum25ce5661997-08-02 03:10:38 +000060 if (PyDict_GetItemString(sd, name) == NULL)
Guido van Rossum5ad58c61992-01-26 18:15:48 +000061 return 0;
62 else
Guido van Rossum25ce5661997-08-02 03:10:38 +000063 return PyDict_DelItemString(sd, name);
Guido van Rossum5ad58c61992-01-26 18:15:48 +000064 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000065 else
Guido van Rossum25ce5661997-08-02 03:10:38 +000066 return PyDict_SetItemString(sd, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000067}
68
Guido van Rossum65bf9f21997-04-29 18:33:38 +000069static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000070sys_displayhook(PyObject *self, PyObject *o)
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000071{
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000072 PyObject *outf;
Nicholas Bastine5662ae2004-03-24 22:22:12 +000073 PyInterpreterState *interp = PyThreadState_GET()->interp;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000074 PyObject *modules = interp->modules;
Georg Brandl1a3284e2007-12-02 09:40:06 +000075 PyObject *builtins = PyDict_GetItemString(modules, "builtins");
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000076
Moshe Zadka03897ea2001-07-23 13:32:43 +000077 if (builtins == NULL) {
Georg Brandl1a3284e2007-12-02 09:40:06 +000078 PyErr_SetString(PyExc_RuntimeError, "lost builtins module");
Moshe Zadka03897ea2001-07-23 13:32:43 +000079 return NULL;
80 }
81
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000082 /* 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;
Greg Steinceb9b7c2001-01-11 09:27:34 +000091 outf = PySys_GetObject("stdout");
Christian Heimes2be03732007-11-15 02:26:46 +000092 if (outf == NULL || outf == Py_None) {
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000093 PyErr_SetString(PyExc_RuntimeError, "lost sys.stdout");
94 return NULL;
95 }
Greg Steinceb9b7c2001-01-11 09:27:34 +000096 if (PyFile_WriteObject(o, outf, 0) != 0)
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000097 return NULL;
Guido van Rossum79139b22007-02-09 23:20:19 +000098 if (PyFile_WriteString("\n", outf) != 0)
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +000099 return NULL;
100 if (PyObject_SetAttrString(builtins, "_", o) != 0)
101 return NULL;
102 Py_INCREF(Py_None);
103 return Py_None;
104}
105
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000106PyDoc_STRVAR(displayhook_doc,
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000107"displayhook(object) -> None\n"
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000108"\n"
Georg Brandl1a3284e2007-12-02 09:40:06 +0000109"Print an object to sys.stdout and also save it in builtins.\n"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000110);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000111
112static PyObject *
113sys_excepthook(PyObject* self, PyObject* args)
114{
115 PyObject *exc, *value, *tb;
Fred Drakea7688822001-10-24 20:47:48 +0000116 if (!PyArg_UnpackTuple(args, "excepthook", 3, 3, &exc, &value, &tb))
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000117 return NULL;
118 PyErr_Display(exc, value, tb);
119 Py_INCREF(Py_None);
120 return Py_None;
121}
122
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000123PyDoc_STRVAR(excepthook_doc,
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000124"excepthook(exctype, value, traceback) -> None\n"
125"\n"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000126"Handle an exception by displaying it with a traceback on sys.stderr.\n"
127);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000128
129static PyObject *
Guido van Rossum46d3dc32003-03-01 03:20:41 +0000130sys_exc_info(PyObject *self, PyObject *noargs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000131{
132 PyThreadState *tstate;
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000133 tstate = PyThreadState_GET();
Guido van Rossuma027efa1997-05-05 20:56:21 +0000134 return Py_BuildValue(
135 "(OOO)",
136 tstate->exc_type != NULL ? tstate->exc_type : Py_None,
137 tstate->exc_value != NULL ? tstate->exc_value : Py_None,
138 tstate->exc_traceback != NULL ?
139 tstate->exc_traceback : Py_None);
140}
141
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000142PyDoc_STRVAR(exc_info_doc,
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000143"exc_info() -> (type, value, traceback)\n\
144\n\
Guido van Rossum46d3dc32003-03-01 03:20:41 +0000145Return information about the most recent exception caught by an except\n\
146clause in the current stack frame or in an older stack frame."
147);
148
149static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000150sys_exit(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000151{
Neal Norwitz0c766a02002-03-27 13:03:09 +0000152 PyObject *exit_code = 0;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000153 if (!PyArg_UnpackTuple(args, "exit", 0, 1, &exit_code))
Neal Norwitz0c766a02002-03-27 13:03:09 +0000154 return NULL;
Guido van Rossum6a468bf1991-12-31 13:15:35 +0000155 /* Raise SystemExit so callers may catch it or clean up. */
Neal Norwitz0c766a02002-03-27 13:03:09 +0000156 PyErr_SetObject(PyExc_SystemExit, exit_code);
Guido van Rossum6a468bf1991-12-31 13:15:35 +0000157 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000158}
159
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000160PyDoc_STRVAR(exit_doc,
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000161"exit([status])\n\
162\n\
163Exit the interpreter by raising SystemExit(status).\n\
164If the status is omitted or None, it defaults to zero (i.e., success).\n\
Neil Schemenauer0f2103f2002-03-23 20:46:35 +0000165If the status is numeric, it will be used as the system exit status.\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000166If it is another kind of object, it will be printed and the system\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000167exit status will be one (i.e., failure)."
168);
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000169
Martin v. Löwis107b7da2001-11-09 20:59:39 +0000170
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000171static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000172sys_getdefaultencoding(PyObject *self)
Fred Drake8b4d01d2000-05-09 19:57:01 +0000173{
Walter Dörwald360b01a2007-06-05 20:15:52 +0000174 return PyUnicode_FromString(PyUnicode_GetDefaultEncoding());
Fred Drake8b4d01d2000-05-09 19:57:01 +0000175}
176
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000177PyDoc_STRVAR(getdefaultencoding_doc,
Marc-André Lemburg99964b82000-06-07 09:13:41 +0000178"getdefaultencoding() -> string\n\
Fred Drake8b4d01d2000-05-09 19:57:01 +0000179\n\
180Return the current default string encoding used by the Unicode \n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000181implementation."
182);
Fred Drake8b4d01d2000-05-09 19:57:01 +0000183
184static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000185sys_setdefaultencoding(PyObject *self, PyObject *args)
Fred Drake8b4d01d2000-05-09 19:57:01 +0000186{
187 char *encoding;
Marc-André Lemburg99964b82000-06-07 09:13:41 +0000188 if (!PyArg_ParseTuple(args, "s:setdefaultencoding", &encoding))
Fred Drake8b4d01d2000-05-09 19:57:01 +0000189 return NULL;
190 if (PyUnicode_SetDefaultEncoding(encoding))
191 return NULL;
192 Py_INCREF(Py_None);
193 return Py_None;
194}
195
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000196PyDoc_STRVAR(setdefaultencoding_doc,
Marc-André Lemburg99964b82000-06-07 09:13:41 +0000197"setdefaultencoding(encoding)\n\
Fred Drake8b4d01d2000-05-09 19:57:01 +0000198\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000199Set the current default string encoding used by the Unicode implementation."
200);
Fred Drake8b4d01d2000-05-09 19:57:01 +0000201
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000202static PyObject *
203sys_getfilesystemencoding(PyObject *self)
204{
205 if (Py_FileSystemDefaultEncoding)
Walter Dörwald360b01a2007-06-05 20:15:52 +0000206 return PyUnicode_FromString(Py_FileSystemDefaultEncoding);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000207 Py_INCREF(Py_None);
208 return Py_None;
209}
210
211PyDoc_STRVAR(getfilesystemencoding_doc,
212"getfilesystemencoding() -> string\n\
213\n\
214Return the encoding used to convert Unicode filenames in\n\
215operating system filenames."
216);
217
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000218
Georg Brandl66a796e2006-12-19 20:50:34 +0000219
220static PyObject *
221sys_intern(PyObject *self, PyObject *args)
222{
223 PyObject *s;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000224 if (!PyArg_ParseTuple(args, "U:intern", &s))
Georg Brandl66a796e2006-12-19 20:50:34 +0000225 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000226 if (PyUnicode_CheckExact(s)) {
Walter Dörwald2a0c0812007-06-05 20:22:04 +0000227 Py_INCREF(s);
228 PyUnicode_InternInPlace(&s);
229 return s;
230 }
231 else {
232 PyErr_Format(PyExc_TypeError,
233 "can't intern %.400s", s->ob_type->tp_name);
Georg Brandl66a796e2006-12-19 20:50:34 +0000234 return NULL;
235 }
Georg Brandl66a796e2006-12-19 20:50:34 +0000236}
237
238PyDoc_STRVAR(intern_doc,
239"intern(string) -> string\n\
240\n\
241``Intern'' the given string. This enters the string in the (global)\n\
242table of interned strings whose purpose is to speed up dictionary lookups.\n\
243Return the string itself or the previously interned string object with the\n\
244same value.");
245
246
Fred Drake5755ce62001-06-27 19:19:46 +0000247/*
248 * Cached interned string objects used for calling the profile and
249 * trace functions. Initialized by trace_init().
250 */
Nicholas Bastinc69ebe82004-03-24 21:57:10 +0000251static PyObject *whatstrings[7] = {NULL, NULL, NULL, NULL, NULL, NULL, NULL};
Fred Drake5755ce62001-06-27 19:19:46 +0000252
253static int
254trace_init(void)
255{
Nicholas Bastinc69ebe82004-03-24 21:57:10 +0000256 static char *whatnames[7] = {"call", "exception", "line", "return",
257 "c_call", "c_exception", "c_return"};
Fred Drake5755ce62001-06-27 19:19:46 +0000258 PyObject *name;
259 int i;
Nicholas Bastinc69ebe82004-03-24 21:57:10 +0000260 for (i = 0; i < 7; ++i) {
Fred Drake5755ce62001-06-27 19:19:46 +0000261 if (whatstrings[i] == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +0000262 name = PyUnicode_InternFromString(whatnames[i]);
Fred Drake5755ce62001-06-27 19:19:46 +0000263 if (name == NULL)
264 return -1;
265 whatstrings[i] = name;
266 }
267 }
268 return 0;
269}
270
271
272static PyObject *
273call_trampoline(PyThreadState *tstate, PyObject* callback,
274 PyFrameObject *frame, int what, PyObject *arg)
275{
276 PyObject *args = PyTuple_New(3);
277 PyObject *whatstr;
278 PyObject *result;
279
280 if (args == NULL)
281 return NULL;
282 Py_INCREF(frame);
283 whatstr = whatstrings[what];
284 Py_INCREF(whatstr);
285 if (arg == NULL)
286 arg = Py_None;
287 Py_INCREF(arg);
288 PyTuple_SET_ITEM(args, 0, (PyObject *)frame);
289 PyTuple_SET_ITEM(args, 1, whatstr);
290 PyTuple_SET_ITEM(args, 2, arg);
291
292 /* call the Python-level function */
293 PyFrame_FastToLocals(frame);
294 result = PyEval_CallObject(callback, args);
295 PyFrame_LocalsToFast(frame, 1);
296 if (result == NULL)
297 PyTraceBack_Here(frame);
298
299 /* cleanup */
300 Py_DECREF(args);
301 return result;
302}
303
304static int
305profile_trampoline(PyObject *self, PyFrameObject *frame,
306 int what, PyObject *arg)
307{
308 PyThreadState *tstate = frame->f_tstate;
309 PyObject *result;
310
Fred Drake8f51f542001-10-04 14:48:42 +0000311 if (arg == NULL)
312 arg = Py_None;
Fred Drake5755ce62001-06-27 19:19:46 +0000313 result = call_trampoline(tstate, self, frame, what, arg);
314 if (result == NULL) {
315 PyEval_SetProfile(NULL, NULL);
316 return -1;
317 }
318 Py_DECREF(result);
319 return 0;
320}
321
322static int
323trace_trampoline(PyObject *self, PyFrameObject *frame,
324 int what, PyObject *arg)
325{
326 PyThreadState *tstate = frame->f_tstate;
327 PyObject *callback;
328 PyObject *result;
329
330 if (what == PyTrace_CALL)
331 callback = self;
332 else
333 callback = frame->f_trace;
334 if (callback == NULL)
335 return 0;
336 result = call_trampoline(tstate, callback, frame, what, arg);
337 if (result == NULL) {
338 PyEval_SetTrace(NULL, NULL);
339 Py_XDECREF(frame->f_trace);
340 frame->f_trace = NULL;
341 return -1;
342 }
343 if (result != Py_None) {
344 PyObject *temp = frame->f_trace;
345 frame->f_trace = NULL;
346 Py_XDECREF(temp);
347 frame->f_trace = result;
348 }
349 else {
350 Py_DECREF(result);
351 }
352 return 0;
353}
Fred Draked0838392001-06-16 21:02:31 +0000354
Fred Drake8b4d01d2000-05-09 19:57:01 +0000355static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000356sys_settrace(PyObject *self, PyObject *args)
Guido van Rossume2437a11992-03-23 18:20:18 +0000357{
Fred Drake5755ce62001-06-27 19:19:46 +0000358 if (trace_init() == -1)
Fred Draked0838392001-06-16 21:02:31 +0000359 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000360 if (args == Py_None)
Fred Drake5755ce62001-06-27 19:19:46 +0000361 PyEval_SetTrace(NULL, NULL);
Guido van Rossume765f7d1992-04-05 14:17:55 +0000362 else
Fred Drake5755ce62001-06-27 19:19:46 +0000363 PyEval_SetTrace(trace_trampoline, args);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000364 Py_INCREF(Py_None);
365 return Py_None;
Guido van Rossume2437a11992-03-23 18:20:18 +0000366}
367
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000368PyDoc_STRVAR(settrace_doc,
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000369"settrace(function)\n\
370\n\
371Set the global debug tracing function. It will be called on each\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000372function call. See the debugger chapter in the library manual."
373);
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000374
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000375static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000376sys_setprofile(PyObject *self, PyObject *args)
Guido van Rossume2437a11992-03-23 18:20:18 +0000377{
Fred Drake5755ce62001-06-27 19:19:46 +0000378 if (trace_init() == -1)
Fred Draked0838392001-06-16 21:02:31 +0000379 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000380 if (args == Py_None)
Fred Drake5755ce62001-06-27 19:19:46 +0000381 PyEval_SetProfile(NULL, NULL);
Guido van Rossume765f7d1992-04-05 14:17:55 +0000382 else
Fred Drake5755ce62001-06-27 19:19:46 +0000383 PyEval_SetProfile(profile_trampoline, args);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000384 Py_INCREF(Py_None);
385 return Py_None;
Guido van Rossume2437a11992-03-23 18:20:18 +0000386}
387
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000388PyDoc_STRVAR(setprofile_doc,
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000389"setprofile(function)\n\
390\n\
391Set the profiling function. It will be called on each function call\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000392and return. See the profiler chapter in the library manual."
393);
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000394
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000395static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000396sys_setcheckinterval(PyObject *self, PyObject *args)
Guido van Rossuma0d7a231995-01-09 17:46:13 +0000397{
Skip Montanarod581d772002-09-03 20:10:45 +0000398 if (!PyArg_ParseTuple(args, "i:setcheckinterval", &_Py_CheckInterval))
Guido van Rossuma0d7a231995-01-09 17:46:13 +0000399 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000400 Py_INCREF(Py_None);
401 return Py_None;
Guido van Rossuma0d7a231995-01-09 17:46:13 +0000402}
403
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000404PyDoc_STRVAR(setcheckinterval_doc,
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000405"setcheckinterval(n)\n\
406\n\
407Tell the Python interpreter to check for asynchronous events every\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000408n instructions. This also affects how often thread switches occur."
409);
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000410
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000411static PyObject *
Tim Peterse5e065b2003-07-06 18:36:54 +0000412sys_getcheckinterval(PyObject *self, PyObject *args)
413{
Christian Heimes217cfd12007-12-02 14:31:20 +0000414 return PyLong_FromLong(_Py_CheckInterval);
Tim Peterse5e065b2003-07-06 18:36:54 +0000415}
416
417PyDoc_STRVAR(getcheckinterval_doc,
418"getcheckinterval() -> current check interval; see setcheckinterval()."
419);
420
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000421#ifdef WITH_TSC
422static PyObject *
423sys_settscdump(PyObject *self, PyObject *args)
424{
425 int bool;
426 PyThreadState *tstate = PyThreadState_Get();
427
428 if (!PyArg_ParseTuple(args, "i:settscdump", &bool))
429 return NULL;
430 if (bool)
431 tstate->interp->tscdump = 1;
432 else
433 tstate->interp->tscdump = 0;
434 Py_INCREF(Py_None);
435 return Py_None;
Tim Peters216b78b2006-01-06 02:40:53 +0000436
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000437}
438
Tim Peters216b78b2006-01-06 02:40:53 +0000439PyDoc_STRVAR(settscdump_doc,
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000440"settscdump(bool)\n\
441\n\
442If true, tell the Python interpreter to dump VM measurements to\n\
443stderr. If false, turn off dump. The measurements are based on the\n\
Michael W. Hudson800ba232004-08-12 18:19:17 +0000444processor's time-stamp counter."
Tim Peters216b78b2006-01-06 02:40:53 +0000445);
Neal Norwitz0f5aed42004-06-13 20:32:17 +0000446#endif /* TSC */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000447
Tim Peterse5e065b2003-07-06 18:36:54 +0000448static PyObject *
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000449sys_setrecursionlimit(PyObject *self, PyObject *args)
450{
451 int new_limit;
452 if (!PyArg_ParseTuple(args, "i:setrecursionlimit", &new_limit))
453 return NULL;
454 if (new_limit <= 0) {
Tim Peters216b78b2006-01-06 02:40:53 +0000455 PyErr_SetString(PyExc_ValueError,
456 "recursion limit must be positive");
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000457 return NULL;
458 }
459 Py_SetRecursionLimit(new_limit);
460 Py_INCREF(Py_None);
461 return Py_None;
462}
463
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000464PyDoc_STRVAR(setrecursionlimit_doc,
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000465"setrecursionlimit(n)\n\
466\n\
467Set the maximum depth of the Python interpreter stack to n. This\n\
468limit prevents infinite recursion from causing an overflow of the C\n\
469stack and crashing Python. The highest possible limit is platform-\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000470dependent."
471);
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000472
473static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000474sys_getrecursionlimit(PyObject *self)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000475{
Christian Heimes217cfd12007-12-02 14:31:20 +0000476 return PyLong_FromLong(Py_GetRecursionLimit());
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000477}
478
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000479PyDoc_STRVAR(getrecursionlimit_doc,
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000480"getrecursionlimit()\n\
481\n\
482Return the current value of the recursion limit, the maximum depth\n\
483of the Python interpreter stack. This limit prevents infinite\n\
Jack Jansene739a0d2002-06-26 20:39:20 +0000484recursion from causing an overflow of the C stack and crashing Python."
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000485);
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000486
Mark Hammond8696ebc2002-10-08 02:44:31 +0000487#ifdef MS_WINDOWS
488PyDoc_STRVAR(getwindowsversion_doc,
489"getwindowsversion()\n\
490\n\
491Return information about the running version of Windows.\n\
492The result is a tuple of (major, minor, build, platform, text)\n\
493All elements are numbers, except text which is a string.\n\
494Platform may be 0 for win32s, 1 for Windows 9x/ME, 2 for Windows NT/2000/XP\n\
495"
496);
497
498static PyObject *
499sys_getwindowsversion(PyObject *self)
500{
501 OSVERSIONINFO ver;
502 ver.dwOSVersionInfoSize = sizeof(ver);
503 if (!GetVersionEx(&ver))
504 return PyErr_SetFromWindowsErr(0);
505 return Py_BuildValue("HHHHs",
506 ver.dwMajorVersion,
507 ver.dwMinorVersion,
508 ver.dwBuildNumber,
509 ver.dwPlatformId,
510 ver.szCSDVersion);
511}
512
513#endif /* MS_WINDOWS */
514
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000515#ifdef HAVE_DLOPEN
516static PyObject *
517sys_setdlopenflags(PyObject *self, PyObject *args)
518{
519 int new_val;
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000520 PyThreadState *tstate = PyThreadState_GET();
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000521 if (!PyArg_ParseTuple(args, "i:setdlopenflags", &new_val))
522 return NULL;
523 if (!tstate)
524 return NULL;
525 tstate->interp->dlopenflags = new_val;
526 Py_INCREF(Py_None);
527 return Py_None;
528}
529
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000530PyDoc_STRVAR(setdlopenflags_doc,
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000531"setdlopenflags(n) -> None\n\
532\n\
533Set the flags that will be used for dlopen() calls. Among other\n\
Neal Norwitz2a47c0f2002-01-29 00:53:41 +0000534things, this will enable a lazy resolving of symbols when importing\n\
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000535a module, if called as sys.setdlopenflags(0)\n\
Neal Norwitz2a47c0f2002-01-29 00:53:41 +0000536To share symbols across extension modules, call as\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000537sys.setdlopenflags(dl.RTLD_NOW|dl.RTLD_GLOBAL)"
538);
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000539
540static PyObject *
541sys_getdlopenflags(PyObject *self, PyObject *args)
542{
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000543 PyThreadState *tstate = PyThreadState_GET();
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000544 if (!tstate)
545 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +0000546 return PyLong_FromLong(tstate->interp->dlopenflags);
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000547}
548
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000549PyDoc_STRVAR(getdlopenflags_doc,
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000550"getdlopenflags() -> int\n\
551\n\
552Return the current value of the flags that are used for dlopen()\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000553calls. The flag constants are defined in the dl module."
554);
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000555#endif
556
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000557#ifdef USE_MALLOPT
558/* Link with -lmalloc (or -lmpc) on an SGI */
559#include <malloc.h>
560
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000561static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000562sys_mdebug(PyObject *self, PyObject *args)
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000563{
564 int flag;
Guido van Rossumffc0f4f2000-03-31 00:38:29 +0000565 if (!PyArg_ParseTuple(args, "i:mdebug", &flag))
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000566 return NULL;
567 mallopt(M_DEBUG, flag);
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000568 Py_INCREF(Py_None);
569 return Py_None;
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000570}
571#endif /* USE_MALLOPT */
572
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000573static PyObject *
Fred Drakea7688822001-10-24 20:47:48 +0000574sys_getrefcount(PyObject *self, PyObject *arg)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000575{
Christian Heimes217cfd12007-12-02 14:31:20 +0000576 return PyLong_FromSsize_t(arg->ob_refcnt);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000577}
578
Tim Peters4be93d02002-07-07 19:59:50 +0000579#ifdef Py_REF_DEBUG
Mark Hammond440d8982000-06-20 08:12:48 +0000580static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000581sys_gettotalrefcount(PyObject *self)
Mark Hammond440d8982000-06-20 08:12:48 +0000582{
Christian Heimes217cfd12007-12-02 14:31:20 +0000583 return PyLong_FromSsize_t(_Py_GetRefTotal());
Mark Hammond440d8982000-06-20 08:12:48 +0000584}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000585#endif /* Py_REF_DEBUG */
Mark Hammond440d8982000-06-20 08:12:48 +0000586
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000587PyDoc_STRVAR(getrefcount_doc,
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000588"getrefcount(object) -> integer\n\
589\n\
Fred Drakeba3ff1b2002-06-20 21:36:19 +0000590Return the reference count of object. The count returned is generally\n\
591one higher than you might expect, because it includes the (temporary)\n\
592reference as an argument to getrefcount()."
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000593);
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000594
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000595#ifdef COUNT_ALLOCS
596static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000597sys_getcounts(PyObject *self)
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000598{
Tim Petersdbd9ba62000-07-09 03:09:57 +0000599 extern PyObject *get_counts(void);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000600
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000601 return get_counts();
602}
603#endif
604
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000605PyDoc_STRVAR(getframe_doc,
Barry Warsawb6a54d22000-12-06 21:47:46 +0000606"_getframe([depth]) -> frameobject\n\
607\n\
608Return a frame object from the call stack. If optional integer depth is\n\
609given, return the frame object that many calls below the top of the stack.\n\
610If that is deeper than the call stack, ValueError is raised. The default\n\
611for depth is zero, returning the frame at the top of the call stack.\n\
612\n\
613This function should be used for internal and specialized\n\
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000614purposes only."
615);
Barry Warsawb6a54d22000-12-06 21:47:46 +0000616
617static PyObject *
618sys_getframe(PyObject *self, PyObject *args)
619{
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000620 PyFrameObject *f = PyThreadState_GET()->frame;
Barry Warsawb6a54d22000-12-06 21:47:46 +0000621 int depth = -1;
622
623 if (!PyArg_ParseTuple(args, "|i:_getframe", &depth))
624 return NULL;
625
626 while (depth > 0 && f != NULL) {
627 f = f->f_back;
628 --depth;
629 }
630 if (f == NULL) {
631 PyErr_SetString(PyExc_ValueError,
632 "call stack is not deep enough");
633 return NULL;
634 }
635 Py_INCREF(f);
636 return (PyObject*)f;
637}
638
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000639PyDoc_STRVAR(current_frames_doc,
640"_current_frames() -> dictionary\n\
641\n\
642Return a dictionary mapping each current thread T's thread id to T's\n\
643current stack frame.\n\
644\n\
645This function should be used for specialized purposes only."
646);
647
648static PyObject *
649sys_current_frames(PyObject *self, PyObject *noargs)
650{
651 return _PyThread_CurrentFrames();
652}
653
Guido van Rossuma12fe4e2003-04-09 19:06:21 +0000654PyDoc_STRVAR(call_tracing_doc,
655"call_tracing(func, args) -> object\n\
656\n\
657Call func(*args), while tracing is enabled. The tracing state is\n\
658saved, and restored afterwards. This is intended to be called from\n\
659a debugger from a checkpoint, to recursively debug some other code."
660);
661
662static PyObject *
663sys_call_tracing(PyObject *self, PyObject *args)
664{
665 PyObject *func, *funcargs;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000666 if (!PyArg_UnpackTuple(args, "call_tracing", 2, 2, &func, &funcargs))
Guido van Rossuma12fe4e2003-04-09 19:06:21 +0000667 return NULL;
668 return _PyEval_CallTracing(func, funcargs);
669}
670
Jeremy Hylton985eba52003-02-05 23:13:00 +0000671PyDoc_STRVAR(callstats_doc,
672"callstats() -> tuple of integers\n\
673\n\
674Return a tuple of function call statistics, if CALL_PROFILE was defined\n\
675when Python was built. Otherwise, return None.\n\
676\n\
677When enabled, this function returns detailed, implementation-specific\n\
678details about the number of function calls executed. The return value is\n\
679a 11-tuple where the entries in the tuple are counts of:\n\
6800. all function calls\n\
6811. calls to PyFunction_Type objects\n\
6822. PyFunction calls that do not create an argument tuple\n\
6833. PyFunction calls that do not create an argument tuple\n\
684 and bypass PyEval_EvalCodeEx()\n\
6854. PyMethod calls\n\
6865. PyMethod calls on bound methods\n\
6876. PyType calls\n\
6887. PyCFunction calls\n\
6898. generator calls\n\
6909. All other calls\n\
69110. Number of stack pops performed by call_function()"
692);
Barry Warsawb6a54d22000-12-06 21:47:46 +0000693
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000694#ifdef __cplusplus
695extern "C" {
696#endif
697
Guido van Rossum7f3f2c11996-05-23 22:45:41 +0000698#ifdef Py_TRACE_REFS
Guido van Rossumded690f1996-05-24 20:48:31 +0000699/* Defined in objects.c because it uses static globals if that file */
Tim Petersdbd9ba62000-07-09 03:09:57 +0000700extern PyObject *_Py_GetObjects(PyObject *, PyObject *);
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000701#endif
Guido van Rossumded690f1996-05-24 20:48:31 +0000702
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000703#ifdef DYNAMIC_EXECUTION_PROFILE
704/* Defined in ceval.c because it uses static globals if that file */
Tim Petersdbd9ba62000-07-09 03:09:57 +0000705extern PyObject *_Py_GetDXProfile(PyObject *, PyObject *);
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000706#endif
707
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000708#ifdef __cplusplus
709}
710#endif
711
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000712static PyMethodDef sys_methods[] = {
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000713 /* Might as well keep this in alphabetic order */
Tim Peters216b78b2006-01-06 02:40:53 +0000714 {"callstats", (PyCFunction)PyEval_GetCallStats, METH_NOARGS,
Jeremy Hylton985eba52003-02-05 23:13:00 +0000715 callstats_doc},
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000716 {"_current_frames", sys_current_frames, METH_NOARGS,
717 current_frames_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000718 {"displayhook", sys_displayhook, METH_O, displayhook_doc},
Guido van Rossum46d3dc32003-03-01 03:20:41 +0000719 {"exc_info", sys_exc_info, METH_NOARGS, exc_info_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000720 {"excepthook", sys_excepthook, METH_VARARGS, excepthook_doc},
Neal Norwitz0c766a02002-03-27 13:03:09 +0000721 {"exit", sys_exit, METH_VARARGS, exit_doc},
Tim Peters216b78b2006-01-06 02:40:53 +0000722 {"getdefaultencoding", (PyCFunction)sys_getdefaultencoding,
723 METH_NOARGS, getdefaultencoding_doc},
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000724#ifdef HAVE_DLOPEN
Tim Peters216b78b2006-01-06 02:40:53 +0000725 {"getdlopenflags", (PyCFunction)sys_getdlopenflags, METH_NOARGS,
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000726 getdlopenflags_doc},
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000727#endif
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000728#ifdef COUNT_ALLOCS
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000729 {"getcounts", (PyCFunction)sys_getcounts, METH_NOARGS},
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000730#endif
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000731#ifdef DYNAMIC_EXECUTION_PROFILE
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000732 {"getdxp", _Py_GetDXProfile, METH_VARARGS},
Guido van Rossum43f1b8d1997-01-24 04:07:45 +0000733#endif
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000734 {"getfilesystemencoding", (PyCFunction)sys_getfilesystemencoding,
Tim Peters216b78b2006-01-06 02:40:53 +0000735 METH_NOARGS, getfilesystemencoding_doc},
Guido van Rossum7f3f2c11996-05-23 22:45:41 +0000736#ifdef Py_TRACE_REFS
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000737 {"getobjects", _Py_GetObjects, METH_VARARGS},
Tim Peters4be93d02002-07-07 19:59:50 +0000738#endif
739#ifdef Py_REF_DEBUG
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000740 {"gettotalrefcount", (PyCFunction)sys_gettotalrefcount, METH_NOARGS},
Sjoerd Mullender6ec3c651995-08-29 09:18:14 +0000741#endif
Fred Drakea7688822001-10-24 20:47:48 +0000742 {"getrefcount", (PyCFunction)sys_getrefcount, METH_O, getrefcount_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000743 {"getrecursionlimit", (PyCFunction)sys_getrecursionlimit, METH_NOARGS,
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000744 getrecursionlimit_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000745 {"_getframe", sys_getframe, METH_VARARGS, getframe_doc},
Mark Hammond8696ebc2002-10-08 02:44:31 +0000746#ifdef MS_WINDOWS
747 {"getwindowsversion", (PyCFunction)sys_getwindowsversion, METH_NOARGS,
748 getwindowsversion_doc},
749#endif /* MS_WINDOWS */
Georg Brandl66a796e2006-12-19 20:50:34 +0000750 {"intern", sys_intern, METH_VARARGS, intern_doc},
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000751#ifdef USE_MALLOPT
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000752 {"mdebug", sys_mdebug, METH_VARARGS},
Guido van Rossum14b4adb1992-09-03 20:25:30 +0000753#endif
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000754 {"setdefaultencoding", sys_setdefaultencoding, METH_VARARGS,
Tim Peters216b78b2006-01-06 02:40:53 +0000755 setdefaultencoding_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000756 {"setcheckinterval", sys_setcheckinterval, METH_VARARGS,
Tim Peters216b78b2006-01-06 02:40:53 +0000757 setcheckinterval_doc},
Tim Peterse5e065b2003-07-06 18:36:54 +0000758 {"getcheckinterval", sys_getcheckinterval, METH_NOARGS,
Tim Peters216b78b2006-01-06 02:40:53 +0000759 getcheckinterval_doc},
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000760#ifdef HAVE_DLOPEN
Tim Peters216b78b2006-01-06 02:40:53 +0000761 {"setdlopenflags", sys_setdlopenflags, METH_VARARGS,
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000762 setdlopenflags_doc},
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000763#endif
Neal Norwitz290d31e2002-03-03 15:12:58 +0000764 {"setprofile", sys_setprofile, METH_O, setprofile_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000765 {"setrecursionlimit", sys_setrecursionlimit, METH_VARARGS,
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000766 setrecursionlimit_doc},
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000767#ifdef WITH_TSC
768 {"settscdump", sys_settscdump, METH_VARARGS, settscdump_doc},
769#endif
Neal Norwitz290d31e2002-03-03 15:12:58 +0000770 {"settrace", sys_settrace, METH_O, settrace_doc},
Guido van Rossuma12fe4e2003-04-09 19:06:21 +0000771 {"call_tracing", sys_call_tracing, METH_VARARGS, call_tracing_doc},
Guido van Rossum3f5da241990-12-20 15:06:42 +0000772 {NULL, NULL} /* sentinel */
773};
774
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000775static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000776list_builtin_module_names(void)
Guido van Rossum34679b71993-01-26 13:33:44 +0000777{
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000778 PyObject *list = PyList_New(0);
Guido van Rossum34679b71993-01-26 13:33:44 +0000779 int i;
780 if (list == NULL)
781 return NULL;
Guido van Rossum25c649f1997-11-04 17:04:34 +0000782 for (i = 0; PyImport_Inittab[i].name != NULL; i++) {
Martin v. Löwis5b222132007-06-10 09:51:05 +0000783 PyObject *name = PyUnicode_FromString(
Guido van Rossum25c649f1997-11-04 17:04:34 +0000784 PyImport_Inittab[i].name);
Guido van Rossum34679b71993-01-26 13:33:44 +0000785 if (name == NULL)
786 break;
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000787 PyList_Append(list, name);
788 Py_DECREF(name);
Guido van Rossum34679b71993-01-26 13:33:44 +0000789 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000790 if (PyList_Sort(list) != 0) {
791 Py_DECREF(list);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000792 list = NULL;
793 }
Guido van Rossum8f49e121997-01-06 22:55:54 +0000794 if (list) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +0000795 PyObject *v = PyList_AsTuple(list);
796 Py_DECREF(list);
Guido van Rossum8f49e121997-01-06 22:55:54 +0000797 list = v;
798 }
Guido van Rossum34679b71993-01-26 13:33:44 +0000799 return list;
800}
801
Guido van Rossum23fff912000-12-15 22:02:05 +0000802static PyObject *warnoptions = NULL;
803
804void
805PySys_ResetWarnOptions(void)
806{
807 if (warnoptions == NULL || !PyList_Check(warnoptions))
808 return;
809 PyList_SetSlice(warnoptions, 0, PyList_GET_SIZE(warnoptions), NULL);
810}
811
812void
Neal Norwitzf3081322007-08-25 00:32:45 +0000813PySys_AddWarnOption(const char *s)
Guido van Rossum23fff912000-12-15 22:02:05 +0000814{
815 PyObject *str;
816
817 if (warnoptions == NULL || !PyList_Check(warnoptions)) {
818 Py_XDECREF(warnoptions);
819 warnoptions = PyList_New(0);
820 if (warnoptions == NULL)
821 return;
822 }
Neal Norwitzaefab1e2007-08-25 01:04:34 +0000823 str = PyUnicode_FromString(s);
Guido van Rossum23fff912000-12-15 22:02:05 +0000824 if (str != NULL) {
825 PyList_Append(warnoptions, str);
826 Py_DECREF(str);
827 }
828}
829
Guido van Rossum40552d01998-08-06 03:34:39 +0000830/* XXX This doc string is too long to be a single string literal in VC++ 5.0.
831 Two literals concatenated works just fine. If you have a K&R compiler
832 or other abomination that however *does* understand longer strings,
833 get rid of the !!! comment in the middle and the quotes that surround it. */
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000834PyDoc_VAR(sys_doc) =
835PyDoc_STR(
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000836"This module provides access to some objects used or maintained by the\n\
837interpreter and to functions that interact strongly with the interpreter.\n\
838\n\
839Dynamic objects:\n\
840\n\
841argv -- command line arguments; argv[0] is the script pathname if known\n\
842path -- module search path; path[0] is the script directory, else ''\n\
843modules -- dictionary of loaded modules\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000844\n\
845displayhook -- called to show results in an interactive session\n\
846excepthook -- called to handle any uncaught exception other than SystemExit\n\
847 To customize printing in an interactive session or to install a custom\n\
848 top-level exception handler, assign other functions to replace these.\n\
849\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000850stdin -- standard input file object; used by raw_input() and input()\n\
Georg Brandl88fc6642007-02-09 21:28:07 +0000851stdout -- standard output file object; used by print()\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000852stderr -- standard error object; used for error messages\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000853 By assigning other file objects (or objects that behave like files)\n\
854 to these, it is possible to redirect all of the interpreter's I/O.\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000855\n\
856last_type -- type of last uncaught exception\n\
857last_value -- value of last uncaught exception\n\
858last_traceback -- traceback of last uncaught exception\n\
859 These three are only available in an interactive session after a\n\
860 traceback has been printed.\n\
Guido van Rossuma71b5f41999-01-14 19:07:00 +0000861"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000862)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000863/* concatenating string here */
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000864PyDoc_STR(
Guido van Rossuma71b5f41999-01-14 19:07:00 +0000865"\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000866Static objects:\n\
867\n\
Christian Heimes2d378ab2007-12-15 01:28:04 +0000868float_info -- a dict with information about the float implementation.\n\
Thomas Woutersd2cf20e2007-08-30 22:57:53 +0000869maxsize -- the largest supported length of containers.\n\
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000870maxunicode -- the largest supported character\n\
Neal Norwitz2a47c0f2002-01-29 00:53:41 +0000871builtin_module_names -- tuple of module names built into this interpreter\n\
Christian Heimes2d378ab2007-12-15 01:28:04 +0000872subversion -- subversion information of the build as tuple\n\
Fred Drake801c08d2000-04-13 15:29:10 +0000873version -- the version of this interpreter as a string\n\
874version_info -- version information as a tuple\n\
875hexversion -- version information encoded as a single integer\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000876copyright -- copyright notice pertaining to this interpreter\n\
877platform -- platform identifier\n\
878executable -- pathname of this Python interpreter\n\
879prefix -- prefix used to find the Python library\n\
880exec_prefix -- prefix used to find the machine-specific Python library\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000881"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000882)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000883#ifdef MS_WINDOWS
884/* concatenating string here */
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000885PyDoc_STR(
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000886"dllhandle -- [Windows only] integer handle of the Python DLL\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000887winver -- [Windows only] version number of the Python DLL\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000888"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000889)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000890#endif /* MS_WINDOWS */
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000891PyDoc_STR(
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000892"__stdin__ -- the original stdin; don't touch!\n\
893__stdout__ -- the original stdout; don't touch!\n\
894__stderr__ -- the original stderr; don't touch!\n\
895__displayhook__ -- the original displayhook; don't touch!\n\
896__excepthook__ -- the original excepthook; don't touch!\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000897\n\
898Functions:\n\
899\n\
Georg Brandl1a3284e2007-12-02 09:40:06 +0000900displayhook() -- print an object to the screen, and save it in builtins._\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000901excepthook() -- print an exception and its traceback to sys.stderr\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000902exc_info() -- return thread-safe information about the current exception\n\
903exit() -- exit the interpreter by raising SystemExit\n\
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000904getdlopenflags() -- returns flags to be used for dlopen() calls\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000905getrefcount() -- return the reference count for an object (plus one :-)\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000906getrecursionlimit() -- return the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000907setcheckinterval() -- control how often the interpreter checks for events\n\
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000908setdlopenflags() -- set the flags to be used for dlopen() calls\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000909setprofile() -- set the global profiling function\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000910setrecursionlimit() -- set the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000911settrace() -- set the global debug tracing function\n\
Fred Drakeccede592000-08-14 20:59:57 +0000912"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000913)
Fred Drakeccede592000-08-14 20:59:57 +0000914/* end of sys_doc */ ;
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000915
Martin v. Löwis43b57802006-01-05 23:38:54 +0000916/* Subversion branch and revision management */
917static const char _patchlevel_revision[] = PY_PATCHLEVEL_REVISION;
918static const char headurl[] = "$HeadURL$";
919static int svn_initialized;
920static char patchlevel_revision[50]; /* Just the number */
921static char branch[50];
922static char shortbranch[50];
923static const char *svn_revision;
924
Tim Peterse86e7a52006-01-06 02:42:46 +0000925static void
926svnversion_init(void)
Martin v. Löwis43b57802006-01-05 23:38:54 +0000927{
928 const char *python, *br_start, *br_end, *br_end2, *svnversion;
Martin v. Löwisd96ee902006-02-16 14:37:16 +0000929 Py_ssize_t len;
Collin Winterd5a5f5d2007-08-22 19:45:07 +0000930 int istag = 0;
Martin v. Löwis43b57802006-01-05 23:38:54 +0000931
932 if (svn_initialized)
933 return;
934
935 python = strstr(headurl, "/python/");
Collin Winterd5a5f5d2007-08-22 19:45:07 +0000936 if (!python) {
937 strcpy(branch, "unknown branch");
938 strcpy(shortbranch, "unknown");
Tim Peters216b78b2006-01-06 02:40:53 +0000939 }
Martin v. Löwis43b57802006-01-05 23:38:54 +0000940 else {
Collin Winterd5a5f5d2007-08-22 19:45:07 +0000941 br_start = python + 8;
942 br_end = strchr(br_start, '/');
943 assert(br_end);
944
945 /* Works even for trunk,
946 as we are in trunk/Python/sysmodule.c */
947 br_end2 = strchr(br_end+1, '/');
948
949 istag = strncmp(br_start, "tags", 4) == 0;
950 if (strncmp(br_start, "trunk", 5) == 0) {
951 strcpy(branch, "trunk");
952 strcpy(shortbranch, "trunk");
953 }
954 else if (istag || strncmp(br_start, "branches", 8) == 0) {
955 len = br_end2 - br_start;
956 strncpy(branch, br_start, len);
957 branch[len] = '\0';
958
959 len = br_end2 - (br_end + 1);
960 strncpy(shortbranch, br_end + 1, len);
961 shortbranch[len] = '\0';
962 }
963 else {
964 Py_FatalError("bad HeadURL");
965 return;
966 }
Martin v. Löwis43b57802006-01-05 23:38:54 +0000967 }
968
969
970 svnversion = _Py_svnversion();
971 if (strcmp(svnversion, "exported") != 0)
972 svn_revision = svnversion;
973 else if (istag) {
974 len = strlen(_patchlevel_revision);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000975 assert(len >= 13);
976 assert(len < (sizeof(patchlevel_revision) + 13));
Martin v. Löwis43b57802006-01-05 23:38:54 +0000977 strncpy(patchlevel_revision, _patchlevel_revision + 11,
978 len - 13);
979 patchlevel_revision[len - 13] = '\0';
980 svn_revision = patchlevel_revision;
981 }
982 else
983 svn_revision = "";
Tim Peters216b78b2006-01-06 02:40:53 +0000984
Martin v. Löwis43b57802006-01-05 23:38:54 +0000985 svn_initialized = 1;
986}
987
988/* Return svnversion output if available.
989 Else return Revision of patchlevel.h if on branch.
990 Else return empty string */
991const char*
992Py_SubversionRevision()
993{
994 svnversion_init();
995 return svn_revision;
996}
997
998const char*
999Py_SubversionShortBranch()
1000{
1001 svnversion_init();
1002 return shortbranch;
1003}
1004
Guido van Rossum25ce5661997-08-02 03:10:38 +00001005PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001006_PySys_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001007{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001008 PyObject *m, *v, *sysdict;
Fred Drake6d27c1e2000-04-13 20:03:20 +00001009 char *s;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001010
Guido van Rossumc3bc31e1998-06-27 19:43:25 +00001011 m = Py_InitModule3("sys", sys_methods, sys_doc);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001012 if (m == NULL)
1013 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001014 sysdict = PyModule_GetDict(m);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001015
Neal Norwitz11bd1192005-10-03 00:54:56 +00001016 {
1017 /* XXX: does this work on Win/Win64? (see posix_fstat) */
1018 struct stat sb;
1019 if (fstat(fileno(stdin), &sb) == 0 &&
1020 S_ISDIR(sb.st_mode)) {
Neal Norwitz72c2c062006-03-09 05:58:11 +00001021 /* There's nothing more we can do. */
1022 /* Py_FatalError() will core dump, so just exit. */
1023 PySys_WriteStderr("Python error: <stdin> is a directory, cannot continue\n");
1024 exit(EXIT_FAILURE);
Neal Norwitz11bd1192005-10-03 00:54:56 +00001025 }
1026 }
1027
Nick Coghlane7837002007-12-21 07:31:17 +00001028 /* stdin/stdout/stderr are now set by pythonrun.c */
Martin v. Löwis5467d4c2003-05-10 07:10:12 +00001029
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001030 PyDict_SetItemString(sysdict, "__displayhook__",
1031 PyDict_GetItemString(sysdict, "displayhook"));
1032 PyDict_SetItemString(sysdict, "__excepthook__",
1033 PyDict_GetItemString(sysdict, "excepthook"));
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001034 PyDict_SetItemString(sysdict, "version",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001035 v = PyUnicode_FromString(Py_GetVersion()));
Barry Warsaw54892c41999-01-27 16:33:19 +00001036 Py_XDECREF(v);
Guido van Rossume0d7dae1999-01-03 12:55:39 +00001037 PyDict_SetItemString(sysdict, "hexversion",
Christian Heimes217cfd12007-12-02 14:31:20 +00001038 v = PyLong_FromLong(PY_VERSION_HEX));
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001039 Py_XDECREF(v);
Martin v. Löwis43b57802006-01-05 23:38:54 +00001040 svnversion_init();
Walter Dörwald94093ff2007-06-07 18:41:59 +00001041 v = Py_BuildValue("(UUU)", "CPython", branch, svn_revision);
Martin v. Löwis43b57802006-01-05 23:38:54 +00001042 PyDict_SetItemString(sysdict, "subversion", v);
Barry Warsaw2a38a862005-12-18 01:27:35 +00001043 Py_XDECREF(v);
Christian Heimes790c8232008-01-07 21:14:23 +00001044 PyDict_SetItemString(sysdict, "dont_write_bytecode",
1045 v = PyBool_FromLong(Py_DontWriteBytecodeFlag));
1046 Py_XDECREF(v);
Fred Drake93a20bf2000-04-13 17:44:51 +00001047 /*
1048 * These release level checks are mutually exclusive and cover
1049 * the field, so don't get too fancy with the pre-processor!
1050 */
1051#if PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_ALPHA
Fred Drake6d27c1e2000-04-13 20:03:20 +00001052 s = "alpha";
Fred Drake592f2d62000-08-31 15:21:11 +00001053#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_BETA
Fred Drake6d27c1e2000-04-13 20:03:20 +00001054 s = "beta";
Fred Drake592f2d62000-08-31 15:21:11 +00001055#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_GAMMA
Fred Drake6d27c1e2000-04-13 20:03:20 +00001056 s = "candidate";
Fred Drake592f2d62000-08-31 15:21:11 +00001057#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_FINAL
Fred Drake6d27c1e2000-04-13 20:03:20 +00001058 s = "final";
Fred Drake93a20bf2000-04-13 17:44:51 +00001059#endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001060
1061#define SET_SYS_FROM_STRING(key, value) \
1062 v = value; \
1063 if (v != NULL) \
1064 PyDict_SetItemString(sysdict, key, v); \
1065 Py_XDECREF(v)
1066
1067 SET_SYS_FROM_STRING("version_info",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001068 Py_BuildValue("iiiUi", PY_MAJOR_VERSION,
Fred Drake801c08d2000-04-13 15:29:10 +00001069 PY_MINOR_VERSION,
Fred Drake6d27c1e2000-04-13 20:03:20 +00001070 PY_MICRO_VERSION, s,
Fred Drake93a20bf2000-04-13 17:44:51 +00001071 PY_RELEASE_SERIAL));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001072 SET_SYS_FROM_STRING("api_version",
Christian Heimes217cfd12007-12-02 14:31:20 +00001073 PyLong_FromLong(PYTHON_API_VERSION));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001074 SET_SYS_FROM_STRING("copyright",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001075 PyUnicode_FromString(Py_GetCopyright()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001076 SET_SYS_FROM_STRING("platform",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001077 PyUnicode_FromString(Py_GetPlatform()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001078 SET_SYS_FROM_STRING("executable",
Christian Heimeseb9b39b2007-11-14 00:15:01 +00001079 PyUnicode_DecodeFSDefault(
1080 Py_GetProgramFullPath()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001081 SET_SYS_FROM_STRING("prefix",
Christian Heimeseb9b39b2007-11-14 00:15:01 +00001082 PyUnicode_DecodeFSDefault(Py_GetPrefix()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001083 SET_SYS_FROM_STRING("exec_prefix",
Christian Heimeseb9b39b2007-11-14 00:15:01 +00001084 PyUnicode_DecodeFSDefault(Py_GetExecPrefix()));
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001085 SET_SYS_FROM_STRING("maxsize",
Christian Heimes217cfd12007-12-02 14:31:20 +00001086 PyLong_FromSsize_t(PY_SSIZE_T_MAX));
Christian Heimes93852662007-12-01 12:22:32 +00001087 SET_SYS_FROM_STRING("float_info",
1088 PyFloat_GetInfo());
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001089 SET_SYS_FROM_STRING("maxunicode",
Christian Heimes217cfd12007-12-02 14:31:20 +00001090 PyLong_FromLong(PyUnicode_GetMax()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001091 SET_SYS_FROM_STRING("builtin_module_names",
1092 list_builtin_module_names());
Fred Drake099325e2000-08-14 15:47:03 +00001093 {
1094 /* Assumes that longs are at least 2 bytes long.
1095 Should be safe! */
1096 unsigned long number = 1;
Fred Drakea2b6ad62000-08-15 04:24:43 +00001097 char *value;
Fred Drake099325e2000-08-14 15:47:03 +00001098
1099 s = (char *) &number;
1100 if (s[0] == 0)
Fred Drakea2b6ad62000-08-15 04:24:43 +00001101 value = "big";
Fred Drake099325e2000-08-14 15:47:03 +00001102 else
Fred Drakea2b6ad62000-08-15 04:24:43 +00001103 value = "little";
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001104 SET_SYS_FROM_STRING("byteorder",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001105 PyUnicode_FromString(value));
Fred Drake099325e2000-08-14 15:47:03 +00001106 }
Guido van Rossum8b9ea871996-08-23 18:14:47 +00001107#ifdef MS_COREDLL
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001108 SET_SYS_FROM_STRING("dllhandle",
1109 PyLong_FromVoidPtr(PyWin_DLLhModule));
1110 SET_SYS_FROM_STRING("winver",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001111 PyUnicode_FromString(PyWin_DLLVersionString));
Guido van Rossumc606fe11996-04-09 02:37:57 +00001112#endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001113#undef SET_SYS_FROM_STRING
Guido van Rossum23fff912000-12-15 22:02:05 +00001114 if (warnoptions == NULL) {
1115 warnoptions = PyList_New(0);
1116 }
1117 else {
1118 Py_INCREF(warnoptions);
1119 }
1120 if (warnoptions != NULL) {
Guido van Rossum03df3b32001-01-13 22:06:05 +00001121 PyDict_SetItemString(sysdict, "warnoptions", warnoptions);
Guido van Rossum23fff912000-12-15 22:02:05 +00001122 }
Tim Peters216b78b2006-01-06 02:40:53 +00001123
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001124 if (PyErr_Occurred())
Guido van Rossum25ce5661997-08-02 03:10:38 +00001125 return NULL;
1126 return m;
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001127}
1128
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001129static PyObject *
Neal Norwitzf3081322007-08-25 00:32:45 +00001130makepathobject(const char *path, int delim)
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001131{
Guido van Rossum3f5da241990-12-20 15:06:42 +00001132 int i, n;
Neal Norwitzf3081322007-08-25 00:32:45 +00001133 const char *p;
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001134 PyObject *v, *w;
Tim Peters216b78b2006-01-06 02:40:53 +00001135
Guido van Rossum3f5da241990-12-20 15:06:42 +00001136 n = 1;
1137 p = path;
1138 while ((p = strchr(p, delim)) != NULL) {
1139 n++;
1140 p++;
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001141 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001142 v = PyList_New(n);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001143 if (v == NULL)
1144 return NULL;
1145 for (i = 0; ; i++) {
1146 p = strchr(path, delim);
1147 if (p == NULL)
1148 p = strchr(path, '\0'); /* End of string */
Christian Heimes5894ba72007-11-04 11:43:14 +00001149 w = PyUnicode_DecodeFSDefaultAndSize(path, (Py_ssize_t) (p - path));
Guido van Rossum3f5da241990-12-20 15:06:42 +00001150 if (w == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001151 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001152 return NULL;
1153 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001154 PyList_SetItem(v, i, w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001155 if (*p == '\0')
1156 break;
1157 path = p+1;
1158 }
1159 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001160}
1161
1162void
Neal Norwitzf3081322007-08-25 00:32:45 +00001163PySys_SetPath(const char *path)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001164{
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001165 PyObject *v;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001166 if ((v = makepathobject(path, DELIM)) == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001167 Py_FatalError("can't create sys.path");
1168 if (PySys_SetObject("path", v) != 0)
1169 Py_FatalError("can't assign sys.path");
1170 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001171}
1172
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001173static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001174makeargvobject(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001175{
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001176 PyObject *av;
Guido van Rossumee3a2991992-01-14 18:42:53 +00001177 if (argc <= 0 || argv == NULL) {
1178 /* Ensure at least one (empty) argument is seen */
1179 static char *empty_argv[1] = {""};
1180 argv = empty_argv;
1181 argc = 1;
1182 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001183 av = PyList_New(argc);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001184 if (av != NULL) {
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001185 int i;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001186 for (i = 0; i < argc; i++) {
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001187#ifdef __VMS
1188 PyObject *v;
1189
1190 /* argv[0] is the script pathname if known */
1191 if (i == 0) {
1192 char* fn = decc$translate_vms(argv[0]);
1193 if ((fn == (char *)0) || fn == (char *)-1)
Guido van Rossum96ca6912007-06-15 03:49:03 +00001194 v = PyUnicode_FromString(argv[0]);
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001195 else
Guido van Rossum96ca6912007-06-15 03:49:03 +00001196 v = PyUnicode_FromString(
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001197 decc$translate_vms(argv[0]));
1198 } else
Guido van Rossum96ca6912007-06-15 03:49:03 +00001199 v = PyUnicode_FromString(argv[i]);
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001200#else
Guido van Rossum96ca6912007-06-15 03:49:03 +00001201 PyObject *v = PyUnicode_FromString(argv[i]);
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001202#endif
Guido van Rossum3f5da241990-12-20 15:06:42 +00001203 if (v == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001204 Py_DECREF(av);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001205 av = NULL;
1206 break;
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001207 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001208 PyList_SetItem(av, i, v);
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001209 }
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001210 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001211 return av;
1212}
1213
1214void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001215PySys_SetArgv(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001216{
Guido van Rossum162e38c2003-02-19 15:25:10 +00001217#if defined(HAVE_REALPATH)
1218 char fullpath[MAXPATHLEN];
1219#elif defined(MS_WINDOWS)
Thomas Heller27bb71e2003-01-08 14:33:48 +00001220 char fullpath[MAX_PATH];
1221#endif
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001222 PyObject *av = makeargvobject(argc, argv);
1223 PyObject *path = PySys_GetObject("path");
Guido van Rossum3f5da241990-12-20 15:06:42 +00001224 if (av == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001225 Py_FatalError("no mem for sys.argv");
1226 if (PySys_SetObject("argv", av) != 0)
1227 Py_FatalError("can't assign sys.argv");
Guido van Rossum94a96671996-07-30 20:35:50 +00001228 if (path != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +00001229 char *argv0 = argv[0];
Guido van Rossum94a96671996-07-30 20:35:50 +00001230 char *p = NULL;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001231 Py_ssize_t n = 0;
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001232 PyObject *a;
Guido van Rossumc474dea1997-04-25 15:38:31 +00001233#ifdef HAVE_READLINK
1234 char link[MAXPATHLEN+1];
1235 char argv0copy[2*MAXPATHLEN+1];
1236 int nr = 0;
Georg Brandl69537722005-09-15 13:00:34 +00001237 if (argc > 0 && argv0 != NULL && strcmp(argv0, "-c") != 0)
Guido van Rossumc474dea1997-04-25 15:38:31 +00001238 nr = readlink(argv0, link, MAXPATHLEN);
1239 if (nr > 0) {
1240 /* It's a symlink */
1241 link[nr] = '\0';
1242 if (link[0] == SEP)
1243 argv0 = link; /* Link to absolute path */
1244 else if (strchr(link, SEP) == NULL)
1245 ; /* Link without path */
1246 else {
1247 /* Must join(dirname(argv0), link) */
1248 char *q = strrchr(argv0, SEP);
1249 if (q == NULL)
1250 argv0 = link; /* argv0 without path */
1251 else {
1252 /* Must make a copy */
1253 strcpy(argv0copy, argv0);
1254 q = strrchr(argv0copy, SEP);
1255 strcpy(q+1, link);
1256 argv0 = argv0copy;
1257 }
1258 }
1259 }
1260#endif /* HAVE_READLINK */
Guido van Rossumcc883411996-09-10 14:44:21 +00001261#if SEP == '\\' /* Special case for MS filename syntax */
Georg Brandl69537722005-09-15 13:00:34 +00001262 if (argc > 0 && argv0 != NULL && strcmp(argv0, "-c") != 0) {
Guido van Rossumcc883411996-09-10 14:44:21 +00001263 char *q;
Thomas Heller27bb71e2003-01-08 14:33:48 +00001264#ifdef MS_WINDOWS
1265 char *ptemp;
1266 if (GetFullPathName(argv0,
1267 sizeof(fullpath),
1268 fullpath,
1269 &ptemp)) {
1270 argv0 = fullpath;
1271 }
1272#endif
Guido van Rossumc474dea1997-04-25 15:38:31 +00001273 p = strrchr(argv0, SEP);
Guido van Rossumcc883411996-09-10 14:44:21 +00001274 /* Test for alternate separator */
Guido van Rossumc474dea1997-04-25 15:38:31 +00001275 q = strrchr(p ? p : argv0, '/');
Guido van Rossumcc883411996-09-10 14:44:21 +00001276 if (q != NULL)
1277 p = q;
1278 if (p != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +00001279 n = p + 1 - argv0;
Guido van Rossumcc883411996-09-10 14:44:21 +00001280 if (n > 1 && p[-1] != ':')
1281 n--; /* Drop trailing separator */
1282 }
1283 }
1284#else /* All other filename syntaxes */
Georg Brandl69537722005-09-15 13:00:34 +00001285 if (argc > 0 && argv0 != NULL && strcmp(argv0, "-c") != 0) {
Guido van Rossum162e38c2003-02-19 15:25:10 +00001286#if defined(HAVE_REALPATH)
1287 if (realpath(argv0, fullpath)) {
1288 argv0 = fullpath;
1289 }
1290#endif
Guido van Rossumc474dea1997-04-25 15:38:31 +00001291 p = strrchr(argv0, SEP);
Guido van Rossum162e38c2003-02-19 15:25:10 +00001292 }
Guido van Rossumcc883411996-09-10 14:44:21 +00001293 if (p != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +00001294 n = p + 1 - argv0;
Guido van Rossumcc883411996-09-10 14:44:21 +00001295#if SEP == '/' /* Special case for Unix filename syntax */
1296 if (n > 1)
1297 n--; /* Drop trailing separator */
1298#endif /* Unix */
1299 }
1300#endif /* All others */
Guido van Rossum96ca6912007-06-15 03:49:03 +00001301 a = PyUnicode_FromStringAndSize(argv0, n);
Guido van Rossum94a96671996-07-30 20:35:50 +00001302 if (a == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001303 Py_FatalError("no mem for sys.path insertion");
1304 if (PyList_Insert(path, 0, a) < 0)
1305 Py_FatalError("sys.path.insert(0) failed");
1306 Py_DECREF(a);
Guido van Rossuma63d9f41996-07-24 01:31:37 +00001307 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001308 Py_DECREF(av);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001309}
Guido van Rossuma890e681998-05-12 14:59:24 +00001310
1311
1312/* APIs to write to sys.stdout or sys.stderr using a printf-like interface.
1313 Adapted from code submitted by Just van Rossum.
1314
1315 PySys_WriteStdout(format, ...)
1316 PySys_WriteStderr(format, ...)
1317
1318 The first function writes to sys.stdout; the second to sys.stderr. When
1319 there is a problem, they write to the real (C level) stdout or stderr;
Guido van Rossum8442af31998-10-12 18:22:10 +00001320 no exceptions are raised.
Guido van Rossuma890e681998-05-12 14:59:24 +00001321
1322 Both take a printf-style format string as their first argument followed
1323 by a variable length argument list determined by the format string.
1324
1325 *** WARNING ***
1326
1327 The format should limit the total size of the formatted output string to
1328 1000 bytes. In particular, this means that no unrestricted "%s" formats
1329 should occur; these should be limited using "%.<N>s where <N> is a
1330 decimal number calculated so that <N> plus the maximum size of other
1331 formatted text does not exceed 1000 bytes. Also watch out for "%f",
1332 which can print hundreds of digits for very large numbers.
1333
1334 */
1335
1336static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001337mywrite(char *name, FILE *fp, const char *format, va_list va)
Guido van Rossuma890e681998-05-12 14:59:24 +00001338{
1339 PyObject *file;
Guido van Rossum8442af31998-10-12 18:22:10 +00001340 PyObject *error_type, *error_value, *error_traceback;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001341 char buffer[1001];
1342 int written;
Guido van Rossuma890e681998-05-12 14:59:24 +00001343
Guido van Rossum8442af31998-10-12 18:22:10 +00001344 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +00001345 file = PySys_GetObject(name);
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001346 written = PyOS_vsnprintf(buffer, sizeof(buffer), format, va);
1347 if (PyFile_WriteString(buffer, file) != 0) {
1348 PyErr_Clear();
1349 fputs(buffer, fp);
1350 }
1351 if (written < 0 || (size_t)written >= sizeof(buffer)) {
1352 const char *truncated = "... truncated";
1353 if (PyFile_WriteString(truncated, file) != 0) {
Guido van Rossuma890e681998-05-12 14:59:24 +00001354 PyErr_Clear();
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001355 fputs(truncated, fp);
Jeremy Hylton5d3d1342001-11-28 21:44:53 +00001356 }
Guido van Rossuma890e681998-05-12 14:59:24 +00001357 }
Guido van Rossum8442af31998-10-12 18:22:10 +00001358 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +00001359}
1360
1361void
Guido van Rossuma890e681998-05-12 14:59:24 +00001362PySys_WriteStdout(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +00001363{
1364 va_list va;
1365
Guido van Rossuma890e681998-05-12 14:59:24 +00001366 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +00001367 mywrite("stdout", stdout, format, va);
1368 va_end(va);
1369}
1370
1371void
Guido van Rossuma890e681998-05-12 14:59:24 +00001372PySys_WriteStderr(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +00001373{
1374 va_list va;
1375
Guido van Rossuma890e681998-05-12 14:59:24 +00001376 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +00001377 mywrite("stderr", stderr, format, va);
1378 va_end(va);
1379}