%%% coding: latin-1
%%% ----------------------------------------------------------------------------
%%% Copyright (c) 2009, Erlang Training and Consulting Ltd.
%%% All rights reserved.
%%%
%%% Redistribution and use in source and binary forms, with or without
%%% modification, are permitted provided that the following conditions are met:
%%%    * Redistributions of source code must retain the above copyright
%%%      notice, this list of conditions and the following disclaimer.
%%%    * Redistributions in binary form must reproduce the above copyright
%%%      notice, this list of conditions and the following disclaimer in the
%%%      documentation and/or other materials provided with the distribution.
%%%    * Neither the name of Erlang Training and Consulting Ltd. nor the
%%%      names of its contributors may be used to endorse or promote products
%%%      derived from this software without specific prior written permission.
%%%
%%% THIS SOFTWARE IS PROVIDED BY Erlang Training and Consulting Ltd. ''AS IS''
%%% AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
%%% IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
%%% ARE DISCLAIMED. IN NO EVENT SHALL Erlang Training and Consulting Ltd. BE
%%% LIABLE SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
%%% BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
%%% WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
%%% OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
%%% ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
%%% ----------------------------------------------------------------------------

%%% @private
%%% @author Oscar Hellstrom <oscar@hellstrom.st>
%%% @doc
%%% This module implements the HTTP request handling. This should normally
%%% not be called directly since it should be spawned by the lhttpc module.
-module(lhttpc_client).

%% Throw is used correctly in this module as a nonlocal return within a large function
-elvis([{elvis_style, no_throw, disable}]).

-export([request/1]).

%% @todo Remove this once https://github.com/inaka/elvis_core/issues/170 is fixed.
-elvis([{elvis_style, atom_naming_convention, disable}]).

-record(client_state,
        {req_id :: term(),
         host :: string(),
         port = 80 :: integer(),
         ssl = false :: true | false,
         method :: string(),
         request :: iolist() | undefined,
         request_headers :: lhttpc:headers(),
         socket,
         connect_timeout = infinity :: timeout(),
         connect_options = [] :: [any()],
         attempts :: integer(),
         requester :: pid(),
         partial_upload = false :: true | false,
         chunked_upload = false :: true | false,
         upload_window :: non_neg_integer() | infinity,
         partial_download = false :: true | false,
         download_window = infinity :: timeout(),
         part_size ::
             non_neg_integer() |
             infinity}).        %% in case of infinity we read whatever data we can get from
                                %% the wire at that point or in case of chunked one chunk

-define(CONNECTION_HDR(HDRS, DEFAULT),
        string:to_lower(
            lhttpc_lib:header_value("connection", HDRS, DEFAULT))).

-spec request(#{id := term(),
                from := pid(),
                host := string(),
                port := 1..65535,
                ssl := true | false,
                path := string(),
                method := string() | atom(),
                headers := lhttpc:headers(),
                body := iolist(),
                options := lhttpc:options()}) ->
                 no_return().
