Remove option around client_addr

This commit is contained in:
Stéphane Lavergne 2023-08-08 15:18:39 -04:00
parent d5f783c159
commit 03596c1a08
4 changed files with 11 additions and 10 deletions

View file

@ -187,7 +187,7 @@ end
(** A TCP server abstraction. *)
module TCP_server = struct
type conn_handler = {
handle: ?client_addr:Unix.sockaddr -> Input.t -> Output.t -> unit; (** Handle client connection *)
handle: client_addr:Unix.sockaddr -> Input.t -> Output.t -> unit; (** Handle client connection *)
}
type t = {

View file

@ -164,7 +164,7 @@ module Request = struct
type 'body t = {
meth: Meth.t;
host: string;
client_addr: Unix.sockaddr option;
client_addr: Unix.sockaddr;
headers: Headers.t;
http_version: int * int;
path: string;
@ -246,7 +246,7 @@ module Request = struct
bad_reqf 400 "body is too short by %d bytes" size)
(* parse request, but not body (yet) *)
let parse_req_start ?client_addr ~get_time_s ~buf (bs : byte_stream) :
let parse_req_start ~client_addr ~get_time_s ~buf (bs : byte_stream) :
unit t option resp_result =
try
let line = Byte_stream.read_line ~buf bs in
@ -342,8 +342,8 @@ module Request = struct
| e -> bad_reqf 500 "failed to read body: %s" (Printexc.to_string e)
module Internal_ = struct
let parse_req_start ?(buf = Buf.create ()) ~get_time_s bs =
parse_req_start ~get_time_s ~buf bs |> unwrap_resp_result
let parse_req_start ?(buf = Buf.create ()) ~client_addr ~get_time_s bs =
parse_req_start ~client_addr ~get_time_s ~buf bs |> unwrap_resp_result
let parse_body ?(buf = Buf.create ()) req bs : _ t =
parse_body_ ~tr_stream:(fun s -> s) ~buf { req with body = bs }
@ -1017,7 +1017,7 @@ let find_map f l =
aux f l
(* handle client on [ic] and [oc] *)
let client_handle_for (self : t) ?client_addr ic oc : unit =
let client_handle_for (self : t) ~client_addr ic oc : unit =
Pool.with_resource self.buf_pool @@ fun buf ->
Pool.with_resource self.buf_pool @@ fun buf_res ->
let is = Byte_stream.of_input ~buf_size:self.buf_size ic in
@ -1025,7 +1025,7 @@ let client_handle_for (self : t) ?client_addr ic oc : unit =
while !continue && running self do
_debug (fun k -> k "read next request");
let (module B) = self.backend in
match Request.parse_req_start ?client_addr ~get_time_s:B.get_time_s ~buf is with
match Request.parse_req_start ~client_addr ~get_time_s:B.get_time_s ~buf is with
| Ok None -> continue := false (* client is done *)
| Error (c, s) ->
(* connection error, close *)

View file

@ -67,7 +67,7 @@ module Request : sig
meth: Meth.t; (** HTTP method for this request. *)
host: string;
(** Host header, mandatory. It can also be found in {!headers}. *)
client_addr : Unix.sockaddr option; (** Client address. Available since NEXT_RELEASE. *)
client_addr : Unix.sockaddr; (** Client address. Available since NEXT_RELEASE. *)
headers: Headers.t; (** List of headers. *)
http_version: int * int;
(** HTTP version. This should be either [1, 0] or [1, 1]. *)
@ -162,7 +162,7 @@ module Request : sig
(* for testing purpose, do not use. There is no guarantee of stability. *)
module Internal_ : sig
val parse_req_start :
?buf:buf -> get_time_s:(unit -> float) -> byte_stream -> unit t option
?buf:buf -> client_addr:Unix.sockaddr -> get_time_s:(unit -> float) -> byte_stream -> unit t option
val parse_body : ?buf:buf -> unit t -> byte_stream -> byte_stream t
end

View file

@ -10,7 +10,8 @@ let () =
salutationsSOMEJUNK"
in
let str = Tiny_httpd.Byte_stream.of_string q in
let r = Request.Internal_.parse_req_start ~get_time_s:(fun _ -> 0.) str in
let client_addr = Unix.(ADDR_INET (inet_addr_loopback, 1024)) in
let r = Request.Internal_.parse_req_start ~client_addr ~get_time_s:(fun _ -> 0.) str in
match r with
| None -> failwith "should parse"
| Some req ->