forked from mirrors/kore
230 lines
8.0 KiB
C
230 lines
8.0 KiB
C
/*
|
|
* Copyright (c) 2017-2018 Joris Vink <joris@coders.se>
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
static PyObject *python_kore_log(PyObject *, PyObject *);
|
|
static PyObject *python_kore_fatal(PyObject *, PyObject *);
|
|
static PyObject *python_kore_listen(PyObject *, PyObject *);
|
|
|
|
#if defined(KORE_USE_PGSQL)
|
|
static PyObject *python_kore_pgsql_register(PyObject *, PyObject *);
|
|
#endif
|
|
|
|
static PyObject *python_websocket_broadcast(PyObject *, PyObject *);
|
|
|
|
#define METHOD(n, c, a) { n, (PyCFunction)c, a, NULL }
|
|
#define GETTER(n, g) { n, (getter)g, NULL, NULL, NULL }
|
|
#define SETTER(n, s) { n, NULL, (setter)g, NULL, NULL }
|
|
#define GETSET(n, g, s) { n, (getter)g, (setter)s, NULL, NULL }
|
|
|
|
static struct PyMethodDef pykore_methods[] = {
|
|
METHOD("log", python_kore_log, METH_VARARGS),
|
|
METHOD("fatal", python_kore_fatal, METH_VARARGS),
|
|
METHOD("listen", python_kore_listen, METH_VARARGS),
|
|
METHOD("websocket_broadcast", python_websocket_broadcast, METH_VARARGS),
|
|
#if defined(KORE_USE_PGSQL)
|
|
METHOD("register_database", python_kore_pgsql_register, METH_VARARGS),
|
|
#endif
|
|
{ NULL, NULL, 0, NULL }
|
|
};
|
|
|
|
static struct PyModuleDef pykore_module = {
|
|
PyModuleDef_HEAD_INIT, "kore", NULL, -1, pykore_methods
|
|
};
|
|
|
|
struct pyconnection {
|
|
PyObject_HEAD
|
|
struct connection *c;
|
|
};
|
|
|
|
static PyObject *pyconnection_disconnect(struct pyconnection *, PyObject *);
|
|
static PyObject *pyconnection_websocket_send(struct pyconnection *, PyObject *);
|
|
|
|
static PyMethodDef pyconnection_methods[] = {
|
|
METHOD("disconnect", pyconnection_disconnect, METH_NOARGS),
|
|
METHOD("websocket_send", pyconnection_websocket_send, METH_VARARGS),
|
|
METHOD(NULL, NULL, -1),
|
|
};
|
|
|
|
static PyObject *pyconnection_get_fd(struct pyconnection *, void *);
|
|
static PyObject *pyconnection_get_addr(struct pyconnection *, void *);
|
|
|
|
static PyGetSetDef pyconnection_getset[] = {
|
|
GETTER("fd", pyconnection_get_fd),
|
|
GETTER("addr", pyconnection_get_addr),
|
|
GETTER(NULL, NULL),
|
|
};
|
|
|
|
static void pyconnection_dealloc(struct pyconnection *);
|
|
|
|
static PyTypeObject pyconnection_type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
.tp_name = "kore.connection",
|
|
.tp_doc = "struct connection",
|
|
.tp_getset = pyconnection_getset,
|
|
.tp_methods = pyconnection_methods,
|
|
.tp_basicsize = sizeof(struct pyconnection),
|
|
.tp_dealloc = (destructor)pyconnection_dealloc,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
};
|
|
|
|
struct pyhttp_request {
|
|
PyObject_HEAD
|
|
struct http_request *req;
|
|
PyObject *data;
|
|
};
|
|
|
|
struct pyhttp_file {
|
|
PyObject_HEAD
|
|
struct http_file *file;
|
|
};
|
|
|
|
static void pyhttp_dealloc(struct pyhttp_request *);
|
|
static void pyhttp_file_dealloc(struct pyhttp_file *);
|
|
|
|
#if defined(KORE_USE_PGSQL)
|
|
static PyObject *pyhttp_pgsql(struct pyhttp_request *, PyObject *);
|
|
#endif
|
|
static PyObject *pyhttp_cookie(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_response(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_argument(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_body_read(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_file_lookup(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_populate_get(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_populate_post(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_populate_multi(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_populate_cookies(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_request_header(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_response_header(struct pyhttp_request *, PyObject *);
|
|
static PyObject *pyhttp_websocket_handshake(struct pyhttp_request *,
|
|
PyObject *);
|
|
|
|
static PyMethodDef pyhttp_request_methods[] = {
|
|
#if defined(KORE_USE_PGSQL)
|
|
METHOD("pgsql", pyhttp_pgsql, METH_VARARGS),
|
|
#endif
|
|
METHOD("cookie", pyhttp_cookie, METH_VARARGS),
|
|
METHOD("response", pyhttp_response, METH_VARARGS),
|
|
METHOD("argument", pyhttp_argument, METH_VARARGS),
|
|
METHOD("body_read", pyhttp_body_read, METH_VARARGS),
|
|
METHOD("file_lookup", pyhttp_file_lookup, METH_VARARGS),
|
|
METHOD("populate_get", pyhttp_populate_get, METH_NOARGS),
|
|
METHOD("populate_post", pyhttp_populate_post, METH_NOARGS),
|
|
METHOD("populate_multi", pyhttp_populate_multi, METH_NOARGS),
|
|
METHOD("populate_cookies", pyhttp_populate_cookies, METH_NOARGS),
|
|
METHOD("request_header", pyhttp_request_header, METH_VARARGS),
|
|
METHOD("response_header", pyhttp_response_header, METH_VARARGS),
|
|
METHOD("websocket_handshake", pyhttp_websocket_handshake, METH_VARARGS),
|
|
METHOD(NULL, NULL, -1)
|
|
};
|
|
|
|
static PyObject *pyhttp_get_host(struct pyhttp_request *, void *);
|
|
static PyObject *pyhttp_get_path(struct pyhttp_request *, void *);
|
|
static PyObject *pyhttp_get_body(struct pyhttp_request *, void *);
|
|
static PyObject *pyhttp_get_agent(struct pyhttp_request *, void *);
|
|
static PyObject *pyhttp_get_method(struct pyhttp_request *, void *);
|
|
static PyObject *pyhttp_get_body_path(struct pyhttp_request *, void *);
|
|
static PyObject *pyhttp_get_connection(struct pyhttp_request *, void *);
|
|
|
|
static PyGetSetDef pyhttp_request_getset[] = {
|
|
GETTER("host", pyhttp_get_host),
|
|
GETTER("path", pyhttp_get_path),
|
|
GETTER("body", pyhttp_get_body),
|
|
GETTER("agent", pyhttp_get_agent),
|
|
GETTER("method", pyhttp_get_method),
|
|
GETTER("body_path", pyhttp_get_body_path),
|
|
GETTER("connection", pyhttp_get_connection),
|
|
GETTER(NULL, NULL)
|
|
};
|
|
|
|
static PyTypeObject pyhttp_request_type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
.tp_name = "kore.http_request",
|
|
.tp_doc = "struct http_request",
|
|
.tp_getset = pyhttp_request_getset,
|
|
.tp_methods = pyhttp_request_methods,
|
|
.tp_dealloc = (destructor)pyhttp_dealloc,
|
|
.tp_basicsize = sizeof(struct pyhttp_request),
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
};
|
|
|
|
static PyObject *pyhttp_file_read(struct pyhttp_file *, PyObject *);
|
|
|
|
static PyMethodDef pyhttp_file_methods[] = {
|
|
METHOD("read", pyhttp_file_read, METH_VARARGS),
|
|
METHOD(NULL, NULL, -1)
|
|
};
|
|
|
|
static PyObject *pyhttp_file_get_name(struct pyhttp_file *, void *);
|
|
static PyObject *pyhttp_file_get_filename(struct pyhttp_file *, void *);
|
|
|
|
static PyGetSetDef pyhttp_file_getset[] = {
|
|
GETTER("name", pyhttp_file_get_name),
|
|
GETTER("filename", pyhttp_file_get_filename),
|
|
GETTER(NULL, NULL)
|
|
};
|
|
|
|
static PyTypeObject pyhttp_file_type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
.tp_name = "kore.http_file",
|
|
.tp_doc = "struct http_file",
|
|
.tp_getset = pyhttp_file_getset,
|
|
.tp_methods = pyhttp_file_methods,
|
|
.tp_dealloc = (destructor)pyhttp_file_dealloc,
|
|
.tp_basicsize = sizeof(struct pyhttp_file),
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
};
|
|
|
|
#if defined(KORE_USE_PGSQL)
|
|
|
|
#define PYKORE_PGSQL_PREINIT 1
|
|
#define PYKORE_PGSQL_INITIALIZE 2
|
|
#define PYKORE_PGSQL_QUERY 3
|
|
#define PYKORE_PGSQL_WAIT 4
|
|
|
|
struct pykore_pgsql {
|
|
PyObject_HEAD
|
|
int state;
|
|
char *db;
|
|
char *query;
|
|
struct http_request *req;
|
|
PyObject *result;
|
|
struct kore_pgsql sql;
|
|
};
|
|
|
|
static void pykore_pgsql_dealloc(struct pykore_pgsql *);
|
|
int pykore_pgsql_result(struct pykore_pgsql *);
|
|
|
|
static PyObject *pykore_pgsql_await(PyObject *);
|
|
static PyObject *pykore_pgsql_iternext(struct pykore_pgsql *);
|
|
|
|
static PyAsyncMethods pykore_pgsql_async = {
|
|
(unaryfunc)pykore_pgsql_await,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
static PyTypeObject pykore_pgsql_type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
.tp_name = "kore.pgsql",
|
|
.tp_doc = "struct kore_pgsql",
|
|
.tp_as_async = &pykore_pgsql_async,
|
|
.tp_iternext = (iternextfunc)pykore_pgsql_iternext,
|
|
.tp_basicsize = sizeof(struct pykore_pgsql),
|
|
.tp_dealloc = (destructor)pykore_pgsql_dealloc,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
};
|
|
#endif
|