2 -- Copyright (C) 2008-2010 Matthew Wild
3 -- Copyright (C) 2008-2010 Waqas Hussain
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
9 module:depends("http");
10 local server = require"net.http.server";
11 local lfs = require "lfs";
13 local os_date = os.date;
15 local stat = lfs.attributes;
16 local build_path = require"socket.url".build_path;
17 local path_sep = package.config:sub(1,1);
19 local base_path = module:get_option_string("http_files_dir", module:get_option_string("http_path"));
20 local cache_size = module:get_option_number("http_files_cache_size", 128);
21 local cache_max_file_size = module:get_option_number("http_files_cache_max_file_size", 4096);
22 local dir_indices = module:get_option("http_index_files", { "index.html", "index.htm" });
23 local directory_index = module:get_option_boolean("http_dir_listing");
25 local mime_map = module:shared("/*/http_files/mime").types;
28 html = "text/html", htm = "text/html",
29 xml = "application/xml",
32 js = "application/javascript",
35 jpeg = "image/jpeg", jpg = "image/jpeg",
36 svg = "image/svg+xml",
38 module:shared("/*/http_files/mime").types = mime_map;
40 local mime_types, err = open(module:get_option_string("mime_types_file", "/etc/mime.types"),"r");
42 local mime_data = mime_types:read("*a");
44 setmetatable(mime_map, {
45 __index = function(t, ext)
46 local typ = mime_data:match("\n(%S+)[^\n]*%s"..(ext:lower()).."%s") or "application/octet-stream";
54 local forbidden_chars_pattern = "[/%z]";
55 if prosody.platform == "windows" then
56 forbidden_chars_pattern = "[/%z\001-\031\127\"*:<>?|]"
59 local urldecode = require "util.http".urldecode;
60 function sanitize_path(path)
61 if not path then return end
65 for component in path:gmatch("([^/]+)") do
66 component = urldecode(component);
67 if component:find(forbidden_chars_pattern) then
69 elseif component == ".." then
75 elseif component ~= "." then
80 if path:sub(-1,-1) == "/" then
83 return "/"..table.concat(out, "/");
86 local cache = require "util.cache".new(cache_size);
89 if type(opts) ~= "table" then -- assume path string
90 opts = { path = opts };
92 local base_path = opts.path;
93 local dir_indices = opts.index_files or dir_indices;
94 local directory_index = opts.directory_index;
95 local function serve_file(event, path)
96 local request, response = event.request, event.response;
97 local sanitized_path = sanitize_path(path);
98 if path and not sanitized_path then
101 path = sanitized_path;
102 local orig_path = sanitize_path(request.path);
103 local full_path = base_path .. (path or ""):gsub("/", path_sep);
104 local attr = stat(full_path:match("^.*[^\\/]")); -- Strip trailing path separator because Windows
109 local request_headers, response_headers = request.headers, response.headers;
111 local last_modified = os_date('!%a, %d %b %Y %H:%M:%S GMT', attr.modification);
112 response_headers.last_modified = last_modified;
114 local etag = ('"%02x-%x-%x-%x"'):format(attr.dev or 0, attr.ino or 0, attr.size or 0, attr.modification or 0);
115 response_headers.etag = etag;
117 local if_none_match = request_headers.if_none_match
118 local if_modified_since = request_headers.if_modified_since;
119 if etag == if_none_match
120 or (not if_none_match and last_modified == if_modified_since) then
124 local data = cache:get(orig_path);
125 if data and data.etag == etag then
126 response_headers.content_type = data.content_type;
128 elseif attr.mode == "directory" and path then
129 if full_path:sub(-1) ~= "/" then
130 local path = { is_absolute = true, is_directory = true };
131 for dir in orig_path:gmatch("[^/]+") do path[#path+1]=dir; end
132 response_headers.location = build_path(path);
135 for i=1,#dir_indices do
136 if stat(full_path..dir_indices[i], "mode") == "file" then
137 return serve_file(event, path..dir_indices[i]);
141 if directory_index then
142 data = server._events.fire_event("directory-index", { path = request.path, full_path = full_path });
147 cache[orig_path] = { data = data, content_type = mime_map.html; etag = etag; };
148 response_headers.content_type = mime_map.html;
151 local f, err = open(full_path, "rb");
153 module:log("debug", "Could not open %s. Error was %s", full_path, err);
156 local ext = full_path:match("%.([^./]+)$");
157 local content_type = ext and mime_map[ext];
158 response_headers.content_type = content_type;
159 if attr.size > cache_max_file_size then
160 response_headers.content_length = attr.size;
161 module:log("debug", "%d > cache_max_file_size", attr.size);
162 return response:send_file(f);
167 cache:set(orig_path, { data = data; content_type = content_type; etag = etag });
170 return response:send(data);
176 function wrap_route(routes)
177 for route,handler in pairs(routes) do
178 if type(handler) ~= "function" then
179 routes[route] = serve(handler);
186 module:provides("http", {
190 directory_index = directory_index;
195 module:log("debug", "http_files_dir not set, assuming use by some other module");