unikernel/duniverse/ocaml-h1/examples/lib/dune

@@ -0,0 +1,4 @@
+(library
+ (name        h1_examples)
+ (libraries   h1 base stdio)
+ (flags (:standard -safe-string)))

unikernel/duniverse/ocaml-h1/examples/lib/h1_examples.ml

@@ -0,0 +1,99 @@
+open Base
+open H1
+module Format = Stdlib.Format
+
+let print_string = Stdio.(Out_channel.output_string stdout)
+
+let text = "CHAPTER I. Down the Rabbit-Hole  Alice was beginning to get very tired of sitting by her sister on the bank, and of having nothing to do: once or twice she had peeped into the book her sister was reading, but it had no pictures or conversations in it, <and what is the use of a book,> thought Alice <without pictures or conversations?> So she was considering in her own mind (as well as she could, for the hot day made her feel very sleepy and stupid), whether the pleasure of making a daisy-chain would be worth the trouble of getting up and picking the daisies, when suddenly a White Rabbit with pink eyes ran close by her. There was nothing so very remarkable in that; nor did Alice think it so very much out of the way to hear the Rabbit say to itself, <Oh dear! Oh dear! I shall be late!> (when she thought it over afterwards, it occurred to her that she ought to have wondered at this, but at the time it all seemed quite natural); but when the Rabbit actually took a watch out of its waistcoat-pocket, and looked at it, and then hurried on, Alice started to her feet, for it flashed across her mind that she had never before seen a rabbit with either a waistcoat-pocket, or a watch to take out of it, and burning with curiosity, she ran across the field after it, and fortunately was just in time to see it pop down a large rabbit-hole under the hedge. In another moment down went Alice after it, never once considering how in the world she was to get out again. The rabbit-hole went straight on like a tunnel for some way, and then dipped suddenly down, so suddenly that Alice had not a moment to think about stopping herself before she found herself falling down a very deep well. Either the well was very deep, or she fell very slowly, for she had plenty of time as she went down to look about her and to wonder what was going to happen next. First, she tried to look down and make out what she was coming to, but it was too dark to see anything; then she looked at the sides of the well, and noticed that they were filled with cupboards......"
+
+let text = Bstr.of_string text
+
+module Client = struct
+ exception Response_error
+
+  let error_handler error =
+    let error =
+      match error with
+      | `Malformed_response err -> Format.sprintf "Malformed response: %s" err
+      | `Invalid_response_body_length _ -> "Invalid body length"
+      | `Exn exn -> Format.sprintf "Exn raised: %s" (Exn.to_string exn)
+    in
+    Format.eprintf "Error handling response: %s\n%!" error;
+  ;;
+
+  let print ~on_eof response response_body =
+    match response with
+    | { Response.status = `OK; _ } as response ->
+      Format.fprintf Format.std_formatter "%a\n%!" Response.pp_hum response;
+      let rec on_read bs ~off ~len =
+        Bstr.sub_string ~off ~len bs |> print_string;
+        Body.Reader.schedule_read response_body ~on_read ~on_eof
+      in
+      Body.Reader.schedule_read response_body ~on_read ~on_eof;
+    | response ->
+      Format.fprintf Format.err_formatter "%a\n%!" Response.pp_hum response;
+      Stdlib.exit 1
+  ;;
+end
+
+module Server = struct
+  let echo_post reqd =
+    match Reqd.request reqd  with
+    | { Request.meth = `POST; headers; _ } ->
+      let response =
+        let content_type =
+          match Headers.get headers "content-type" with
+          | None   -> "application/octet-stream"
+          | Some x -> x
+        in
+        Response.create ~headers:(Headers.of_list ["content-type", content_type; "connection", "close"]) `OK
+      in
+      let request_body  = Reqd.request_body reqd in
+      let response_body = Reqd.respond_with_streaming reqd response in
+      let rec on_read buffer ~off ~len =
+        Body.Writer.write_bigstring response_body buffer ~off ~len;
+        Body.Reader.schedule_read request_body ~on_eof ~on_read;
+      and on_eof () =
+        Body.Writer.close response_body
+      in
+      Body.Reader.schedule_read (Reqd.request_body reqd) ~on_eof ~on_read
+    | _ ->
+      let headers = Headers.of_list [ "connection", "close" ] in
+      Reqd.respond_with_string reqd (Response.create ~headers `Method_not_allowed) ""
+  ;;
+
+  let benchmark =
+    let headers = Headers.of_list ["content-length", Int.to_string (Bstr.length text)] in
+    let handler reqd =
+      let { Request.target; _ } = Reqd.request reqd in
+      let request_body          = Reqd.request_body reqd in
+      Body.Reader.close request_body;
+      match target with
+      | "/" -> Reqd.respond_with_bigstring reqd (Response.create ~headers `OK) text;
+      | _   -> Reqd.respond_with_string    reqd (Response.create `Not_found) "Route not found"
+    in
+    handler
+  ;;
+
+  let error_handler ?request:_ error start_response =
+    let response_body = start_response Headers.empty in
+    begin match error with
+    | `Exn exn ->
+      Body.Writer.write_string response_body (Exn.to_string exn);
+      Body.Writer.write_string response_body "\n";
+    | #Status.standard as error ->
+      Body.Writer.write_string response_body (Status.default_reason_phrase error)
+    end;
+    Body.Writer.close response_body
+  ;;
+
+  let upgrade reqd =
+    if Request.is_upgrade (Reqd.request reqd) then (
+      let headers = Headers.of_list [ "connection", "upgrade" ] in
+      Reqd.respond_with_upgrade reqd headers;
+    ) else (
+      let headers = Headers.of_list [ "connection", "close" ] in
+      Reqd.respond_with_string reqd (Response.create ~headers `Not_found) ""
+    )
+  ;;
+end

unikernel/duniverse/ocaml-h1/examples/lwt/dune

@@ -0,0 +1,9 @@
+(executables
+ (libraries h1 h1-lwt-unix h1_examples base stdio lwt lwt.unix)
+ (optional true)
+ (names lwt_get lwt_post lwt_echo_post lwt_echo_upgrade lwt_chunked))
+
+(alias
+ (name runtest)
+ (package h1-lwt-unix)
+ (deps (glob_files *.exe)))

unikernel/duniverse/ocaml-h1/examples/lwt/lwt_chunked.ml

@@ -0,0 +1,40 @@
+open Base
+open Lwt.Infix
+module Arg = Stdlib.Arg
+
+open H1_lwt_unix
+
+let request_handler (_ : Unix.sockaddr) reqd =
+  let body = H1.Reqd.respond_with_streaming reqd (H1.Response.create ~headers:(H1.Headers.of_list ["connection", "close"]) `OK) in
+  let rec respond_loop i =
+    H1.Body.Writer.write_string body (Printf.sprintf "Chunk %i\n" i);
+    H1.Body.Writer.flush_with_reason body (function
+    | `Closed  -> Stdio.print_endline "closed"
+    | `Written -> Stdio.print_endline "written"; Lwt.bind (Lwt_unix.sleep 5.) (fun () -> respond_loop (i+1)) |> ignore
+    );
+    Lwt.return_unit
+  in ignore (respond_loop 0)
+
+let error_handler (_ : Unix.sockaddr) = H1_examples.Server.error_handler
+
+let main port =
+  let listen_address = Unix.(ADDR_INET (inet_addr_loopback, port)) in
+  Lwt.async (fun () ->
+    Lwt_io.establish_server_with_client_socket
+      listen_address
+      (Server.create_connection_handler ~upgrade_handler:None ~request_handler ~error_handler)
+    >|= fun _server ->
+      Stdio.printf "Listening on port %i.\n" port);
+  let forever, _ = Lwt.wait () in
+  Lwt_main.run forever
+;;
+
+let () =
+  Stdlib.Sys.set_signal Stdlib.Sys.sigpipe Stdlib.Sys.Signal_ignore;
+  let port = ref 8080 in
+  Arg.parse
+    ["-p", Arg.Set_int port, " Listening port number (8080 by default)"]
+    ignore
+    "Echoes POST requests. Runs forever.";
+  main !port
+;;

unikernel/duniverse/ocaml-h1/examples/lwt/lwt_echo_post.ml

@@ -0,0 +1,36 @@
+open Base
+open Lwt.Infix
+module Arg = Stdlib.Arg
+
+open H1_lwt_unix
+
+let request_handler (_ : Unix.sockaddr) = H1_examples.Server.echo_post
+let error_handler (_ : Unix.sockaddr) = H1_examples.Server.error_handler
+
+let main port =
+  let listen_address = Unix.(ADDR_INET (inet_addr_loopback, port)) in
+  Lwt.async (fun () ->
+    Lwt_io.establish_server_with_client_socket
+      listen_address
+      (Server.create_connection_handler
+         ~request_handler
+         ~error_handler
+         ~upgrade_handler:None)
+    >|= fun _server ->
+      Stdio.printf "Listening on port %i and echoing POST requests.\n" port;
+      Stdio.printf "To send a POST request, try one of the following\n\n";
+      Stdio.printf "  echo \"Testing echo POST\" | dune exec examples/async/async_post.exe\n";
+      Stdio.printf "  echo \"Testing echo POST\" | dune exec examples/lwt/lwt_post.exe\n";
+      Stdio.printf "  echo \"Testing echo POST\" | curl -XPOST --data @- http://localhost:%d\n\n%!" port);
+  let forever, _ = Lwt.wait () in
+  Lwt_main.run forever
+;;
+
+let () =
+  let port = ref 8080 in
+  Arg.parse
+    ["-p", Arg.Set_int port, " Listening port number (8080 by default)"]
+    ignore
+    "Echoes POST requests. Runs forever.";
+  main !port
+;;

unikernel/duniverse/ocaml-h1/examples/lwt/lwt_echo_upgrade.ml

@@ -0,0 +1,47 @@
+open Base
+open Lwt.Infix
+module Arg = Stdlib.Arg
+
+open H1_lwt_unix
+
+let request_handler (_ : Unix.sockaddr) = H1_examples.Server.upgrade
+let error_handler (_ : Unix.sockaddr) = H1_examples.Server.error_handler
+
+let upgrade_handler (_ : Unix.sockaddr) (fd : Lwt_unix.file_descr) =
+  let input = Lwt_io.of_fd fd ~mode:Input in
+  let output = Lwt_io.of_fd fd ~mode:Output in
+  let rec loop () =
+    Lwt_io.read input ~count:4096
+    >>= function
+    | "" -> Lwt.return_unit
+    | data -> Lwt_io.write output data >>= loop
+  in
+  loop ()
+;;
+
+let main port =
+  let listen_address = Unix.(ADDR_INET (inet_addr_loopback, port)) in
+  Lwt.async (fun () ->
+    Lwt_io.establish_server_with_client_socket
+      listen_address
+      (Server.create_connection_handler
+         ~request_handler
+         ~error_handler
+         ~upgrade_handler:(Some upgrade_handler))
+    >|= fun _server ->
+      Stdio.printf "Listening on port %i, upgrading, and echoing data.\n" port;
+      Stdio.printf "To send an interactive upgrade request, try\n\n";
+      Stdio.printf "  examples/script/upgrade-connect\n%!");
+  let forever, _ = Lwt.wait () in
+  Lwt_main.run forever
+;;
+
+let () =
+  let port = ref 8080 in
+  Arg.parse
+    ["-p", Arg.Set_int port, " Listening port number (8080 by default)"]
+    ignore
+    "Echoes POST requests. Runs forever.";
+  main !port
+;;
+

unikernel/duniverse/ocaml-h1/examples/lwt/lwt_get.ml

@@ -0,0 +1,43 @@
+open Base
+open Lwt.Infix
+module Arg = Stdlib.Arg
+
+open H1
+open H1_lwt_unix
+
+let main port host =
+  Lwt_unix.getaddrinfo host (Int.to_string port) [Unix.(AI_FAMILY PF_INET)]
+  >>= fun addresses ->
+  let socket = Lwt_unix.socket Unix.PF_INET Unix.SOCK_STREAM 0 in
+  Lwt_unix.connect socket (List.hd_exn addresses).Unix.ai_addr
+  >>= fun () ->
+  let finished, notify_finished = Lwt.wait () in
+  let response_handler =
+    H1_examples.Client.print ~on_eof:(Lwt.wakeup_later notify_finished)
+  in
+  let headers = Httpun_types.Headers.of_list [ "host", host ] in
+  let request_body =
+    Client.request
+      ~error_handler:H1_examples.Client.error_handler
+      ~response_handler
+      socket
+      (Request.create ~headers `GET "/")
+  in
+  Body.Writer.close request_body;
+  finished
+;;
+
+let () =
+  let host = ref None in
+  let port = ref 80 in
+  Arg.parse
+    ["-p", Set_int port, " Port number (80 by default)"]
+    (fun host_argument -> host := Some host_argument)
+    "lwt_get.exe [-p N] HOST";
+  let host =
+    match !host with
+    | None -> failwith "No hostname provided"
+    | Some host -> host
+  in
+  Lwt_main.run (main !port host)
+;;

unikernel/duniverse/ocaml-h1/examples/lwt/lwt_post.ml

@@ -0,0 +1,53 @@
+open Base
+open Lwt.Infix
+module Arg = Stdlib.Arg
+
+open H1
+open H1_lwt_unix
+
+let main port host =
+  Lwt_io.(read stdin)
+  >>= fun body ->
+  Lwt_unix.getaddrinfo host (Int.to_string port) [Unix.(AI_FAMILY PF_INET)]
+  >>= fun addresses ->
+  let socket = Lwt_unix.socket Unix.PF_INET Unix.SOCK_STREAM 0 in
+  Lwt_unix.connect socket (List.hd_exn addresses).Unix.ai_addr
+  >>= fun () ->
+  let finished, notify_finished = Lwt.wait () in
+  let response_handler =
+    H1_examples.Client.print ~on_eof:(Lwt.wakeup_later notify_finished)
+  in
+  let headers =
+    Httpun_types.Headers.of_list
+    [ "content-length"   , (Int.to_string (String.length body))
+    ; "connection"       , "close"
+    ; "host"             , host
+    ]
+  in
+  let request_body =
+    Client.request
+      ~error_handler:H1_examples.Client.error_handler
+      ~response_handler
+      socket
+      (Request.create ~headers `POST "/")
+  in
+  Body.Writer.write_string request_body body;
+  Body.Writer.close request_body;
+  finished
+;;
+
+let () =
+  let host = ref None in
+  let port = ref 8080 in
+
+  Arg.parse
+    ["-p", Set_int port, " Port number (8080 by default)"]
+    (fun host_argument -> host := Some host_argument)
+    "lwt_get.exe [-p N] HOST";
+  let host =
+    match !host with
+    | None -> failwith "No hostname provided"
+    | Some host -> host
+  in
+  Lwt_main.run (main !port host)
+;;

unikernel/duniverse/ocaml-h1/examples/script/upgrade-connect

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+function headers {
+  printf "\
+GET / HTTP/1.1\r
+Host: localhost\r
+Connection: upgrade\r
+\r
+"
+}
+
+( headers; echo hello; cat; echo bye ) | nc localhost 8080 --close