Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 1 | |
| 2 | /* fcntl module */ |
| 3 | |
Thomas Wouters | 26cc63f | 2006-03-02 00:21:10 +0000 | [diff] [blame] | 4 | #define PY_SSIZE_T_CLEAN |
| 5 | |
Roger E. Masse | 919213a | 1996-12-17 17:42:22 +0000 | [diff] [blame] | 6 | #include "Python.h" |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 7 | |
Guido van Rossum | a376cc5 | 1996-12-05 23:43:35 +0000 | [diff] [blame] | 8 | #ifdef HAVE_SYS_FILE_H |
| 9 | #include <sys/file.h> |
| 10 | #endif |
| 11 | |
Guido van Rossum | 3d65fa3 | 1996-12-09 18:49:14 +0000 | [diff] [blame] | 12 | #include <sys/ioctl.h> |
Guido van Rossum | 3c0b79c | 1996-06-11 15:11:34 +0000 | [diff] [blame] | 13 | #include <fcntl.h> |
Martin v. Löwis | 14e73b1 | 2003-01-01 09:51:12 +0000 | [diff] [blame] | 14 | #ifdef HAVE_STROPTS_H |
| 15 | #include <stropts.h> |
| 16 | #endif |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 17 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 18 | /*[clinic input] |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 19 | module fcntl |
| 20 | [clinic start generated code]*/ |
Serhiy Storchaka | 1009bf1 | 2015-04-03 23:53:51 +0300 | [diff] [blame^] | 21 | /*[clinic end generated code: output=da39a3ee5e6b4b0d input=124b58387c158179]*/ |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 22 | |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 23 | static int |
| 24 | conv_descriptor(PyObject *object, int *target) |
| 25 | { |
| 26 | int fd = PyObject_AsFileDescriptor(object); |
| 27 | |
| 28 | if (fd < 0) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 29 | return 0; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 30 | *target = fd; |
| 31 | return 1; |
| 32 | } |
| 33 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 34 | /* Must come after conv_descriptor definition. */ |
| 35 | #include "clinic/fcntlmodule.c.h" |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 36 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 37 | /*[clinic input] |
| 38 | fcntl.fcntl |
| 39 | |
| 40 | fd: object(type='int', converter='conv_descriptor') |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 41 | cmd as code: int |
| 42 | arg: object(c_default='NULL') = 0 |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 43 | / |
| 44 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 45 | Perform the operation `cmd` on file descriptor fd. |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 46 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 47 | The values used for `cmd` are operating system dependent, and are available |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 48 | as constants in the fcntl module, using the same names as used in |
| 49 | the relevant C header files. The argument arg is optional, and |
| 50 | defaults to 0; it may be an int or a string. If arg is given as a string, |
| 51 | the return value of fcntl is a string of that length, containing the |
| 52 | resulting value put in the arg buffer by the operating system. The length |
| 53 | of the arg string is not allowed to exceed 1024 bytes. If the arg given |
| 54 | is an integer or if none is specified, the result value is an integer |
| 55 | corresponding to the return value of the fcntl call in the C code. |
| 56 | [clinic start generated code]*/ |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 57 | |
Roger E. Masse | 919213a | 1996-12-17 17:42:22 +0000 | [diff] [blame] | 58 | static PyObject * |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 59 | fcntl_fcntl_impl(PyModuleDef *module, int fd, int code, PyObject *arg) |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 60 | /*[clinic end generated code: output=afc5bfa74a03ef0d input=8cefbe59b29efbe2]*/ |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 61 | { |
Serhiy Storchaka | 5a8dacf | 2014-11-10 11:25:50 +0200 | [diff] [blame] | 62 | unsigned int int_arg = 0; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 63 | int ret; |
| 64 | char *str; |
| 65 | Py_ssize_t len; |
| 66 | char buf[1024]; |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 67 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 68 | if (arg != NULL) { |
| 69 | int parse_result; |
| 70 | |
| 71 | if (PyArg_Parse(arg, "s#", &str, &len)) { |
| 72 | if ((size_t)len > sizeof buf) { |
| 73 | PyErr_SetString(PyExc_ValueError, |
| 74 | "fcntl string arg too long"); |
| 75 | return NULL; |
| 76 | } |
| 77 | memcpy(buf, str, len); |
| 78 | Py_BEGIN_ALLOW_THREADS |
| 79 | ret = fcntl(fd, code, buf); |
| 80 | Py_END_ALLOW_THREADS |
| 81 | if (ret < 0) { |
| 82 | PyErr_SetFromErrno(PyExc_IOError); |
| 83 | return NULL; |
| 84 | } |
| 85 | return PyBytes_FromStringAndSize(buf, len); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 86 | } |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 87 | |
| 88 | PyErr_Clear(); |
| 89 | parse_result = PyArg_Parse(arg, |
Serhiy Storchaka | 5a8dacf | 2014-11-10 11:25:50 +0200 | [diff] [blame] | 90 | "I;fcntl requires a file or file descriptor," |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 91 | " an integer and optionally a third integer or a string", |
| 92 | &int_arg); |
| 93 | if (!parse_result) { |
| 94 | return NULL; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 95 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 96 | } |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 97 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 98 | Py_BEGIN_ALLOW_THREADS |
Serhiy Storchaka | 5a8dacf | 2014-11-10 11:25:50 +0200 | [diff] [blame] | 99 | ret = fcntl(fd, code, (int)int_arg); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 100 | Py_END_ALLOW_THREADS |
| 101 | if (ret < 0) { |
| 102 | PyErr_SetFromErrno(PyExc_IOError); |
| 103 | return NULL; |
| 104 | } |
| 105 | return PyLong_FromLong((long)ret); |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 106 | } |
| 107 | |
Guido van Rossum | 185ead6 | 1998-11-23 15:32:55 +0000 | [diff] [blame] | 108 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 109 | /*[clinic input] |
| 110 | fcntl.ioctl |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 111 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 112 | fd: object(type='int', converter='conv_descriptor') |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 113 | request as code: unsigned_int(bitwise=True) |
| 114 | arg as ob_arg: object(c_default='NULL') = 0 |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 115 | mutate_flag as mutate_arg: bool = True |
| 116 | / |
| 117 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 118 | Perform the operation `request` on file descriptor `fd`. |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 119 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 120 | The values used for `request` are operating system dependent, and are available |
| 121 | as constants in the fcntl or termios library modules, using the same names as |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 122 | used in the relevant C header files. |
| 123 | |
| 124 | The argument `arg` is optional, and defaults to 0; it may be an int or a |
| 125 | buffer containing character data (most likely a string or an array). |
| 126 | |
| 127 | If the argument is a mutable buffer (such as an array) and if the |
| 128 | mutate_flag argument (which is only allowed in this case) is true then the |
| 129 | buffer is (in effect) passed to the operating system and changes made by |
| 130 | the OS will be reflected in the contents of the buffer after the call has |
| 131 | returned. The return value is the integer returned by the ioctl system |
| 132 | call. |
| 133 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 134 | If the argument is a mutable buffer and the mutable_flag argument is false, |
| 135 | the behavior is as if a string had been passed. |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 136 | |
| 137 | If the argument is an immutable buffer (most likely a string) then a copy |
| 138 | of the buffer is passed to the operating system and the return value is a |
| 139 | string of the same length containing whatever the operating system put in |
| 140 | the buffer. The length of the arg buffer in this case is not allowed to |
| 141 | exceed 1024 bytes. |
| 142 | |
| 143 | If the arg given is an integer or if none is specified, the result value is |
| 144 | an integer corresponding to the return value of the ioctl call in the C |
| 145 | code. |
| 146 | [clinic start generated code]*/ |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 147 | |
Roger E. Masse | 919213a | 1996-12-17 17:42:22 +0000 | [diff] [blame] | 148 | static PyObject * |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 149 | fcntl_ioctl_impl(PyModuleDef *module, int fd, unsigned int code, PyObject *ob_arg, int mutate_arg) |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 150 | /*[clinic end generated code: output=ad47738c118622bf input=ede70c433cccbbb2]*/ |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 151 | { |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 152 | #define IOCTL_BUFSZ 1024 |
Serhiy Storchaka | 483405b | 2015-02-17 10:14:30 +0200 | [diff] [blame] | 153 | /* We use the unsigned non-checked 'I' format for the 'code' parameter |
| 154 | because the system expects it to be a 32bit bit field value |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 155 | regardless of it being passed as an int or unsigned long on |
| 156 | various platforms. See the termios.TIOCSWINSZ constant across |
R David Murray | d5a2f0b | 2013-11-07 10:51:07 -0500 | [diff] [blame] | 157 | platforms for an example of this. |
Christian Heimes | e25f35e | 2008-03-20 10:49:03 +0000 | [diff] [blame] | 158 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 159 | If any of the 64bit platforms ever decide to use more than 32bits |
| 160 | in their unsigned long ioctl codes this will break and need |
| 161 | special casing based on the platform being built on. |
| 162 | */ |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 163 | int arg = 0; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 164 | int ret; |
| 165 | Py_buffer pstr; |
| 166 | char *str; |
| 167 | Py_ssize_t len; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 168 | char buf[IOCTL_BUFSZ+1]; /* argument plus NUL byte */ |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 169 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 170 | if (ob_arg != NULL) { |
| 171 | if (PyArg_Parse(ob_arg, "w*:ioctl", &pstr)) { |
| 172 | char *arg; |
| 173 | str = pstr.buf; |
| 174 | len = pstr.len; |
Michael W. Hudson | f008998 | 2003-03-03 12:29:42 +0000 | [diff] [blame] | 175 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 176 | if (mutate_arg) { |
| 177 | if (len <= IOCTL_BUFSZ) { |
| 178 | memcpy(buf, str, len); |
| 179 | buf[len] = '\0'; |
| 180 | arg = buf; |
| 181 | } |
| 182 | else { |
| 183 | arg = str; |
| 184 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 185 | } |
| 186 | else { |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 187 | if (len > IOCTL_BUFSZ) { |
| 188 | PyBuffer_Release(&pstr); |
| 189 | PyErr_SetString(PyExc_ValueError, |
| 190 | "ioctl string arg too long"); |
| 191 | return NULL; |
| 192 | } |
| 193 | else { |
| 194 | memcpy(buf, str, len); |
| 195 | buf[len] = '\0'; |
| 196 | arg = buf; |
| 197 | } |
| 198 | } |
| 199 | if (buf == arg) { |
| 200 | Py_BEGIN_ALLOW_THREADS /* think array.resize() */ |
| 201 | ret = ioctl(fd, code, arg); |
| 202 | Py_END_ALLOW_THREADS |
| 203 | } |
| 204 | else { |
| 205 | ret = ioctl(fd, code, arg); |
| 206 | } |
| 207 | if (mutate_arg && (len <= IOCTL_BUFSZ)) { |
| 208 | memcpy(str, buf, len); |
| 209 | } |
| 210 | PyBuffer_Release(&pstr); /* No further access to str below this point */ |
| 211 | if (ret < 0) { |
| 212 | PyErr_SetFromErrno(PyExc_IOError); |
| 213 | return NULL; |
| 214 | } |
| 215 | if (mutate_arg) { |
| 216 | return PyLong_FromLong(ret); |
| 217 | } |
| 218 | else { |
| 219 | return PyBytes_FromStringAndSize(buf, len); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 220 | } |
| 221 | } |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 222 | |
| 223 | PyErr_Clear(); |
| 224 | if (PyArg_Parse(ob_arg, "s*:ioctl", &pstr)) { |
| 225 | str = pstr.buf; |
| 226 | len = pstr.len; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 227 | if (len > IOCTL_BUFSZ) { |
| 228 | PyBuffer_Release(&pstr); |
| 229 | PyErr_SetString(PyExc_ValueError, |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 230 | "ioctl string arg too long"); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 231 | return NULL; |
| 232 | } |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 233 | memcpy(buf, str, len); |
| 234 | buf[len] = '\0'; |
| 235 | Py_BEGIN_ALLOW_THREADS |
| 236 | ret = ioctl(fd, code, buf); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 237 | Py_END_ALLOW_THREADS |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 238 | if (ret < 0) { |
| 239 | PyBuffer_Release(&pstr); |
| 240 | PyErr_SetFromErrno(PyExc_IOError); |
| 241 | return NULL; |
| 242 | } |
| 243 | PyBuffer_Release(&pstr); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 244 | return PyBytes_FromStringAndSize(buf, len); |
| 245 | } |
Michael W. Hudson | f008998 | 2003-03-03 12:29:42 +0000 | [diff] [blame] | 246 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 247 | PyErr_Clear(); |
| 248 | if (!PyArg_Parse(ob_arg, |
| 249 | "i;ioctl requires a file or file descriptor," |
| 250 | " an integer and optionally an integer or buffer argument", |
| 251 | &arg)) { |
| 252 | return NULL; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 253 | } |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 254 | // Fall-through to outside the 'if' statement. |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 255 | } |
| 256 | Py_BEGIN_ALLOW_THREADS |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 257 | ret = ioctl(fd, code, arg); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 258 | Py_END_ALLOW_THREADS |
| 259 | if (ret < 0) { |
| 260 | PyErr_SetFromErrno(PyExc_IOError); |
| 261 | return NULL; |
| 262 | } |
| 263 | return PyLong_FromLong((long)ret); |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 264 | #undef IOCTL_BUFSZ |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 265 | } |
| 266 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 267 | /*[clinic input] |
| 268 | fcntl.flock |
Guido van Rossum | 185ead6 | 1998-11-23 15:32:55 +0000 | [diff] [blame] | 269 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 270 | fd: object(type='int', converter='conv_descriptor') |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 271 | operation as code: int |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 272 | / |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 273 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 274 | Perform the lock operation `operation` on file descriptor `fd`. |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 275 | |
| 276 | See the Unix manual page for flock(2) for details (On some systems, this |
| 277 | function is emulated using fcntl()). |
| 278 | [clinic start generated code]*/ |
Guido van Rossum | 3539b1e | 1996-05-23 22:56:38 +0000 | [diff] [blame] | 279 | |
Roger E. Masse | 919213a | 1996-12-17 17:42:22 +0000 | [diff] [blame] | 280 | static PyObject * |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 281 | fcntl_flock_impl(PyModuleDef *module, int fd, int code) |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 282 | /*[clinic end generated code: output=c9035133a7dbfc96 input=b70a0a41ca22a8a0]*/ |
Guido van Rossum | 3539b1e | 1996-05-23 22:56:38 +0000 | [diff] [blame] | 283 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 284 | int ret; |
Guido van Rossum | 3539b1e | 1996-05-23 22:56:38 +0000 | [diff] [blame] | 285 | |
Guido van Rossum | 3c0b79c | 1996-06-11 15:11:34 +0000 | [diff] [blame] | 286 | #ifdef HAVE_FLOCK |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 287 | Py_BEGIN_ALLOW_THREADS |
| 288 | ret = flock(fd, code); |
| 289 | Py_END_ALLOW_THREADS |
Guido van Rossum | 3c0b79c | 1996-06-11 15:11:34 +0000 | [diff] [blame] | 290 | #else |
| 291 | |
| 292 | #ifndef LOCK_SH |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 293 | #define LOCK_SH 1 /* shared lock */ |
| 294 | #define LOCK_EX 2 /* exclusive lock */ |
| 295 | #define LOCK_NB 4 /* don't block when locking */ |
| 296 | #define LOCK_UN 8 /* unlock */ |
Guido van Rossum | 3c0b79c | 1996-06-11 15:11:34 +0000 | [diff] [blame] | 297 | #endif |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 298 | { |
| 299 | struct flock l; |
| 300 | if (code == LOCK_UN) |
| 301 | l.l_type = F_UNLCK; |
| 302 | else if (code & LOCK_SH) |
| 303 | l.l_type = F_RDLCK; |
| 304 | else if (code & LOCK_EX) |
| 305 | l.l_type = F_WRLCK; |
| 306 | else { |
| 307 | PyErr_SetString(PyExc_ValueError, |
| 308 | "unrecognized flock argument"); |
| 309 | return NULL; |
| 310 | } |
| 311 | l.l_whence = l.l_start = l.l_len = 0; |
| 312 | Py_BEGIN_ALLOW_THREADS |
| 313 | ret = fcntl(fd, (code & LOCK_NB) ? F_SETLK : F_SETLKW, &l); |
| 314 | Py_END_ALLOW_THREADS |
| 315 | } |
Guido van Rossum | 3c0b79c | 1996-06-11 15:11:34 +0000 | [diff] [blame] | 316 | #endif /* HAVE_FLOCK */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 317 | if (ret < 0) { |
| 318 | PyErr_SetFromErrno(PyExc_IOError); |
| 319 | return NULL; |
| 320 | } |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 321 | Py_RETURN_NONE; |
Guido van Rossum | 3539b1e | 1996-05-23 22:56:38 +0000 | [diff] [blame] | 322 | } |
| 323 | |
Guido van Rossum | 185ead6 | 1998-11-23 15:32:55 +0000 | [diff] [blame] | 324 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 325 | /*[clinic input] |
| 326 | fcntl.lockf |
Guido van Rossum | 185ead6 | 1998-11-23 15:32:55 +0000 | [diff] [blame] | 327 | |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 328 | fd: object(type='int', converter='conv_descriptor') |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 329 | cmd as code: int |
| 330 | len as lenobj: object(c_default='NULL') = 0 |
| 331 | start as startobj: object(c_default='NULL') = 0 |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 332 | whence: int = 0 |
| 333 | / |
| 334 | |
| 335 | A wrapper around the fcntl() locking calls. |
| 336 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 337 | `fd` is the file descriptor of the file to lock or unlock, and operation is one |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 338 | of the following values: |
| 339 | |
| 340 | LOCK_UN - unlock |
| 341 | LOCK_SH - acquire a shared lock |
| 342 | LOCK_EX - acquire an exclusive lock |
| 343 | |
| 344 | When operation is LOCK_SH or LOCK_EX, it can also be bitwise ORed with |
| 345 | LOCK_NB to avoid blocking on lock acquisition. If LOCK_NB is used and the |
| 346 | lock cannot be acquired, an IOError will be raised and the exception will |
| 347 | have an errno attribute set to EACCES or EAGAIN (depending on the operating |
| 348 | system -- for portability, check for either value). |
| 349 | |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 350 | `len` is the number of bytes to lock, with the default meaning to lock to |
| 351 | EOF. `start` is the byte offset, relative to `whence`, to that the lock |
| 352 | starts. `whence` is as with fileobj.seek(), specifically: |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 353 | |
| 354 | 0 - relative to the start of the file (SEEK_SET) |
| 355 | 1 - relative to the current buffer position (SEEK_CUR) |
| 356 | 2 - relative to the end of the file (SEEK_END) |
| 357 | [clinic start generated code]*/ |
| 358 | |
Roger E. Masse | 919213a | 1996-12-17 17:42:22 +0000 | [diff] [blame] | 359 | static PyObject * |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 360 | fcntl_lockf_impl(PyModuleDef *module, int fd, int code, PyObject *lenobj, PyObject *startobj, int whence) |
Serhiy Storchaka | 17d3a58 | 2015-03-20 20:04:21 +0200 | [diff] [blame] | 361 | /*[clinic end generated code: output=5536df2892bf3ce9 input=9c594391de821f24]*/ |
Guido van Rossum | c864364 | 1996-09-11 23:17:20 +0000 | [diff] [blame] | 362 | { |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 363 | int ret; |
Guido van Rossum | c864364 | 1996-09-11 23:17:20 +0000 | [diff] [blame] | 364 | |
Guido van Rossum | c864364 | 1996-09-11 23:17:20 +0000 | [diff] [blame] | 365 | #ifndef LOCK_SH |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 366 | #define LOCK_SH 1 /* shared lock */ |
| 367 | #define LOCK_EX 2 /* exclusive lock */ |
| 368 | #define LOCK_NB 4 /* don't block when locking */ |
| 369 | #define LOCK_UN 8 /* unlock */ |
Andrew MacIntyre | 7bf6833 | 2002-03-03 02:59:16 +0000 | [diff] [blame] | 370 | #endif /* LOCK_SH */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 371 | { |
| 372 | struct flock l; |
| 373 | if (code == LOCK_UN) |
| 374 | l.l_type = F_UNLCK; |
| 375 | else if (code & LOCK_SH) |
| 376 | l.l_type = F_RDLCK; |
| 377 | else if (code & LOCK_EX) |
| 378 | l.l_type = F_WRLCK; |
| 379 | else { |
| 380 | PyErr_SetString(PyExc_ValueError, |
| 381 | "unrecognized lockf argument"); |
| 382 | return NULL; |
| 383 | } |
| 384 | l.l_start = l.l_len = 0; |
| 385 | if (startobj != NULL) { |
Guido van Rossum | 056bad9 | 1999-01-06 18:44:23 +0000 | [diff] [blame] | 386 | #if !defined(HAVE_LARGEFILE_SUPPORT) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 387 | l.l_start = PyLong_AsLong(startobj); |
Guido van Rossum | 056bad9 | 1999-01-06 18:44:23 +0000 | [diff] [blame] | 388 | #else |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 389 | l.l_start = PyLong_Check(startobj) ? |
| 390 | PyLong_AsLongLong(startobj) : |
| 391 | PyLong_AsLong(startobj); |
Guido van Rossum | 056bad9 | 1999-01-06 18:44:23 +0000 | [diff] [blame] | 392 | #endif |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 393 | if (PyErr_Occurred()) |
| 394 | return NULL; |
| 395 | } |
| 396 | if (lenobj != NULL) { |
Guido van Rossum | 056bad9 | 1999-01-06 18:44:23 +0000 | [diff] [blame] | 397 | #if !defined(HAVE_LARGEFILE_SUPPORT) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 398 | l.l_len = PyLong_AsLong(lenobj); |
Guido van Rossum | 056bad9 | 1999-01-06 18:44:23 +0000 | [diff] [blame] | 399 | #else |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 400 | l.l_len = PyLong_Check(lenobj) ? |
| 401 | PyLong_AsLongLong(lenobj) : |
| 402 | PyLong_AsLong(lenobj); |
Guido van Rossum | 056bad9 | 1999-01-06 18:44:23 +0000 | [diff] [blame] | 403 | #endif |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 404 | if (PyErr_Occurred()) |
| 405 | return NULL; |
| 406 | } |
| 407 | l.l_whence = whence; |
| 408 | Py_BEGIN_ALLOW_THREADS |
| 409 | ret = fcntl(fd, (code & LOCK_NB) ? F_SETLK : F_SETLKW, &l); |
| 410 | Py_END_ALLOW_THREADS |
| 411 | } |
| 412 | if (ret < 0) { |
| 413 | PyErr_SetFromErrno(PyExc_IOError); |
| 414 | return NULL; |
| 415 | } |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 416 | Py_RETURN_NONE; |
Guido van Rossum | c864364 | 1996-09-11 23:17:20 +0000 | [diff] [blame] | 417 | } |
Guido van Rossum | 3539b1e | 1996-05-23 22:56:38 +0000 | [diff] [blame] | 418 | |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 419 | /* List of functions */ |
| 420 | |
Roger E. Masse | 919213a | 1996-12-17 17:42:22 +0000 | [diff] [blame] | 421 | static PyMethodDef fcntl_methods[] = { |
Brett Cannon | b7299dd | 2014-11-09 20:22:01 -0500 | [diff] [blame] | 422 | FCNTL_FCNTL_METHODDEF |
| 423 | FCNTL_IOCTL_METHODDEF |
| 424 | FCNTL_FLOCK_METHODDEF |
| 425 | FCNTL_LOCKF_METHODDEF |
| 426 | {NULL, NULL} /* sentinel */ |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 427 | }; |
| 428 | |
| 429 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 430 | PyDoc_STRVAR(module_doc, |
Guido van Rossum | 185ead6 | 1998-11-23 15:32:55 +0000 | [diff] [blame] | 431 | "This module performs file control and I/O control on file \n\ |
| 432 | descriptors. It is an interface to the fcntl() and ioctl() Unix\n\ |
| 433 | routines. File descriptors can be obtained with the fileno() method of\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 434 | a file or socket object."); |
Guido van Rossum | 185ead6 | 1998-11-23 15:32:55 +0000 | [diff] [blame] | 435 | |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 436 | /* Module initialisation */ |
| 437 | |
Martin v. Löwis | 14e73b1 | 2003-01-01 09:51:12 +0000 | [diff] [blame] | 438 | |
Guido van Rossum | f4e32c7 | 1997-07-31 19:39:54 +0000 | [diff] [blame] | 439 | static int |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 440 | all_ins(PyObject* m) |
Guido van Rossum | f4e32c7 | 1997-07-31 19:39:54 +0000 | [diff] [blame] | 441 | { |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 442 | if (PyModule_AddIntMacro(m, LOCK_SH)) return -1; |
| 443 | if (PyModule_AddIntMacro(m, LOCK_EX)) return -1; |
| 444 | if (PyModule_AddIntMacro(m, LOCK_NB)) return -1; |
| 445 | if (PyModule_AddIntMacro(m, LOCK_UN)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 446 | /* GNU extensions, as of glibc 2.2.4 */ |
| 447 | #ifdef LOCK_MAND |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 448 | if (PyModule_AddIntMacro(m, LOCK_MAND)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 449 | #endif |
| 450 | #ifdef LOCK_READ |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 451 | if (PyModule_AddIntMacro(m, LOCK_READ)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 452 | #endif |
| 453 | #ifdef LOCK_WRITE |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 454 | if (PyModule_AddIntMacro(m, LOCK_WRITE)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 455 | #endif |
| 456 | #ifdef LOCK_RW |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 457 | if (PyModule_AddIntMacro(m, LOCK_RW)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 458 | #endif |
| 459 | |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 460 | #ifdef F_DUPFD |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 461 | if (PyModule_AddIntMacro(m, F_DUPFD)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 462 | #endif |
Victor Stinner | 2716d53 | 2013-01-08 00:52:40 +0100 | [diff] [blame] | 463 | #ifdef F_DUPFD_CLOEXEC |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 464 | if (PyModule_AddIntMacro(m, F_DUPFD_CLOEXEC)) return -1; |
Victor Stinner | 2716d53 | 2013-01-08 00:52:40 +0100 | [diff] [blame] | 465 | #endif |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 466 | #ifdef F_GETFD |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 467 | if (PyModule_AddIntMacro(m, F_GETFD)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 468 | #endif |
| 469 | #ifdef F_SETFD |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 470 | if (PyModule_AddIntMacro(m, F_SETFD)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 471 | #endif |
| 472 | #ifdef F_GETFL |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 473 | if (PyModule_AddIntMacro(m, F_GETFL)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 474 | #endif |
| 475 | #ifdef F_SETFL |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 476 | if (PyModule_AddIntMacro(m, F_SETFL)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 477 | #endif |
| 478 | #ifdef F_GETLK |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 479 | if (PyModule_AddIntMacro(m, F_GETLK)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 480 | #endif |
| 481 | #ifdef F_SETLK |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 482 | if (PyModule_AddIntMacro(m, F_SETLK)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 483 | #endif |
| 484 | #ifdef F_SETLKW |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 485 | if (PyModule_AddIntMacro(m, F_SETLKW)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 486 | #endif |
| 487 | #ifdef F_GETOWN |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 488 | if (PyModule_AddIntMacro(m, F_GETOWN)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 489 | #endif |
| 490 | #ifdef F_SETOWN |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 491 | if (PyModule_AddIntMacro(m, F_SETOWN)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 492 | #endif |
| 493 | #ifdef F_GETSIG |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 494 | if (PyModule_AddIntMacro(m, F_GETSIG)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 495 | #endif |
| 496 | #ifdef F_SETSIG |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 497 | if (PyModule_AddIntMacro(m, F_SETSIG)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 498 | #endif |
| 499 | #ifdef F_RDLCK |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 500 | if (PyModule_AddIntMacro(m, F_RDLCK)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 501 | #endif |
| 502 | #ifdef F_WRLCK |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 503 | if (PyModule_AddIntMacro(m, F_WRLCK)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 504 | #endif |
| 505 | #ifdef F_UNLCK |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 506 | if (PyModule_AddIntMacro(m, F_UNLCK)) return -1; |
Fred Drake | 152a25e | 2001-05-09 21:02:02 +0000 | [diff] [blame] | 507 | #endif |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 508 | /* LFS constants */ |
| 509 | #ifdef F_GETLK64 |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 510 | if (PyModule_AddIntMacro(m, F_GETLK64)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 511 | #endif |
| 512 | #ifdef F_SETLK64 |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 513 | if (PyModule_AddIntMacro(m, F_SETLK64)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 514 | #endif |
| 515 | #ifdef F_SETLKW64 |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 516 | if (PyModule_AddIntMacro(m, F_SETLKW64)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 517 | #endif |
| 518 | /* GNU extensions, as of glibc 2.2.4. */ |
Alexandre Vassalotti | bee3253 | 2008-05-16 18:15:12 +0000 | [diff] [blame] | 519 | #ifdef FASYNC |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 520 | if (PyModule_AddIntMacro(m, FASYNC)) return -1; |
Alexandre Vassalotti | bee3253 | 2008-05-16 18:15:12 +0000 | [diff] [blame] | 521 | #endif |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 522 | #ifdef F_SETLEASE |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 523 | if (PyModule_AddIntMacro(m, F_SETLEASE)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 524 | #endif |
| 525 | #ifdef F_GETLEASE |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 526 | if (PyModule_AddIntMacro(m, F_GETLEASE)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 527 | #endif |
| 528 | #ifdef F_NOTIFY |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 529 | if (PyModule_AddIntMacro(m, F_NOTIFY)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 530 | #endif |
| 531 | /* Old BSD flock(). */ |
| 532 | #ifdef F_EXLCK |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 533 | if (PyModule_AddIntMacro(m, F_EXLCK)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 534 | #endif |
| 535 | #ifdef F_SHLCK |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 536 | if (PyModule_AddIntMacro(m, F_SHLCK)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 537 | #endif |
| 538 | |
Charles-François Natali | 23e1ecb | 2011-11-02 18:58:25 +0100 | [diff] [blame] | 539 | /* OS X specifics */ |
Georg Brandl | 6aa2d1f | 2008-08-12 08:35:52 +0000 | [diff] [blame] | 540 | #ifdef F_FULLFSYNC |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 541 | if (PyModule_AddIntMacro(m, F_FULLFSYNC)) return -1; |
Georg Brandl | 6aa2d1f | 2008-08-12 08:35:52 +0000 | [diff] [blame] | 542 | #endif |
Charles-François Natali | 23e1ecb | 2011-11-02 18:58:25 +0100 | [diff] [blame] | 543 | #ifdef F_NOCACHE |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 544 | if (PyModule_AddIntMacro(m, F_NOCACHE)) return -1; |
Charles-François Natali | 23e1ecb | 2011-11-02 18:58:25 +0100 | [diff] [blame] | 545 | #endif |
Georg Brandl | 6aa2d1f | 2008-08-12 08:35:52 +0000 | [diff] [blame] | 546 | |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 547 | /* For F_{GET|SET}FL */ |
| 548 | #ifdef FD_CLOEXEC |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 549 | if (PyModule_AddIntMacro(m, FD_CLOEXEC)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 550 | #endif |
| 551 | |
| 552 | /* For F_NOTIFY */ |
| 553 | #ifdef DN_ACCESS |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 554 | if (PyModule_AddIntMacro(m, DN_ACCESS)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 555 | #endif |
| 556 | #ifdef DN_MODIFY |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 557 | if (PyModule_AddIntMacro(m, DN_MODIFY)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 558 | #endif |
| 559 | #ifdef DN_CREATE |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 560 | if (PyModule_AddIntMacro(m, DN_CREATE)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 561 | #endif |
| 562 | #ifdef DN_DELETE |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 563 | if (PyModule_AddIntMacro(m, DN_DELETE)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 564 | #endif |
| 565 | #ifdef DN_RENAME |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 566 | if (PyModule_AddIntMacro(m, DN_RENAME)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 567 | #endif |
| 568 | #ifdef DN_ATTRIB |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 569 | if (PyModule_AddIntMacro(m, DN_ATTRIB)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 570 | #endif |
| 571 | #ifdef DN_MULTISHOT |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 572 | if (PyModule_AddIntMacro(m, DN_MULTISHOT)) return -1; |
Martin v. Löwis | 1baeba6 | 2001-12-28 21:08:12 +0000 | [diff] [blame] | 573 | #endif |
| 574 | |
Martin v. Löwis | 14e73b1 | 2003-01-01 09:51:12 +0000 | [diff] [blame] | 575 | #ifdef HAVE_STROPTS_H |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 576 | /* Unix 98 guarantees that these are in stropts.h. */ |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 577 | if (PyModule_AddIntMacro(m, I_PUSH)) return -1; |
| 578 | if (PyModule_AddIntMacro(m, I_POP)) return -1; |
| 579 | if (PyModule_AddIntMacro(m, I_LOOK)) return -1; |
| 580 | if (PyModule_AddIntMacro(m, I_FLUSH)) return -1; |
| 581 | if (PyModule_AddIntMacro(m, I_FLUSHBAND)) return -1; |
| 582 | if (PyModule_AddIntMacro(m, I_SETSIG)) return -1; |
| 583 | if (PyModule_AddIntMacro(m, I_GETSIG)) return -1; |
| 584 | if (PyModule_AddIntMacro(m, I_FIND)) return -1; |
| 585 | if (PyModule_AddIntMacro(m, I_PEEK)) return -1; |
| 586 | if (PyModule_AddIntMacro(m, I_SRDOPT)) return -1; |
| 587 | if (PyModule_AddIntMacro(m, I_GRDOPT)) return -1; |
| 588 | if (PyModule_AddIntMacro(m, I_NREAD)) return -1; |
| 589 | if (PyModule_AddIntMacro(m, I_FDINSERT)) return -1; |
| 590 | if (PyModule_AddIntMacro(m, I_STR)) return -1; |
| 591 | if (PyModule_AddIntMacro(m, I_SWROPT)) return -1; |
Michael W. Hudson | 505c4c2 | 2003-05-09 10:45:20 +0000 | [diff] [blame] | 592 | #ifdef I_GWROPT |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 593 | /* despite the comment above, old-ish glibcs miss a couple... */ |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 594 | if (PyModule_AddIntMacro(m, I_GWROPT)) return -1; |
Michael W. Hudson | 505c4c2 | 2003-05-09 10:45:20 +0000 | [diff] [blame] | 595 | #endif |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 596 | if (PyModule_AddIntMacro(m, I_SENDFD)) return -1; |
| 597 | if (PyModule_AddIntMacro(m, I_RECVFD)) return -1; |
| 598 | if (PyModule_AddIntMacro(m, I_LIST)) return -1; |
| 599 | if (PyModule_AddIntMacro(m, I_ATMARK)) return -1; |
| 600 | if (PyModule_AddIntMacro(m, I_CKBAND)) return -1; |
| 601 | if (PyModule_AddIntMacro(m, I_GETBAND)) return -1; |
| 602 | if (PyModule_AddIntMacro(m, I_CANPUT)) return -1; |
| 603 | if (PyModule_AddIntMacro(m, I_SETCLTIME)) return -1; |
Michael W. Hudson | 505c4c2 | 2003-05-09 10:45:20 +0000 | [diff] [blame] | 604 | #ifdef I_GETCLTIME |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 605 | if (PyModule_AddIntMacro(m, I_GETCLTIME)) return -1; |
Michael W. Hudson | 505c4c2 | 2003-05-09 10:45:20 +0000 | [diff] [blame] | 606 | #endif |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 607 | if (PyModule_AddIntMacro(m, I_LINK)) return -1; |
| 608 | if (PyModule_AddIntMacro(m, I_UNLINK)) return -1; |
| 609 | if (PyModule_AddIntMacro(m, I_PLINK)) return -1; |
| 610 | if (PyModule_AddIntMacro(m, I_PUNLINK)) return -1; |
Martin v. Löwis | 14e73b1 | 2003-01-01 09:51:12 +0000 | [diff] [blame] | 611 | #endif |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 612 | |
| 613 | return 0; |
Guido van Rossum | f4e32c7 | 1997-07-31 19:39:54 +0000 | [diff] [blame] | 614 | } |
| 615 | |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 616 | |
| 617 | static struct PyModuleDef fcntlmodule = { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 618 | PyModuleDef_HEAD_INIT, |
| 619 | "fcntl", |
| 620 | module_doc, |
| 621 | -1, |
| 622 | fcntl_methods, |
| 623 | NULL, |
| 624 | NULL, |
| 625 | NULL, |
| 626 | NULL |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 627 | }; |
| 628 | |
Mark Hammond | fe51c6d | 2002-08-02 02:27:13 +0000 | [diff] [blame] | 629 | PyMODINIT_FUNC |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 630 | PyInit_fcntl(void) |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 631 | { |
Charles-Francois Natali | 74ca886 | 2013-05-20 19:13:19 +0200 | [diff] [blame] | 632 | PyObject *m; |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 633 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 634 | /* Create the module and add the functions and documentation */ |
| 635 | m = PyModule_Create(&fcntlmodule); |
| 636 | if (m == NULL) |
| 637 | return NULL; |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 638 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 639 | /* Add some symbolic constants to the module */ |
Charles-François Natali | 5abca14 | 2013-12-01 14:30:47 +0100 | [diff] [blame] | 640 | if (all_ins(m) < 0) |
| 641 | return NULL; |
| 642 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 643 | return m; |
Guido van Rossum | 0297512 | 1992-08-17 08:55:12 +0000 | [diff] [blame] | 644 | } |