[ashd.git] / python3 / htp.c

/*
    ashd - A Sane HTTP Daemon
    Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <Python.h>
#include <errno.h>

#include <ashd/utils.h>
#include <ashd/proc.h>

static PyObject *p_recvfd(PyObject *self, PyObject *args)
{
    int fd, ret;
    char *data;
    size_t dlen;
    PyObject *ro;
    
    fd = 0;
    if(!PyArg_ParseTuple(args, "|i", &fd))
	return(NULL);
    Py_BEGIN_ALLOW_THREADS;
    ret = recvfd(fd, &data, &dlen);
    Py_END_ALLOW_THREADS;
    if(ret < 0) {
	if(errno == 0)
	    return(Py_BuildValue("OO", Py_None, Py_None));
	PyErr_SetFromErrno(PyExc_OSError);
	return(NULL);
    }
    ro = Py_BuildValue("Ni", PyBytes_FromStringAndSize(data, dlen), ret);
    free(data);
    return(ro);
}

static PyObject *p_sendfd(PyObject *self, PyObject *args)
{
    int sock, fd, ret;
    Py_buffer data;
    
    if(!PyArg_ParseTuple(args, "iiy*", &sock, &fd, &data))
	return(NULL);
    Py_BEGIN_ALLOW_THREADS;
    ret = sendfd(sock, fd, data.buf, data.len);
    Py_END_ALLOW_THREADS;
    PyBuffer_Release(&data);
    if(ret < 0) {
	PyErr_SetFromErrno(PyExc_OSError);
	return(NULL);
    }
    Py_RETURN_NONE;
}

static PyMethodDef methods[] = {
    {"recvfd", p_recvfd, METH_VARARGS, "Receive a datagram and a file descriptor"},
    {"sendfd", p_sendfd, METH_VARARGS, "Send a datagram and a file descriptor"},
    {NULL, NULL, 0, NULL}
};

static struct PyModuleDef module = {
    PyModuleDef_HEAD_INIT,
    .m_name = "htlib",
    .m_size = -1,
    .m_methods = methods,
};

PyMODINIT_FUNC PyInit_htlib(void)
{
    return(PyModule_Create(&module));
}
Commit	Line	Data
	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	Py_BEGIN_ALLOW_THREADS;
	36	ret = recvfd(fd, &data, &dlen);
	37	Py_END_ALLOW_THREADS;
	38	if(ret < 0) {
	39	if(errno == 0)
	40	return(Py_BuildValue("OO", Py_None, Py_None));
	41	PyErr_SetFromErrno(PyExc_OSError);
	42	return(NULL);
	43	}
	44	ro = Py_BuildValue("Ni", PyBytes_FromStringAndSize(data, dlen), ret);
	45	free(data);
	46	return(ro);
	47	}
	48
	49	static PyObject p_sendfd(PyObject self, PyObject *args)
	50	{
	51	int sock, fd, ret;
	52	Py_buffer data;
	53
	54	if(!PyArg_ParseTuple(args, "iiy*", &sock, &fd, &data))
	55	return(NULL);
	56	Py_BEGIN_ALLOW_THREADS;
	57	ret = sendfd(sock, fd, data.buf, data.len);
	58	Py_END_ALLOW_THREADS;
	59	PyBuffer_Release(&data);
	60	if(ret < 0) {
	61	PyErr_SetFromErrno(PyExc_OSError);
	62	return(NULL);
	63	}
	64	Py_RETURN_NONE;
	65	}
	66
	67	static PyMethodDef methods[] = {
	68	{"recvfd", p_recvfd, METH_VARARGS, "Receive a datagram and a file descriptor"},
	69	{"sendfd", p_sendfd, METH_VARARGS, "Send a datagram and a file descriptor"},
	70	{NULL, NULL, 0, NULL}
	71	};
	72
	73	static struct PyModuleDef module = {
	74	PyModuleDef_HEAD_INIT,
	75	.m_name = "htlib",
	76	.m_size = -1,
	77	.m_methods = methods,
	78	};
	79
	80	PyMODINIT_FUNC PyInit_htlib(void)
	81	{
	82	return(PyModule_Create(&module));
	83	}