client: fix infinite loop when client handshake disconnects (#73)

anmonteiro · Sep 2, 2024 · d8318ad · d8318ad
1 parent 9aacc12
commit d8318ad
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/client_connection.ml b/lib/client_connection.ml
@@ -129,7 +129,7 @@ let read t bs ~off ~len =
 
 let read_eof t bs ~off ~len =
   match t.state with
-  | Handshake handshake -> Client_handshake.read handshake bs ~off ~len
+  | Handshake handshake -> Client_handshake.read_eof handshake bs ~off ~len
   | Websocket websocket -> Websocket_connection.read_eof websocket bs ~off ~len
 
 let next_write_operation t =

diff --git a/lib/client_handshake.ml b/lib/client_handshake.ml
@@ -35,6 +35,9 @@ let next_write_operation t =
 let read t =
   Httpun.Client_connection.read t.connection
 
+let read_eof t =
+  Httpun.Client_connection.read_eof t.connection
+
 let yield_reader t =
   Httpun.Client_connection.yield_reader t.connection