mirror of
https://github.com/python/cpython
synced 2024-09-06 02:18:37 +00:00
233 lines
4.9 KiB
C
Executable file
233 lines
4.9 KiB
C
Executable file
/*********************************************************
|
|
|
|
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>
|
|
|
|
// 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 PyInt_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 PyInt_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 PyInt_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);
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
|
|
static void
|
|
insertint(PyObject *d, char *name, int value)
|
|
{
|
|
PyObject *v = PyInt_FromLong((long) value);
|
|
if (v == NULL) {
|
|
/* Don't bother reporting this error */
|
|
PyErr_Clear();
|
|
}
|
|
else {
|
|
PyDict_SetItemString(d, name, v);
|
|
Py_DECREF(v);
|
|
}
|
|
}
|
|
|
|
|
|
/* 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},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
PyMODINIT_FUNC
|
|
initmsvcrt(void)
|
|
{
|
|
PyObject *m = Py_InitModule("msvcrt", msvcrt_functions);
|
|
PyObject *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);
|
|
}
|