#include "Python.h"

#include "code.h"

#include "frameobject.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;

extern const char *PyWin_DLLVersionString;

#ifdef __VMS

#include <unixlib.h>

#ifdef HAVE_LANGINFO_H

#include <locale.h>

#include <langinfo.h>

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);

}

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);

}

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_GetAttrString(outf, "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_GetAttrString(outf, "buffer");

if (buffer) {

result = PyObject_CallMethod(buffer, "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 = PyDict_GetItemString(modules, "builtins");

int err;

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_SetAttrString(builtins, "_", Py_None) != 0)

return NULL;

outf = PySys_GetObject("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

PyErr_Clear();

err = sys_displayhook_unencodable(outf, o);

if (err)

return NULL;

}

else {

return NULL;

}

}

if (PyFile_WriteString("\n", outf) != 0)

return NULL;

if (PyObject_SetAttrString(builtins, "_", o) != 0)

return NULL;

Py_INCREF(Py_None);

return Py_None;

}

PyDoc_STRVAR(displayhook_doc,

);

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,

);

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,

);

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,

);

static PyObject *

sys_getdefaultencoding(PyObject *self)

{

return PyUnicode_FromString(PyUnicode_GetDefaultEncoding());

}

PyDoc_STRVAR(getdefaultencoding_doc,

);

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,

);

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,

same value.");

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(PyThreadState *tstate, PyObject* callback,

PyFrameObject *frame, int what, PyObject *arg)

{

PyObject *args = PyTuple_New(3);

PyObject *whatstr;

PyObject *result;

if (args == NULL)

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 */

PyFrame_FastToLocals(frame);

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)

{

PyThreadState *tstate = frame->f_tstate;

PyObject *result;

if (arg == NULL)

arg = Py_None;

result = call_trampoline(tstate, 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)

{

PyThreadState *tstate = frame->f_tstate;

PyObject *callback;

PyObject *result;

if (what == PyTrace_CALL)

callback = self;

else

callback = frame->f_trace;

if (callback == NULL)

return 0;

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

if (result == NULL) {

PyEval_SetTrace(NULL, NULL);

Py_XDECREF(frame->f_trace);

frame->f_trace = NULL;

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,

);

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,

);

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,

);

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,

);

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,

);

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,

);

#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,

);

static PyObject *

sys_getswitchinterval(PyObject *self, PyObject *args)

{

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

}

PyDoc_STRVAR(getswitchinterval_doc,

);

#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,

);

#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 PyTypeObject Hash_InfoType;

PyDoc_STRVAR(hash_info_doc,

numeric 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"},

{NULL, NULL}

};

static PyStructSequence_Desc hash_info_desc = {

"sys.hash_info",

hash_info_doc,

hash_info_fields,

5,

};

static PyObject *

get_hash_info(void)

{

PyObject *hash_info;

int field = 0;

hash_info = PyStructSequence_New(&Hash_InfoType);

if (hash_info == NULL)

return NULL;

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));

if (PyErr_Occurred()) {

Py_CLEAR(hash_info);

return NULL;

}

return hash_info;

}

PyDoc_STRVAR(setrecursionlimit_doc,

);

static PyObject *

sys_getrecursionlimit(PyObject *self)

{

return PyLong_FromLong(Py_GetRecursionLimit());

}

PyDoc_STRVAR(getrecursionlimit_doc,

);

#ifdef MS_WINDOWS

PyDoc_STRVAR(getwindowsversion_doc,

);

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,

};

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));

return version;

}

#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,

h2py script.");

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,

The flag constants are defined in the ctypes and DLFCN modules.");

#endif /* HAVE_DLOPEN */

#ifdef USE_MALLOPT

#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 */

static PyObject *

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

{

PyObject *res = NULL;

static PyObject *str__sizeof__ = NULL, *gc_head_size = NULL;

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

PyObject *o, *dflt = NULL;

PyObject *method;

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

kwlist, &o, &dflt))

return NULL;

/* Initialize static variable for GC head size */

if (gc_head_size == NULL) {

gc_head_size = PyLong_FromSsize_t(sizeof(PyGC_Head));

if (gc_head_size == NULL)

return NULL;

}

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

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

return NULL;

method = _PyObject_LookupSpecial(o, "__sizeof__",

&str__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);

}

/* Has a default value been given */

if ((res == NULL) && (dflt != NULL) &&

PyErr_ExceptionMatches(PyExc_TypeError))

{

PyErr_Clear();

Py_INCREF(dflt);

return dflt;

}

else if (res == NULL)

return res;

/* add gc_head size */

if (PyObject_IS_GC(o)) {

PyObject *tmp = res;

res = PyNumber_Add(tmp, gc_head_size);

Py_DECREF(tmp);

}

return res;

}

PyDoc_STRVAR(getsizeof_doc,

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());

}

#endif /* Py_REF_DEBUG */

PyDoc_STRVAR(getrefcount_doc,

);

#ifdef COUNT_ALLOCS

static PyObject *

sys_getcounts(PyObject *self)

{

extern PyObject *get_counts(void);

return get_counts();

}

PyDoc_STRVAR(getframe_doc,

);

static PyObject *

sys_getframe(PyObject *self, PyObject *args)

{

PyFrameObject *f = PyThreadState_GET()->frame;

int depth = -1;

if (!PyArg_ParseTuple(args, "|i:_getframe", &depth))

return NULL;

while (depth > 0 && f != NULL) {

f = f->f_back;

--depth;

}

if (f == NULL) {

PyErr_SetString(PyExc_ValueError,

"call stack is not deep enough");

return NULL;

}

Py_INCREF(f);

return (PyObject*)f;

}

PyDoc_STRVAR(current_frames_doc,

);

static PyObject *

sys_current_frames(PyObject *self, PyObject *noargs)

{

return _PyThread_CurrentFrames();

}

PyDoc_STRVAR(call_tracing_doc,

);

static PyObject *

sys_call_tracing(PyObject *self, PyObject *args)

{

PyObject *func, *funcargs;

if (!PyArg_ParseTuple(args, "OO!:call_tracing", &func, &PyTuple_Type, &funcargs))

return NULL;

return _PyEval_CallTracing(func, funcargs);

}

PyDoc_STRVAR(callstats_doc,

);

#ifdef __cplusplus

extern "C" {

#ifdef Py_TRACE_REFS

extern PyObject *_Py_GetObjects(PyObject *, PyObject *);