summaryrefslogtreecommitdiffhomepage
path: root/src/nxt_python_wsgi.c
diff options
context:
space:
mode:
authorIgor Sysoev <igor@sysoev.ru>2017-01-17 20:00:00 +0300
committerIgor Sysoev <igor@sysoev.ru>2017-01-17 20:00:00 +0300
commit16cbf3c076a0aca6d47adaf3f719493674cf2363 (patch)
treee6530480020f62a2bdbf249988ec3e2a751d3927 /src/nxt_python_wsgi.c
downloadunit-16cbf3c076a0aca6d47adaf3f719493674cf2363.tar.gz
unit-16cbf3c076a0aca6d47adaf3f719493674cf2363.tar.bz2
Initial version.
Diffstat (limited to 'src/nxt_python_wsgi.c')
-rw-r--r--src/nxt_python_wsgi.c973
1 files changed, 973 insertions, 0 deletions
diff --git a/src/nxt_python_wsgi.c b/src/nxt_python_wsgi.c
new file mode 100644
index 00000000..c0b03efe
--- /dev/null
+++ b/src/nxt_python_wsgi.c
@@ -0,0 +1,973 @@
+
+/*
+ * Copyright (C) Valentin V. Bartenev
+ * Copyright (C) NGINX, Inc.
+ */
+
+
+#include <Python.h>
+
+#include <compile.h>
+#include <node.h>
+
+#ifdef _DARWIN_C_SOURCE
+#undef _DARWIN_C_SOURCE
+#endif
+
+#include <nxt_main.h>
+#include <nxt_cycle.h>
+#include <nxt_application.h>
+
+
+typedef struct {
+ PyObject_HEAD
+ //nxt_app_request_t *request;
+} nxt_py_input_t;
+
+
+typedef struct {
+ PyObject_HEAD
+ //nxt_app_request_t *request;
+} nxt_py_error_t;
+
+
+static nxt_int_t nxt_python_init(nxt_thread_t *thr);
+static nxt_int_t nxt_python_run(nxt_app_request_t *r);
+
+static PyObject *nxt_python_create_environ(nxt_thread_t *thr);
+static PyObject *nxt_python_get_environ(nxt_app_request_t *r);
+
+static PyObject *nxt_py_start_resp(PyObject *self, PyObject *args);
+
+static void nxt_py_input_dealloc(nxt_py_input_t *self);
+static PyObject *nxt_py_input_read(nxt_py_input_t *self, PyObject *args);
+static PyObject *nxt_py_input_readline(nxt_py_input_t *self, PyObject *args);
+static PyObject *nxt_py_input_readlines(nxt_py_input_t *self, PyObject *args);
+
+
+extern nxt_int_t nxt_python_wsgi_init(nxt_thread_t *thr, nxt_cycle_t *cycle);
+
+
+nxt_application_module_t nxt_python_module = {
+ nxt_python_init,
+ NULL,
+ NULL,
+ nxt_python_run,
+};
+
+
+static PyMethodDef nxt_py_start_resp_method[] = {
+ {"nginman_start_response", nxt_py_start_resp, METH_VARARGS, ""}
+};
+
+
+static PyMethodDef nxt_py_input_methods[] = {
+ { "read", (PyCFunction) nxt_py_input_read, METH_VARARGS, 0 },
+ { "readline", (PyCFunction) nxt_py_input_readline, METH_VARARGS, 0 },
+ { "readlines", (PyCFunction) nxt_py_input_readlines, METH_VARARGS, 0 },
+ { NULL, NULL, 0, 0 }
+};
+
+
+static PyTypeObject nxt_py_input_type = {
+ PyVarObject_HEAD_INIT(NULL, 0)
+ "nginman._input", /* tp_name */
+ (int) sizeof(nxt_py_input_t), /* tp_basicsize */
+ 0, /* tp_itemsize */
+ (destructor) nxt_py_input_dealloc, /* tp_dealloc */
+ 0, /* tp_print */
+ 0, /* tp_getattr */
+ 0, /* tp_setattr */
+ 0, /* tp_compare */
+ 0, /* tp_repr */
+ 0, /* tp_as_number */
+ 0, /* tp_as_sequence */
+ 0, /* tp_as_mapping */
+ 0, /* tp_hash */
+ 0, /* tp_call */
+ 0, /* tp_str */
+ 0, /* tp_getattro */
+ 0, /* tp_setattro */
+ 0, /* tp_as_buffer */
+ Py_TPFLAGS_DEFAULT, /* tp_flags */
+ "nginman input object.", /* tp_doc */
+ 0, /* tp_traverse */
+ 0, /* tp_clear */
+ 0, /* tp_richcompare */
+ 0, /* tp_weaklistoffset */
+ 0, /* tp_iter */
+ 0, /* tp_iternext */
+ nxt_py_input_methods, /* tp_methods */
+ 0, /* tp_members */
+ 0, /* tp_getset */
+ 0, /* tp_base */
+ 0, /* tp_dict */
+ 0, /* tp_descr_get */
+ 0, /* tp_descr_set */
+ 0, /* tp_dictoffset */
+ 0, /* tp_init */
+ 0, /* tp_alloc */
+ 0, /* tp_new */
+ 0, /* tp_free */
+ 0, /* tp_is_gc */
+ 0, /* tp_bases */
+ 0, /* tp_mro - method resolution order */
+ 0, /* tp_cache */
+ 0, /* tp_subclasses */
+ 0, /* tp_weaklist */
+ 0, /* tp_del */
+ 0, /* tp_version_tag */
+};
+
+
+static char *nxt_py_module;
+
+static PyObject *nxt_py_application;
+static PyObject *nxt_py_start_resp_obj;
+static PyObject *nxt_py_environ_ptyp;
+
+static nxt_app_request_t *nxt_app_request;
+
+
+nxt_int_t
+nxt_python_wsgi_init(nxt_thread_t *thr, nxt_cycle_t *cycle)
+{
+ char **argv;
+ u_char *p, *dir;
+
+ PyObject *obj, *pypath;
+
+ argv = nxt_process_argv;
+
+ while (*argv != NULL) {
+ p = (u_char *) *argv++;
+
+ if (nxt_strcmp(p, "--py-module") == 0) {
+ if (*argv == NULL) {
+ nxt_log_emerg(thr->log,
+ "no argument for option \"--py-module\"");
+ return NXT_ERROR;
+ }
+
+ nxt_py_module = *argv++;
+
+ nxt_log_error(NXT_LOG_INFO, thr->log, "python module: \"%s\"",
+ nxt_py_module);
+
+ break;
+ }
+ }
+
+ if (nxt_py_module == NULL) {
+ return NXT_OK;
+ }
+
+ Py_InitializeEx(0);
+
+ obj = NULL;
+ argv = nxt_process_argv;
+
+ while (*argv != NULL) {
+ p = (u_char *) *argv++;
+
+ if (nxt_strcmp(p, "--py-path") == 0) {
+ if (*argv == NULL) {
+ nxt_log_emerg(thr->log, "no argument for option \"--py-path\"");
+ goto fail;
+ }
+
+ dir = (u_char *) *argv++;
+
+ nxt_log_error(NXT_LOG_INFO, thr->log, "python path \"%s\"", dir);
+
+ obj = PyString_FromString((char *) dir);
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed create string object \"%s\"", dir);
+ goto fail;
+ }
+
+ pypath = PySys_GetObject((char *) "path");
+
+ if (nxt_slow_path(pypath == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to get \"sys.path\" list");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyList_Insert(pypath, 0, obj) != 0)) {
+ nxt_log_alert(thr->log,
+ "Python failed to insert \"%s\" into \"sys.path\"", dir);
+ goto fail;
+ }
+
+ Py_DECREF(obj);
+ obj = NULL;
+
+ continue;
+ }
+ }
+
+ obj = PyCFunction_New(nxt_py_start_resp_method, NULL);
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to initialize the \"start_response\" function");
+ goto fail;
+ }
+
+ nxt_py_start_resp_obj = obj;
+
+ obj = nxt_python_create_environ(thr);
+
+ if (obj == NULL) {
+ goto fail;
+ }
+
+ nxt_py_environ_ptyp = obj;
+
+
+ obj = Py_BuildValue("[s]", "nginman");
+ if (obj == NULL) {
+ nxt_log_alert(thr->log,
+ "Python failed to create the \"sys.argv\" list");
+ goto fail;
+ }
+
+ if (PySys_SetObject((char *) "argv", obj) != 0) {
+ nxt_log_alert(thr->log, "Python failed to set the \"sys.argv\" list");
+ goto fail;
+ }
+
+ Py_DECREF(obj);
+
+ return NXT_OK;
+
+fail:
+
+ Py_XDECREF(obj);
+ Py_XDECREF(nxt_py_start_resp_obj);
+
+ Py_Finalize();
+
+ return NXT_ERROR;
+}
+
+
+static nxt_int_t
+nxt_python_init(nxt_thread_t *thr)
+{
+ PyObject *module, *obj;
+
+#if 0
+ FILE *fp;
+ PyObject *co;
+ struct _node *node;
+
+ chdir((char *) dir);
+ fp = fopen((char *) script, "r");
+
+ if (fp == NULL) {
+ nxt_log_debug(thr->log, "fopen failed");
+ return NXT_ERROR;
+ }
+
+
+ Py_SetProgramName((char *) "python mysite/wsgi.py");
+ Py_InitializeEx(0);
+
+ node = PyParser_SimpleParseFile(fp, (char *) script, Py_file_input);
+
+ fclose(fp);
+
+ if (node == NULL) {
+ nxt_log_debug(thr->log, "BAD node");
+ return NXT_ERROR;
+ }
+
+ co = (PyObject *) PyNode_Compile(node, (char *) script);
+
+ PyNode_Free(node);
+
+ if (co == NULL) {
+ nxt_log_debug(thr->log, "BAD co");
+ return NXT_ERROR;
+ }
+
+ pModule = PyImport_ExecCodeModuleEx((char *) "_wsgi_nginman", co, (char *) script);
+
+ Py_XDECREF(co);
+#endif
+
+ PyOS_AfterFork();
+
+ module = PyImport_ImportModule(nxt_py_module);
+
+ if (nxt_slow_path(module == NULL)) {
+ nxt_log_emerg(thr->log, "Python failed to import module \"%s\"",
+ nxt_py_module);
+ return NXT_ERROR;
+ }
+
+ obj = PyDict_GetItemString(PyModule_GetDict(module), "application");
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_emerg(thr->log, "Python failed to get \"application\" "
+ "from module \"%s\"", nxt_py_module);
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyCallable_Check(obj) == 0)) {
+ nxt_log_emerg(thr->log, "\"application\" in module \"%s\" "
+ "is not a callable object", nxt_py_module);
+ goto fail;
+ }
+
+ Py_INCREF(obj);
+ Py_DECREF(module);
+
+ nxt_py_application = obj;
+
+ return NXT_OK;
+
+fail:
+
+ Py_DECREF(module);
+
+ return NXT_ERROR;
+}
+
+
+static nxt_int_t
+nxt_python_run(nxt_app_request_t *r)
+{
+ u_char *buf;
+ size_t size;
+ PyObject *result, *iterator, *item, *args, *environ;
+
+ nxt_app_request = r;
+
+ environ = nxt_python_get_environ(r);
+
+ if (nxt_slow_path(environ == NULL)) {
+ return NXT_ERROR;
+ }
+
+ args = PyTuple_New(2);
+
+ if (nxt_slow_path(args == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create arguments tuple");
+ return NXT_ERROR;
+ }
+
+ PyTuple_SET_ITEM(args, 0, environ);
+
+ Py_INCREF(nxt_py_start_resp_obj);
+ PyTuple_SET_ITEM(args, 1, nxt_py_start_resp_obj);
+
+ result = PyObject_CallObject(nxt_py_application, args);
+
+ Py_DECREF(args);
+
+ if (nxt_slow_path(result == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to call the application");
+ PyErr_Print();
+ return NXT_ERROR;
+ }
+
+ iterator = PyObject_GetIter(result);
+
+ Py_DECREF(result);
+
+ if (nxt_slow_path(iterator == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "the application returned not an iterable object");
+ return NXT_ERROR;
+ }
+
+ while((item = PyIter_Next(iterator))) {
+
+ if (nxt_slow_path(PyString_Check(item) == 0)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "the application returned not a string object");
+
+ Py_DECREF(item);
+ Py_DECREF(iterator);
+
+ return NXT_ERROR;
+ }
+
+ size = PyString_GET_SIZE(item);
+ buf = (u_char *) PyString_AS_STRING(item);
+
+ nxt_app_write(r, buf, size);
+
+ Py_DECREF(item);
+ }
+
+ Py_DECREF(iterator);
+
+ if (nxt_slow_path(PyErr_Occurred() != NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log, "an application error occurred");
+ PyErr_Print();
+ return NXT_ERROR;
+ }
+
+ return NXT_OK;
+}
+
+
+static PyObject *
+nxt_python_create_environ(nxt_thread_t *thr)
+{
+ PyObject *obj, *stderr, *environ;
+
+ environ = PyDict_New();
+
+ if (nxt_slow_path(environ == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to create the \"environ\" dictionary");
+ return NULL;
+ }
+
+ obj = Py_BuildValue("(ii)", 1, 0);
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to build the \"wsgi.version\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "wsgi.version", obj) != 0))
+ {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"wsgi.version\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(obj);
+ obj = NULL;
+
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "wsgi.multithread",
+ Py_False)
+ != 0))
+ {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"wsgi.multithread\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "wsgi.multiprocess",
+ Py_True)
+ != 0))
+ {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"wsgi.multiprocess\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "wsgi.run_once",
+ Py_False)
+ != 0))
+ {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"wsgi.run_once\" environ value");
+ goto fail;
+ }
+
+
+ obj = PyString_FromString("http");
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to create the \"wsgi.url_scheme\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "wsgi.url_scheme", obj)
+ != 0))
+ {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"wsgi.url_scheme\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(obj);
+ obj = NULL;
+
+
+ if (nxt_slow_path(PyType_Ready(&nxt_py_input_type) != 0)) {
+ nxt_log_alert(thr->log,
+ "Python failed to initialize the \"wsgi.input\" type object");
+ goto fail;
+ }
+
+ obj = (PyObject *) PyObject_New(nxt_py_input_t, &nxt_py_input_type);
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to create the \"wsgi.input\" object");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "wsgi.input", obj) != 0)) {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"wsgi.input\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(obj);
+ obj = NULL;
+
+
+ stderr = PySys_GetObject((char *) "stderr");
+
+ if (nxt_slow_path(stderr == NULL)) {
+ nxt_log_alert(thr->log, "Python failed to get \"sys.stderr\" object");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "wsgi.error", stderr) != 0))
+ {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"wsgi.error\" environ value");
+ goto fail;
+ }
+
+
+ obj = PyString_FromString("localhost");
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to create the \"SERVER_NAME\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "SERVER_NAME", obj) != 0)) {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"SERVER_NAME\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(obj);
+
+
+ obj = PyString_FromString("80");
+
+ if (nxt_slow_path(obj == NULL)) {
+ nxt_log_alert(thr->log,
+ "Python failed to create the \"SERVER_PORT\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "SERVER_PORT", obj) != 0)) {
+ nxt_log_alert(thr->log,
+ "Python failed to set the \"SERVER_PORT\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(obj);
+
+ return environ;
+
+fail:
+
+ Py_XDECREF(obj);
+ Py_DECREF(environ);
+
+ return NULL;
+}
+
+
+static PyObject *
+nxt_python_get_environ(nxt_app_request_t *r)
+{
+ u_char *p, ch, *query;
+ nxt_str_t *str;
+ nxt_uint_t i, n;
+ nxt_app_header_field_t *fld;
+
+ PyObject *environ, *value;
+
+ static const u_char prefix[5] = "HTTP_";
+
+ static u_char key[256];
+
+ environ = PyDict_Copy(nxt_py_environ_ptyp);
+
+ if (nxt_slow_path(environ == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"environ\" dictionary");
+ return NULL;
+ }
+
+ value = PyString_FromStringAndSize((char *) r->header.version.data,
+ r->header.version.len);
+
+ if (nxt_slow_path(value == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"SERVER_PROTOCOL\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "SERVER_PROTOCOL", value)
+ != 0))
+ {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"SERVER_PROTOCOL\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(value);
+
+ value = PyString_FromStringAndSize((char *) r->header.method.data,
+ r->header.method.len);
+
+ if (nxt_slow_path(value == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"REQUEST_METHOD\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "REQUEST_METHOD", value)
+ != 0))
+ {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"REQUEST_METHOD\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(value);
+
+ value = PyString_FromStringAndSize((char *) r->header.path.data,
+ r->header.path.len);
+
+ if (nxt_slow_path(value == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"REQUEST_URI\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "REQUEST_URI", value)
+ != 0))
+ {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"REQUEST_URI\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(value);
+
+ query = nxt_memchr(r->header.path.data, '?', r->header.path.len);
+
+ if (query != NULL) {
+ value = PyString_FromStringAndSize((char *) r->header.path.data,
+ query - r->header.path.data);
+
+ query++;
+
+ } else {
+ value = PyString_FromStringAndSize((char *) r->header.path.data,
+ r->header.path.len);
+ }
+
+ if (nxt_slow_path(value == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"PATH_INFO\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "PATH_INFO", value) != 0)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"PATH_INFO\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(value);
+
+ if (query != NULL) {
+ value = PyString_FromStringAndSize((char *) query,
+ r->header.path.data
+ + r->header.path.len - query);
+
+ if (nxt_slow_path(value == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"QUERY_STRING\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "QUERY_STRING", value)
+ != 0))
+ {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"QUERY_STRING\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(value);
+ }
+
+ if (r->header.content_length != NULL) {
+ str = r->header.content_length;
+
+ value = PyString_FromStringAndSize((char *) str->data, str->len);
+
+ if (nxt_slow_path(value == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"CONTENT_LENGTH\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "CONTENT_LENGTH", value)
+ != 0))
+ {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"CONTENT_LENGTH\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(value);
+ }
+
+ if (r->header.content_type != NULL) {
+ str = r->header.content_type;
+
+ value = PyString_FromStringAndSize((char *) str->data, str->len);
+
+ if (nxt_slow_path(value == NULL)) {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to create the \"CONTENT_TYPE\" environ value");
+ goto fail;
+ }
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, "CONTENT_TYPE", value)
+ != 0))
+ {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"CONTENT_TYPE\" environ value");
+ goto fail;
+ }
+
+ Py_DECREF(value);
+ }
+
+ nxt_memcpy(key, prefix, sizeof(prefix));
+
+ for (i = 0; i < r->header.fields_num; i++) {
+ fld = &r->header.fields[i];
+ p = key + sizeof(prefix);
+
+ for (n = 0; n < fld->name.len; n++, p++) {
+
+ ch = fld->name.data[n];
+
+ if (ch >= 'a' && ch <= 'z') {
+ *p = ch & ~0x20;
+ continue;
+ }
+
+ if (ch == '-') {
+ *p = '_';
+ continue;
+ }
+
+ *p = ch;
+ }
+
+ *p = '\0';
+
+ value = PyString_FromStringAndSize((char *) fld->value.data,
+ fld->value.len);
+
+ if (nxt_slow_path(PyDict_SetItemString(environ, (char *) key, value)
+ != 0))
+ {
+ nxt_log_error(NXT_LOG_ERR, r->log,
+ "Python failed to set the \"%s\" environ value", key);
+ goto fail;
+ }
+
+ Py_DECREF(value);
+ }
+
+ return environ;
+
+fail:
+
+ Py_XDECREF(value);
+ Py_DECREF(environ);
+
+ return NULL;
+}
+
+
+static PyObject *
+nxt_py_start_resp(PyObject *self, PyObject *args)
+{
+ u_char *p, buf[4096];
+ PyObject *headers, *tuple, *string;
+ nxt_str_t str;
+ nxt_uint_t i, n;
+
+ static const u_char resp[] = "HTTP/1.1 ";
+
+ static const u_char default_headers[]
+ = "Server: nginman/0.1\r\n"
+ "Connection: close\r\n";
+
+ n = PyTuple_GET_SIZE(args);
+
+ if (n < 2 || n > 3) {
+ return PyErr_Format(PyExc_TypeError, "invalid number of arguments");
+ }
+
+ string = PyTuple_GET_ITEM(args, 0);
+
+ if (!PyString_Check(string)) {
+ return PyErr_Format(PyExc_TypeError,
+ "the first argument is not a string");
+ }
+
+ str.len = PyString_GET_SIZE(string);
+ str.data = (u_char *) PyString_AS_STRING(string);
+
+ p = nxt_cpymem(buf, resp, sizeof(resp) - 1);
+ p = nxt_cpymem(p, str.data, str.len);
+
+ *p++ = '\r'; *p++ = '\n';
+
+ p = nxt_cpymem(p, default_headers, sizeof(default_headers) - 1);
+
+ headers = PyTuple_GET_ITEM(args, 1);
+
+ if (!PyList_Check(headers)) {
+ return PyErr_Format(PyExc_TypeError,
+ "the second argument is not a response headers list");
+ }
+
+ for (i = 0; i < (nxt_uint_t) PyList_GET_SIZE(headers); i++) {
+ tuple = PyList_GET_ITEM(headers, i);
+
+ if (!PyTuple_Check(tuple)) {
+ return PyErr_Format(PyExc_TypeError,
+ "the response headers must be a list of tuples");
+ }
+
+ if (PyTuple_GET_SIZE(tuple) != 2) {
+ return PyErr_Format(PyExc_TypeError,
+ "each header must be a tuple of two items");
+ }
+
+ string = PyTuple_GET_ITEM(tuple, 0);
+
+ if (!PyString_Check(string)) {
+ return PyErr_Format(PyExc_TypeError,
+ "all response headers names must be strings");
+ }
+
+ str.len = PyString_GET_SIZE(string);
+ str.data = (u_char *) PyString_AS_STRING(string);
+
+ p = nxt_cpymem(p, str.data, str.len);
+
+ *p++ = ':'; *p++ = ' ';
+
+ string = PyTuple_GET_ITEM(tuple, 1);
+
+ if (!PyString_Check(string)) {
+ return PyErr_Format(PyExc_TypeError,
+ "all response headers values must be strings");
+ }
+
+ str.len = PyString_GET_SIZE(string);
+ str.data = (u_char *) PyString_AS_STRING(string);
+
+ p = nxt_cpymem(p, str.data, str.len);
+
+ *p++ = '\r'; *p++ = '\n';
+ }
+
+ *p++ = '\r'; *p++ = '\n';
+
+ nxt_app_write(nxt_app_request, buf, p - buf);
+
+ return args;
+}
+
+
+static void
+nxt_py_input_dealloc(nxt_py_input_t *self)
+{
+ PyObject_Del(self);
+}
+
+
+static PyObject *
+nxt_py_input_read(nxt_py_input_t *self, PyObject *args)
+{
+ u_char *buf;
+ PyObject *body, *obj;
+ Py_ssize_t size;
+ nxt_uint_t n;
+
+ nxt_app_request_t *r = nxt_app_request;
+
+ size = r->body_rest;
+
+ n = PyTuple_GET_SIZE(args);
+
+ if (n > 0) {
+ if (n != 1) {
+ return PyErr_Format(PyExc_TypeError, "invalid number of arguments");
+ }
+
+ obj = PyTuple_GET_ITEM(args, 0);
+
+ size = PyNumber_AsSsize_t(obj, PyExc_OverflowError);
+
+ if (nxt_slow_path(size < 0)) {
+ if (size == -1 && PyErr_Occurred()) {
+ return NULL;
+ }
+
+ return PyErr_Format(PyExc_ValueError,
+ "the read body size cannot be zero or less");
+ }
+
+ if (size == 0 || size > r->body_rest) {
+ size = r->body_rest;
+ }
+ }
+
+ body = PyString_FromStringAndSize(NULL, size);
+
+ if (nxt_slow_path(body == NULL)) {
+ return NULL;
+ }
+
+ buf = (u_char *) PyString_AS_STRING(body);
+
+ if (nxt_app_http_read_body(r, buf, size) != NXT_OK) {
+ return PyErr_Format(PyExc_IOError, "failed to read body");
+ }
+
+ return body;
+}
+
+
+static PyObject *
+nxt_py_input_readline(nxt_py_input_t *self, PyObject *args)
+{
+ return PyString_FromString("");
+}
+
+
+static PyObject *
+nxt_py_input_readlines(nxt_py_input_t *self, PyObject *args)
+{
+ return PyList_New(0);
+}