blob: 6fcbe75eef53c505d0ff3c74996a1f05e2cf83a0 [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;
75 PyObject *builtins = PyDict_GetItemString(modules, "__builtin__");
76
Moshe Zadka03897ea2001-07-23 13:32:43 +000077 if (builtins == NULL) {
78 PyErr_SetString(PyExc_RuntimeError, "lost __builtin__");
79 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");
92 if (outf == NULL) {
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"
Brett Cannon51185172006-03-02 22:07:40 +0000109"Print an object to sys.stdout and also save it in __builtin__.\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{
414 return PyInt_FromLong(_Py_CheckInterval);
415}
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{
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000476 return PyInt_FromLong(Py_GetRecursionLimit());
477}
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;
546 return PyInt_FromLong(tstate->interp->dlopenflags);
547}
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{
Martin v. Löwis725507b2006-03-07 12:08:51 +0000576 return PyInt_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{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000583 return PyInt_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\
868maxint -- the largest supported integer (the smallest is -maxint-1)\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\
Fred Drake801c08d2000-04-13 15:29:10 +0000872version -- the version of this interpreter as a string\n\
873version_info -- version information as a tuple\n\
874hexversion -- version information encoded as a single integer\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000875copyright -- copyright notice pertaining to this interpreter\n\
876platform -- platform identifier\n\
877executable -- pathname of this Python interpreter\n\
878prefix -- prefix used to find the Python library\n\
879exec_prefix -- prefix used to find the machine-specific Python library\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000880"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000881)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000882#ifdef MS_WINDOWS
883/* concatenating string here */
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000884PyDoc_STR(
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000885"dllhandle -- [Windows only] integer handle of the Python DLL\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000886winver -- [Windows only] version number of the Python DLL\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000887"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000888)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000889#endif /* MS_WINDOWS */
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000890PyDoc_STR(
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000891"__stdin__ -- the original stdin; don't touch!\n\
892__stdout__ -- the original stdout; don't touch!\n\
893__stderr__ -- the original stderr; don't touch!\n\
894__displayhook__ -- the original displayhook; don't touch!\n\
895__excepthook__ -- the original excepthook; don't touch!\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000896\n\
897Functions:\n\
898\n\
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +0000899displayhook() -- print an object to the screen, and save it in __builtin__._\n\
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000900excepthook() -- print an exception and its traceback to sys.stderr\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000901exc_info() -- return thread-safe information about the current exception\n\
902exit() -- exit the interpreter by raising SystemExit\n\
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000903getdlopenflags() -- returns flags to be used for dlopen() calls\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000904getrefcount() -- return the reference count for an object (plus one :-)\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000905getrecursionlimit() -- return the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000906setcheckinterval() -- control how often the interpreter checks for events\n\
Martin v. Löwisf0473d52001-07-18 16:17:16 +0000907setdlopenflags() -- set the flags to be used for dlopen() calls\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000908setprofile() -- set the global profiling function\n\
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000909setrecursionlimit() -- set the max recursion depth for the interpreter\n\
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000910settrace() -- set the global debug tracing function\n\
Fred Drakeccede592000-08-14 20:59:57 +0000911"
Martin v. Löwisa3fb4f72002-06-09 13:33:54 +0000912)
Fred Drakeccede592000-08-14 20:59:57 +0000913/* end of sys_doc */ ;
Guido van Rossumc3bc31e1998-06-27 19:43:25 +0000914
Martin v. Löwis43b57802006-01-05 23:38:54 +0000915/* Subversion branch and revision management */
916static const char _patchlevel_revision[] = PY_PATCHLEVEL_REVISION;
917static const char headurl[] = "$HeadURL$";
918static int svn_initialized;
919static char patchlevel_revision[50]; /* Just the number */
920static char branch[50];
921static char shortbranch[50];
922static const char *svn_revision;
923
Tim Peterse86e7a52006-01-06 02:42:46 +0000924static void
925svnversion_init(void)
Martin v. Löwis43b57802006-01-05 23:38:54 +0000926{
927 const char *python, *br_start, *br_end, *br_end2, *svnversion;
Martin v. Löwisd96ee902006-02-16 14:37:16 +0000928 Py_ssize_t len;
Collin Winterd5a5f5d2007-08-22 19:45:07 +0000929 int istag = 0;
Martin v. Löwis43b57802006-01-05 23:38:54 +0000930
931 if (svn_initialized)
932 return;
933
934 python = strstr(headurl, "/python/");
Collin Winterd5a5f5d2007-08-22 19:45:07 +0000935 if (!python) {
936 strcpy(branch, "unknown branch");
937 strcpy(shortbranch, "unknown");
Tim Peters216b78b2006-01-06 02:40:53 +0000938 }
Martin v. Löwis43b57802006-01-05 23:38:54 +0000939 else {
Collin Winterd5a5f5d2007-08-22 19:45:07 +0000940 br_start = python + 8;
941 br_end = strchr(br_start, '/');
942 assert(br_end);
943
944 /* Works even for trunk,
945 as we are in trunk/Python/sysmodule.c */
946 br_end2 = strchr(br_end+1, '/');
947
948 istag = strncmp(br_start, "tags", 4) == 0;
949 if (strncmp(br_start, "trunk", 5) == 0) {
950 strcpy(branch, "trunk");
951 strcpy(shortbranch, "trunk");
952 }
953 else if (istag || strncmp(br_start, "branches", 8) == 0) {
954 len = br_end2 - br_start;
955 strncpy(branch, br_start, len);
956 branch[len] = '\0';
957
958 len = br_end2 - (br_end + 1);
959 strncpy(shortbranch, br_end + 1, len);
960 shortbranch[len] = '\0';
961 }
962 else {
963 Py_FatalError("bad HeadURL");
964 return;
965 }
Martin v. Löwis43b57802006-01-05 23:38:54 +0000966 }
967
968
969 svnversion = _Py_svnversion();
970 if (strcmp(svnversion, "exported") != 0)
971 svn_revision = svnversion;
972 else if (istag) {
973 len = strlen(_patchlevel_revision);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000974 assert(len >= 13);
975 assert(len < (sizeof(patchlevel_revision) + 13));
Martin v. Löwis43b57802006-01-05 23:38:54 +0000976 strncpy(patchlevel_revision, _patchlevel_revision + 11,
977 len - 13);
978 patchlevel_revision[len - 13] = '\0';
979 svn_revision = patchlevel_revision;
980 }
981 else
982 svn_revision = "";
Tim Peters216b78b2006-01-06 02:40:53 +0000983
Martin v. Löwis43b57802006-01-05 23:38:54 +0000984 svn_initialized = 1;
985}
986
987/* Return svnversion output if available.
988 Else return Revision of patchlevel.h if on branch.
989 Else return empty string */
990const char*
991Py_SubversionRevision()
992{
993 svnversion_init();
994 return svn_revision;
995}
996
997const char*
998Py_SubversionShortBranch()
999{
1000 svnversion_init();
1001 return shortbranch;
1002}
1003
Guido van Rossum25ce5661997-08-02 03:10:38 +00001004PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001005_PySys_Init(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001006{
Guido van Rossum25ce5661997-08-02 03:10:38 +00001007 PyObject *m, *v, *sysdict;
Fred Drake6d27c1e2000-04-13 20:03:20 +00001008 char *s;
Guido van Rossum25ce5661997-08-02 03:10:38 +00001009
Guido van Rossumc3bc31e1998-06-27 19:43:25 +00001010 m = Py_InitModule3("sys", sys_methods, sys_doc);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001011 if (m == NULL)
1012 return NULL;
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001013 sysdict = PyModule_GetDict(m);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001014
Neal Norwitz11bd1192005-10-03 00:54:56 +00001015 {
1016 /* XXX: does this work on Win/Win64? (see posix_fstat) */
1017 struct stat sb;
1018 if (fstat(fileno(stdin), &sb) == 0 &&
1019 S_ISDIR(sb.st_mode)) {
Neal Norwitz72c2c062006-03-09 05:58:11 +00001020 /* There's nothing more we can do. */
1021 /* Py_FatalError() will core dump, so just exit. */
1022 PySys_WriteStderr("Python error: <stdin> is a directory, cannot continue\n");
1023 exit(EXIT_FAILURE);
Neal Norwitz11bd1192005-10-03 00:54:56 +00001024 }
1025 }
1026
Guido van Rossum6f376c42007-05-24 14:31:33 +00001027 /* stdin/stdout/stderr are now set by site.py. */
Martin v. Löwis5467d4c2003-05-10 07:10:12 +00001028
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001029 PyDict_SetItemString(sysdict, "__displayhook__",
1030 PyDict_GetItemString(sysdict, "displayhook"));
1031 PyDict_SetItemString(sysdict, "__excepthook__",
1032 PyDict_GetItemString(sysdict, "excepthook"));
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001033 PyDict_SetItemString(sysdict, "version",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001034 v = PyUnicode_FromString(Py_GetVersion()));
Barry Warsaw54892c41999-01-27 16:33:19 +00001035 Py_XDECREF(v);
Guido van Rossume0d7dae1999-01-03 12:55:39 +00001036 PyDict_SetItemString(sysdict, "hexversion",
1037 v = PyInt_FromLong(PY_VERSION_HEX));
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001038 Py_XDECREF(v);
Martin v. Löwis43b57802006-01-05 23:38:54 +00001039 svnversion_init();
Walter Dörwald94093ff2007-06-07 18:41:59 +00001040 v = Py_BuildValue("(UUU)", "CPython", branch, svn_revision);
Martin v. Löwis43b57802006-01-05 23:38:54 +00001041 PyDict_SetItemString(sysdict, "subversion", v);
Barry Warsaw2a38a862005-12-18 01:27:35 +00001042 Py_XDECREF(v);
Fred Drake93a20bf2000-04-13 17:44:51 +00001043 /*
1044 * These release level checks are mutually exclusive and cover
1045 * the field, so don't get too fancy with the pre-processor!
1046 */
1047#if PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_ALPHA
Fred Drake6d27c1e2000-04-13 20:03:20 +00001048 s = "alpha";
Fred Drake592f2d62000-08-31 15:21:11 +00001049#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_BETA
Fred Drake6d27c1e2000-04-13 20:03:20 +00001050 s = "beta";
Fred Drake592f2d62000-08-31 15:21:11 +00001051#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_GAMMA
Fred Drake6d27c1e2000-04-13 20:03:20 +00001052 s = "candidate";
Fred Drake592f2d62000-08-31 15:21:11 +00001053#elif PY_RELEASE_LEVEL == PY_RELEASE_LEVEL_FINAL
Fred Drake6d27c1e2000-04-13 20:03:20 +00001054 s = "final";
Fred Drake93a20bf2000-04-13 17:44:51 +00001055#endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001056
1057#define SET_SYS_FROM_STRING(key, value) \
1058 v = value; \
1059 if (v != NULL) \
1060 PyDict_SetItemString(sysdict, key, v); \
1061 Py_XDECREF(v)
1062
1063 SET_SYS_FROM_STRING("version_info",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001064 Py_BuildValue("iiiUi", PY_MAJOR_VERSION,
Fred Drake801c08d2000-04-13 15:29:10 +00001065 PY_MINOR_VERSION,
Fred Drake6d27c1e2000-04-13 20:03:20 +00001066 PY_MICRO_VERSION, s,
Fred Drake93a20bf2000-04-13 17:44:51 +00001067 PY_RELEASE_SERIAL));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001068 SET_SYS_FROM_STRING("api_version",
1069 PyInt_FromLong(PYTHON_API_VERSION));
1070 SET_SYS_FROM_STRING("copyright",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001071 PyUnicode_FromString(Py_GetCopyright()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001072 SET_SYS_FROM_STRING("platform",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001073 PyUnicode_FromString(Py_GetPlatform()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001074 SET_SYS_FROM_STRING("executable",
Neal Norwitzaefab1e2007-08-25 01:04:34 +00001075 PyUnicode_FromString(Py_GetProgramFullPath()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001076 SET_SYS_FROM_STRING("prefix",
Neal Norwitzaefab1e2007-08-25 01:04:34 +00001077 PyUnicode_FromString(Py_GetPrefix()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001078 SET_SYS_FROM_STRING("exec_prefix",
Neal Norwitzaefab1e2007-08-25 01:04:34 +00001079 PyUnicode_FromString(Py_GetExecPrefix()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001080 SET_SYS_FROM_STRING("maxint",
1081 PyInt_FromLong(PyInt_GetMax()));
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001082 SET_SYS_FROM_STRING("maxsize",
Neal Norwitz56cdf682007-10-26 06:05:28 +00001083 PyInt_FromSsize_t(PY_SSIZE_T_MAX));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001084 SET_SYS_FROM_STRING("maxunicode",
1085 PyInt_FromLong(PyUnicode_GetMax()));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001086 SET_SYS_FROM_STRING("builtin_module_names",
1087 list_builtin_module_names());
Fred Drake099325e2000-08-14 15:47:03 +00001088 {
1089 /* Assumes that longs are at least 2 bytes long.
1090 Should be safe! */
1091 unsigned long number = 1;
Fred Drakea2b6ad62000-08-15 04:24:43 +00001092 char *value;
Fred Drake099325e2000-08-14 15:47:03 +00001093
1094 s = (char *) &number;
1095 if (s[0] == 0)
Fred Drakea2b6ad62000-08-15 04:24:43 +00001096 value = "big";
Fred Drake099325e2000-08-14 15:47:03 +00001097 else
Fred Drakea2b6ad62000-08-15 04:24:43 +00001098 value = "little";
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001099 SET_SYS_FROM_STRING("byteorder",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001100 PyUnicode_FromString(value));
Fred Drake099325e2000-08-14 15:47:03 +00001101 }
Guido van Rossum8b9ea871996-08-23 18:14:47 +00001102#ifdef MS_COREDLL
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001103 SET_SYS_FROM_STRING("dllhandle",
1104 PyLong_FromVoidPtr(PyWin_DLLhModule));
1105 SET_SYS_FROM_STRING("winver",
Walter Dörwald94093ff2007-06-07 18:41:59 +00001106 PyUnicode_FromString(PyWin_DLLVersionString));
Guido van Rossumc606fe11996-04-09 02:37:57 +00001107#endif
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001108#undef SET_SYS_FROM_STRING
Guido van Rossum23fff912000-12-15 22:02:05 +00001109 if (warnoptions == NULL) {
1110 warnoptions = PyList_New(0);
1111 }
1112 else {
1113 Py_INCREF(warnoptions);
1114 }
1115 if (warnoptions != NULL) {
Guido van Rossum03df3b32001-01-13 22:06:05 +00001116 PyDict_SetItemString(sysdict, "warnoptions", warnoptions);
Guido van Rossum23fff912000-12-15 22:02:05 +00001117 }
Tim Peters216b78b2006-01-06 02:40:53 +00001118
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001119 if (PyErr_Occurred())
Guido van Rossum25ce5661997-08-02 03:10:38 +00001120 return NULL;
1121 return m;
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001122}
1123
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001124static PyObject *
Neal Norwitzf3081322007-08-25 00:32:45 +00001125makepathobject(const char *path, int delim)
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001126{
Guido van Rossum3f5da241990-12-20 15:06:42 +00001127 int i, n;
Neal Norwitzf3081322007-08-25 00:32:45 +00001128 const char *p;
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001129 PyObject *v, *w;
Tim Peters216b78b2006-01-06 02:40:53 +00001130
Guido van Rossum3f5da241990-12-20 15:06:42 +00001131 n = 1;
1132 p = path;
1133 while ((p = strchr(p, delim)) != NULL) {
1134 n++;
1135 p++;
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001136 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001137 v = PyList_New(n);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001138 if (v == NULL)
1139 return NULL;
1140 for (i = 0; ; i++) {
1141 p = strchr(path, delim);
1142 if (p == NULL)
1143 p = strchr(path, '\0'); /* End of string */
Christian Heimes5894ba72007-11-04 11:43:14 +00001144 w = PyUnicode_DecodeFSDefaultAndSize(path, (Py_ssize_t) (p - path));
Guido van Rossum3f5da241990-12-20 15:06:42 +00001145 if (w == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001146 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001147 return NULL;
1148 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001149 PyList_SetItem(v, i, w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001150 if (*p == '\0')
1151 break;
1152 path = p+1;
1153 }
1154 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001155}
1156
1157void
Neal Norwitzf3081322007-08-25 00:32:45 +00001158PySys_SetPath(const char *path)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001159{
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001160 PyObject *v;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001161 if ((v = makepathobject(path, DELIM)) == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001162 Py_FatalError("can't create sys.path");
1163 if (PySys_SetObject("path", v) != 0)
1164 Py_FatalError("can't assign sys.path");
1165 Py_DECREF(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001166}
1167
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001168static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001169makeargvobject(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001170{
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001171 PyObject *av;
Guido van Rossumee3a2991992-01-14 18:42:53 +00001172 if (argc <= 0 || argv == NULL) {
1173 /* Ensure at least one (empty) argument is seen */
1174 static char *empty_argv[1] = {""};
1175 argv = empty_argv;
1176 argc = 1;
1177 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001178 av = PyList_New(argc);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001179 if (av != NULL) {
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001180 int i;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001181 for (i = 0; i < argc; i++) {
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001182#ifdef __VMS
1183 PyObject *v;
1184
1185 /* argv[0] is the script pathname if known */
1186 if (i == 0) {
1187 char* fn = decc$translate_vms(argv[0]);
1188 if ((fn == (char *)0) || fn == (char *)-1)
Guido van Rossum96ca6912007-06-15 03:49:03 +00001189 v = PyUnicode_FromString(argv[0]);
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001190 else
Guido van Rossum96ca6912007-06-15 03:49:03 +00001191 v = PyUnicode_FromString(
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001192 decc$translate_vms(argv[0]));
1193 } else
Guido van Rossum96ca6912007-06-15 03:49:03 +00001194 v = PyUnicode_FromString(argv[i]);
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001195#else
Guido van Rossum96ca6912007-06-15 03:49:03 +00001196 PyObject *v = PyUnicode_FromString(argv[i]);
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00001197#endif
Guido van Rossum3f5da241990-12-20 15:06:42 +00001198 if (v == NULL) {
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001199 Py_DECREF(av);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001200 av = NULL;
1201 break;
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001202 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001203 PyList_SetItem(av, i, v);
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001204 }
Guido van Rossum5b3138b1990-11-18 17:41:40 +00001205 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001206 return av;
1207}
1208
1209void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001210PySys_SetArgv(int argc, char **argv)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001211{
Guido van Rossum162e38c2003-02-19 15:25:10 +00001212#if defined(HAVE_REALPATH)
1213 char fullpath[MAXPATHLEN];
1214#elif defined(MS_WINDOWS)
Thomas Heller27bb71e2003-01-08 14:33:48 +00001215 char fullpath[MAX_PATH];
1216#endif
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001217 PyObject *av = makeargvobject(argc, argv);
1218 PyObject *path = PySys_GetObject("path");
Guido van Rossum3f5da241990-12-20 15:06:42 +00001219 if (av == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001220 Py_FatalError("no mem for sys.argv");
1221 if (PySys_SetObject("argv", av) != 0)
1222 Py_FatalError("can't assign sys.argv");
Guido van Rossum94a96671996-07-30 20:35:50 +00001223 if (path != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +00001224 char *argv0 = argv[0];
Guido van Rossum94a96671996-07-30 20:35:50 +00001225 char *p = NULL;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001226 Py_ssize_t n = 0;
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001227 PyObject *a;
Guido van Rossumc474dea1997-04-25 15:38:31 +00001228#ifdef HAVE_READLINK
1229 char link[MAXPATHLEN+1];
1230 char argv0copy[2*MAXPATHLEN+1];
1231 int nr = 0;
Georg Brandl69537722005-09-15 13:00:34 +00001232 if (argc > 0 && argv0 != NULL && strcmp(argv0, "-c") != 0)
Guido van Rossumc474dea1997-04-25 15:38:31 +00001233 nr = readlink(argv0, link, MAXPATHLEN);
1234 if (nr > 0) {
1235 /* It's a symlink */
1236 link[nr] = '\0';
1237 if (link[0] == SEP)
1238 argv0 = link; /* Link to absolute path */
1239 else if (strchr(link, SEP) == NULL)
1240 ; /* Link without path */
1241 else {
1242 /* Must join(dirname(argv0), link) */
1243 char *q = strrchr(argv0, SEP);
1244 if (q == NULL)
1245 argv0 = link; /* argv0 without path */
1246 else {
1247 /* Must make a copy */
1248 strcpy(argv0copy, argv0);
1249 q = strrchr(argv0copy, SEP);
1250 strcpy(q+1, link);
1251 argv0 = argv0copy;
1252 }
1253 }
1254 }
1255#endif /* HAVE_READLINK */
Guido van Rossumcc883411996-09-10 14:44:21 +00001256#if SEP == '\\' /* Special case for MS filename syntax */
Georg Brandl69537722005-09-15 13:00:34 +00001257 if (argc > 0 && argv0 != NULL && strcmp(argv0, "-c") != 0) {
Guido van Rossumcc883411996-09-10 14:44:21 +00001258 char *q;
Thomas Heller27bb71e2003-01-08 14:33:48 +00001259#ifdef MS_WINDOWS
1260 char *ptemp;
1261 if (GetFullPathName(argv0,
1262 sizeof(fullpath),
1263 fullpath,
1264 &ptemp)) {
1265 argv0 = fullpath;
1266 }
1267#endif
Guido van Rossumc474dea1997-04-25 15:38:31 +00001268 p = strrchr(argv0, SEP);
Guido van Rossumcc883411996-09-10 14:44:21 +00001269 /* Test for alternate separator */
Guido van Rossumc474dea1997-04-25 15:38:31 +00001270 q = strrchr(p ? p : argv0, '/');
Guido van Rossumcc883411996-09-10 14:44:21 +00001271 if (q != NULL)
1272 p = q;
1273 if (p != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +00001274 n = p + 1 - argv0;
Guido van Rossumcc883411996-09-10 14:44:21 +00001275 if (n > 1 && p[-1] != ':')
1276 n--; /* Drop trailing separator */
1277 }
1278 }
1279#else /* All other filename syntaxes */
Georg Brandl69537722005-09-15 13:00:34 +00001280 if (argc > 0 && argv0 != NULL && strcmp(argv0, "-c") != 0) {
Guido van Rossum162e38c2003-02-19 15:25:10 +00001281#if defined(HAVE_REALPATH)
1282 if (realpath(argv0, fullpath)) {
1283 argv0 = fullpath;
1284 }
1285#endif
Guido van Rossumc474dea1997-04-25 15:38:31 +00001286 p = strrchr(argv0, SEP);
Guido van Rossum162e38c2003-02-19 15:25:10 +00001287 }
Guido van Rossumcc883411996-09-10 14:44:21 +00001288 if (p != NULL) {
Guido van Rossumc474dea1997-04-25 15:38:31 +00001289 n = p + 1 - argv0;
Guido van Rossumcc883411996-09-10 14:44:21 +00001290#if SEP == '/' /* Special case for Unix filename syntax */
1291 if (n > 1)
1292 n--; /* Drop trailing separator */
1293#endif /* Unix */
1294 }
1295#endif /* All others */
Guido van Rossum96ca6912007-06-15 03:49:03 +00001296 a = PyUnicode_FromStringAndSize(argv0, n);
Guido van Rossum94a96671996-07-30 20:35:50 +00001297 if (a == NULL)
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001298 Py_FatalError("no mem for sys.path insertion");
1299 if (PyList_Insert(path, 0, a) < 0)
1300 Py_FatalError("sys.path.insert(0) failed");
1301 Py_DECREF(a);
Guido van Rossuma63d9f41996-07-24 01:31:37 +00001302 }
Guido van Rossum65bf9f21997-04-29 18:33:38 +00001303 Py_DECREF(av);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001304}
Guido van Rossuma890e681998-05-12 14:59:24 +00001305
1306
1307/* APIs to write to sys.stdout or sys.stderr using a printf-like interface.
1308 Adapted from code submitted by Just van Rossum.
1309
1310 PySys_WriteStdout(format, ...)
1311 PySys_WriteStderr(format, ...)
1312
1313 The first function writes to sys.stdout; the second to sys.stderr. When
1314 there is a problem, they write to the real (C level) stdout or stderr;
Guido van Rossum8442af31998-10-12 18:22:10 +00001315 no exceptions are raised.
Guido van Rossuma890e681998-05-12 14:59:24 +00001316
1317 Both take a printf-style format string as their first argument followed
1318 by a variable length argument list determined by the format string.
1319
1320 *** WARNING ***
1321
1322 The format should limit the total size of the formatted output string to
1323 1000 bytes. In particular, this means that no unrestricted "%s" formats
1324 should occur; these should be limited using "%.<N>s where <N> is a
1325 decimal number calculated so that <N> plus the maximum size of other
1326 formatted text does not exceed 1000 bytes. Also watch out for "%f",
1327 which can print hundreds of digits for very large numbers.
1328
1329 */
1330
1331static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001332mywrite(char *name, FILE *fp, const char *format, va_list va)
Guido van Rossuma890e681998-05-12 14:59:24 +00001333{
1334 PyObject *file;
Guido van Rossum8442af31998-10-12 18:22:10 +00001335 PyObject *error_type, *error_value, *error_traceback;
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001336 char buffer[1001];
1337 int written;
Guido van Rossuma890e681998-05-12 14:59:24 +00001338
Guido van Rossum8442af31998-10-12 18:22:10 +00001339 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +00001340 file = PySys_GetObject(name);
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001341 written = PyOS_vsnprintf(buffer, sizeof(buffer), format, va);
1342 if (PyFile_WriteString(buffer, file) != 0) {
1343 PyErr_Clear();
1344 fputs(buffer, fp);
1345 }
1346 if (written < 0 || (size_t)written >= sizeof(buffer)) {
1347 const char *truncated = "... truncated";
1348 if (PyFile_WriteString(truncated, file) != 0) {
Guido van Rossuma890e681998-05-12 14:59:24 +00001349 PyErr_Clear();
Guido van Rossumda5b8f22007-06-12 23:30:11 +00001350 fputs(truncated, fp);
Jeremy Hylton5d3d1342001-11-28 21:44:53 +00001351 }
Guido van Rossuma890e681998-05-12 14:59:24 +00001352 }
Guido van Rossum8442af31998-10-12 18:22:10 +00001353 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossuma890e681998-05-12 14:59:24 +00001354}
1355
1356void
Guido van Rossuma890e681998-05-12 14:59:24 +00001357PySys_WriteStdout(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +00001358{
1359 va_list va;
1360
Guido van Rossuma890e681998-05-12 14:59:24 +00001361 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +00001362 mywrite("stdout", stdout, format, va);
1363 va_end(va);
1364}
1365
1366void
Guido van Rossuma890e681998-05-12 14:59:24 +00001367PySys_WriteStderr(const char *format, ...)
Guido van Rossuma890e681998-05-12 14:59:24 +00001368{
1369 va_list va;
1370
Guido van Rossuma890e681998-05-12 14:59:24 +00001371 va_start(va, format);
Guido van Rossuma890e681998-05-12 14:59:24 +00001372 mywrite("stderr", stderr, format, va);
1373 va_end(va);
1374}