request(#{id := ReqId, from := From} = Arg) ->
    Result =
        try
            execute(Arg)
        catch
            Reason ->
                {response, ReqId, self(), {error, Reason}};
            error:closed ->
                {response, ReqId, self(), {error, connection_closed}};
            error:Error:Stack ->
                {exit, ReqId, self(), {Error, Stack}}
        end,
    case Result of
        {response, _, _, {ok, {no_return, _}}} ->
            ok;
        _Else ->
            From ! Result
    end,
    % Don't send back {'EXIT', Pid, normal} if the process
    % calling us is trapping exits
    unlink(From),
    ok.

execute(#{id := ReqId,
          from := From,
          host := Host,
          port := Port,
          ssl := SSL,
          path := Path,
          method := Method,
          headers := Hdrs,
          body := Body,
          options := Options}) ->
    UploadWindowSize = proplists:get_value(partial_upload, Options),
    PartialUpload = proplists:is_defined(partial_upload, Options),
    PartialDownload = proplists:is_defined(partial_download, Options),
    PartialDownloadOptions = proplists:get_value(partial_download, Options, []),
    NormalizedMethod = lhttpc_lib:normalize_method(Method),
    MaxConnections = proplists:get_value(max_connections, Options, 10),
    ConnectionTimeout = proplists:get_value(connection_timeout, Options, infinity),
    {ChunkedUpload, Request} =
        lhttpc_lib:format_request(Path, NormalizedMethod, Hdrs, Host, Port, Body, PartialUpload),
    Socket =
        case lhttpc_lb:checkout(Host, Port, SSL, MaxConnections, ConnectionTimeout) of
            {ok, S} ->
                S; % Re-using HTTP/1.1 connections
            retry_later ->
                throw(retry_later);
            no_socket ->
                undefined % Opening a new HTTP/1.1 connection
        end,
    State =
        #client_state{req_id = ReqId,
                      host = Host,
                      port = Port,
                      ssl = SSL,
                      method = NormalizedMethod,
                      request = Request,
                      requester = From,
                      request_headers = Hdrs,
                      socket = Socket,
                      connect_timeout = proplists:get_value(connect_timeout, Options, infinity),
                      connect_options = proplists:get_value(connect_options, Options, []),
                      attempts = 1 + proplists:get_value(send_retry, Options, 1),
                      partial_upload = PartialUpload,
                      upload_window = UploadWindowSize,
                      chunked_upload = ChunkedUpload,
                      partial_download = PartialDownload,
                      download_window =
                          proplists:get_value(window_size, PartialDownloadOptions, infinity),
                      part_size = proplists:get_value(part_size, PartialDownloadOptions, infinity)},
    Response =
        case send_request(State) of
            {R, undefined} ->
                {ok, R};
            {R, NewSocket} ->
                % The socket we ended up doing the request over is returned
                % here, it might be the same as Socket, but we don't know.
                lhttpc_lb:checkin(Host, Port, SSL, NewSocket),
                {ok, R}
        end,
    {response, ReqId, self(), Response}.

