mirror of
https://github.com/python/cpython.git
synced 2024-11-26 19:34:19 +08:00
407 lines
10 KiB
C
407 lines
10 KiB
C
|
|
/* fcntl module */
|
|
|
|
#include "Python.h"
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_FILE_H
|
|
#include <sys/file.h>
|
|
#endif
|
|
|
|
#include <sys/ioctl.h>
|
|
#include <fcntl.h>
|
|
|
|
|
|
static int
|
|
conv_descriptor(PyObject *object, int *target)
|
|
{
|
|
int fd = PyObject_AsFileDescriptor(object);
|
|
|
|
if (fd < 0)
|
|
return 0;
|
|
*target = fd;
|
|
return 1;
|
|
}
|
|
|
|
|
|
/* fcntl(fd, opt, [arg]) */
|
|
|
|
static PyObject *
|
|
fcntl_fcntl(PyObject *self, PyObject *args)
|
|
{
|
|
int fd;
|
|
int code;
|
|
int arg;
|
|
int ret;
|
|
char *str;
|
|
int len;
|
|
char buf[1024];
|
|
|
|
if (PyArg_ParseTuple(args, "O&is#:fcntl",
|
|
conv_descriptor, &fd, &code, &str, &len)) {
|
|
if (len > sizeof buf) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"fcntl string arg too long");
|
|
return NULL;
|
|
}
|
|
memcpy(buf, str, len);
|
|
Py_BEGIN_ALLOW_THREADS
|
|
ret = fcntl(fd, code, buf);
|
|
Py_END_ALLOW_THREADS
|
|
if (ret < 0) {
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return NULL;
|
|
}
|
|
return PyString_FromStringAndSize(buf, len);
|
|
}
|
|
|
|
PyErr_Clear();
|
|
arg = 0;
|
|
if (!PyArg_ParseTuple(args,
|
|
"O&i|i;fcntl requires a file or file descriptor,"
|
|
" an integer and optionally a third integer or a string",
|
|
conv_descriptor, &fd, &code, &arg)) {
|
|
return NULL;
|
|
}
|
|
Py_BEGIN_ALLOW_THREADS
|
|
ret = fcntl(fd, code, arg);
|
|
Py_END_ALLOW_THREADS
|
|
if (ret < 0) {
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return NULL;
|
|
}
|
|
return PyInt_FromLong((long)ret);
|
|
}
|
|
|
|
static char fcntl_doc [] =
|
|
|
|
"fcntl(fd, opt, [arg])\n\
|
|
\n\
|
|
Perform the requested operation on file descriptor fd. The operation\n\
|
|
is defined by op and is operating system dependent. These constants are\n\
|
|
available from the fcntl module. The argument arg is optional, and\n\
|
|
defaults to 0; it may be an int or a string. If arg is given as a string,\n\
|
|
the return value of fcntl is a string of that length, containing the\n\
|
|
resulting value put in the arg buffer by the operating system.The length\n\
|
|
of the arg string is not allowed to exceed 1024 bytes. If the arg given\n\
|
|
is an integer or if none is specified, the result value is an integer\n\
|
|
corresponding to the return value of the fcntl call in the C code.";
|
|
|
|
|
|
/* ioctl(fd, opt, [arg]) */
|
|
|
|
static PyObject *
|
|
fcntl_ioctl(PyObject *self, PyObject *args)
|
|
{
|
|
int fd;
|
|
int code;
|
|
int arg;
|
|
int ret;
|
|
char *str;
|
|
int len;
|
|
char buf[1024];
|
|
|
|
if (PyArg_ParseTuple(args, "O&is#:ioctl",
|
|
conv_descriptor, &fd, &code, &str, &len)) {
|
|
if (len > sizeof buf) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"ioctl string arg too long");
|
|
return NULL;
|
|
}
|
|
memcpy(buf, str, len);
|
|
Py_BEGIN_ALLOW_THREADS
|
|
ret = ioctl(fd, code, buf);
|
|
Py_END_ALLOW_THREADS
|
|
if (ret < 0) {
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return NULL;
|
|
}
|
|
return PyString_FromStringAndSize(buf, len);
|
|
}
|
|
|
|
PyErr_Clear();
|
|
arg = 0;
|
|
if (!PyArg_ParseTuple(args,
|
|
"O&i|i;ioctl requires a file or file descriptor,"
|
|
" an integer and optionally a third integer or a string",
|
|
conv_descriptor, &fd, &code, &arg)) {
|
|
return NULL;
|
|
}
|
|
Py_BEGIN_ALLOW_THREADS
|
|
ret = ioctl(fd, code, arg);
|
|
Py_END_ALLOW_THREADS
|
|
if (ret < 0) {
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return NULL;
|
|
}
|
|
return PyInt_FromLong((long)ret);
|
|
}
|
|
|
|
static char ioctl_doc [] =
|
|
"ioctl(fd, opt, [arg])\n\
|
|
\n\
|
|
Perform the requested operation on file descriptor fd. The operation\n\
|
|
is defined by op and is operating system dependent. Typically these\n\
|
|
codes can be retrieved from the library module IOCTL. The argument arg\n\
|
|
is optional, and defaults to 0; it may be an int or a string. If arg is\n\
|
|
given as a string, the return value of ioctl is a string of that length,\n\
|
|
containing the resulting value put in the arg buffer by the operating system.\n\
|
|
The length of the arg string is not allowed to exceed 1024 bytes. If the arg\n\
|
|
given is an integer or if none is specified, the result value is an integer\n\
|
|
corresponding to the return value of the ioctl call in the C code.";
|
|
|
|
|
|
/* flock(fd, operation) */
|
|
|
|
static PyObject *
|
|
fcntl_flock(PyObject *self, PyObject *args)
|
|
{
|
|
int fd;
|
|
int code;
|
|
int ret;
|
|
|
|
if (!PyArg_ParseTuple(args, "O&i:flock",
|
|
conv_descriptor, &fd, &code))
|
|
return NULL;
|
|
|
|
#ifdef HAVE_FLOCK
|
|
Py_BEGIN_ALLOW_THREADS
|
|
ret = flock(fd, code);
|
|
Py_END_ALLOW_THREADS
|
|
#else
|
|
|
|
#ifndef LOCK_SH
|
|
#define LOCK_SH 1 /* shared lock */
|
|
#define LOCK_EX 2 /* exclusive lock */
|
|
#define LOCK_NB 4 /* don't block when locking */
|
|
#define LOCK_UN 8 /* unlock */
|
|
#endif
|
|
{
|
|
struct flock l;
|
|
if (code == LOCK_UN)
|
|
l.l_type = F_UNLCK;
|
|
else if (code & LOCK_SH)
|
|
l.l_type = F_RDLCK;
|
|
else if (code & LOCK_EX)
|
|
l.l_type = F_WRLCK;
|
|
else {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"unrecognized flock argument");
|
|
return NULL;
|
|
}
|
|
l.l_whence = l.l_start = l.l_len = 0;
|
|
Py_BEGIN_ALLOW_THREADS
|
|
ret = fcntl(fd, (code & LOCK_NB) ? F_SETLK : F_SETLKW, &l);
|
|
Py_END_ALLOW_THREADS
|
|
}
|
|
#endif /* HAVE_FLOCK */
|
|
if (ret < 0) {
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return NULL;
|
|
}
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static char flock_doc [] =
|
|
"flock(fd, operation)\n\
|
|
\n\
|
|
Perform the lock operation op on file descriptor fd. See the Unix \n\
|
|
manual flock(3) for details. (On some systems, this function is\n\
|
|
emulated using fcntl().)";
|
|
|
|
|
|
/* lockf(fd, operation) */
|
|
static PyObject *
|
|
fcntl_lockf(PyObject *self, PyObject *args)
|
|
{
|
|
int fd, code, ret, whence = 0;
|
|
PyObject *lenobj = NULL, *startobj = NULL;
|
|
|
|
if (!PyArg_ParseTuple(args, "O&i|OOi:lockf",
|
|
conv_descriptor, &fd, &code,
|
|
&lenobj, &startobj, &whence))
|
|
return NULL;
|
|
|
|
#ifndef LOCK_SH
|
|
#define LOCK_SH 1 /* shared lock */
|
|
#define LOCK_EX 2 /* exclusive lock */
|
|
#define LOCK_NB 4 /* don't block when locking */
|
|
#define LOCK_UN 8 /* unlock */
|
|
#endif
|
|
{
|
|
struct flock l;
|
|
if (code == LOCK_UN)
|
|
l.l_type = F_UNLCK;
|
|
else if (code & LOCK_SH)
|
|
l.l_type = F_RDLCK;
|
|
else if (code & LOCK_EX)
|
|
l.l_type = F_WRLCK;
|
|
else {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"unrecognized flock argument");
|
|
return NULL;
|
|
}
|
|
l.l_start = l.l_len = 0;
|
|
if (startobj != NULL) {
|
|
#if !defined(HAVE_LARGEFILE_SUPPORT)
|
|
l.l_start = PyInt_AsLong(startobj);
|
|
#else
|
|
l.l_start = PyLong_Check(startobj) ?
|
|
PyLong_AsLongLong(startobj) :
|
|
PyInt_AsLong(startobj);
|
|
#endif
|
|
if (PyErr_Occurred())
|
|
return NULL;
|
|
}
|
|
if (lenobj != NULL) {
|
|
#if !defined(HAVE_LARGEFILE_SUPPORT)
|
|
l.l_len = PyInt_AsLong(lenobj);
|
|
#else
|
|
l.l_len = PyLong_Check(lenobj) ?
|
|
PyLong_AsLongLong(lenobj) :
|
|
PyInt_AsLong(lenobj);
|
|
#endif
|
|
if (PyErr_Occurred())
|
|
return NULL;
|
|
}
|
|
l.l_whence = whence;
|
|
Py_BEGIN_ALLOW_THREADS
|
|
ret = fcntl(fd, (code & LOCK_NB) ? F_SETLK : F_SETLKW, &l);
|
|
Py_END_ALLOW_THREADS
|
|
}
|
|
if (ret < 0) {
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return NULL;
|
|
}
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static char lockf_doc [] =
|
|
"lockf (fd, operation, length=0, start=0, whence=0)\n\
|
|
\n\
|
|
This is essentially a wrapper around the fcntl() locking calls. fd is the\n\
|
|
file descriptor of the file to lock or unlock, and operation is one of the\n\
|
|
following values:\n\
|
|
\n\
|
|
LOCK_UN - unlock\n\
|
|
LOCK_SH - acquire a shared lock\n\
|
|
LOCK_EX - acquire an exclusive lock\n\
|
|
\n\
|
|
When operation is LOCK_SH or LOCK_EX, it can also be bit-wise OR'd with\n\
|
|
LOCK_NB to avoid blocking on lock acquisition. If LOCK_NB is used and the\n\
|
|
lock cannot be acquired, an IOError will be raised and the exception will\n\
|
|
have an errno attribute set to EACCES or EAGAIN (depending on the operating\n\
|
|
system -- for portability, check for either value).\n\
|
|
\n\
|
|
length is the number of bytes to lock, with the default meaning to lock to\n\
|
|
EOF. start is the byte offset, relative to whence, to that the lock\n\
|
|
starts. whence is as with fileobj.seek(), specifically:\n\
|
|
\n\
|
|
0 - relative to the start of the file (SEEK_SET)\n\
|
|
1 - relative to the current buffer position (SEEK_CUR)\n\
|
|
2 - relative to the end of the file (SEEK_END)";
|
|
|
|
/* List of functions */
|
|
|
|
static PyMethodDef fcntl_methods[] = {
|
|
{"fcntl", fcntl_fcntl, METH_VARARGS, fcntl_doc},
|
|
{"ioctl", fcntl_ioctl, METH_VARARGS, ioctl_doc},
|
|
{"flock", fcntl_flock, METH_VARARGS, flock_doc},
|
|
{"lockf", fcntl_lockf, METH_VARARGS, lockf_doc},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
static char module_doc [] =
|
|
|
|
"This module performs file control and I/O control on file \n\
|
|
descriptors. It is an interface to the fcntl() and ioctl() Unix\n\
|
|
routines. File descriptors can be obtained with the fileno() method of\n\
|
|
a file or socket object.";
|
|
|
|
/* Module initialisation */
|
|
|
|
static int
|
|
ins(PyObject* d, char* symbol, long value)
|
|
{
|
|
PyObject* v = PyInt_FromLong(value);
|
|
if (!v || PyDict_SetItemString(d, symbol, v) < 0)
|
|
return -1;
|
|
|
|
Py_DECREF(v);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
all_ins(PyObject* d)
|
|
{
|
|
if (ins(d, "LOCK_SH", (long)LOCK_SH)) return -1;
|
|
if (ins(d, "LOCK_EX", (long)LOCK_EX)) return -1;
|
|
if (ins(d, "LOCK_NB", (long)LOCK_NB)) return -1;
|
|
if (ins(d, "LOCK_UN", (long)LOCK_UN)) return -1;
|
|
#ifdef F_DUPFD
|
|
if (ins(d, "F_DUPFD", (long)F_DUPFD)) return -1;
|
|
#endif
|
|
#ifdef F_GETFD
|
|
if (ins(d, "F_GETFD", (long)F_GETFD)) return -1;
|
|
#endif
|
|
#ifdef F_SETFD
|
|
if (ins(d, "F_SETFD", (long)F_SETFD)) return -1;
|
|
#endif
|
|
#ifdef F_GETFL
|
|
if (ins(d, "F_GETFL", (long)F_GETFL)) return -1;
|
|
#endif
|
|
#ifdef F_SETFL
|
|
if (ins(d, "F_SETFL", (long)F_SETFL)) return -1;
|
|
#endif
|
|
#ifdef F_GETLK
|
|
if (ins(d, "F_GETLK", (long)F_GETLK)) return -1;
|
|
#endif
|
|
#ifdef F_SETLK
|
|
if (ins(d, "F_SETLK", (long)F_SETLK)) return -1;
|
|
#endif
|
|
#ifdef F_SETLKW
|
|
if (ins(d, "F_SETLKW", (long)F_SETLKW)) return -1;
|
|
#endif
|
|
#ifdef F_GETOWN
|
|
if (ins(d, "F_GETOWN", (long)F_GETOWN)) return -1;
|
|
#endif
|
|
#ifdef F_SETOWN
|
|
if (ins(d, "F_SETOWN", (long)F_SETOWN)) return -1;
|
|
#endif
|
|
#ifdef F_GETSIG
|
|
if (ins(d, "F_GETSIG", (long)F_GETSIG)) return -1;
|
|
#endif
|
|
#ifdef F_SETSIG
|
|
if (ins(d, "F_SETSIG", (long)F_SETSIG)) return -1;
|
|
#endif
|
|
#ifdef F_RDLCK
|
|
if (ins(d, "F_RDLCK", (long)F_RDLCK)) return -1;
|
|
#endif
|
|
#ifdef F_WRLCK
|
|
if (ins(d, "F_WRLCK", (long)F_WRLCK)) return -1;
|
|
#endif
|
|
#ifdef F_UNLCK
|
|
if (ins(d, "F_UNLCK", (long)F_UNLCK)) return -1;
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
DL_EXPORT(void)
|
|
initfcntl(void)
|
|
{
|
|
PyObject *m, *d;
|
|
|
|
/* Create the module and add the functions and documentation */
|
|
m = Py_InitModule3("fcntl", fcntl_methods, module_doc);
|
|
|
|
/* Add some symbolic constants to the module */
|
|
d = PyModule_GetDict(m);
|
|
all_ins(d);
|
|
}
|