diff options
Diffstat (limited to 'src/nxt_python_wsgi.c')
| -rw-r--r-- | src/nxt_python_wsgi.c | 973 |
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); +} |