/* System module */

/*

Various bits of information used by the interpreter are collected in

module 'sys'.

Function member:

- exit(sts): raise SystemExit

Data members:

- stdin, stdout, stderr: standard file objects

- modules: the table of modules (dictionary)

- path: module search path (list of strings)

- argv: script arguments (list of strings)

- ps1, ps2: optional primary and secondary prompts (strings)

*/

#include "Python.h"

#include "code.h"

#include "frameobject.h"

#include "pythread.h"

#include "osdefs.h"

#ifdef MS_WINDOWS

#define WIN32_LEAN_AND_MEAN

#include <windows.h>

#endif /* MS_WINDOWS */

#ifdef MS_COREDLL

extern void *PyWin_DLLhModule;

/* A string loaded from the DLL at startup: */

extern const char *PyWin_DLLVersionString;

#endif

#ifdef HAVE_LANGINFO_H

#include <locale.h>

#include <langinfo.h>

#endif

_Py_IDENTIFIER(_);

_Py_IDENTIFIER(__sizeof__);

_Py_IDENTIFIER(buffer);

_Py_IDENTIFIER(builtins);

_Py_IDENTIFIER(encoding);

_Py_IDENTIFIER(path);

_Py_IDENTIFIER(stdout);

_Py_IDENTIFIER(stderr);

_Py_IDENTIFIER(write);

PyObject *

_PySys_GetObjectId(_Py_Identifier *key)

{

PyThreadState *tstate = PyThreadState_GET();

PyObject *sd = tstate->interp->sysdict;

if (sd == NULL)

return NULL;

return _PyDict_GetItemId(sd, key);

}

PyObject *

PySys_GetObject(const char *name)

{

PyThreadState *tstate = PyThreadState_GET();

PyObject *sd = tstate->interp->sysdict;

if (sd == NULL)

return NULL;

return PyDict_GetItemString(sd, name);

}

int

_PySys_SetObjectId(_Py_Identifier *key, PyObject *v)

{

PyThreadState *tstate = PyThreadState_GET();

PyObject *sd = tstate->interp->sysdict;

if (v == NULL) {

if (_PyDict_GetItemId(sd, key) == NULL)

return 0;

else

return _PyDict_DelItemId(sd, key);

}

else

return _PyDict_SetItemId(sd, key, v);

}

int

PySys_SetObject(const char *name, PyObject *v)

{

PyThreadState *tstate = PyThreadState_GET();

PyObject *sd = tstate->interp->sysdict;

if (v == NULL) {

if (PyDict_GetItemString(sd, name) == NULL)

return 0;

else

return PyDict_DelItemString(sd, name);

}

else

return PyDict_SetItemString(sd, name, v);

}

/* Write repr(o) to sys.stdout using sys.stdout.encoding and 'backslashreplace'

error handler. If sys.stdout has a buffer attribute, use

sys.stdout.buffer.write(encoded), otherwise redecode the string and use

sys.stdout.write(redecoded).

Helper function for sys_displayhook(). */

static int

sys_displayhook_unencodable(PyObject *outf, PyObject *o)

{

PyObject *stdout_encoding = NULL;

PyObject *encoded, *escaped_str, *repr_str, *buffer, *result;

char *stdout_encoding_str;

int ret;

stdout_encoding = _PyObject_GetAttrId(outf, &PyId_encoding);

if (stdout_encoding == NULL)

goto error;

stdout_encoding_str = _PyUnicode_AsString(stdout_encoding);

if (stdout_encoding_str == NULL)

goto error;

repr_str = PyObject_Repr(o);

if (repr_str == NULL)

goto error;

encoded = PyUnicode_AsEncodedString(repr_str,

stdout_encoding_str,

"backslashreplace");

Py_DECREF(repr_str);

if (encoded == NULL)

goto error;

buffer = _PyObject_GetAttrId(outf, &PyId_buffer);

if (buffer) {

result = _PyObject_CallMethodId(buffer, &PyId_write, "(O)", encoded);

Py_DECREF(buffer);

Py_DECREF(encoded);

if (result == NULL)

goto error;

Py_DECREF(result);

}

else {

PyErr_Clear();

escaped_str = PyUnicode_FromEncodedObject(encoded,

stdout_encoding_str,

"strict");

Py_DECREF(encoded);

if (PyFile_WriteObject(escaped_str, outf, Py_PRINT_RAW) != 0) {

Py_DECREF(escaped_str);

goto error;

}

Py_DECREF(escaped_str);

}

ret = 0;

goto finally;

error:

ret = -1;

finally:

Py_XDECREF(stdout_encoding);

return ret;

}

