|  | 
 | /* fcntl module */ | 
 |  | 
 | #include "Python.h" | 
 |  | 
 | #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; | 
 |  | 
 | #if defined(PYOS_OS2) && defined(PYCC_GCC) | 
 | 	PyErr_SetString(PyExc_NotImplementedError, | 
 | 			"lockf not supported on OS/2 (EMX)"); | 
 | 	return NULL; | 
 | #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  /* LOCK_SH */ | 
 | 	{ | 
 | 		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; | 
 | #endif  /* defined(PYOS_OS2) && defined(PYCC_GCC) */ | 
 | } | 
 |  | 
 | 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; | 
 | /* GNU extensions, as of glibc 2.2.4 */ | 
 | #ifdef LOCK_MAND | 
 |         if (ins(d, "LOCK_MAND", (long)LOCK_MAND)) return -1; | 
 | #endif | 
 | #ifdef LOCK_READ | 
 |         if (ins(d, "LOCK_READ", (long)LOCK_READ)) return -1; | 
 | #endif | 
 | #ifdef LOCK_WRITE | 
 |         if (ins(d, "LOCK_WRITE", (long)LOCK_WRITE)) return -1; | 
 | #endif | 
 | #ifdef LOCK_RW | 
 |         if (ins(d, "LOCK_RW", (long)LOCK_RW)) return -1; | 
 | #endif | 
 |  | 
 | #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 | 
 | /* LFS constants */ | 
 | #ifdef F_GETLK64 | 
 |         if (ins(d, "F_GETLK64", (long)F_GETLK64)) return -1; | 
 | #endif | 
 | #ifdef F_SETLK64 | 
 |         if (ins(d, "F_SETLK64", (long)F_SETLK64)) return -1; | 
 | #endif | 
 | #ifdef F_SETLKW64 | 
 |         if (ins(d, "F_SETLKW64", (long)F_SETLKW64)) return -1; | 
 | #endif | 
 | /* GNU extensions, as of glibc 2.2.4. */ | 
 | #ifdef F_SETLEASE | 
 |         if (ins(d, "F_SETLEASE", (long)F_SETLEASE)) return -1; | 
 | #endif | 
 | #ifdef F_GETLEASE | 
 |         if (ins(d, "F_GETLEASE", (long)F_GETLEASE)) return -1; | 
 | #endif | 
 | #ifdef F_NOTIFY | 
 |         if (ins(d, "F_NOTIFY", (long)F_NOTIFY)) return -1; | 
 | #endif | 
 | /* Old BSD flock(). */ | 
 | #ifdef F_EXLCK | 
 |         if (ins(d, "F_EXLCK", (long)F_EXLCK)) return -1; | 
 | #endif | 
 | #ifdef F_SHLCK | 
 |         if (ins(d, "F_SHLCK", (long)F_SHLCK)) return -1; | 
 | #endif | 
 |  | 
 | /* For F_{GET|SET}FL */ | 
 | #ifdef FD_CLOEXEC | 
 |         if (ins(d, "FD_CLOEXEC", (long)FD_CLOEXEC)) return -1; | 
 | #endif | 
 |  | 
 | /* For F_NOTIFY */ | 
 | #ifdef DN_ACCESS | 
 |         if (ins(d, "DN_ACCESS", (long)DN_ACCESS)) return -1; | 
 | #endif | 
 | #ifdef DN_MODIFY | 
 |         if (ins(d, "DN_MODIFY", (long)DN_MODIFY)) return -1; | 
 | #endif | 
 | #ifdef DN_CREATE | 
 |         if (ins(d, "DN_CREATE", (long)DN_CREATE)) return -1; | 
 | #endif | 
 | #ifdef DN_DELETE | 
 |         if (ins(d, "DN_DELETE", (long)DN_DELETE)) return -1; | 
 | #endif | 
 | #ifdef DN_RENAME | 
 |         if (ins(d, "DN_RENAME", (long)DN_RENAME)) return -1; | 
 | #endif | 
 | #ifdef DN_ATTRIB | 
 |         if (ins(d, "DN_ATTRIB", (long)DN_ATTRIB)) return -1; | 
 | #endif | 
 | #ifdef DN_MULTISHOT | 
 |         if (ins(d, "DN_MULTISHOT", (long)DN_MULTISHOT)) 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); | 
 | } |