PC/msvcrtmodule.c - platform/external/python/cpython3 - Gitiles

 /*********************************************************

 	msvcrtmodule.c

 	A Python interface to the Microsoft Visual C Runtime
 	Library, providing access to those non-portable, but
 	still useful routines.

 	Only ever compiled with an MS compiler, so no attempt
 	has been made to avoid MS language extensions, etc...

 	This may only work on NT or 95...

 	Author: Mark Hammond and Guido van Rossum.
 	Maintenance: Guido van Rossum.

 ***********************************************************/

 #include "Python.h"
 #include "malloc.h"
 #include <io.h>
 #include <conio.h>
 #include <sys/locking.h>
 #include <crtdbg.h>
 #include <windows.h>

 // Force the malloc heap to clean itself up, and free unused blocks
 // back to the OS.  (According to the docs, only works on NT.)
 static PyObject *
 msvcrt_heapmin(PyObject *self, PyObject *args)
 {
 	if (!PyArg_ParseTuple(args, ":heapmin"))
 		return NULL;

 	if (_heapmin() != 0)
 		return PyErr_SetFromErrno(PyExc_IOError);

 	Py_INCREF(Py_None);
 	return Py_None;
 }

 // Perform locking operations on a C runtime file descriptor.
 static PyObject *
 msvcrt_locking(PyObject *self, PyObject *args)
 {
 	int fd;
 	int mode;
 	long nbytes;
 	int err;

 	if (!PyArg_ParseTuple(args, "iil:locking", &fd, &mode, &nbytes))
 		return NULL;

 	Py_BEGIN_ALLOW_THREADS
 	err = _locking(fd, mode, nbytes);
 	Py_END_ALLOW_THREADS
 	if (err != 0)
 		return PyErr_SetFromErrno(PyExc_IOError);

 	Py_INCREF(Py_None);
 	return Py_None;
 }

 // Set the file translation mode for a C runtime file descriptor.
 static PyObject *
 msvcrt_setmode(PyObject *self, PyObject *args)
 {
 	int fd;
 	int flags;
 	if (!PyArg_ParseTuple(args,"ii:setmode", &fd, &flags))
 		return NULL;

 	flags = _setmode(fd, flags);
 	if (flags == -1)
 		return PyErr_SetFromErrno(PyExc_IOError);

 	return PyLong_FromLong(flags);
 }

 // Convert an OS file handle to a C runtime file descriptor.
 static PyObject *
 msvcrt_open_osfhandle(PyObject *self, PyObject *args)
 {
 	long handle;
 	int flags;
 	int fd;

 	if (!PyArg_ParseTuple(args, "li:open_osfhandle", &handle, &flags))
 		return NULL;

 	fd = _open_osfhandle(handle, flags);
 	if (fd == -1)
 		return PyErr_SetFromErrno(PyExc_IOError);

 	return PyLong_FromLong(fd);
 }

 // Convert a C runtime file descriptor to an OS file handle.
 static PyObject *
 msvcrt_get_osfhandle(PyObject *self, PyObject *args)
 {
 	int fd;
 	Py_intptr_t handle;

 	if (!PyArg_ParseTuple(args,"i:get_osfhandle", &fd))
 		return NULL;

 	handle = _get_osfhandle(fd);
 	if (handle == -1)
 		return PyErr_SetFromErrno(PyExc_IOError);

 	/* technically 'handle' is not a pointer, but a integer as
 	   large as a pointer, Python's *VoidPtr interface is the
 	   most appropriate here */
 	return PyLong_FromVoidPtr((void*)handle);
 }

 /* Console I/O */

 static PyObject *
 msvcrt_kbhit(PyObject *self, PyObject *args)
 {
 	int ok;

 	if (!PyArg_ParseTuple(args, ":kbhit"))
 		return NULL;

 	ok = _kbhit();
 	return PyLong_FromLong(ok);
 }

 static PyObject *
 msvcrt_getch(PyObject *self, PyObject *args)
 {
 	int ch;
 	char s[1];

 	if (!PyArg_ParseTuple(args, ":getch"))
 		return NULL;

 	Py_BEGIN_ALLOW_THREADS
 	ch = _getch();
 	Py_END_ALLOW_THREADS
 	s[0] = ch;
 	return PyString_FromStringAndSize(s, 1);
 }

 #if _MSC_VER >= 1300
 static PyObject *
 msvcrt_getwch(PyObject *self, PyObject *args)
 {
 	Py_UNICODE ch;
 	Py_UNICODE u[1];

 	if (!PyArg_ParseTuple(args, ":getwch"))
 		return NULL;

 	Py_BEGIN_ALLOW_THREADS
 	ch = _getwch();
 	Py_END_ALLOW_THREADS
 	u[0] = ch;
 	return PyUnicode_FromUnicode(u, 1);
 }
 #endif

 static PyObject *
 msvcrt_getche(PyObject *self, PyObject *args)
 {
 	int ch;
 	char s[1];

 	if (!PyArg_ParseTuple(args, ":getche"))
 		return NULL;

 	Py_BEGIN_ALLOW_THREADS
 	ch = _getche();
 	Py_END_ALLOW_THREADS
 	s[0] = ch;
 	return PyString_FromStringAndSize(s, 1);
 }

 #if _MSC_VER >= 1300
 static PyObject *
 msvcrt_getwche(PyObject *self, PyObject *args)
 {
 	Py_UNICODE ch;
 	Py_UNICODE s[1];

 	if (!PyArg_ParseTuple(args, ":getwche"))
 		return NULL;

 	Py_BEGIN_ALLOW_THREADS
 	ch = _getwche();
 	Py_END_ALLOW_THREADS
 	s[0] = ch;
 	return PyUnicode_FromUnicode(s, 1);
 }
 #endif

 static PyObject *
 msvcrt_putch(PyObject *self, PyObject *args)
 {
 	char ch;

 	if (!PyArg_ParseTuple(args, "c:putch", &ch))
 		return NULL;

 	_putch(ch);
 	Py_INCREF(Py_None);
 	return Py_None;
 }


 #if _MSC_VER >= 1300
 static PyObject *
 msvcrt_putwch(PyObject *self, PyObject *args)
 {
 	Py_UNICODE *ch;
 	int size;

 	if (!PyArg_ParseTuple(args, "u#:putwch", &ch, &size))
 		return NULL;

 	if (size == 0) {
 		PyErr_SetString(PyExc_ValueError,
 			"Expected unicode string of length 1");
 		return NULL;
 	}
 	_putwch(*ch);
 	Py_RETURN_NONE;

 }
 #endif

 static PyObject *
 msvcrt_ungetch(PyObject *self, PyObject *args)
 {
 	char ch;

 	if (!PyArg_ParseTuple(args, "c:ungetch", &ch))
 		return NULL;

 	if (_ungetch(ch) == EOF)
 		return PyErr_SetFromErrno(PyExc_IOError);
 	Py_INCREF(Py_None);
 	return Py_None;
 }

 #if _MSC_VER >= 1300
 static PyObject *
 msvcrt_ungetwch(PyObject *self, PyObject *args)
 {
 	Py_UNICODE ch;

 	if (!PyArg_ParseTuple(args, "u:ungetwch", &ch))
 		return NULL;

 	if (_ungetch(ch) == EOF)
 		return PyErr_SetFromErrno(PyExc_IOError);
 	Py_INCREF(Py_None);
 	return Py_None;
 }
 #endif

 static void
 insertint(PyObject *d, char *name, int value)
 {
 	PyObject *v = PyLong_FromLong((long) value);
 	if (v == NULL) {
 		/* Don't bother reporting this error */
 		PyErr_Clear();
 	}
 	else {
 		PyDict_SetItemString(d, name, v);
 		Py_DECREF(v);
 	}
 }

 #ifdef _DEBUG

 static PyObject*
 msvcrt_setreportfile(PyObject *self, PyObject *args)
 {
 	int type, file;
 	_HFILE res;

 	if (!PyArg_ParseTuple(args, "ii", &type, &file))
 		return NULL;
 	res = _CrtSetReportFile(type, (_HFILE)file);
 	return PyLong_FromLong((long)res);
 	Py_INCREF(Py_None);
 	return Py_None;
 }

 static PyObject*
 msvcrt_setreportmode(PyObject *self, PyObject *args)
 {
 	int type, mode;
 	int res;

 	if (!PyArg_ParseTuple(args, "ii", &type, &mode))
 		return NULL;
 	res = _CrtSetReportMode(type, mode);
 	if (res == -1)
 	    return PyErr_SetFromErrno(PyExc_IOError);
 	return PyLong_FromLong(res);
 }

 static PyObject*
 msvcrt_seterrormode(PyObject *self, PyObject *args)
 {
 	int mode, res;

 	if (!PyArg_ParseTuple(args, "i", &mode))
 		return NULL;
 	res = _set_error_mode(mode);
 	return PyLong_FromLong(res);
 }

 #endif

 static PyObject*
 seterrormode(PyObject *self, PyObject *args)
 {
 	unsigned int mode, res;

 	if (!PyArg_ParseTuple(args, "I", &mode))
 		return NULL;
 	res = SetErrorMode(mode);
 	return PyLong_FromUnsignedLong(res);
 }


 /* List of functions exported by this module */
 static struct PyMethodDef msvcrt_functions[] = {
 	{"heapmin",		msvcrt_heapmin, METH_VARARGS},
 	{"locking",             msvcrt_locking, METH_VARARGS},
 	{"setmode",		msvcrt_setmode, METH_VARARGS},
 	{"open_osfhandle",	msvcrt_open_osfhandle, METH_VARARGS},
 	{"get_osfhandle",	msvcrt_get_osfhandle, METH_VARARGS},
 	{"kbhit",		msvcrt_kbhit, METH_VARARGS},
 	{"getch",		msvcrt_getch, METH_VARARGS},
 	{"getche",		msvcrt_getche, METH_VARARGS},
 	{"putch",		msvcrt_putch, METH_VARARGS},
 	{"ungetch",		msvcrt_ungetch, METH_VARARGS},
 	{"SetErrorMode",	seterrormode, METH_VARARGS},
 #ifdef _DEBUG
 	{"CrtSetReportFile",	msvcrt_setreportfile, METH_VARARGS},
 	{"CrtSetReportMode",	msvcrt_setreportmode, METH_VARARGS},
 	{"set_error_mode",	msvcrt_seterrormode, METH_VARARGS},
 #endif
 #if _MSC_VER >= 1300
 	{"getwch",		msvcrt_getwch, METH_VARARGS},
 	{"getwche",		msvcrt_getwche, METH_VARARGS},
 	{"putwch",		msvcrt_putwch, METH_VARARGS},
 	{"ungetwch",		msvcrt_ungetwch, METH_VARARGS},
 #endif
 	{NULL,			NULL}
 };

 PyMODINIT_FUNC
 initmsvcrt(void)
 {
 	PyObject *d;
 	PyObject *m = Py_InitModule("msvcrt", msvcrt_functions);
 	if (m == NULL)
 		return;
 	d = PyModule_GetDict(m);

 	/* constants for the locking() function's mode argument */
 	insertint(d, "LK_LOCK", _LK_LOCK);
 	insertint(d, "LK_NBLCK", _LK_NBLCK);
 	insertint(d, "LK_NBRLCK", _LK_NBRLCK);
 	insertint(d, "LK_RLCK", _LK_RLCK);
 	insertint(d, "LK_UNLCK", _LK_UNLCK);
 	insertint(d, "SEM_FAILCRITICALERRORS", SEM_FAILCRITICALERRORS);
 	insertint(d, "SEM_NOALIGNMENTFAULTEXCEPT", SEM_NOALIGNMENTFAULTEXCEPT);
 	insertint(d, "SEM_NOGPFAULTERRORBOX", SEM_NOGPFAULTERRORBOX);
 	insertint(d, "SEM_NOOPENFILEERRORBOX", SEM_NOOPENFILEERRORBOX);
 #ifdef _DEBUG
 	insertint(d, "CRT_WARN", _CRT_WARN);
 	insertint(d, "CRT_ERROR", _CRT_ERROR);
 	insertint(d, "CRT_ASSERT", _CRT_ASSERT);
 	insertint(d, "CRTDBG_MODE_DEBUG", _CRTDBG_MODE_DEBUG);
 	insertint(d, "CRTDBG_MODE_FILE", _CRTDBG_MODE_FILE);
 	insertint(d, "CRTDBG_MODE_WNDW", _CRTDBG_MODE_WNDW);
 	insertint(d, "CRTDBG_REPORT_MODE", _CRTDBG_REPORT_MODE);
 	insertint(d, "CRTDBG_FILE_STDERR", (int)_CRTDBG_FILE_STDERR);
 	insertint(d, "CRTDBG_FILE_STDOUT", (int)_CRTDBG_FILE_STDOUT);
 	insertint(d, "CRTDBG_REPORT_FILE", (int)_CRTDBG_REPORT_FILE);
 #endif
 }
	/*********************************************************

	msvcrtmodule.c

	A Python interface to the Microsoft Visual C Runtime
	Library, providing access to those non-portable, but
	still useful routines.

	Only ever compiled with an MS compiler, so no attempt
	has been made to avoid MS language extensions, etc...

	This may only work on NT or 95...

	Author: Mark Hammond and Guido van Rossum.
	Maintenance: Guido van Rossum.

	***********************************************************/

	#include "Python.h"
	#include "malloc.h"
	#include <io.h>
	#include <conio.h>
	#include <sys/locking.h>
	#include <crtdbg.h>
	#include <windows.h>

	// Force the malloc heap to clean itself up, and free unused blocks
	// back to the OS. (According to the docs, only works on NT.)
	static PyObject *
	msvcrt_heapmin(PyObject self, PyObject args)
	{
	if (!PyArg_ParseTuple(args, ":heapmin"))
	return NULL;

	if (_heapmin() != 0)
	return PyErr_SetFromErrno(PyExc_IOError);

	Py_INCREF(Py_None);
	return Py_None;
	}

	// Perform locking operations on a C runtime file descriptor.
	static PyObject *
	msvcrt_locking(PyObject self, PyObject args)
	{
	int fd;
	int mode;
	long nbytes;
	int err;

	if (!PyArg_ParseTuple(args, "iil:locking", &fd, &mode, &nbytes))
	return NULL;

	Py_BEGIN_ALLOW_THREADS
	err = _locking(fd, mode, nbytes);
	Py_END_ALLOW_THREADS
	if (err != 0)
	return PyErr_SetFromErrno(PyExc_IOError);

	Py_INCREF(Py_None);
	return Py_None;
	}

	// Set the file translation mode for a C runtime file descriptor.
	static PyObject *
	msvcrt_setmode(PyObject self, PyObject args)
	{
	int fd;
	int flags;
	if (!PyArg_ParseTuple(args,"ii:setmode", &fd, &flags))
	return NULL;

	flags = _setmode(fd, flags);
	if (flags == -1)
	return PyErr_SetFromErrno(PyExc_IOError);

	return PyLong_FromLong(flags);
	}

	// Convert an OS file handle to a C runtime file descriptor.
	static PyObject *
	msvcrt_open_osfhandle(PyObject self, PyObject args)
	{
	long handle;
	int flags;
	int fd;

	if (!PyArg_ParseTuple(args, "li:open_osfhandle", &handle, &flags))
	return NULL;

	fd = _open_osfhandle(handle, flags);
	if (fd == -1)
	return PyErr_SetFromErrno(PyExc_IOError);

	return PyLong_FromLong(fd);
	}

	// Convert a C runtime file descriptor to an OS file handle.
	static PyObject *
	msvcrt_get_osfhandle(PyObject self, PyObject args)
	{
	int fd;
	Py_intptr_t handle;

	if (!PyArg_ParseTuple(args,"i:get_osfhandle", &fd))
	return NULL;

	handle = _get_osfhandle(fd);
	if (handle == -1)
	return PyErr_SetFromErrno(PyExc_IOError);

	/* technically 'handle' is not a pointer, but a integer as
	large as a pointer, Python's *VoidPtr interface is the
	most appropriate here */
	return PyLong_FromVoidPtr((void*)handle);
	}

	/* Console I/O */

	static PyObject *
	msvcrt_kbhit(PyObject self, PyObject args)
	{
	int ok;

	if (!PyArg_ParseTuple(args, ":kbhit"))
	return NULL;

	ok = _kbhit();
	return PyLong_FromLong(ok);
	}

	static PyObject *
	msvcrt_getch(PyObject self, PyObject args)
	{
	int ch;
	char s[1];

	if (!PyArg_ParseTuple(args, ":getch"))
	return NULL;

	Py_BEGIN_ALLOW_THREADS
	ch = _getch();
	Py_END_ALLOW_THREADS
	s[0] = ch;
	return PyString_FromStringAndSize(s, 1);
	}

	#if _MSC_VER >= 1300
	static PyObject *
	msvcrt_getwch(PyObject self, PyObject args)
	{
	Py_UNICODE ch;
	Py_UNICODE u[1];

	if (!PyArg_ParseTuple(args, ":getwch"))
	return NULL;

	Py_BEGIN_ALLOW_THREADS
	ch = _getwch();
	Py_END_ALLOW_THREADS
	u[0] = ch;
	return PyUnicode_FromUnicode(u, 1);
	}
	#endif

	static PyObject *
	msvcrt_getche(PyObject self, PyObject args)
	{
	int ch;
	char s[1];

	if (!PyArg_ParseTuple(args, ":getche"))
	return NULL;

	Py_BEGIN_ALLOW_THREADS
	ch = _getche();
	Py_END_ALLOW_THREADS
	s[0] = ch;
	return PyString_FromStringAndSize(s, 1);
	}

	#if _MSC_VER >= 1300
	static PyObject *
	msvcrt_getwche(PyObject self, PyObject args)
	{
	Py_UNICODE ch;
	Py_UNICODE s[1];

	if (!PyArg_ParseTuple(args, ":getwche"))
	return NULL;

	Py_BEGIN_ALLOW_THREADS
	ch = _getwche();
	Py_END_ALLOW_THREADS
	s[0] = ch;
	return PyUnicode_FromUnicode(s, 1);
	}
	#endif

	static PyObject *
	msvcrt_putch(PyObject self, PyObject args)
	{
	char ch;

	if (!PyArg_ParseTuple(args, "c:putch", &ch))
	return NULL;

	_putch(ch);
	Py_INCREF(Py_None);
	return Py_None;
	}


	#if _MSC_VER >= 1300
	static PyObject *
	msvcrt_putwch(PyObject self, PyObject args)
	{
	Py_UNICODE *ch;
	int size;

	if (!PyArg_ParseTuple(args, "u#:putwch", &ch, &size))
	return NULL;

	if (size == 0) {
	PyErr_SetString(PyExc_ValueError,
	"Expected unicode string of length 1");
	return NULL;
	}
	_putwch(*ch);
	Py_RETURN_NONE;

	}
	#endif

	static PyObject *
	msvcrt_ungetch(PyObject self, PyObject args)
	{
	char ch;

	if (!PyArg_ParseTuple(args, "c:ungetch", &ch))
	return NULL;

	if (_ungetch(ch) == EOF)
	return PyErr_SetFromErrno(PyExc_IOError);
	Py_INCREF(Py_None);
	return Py_None;
	}

	#if _MSC_VER >= 1300
	static PyObject *
	msvcrt_ungetwch(PyObject self, PyObject args)
	{
	Py_UNICODE ch;

	if (!PyArg_ParseTuple(args, "u:ungetwch", &ch))
	return NULL;

	if (_ungetch(ch) == EOF)
	return PyErr_SetFromErrno(PyExc_IOError);
	Py_INCREF(Py_None);
	return Py_None;
	}
	#endif

	static void
	insertint(PyObject d, char name, int value)
	{
	PyObject *v = PyLong_FromLong((long) value);
	if (v == NULL) {
	/* Don't bother reporting this error */
	PyErr_Clear();
	}
	else {
	PyDict_SetItemString(d, name, v);
	Py_DECREF(v);
	}
	}

	#ifdef _DEBUG

	static PyObject*
	msvcrt_setreportfile(PyObject self, PyObject args)
	{
	int type, file;
	_HFILE res;

	if (!PyArg_ParseTuple(args, "ii", &type, &file))
	return NULL;
	res = _CrtSetReportFile(type, (_HFILE)file);
	return PyLong_FromLong((long)res);
	Py_INCREF(Py_None);
	return Py_None;
	}

	static PyObject*
	msvcrt_setreportmode(PyObject self, PyObject args)
	{
	int type, mode;
	int res;

	if (!PyArg_ParseTuple(args, "ii", &type, &mode))
	return NULL;
	res = _CrtSetReportMode(type, mode);
	if (res == -1)
	return PyErr_SetFromErrno(PyExc_IOError);
	return PyLong_FromLong(res);
	}

	static PyObject*
	msvcrt_seterrormode(PyObject self, PyObject args)
	{
	int mode, res;

	if (!PyArg_ParseTuple(args, "i", &mode))
	return NULL;
	res = _set_error_mode(mode);
	return PyLong_FromLong(res);
	}

	#endif

	static PyObject*
	seterrormode(PyObject self, PyObject args)
	{
	unsigned int mode, res;

	if (!PyArg_ParseTuple(args, "I", &mode))
	return NULL;
	res = SetErrorMode(mode);
	return PyLong_FromUnsignedLong(res);
	}


	/* List of functions exported by this module */
	static struct PyMethodDef msvcrt_functions[] = {
	{"heapmin", msvcrt_heapmin, METH_VARARGS},
	{"locking", msvcrt_locking, METH_VARARGS},
	{"setmode", msvcrt_setmode, METH_VARARGS},
	{"open_osfhandle", msvcrt_open_osfhandle, METH_VARARGS},
	{"get_osfhandle", msvcrt_get_osfhandle, METH_VARARGS},
	{"kbhit", msvcrt_kbhit, METH_VARARGS},
	{"getch", msvcrt_getch, METH_VARARGS},
	{"getche", msvcrt_getche, METH_VARARGS},
	{"putch", msvcrt_putch, METH_VARARGS},
	{"ungetch", msvcrt_ungetch, METH_VARARGS},
	{"SetErrorMode", seterrormode, METH_VARARGS},
	#ifdef _DEBUG
	{"CrtSetReportFile", msvcrt_setreportfile, METH_VARARGS},
	{"CrtSetReportMode", msvcrt_setreportmode, METH_VARARGS},
	{"set_error_mode", msvcrt_seterrormode, METH_VARARGS},
	#endif
	#if _MSC_VER >= 1300
	{"getwch", msvcrt_getwch, METH_VARARGS},
	{"getwche", msvcrt_getwche, METH_VARARGS},
	{"putwch", msvcrt_putwch, METH_VARARGS},
	{"ungetwch", msvcrt_ungetwch, METH_VARARGS},
	#endif
	{NULL, NULL}
	};

	PyMODINIT_FUNC
	initmsvcrt(void)
	{
	PyObject *d;
	PyObject *m = Py_InitModule("msvcrt", msvcrt_functions);
	if (m == NULL)
	return;
	d = PyModule_GetDict(m);

	/* constants for the locking() function's mode argument */
	insertint(d, "LK_LOCK", _LK_LOCK);
	insertint(d, "LK_NBLCK", _LK_NBLCK);
	insertint(d, "LK_NBRLCK", _LK_NBRLCK);
	insertint(d, "LK_RLCK", _LK_RLCK);
	insertint(d, "LK_UNLCK", _LK_UNLCK);
	insertint(d, "SEM_FAILCRITICALERRORS", SEM_FAILCRITICALERRORS);
	insertint(d, "SEM_NOALIGNMENTFAULTEXCEPT", SEM_NOALIGNMENTFAULTEXCEPT);
	insertint(d, "SEM_NOGPFAULTERRORBOX", SEM_NOGPFAULTERRORBOX);
	insertint(d, "SEM_NOOPENFILEERRORBOX", SEM_NOOPENFILEERRORBOX);
	#ifdef _DEBUG
	insertint(d, "CRT_WARN", _CRT_WARN);
	insertint(d, "CRT_ERROR", _CRT_ERROR);
	insertint(d, "CRT_ASSERT", _CRT_ASSERT);
	insertint(d, "CRTDBG_MODE_DEBUG", _CRTDBG_MODE_DEBUG);
	insertint(d, "CRTDBG_MODE_FILE", _CRTDBG_MODE_FILE);
	insertint(d, "CRTDBG_MODE_WNDW", _CRTDBG_MODE_WNDW);
	insertint(d, "CRTDBG_REPORT_MODE", _CRTDBG_REPORT_MODE);
	insertint(d, "CRTDBG_FILE_STDERR", (int)_CRTDBG_FILE_STDERR);
	insertint(d, "CRTDBG_FILE_STDOUT", (int)_CRTDBG_FILE_STDOUT);
	insertint(d, "CRTDBG_REPORT_FILE", (int)_CRTDBG_REPORT_FILE);
	#endif
	}