static PyObject *

sys_displayhook(PyObject *self, PyObject *o)

{

PyObject *outf;

PyInterpreterState *interp = PyThreadState_GET()->interp;

PyObject *modules = interp->modules;

PyObject *builtins;

static PyObject *newline = NULL;

int err;

builtins = _PyDict_GetItemId(modules, &PyId_builtins);

if (builtins == NULL) {

PyErr_SetString(PyExc_RuntimeError, "lost builtins module");

return NULL;

}

/* Print value except if None */

/* After printing, also assign to '_' */

/* Before, set '_' to None to avoid recursion */

if (o == Py_None) {

Py_INCREF(Py_None);

return Py_None;

}

if (_PyObject_SetAttrId(builtins, &PyId__, Py_None) != 0)

return NULL;

outf = _PySys_GetObjectId(&PyId_stdout);

if (outf == NULL || outf == Py_None) {

PyErr_SetString(PyExc_RuntimeError, "lost sys.stdout");

return NULL;

}

if (PyFile_WriteObject(o, outf, 0) != 0) {

if (PyErr_ExceptionMatches(PyExc_UnicodeEncodeError)) {

/* repr(o) is not encodable to sys.stdout.encoding with

* sys.stdout.errors error handler (which is probably 'strict') */

PyErr_Clear();

err = sys_displayhook_unencodable(outf, o);

if (err)

return NULL;

}

else {

return NULL;

}

}

if (newline == NULL) {

newline = PyUnicode_FromString("\n");

if (newline == NULL)

return NULL;

}

if (PyFile_WriteObject(newline, outf, Py_PRINT_RAW) != 0)

return NULL;

if (_PyObject_SetAttrId(builtins, &PyId__, o) != 0)

return NULL;

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(displayhook_doc,

"displayhook(object) -> None\n"

"\n"

"Print an object to sys.stdout and also save it in builtins._\n"

);

static PyObject *

sys_excepthook(PyObject* self, PyObject* args)

{

PyObject *exc, *value, *tb;

if (!PyArg_UnpackTuple(args, "excepthook", 3, 3, &exc, &value, &tb))

return NULL;

PyErr_Display(exc, value, tb);

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(excepthook_doc,

"excepthook(exctype, value, traceback) -> None\n"

"\n"

"Handle an exception by displaying it with a traceback on sys.stderr.\n"

);

static PyObject *

sys_exc_info(PyObject *self, PyObject *noargs)

{

PyThreadState *tstate;

tstate = PyThreadState_GET();

return Py_BuildValue(

"(OOO)",

tstate->exc_type != NULL ? tstate->exc_type : Py_None,

tstate->exc_value != NULL ? tstate->exc_value : Py_None,

tstate->exc_traceback != NULL ?

tstate->exc_traceback : Py_None);

}

PyDoc_STRVAR(exc_info_doc,

"exc_info() -> (type, value, traceback)\n\

\n\

Return information about the most recent exception caught by an except\n\

clause in the current stack frame or in an older stack frame."

);

static PyObject *

sys_exit(PyObject *self, PyObject *args)

{

PyObject *exit_code = 0;

if (!PyArg_UnpackTuple(args, "exit", 0, 1, &exit_code))

return NULL;

/* Raise SystemExit so callers may catch it or clean up. */

PyErr_SetObject(PyExc_SystemExit, exit_code);

return NULL;

}

PyDoc_STRVAR(exit_doc,

"exit([status])\n\

\n\

Exit the interpreter by raising SystemExit(status).\n\

If the status is omitted or None, it defaults to zero (i.e., success).\n\

If the status is an integer, it will be used as the system exit status.\n\

If it is another kind of object, it will be printed and the system\n\

exit status will be one (i.e., failure)."

);

static PyObject *

sys_getdefaultencoding(PyObject *self)

{

return PyUnicode_FromString(PyUnicode_GetDefaultEncoding());

}

PyDoc_STRVAR(getdefaultencoding_doc,

"getdefaultencoding() -> string\n\

\n\

Return the current default string encoding used by the Unicode \n\

implementation."

);

static PyObject *

sys_getfilesystemencoding(PyObject *self)

{

if (Py_FileSystemDefaultEncoding)

return PyUnicode_FromString(Py_FileSystemDefaultEncoding);

PyErr_SetString(PyExc_RuntimeError,

"filesystem encoding is not initialized");

return NULL;

}

PyDoc_STRVAR(getfilesystemencoding_doc,

"getfilesystemencoding() -> string\n\

\n\

Return the encoding used to convert Unicode filenames in\n\

operating system filenames."

);

static PyObject *

sys_intern(PyObject *self, PyObject *args)

{

PyObject *s;

if (!PyArg_ParseTuple(args, "U:intern", &s))

return NULL;

if (PyUnicode_CheckExact(s)) {

Py_INCREF(s);

PyUnicode_InternInPlace(&s);

return s;

}

else {

PyErr_Format(PyExc_TypeError,

"can't intern %.400s", s->ob_type->tp_name);

return NULL;

}

}

PyDoc_STRVAR(intern_doc,

"intern(string) -> string\n\

\n\

``Intern'' the given string. This enters the string in the (global)\n\

table of interned strings whose purpose is to speed up dictionary lookups.\n\

Return the string itself or the previously interned string object with the\n\

same value.");

/*

* Cached interned string objects used for calling the profile and

* trace functions. Initialized by trace_init().

*/

static PyObject *whatstrings[7] = {NULL, NULL, NULL, NULL, NULL, NULL, NULL};

static int

trace_init(void)

{

static char *whatnames[7] = {"call", "exception", "line", "return",

"c_call", "c_exception", "c_return"};

PyObject *name;

int i;

for (i = 0; i < 7; ++i) {

if (whatstrings[i] == NULL) {

name = PyUnicode_InternFromString(whatnames[i]);

if (name == NULL)

return -1;

whatstrings[i] = name;

}

}

return 0;

}

static PyObject *

call_trampoline(PyObject* callback,

PyFrameObject *frame, int what, PyObject *arg)

{

PyObject *args;

PyObject *whatstr;

PyObject *result;

args = PyTuple_New(3);

if (args == NULL)

return NULL;

if (PyFrame_FastToLocalsWithError(frame) < 0)

return NULL;

Py_INCREF(frame);

whatstr = whatstrings[what];

Py_INCREF(whatstr);

if (arg == NULL)

arg = Py_None;

Py_INCREF(arg);

PyTuple_SET_ITEM(args, 0, (PyObject *)frame);

PyTuple_SET_ITEM(args, 1, whatstr);

PyTuple_SET_ITEM(args, 2, arg);

/* call the Python-level function */

result = PyEval_CallObject(callback, args);

PyFrame_LocalsToFast(frame, 1);

if (result == NULL)

PyTraceBack_Here(frame);

/* cleanup */

Py_DECREF(args);

return result;

}

static int

profile_trampoline(PyObject *self, PyFrameObject *frame,

int what, PyObject *arg)

{

PyObject *result;

if (arg == NULL)

arg = Py_None;

result = call_trampoline(self, frame, what, arg);

if (result == NULL) {

PyEval_SetProfile(NULL, NULL);

return -1;

}

Py_DECREF(result);

return 0;

}

static int

trace_trampoline(PyObject *self, PyFrameObject *frame,

int what, PyObject *arg)

{

PyObject *callback;

PyObject *result;

if (what == PyTrace_CALL)

callback = self;

else

callback = frame->f_trace;

if (callback == NULL)

return 0;

result = call_trampoline(callback, frame, what, arg);

if (result == NULL) {

PyEval_SetTrace(NULL, NULL);

Py_CLEAR(frame->f_trace);

return -1;

}

if (result != Py_None) {

PyObject *temp = frame->f_trace;

frame->f_trace = NULL;

Py_XDECREF(temp);

frame->f_trace = result;

}

else {

Py_DECREF(result);

}

return 0;

}

static PyObject *

sys_settrace(PyObject *self, PyObject *args)

{

if (trace_init() == -1)

return NULL;

if (args == Py_None)

PyEval_SetTrace(NULL, NULL);

else

PyEval_SetTrace(trace_trampoline, args);

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(settrace_doc,

"settrace(function)\n\

\n\

Set the global debug tracing function. It will be called on each\n\

function call. See the debugger chapter in the library manual."

);

static PyObject *

sys_gettrace(PyObject *self, PyObject *args)

{

PyThreadState *tstate = PyThreadState_GET();

PyObject *temp = tstate->c_traceobj;

if (temp == NULL)

temp = Py_None;

Py_INCREF(temp);

return temp;

}

PyDoc_STRVAR(gettrace_doc,

"gettrace()\n\

\n\

Return the global debug tracing function set with sys.settrace.\n\

See the debugger chapter in the library manual."

);

static PyObject *

sys_setprofile(PyObject *self, PyObject *args)

{

if (trace_init() == -1)

return NULL;

if (args == Py_None)

PyEval_SetProfile(NULL, NULL);

else

PyEval_SetProfile(profile_trampoline, args);

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(setprofile_doc,

"setprofile(function)\n\

\n\

Set the profiling function. It will be called on each function call\n\

and return. See the profiler chapter in the library manual."

);

static PyObject *

sys_getprofile(PyObject *self, PyObject *args)

{

PyThreadState *tstate = PyThreadState_GET();

PyObject *temp = tstate->c_profileobj;

if (temp == NULL)

temp = Py_None;

Py_INCREF(temp);

return temp;

}

PyDoc_STRVAR(getprofile_doc,

"getprofile()\n\

\n\

Return the profiling function set with sys.setprofile.\n\

See the profiler chapter in the library manual."

);

static int _check_interval = 100;

static PyObject *

sys_setcheckinterval(PyObject *self, PyObject *args)

{

if (PyErr_WarnEx(PyExc_DeprecationWarning,

"sys.getcheckinterval() and sys.setcheckinterval() "

"are deprecated. Use sys.setswitchinterval() "

"instead.", 1) < 0)

return NULL;

if (!PyArg_ParseTuple(args, "i:setcheckinterval", &_check_interval))

return NULL;

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(setcheckinterval_doc,

"setcheckinterval(n)\n\

\n\

Tell the Python interpreter to check for asynchronous events every\n\

n instructions. This also affects how often thread switches occur."

);

static PyObject *

sys_getcheckinterval(PyObject *self, PyObject *args)

{

if (PyErr_WarnEx(PyExc_DeprecationWarning,

"sys.getcheckinterval() and sys.setcheckinterval() "

"are deprecated. Use sys.getswitchinterval() "

"instead.", 1) < 0)

return NULL;

return PyLong_FromLong(_check_interval);

}

PyDoc_STRVAR(getcheckinterval_doc,

"getcheckinterval() -> current check interval; see setcheckinterval()."

);

#ifdef WITH_THREAD

static PyObject *

sys_setswitchinterval(PyObject *self, PyObject *args)

{

double d;

if (!PyArg_ParseTuple(args, "d:setswitchinterval", &d))

return NULL;

if (d <= 0.0) {

PyErr_SetString(PyExc_ValueError,

"switch interval must be strictly positive");

return NULL;

}

_PyEval_SetSwitchInterval((unsigned long) (1e6 * d));

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(setswitchinterval_doc,

"setswitchinterval(n)\n\

\n\

Set the ideal thread switching delay inside the Python interpreter\n\

The actual frequency of switching threads can be lower if the\n\

interpreter executes long sequences of uninterruptible code\n\

(this is implementation-specific and workload-dependent).\n\

\n\

The parameter must represent the desired switching delay in seconds\n\

A typical value is 0.005 (5 milliseconds)."

);

static PyObject *

sys_getswitchinterval(PyObject *self, PyObject *args)

{

return PyFloat_FromDouble(1e-6 * _PyEval_GetSwitchInterval());

}

PyDoc_STRVAR(getswitchinterval_doc,

"getswitchinterval() -> current thread switch interval; see setswitchinterval()."

);

#endif /* WITH_THREAD */

#ifdef WITH_TSC

static PyObject *

sys_settscdump(PyObject *self, PyObject *args)

{

int bool;

PyThreadState *tstate = PyThreadState_Get();

if (!PyArg_ParseTuple(args, "i:settscdump", &bool))

return NULL;

if (bool)

tstate->interp->tscdump = 1;

else

tstate->interp->tscdump = 0;

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(settscdump_doc,

"settscdump(bool)\n\

\n\

If true, tell the Python interpreter to dump VM measurements to\n\

stderr. If false, turn off dump. The measurements are based on the\n\

processor's time-stamp counter."

);

#endif /* TSC */

static PyObject *

sys_setrecursionlimit(PyObject *self, PyObject *args)

{

int new_limit;

if (!PyArg_ParseTuple(args, "i:setrecursionlimit", &new_limit))

return NULL;

if (new_limit <= 0) {

PyErr_SetString(PyExc_ValueError,

"recursion limit must be positive");

return NULL;

}

Py_SetRecursionLimit(new_limit);

Py_INCREF(Py_None);

return Py_None;

}

static PyObject *

sys_set_coroutine_wrapper(PyObject *self, PyObject *wrapper)

{

if (wrapper != Py_None) {

if (!PyCallable_Check(wrapper)) {

PyErr_Format(PyExc_TypeError,

"callable expected, got %.50s",

Py_TYPE(wrapper)->tp_name);

return NULL;

}

PyEval_SetCoroutineWrapper(wrapper);

}

else {

PyEval_SetCoroutineWrapper(NULL);

}

Py_RETURN_NONE;

}

PyDoc_STRVAR(set_coroutine_wrapper_doc,

"set_coroutine_wrapper(wrapper)\n\

\n\

Set a wrapper for coroutine objects."

);

static PyObject *

sys_get_coroutine_wrapper(PyObject *self, PyObject *args)

{

PyObject *wrapper = PyEval_GetCoroutineWrapper();

if (wrapper == NULL) {

wrapper = Py_None;

}

Py_INCREF(wrapper);

return wrapper;

}

PyDoc_STRVAR(get_coroutine_wrapper_doc,

"get_coroutine_wrapper()\n\

\n\

Return the wrapper for coroutine objects set by sys.set_coroutine_wrapper."

);

static PyTypeObject Hash_InfoType;

PyDoc_STRVAR(hash_info_doc,

"hash_info\n\

\n\

A struct sequence providing parameters used for computing\n\

hashes. The attributes are read only.");

static PyStructSequence_Field hash_info_fields[] = {

{"width", "width of the type used for hashing, in bits"},

{"modulus", "prime number giving the modulus on which the hash "

"function is based"},

{"inf", "value to be used for hash of a positive infinity"},

{"nan", "value to be used for hash of a nan"},

{"imag", "multiplier used for the imaginary part of a complex number"},

{"algorithm", "name of the algorithm for hashing of str, bytes and "

"memoryviews"},

{"hash_bits", "internal output size of hash algorithm"},

{"seed_bits", "seed size of hash algorithm"},

{"cutoff", "small string optimization cutoff"},

{NULL, NULL}

};

static PyStructSequence_Desc hash_info_desc = {

"sys.hash_info",

hash_info_doc,

hash_info_fields,

9,

};

static PyObject *

get_hash_info(void)

{

PyObject *hash_info;

int field = 0;

PyHash_FuncDef *hashfunc;

hash_info = PyStructSequence_New(&Hash_InfoType);

if (hash_info == NULL)

return NULL;

hashfunc = PyHash_GetFuncDef();

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromLong(8*sizeof(Py_hash_t)));

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromSsize_t(_PyHASH_MODULUS));

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromLong(_PyHASH_INF));

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromLong(_PyHASH_NAN));

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromLong(_PyHASH_IMAG));

PyStructSequence_SET_ITEM(hash_info, field++,

PyUnicode_FromString(hashfunc->name));

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromLong(hashfunc->hash_bits));

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromLong(hashfunc->seed_bits));

PyStructSequence_SET_ITEM(hash_info, field++,

PyLong_FromLong(Py_HASH_CUTOFF));

if (PyErr_Occurred()) {

Py_CLEAR(hash_info);

return NULL;

}

return hash_info;

}

PyDoc_STRVAR(setrecursionlimit_doc,

"setrecursionlimit(n)\n\

\n\

Set the maximum depth of the Python interpreter stack to n. This\n\

limit prevents infinite recursion from causing an overflow of the C\n\

stack and crashing Python. The highest possible limit is platform-\n\

dependent."

);

static PyObject *

sys_getrecursionlimit(PyObject *self)

{

return PyLong_FromLong(Py_GetRecursionLimit());

}

PyDoc_STRVAR(getrecursionlimit_doc,

"getrecursionlimit()\n\

\n\

Return the current value of the recursion limit, the maximum depth\n\

of the Python interpreter stack. This limit prevents infinite\n\

recursion from causing an overflow of the C stack and crashing Python."

);

#ifdef MS_WINDOWS

PyDoc_STRVAR(getwindowsversion_doc,

"getwindowsversion()\n\

\n\

Return information about the running version of Windows as a named tuple.\n\

The members are named: major, minor, build, platform, service_pack,\n\

service_pack_major, service_pack_minor, suite_mask, and product_type. For\n\

backward compatibility, only the first 5 items are available by indexing.\n\

All elements are numbers, except service_pack which is a string. Platform\n\

may be 0 for win32s, 1 for Windows 9x/ME, 2 for Windows NT/2000/XP/Vista/7,\n\

3 for Windows CE. Product_type may be 1 for a workstation, 2 for a domain\n\

controller, 3 for a server."

);

static PyTypeObject WindowsVersionType = {0, 0, 0, 0, 0, 0};

static PyStructSequence_Field windows_version_fields[] = {

{"major", "Major version number"},

{"minor", "Minor version number"},

{"build", "Build number"},

{"platform", "Operating system platform"},

{"service_pack", "Latest Service Pack installed on the system"},

{"service_pack_major", "Service Pack major version number"},

{"service_pack_minor", "Service Pack minor version number"},

{"suite_mask", "Bit mask identifying available product suites"},

{"product_type", "System product type"},

{0}

};

static PyStructSequence_Desc windows_version_desc = {

"sys.getwindowsversion", /* name */

getwindowsversion_doc, /* doc */

windows_version_fields, /* fields */

5 /* For backward compatibility,

only the first 5 items are accessible

via indexing, the rest are name only */

};

/* Disable deprecation warnings about GetVersionEx as the result is

being passed straight through to the caller, who is responsible for

using it correctly. */

#pragma warning(push)

#pragma warning(disable:4996)

static PyObject *

sys_getwindowsversion(PyObject *self)

{

PyObject *version;

int pos = 0;

OSVERSIONINFOEX ver;

ver.dwOSVersionInfoSize = sizeof(ver);

if (!GetVersionEx((OSVERSIONINFO*) &ver))

return PyErr_SetFromWindowsErr(0);

version = PyStructSequence_New(&WindowsVersionType);

if (version == NULL)

return NULL;

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.dwMajorVersion));

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.dwMinorVersion));

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.dwBuildNumber));

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.dwPlatformId));

