Commit | Line | Data |
---|---|---|
5b1830f3 MW |
1 | ### -*-pyrex-*- |
2 | ### | |
3 | ### Messing with file descriptors | |
4 | ### | |
5 | ### (c) 2007 Straylight/Edgeware | |
6 | ### | |
ae68e889 | 7 | |
5b1830f3 MW |
8 | ###----- Licensing notice --------------------------------------------------- |
9 | ### | |
10 | ### This file is part of the Python interface to mLib. | |
11 | ### | |
12 | ### mLib/Python is free software; you can redistribute it and/or modify | |
13 | ### it under the terms of the GNU General Public License as published by | |
14 | ### the Free Software Foundation; either version 2 of the License, or | |
15 | ### (at your option) any later version. | |
16 | ### | |
17 | ### mLib/Python is distributed in the hope that it will be useful, | |
18 | ### but WITHOUT ANY WARRANTY; without even the implied warranty of | |
19 | ### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
20 | ### GNU General Public License for more details. | |
21 | ### | |
22 | ### You should have received a copy of the GNU General Public License | |
23 | ### along with mLib/Python; if not, write to the Free Software Foundation, | |
24 | ### Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | |
ae68e889 MW |
25 | |
26 | def fdflags(file, | |
27 | unsigned fbic = 0, unsigned fxor = 0, | |
28 | unsigned fdbic = 0, unsigned fdxor = 0): | |
addc0c37 MW |
29 | """ |
30 | fdflags(FILE, [fbic = 0], [fxor = 0], [fdbic = 0], [fdxor = 0]) | |
31 | ||
32 | Set fcntl(2) file and descriptor flags. If these are FL and FD, then | |
33 | update: | |
34 | ||
35 | * FL = (FL & ~FBIC) ^ FXOR | |
36 | * FD = (FD & ~FDBIC) ^ FDXOR | |
37 | ||
38 | FILE may be integer file descriptor or an object with `fileno' method. | |
39 | """ | |
ae68e889 | 40 | cdef int rc |
f0bf14e9 | 41 | rc = _fdflags(_getfd(file), fbic, fxor, fdbic, fdxor) |
ae68e889 MW |
42 | if rc < 0: |
43 | _oserror() | |
44 | return rc | |
45 | ||
46 | def fdsend(sock, file, buffer): | |
addc0c37 MW |
47 | """ |
48 | fdsend(SOCK, FILE, BUFFER) -> RC: | |
49 | send FILE over Unix-domain socket SOCK, along with BUFFER | |
50 | """ | |
ae68e889 | 51 | cdef void *p |
78911cdb | 52 | cdef Py_ssize_t len |
ae68e889 | 53 | cdef int rc |
704500e1 | 54 | PyObject_AsReadBuffer(buffer, <cvp *>&p, &len) |
ae68e889 MW |
55 | rc = fdpass_send(_getfd(sock), _getfd(file), p, len) |
56 | if rc < 0: | |
57 | _oserror() | |
58 | return rc | |
59 | ||
60 | def fdrecv(sock, unsigned size): | |
addc0c37 MW |
61 | """ |
62 | fdrecv(SOCK, SIZE) -> FD, BUFFER | |
63 | receive file FD and BUFFER of length up to SIZE from Unix-domain SOCK | |
64 | """ | |
ae68e889 MW |
65 | cdef void *p |
66 | cdef buf | |
78911cdb | 67 | cdef Py_ssize_t len |
ae68e889 MW |
68 | cdef PyObject *obj |
69 | cdef int fd | |
23bff39b | 70 | buf = PyString_FromStringAndSize(NULL, size) |
ae68e889 MW |
71 | p = PyString_AS_STRING(buf) |
72 | len = fdpass_recv(_getfd(sock), &fd, p, size) | |
73 | if len < 0: | |
74 | _oserror() | |
75 | obj = <PyObject *>buf | |
76 | _PyString_Resize(&obj, len) | |
77 | return fd, <object>obj | |
78 | ||
5b1830f3 | 79 | ###----- That's all, folks -------------------------------------------------- |