(* OCaml HTTP - do it yourself (fully OCaml) HTTP daemon Copyright (C) <2002> Stefano Zacchiroli 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 2 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, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA *) open Printf;; open Http_common;; open Http_parser;; let debug = true let debug_print str = if debug then begin prerr_endline ("DEBUG: " ^ str); flush stderr end let default_addr = "0.0.0.0" let default_port = 80 let default_timeout = 300 let default_mode = `Fork (** send raw data on outchan, flushing it afterwards *) let send_raw ~data outchan = output_string outchan data; flush outchan let send_CRLF = send_raw ~data:crlf (** TODO perform some sanity test on header and value *) let send_header ~header ~value = send_raw ~data:(header ^ ": " ^ value ^ crlf) let send_headers ~headers outchan = List.iter (fun (header, value) -> send_header ~header ~value outchan) headers (** internal: parse a code argument from a function which have two optional arguments "code" and "status" *) let get_code_argument func_name = fun ~code ~status -> (match code, status with | Some c, None -> c | None, Some s -> code_of_status s | Some _, Some _ -> failwith (func_name ^ " you must give 'code' or 'status', not both") | None, None -> failwith (func_name ^ " you must give 'code' or 'status', not none")) (** internal: low level for send_status_line *) let send_status_line' ~version ~code = let status_line = String.concat " " [ string_of_version version; string_of_int code; reason_phrase_of_code code ] in send_raw ~data:(status_line ^ crlf) let send_status_line ?(version = http_version) ?code ?status outchan = send_status_line' ~version ~code:(get_code_argument "Daemon.send_status_line" ~code ~status) outchan (* FIXME duplication of code between this and response#addBasicHeaders *) let send_basic_headers ?(version = http_version) ?code ?status outchan = send_status_line' ~version ~code:(get_code_argument "Daemon.send_basic_headers" ~code ~status) outchan; send_headers ~headers:["Date", Http_misc.date_822 (); "Server", server_string] outchan (** internal: given a status code and an additional body return a string representing an HTML document that explains the meaning of given status code. Additional data can be added to the body via 'body' argument *) let foo_body code body = let reason_phrase = reason_phrase_of_code code in sprintf " %d %s

%d - %s

%s " code reason_phrase code reason_phrase body (** internal: send a fooish body explaining in HTML form the 'reason phrase' of an HTTP response; body, if given, will be appended to the body *) let send_foo_body code body = send_raw ~data:(foo_body code body) (* TODO add the computation of Content-Length header *) let respond ?(body = "") ?(headers = []) ?(version = http_version) ?(code = 200) ?status outchan = let code = match status with | None -> code | Some s -> code_of_status s in send_basic_headers ~version ~code outchan; send_headers ~headers outchan; send_CRLF outchan; send_raw ~data:body outchan (** internal: low level for respond_redirect, respond_error, ... This function send a status line corresponding to a given code, some basic headers, the additional headers (if given) and an HTML page containing the reason phrase; if body is given it will be included in the body of the HTML page *) let send_empty_response func_name ?(is_valid_status = fun _ -> true) ?(headers = []) ~body () = fun ?(version = http_version) ?code ?status outchan -> let code = get_code_argument func_name ~code ~status in if not (is_valid_status code) then failwith (sprintf "'%d' isn't a valid status code for %s" code func_name) else begin (* status code suitable for answering *) let headers = [ "Connection", "close"; "Content-Type", "text/html; charset=iso-8859-1" ] @ headers in let body = (foo_body code body) ^ body in respond ~version ~code ~headers ~body outchan (* (* OLD VERSION, now use 'respond' function *) send_basic_headers ~version ~code outchan; send_header ~header:"Connection" ~value:"close" outchan; send_header ~header:"Content-Type" ~value:"text/html; charset=iso-8859-1" outchan; send_headers ~headers outchan; send_CRLF outchan; send_foo_body ~code ~body outchan *) end (* TODO sanity tests on location *) let respond_redirect ~location ?(body = "") ?(version = http_version) ?(code = 301) ?status outchan = let code = match status with | None -> code | Some (s: Http_types.redirection_status) -> code_of_status s in send_empty_response "Daemon.respond_redirect" ~is_valid_status:is_redirection ~headers:["Location", location] ~body () ~version ~code outchan let respond_error ?(body = "") ?(version = http_version) ?(code = 400) ?status outchan = let code = match status with | None -> code | Some s -> code_of_status s in send_empty_response "Daemon.respond_error" ~is_valid_status:is_error ~body () ~version ~code outchan let respond_not_found ~url ?(version = http_version) outchan = send_empty_response "Daemon.respond_not_found" ~body:"" () ~version ~code:404 outchan let respond_forbidden ~url ?(version = http_version) outchan = send_empty_response "Daemon.respond_permission_denied" ~body:"" () ~version ~code:403 outchan let send_file ?name ?file outchan = let buflen = 1024 in let buf = String.make buflen ' ' in let (file, cleanup) = (match (name, file) with | Some n, None -> (* if we open the file, we close it before returning *) let f = open_in n in f, (fun () -> close_in f) | None, Some f -> (f, (fun () -> ())) | _ -> failwith "Daemon.send_file: either name or file must be given") in try while true do let bytes = input file buf 0 buflen in if bytes = 0 then raise End_of_file else output outchan buf 0 bytes done; assert false with End_of_file -> begin flush outchan; cleanup () end (* TODO interface is too ugly to advertise this function in .mli *) (** create a minimal HTML directory listing of a given directory and send it over an out_channel, directory is passed as a dir_handle; name is the directory name, used for pretty printing purposes; path is the opened dir path, used to test its contents with stat *) let send_dir_listing ~dir ~name ~path outchan = fprintf outchan "\n%s\n\n" name; let (dirs, files) = List.partition (fun e -> Http_misc.is_directory (path ^ e)) (Http_misc.ls dir) in List.iter (fun d -> fprintf outchan "%s/
\n" d d) (List.sort compare dirs); List.iter (fun f -> fprintf outchan "%s
\n" f f) (List.sort compare files); fprintf outchan "\n"; flush outchan let respond_file ~fname ?(version = http_version) outchan = (** ASSUMPTION: 'fname' doesn't begin with a "/"; it's relative to the current document root (usually the daemon's cwd) *) let droot = Sys.getcwd () in (* document root *) let path = droot ^ "/" ^ fname in (* full path to the desired file *) if not (Sys.file_exists path) then (* file not found *) respond_not_found ~url:fname outchan else begin try if Http_misc.is_directory path then begin (* file found, is a dir *) let dir = Unix.opendir path in send_basic_headers ~version ~code:200 outchan; send_header "Content-Type" "text/html" outchan; send_CRLF outchan; send_dir_listing ~dir ~name:fname ~path outchan; Unix.closedir dir end else begin (* file found, is something else *) let file = open_in fname in send_basic_headers ~version ~code:200 outchan; send_header ~header:"Content-Length" ~value:(string_of_int (Http_misc.filesize fname)) outchan; send_CRLF outchan; send_file ~file outchan; close_in file end with | Unix.Unix_error (Unix.EACCES, s, _) when (s = fname) -> respond_forbidden ~url:fname ~version outchan | Sys_error s when (Pcre.pmatch ~rex:(Pcre.regexp (fname ^ ": Permission denied")) s) -> respond_forbidden ~url:fname ~version outchan end let respond_with (res: Http_types.response) outchan = res#serialize outchan; flush outchan (* TODO support also chroot to 'root', not only chdir *) (* curried request *) let start ?(addr = default_addr) ?(port = default_port) ?(timeout = Some default_timeout) ?(mode = default_mode) ?root callback = (match root with (* chdir to document root *) | Some dir -> Sys.chdir dir | None -> ()); let sockaddr = Unix.ADDR_INET (Unix.inet_addr_of_string addr, port) in let daemon_callback inchan outchan = try let (path, parameters) = Http_parser.parse_request inchan in callback path parameters outchan; flush outchan with | End_of_file -> respond_error ~code:400 ~body:"Unexpected End Of File" outchan | Malformed_request req -> respond_error ~code:400 ~body:( "request 1st line format should be: ' '" ^ "
\nwhile received request 1st line was:
\n" ^ req) outchan | Unsupported_method meth -> respond_error ~code:501 ~body:("Method '" ^ meth ^ "' isn't supported (yet)") outchan | Malformed_request_URI uri -> respond_error ~code:400 ~body:("Malformed URL: '" ^ uri ^ "'") outchan | Unsupported_HTTP_version version -> respond_error ~code:505 ~body:("HTTP version '" ^ version ^ "' isn't supported (yet)") outchan | Malformed_query query -> respond_error ~code:400 ~body:(sprintf "Malformed query string '%s'" query) outchan | Malformed_query_part (binding, query) -> respond_error ~code:400 ~body:( sprintf "Malformed query part '%s' in query '%s'" binding query) outchan in match mode with | `Single -> Http_tcp_server.simple ~sockaddr ~timeout daemon_callback | `Fork -> Http_tcp_server.ocaml_builtin ~sockaddr ~timeout daemon_callback | `Thread -> Http_tcp_server.thread ~sockaddr ~timeout daemon_callback (* OO request *) let start' ?(addr = default_addr) ?(port = default_port) ?(timeout = Some default_timeout) ?(mode = default_mode) ?root callback = let wrapper path params outchan = let req = new Http_request.request ~path ~params in callback req outchan in match root with | None -> start ~addr ~port ~timeout ~mode wrapper | Some root -> start ~addr ~port ~timeout ~mode ~root wrapper module Trivial = struct let callback path _ outchan = if not (Pcre.pmatch ~rex:(Pcre.regexp "^/") path) then respond_error ~code:400 outchan else respond_file ~fname:(Http_misc.strip_heading_slash path) outchan let start ?(addr = default_addr) ?(port = default_port) () = start ~addr ~port callback end