ported to ocaml 3.08

[helm.git] / helm / DEVEL / ocaml-http / http_client.ml

diff --git a/helm/DEVEL/ocaml-http/http_client.ml b/helm/DEVEL/ocaml-http/http_client.ml
index e0f9d0f1158e61199089b1cccae4aff597047139..1a89bba8aa6d525dc0bd84d4b21723efe9ea15f2 100644 (file)
--- a/helm/DEVEL/ocaml-http/http_client.ml
+++ b/helm/DEVEL/ocaml-http/http_client.ml
@@ -1,6 +1,30 @@
 
+(*
+  OCaml HTTP - do it yourself (fully OCaml) HTTP daemon
+
+  Copyright (C) <2002-2004> Stefano Zacchiroli <zack@cs.unibo.it>
+
+  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
+
+exception Http_error of (int * string)  (* code, body *)
+
 let http_scheme_RE = Pcre.regexp ~flags:[`CASELESS] "^http://"
 let url_RE = Pcre.regexp "^([\\w.-]+)(:(\\d+))?(/.*)?$"
 
@@ -28,12 +52,33 @@ let init_socket addr port =
   let inchan = Unix.in_channel_of_descr suck in
   (inchan, outchan)
 
-let http_get_iter_buf ~callback url =
+let submit_request kind url =
   let (address, port, path) = parse_url url in
-  let buf = String.create tcp_bufsiz in
   let (inchan, outchan) = init_socket address port in
-  output_string outchan (sprintf "GET %s HTTP/1.0\r\n\r\n" path);
+  let req_string = match kind with `GET -> "GET" | `HEAD -> "HEAD" in
+  output_string outchan (sprintf "%s %s HTTP/1.0\r\n" req_string path);
+  output_string outchan (sprintf "Host: %s\r\n\r\n" address);
   flush outchan;
+  (inchan, outchan)
+
+let http_head url =
+  let (inchan, outchan) = submit_request `HEAD url in
+  let (_, status) = Http_parser.parse_response_fst_line inchan in
+  (match code_of_status status with
+  | 200 -> ()
+  | code -> raise (Http_error (code, "")));
+  let buf = Http_misc.buf_of_inchan inchan in
+  close_in inchan; (* close also outchan, same fd *)
+  Buffer.contents buf
+
+let http_get_iter callback url =
+  let (inchan, outchan) = submit_request `GET url in
+  let buf = String.create tcp_bufsiz in
+  let (_, status) = Http_parser.parse_response_fst_line inchan in
+  (match code_of_status status with
+  | 200 -> ()
+  | code -> raise (Http_error (code, "")));
+  ignore (Http_parser.parse_headers inchan);
   (try
     while true do
       match input inchan buf 0 tcp_bufsiz with
@@ -50,6 +95,6 @@ let http_get_iter_buf ~callback url =
 
 let http_get url =
   let buf = Buffer.create 10240 in
-  http_get_iter_buf ~callback:(Buffer.add_string buf) url;
+  http_get_iter (Buffer.add_string buf) url;
   Buffer.contents buf