]> matita.cs.unibo.it Git - helm.git/blobdiff - helm/DEVEL/ocaml-http/http_daemon.ml
- added main that starts a new http_daemon given a daemon_spec (see
[helm.git] / helm / DEVEL / ocaml-http / http_daemon.ml
index f29b4c1d16390084af88c6268611e10a13ee7610..1bbf83b20760080395bf73128fcbc0be31090797 100644 (file)
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *)
 
-open Printf;;
+open Printf
 
-open Http_common;;
-open Http_types;;
-open Http_constants;;
-open Http_parser;;
+open Http_common
+open Http_types
+open Http_constants
+open Http_parser
 
   (** send raw data on outchan, flushing it afterwards *)
 let send_raw ~data outchan =
@@ -130,8 +130,11 @@ let respond_forbidden ~url ?version outchan =
   (`Code 403) outchan
 
 let respond_unauthorized ?version ?(realm = server_string) outchan =
+  let body =
+    sprintf "401 - Unauthorized - Authentication failed for realm \"%s\"" realm
+  in
   respond ~headers:["WWW-Authenticate", sprintf "Basic realm=\"%s\"" realm]
-    ~code:(`Code 401) outchan
+    ~code:(`Code 401) ~body outchan
 
 let send_file ~src outchan =
   let buflen = 1024 in
@@ -303,56 +306,103 @@ let server_of_mode = function
   callbacks keep on living until the end or are them all killed immediatly?
   The right semantics should obviously be the first one *)
 
+let handle_manual_auth outchan f =
+  try
+    f ()
+  with
+  | Unauthorized realm -> respond_unauthorized ~realm outchan
+  | Again -> ()
+
+let handle_auth req spec outchan =
+  try
+    (match (spec.auth, req#authorization) with
+    | None, _ -> spec.callback req outchan  (* no auth required *)
+    | Some (realm, `Basic (spec_username, spec_password)),
+      Some (`Basic (username, password))
+      when (username = spec_username) && (password = spec_password) ->
+        (* auth ok *)
+        spec.callback req outchan
+    | Some (realm, _), _ -> raise (Unauthorized realm)) (* auth failure *)
+  with
+  | Unauthorized realm -> respond_unauthorized ~realm outchan
+  | Again -> ()
+
   (* TODO support also chroot to 'root', not only chdir *)
+  (* TODO deprecated: remove from future versions *)
   (* curried request *)
 let start
   ?(addr = default_addr) ?(port = default_port)
-  ?(timeout = Some default_timeout) ?(mode = default_mode) ?root callback
+  ?(timeout = default_timeout) ?(mode = default_mode) ?root callback
   =
+  Http_misc.warn
+    "Http_daemon.start is deprecated in favour of Http_daemon.main and will be removed in future versions of the library";
   chdir_to_document_root root;
   let sockaddr = Http_misc.build_sockaddr (addr, port) in
   let daemon_callback inchan outchan =
-    try
+    handle_manual_auth outchan (fun () ->
       let (path, parameters) = safe_parse_request inchan outchan in
       callback path parameters outchan;
-      flush outchan
-    with
-    | Unauthorized realm -> respond_unauthorized ~realm outchan
-    | Again -> ()
+      flush outchan);
   in
   try
     (server_of_mode mode) ~sockaddr ~timeout daemon_callback 
   with Quit -> ()
 
   (* OO request *)
+  (* TODO deprecated: remove from future versions *)
 let start'
   ?(addr = default_addr) ?(port = default_port)
-  ?(timeout = Some default_timeout) ?(mode = default_mode) ?root callback
-  =
+  ?(timeout = default_timeout) ?(mode = default_mode) ?root callback
+=
+  Http_misc.warn
+    "Http_daemon.start' is deprecated in favour of Http_daemon.main and will be removed in future versions of the library";
   chdir_to_document_root root;
   let sockaddr = Http_misc.build_sockaddr (addr, port) in
   let daemon_callback inchan outchan =
-    try
+    handle_manual_auth outchan (fun () ->
       let req = safe_parse_request' inchan outchan in
       callback req outchan;
-      flush outchan
-    with
-    | Unauthorized realm -> respond_unauthorized ~realm outchan
-    | Again -> ()
+      flush outchan)
   in
   try
     (server_of_mode mode) ~sockaddr ~timeout daemon_callback 
   with Quit -> ()
 
+let main spec =
+  chdir_to_document_root spec.root_dir;
+  let sockaddr = Http_misc.build_sockaddr (spec.address, spec.port) in
+  let daemon_callback inchan outchan =
+    try
+      let req = safe_parse_request' inchan outchan in
+      handle_auth req spec outchan;
+      flush outchan
+    with exn ->
+      (match spec.exn_handler with
+      | Some f ->
+          debug_print "uncaught exception: executing handler";
+          f exn outchan
+      | None ->
+          debug_print "uncaught exception but no handler given: re-raising";
+          raise exn)
+  in
+  try
+    (server_of_mode spec.mode) ~sockaddr ~timeout:spec.timeout daemon_callback 
+  with Quit -> ()
+
 module Trivial =
   struct
-    let callback path _ outchan =
-      if not (Pcre.pmatch ~rex:(Pcre.regexp "^/") path) then
+    let heading_slash_RE = Pcre.regexp "^/"
+
+    let trivial_callback req outchan =
+      let path = req#path in
+      if not (Pcre.pmatch ~rex:heading_slash_RE path) then
         respond_error ~code:(`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
+
+    let callback = trivial_callback
+
+    let main spec = main { spec with callback = trivial_callback }
   end
 
   (* @param inchan input channel connected to client
@@ -412,3 +462,34 @@ class daemon ?(addr = "0.0.0.0") ?(port = 80) () =
 
   end
 
+open Http_constants
+
+let default_spec = {
+  address = default_addr;
+  auth = default_auth;
+  callback = default_callback;
+  mode = default_mode;
+  port = default_port;
+  root_dir = default_root_dir;
+  exn_handler = default_exn_handler;
+  timeout = default_timeout;
+}
+
+let daemon_spec
+  ?(address = default_addr) ?(auth = default_auth)
+  ?(callback = default_callback) ?(mode = default_mode) ?(port = default_port)
+  ?(root_dir = default_root_dir) ?(exn_handler = default_exn_handler)
+  ?(timeout = default_timeout)
+  ()
+=
+  { default_spec with
+      address = address;
+      auth = auth;
+      callback = callback;
+      mode = mode;
+      port = port;
+      root_dir = root_dir;
+      exn_handler = exn_handler;
+      timeout = timeout;
+  }
+