plugins/mod_httpserver.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2009 Matthew Wild
   3 -- Copyright (C) 2008-2009 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10 local httpserver = require "net.httpserver";
  11
  12 local open = io.open;
  13 local t_concat = table.concat;
  14
  15 local http_base = config.get("*", "core", "http_path") or "www_files";
  16
  17 local response_400 = { status = "400 Bad Request", body = "<h1>Bad Request</h1>Sorry, we didn't understand your request :(" };
  18 local response_404 = { status = "404 Not Found", body = "<h1>Page Not Found</h1>Sorry, we couldn't find what you were looking for :(" };
  19
  20 local function preprocess_path(path)
  21         if path:sub(1,1) ~= "/" then
  22                 path = "/"..path;
  23         end
  24         local level = 0;
  25         for component in path:gmatch("([^/]+)/") do
  26                 if component == ".." then
  27                         level = level - 1;
  28                 elseif component ~= "." then
  29                         level = level + 1;
  30                 end
  31                 if level < 0 then
  32                         return nil;
  33                 end
  34         end
  35         return path;
  36 end
  37
  38 function serve_file(path)
  39         local f, err = open(http_base..path, "r");
  40         if not f then return response_404; end
  41         local data = f:read("*a");
  42         f:close();
  43         return data;
  44 end
  45
  46 local function handle_file_request(method, body, request)
  47         local path = preprocess_path(request.url.path);
  48         if not path then return response_400; end
  49         path = path:gsub("^/[^/]+", ""); -- Strip /files/
  50         return serve_file(path);
  51 end
  52
  53 local function handle_default_request(method, body, request)
  54         local path = preprocess_path(request.url.path);
  55         if not path then return response_400; end
  56         return serve_file(path);
  57 end
  58
  59 local ports = config.get(module.host, "core", "http_ports") or { 5280 };
  60 httpserver.set_default_handler(handle_default_request);
  61 httpserver.new_from_config(ports, handle_file_request, { base = "files" });