Merge sasl branch
[prosody.git] / net / http / server.lua
index 8b91b4d3fba15fb1ada7b717a1f1a634f50c71d5..dec7da1999c22c4cb6ae57ae83512b06bef369d7 100644 (file)
@@ -9,11 +9,9 @@ local pairs = pairs;
 local s_upper = string.upper;
 local setmetatable = setmetatable;
 local xpcall = xpcall;
-local debug = debug;
+local traceback = debug.traceback;
 local tostring = tostring;
 local codes = require "net.http.codes";
-local _G = _G;
-local legacy_httpserver = require "net.httpserver";
 
 local _M = {};
 
@@ -29,8 +27,11 @@ local function is_wildcard_match(wildcard_event, event)
        return wildcard_event:sub(1, -2) == event:sub(1, #wildcard_event-1);
 end
 
+local recent_wildcard_events, max_cached_wildcard_events = {}, 10000;
+
 local event_map = events._event_map;
 setmetatable(events._handlers, {
+       -- Called when firing an event that doesn't exist (but may match a wildcard handler)
        __index = function (handlers, curr_event)
                if is_wildcard_event(curr_event) then return; end -- Wildcard events cannot be fired
                -- Find all handlers that could match this event, sort them
@@ -60,6 +61,12 @@ setmetatable(events._handlers, {
                        handlers_array = false;
                end
                rawset(handlers, curr_event, handlers_array);
+               if not event_map[curr_event] then -- Only wildcard handlers match, if any
+                       table.insert(recent_wildcard_events, curr_event);
+                       if #recent_wildcard_events > max_cached_wildcard_events then
+                               rawset(handlers, table.remove(recent_wildcard_events, 1), nil);
+                       end
+               end
                return handlers_array;
        end;
        __newindex = function (handlers, curr_event, handlers_array)
@@ -81,7 +88,7 @@ local _1, _2, _3;
 local function _handle_request() return handle_request(_1, _2, _3); end
 
 local last_err;
-local function _traceback_handler(err) last_err = err; log("error", "Traceback[http]: %s: %s", tostring(err), debug.traceback()); end
+local function _traceback_handler(err) last_err = err; log("error", "Traceback[httpserver]: %s", traceback(tostring(err), 2)); end
 events.add_handler("http-error", function (error)
        return "Error processing request: "..codes[error.code]..". Check your error log for more information.";
 end, -1);
@@ -90,30 +97,31 @@ function listener.onconnect(conn)
        local secure = conn:ssl() and true or nil;
        local pending = {};
        local waiting = false;
-       local function process_next(last_response)
-               --if waiting then log("debug", "can't process_next, waiting"); return; end
-               if sessions[conn] and #pending > 0 then
+       local function process_next()
+               if waiting then log("debug", "can't process_next, waiting"); return; end
+               waiting = true;
+               while sessions[conn] and #pending > 0 do
                        local request = t_remove(pending);
                        --log("debug", "process_next: %s", request.path);
-                       waiting = true;
                        --handle_request(conn, request, process_next);
                        _1, _2, _3 = conn, request, process_next;
                        if not xpcall(_handle_request, _traceback_handler) then
                                conn:write("HTTP/1.0 500 Internal Server Error\r\n\r\n"..events.fire_event("http-error", { code = 500, private_message = last_err }));
                                conn:close();
                        end
-               else
-                       --log("debug", "ready for more");
-                       waiting = false;
                end
+               --log("debug", "ready for more");
+               waiting = false;
        end
        local function success_cb(request)
                --log("debug", "success_cb: %s", request.path);
+               if waiting then
+                       log("error", "http connection handler is not reentrant: %s", request.path);
+                       assert(false, "http connection handler is not reentrant");
+               end
                request.secure = secure;
                t_insert(pending, request);
-               if not waiting then
-                       process_next();
-               end
+               process_next();
        end
        local function error_cb(err)
                log("debug", "error_cb: %s", err or "<nil>");
@@ -158,12 +166,23 @@ function handle_request(conn, request, finish_cb)
 
        local date_header = os_date('!%a, %d %b %Y %H:%M:%S GMT'); -- FIXME use
        local conn_header = request.headers.connection;
-       local keep_alive = conn_header == "Keep-Alive" or (request.httpversion == "1.1" and conn_header ~= "close");
+       conn_header = conn_header and ","..conn_header:gsub("[ \t]", ""):lower().."," or ""
+       local httpversion = request.httpversion
+       local persistent = conn_header:find(",Keep-Alive,", 1, true)
+               or (httpversion == "1.1" and not conn_header:find(",close,", 1, true));
+
+       local response_conn_header;
+       if persistent then
+               response_conn_header = "Keep-Alive";
+       else
+               response_conn_header = httpversion == "1.1" and "close" or nil
+       end
 
        local response = {
                request = request;
                status_code = 200;
-               headers = { date = date_header, connection = (keep_alive and "Keep-Alive" or "close") };
+               headers = { date = date_header, connection = response_conn_header };
+               persistent = persistent;
                conn = conn;
                send = _M.send_response;
                finish_cb = finish_cb;
@@ -198,7 +217,7 @@ function handle_request(conn, request, finish_cb)
        local result = events.fire_event(event, payload);
        if result ~= nil then
                if result ~= true then
-                       local code, body = 200, "";
+                       local body;
                        local result_type = type(result);
                        if result_type == "number" then
                                response.status_code = result;
@@ -208,10 +227,14 @@ function handle_request(conn, request, finish_cb)
                        elseif result_type == "string" then
                                body = result;
                        elseif result_type == "table" then
-                               body = result.body;
-                               result.body = nil;
                                for k, v in pairs(result) do
-                                       response[k] = v;
+                                       if k ~= "headers" then
+                                               response[k] = v;
+                                       else
+                                               for header_name, header_value in pairs(v) do
+                                                       response.headers[header_name] = header_value;
+                                               end
+                                       end
                                end
                        end
                        response:send(body);
@@ -230,7 +253,7 @@ function _M.send_response(response, body)
        
        local status_line = "HTTP/"..response.request.httpversion.." "..(response.status or codes[response.status_code]);
        local headers = response.headers;
-       body = body or "";
+       body = body or response.body or "";
        headers.content_length = #body;
 
        local output = { status_line };
@@ -245,7 +268,7 @@ function _M.send_response(response, body)
                response:on_destroy();
                response.on_destroy = nil;
        end
-       if headers.connection == "Keep-Alive" then
+       if response.persistent then
                response:finish_cb();
        else
                response.conn:close();
@@ -270,6 +293,9 @@ end
 function _M.set_default_host(host)
        default_host = host;
 end
+function _M.fire_event(event, ...)
+       return events.fire_event(event, ...);
+end
 
 _M.listener = listener;
 _M.codes = codes;