ratequeue: Add some missing includes.
[ashd.git] / python3 / htp.c
1 /*
2     ashd - A Sane HTTP Daemon
3     Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
4
5     This program is free software: you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation, either version 3 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License
16     along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 */
18
19 #include <Python.h>
20 #include <errno.h>
21
22 #include <ashd/utils.h>
23 #include <ashd/proc.h>
24
25 static PyObject *p_recvfd(PyObject *self, PyObject *args)
26 {
27     int fd, ret;
28     char *data;
29     size_t dlen;
30     PyObject *ro;
31     
32     fd = 0;
33     if(!PyArg_ParseTuple(args, "|i", &fd))
34         return(NULL);
35     while(1) {
36         Py_BEGIN_ALLOW_THREADS;
37         ret = recvfd(fd, &data, &dlen);
38         Py_END_ALLOW_THREADS;
39         if(ret < 0) {
40             if(errno == 0)
41                 return(Py_BuildValue("OO", Py_None, Py_None));
42             if(errno == EINTR) {
43                 if(PyErr_CheckSignals())
44                     return(NULL);
45                 continue;
46             }
47             PyErr_SetFromErrno(PyExc_OSError);
48             return(NULL);
49         }
50         ro = Py_BuildValue("Ni", PyBytes_FromStringAndSize(data, dlen), ret);
51         free(data);
52         return(ro);
53     }
54 }
55
56 static PyObject *p_sendfd(PyObject *self, PyObject *args)
57 {
58     int sock, fd, ret;
59     Py_buffer data;
60     
61     if(!PyArg_ParseTuple(args, "iiy*", &sock, &fd, &data))
62         return(NULL);
63     while(1) {
64         Py_BEGIN_ALLOW_THREADS;
65         ret = sendfd(sock, fd, data.buf, data.len);
66         Py_END_ALLOW_THREADS;
67         PyBuffer_Release(&data);
68         if(ret < 0) {
69             if(errno == EINTR) {
70                 if(PyErr_CheckSignals())
71                     return(NULL);
72                 continue;
73             }
74             PyErr_SetFromErrno(PyExc_OSError);
75             return(NULL);
76         }
77         Py_RETURN_NONE;
78     }
79 }
80
81 static PyMethodDef methods[] = {
82     {"recvfd", p_recvfd, METH_VARARGS, "Receive a datagram and a file descriptor"},
83     {"sendfd", p_sendfd, METH_VARARGS, "Send a datagram and a file descriptor"},
84     {NULL, NULL, 0, NULL}
85 };
86
87 static struct PyModuleDef module = {
88     PyModuleDef_HEAD_INIT,
89     .m_name = "htlib",
90     .m_size = -1,
91     .m_methods = methods,
92 };
93
94 PyMODINIT_FUNC PyInit_htlib(void)
95 {
96     return(PyModule_Create(&module));
97 }