send_request(#client_state{attempts = 0}) ->
    % Don't try again if the number of allowed attempts is 0.
    throw(connection_closed);
send_request(#client_state{socket = undefined} = State) ->
    Host = State#client_state.host,
    Port = State#client_state.port,
    SSL = State#client_state.ssl,
    Timeout = State#client_state.connect_timeout,
    ConnectOptions = State#client_state.connect_options,
    SocketOptions = [binary, {packet, http}, {active, false} | ConnectOptions],
    case lhttpc_sock:connect(Host, Port, SocketOptions, Timeout, SSL) of
        {ok, Socket} ->
            send_request(State#client_state{socket = Socket});
        {error, etimedout} ->
            % TCP stack decided to give up
            throw(connect_timeout);
        {error, timeout} ->
            throw(connect_timeout);
        {error, Reason} ->
            erlang:error(Reason)
    end;
send_request(State) ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    Request = State#client_state.request,
    case lhttpc_sock:send(Socket, Request, SSL) of
        ok ->
            if State#client_state.partial_upload ->
                   partial_upload(State);
               not State#client_state.partial_upload ->
                   read_response(State)
            end;
        {error, closed} ->
            lhttpc_sock:close(Socket, SSL),
            NewState =
                State#client_state{socket = undefined, attempts = State#client_state.attempts - 1},
            send_request(NewState);
        {error, Reason} ->
            lhttpc_sock:close(Socket, SSL),
            erlang:error(Reason)
    end.

partial_upload(State) ->
    Response = {ok, {self(), State#client_state.upload_window}},
    State#client_state.requester ! {response, State#client_state.req_id, self(), Response},
    partial_upload_loop(State#client_state{attempts = 1, request = undefined}).

partial_upload_loop(#client_state{requester = Pid} = State) ->
    receive
        {trailers, Pid, Trailers} ->
            send_trailers(State, Trailers),
            read_response(State);
        {body_part, Pid, http_eob} ->
            send_body_part(State, http_eob),
            read_response(State);
        {body_part, Pid, Data} ->
            send_body_part(State, Data),
            Pid ! {ack, self()},
            partial_upload_loop(State)
    end.

send_body_part(#client_state{socket = Socket, ssl = SSL} = State, BodyPart) ->
    Data = encode_body_part(State, BodyPart),
    check_send_result(State, lhttpc_sock:send(Socket, Data, SSL)).

send_trailers(#client_state{chunked_upload = true} = State, Trailers) ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    Data = [<<"0\r\n">>, lhttpc_lib:format_hdrs(Trailers)],
    check_send_result(State, lhttpc_sock:send(Socket, Data, SSL));
send_trailers(#client_state{chunked_upload = false}, _Trailers) ->
    erlang:error(trailers_not_allowed).

encode_body_part(#client_state{chunked_upload = true}, http_eob) ->
    <<"0\r\n\r\n">>; % We don't send trailers after http_eob
encode_body_part(#client_state{chunked_upload = false}, http_eob) ->
    <<>>;
encode_body_part(#client_state{chunked_upload = true}, Data) ->
    Size = integer_to_binary(iolist_size(Data), 16),
    [Size, <<"\r\n">>, Data, <<"\r\n">>];
encode_body_part(#client_state{chunked_upload = false}, Data) ->
    Data.

check_send_result(_State, ok) ->
    ok;
check_send_result(#client_state{socket = Sock, ssl = SSL}, {error, Reason}) ->
    lhttpc_sock:close(Sock, SSL),
    throw(Reason).

read_response(#client_state{socket = Socket, ssl = SSL} = State) ->
    lhttpc_sock:setopts(Socket, [{packet, http}], SSL),
    read_response(State, nil, {nil, nil}, []).

read_response(State, Vsn, {StatusCode, _} = Status, Hdrs) ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    case lhttpc_sock:recv(Socket, SSL) of
        {ok, {http_response, NewVsn, NewStatusCode, Reason}} ->
            NewStatus = {NewStatusCode, Reason},
            read_response(State, NewVsn, NewStatus, Hdrs);
        {ok, {http_header, _, Name, _, Value}} ->
            Header = {lhttpc_lib:maybe_atom_to_list(Name), Value},
            read_response(State, Vsn, Status, [Header | Hdrs]);
        {ok, http_eoh} when StatusCode >= 100, StatusCode =< 199 ->
            % RFC 2616, section 10.1:
            % A client MUST be prepared to accept one or more
            % 1xx status responses prior to a regular
            % response, even if the client does not expect a
            % 100 (Continue) status message. Unexpected 1xx
            % status responses MAY be ignored by a user agent.
            read_response(State, nil, {nil, nil}, []);
        {ok, http_eoh} ->
            lhttpc_sock:setopts(Socket, [{packet, raw}], SSL),
            Response = handle_response_body(State, Vsn, Status, Hdrs),
            NewHdrs = element(2, Response),
            ReqHdrs = State#client_state.request_headers,
            NewSocket = maybe_close_socket(Socket, SSL, Vsn, ReqHdrs, NewHdrs),
            {Response, NewSocket};
        {error, closed} ->
            % Either we only noticed that the socket was closed after we
            % sent the request, the server closed it just after we put
            % the request on the wire or the server has some issues and is
            % closing connections without sending responses.
            % If this the first attempt to send the request, we will try again.
            lhttpc_sock:close(Socket, SSL),
            NewState =
                State#client_state{socket = undefined, attempts = State#client_state.attempts - 1},
            send_request(NewState);
        {error, timeout} ->
            lhttpc_sock:close(Socket, SSL),
            NewState = State#client_state{socket = undefined, attempts = 0},
            send_request(NewState)
    end.

handle_response_body(#client_state{partial_download = false} = State,
                     Vsn,
                     Status,
                     Hdrs) ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    Method = State#client_state.method,
    {Body, NewHdrs} =
        case has_body(Method, element(1, Status), Hdrs) of
            true ->
                read_body(Vsn, Hdrs, SSL, Socket, body_type(Hdrs));
            false ->
                {<<>>, Hdrs}
        end,
    {Status, NewHdrs, Body};
handle_response_body(#client_state{partial_download = true} = State, Vsn, Status, Hdrs) ->
    Method = State#client_state.method,
    case has_body(Method, element(1, Status), Hdrs) of
        true ->
            Response = {ok, {Status, Hdrs, self()}},
            State#client_state.requester ! {response, State#client_state.req_id, self(), Response},
            MonRef = erlang:monitor(process, State#client_state.requester),
            Res = read_partial_body(State, Vsn, Hdrs, body_type(Hdrs)),
            erlang:demonitor(MonRef, [flush]),
            Res;
        false ->
            {Status, Hdrs, undefined}
    end.

has_body("HEAD", _, _) ->
    % HEAD responses aren't allowed to include a body
    false;
has_body("OPTIONS", _, Hdrs) ->
    % OPTIONS can include a body, if Content-Length or Transfer-Encoding
    % indicates it
    ContentLength = lhttpc_lib:header_value("content-length", Hdrs),
    TransferEncoding = lhttpc_lib:header_value("transfer-encoding", Hdrs),
    case {ContentLength, TransferEncoding} of
        {undefined, undefined} ->
            false;
        {_, _} ->
            true
    end;
has_body(_, 204, _) ->
    false; % RFC 2616 10.2.5: 204 No Content
has_body(_, 304, _) ->
    false; % RFC 2616 10.3.5: 304 Not Modified
has_body(_, _, _) ->
    true. % All other responses are assumed to have a body

body_type(Hdrs) ->
    % Find out how to read the entity body from the request.
    % * If we have a Content-Length, just use that and read the complete
    %   entity.
    % * If Transfer-Encoding is set to chunked, we should read one chunk at
    %   the time
    % * If neither of this is true, we need to read until the socket is
    %   closed (AFAIK, this was common in versions before 1.1).
    case lhttpc_lib:header_value("content-length", Hdrs) of
        undefined ->
            TransferEncoding =
                string:to_lower(
                    lhttpc_lib:header_value("transfer-encoding", Hdrs, "undefined")),
            case TransferEncoding of
                "chunked" ->
                    chunked;
                _ ->
                    infinite
            end;
        ContentLength ->
            {fixed_length, list_to_integer(ContentLength)}
    end.

read_partial_body(State, _Vsn, Hdrs, chunked) ->
    Window = State#client_state.download_window,
    read_partial_chunked_body(State, Hdrs, Window, 0, [], 0);
read_partial_body(State, Vsn, Hdrs, infinite) ->
    check_infinite_response(Vsn, Hdrs),
    read_partial_infinite_body(State, Hdrs, State#client_state.download_window);
read_partial_body(State, _Vsn, Hdrs, {fixed_length, ContentLength}) ->
    read_partial_finite_body(State, Hdrs, ContentLength, State#client_state.download_window).

read_body(_Vsn, Hdrs, SSL, Socket, chunked) ->
    read_chunked_body(Socket, SSL, Hdrs, []);
read_body(Vsn, Hdrs, SSL, Socket, infinite) ->
    check_infinite_response(Vsn, Hdrs),
    read_infinite_body(Socket, Hdrs, SSL);
read_body(_Vsn, Hdrs, SSL, Socket, {fixed_length, ContentLength}) ->
    read_length(Hdrs, SSL, Socket, ContentLength).

read_partial_finite_body(#client_state{} = State, Hdrs, 0, _Window) ->
    reply_end_of_body(State, [], Hdrs);
read_partial_finite_body(#client_state{requester = To} = State, Hdrs, ContentLength, 0) ->
    receive
        {ack, To} ->
            read_partial_finite_body(State, Hdrs, ContentLength, 1);
        {'DOWN', _, process, To, _} ->
            exit(normal)
    end;
read_partial_finite_body(State, Hdrs, ContentLength, Window) when Window >= 0 ->
    Bin = read_body_part(State, ContentLength),
    State#client_state.requester ! {body_part, self(), Bin},
    To = State#client_state.requester,
    receive
        {ack, To} ->
            Length = ContentLength - iolist_size(Bin),
            read_partial_finite_body(State, Hdrs, Length, Window);
        {'DOWN', _, process, To, _} ->
            exit(normal)
    after 0 ->
        Length = ContentLength - iolist_size(Bin),
        read_partial_finite_body(State, Hdrs, Length, lhttpc_lib:dec(Window))
    end.

read_body_part(#client_state{part_size = infinity} = State, _ContentLength) ->
    case lhttpc_sock:recv(State#client_state.socket, State#client_state.ssl) of
        {ok, Data} ->
            Data;
        {error, Reason} ->
            erlang:error(Reason)
    end;
read_body_part(#client_state{part_size = PartSize} = State, ContentLength)
    when PartSize =< ContentLength ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    PartSize = State#client_state.part_size,
    case lhttpc_sock:recv(Socket, PartSize, SSL) of
        {ok, Data} ->
            Data;
        {error, Reason} ->
            erlang:error(Reason)
    end;
read_body_part(#client_state{part_size = PartSize} = State, ContentLength)
    when PartSize > ContentLength ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    case lhttpc_sock:recv(Socket, ContentLength, SSL) of
        {ok, Data} ->
            Data;
        {error, Reason} ->
            erlang:error(Reason)
    end.

read_length(Hdrs, SSL, Socket, Length) ->
    case lhttpc_sock:recv(Socket, Length, SSL) of
        {ok, Data} ->
            {Data, Hdrs};
        {error, Reason} ->
            erlang:error(Reason)
    end.

read_partial_chunked_body(State, Hdrs, Window, BufferSize, Buffer, 0) ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    PartSize = State#client_state.part_size,
    case read_chunk_size(Socket, SSL) of
        0 ->
            reply_chunked_part(State, Buffer, Window),
            {Trailers, NewHdrs} = read_trailers(Socket, SSL, [], Hdrs),
            reply_end_of_body(State, Trailers, NewHdrs);
        ChunkSize when PartSize =:= infinity ->
            Chunk = read_chunk(Socket, SSL, ChunkSize),
            NewWindow = reply_chunked_part(State, [Chunk | Buffer], Window),
            read_partial_chunked_body(State, Hdrs, NewWindow, 0, [], 0);
        ChunkSize when BufferSize + ChunkSize >= PartSize ->
            {Chunk, RemSize} = read_partial_chunk(Socket, SSL, PartSize - BufferSize, ChunkSize),
            NewWindow = reply_chunked_part(State, [Chunk | Buffer], Window),
            read_partial_chunked_body(State, Hdrs, NewWindow, 0, [], RemSize);
        ChunkSize ->
            Chunk = read_chunk(Socket, SSL, ChunkSize),
            read_partial_chunked_body(State,
                                      Hdrs,
                                      Window,
                                      BufferSize + ChunkSize,
                                      [Chunk | Buffer],
                                      0)
    end;
read_partial_chunked_body(State, Hdrs, Window, BufferSize, Buffer, RemSize) ->
    Socket = State#client_state.socket,
    SSL = State#client_state.ssl,
    PartSize = State#client_state.part_size,
    if BufferSize + RemSize >= PartSize ->
           {Chunk, NewRemSize} = read_partial_chunk(Socket, SSL, PartSize - BufferSize, RemSize),
           NewWindow = reply_chunked_part(State, [Chunk | Buffer], Window),
           read_partial_chunked_body(State, Hdrs, NewWindow, 0, [], NewRemSize);
       BufferSize + RemSize < PartSize ->
           Chunk = read_chunk(Socket, SSL, RemSize),
           read_partial_chunked_body(State, Hdrs, Window, BufferSize + RemSize, [Chunk | Buffer], 0)
    end.

read_chunk_size(Socket, SSL) ->
    lhttpc_sock:setopts(Socket, [{packet, line}], SSL),
    case lhttpc_sock:recv(Socket, SSL) of
        {ok, ChunkSizeExt} ->
            chunk_size(ChunkSizeExt);
        {error, Reason} ->
            erlang:error(Reason)
    end.

reply_chunked_part(_State, [], Window) ->
    Window;
reply_chunked_part(#client_state{requester = Pid} = State, Buff, 0) ->
    receive
        {ack, Pid} ->
            reply_chunked_part(State, Buff, 1);
        {'DOWN', _, process, Pid, _} ->
            exit(normal)
    end;
reply_chunked_part(#client_state{requester = Pid}, Buffer, Window) ->
    Pid ! {body_part, self(), list_to_binary(lists:reverse(Buffer))},
    receive
        {ack, Pid} ->
            Window;
        {'DOWN', _, process, Pid, _} ->
            exit(normal)
    after 0 ->
        lhttpc_lib:dec(Window)
    end.

read_chunked_body(Socket, SSL, Hdrs, Chunks) ->
    case read_chunk_size(Socket, SSL) of
        0 ->
            Body = list_to_binary(lists:reverse(Chunks)),
            {_, NewHdrs} = read_trailers(Socket, SSL, [], Hdrs),
            {Body, NewHdrs};
        Size ->
            Chunk = read_chunk(Socket, SSL, Size),
            read_chunked_body(Socket, SSL, Hdrs, [Chunk | Chunks])
    end.

chunk_size(Bin) ->
    erlang:list_to_integer(
        lists:reverse(chunk_size(Bin, [])), 16).

chunk_size(<<$;, _/binary>>, Chars) ->
    Chars;
chunk_size(<<"\r\n", _/binary>>, Chars) ->
    Chars;
chunk_size(<<$\s, Binary/binary>>, Chars) ->
    %% Facebook's HTTP server returns a chunk size like "6  \r\n"
    chunk_size(Binary, Chars);
chunk_size(<<Char, Binary/binary>>, Chars) ->
    chunk_size(Binary, [Char | Chars]).

read_partial_chunk(Socket, SSL, ChunkSize, ChunkSize) ->
    {read_chunk(Socket, SSL, ChunkSize), 0};
read_partial_chunk(Socket, SSL, Size, ChunkSize) ->
    lhttpc_sock:setopts(Socket, [{packet, raw}], SSL),
    case lhttpc_sock:recv(Socket, Size, SSL) of
        {ok, Chunk} ->
            {Chunk, ChunkSize - Size};
        {error, Reason} ->
            erlang:error(Reason)
    end.

read_chunk(Socket, SSL, Size) ->
    lhttpc_sock:setopts(Socket, [{packet, raw}], SSL),
    case lhttpc_sock:recv(Socket, Size + 2, SSL) of
        {ok, <<Chunk:Size/binary, "\r\n">>} ->
            Chunk;
        {ok, Data} ->
            erlang:error({invalid_chunk, Data});
        {error, Reason} ->
            erlang:error(Reason)
    end.

read_trailers(Socket, SSL, Trailers, Hdrs) ->
    lhttpc_sock:setopts(Socket, [{packet, httph}], SSL),
    case lhttpc_sock:recv(Socket, SSL) of
        {ok, http_eoh} ->
            {Trailers, Hdrs};
        {ok, {http_header, _, Name, _, Value}} ->
            Header = {lhttpc_lib:maybe_atom_to_list(Name), Value},
            read_trailers(Socket, SSL, [Header | Trailers], [Header | Hdrs]);
        {error, {http_error, Data}} ->
            erlang:error({bad_trailer, Data})
    end.

reply_end_of_body(#client_state{requester = Requester}, Trailers, Hdrs) ->
    Requester ! {http_eob, self(), Trailers},
    {no_return, Hdrs}.

read_partial_infinite_body(#client_state{requester = To} = State, Hdrs, 0) ->
    receive
        {ack, To} ->
            read_partial_infinite_body(State, Hdrs, 1);
        {'DOWN', _, process, To, _} ->
            exit(normal)
    end;
read_partial_infinite_body(#client_state{requester = To} = State, Hdrs, Window)
    when Window >= 0 ->
    case read_infinite_body_part(State) of
        http_eob ->
            reply_end_of_body(State, [], Hdrs);
        Bin ->
            State#client_state.requester ! {body_part, self(), Bin},
            receive
                {ack, To} ->
                    read_partial_infinite_body(State, Hdrs, Window);
                {'DOWN', _, process, To, _} ->
                    exit(normal)
            after 0 ->
                read_partial_infinite_body(State, Hdrs, lhttpc_lib:dec(Window))
            end
    end.

read_infinite_body_part(#client_state{socket = Socket, ssl = SSL}) ->
    case lhttpc_sock:recv(Socket, SSL) of
        {ok, Data} ->
            Data;
        {error, closed} ->
            http_eob;
        {error, Reason} ->
            erlang:error(Reason)
    end.

check_infinite_response({1, Minor}, Hdrs) when Minor >= 1 ->
    HdrValue = lhttpc_lib:header_value("connection", Hdrs, "keep-alive"),
    case string:to_lower(HdrValue) of
        "close" ->
            ok;
        _ ->
            erlang:error(no_content_length)
    end;
check_infinite_response(_, Hdrs) ->
    HdrValue = lhttpc_lib:header_value("connection", Hdrs, "close"),
    case string:to_lower(HdrValue) of
        "keep-alive" ->
            erlang:error(no_content_length);
        _ ->
            ok
    end.

read_infinite_body(Socket, Hdrs, SSL) ->
    read_until_closed(Socket, <<>>, Hdrs, SSL).

read_until_closed(Socket, Acc, Hdrs, SSL) ->
    case lhttpc_sock:recv(Socket, SSL) of
        {ok, Body} ->
            NewAcc = <<Acc/binary, Body/binary>>,
            read_until_closed(Socket, NewAcc, Hdrs, SSL);
        {error, closed} ->
            {Acc, Hdrs};
        {error, Reason} ->
            erlang:error(Reason)
    end.

maybe_close_socket(Socket, SSL, {1, Minor}, ReqHdrs, RespHdrs) when Minor >= 1 ->
    ClientConnection = ?CONNECTION_HDR(ReqHdrs, "keep-alive"),
    ServerConnection = ?CONNECTION_HDR(RespHdrs, "keep-alive"),
    if ClientConnection =:= "close"; ServerConnection =:= "close" ->
           lhttpc_sock:close(Socket, SSL),
           undefined;
       ClientConnection =/= "close", ServerConnection =/= "close" ->
           Socket
    end;
maybe_close_socket(Socket, SSL, _, ReqHdrs, RespHdrs) ->
    ClientConnection = ?CONNECTION_HDR(ReqHdrs, "keep-alive"),
    ServerConnection = ?CONNECTION_HDR(RespHdrs, "close"),
    if ClientConnection =:= "close"; ServerConnection =/= "keep-alive" ->
           lhttpc_sock:close(Socket, SSL),
           undefined;
       ClientConnection =/= "close", ServerConnection =:= "keep-alive" ->
           Socket
    end.