s4:web_server/wsgi.c - fix a counter type
[metze/samba/wip.git] / source4 / web_server / wsgi.c
index 3d5d0b4bf06e8e8ccea827d579d9b784f140d4f0..93dd8fe5c0a24917a670488834fcaddd60fa44b2 100644 (file)
@@ -5,26 +5,39 @@
 
    Implementation of the WSGI interface described in PEP0333 
    (http://www.python.org/dev/peps/pep-0333)
-   
+
    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 "includes.h"
 #include "web_server/web_server.h"
-#include "lib/util/dlinklist.h"
-#include "lib/util/data_blob.h"
-#include <Python.h>
+#include "../lib/util/dlinklist.h"
+#include "../lib/util/data_blob.h"
+#include "lib/tls/tls.h"
+#include "lib/tsocket/tsocket.h"
+
+/* There's no Py_ssize_t in 2.4, apparently */
+#if PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION < 5
+typedef int Py_ssize_t;
+typedef inquiry lenfunc;
+typedef intargfunc ssizeargfunc;
+#endif
+
+#ifndef Py_RETURN_NONE
+#define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
+#endif
 
 typedef struct {
        PyObject_HEAD
@@ -35,7 +48,7 @@ static PyObject *start_response(PyObject *self, PyObject *args, PyObject *kwargs
 {
        PyObject *response_header, *exc_info = NULL;
        char *status;
-       int i;
+       Py_ssize_t i;
        const char *kwnames[] = {
                "status", "response_header", "exc_info", NULL
        };
@@ -89,7 +102,7 @@ static PyObject *start_response(PyObject *self, PyObject *args, PyObject *kwargs
 
        websrv_output_headers(web, status, headers);
 
-       return Py_None;
+       Py_RETURN_NONE;
 }
 
 static PyMethodDef web_request_methods[] = {
@@ -113,7 +126,7 @@ typedef struct {
 static PyObject *py_error_flush(PyObject *self, PyObject *args, PyObject *kwargs)
 {
        /* Nothing to do here */
-       return Py_None;
+       Py_RETURN_NONE;
 }
 
 static PyObject *py_error_write(PyObject *self, PyObject *args, PyObject *kwargs)
@@ -127,7 +140,7 @@ static PyObject *py_error_write(PyObject *self, PyObject *args, PyObject *kwargs
 
        DEBUG(0, ("WSGI App: %s", str));
 
-       return Py_None;
+       Py_RETURN_NONE;
 }
 
 static PyObject *py_error_writelines(PyObject *self, PyObject *args, PyObject *kwargs)
@@ -145,7 +158,7 @@ static PyObject *py_error_writelines(PyObject *self, PyObject *args, PyObject *k
                DEBUG(0, ("WSGI App: %s", str));
        }
 
-       return Py_None;
+       Py_RETURN_NONE;
 }
 
 static PyMethodDef error_Stream_methods[] = {
@@ -165,37 +178,67 @@ PyTypeObject error_Stream_Type = {
 
 typedef struct {
        PyObject_HEAD
+       struct websrv_context *web;
+       size_t offset;
 } input_Stream_Object;
 
-static PyObject *py_input_read(PyObject *self, PyObject *args, PyObject *kwargs)
+static PyObject *py_input_read(PyObject *_self, PyObject *args, PyObject *kwargs)
 {
-       /* FIXME */
-       return NULL;
+       const char *kwnames[] = { "size", NULL };
+       PyObject *ret;
+       input_Stream_Object *self = (input_Stream_Object *)_self;
+       int size = -1;
+
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", discard_const_p(char *, kwnames), &size))
+               return NULL;
+       
+       /* Don't read beyond buffer boundaries */
+       if (size == -1)
+               size = self->web->input.partial.length-self->offset;
+       else
+               size = MIN(size, self->web->input.partial.length-self->offset);
+
+       ret = PyString_FromStringAndSize((char *)self->web->input.partial.data+self->offset, size);
+       self->offset += size;
+
+       return ret;
 }
 
-static PyObject *py_input_readline(PyObject *self, PyObject *args, PyObject *kwargs)
+static PyObject *py_input_readline(PyObject *_self)
 {
        /* FIXME */
+       PyErr_SetString(PyExc_NotImplementedError, 
+                       "readline() not yet implemented");
        return NULL;
 }
 
-static PyObject *py_input_readlines(PyObject *self, PyObject *args, PyObject *kwargs)
+static PyObject *py_input_readlines(PyObject *_self, PyObject *args, PyObject *kwargs)
 {
+       const char *kwnames[] = { "hint", NULL };
+       int hint;
+
+       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", discard_const_p(char *, kwnames), &hint))
+               return NULL;
+       
        /* FIXME */
+       PyErr_SetString(PyExc_NotImplementedError, 
+                       "readlines() not yet implemented");
        return NULL;
 }
 
-static PyObject *py_input___iter__(PyObject *self, PyObject *args, PyObject *kwargs)
+static PyObject *py_input___iter__(PyObject *_self)
 {
        /* FIXME */
+       PyErr_SetString(PyExc_NotImplementedError, 
+                       "__iter__() not yet implemented");
        return NULL;
 }
 
 static PyMethodDef input_Stream_methods[] = {
        { "read", (PyCFunction)py_input_read, METH_VARARGS|METH_KEYWORDS, NULL },
-       { "readline", (PyCFunction)py_input_readline, METH_VARARGS|METH_KEYWORDS, NULL },
+       { "readline", (PyCFunction)py_input_readline, METH_NOARGS, NULL },
        { "readlines", (PyCFunction)py_input_readlines, METH_VARARGS|METH_KEYWORDS, NULL },
-       { "__iter__", (PyCFunction)py_input___iter__, METH_VARARGS|METH_KEYWORDS, NULL },
+       { "__iter__", (PyCFunction)py_input___iter__, METH_NOARGS, NULL },
        { NULL, NULL, 0, NULL }
 };
 
@@ -207,9 +250,11 @@ PyTypeObject input_Stream_Type = {
        .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
 };
 
-static PyObject *Py_InputHttpStream(void *foo)
+static PyObject *Py_InputHttpStream(struct websrv_context *web)
 {
        input_Stream_Object *ret = PyObject_New(input_Stream_Object, &input_Stream_Type);
+       ret->web = web;
+       ret->offset = 0;
        return (PyObject *)ret;
 }
 
@@ -219,24 +264,19 @@ static PyObject *Py_ErrorHttpStream(void)
        return (PyObject *)ret;
 }
 
-static PyObject *create_environ(bool tls, int content_length, const char *content_type, struct http_header *headers, const char *request_method)
+static PyObject *create_environ(bool tls, int content_length, struct http_header *headers, const char *request_method, const char *servername, int serverport, PyObject *inputstream, const char *request_string)
 {
        PyObject *env;
-       PyObject *inputstream, *errorstream;
+       PyObject *errorstream;
        PyObject *py_scheme;
        struct http_header *hdr;
+       char *questionmark;
        
        env = PyDict_New();
        if (env == NULL) {
                return NULL;
        }
 
-       inputstream = Py_InputHttpStream(NULL);
-       if (inputstream == NULL) {
-               Py_DECREF(env);
-               return NULL;
-       }
-
        errorstream = Py_ErrorHttpStream();
        if (errorstream == NULL) {
                Py_DECREF(env);
@@ -251,24 +291,35 @@ static PyObject *create_environ(bool tls, int content_length, const char *conten
        PyDict_SetItemString(env, "wsgi.multiprocess", Py_True);
        PyDict_SetItemString(env, "wsgi.run_once", Py_False);
        PyDict_SetItemString(env, "SERVER_PROTOCOL", PyString_FromString("HTTP/1.0"));
-       if (content_type != NULL) {
-               PyDict_SetItemString(env, "CONTENT_TYPE", PyString_FromString(content_type));
-       }
        if (content_length > 0) {
                PyDict_SetItemString(env, "CONTENT_LENGTH", PyLong_FromLong(content_length));
        }
        PyDict_SetItemString(env, "REQUEST_METHOD", PyString_FromString(request_method));
 
-       /* FIXME: SCRIPT_NAME */
-       /* FIXME: PATH_INFO */
-       /* FIXME: QUERY_STRING */
-       /* FIXME: SERVER_NAME, SERVER_PORT */
-
+       questionmark = strchr(request_string, '?');
+       if (questionmark == NULL) {
+               PyDict_SetItemString(env, "SCRIPT_NAME", PyString_FromString(request_string));
+       } else {
+               PyDict_SetItemString(env, "QUERY_STRING", PyString_FromString(questionmark+1));
+               PyDict_SetItemString(env, "SCRIPT_NAME", PyString_FromStringAndSize(request_string, questionmark-request_string));
+       }
+       
+       PyDict_SetItemString(env, "SERVER_NAME", PyString_FromString(servername));
+       PyDict_SetItemString(env, "SERVER_PORT", PyInt_FromLong(serverport));
        for (hdr = headers; hdr; hdr = hdr->next) {
                char *name;
-               asprintf(&name, "HTTP_%s", hdr->name);
-               PyDict_SetItemString(env, name, PyString_FromString(hdr->value));
-               free(name);
+               if (!strcasecmp(hdr->name, "Content-Type")) {
+                       PyDict_SetItemString(env, "CONTENT_TYPE", PyString_FromString(hdr->value));
+               } else { 
+                       if (asprintf(&name, "HTTP_%s", hdr->name) < 0) {
+                               Py_DECREF(env);
+                               Py_DECREF(inputstream);
+                               PyErr_NoMemory();
+                               return NULL;
+                       }
+                       PyDict_SetItemString(env, name, PyString_FromString(hdr->value));
+                       free(name);
+               }
        }
 
        if (tls) {
@@ -285,16 +336,27 @@ static void wsgi_process_http_input(struct web_server_data *wdata,
                                    struct websrv_context *web)
 {
        PyObject *py_environ, *result, *item, *iter;
-       PyObject *request_handler = wdata->private;
-
+       PyObject *request_handler = (PyObject *)wdata->private_data;
+       struct tsocket_address *my_address = web->conn->local_address;
+       const char *addr = "0.0.0.0";
+       uint16_t port = 0;
        web_request_Object *py_web = PyObject_New(web_request_Object, &web_request_Type);
        py_web->web = web;
 
-       py_environ = create_environ(false /* FIXME: Figure out whether we're using tls */, 
+       if (tsocket_address_is_inet(my_address, "ip")) {
+               addr = tsocket_address_inet_addr_string(my_address, wdata);
+               port = tsocket_address_inet_port(my_address);
+       }
+
+       py_environ = create_environ(tls_enabled(web->conn->socket),
                                    web->input.content_length, 
-                                   web->input.content_type, 
                                    web->input.headers, 
-                                   web->input.post_request?"POST":"GET");
+                                   web->input.post_request?"POST":"GET",
+                                   addr,
+                                   port,
+                                   Py_InputHttpStream(web),
+                                   web->input.url
+                                   );
        if (py_environ == NULL) {
                DEBUG(0, ("Unable to create WSGI environment object\n"));
                return;
@@ -342,6 +404,6 @@ bool wsgi_initialize(struct web_server_data *wdata)
                DEBUG(0, ("Unable to find SWAT\n"));
                return false;
        }
-       wdata->private = py_swat;
+       wdata->private_data = py_swat;
        return true;
 }