PyStructSequence_SET_ITEM(version, pos++, PyUnicode_FromString(ver.szCSDVersion));

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.wServicePackMajor));

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.wServicePackMinor));

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.wSuiteMask));

PyStructSequence_SET_ITEM(version, pos++, PyLong_FromLong(ver.wProductType));

if (PyErr_Occurred()) {

Py_DECREF(version);

return NULL;

}

return version;

}

#pragma warning(pop)

#endif /* MS_WINDOWS */

#ifdef HAVE_DLOPEN

static PyObject *

sys_setdlopenflags(PyObject *self, PyObject *args)

{

int new_val;

PyThreadState *tstate = PyThreadState_GET();

if (!PyArg_ParseTuple(args, "i:setdlopenflags", &new_val))

return NULL;

if (!tstate)

return NULL;

tstate->interp->dlopenflags = new_val;

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(setdlopenflags_doc,

"setdlopenflags(n) -> None\n\

\n\

Set the flags used by the interpreter for dlopen calls, such as when the\n\

interpreter loads extension modules. Among other things, this will enable\n\

a lazy resolving of symbols when importing a module, if called as\n\

sys.setdlopenflags(0). To share symbols across extension modules, call as\n\

sys.setdlopenflags(os.RTLD_GLOBAL). Symbolic names for the flag modules\n\

can be found in the os module (RTLD_xxx constants, e.g. os.RTLD_LAZY).");

static PyObject *

sys_getdlopenflags(PyObject *self, PyObject *args)

{

PyThreadState *tstate = PyThreadState_GET();

if (!tstate)

return NULL;

return PyLong_FromLong(tstate->interp->dlopenflags);

}

PyDoc_STRVAR(getdlopenflags_doc,

"getdlopenflags() -> int\n\

\n\

Return the current value of the flags that are used for dlopen calls.\n\

The flag constants are defined in the os module.");

#endif /* HAVE_DLOPEN */

#ifdef USE_MALLOPT

/* Link with -lmalloc (or -lmpc) on an SGI */

#include <malloc.h>

static PyObject *

sys_mdebug(PyObject *self, PyObject *args)

{

int flag;

if (!PyArg_ParseTuple(args, "i:mdebug", &flag))

return NULL;

mallopt(M_DEBUG, flag);

Py_INCREF(Py_None);

return Py_None;

}

#endif /* USE_MALLOPT */

size_t

_PySys_GetSizeOf(PyObject *o)

{

PyObject *res = NULL;

PyObject *method;

Py_ssize_t size;

/* Make sure the type is initialized. float gets initialized late */

if (PyType_Ready(Py_TYPE(o)) < 0)

return (size_t)-1;

method = _PyObject_LookupSpecial(o, &PyId___sizeof__);

if (method == NULL) {

if (!PyErr_Occurred())

PyErr_Format(PyExc_TypeError,

"Type %.100s doesn't define __sizeof__",

Py_TYPE(o)->tp_name);

}

else {

res = PyObject_CallFunctionObjArgs(method, NULL);

Py_DECREF(method);

}

if (res == NULL)

return (size_t)-1;

size = PyLong_AsSsize_t(res);

Py_DECREF(res);

if (size == -1 && PyErr_Occurred())

return (size_t)-1;

if (size < 0) {

PyErr_SetString(PyExc_ValueError, "__sizeof__() should return >= 0");

return (size_t)-1;

}

/* add gc_head size */

if (PyObject_IS_GC(o))

return ((size_t)size) + sizeof(PyGC_Head);

return (size_t)size;

}

static PyObject *

sys_getsizeof(PyObject *self, PyObject *args, PyObject *kwds)

{

static char *kwlist[] = {"object", "default", 0};

size_t size;

PyObject *o, *dflt = NULL;

if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|O:getsizeof",

kwlist, &o, &dflt))

return NULL;

size = _PySys_GetSizeOf(o);

if (size == (size_t)-1 && PyErr_Occurred()) {

/* Has a default value been given */

if (dflt != NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {

PyErr_Clear();

Py_INCREF(dflt);

return dflt;

}

else

return NULL;

}

return PyLong_FromSize_t(size);

}

PyDoc_STRVAR(getsizeof_doc,

"getsizeof(object, default) -> int\n\

\n\

Return the size of object in bytes.");

static PyObject *

sys_getrefcount(PyObject *self, PyObject *arg)

{

return PyLong_FromSsize_t(arg->ob_refcnt);

}

#ifdef Py_REF_DEBUG

static PyObject *

sys_gettotalrefcount(PyObject *self)

{

return PyLong_FromSsize_t(_Py_GetRefTotal());