Merge pull request #724 from benoitc/fix-gh722

recv: if expected size < BufSize fallback to old behaviour
benoitc · Sep 15, 2023 · 745e795 · 745e795
2 parents abed0ae + 42ea010
commit 745e795
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/hackney_response.erl b/src/hackney_response.erl
@@ -350,8 +350,10 @@ recv(#client{transport=Transport, socket=Skt, recv_timeout=Timeout}) ->
 
 recv(#client{transport=Transport, socket=Skt, recv_timeout=Timeout}, {_BufSize, undefined}) ->
   Transport:recv(Skt, 0, Timeout);
-recv(#client{transport=Transport, socket=Skt, recv_timeout=Timeout}, {BufSize, ExpectedSize}) ->
-  Transport:recv(Skt, ExpectedSize - BufSize, Timeout).
+recv(#client{transport=Transport, socket=Skt, recv_timeout=Timeout}, {BufSize, ExpectedSize}) when ExpectedSize >= BufSize ->
+  Transport:recv(Skt, ExpectedSize - BufSize, Timeout);
+recv(#client{transport=Transport, socket=Skt, recv_timeout=Timeout}, {_BufSize, _ExpectedSize}) ->
+  Transport:recv(Skt, 0, Timeout).
 
 close(#client{socket=nil}=Client) ->
   Client#client{state = closed};