diff options
| -rw-r--r-- | .gitignore | 1 | ||||
| -rw-r--r-- | src/cowboy_client.erl | 268 | ||||
| -rw-r--r-- | src/cowboy_http.erl | 8 | ||||
| -rw-r--r-- | src/cowboy_http_protocol.erl | 23 | ||||
| -rw-r--r-- | src/cowboy_http_req.erl | 7 | ||||
| -rw-r--r-- | src/cowboy_ssl_transport.erl | 7 | ||||
| -rw-r--r-- | src/cowboy_tcp_transport.erl | 6 | ||||
| -rw-r--r-- | test/http_SUITE.erl | 1200 | ||||
| -rw-r--r-- | test/http_handler_long_polling.erl | 2 |
9 files changed, 915 insertions, 607 deletions
@@ -1,3 +1,4 @@ +.cowboy.plt .eunit deps doc/*.css diff --git a/src/cowboy_client.erl b/src/cowboy_client.erl new file mode 100644 index 0000000..21931e1 --- /dev/null +++ b/src/cowboy_client.erl @@ -0,0 +1,268 @@ +%% Copyright (c) 2012, Loïc Hoguin <[email protected]> +%% +%% Permission to use, copy, modify, and/or distribute this software for any +%% purpose with or without fee is hereby granted, provided that the above +%% copyright notice and this permission notice appear in all copies. +%% +%% THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES +%% WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF +%% MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR +%% ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES +%% WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN +%% ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF +%% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + +%% @private +-module(cowboy_client). + +-export([init/1]). +-export([state/1]). +-export([transport/1]). + +-export([connect/4]). +-export([raw_request/2]). +-export([request/3]). +-export([request/4]). +-export([request/5]). +-export([response/1]). +-export([response_body/1]). +-export([skip_body/1]). +-export([stream_status/1]). +-export([stream_headers/1]). +-export([stream_header/1]). +-export([stream_body/1]). + +-record(client, { + state = wait :: wait | request | response | response_body, + opts = [] :: [any()], + socket = undefined :: undefined | inet:socket(), + transport = undefined :: module(), + timeout = 5000 :: timeout(), %% @todo Configurable. + buffer = <<>> :: binary(), + connection = keepalive :: keepalive | close, + version = {1, 1} :: cowboy_http:version(), + response_body = undefined :: undefined | non_neg_integer() +}). + +init(Opts) -> + {ok, #client{opts=Opts}}. + +state(#client{state=State}) -> + State. + +transport(#client{socket=undefined}) -> + {error, notconnected}; +transport(#client{transport=Transport, socket=Socket}) -> + {ok, Transport, Socket}. + +connect(Transport, Host, Port, Client) + when is_binary(Host) -> + connect(Transport, binary_to_list(Host), Port, Client); +connect(Transport, Host, Port, Client=#client{state=State, opts=Opts}) + when is_atom(Transport), is_list(Host), + is_integer(Port), is_record(Client, client), + State =:= wait -> + {ok, Socket} = Transport:connect(Host, Port, Opts), + {ok, Client#client{state=request, socket=Socket, transport=Transport}}. + +raw_request(Data, Client=#client{state=response_body}) -> + {done, Client2} = skip_body(Client), + raw_request(Data, Client2); +raw_request(Data, Client=#client{ + state=State, socket=Socket, transport=Transport}) + when State =:= request -> + ok = Transport:send(Socket, Data), + {ok, Client}. + +request(Method, URL, Client) -> + request(Method, URL, [], <<>>, Client). + +request(Method, URL, Headers, Client) -> + request(Method, URL, Headers, <<>>, Client). + +request(Method, URL, Headers, Body, Client=#client{state=response_body}) -> + {done, Client2} = skip_body(Client), + request(Method, URL, Headers, Body, Client2); +request(Method, URL, Headers, Body, Client=#client{ + state=State, version=Version}) + when State =:= wait; State =:= request -> + {Transport, FullHost, Host, Port, Path} = parse_url(URL), + {ok, Client2} = case State of + wait -> connect(Transport, Host, Port, Client); + request -> {ok, Client} + end, + VersionBin = cowboy_http:version_to_binary(Version), + %% @todo do keepalive too, allow override... + Headers2 = [ + {<<"host">>, FullHost}, + {<<"user-agent">>, <<"Cow">>} + |Headers], + Headers3 = case iolist_size(Body) of + 0 -> Headers2; + Length -> [{<<"content-length">>, integer_to_list(Length)}|Headers2] + end, + HeadersData = [[Name, <<": ">>, Value, <<"\r\n">>] + || {Name, Value} <- Headers3], + Data = [Method, <<" ">>, Path, <<" ">>, VersionBin, <<"\r\n">>, + HeadersData, <<"\r\n">>, Body], + raw_request(Data, Client2). + +parse_url(<< "https://", Rest/binary >>) -> + parse_url(Rest, cowboy_ssl_transport); +parse_url(<< "http://", Rest/binary >>) -> + parse_url(Rest, cowboy_tcp_transport); +parse_url(URL) -> + parse_url(URL, cowboy_tcp_transport). + +parse_url(URL, Transport) -> + case binary:split(URL, <<"/">>) of + [Peer] -> + {Host, Port} = parse_peer(Peer, Transport), + {Transport, Peer, Host, Port, <<"/">>}; + [Peer, Path] -> + {Host, Port} = parse_peer(Peer, Transport), + {Transport, Peer, Host, Port, [<<"/">>, Path]} + end. + +parse_peer(Peer, Transport) -> + case binary:split(Peer, <<":">>) of + [Host] when Transport =:= cowboy_tcp_transport -> + {binary_to_list(Host), 80}; + [Host] when Transport =:= cowboy_ssl_transport -> + {binary_to_list(Host), 443}; + [Host, Port] -> + {binary_to_list(Host), list_to_integer(binary_to_list(Port))} + end. + +response(Client=#client{state=response_body}) -> + {done, Client2} = skip_body(Client), + response(Client2); +response(Client=#client{state=request}) -> + case stream_status(Client) of + {ok, Status, _, Client2} -> + case stream_headers(Client2) of + {ok, Headers, Client3} -> + {ok, Status, Headers, Client3}; + {error, Reason} -> + {error, Reason} + end; + {error, Reason} -> + {error, Reason} + end. + +response_body(Client=#client{state=response_body}) -> + response_body_loop(Client, <<>>). + +response_body_loop(Client, Acc) -> + case stream_body(Client) of + {ok, Data, Client2} -> + response_body_loop(Client2, << Acc/binary, Data/binary >>); + {done, Client2} -> + {ok, Acc, Client2} + end. + +skip_body(Client=#client{state=response_body}) -> + case stream_body(Client) of + {ok, _, Client2} -> skip_body(Client2); + Done -> Done + end. + +stream_status(Client=#client{state=State, buffer=Buffer}) + when State =:= request -> + case binary:split(Buffer, <<"\r\n">>) of + [Line, Rest] -> + parse_status(Client#client{state=response, buffer=Rest}, Line); + _ -> + case recv(Client) of + {ok, Data} -> + Buffer2 = << Buffer/binary, Data/binary >>, + stream_status(Client#client{buffer=Buffer2}); + {error, Reason} -> + {error, Reason} + end + end. + +parse_status(Client, << "HTTP/", High, ".", Low, " ", + S3, S2, S1, " ", StatusStr/binary >>) + when High >= $0, High =< $9, Low >= $0, Low =< $9, + S3 >= $0, S3 =< $9, S2 >= $0, S2 =< $9, S1 >= $0, S1 =< $9 -> + Version = {High - $0, Low - $0}, + Status = (S3 - $0) * 100 + (S2 - $0) * 10 + S1 - $0, + {ok, Status, StatusStr, Client#client{version=Version}}. + +stream_headers(Client=#client{state=State}) + when State =:= response -> + stream_headers(Client, []). + +stream_headers(Client, Acc) -> + case stream_header(Client) of + {ok, Name, Value, Client2} -> + stream_headers(Client2, [{Name, Value}|Acc]); + {done, Client2} -> + {ok, Acc, Client2}; + {error, Reason} -> + {error, Reason} + end. + +stream_header(Client=#client{state=State, buffer=Buffer, + response_body=RespBody}) when State =:= response -> + case binary:split(Buffer, <<"\r\n">>) of + [<<>>, Rest] -> + %% If we have a body, set response_body. + Client2 = case RespBody of + undefined -> Client#client{state=request}; + 0 -> Client#client{state=request}; + _ -> Client#client{state=response_body} + end, + {done, Client2#client{buffer=Rest}}; + [Line, Rest] -> + %% @todo Do a better parsing later on. + [Name, Value] = binary:split(Line, <<": ">>), + Name2 = cowboy_bstr:to_lower(Name), + Client2 = case Name2 of + <<"content-length">> -> + Length = list_to_integer(binary_to_list(Value)), + if Length >= 0 -> ok end, + Client#client{response_body=Length}; + _ -> + Client + end, + {ok, Name2, Value, Client2#client{buffer=Rest}}; + _ -> + case recv(Client) of + {ok, Data} -> + Buffer2 = << Buffer/binary, Data/binary >>, + stream_header(Client#client{buffer=Buffer2}); + {error, Reason} -> + {error, Reason} + end + end. + +stream_body(Client=#client{state=response_body, response_body=RespBody}) + when RespBody =:= undefined; RespBody =:= 0 -> + {done, Client#client{state=request, response_body=undefined}}; +stream_body(Client=#client{state=response_body, buffer=Buffer, + response_body=Length}) when is_integer(Length) -> + case byte_size(Buffer) of + 0 -> + case recv(Client) of + {ok, Body} when byte_size(Body) =< Length -> + Length2 = Length - byte_size(Body), + {ok, Body, Client#client{response_body=Length2}}; + {ok, Data} -> + << Body:Length/binary, Rest/binary >> = Data, + {ok, Body, Client#client{buffer=Rest, + response_body=undefined}}; + {error, Reason} -> + {error, Reason} + end; + N when N =< Length -> + Length2 = Length - N, + {ok, Buffer, Client#client{buffer= <<>>, response_body=Length2}}; + _ -> + << Body:Length/binary, Rest/binary >> = Buffer, + {ok, Body, Client#client{buffer=Rest, response_body=undefined}} + end. + +recv(#client{socket=Socket, transport=Transport, timeout=Timeout}) -> + Transport:recv(Socket, 0, Timeout). diff --git a/src/cowboy_http.erl b/src/cowboy_http.erl index 2f4f982..0289ef3 100644 --- a/src/cowboy_http.erl +++ b/src/cowboy_http.erl @@ -26,7 +26,8 @@ -export([te_chunked/2, te_identity/2, ce_identity/1]). %% Interpretation. --export([connection_to_atom/1, urldecode/1, urldecode/2, urlencode/1, +-export([connection_to_atom/1, version_to_binary/1, + urldecode/1, urldecode/2, urlencode/1, urlencode/2, x_www_form_urlencoded/2]). -type method() :: 'OPTIONS' | 'GET' | 'HEAD' @@ -773,6 +774,11 @@ connection_to_atom([<<"close">>|_Tail]) -> connection_to_atom([_Any|Tail]) -> connection_to_atom(Tail). +%% @doc Convert an HTTP version tuple to its binary form. +-spec version_to_binary(version()) -> binary(). +version_to_binary({1, 1}) -> <<"HTTP/1.1">>; +version_to_binary({1, 0}) -> <<"HTTP/1.0">>. + %% @doc Decode a URL encoded binary. %% @equiv urldecode(Bin, crash) -spec urldecode(binary()) -> binary(). diff --git a/src/cowboy_http_protocol.erl b/src/cowboy_http_protocol.erl index 04abfbc..816c825 100644 --- a/src/cowboy_http_protocol.erl +++ b/src/cowboy_http_protocol.erl @@ -121,16 +121,22 @@ request({http_request, Method, {absoluteURI, _Scheme, _Host, _Port, Path}, request({http_request, Method, {abs_path, Path}, Version}, State); request({http_request, Method, {abs_path, AbsPath}, Version}, State=#state{socket=Socket, transport=Transport, + req_keepalive=Keepalive, max_keepalive=MaxKeepalive, urldecode={URLDecFun, URLDecArg}=URLDec}) -> URLDecode = fun(Bin) -> URLDecFun(Bin, URLDecArg) end, {Path, RawPath, Qs} = cowboy_dispatcher:split_path(AbsPath, URLDecode), - ConnAtom = version_to_connection(Version), + ConnAtom = if Keepalive < MaxKeepalive -> version_to_connection(Version); + true -> close + end, parse_header(#http_req{socket=Socket, transport=Transport, connection=ConnAtom, pid=self(), method=Method, version=Version, path=Path, raw_path=RawPath, raw_qs=Qs, urldecode=URLDec}, State); request({http_request, Method, '*', Version}, - State=#state{socket=Socket, transport=Transport, urldecode=URLDec}) -> - ConnAtom = version_to_connection(Version), + State=#state{socket=Socket, transport=Transport, + req_keepalive=Keepalive, max_keepalive=MaxKeepalive, urldecode=URLDec}) -> + ConnAtom = if Keepalive < MaxKeepalive -> version_to_connection(Version); + true -> close + end, parse_header(#http_req{socket=Socket, transport=Transport, connection=ConnAtom, pid=self(), method=Method, version=Version, path='*', raw_path= <<"*">>, raw_qs= <<>>, urldecode=URLDec}, State); @@ -186,7 +192,9 @@ header({http_header, _I, 'Host', _R, RawHost}, Req=#http_req{ header({http_header, _I, 'Host', _R, _V}, Req, State) -> parse_header(Req, State); header({http_header, _I, 'Connection', _R, Connection}, - Req=#http_req{headers=Headers}, State) -> + Req=#http_req{headers=Headers}, State=#state{ + req_keepalive=Keepalive, max_keepalive=MaxKeepalive}) + when Keepalive < MaxKeepalive -> Req2 = Req#http_req{headers=[{'Connection', Connection}|Headers]}, {ConnTokens, Req3} = cowboy_http_req:parse_header('Connection', Req2), @@ -376,15 +384,14 @@ terminate_request(HandlerState, Req, State) -> next_request(Req, State, HandlerRes). -spec next_request(#http_req{}, #state{}, any()) -> ok. -next_request(Req=#http_req{connection=Conn}, - State=#state{req_keepalive=Keepalive, max_keepalive=MaxKeepalive}, - HandlerRes) -> +next_request(Req=#http_req{connection=Conn}, State=#state{ + req_keepalive=Keepalive}, HandlerRes) -> RespRes = ensure_response(Req), {BodyRes, Buffer} = ensure_body_processed(Req), %% Flush the resp_sent message before moving on. receive {cowboy_http_req, resp_sent} -> ok after 0 -> ok end, case {HandlerRes, BodyRes, RespRes, Conn} of - {ok, ok, ok, keepalive} when Keepalive < MaxKeepalive -> + {ok, ok, ok, keepalive} -> ?MODULE:parse_request(State#state{ buffer=Buffer, req_empty_lines=0, req_keepalive=Keepalive + 1}); diff --git a/src/cowboy_http_req.erl b/src/cowboy_http_req.erl index a6e8834..dfb73e7 100644 --- a/src/cowboy_http_req.erl +++ b/src/cowboy_http_req.erl @@ -456,10 +456,11 @@ stream_body(Req=#http_req{body_state=done}) -> -spec stream_body_recv(#http_req{}) -> {ok, binary(), #http_req{}} | {error, atom()}. -stream_body_recv(Req=#http_req{transport=Transport, socket=Socket}) -> +stream_body_recv(Req=#http_req{ + transport=Transport, socket=Socket, buffer=Buffer}) -> %% @todo Allow configuring the timeout. case Transport:recv(Socket, 0, 5000) of - {ok, Data} -> transfer_decode(Data, Req); + {ok, Data} -> transfer_decode(<< Buffer/binary, Data/binary >>, Req); {error, Reason} -> {error, Reason} end. @@ -477,7 +478,7 @@ transfer_decode(Data, Req=#http_req{ {stream, TransferDecode, TransferState2, ContentDecode}}); %% @todo {header(s) for chunked more -> - stream_body_recv(Req); + stream_body_recv(Req#http_req{buffer=Data}); {done, Length, Rest} -> Req2 = transfer_decode_done(Length, Rest, Req), {done, Req2}; diff --git a/src/cowboy_ssl_transport.erl b/src/cowboy_ssl_transport.erl index 3b130f0..a661622 100644 --- a/src/cowboy_ssl_transport.erl +++ b/src/cowboy_ssl_transport.erl @@ -25,6 +25,7 @@ -module(cowboy_ssl_transport). -export([name/0, messages/0, listen/1, accept/2, recv/3, send/2, setopts/2, controlling_process/2, peername/1, close/1, sockname/1]). +-export([connect/3]). %% @doc Name of this transport API, <em>ssl</em>. -spec name() -> ssl. @@ -37,6 +38,12 @@ name() -> ssl. -spec messages() -> {ssl, ssl_closed, ssl_error}. messages() -> {ssl, ssl_closed, ssl_error}. +%% @private +%% @todo Probably filter Opts? +connect(Host, Port, Opts) when is_list(Host), is_integer(Port) -> + ssl:connect(Host, Port, + Opts ++ [binary, {active, false}, {packet, raw}]). + %% @doc Setup a socket to listen on the given port on the local host. %% %% The available options are: diff --git a/src/cowboy_tcp_transport.erl b/src/cowboy_tcp_transport.erl index f197dd1..079494d 100644 --- a/src/cowboy_tcp_transport.erl +++ b/src/cowboy_tcp_transport.erl @@ -21,6 +21,7 @@ -export([name/0, messages/0, listen/1, accept/2, recv/3, send/2, setopts/2, controlling_process/2, peername/1, close/1, sockname/1]). +-export([connect/3]). %% @doc Name of this transport API, <em>tcp</em>. -spec name() -> tcp. @@ -33,6 +34,11 @@ name() -> tcp. -spec messages() -> {tcp, tcp_closed, tcp_error}. messages() -> {tcp, tcp_closed, tcp_error}. +%% @private +connect(Host, Port, Opts) when is_list(Host), is_integer(Port) -> + gen_tcp:connect(Host, Port, + Opts ++ [binary, {active, false}, {packet, raw}]). + %% @doc Setup a socket to listen on the given port on the local host. %% %% The available options are: diff --git a/test/http_SUITE.erl b/test/http_SUITE.erl index cebc1e6..a906037 100644 --- a/test/http_SUITE.erl +++ b/test/http_SUITE.erl @@ -17,44 +17,102 @@ -include_lib("common_test/include/ct.hrl"). --export([all/0, groups/0, init_per_suite/1, end_per_suite/1, - init_per_group/2, end_per_group/2]). %% ct. --export([chunked_response/1, headers_dupe/1, headers_huge/1, - keepalive_nl/1, max_keepalive/1, nc_rand/1, nc_zero/1, - pipeline/1, raw/1, set_resp_header/1, set_resp_overwrite/1, - set_resp_body/1, stream_body_set_resp/1, response_as_req/1, - static_mimetypes_function/1, static_attribute_etag/1, - static_function_etag/1, multipart/1, te_identity/1, - te_chunked/1, te_chunked_delayed/1]). %% http. --export([http_200/1, http_404/1, handler_errors/1, - file_200/1, file_403/1, dir_403/1, file_404/1, - file_400/1]). %% http and https. --export([http_10_hostless/1, http_10_chunkless/1]). %% misc. --export([rest_simple/1, rest_keepalive/1, rest_keepalive_post/1, - rest_nodelete/1, rest_resource_etags/1]). %% rest. --export([onrequest/1, onrequest_reply/1]). %% hooks. +%% ct. +-export([all/0]). +-export([groups/0]). +-export([init_per_suite/1]). +-export([end_per_suite/1]). +-export([init_per_group/2]). +-export([end_per_group/2]). + +%% Tests. +-export([check_raw_status/1]). +-export([check_status/1]). +-export([chunked_response/1]). +-export([error_chain_handle_after_reply/1]). +-export([error_chain_handle_before_reply/1]). +-export([error_handle_after_reply/1]). +-export([error_init_after_reply/1]). +-export([error_init_reply_handle_error/1]). +-export([headers_dupe/1]). +-export([http10_chunkless/1]). +-export([http10_hostless/1]). +-export([keepalive_max/1]). +-export([keepalive_nl/1]). +-export([multipart/1]). +-export([nc_rand/1]). +-export([nc_zero/1]). +-export([onrequest/1]). +-export([onrequest_reply/1]). +-export([pipeline/1]). +-export([rest_keepalive/1]). +-export([rest_keepalive_post/1]). +-export([rest_nodelete/1]). +-export([rest_resource_etags/1]). +-export([rest_resource_etags_if_none_match/1]). +-export([set_resp_body/1]). +-export([set_resp_header/1]). +-export([set_resp_overwrite/1]). +-export([static_attribute_etag/1]). +-export([static_function_etag/1]). +-export([static_mimetypes_function/1]). +-export([static_test_file/1]). +-export([static_test_file_css/1]). +-export([stream_body_set_resp/1]). +-export([te_chunked/1]). +-export([te_chunked_delayed/1]). +-export([te_identity/1]). %% ct. all() -> - [{group, http}, {group, https}, {group, misc}, {group, rest}, - {group, hooks}]. + [{group, http}, {group, https}, {group, hooks}]. groups() -> - BaseTests = [http_200, http_404, handler_errors, - file_200, file_403, dir_403, file_404, file_400], - [{http, [], [chunked_response, headers_dupe, headers_huge, - keepalive_nl, max_keepalive, nc_rand, nc_zero, pipeline, raw, - set_resp_header, set_resp_overwrite, - set_resp_body, response_as_req, stream_body_set_resp, - static_mimetypes_function, static_attribute_etag, - static_function_etag, multipart, te_identity, te_chunked, - te_chunked_delayed] ++ BaseTests}, - {https, [], BaseTests}, - {misc, [], [http_10_hostless, http_10_chunkless]}, - {rest, [], [rest_simple, rest_keepalive, rest_keepalive_post, - rest_nodelete, rest_resource_etags]}, - {hooks, [], [onrequest, onrequest_reply]}]. + Tests = [ + check_raw_status, + check_status, + chunked_response, + error_chain_handle_after_reply, + error_chain_handle_before_reply, + error_handle_after_reply, + error_init_after_reply, + error_init_reply_handle_error, + headers_dupe, + http10_chunkless, + http10_hostless, + keepalive_max, + keepalive_nl, + multipart, + nc_rand, + nc_zero, + pipeline, + rest_keepalive, + rest_keepalive_post, + rest_nodelete, + rest_resource_etags, + rest_resource_etags_if_none_match, + set_resp_body, + set_resp_header, + set_resp_overwrite, + static_attribute_etag, + static_function_etag, + static_mimetypes_function, + static_test_file, + static_test_file_css, + stream_body_set_resp, + te_chunked, + te_chunked_delayed, + te_identity + ], + [ + {http, [], Tests}, + {https, [], Tests}, + {hooks, [], [ + onrequest, + onrequest_reply + ]} + ]. init_per_suite(Config) -> application:start(inets), @@ -68,57 +126,54 @@ end_per_suite(_Config) -> init_per_group(http, Config) -> Port = 33080, + Transport = cowboy_tcp_transport, Config1 = init_static_dir(Config), cowboy:start_listener(http, 100, - cowboy_tcp_transport, [{port, Port}], - cowboy_http_protocol, [{max_keepalive, 50}, - {dispatch, init_http_dispatch(Config1)}] + Transport, [{port, Port}], + cowboy_http_protocol, [ + {dispatch, init_dispatch(Config1)}, + {max_keepalive, 50}, + {timeout, 500}] ), - [{scheme, "http"}, {port, Port}|Config1]; + {ok, Client} = cowboy_client:init([]), + [{scheme, <<"http">>}, {port, Port}, {opts, []}, + {transport, Transport}, {client, Client}|Config1]; init_per_group(https, Config) -> Port = 33081, + Transport = cowboy_ssl_transport, + Opts = [ + {certfile, ?config(data_dir, Config) ++ "cert.pem"}, + {keyfile, ?config(data_dir, Config) ++ "key.pem"}, + {password, "cowboy"} + ], Config1 = init_static_dir(Config), application:start(crypto), application:start(public_key), application:start(ssl), - DataDir = ?config(data_dir, Config), {ok,_} = cowboy:start_listener(https, 100, - cowboy_ssl_transport, [ - {port, Port}, {certfile, DataDir ++ "cert.pem"}, - {keyfile, DataDir ++ "key.pem"}, {password, "cowboy"}], - cowboy_http_protocol, [{dispatch, init_https_dispatch(Config1)}] + Transport, Opts ++ [{port, Port}], + cowboy_http_protocol, [ + {dispatch, init_dispatch(Config1)}, + {max_keepalive, 50}, + {timeout, 500}] ), - [{scheme, "https"}, {port, Port}|Config1]; -init_per_group(misc, Config) -> - Port = 33082, - {ok,_} = cowboy:start_listener(misc, 100, - cowboy_tcp_transport, [{port, Port}], - cowboy_http_protocol, [{dispatch, [{'_', [ - {[<<"chunked_response">>], chunked_handler, []}, - {[], http_handler, []} - ]}]}]), - [{port, Port}|Config]; -init_per_group(rest, Config) -> - Port = 33083, - {ok,_} = cowboy:start_listener(rest, 100, - cowboy_tcp_transport, [{port, Port}], - cowboy_http_protocol, [{dispatch, [{'_', [ - {[<<"simple">>], rest_simple_resource, []}, - {[<<"forbidden_post">>], rest_forbidden_resource, [true]}, - {[<<"simple_post">>], rest_forbidden_resource, [false]}, - {[<<"nodelete">>], rest_nodelete_resource, []}, - {[<<"resetags">>], rest_resource_etags, []} - ]}]}]), - [{scheme, "http"},{port, Port}|Config]; + {ok, Client} = cowboy_client:init(Opts), + [{scheme, <<"https">>}, {port, Port}, {opts, Opts}, + {transport, Transport}, {client, Client}|Config1]; init_per_group(hooks, Config) -> - Port = 33084, + Port = 33082, + Transport = cowboy_tcp_transport, {ok, _} = cowboy:start_listener(hooks, 100, - cowboy_tcp_transport, [{port, Port}], + Transport, [{port, Port}], cowboy_http_protocol, [ - {dispatch, init_http_dispatch(Config)}, - {onrequest, fun onrequest_hook/1} + {dispatch, init_dispatch(Config)}, + {max_keepalive, 50}, + {onrequest, fun onrequest_hook/1}, + {timeout, 500} ]), - [{scheme, "http"}, {port, Port}|Config]. + {ok, Client} = cowboy_client:init([]), + [{scheme, <<"http">>}, {port, Port}, {opts, []}, + {transport, Transport}, {client, Client}|Config]. end_per_group(https, Config) -> cowboy:stop_listener(https), @@ -130,13 +185,13 @@ end_per_group(https, Config) -> end_per_group(http, Config) -> cowboy:stop_listener(http), end_static_dir(Config); -end_per_group(Listener, _Config) -> - cowboy:stop_listener(Listener), +end_per_group(hooks, _) -> + cowboy:stop_listener(hooks), ok. %% Dispatch configuration. -init_http_dispatch(Config) -> +init_dispatch(Config) -> [ {[<<"localhost">>], [ {[<<"chunked_response">>], chunked_handler, []}, @@ -168,14 +223,15 @@ init_http_dispatch(Config) -> {etag, {fun static_function_etag/2, etag_data}}]}, {[<<"multipart">>], http_handler_multipart, []}, {[<<"echo">>, <<"body">>], http_handler_echo_body, []}, + {[<<"simple">>], rest_simple_resource, []}, + {[<<"forbidden_post">>], rest_forbidden_resource, [true]}, + {[<<"simple_post">>], rest_forbidden_resource, [false]}, + {[<<"nodelete">>], rest_nodelete_resource, []}, + {[<<"resetags">>], rest_resource_etags, []}, {[], http_handler, []} ]} ]. -init_https_dispatch(Config) -> - init_http_dispatch(Config). - - init_static_dir(Config) -> Dir = filename:join(?config(priv_dir, Config), "static"), Level1 = fun(Name) -> filename:join(Dir, Name) end, @@ -199,69 +255,248 @@ end_static_dir(Config) -> ok = file:del_dir(Dir), Config. -%% http. +%% Convenience functions. + +quick_raw(Data, Config) -> + Client = ?config(client, Config), + Transport = ?config(transport, Config), + {ok, Client2} = cowboy_client:connect( + Transport, "localhost", ?config(port, Config), Client), + {ok, Client3} = cowboy_client:raw_request(Data, Client2), + case cowboy_client:response(Client3) of + {ok, Status, _, _} -> Status; + {error, _} -> closed + end. + +build_url(Path, Config) -> + {scheme, Scheme} = lists:keyfind(scheme, 1, Config), + {port, Port} = lists:keyfind(port, 1, Config), + PortBin = list_to_binary(integer_to_list(Port)), + PathBin = list_to_binary(Path), + << Scheme/binary, "://localhost:", PortBin/binary, PathBin/binary >>. + +quick_get(URL, Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url(URL, Config), Client), + {ok, Status, _, _} = cowboy_client:response(Client2), + Status. + +body_to_chunks(_, <<>>, Acc) -> + lists:reverse([<<"0\r\n\r\n">>|Acc]); +body_to_chunks(ChunkSize, Body, Acc) -> + BodySize = byte_size(Body), + ChunkSize2 = case BodySize < ChunkSize of + true -> BodySize; + false -> ChunkSize + end, + << Chunk:ChunkSize2/binary, Rest/binary >> = Body, + ChunkSizeBin = list_to_binary(integer_to_list(ChunkSize2, 16)), + body_to_chunks(ChunkSize, Rest, + [<< ChunkSizeBin/binary, "\r\n", Chunk/binary, "\r\n" >>|Acc]). + +%% Tests. + +check_raw_status(Config) -> + Huge = [$0 || _ <- lists:seq(1, 5000)], + HugeCookie = lists:flatten(["whatever_man_biiiiiiiiiiiig_cookie_me_want_77=" + "Wed Apr 06 2011 10:38:52 GMT-0500 (CDT)" || _ <- lists:seq(1, 40)]), + ResponsePacket = +"HTTP/1.0 302 Found +Location: http://www.google.co.il/ +Cache-Control: private +Content-Type: text/html; charset=UTF-8 +Set-Cookie: PREF=ID=568f67013d4a7afa:FF=0:TM=1323014101:LM=1323014101:S=XqctDWC65MzKT0zC; expires=Tue, 03-Dec-2013 15:55:01 GMT; path=/; domain=.google.com +Date: Sun, 04 Dec 2011 15:55:01 GMT +Server: gws +Content-Length: 221 +X-XSS-Protection: 1; mode=block +X-Frame-Options: SAMEORIGIN + +<HTML><HEAD><meta http-equiv=\"content-type\" content=\"text/html;charset=utf-8\"> +<TITLE>302 Moved</TITLE></HEAD><BODY> +<H1>302 Moved</H1> +The document has moved +<A HREF=\"http://www.google.co.il/\">here</A>. +</BODY></HTML>", + Tests = [ + {200, ["GET / HTTP/1.0\r\nHost: localhost\r\n" + "Set-Cookie: ", HugeCookie, "\r\n\r\n"]}, + {200, "\r\n\r\n\r\n\r\n\r\nGET / HTTP/1.1\r\nHost: localhost\r\n\r\n"}, + {200, "GET http://proxy/ HTTP/1.1\r\nHost: localhost\r\n\r\n"}, + {400, "\n"}, + {400, "Garbage\r\n\r\n"}, + {400, "\r\n\r\n\r\n\r\n\r\n\r\n"}, + {400, "GET / HTTP/1.1\r\nHost: dev-extend.eu\r\n\r\n"}, + {400, "GET http://proxy/ HTTP/1.1\r\n\r\n"}, + {400, ResponsePacket}, + {408, "GET / HTTP/1.1\r\n"}, + {408, "GET / HTTP/1.1\r\nHost: localhost"}, + {408, "GET / HTTP/1.1\r\nHost: localhost\r\n"}, + {408, "GET / HTTP/1.1\r\nHost: localhost\r\n\r"}, + {413, Huge}, + {413, "GET / HTTP/1.1\r\n" ++ Huge}, + {505, "GET / HTTP/1.2\r\nHost: localhost\r\n\r\n"}, + {closed, ""}, + {closed, "\r\n"}, + {closed, "\r\n\r\n"}, + {closed, "GET / HTTP/1.1"} + ], + _ = [{Status, Packet} = begin + Ret = quick_raw(Packet, Config), + {Ret, Packet} + end || {Status, Packet} <- Tests]. + +check_status(Config) -> + Tests = [ + {102, "/long_polling"}, + {200, "/"}, + {200, "/simple"}, + {400, "/static/%2f"}, + {400, "/static/%2e"}, + {400, "/static/%2e%2e"}, + {403, "/static/test_dir"}, + {403, "/static/test_dir/"}, + {403, "/static/test_noread"}, + {404, "/not/found"}, + {404, "/static/not_found"}, + {500, "/handler_errors?case=handler_before_reply"}, + {500, "/handler_errors?case=init_before_reply"}, + {666, "/init_shutdown"} + ], + _ = [{Status, URL} = begin + Ret = quick_get(URL, Config), + {Ret, URL} + end || {Status, URL} <- Tests]. +%% @todo Convert to cowboy_client. chunked_response(Config) -> - {ok, {{"HTTP/1.1", 200, "OK"}, _Headers, "chunked_handler\r\nworks fine!"}} = - httpc:request(build_url("/chunked_response", Config)). + {ok, {{"HTTP/1.1", 200, "OK"}, _, "chunked_handler\r\nworks fine!"}} + = httpc:request(binary_to_list(build_url("/chunked_response", Config))). + +error_chain_handle_after_reply(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/", Config), Client), + {ok, Client3} = cowboy_client:request(<<"GET">>, + build_url("/handler_errors?case=handle_after_reply", Config), Client2), + {ok, 200, _, Client4} = cowboy_client:response(Client3), + {ok, 200, _, Client5} = cowboy_client:response(Client4), + {error, closed} = cowboy_client:response(Client5). + +error_chain_handle_before_reply(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/", Config), Client), + {ok, Client3} = cowboy_client:request(<<"GET">>, + build_url("/handler_errors?case=handle_before_reply", Config), Client2), + {ok, 200, _, Client4} = cowboy_client:response(Client3), + {ok, 500, _, Client5} = cowboy_client:response(Client4), + {error, closed} = cowboy_client:response(Client5). + +error_handle_after_reply(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/handler_errors?case=handle_after_reply", Config), Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {error, closed} = cowboy_client:response(Client3). + +error_init_after_reply(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/handler_errors?case=init_after_reply", Config), Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {error, closed} = cowboy_client:response(Client3). + +error_init_reply_handle_error(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/handler_errors?case=init_reply_handle_error", Config), Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {error, closed} = cowboy_client:response(Client3). headers_dupe(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, "GET /headers/dupe HTTP/1.1\r\n" - "Host: localhost\r\nConnection: keep-alive\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_Start, _Length} = binary:match(Data, <<"Connection: close">>), - nomatch = binary:match(Data, <<"Connection: keep-alive">>), - {error, closed} = gen_tcp:recv(Socket, 0, 1000). - -headers_huge(Config) -> - Cookie = lists:flatten(["whatever_man_biiiiiiiiiiiig_cookie_me_want_77=" - "Wed Apr 06 2011 10:38:52 GMT-0500 (CDT)" || _N <- lists:seq(1, 40)]), - {_Packet, 200} = raw_req(["GET / HTTP/1.0\r\nHost: localhost\r\n" - "Set-Cookie: ", Cookie, "\r\n\r\n"], Config). + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/headers/dupe", Config), Client), + {ok, 200, Headers, Client3} = cowboy_client:response(Client2), + {<<"connection">>, <<"close">>} + = lists:keyfind(<<"connection">>, 1, Headers), + Connections = [H || H = {Name, _} <- Headers, Name =:= <<"connection">>], + 1 = length(Connections), + {error, closed} = cowboy_client:response(Client3). + +http10_chunkless(Config) -> + Client = ?config(client, Config), + Transport = ?config(transport, Config), + {ok, Client2} = cowboy_client:connect( + Transport, "localhost", ?config(port, Config), Client), + Data = "GET /chunked_response HTTP/1.0\r\nHost: localhost\r\n\r\n", + {ok, Client3} = cowboy_client:raw_request(Data, Client2), + {ok, 200, Headers, Client4} = cowboy_client:response(Client3), + false = lists:keyfind(<<"transfer-encoding">>, 1, Headers), + %% Hack: we just try to get 28 bytes and compare. + {ok, Transport, Socket} = cowboy_client:transport(Client4), + Buffer = element(7, Client4), + Buffer2 = case Transport:recv(Socket, 28 - byte_size(Buffer), 1000) of + {ok, Recv} -> << Buffer/binary, Recv/binary >>; + _ -> Buffer + end, + <<"chunked_handler\r\nworks fine!">> = Buffer2. -keepalive_nl(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = keepalive_nl_loop(Socket, 10), - ok = gen_tcp:close(Socket). +http10_hostless(Config) -> + Port10 = ?config(port, Config) + 10, + Name = list_to_atom("http10_hostless_" ++ integer_to_list(Port10)), + cowboy:start_listener(Name, 5, + ?config(transport, Config), ?config(opts, Config) ++ [{port, Port10}], + cowboy_http_protocol, [ + {dispatch, [{'_', [ + {[<<"http1.0">>, <<"hostless">>], http_handler, []}]}]}, + {max_keepalive, 50}, + {timeout, 500}] + ), + 200 = quick_raw("GET /http1.0/hostless HTTP/1.0\r\n\r\n", + [{port, Port10}|Config]), + cowboy:stop_listener(http10). -keepalive_nl_loop(_Socket, 0) -> - ok; -keepalive_nl_loop(Socket, N) -> - ok = gen_tcp:send(Socket, "GET / HTTP/1.1\r\n" - "Host: localhost\r\nConnection: keep-alive\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {0, 12} = binary:match(Data, <<"HTTP/1.1 200">>), - nomatch = binary:match(Data, <<"Connection: close">>), - ok = gen_tcp:send(Socket, "\r\n"), %% extra nl - keepalive_nl_loop(Socket, N - 1). - -max_keepalive(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = max_keepalive_loop(Socket, 50), - {error, closed} = gen_tcp:recv(Socket, 0, 1000). +keepalive_max(Config) -> + Client = ?config(client, Config), + URL = build_url("/", Config), + ok = keepalive_max_loop(Client, URL, 50). -max_keepalive_loop(_Socket, 0) -> +keepalive_max_loop(_, _, 0) -> ok; -max_keepalive_loop(Socket, N) -> - ok = gen_tcp:send(Socket, "GET / HTTP/1.1\r\n" - "Host: localhost\r\nConnection: keep-alive\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {0, 12} = binary:match(Data, <<"HTTP/1.1 200">>), - case N of - 1 -> {_, _} = binary:match(Data, <<"Connection: close">>); - N -> nomatch = binary:match(Data, <<"Connection: close">>) +keepalive_max_loop(Client, URL, N) -> + Headers = [{<<"connection">>, <<"keep-alive">>}], + {ok, Client2} = cowboy_client:request(<<"GET">>, URL, Headers, Client), + {ok, 200, RespHeaders, Client3} = cowboy_client:response(Client2), + Expected = case N of + 1 -> <<"close">>; + N -> <<"keep-alive">> end, - keepalive_nl_loop(Socket, N - 1). + {<<"connection">>, Expected} + = lists:keyfind(<<"connection">>, 1, RespHeaders), + keepalive_max_loop(Client3, URL, N - 1). + +keepalive_nl(Config) -> + Client = ?config(client, Config), + URL = build_url("/", Config), + ok = keepalive_nl_loop(Client, URL, 10). + +keepalive_nl_loop(_, _, 0) -> + ok; +keepalive_nl_loop(Client, URL, N) -> + Headers = [{<<"connection">>, <<"keep-alive">>}], + {ok, Client2} = cowboy_client:request(<<"GET">>, URL, Headers, Client), + {ok, 200, RespHeaders, Client3} = cowboy_client:response(Client2), + {<<"connection">>, <<"keep-alive">>} + = lists:keyfind(<<"connection">>, 1, RespHeaders), + {ok, Transport, Socket} = cowboy_client:transport(Client2), + ok = Transport:send(Socket, <<"\r\n">>), %% empty line + keepalive_nl_loop(Client3, URL, N - 1). multipart(Config) -> - Url = build_url("/multipart", Config), + Client = ?config(client, Config), Body = << "This is a preamble." "\r\n--OHai\r\nX-Name:answer\r\n\r\n42" @@ -269,21 +504,18 @@ multipart(Config) -> "\r\n--OHai--" "This is an epiloque." >>, - Request = {Url, [], "multipart/x-makes-no-sense; boundary=OHai", Body}, - {ok, {{"HTTP/1.1", 200, "OK"}, _Headers, Response}} = - httpc:request(post, Request, [], [{body_format, binary}]), - Parts = binary_to_term(Response), + {ok, Client2} = cowboy_client:request(<<"POST">>, + build_url("/multipart", Config), + [{<<"content-type">>, <<"multipart/x-makes-no-sense; boundary=OHai">>}], + Body, Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {ok, RespBody, _} = cowboy_client:response_body(Client3), + Parts = binary_to_term(RespBody), Parts = [ {[{<<"X-Name">>, <<"answer">>}], <<"42">>}, {[{'Server', <<"Cowboy">>}], <<"It rocks!\r\n">>} ]. -nc_rand(Config) -> - nc_reqs(Config, "/dev/urandom"). - -nc_zero(Config) -> - nc_reqs(Config, "/dev/zero"). - nc_reqs(Config, Input) -> Cat = os:find_executable("cat"), Nc = os:find_executable("nc"), @@ -295,235 +527,207 @@ nc_reqs(Config, Input) -> _Good -> %% Throw garbage at the server then check if it's still up. {port, Port} = lists:keyfind(port, 1, Config), - [nc_run_req(Port, Input) || _N <- lists:seq(1, 100)], - Packet = "GET / HTTP/1.0\r\nHost: localhost\r\n\r\n", - {Packet, 200} = raw_req(Packet, Config) + StrPort = integer_to_list(Port), + [os:cmd("cat " ++ Input ++ " | nc localhost " ++ StrPort) + || _ <- lists:seq(1, 100)], + 200 = quick_get("/", Config) end. -nc_run_req(Port, Input) -> - os:cmd("cat " ++ Input ++ " | nc localhost " ++ integer_to_list(Port)). - -pipeline(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, - "GET / HTTP/1.1\r\nHost: localhost\r\nConnection: keep-alive\r\n\r\n" - "GET / HTTP/1.1\r\nHost: localhost\r\nConnection: keep-alive\r\n\r\n" - "GET / HTTP/1.1\r\nHost: localhost\r\nConnection: keep-alive\r\n\r\n" - "GET / HTTP/1.1\r\nHost: localhost\r\nConnection: keep-alive\r\n\r\n" - "GET / HTTP/1.1\r\nHost: localhost\r\nConnection: close\r\n\r\n"), - Data = pipeline_recv(Socket, <<>>), - Reqs = binary:split(Data, << "\r\n\r\nhttp_handler" >>, [global, trim]), - 5 = length(Reqs), - pipeline_check(Reqs). - -pipeline_check([]) -> - ok; -pipeline_check([Req|Tail]) -> - << "HTTP/1.1 200", _Rest/bits >> = Req, - pipeline_check(Tail). - -pipeline_recv(Socket, SoFar) -> - case gen_tcp:recv(Socket, 0, 6000) of - {ok, Data} -> - pipeline_recv(Socket, << SoFar/binary, Data/binary >>); - {error, closed} -> - ok = gen_tcp:close(Socket), - SoFar - end. - -raw_req(Packet, Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, Packet), - Res = case gen_tcp:recv(Socket, 0, 6000) of - {ok, << "HTTP/1.1 ", Str:24/bits, _Rest/bits >>} -> - list_to_integer(binary_to_list(Str)); - {error, Reason} -> - Reason - end, - gen_tcp:close(Socket), - {Packet, Res}. - -%% Send a raw request. Return the response code and the full response. -raw_resp(Request, Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - Transport = case ?config(scheme, Config) of - "http" -> gen_tcp; - "https" -> ssl - end, - {ok, Socket} = Transport:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = Transport:send(Socket, Request), - {StatusCode, Response} = case recv_loop(Transport, Socket, <<>>) of - {ok, << "HTTP/1.1 ", Str:24/bits, _Rest/bits >> = Bin} -> - {list_to_integer(binary_to_list(Str)), Bin}; - {ok, Bin} -> - {badresp, Bin}; - {error, Reason} -> - {Reason, <<>>} - end, - Transport:close(Socket), - {Response, StatusCode}. - -recv_loop(Transport, Socket, Acc) -> - case Transport:recv(Socket, 0, 6000) of - {ok, Data} -> - recv_loop(Transport, Socket, <<Acc/binary, Data/binary>>); - {error, closed} -> - ok = Transport:close(Socket), - {ok, Acc}; - {error, Reason} -> - {error, Reason} - end. - - - -raw(Config) -> - Huge = [$0 || _N <- lists:seq(1, 5000)], - Tests = [ - {"\r\n\r\n\r\n\r\n\r\nGET / HTTP/1.1\r\nHost: localhost\r\n\r\n", 200}, - {"\n", 400}, - {"Garbage\r\n\r\n", 400}, - {"\r\n\r\n\r\n\r\n\r\n\r\n", 400}, - {"GET / HTTP/1.1\r\nHost: dev-extend.eu\r\n\r\n", 400}, - {"", closed}, - {"\r\n", closed}, - {"\r\n\r\n", closed}, - {"GET / HTTP/1.1", closed}, - {"GET / HTTP/1.1\r\n", 408}, - {"GET / HTTP/1.1\r\nHost: localhost", 408}, - {"GET / HTTP/1.1\r\nHost: localhost\r\n", 408}, - {"GET / HTTP/1.1\r\nHost: localhost\r\n\r", 408}, - {"GET http://proxy/ HTTP/1.1\r\n\r\n", 400}, - {"GET http://proxy/ HTTP/1.1\r\nHost: localhost\r\n\r\n", 200}, - {"GET / HTTP/1.2\r\nHost: localhost\r\n\r\n", 505}, - {"GET /init_shutdown HTTP/1.1\r\nHost: localhost\r\n\r\n", 666}, - {"GET /long_polling HTTP/1.1\r\nHost: localhost\r\n\r\n", 102}, - {Huge, 413}, - {"GET / HTTP/1.1\r\n" ++ Huge, 413} - ], - [{Packet, StatusCode} = raw_req(Packet, Config) - || {Packet, StatusCode} <- Tests]. - -set_resp_header(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, "GET /set_resp/header HTTP/1.1\r\n" - "Host: localhost\r\nConnection: close\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_, _} = binary:match(Data, <<"Vary: Accept">>), - {_, _} = binary:match(Data, <<"Set-Cookie: ">>). +nc_rand(Config) -> + nc_reqs(Config, "/dev/urandom"). -set_resp_overwrite(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, "GET /set_resp/overwrite HTTP/1.1\r\n" - "Host: localhost\r\nConnection: close\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_Start, _Length} = binary:match(Data, <<"Server: DesireDrive/1.0">>). +nc_zero(Config) -> + nc_reqs(Config, "/dev/zero"). -set_resp_body(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, "GET /set_resp/body HTTP/1.1\r\n" - "Host: localhost\r\nConnection: close\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_Start, _Length} = binary:match(Data, <<"\r\n\r\n" - "A flameless dance does not equal a cycle">>). - -response_as_req(Config) -> - Packet = -"HTTP/1.0 302 Found -Location: http://www.google.co.il/ -Cache-Control: private -Content-Type: text/html; charset=UTF-8 -Set-Cookie: PREF=ID=568f67013d4a7afa:FF=0:TM=1323014101:LM=1323014101:S=XqctDWC65MzKT0zC; expires=Tue, 03-Dec-2013 15:55:01 GMT; path=/; domain=.google.com -Date: Sun, 04 Dec 2011 15:55:01 GMT -Server: gws -Content-Length: 221 -X-XSS-Protection: 1; mode=block -X-Frame-Options: SAMEORIGIN +onrequest(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/", Config), Client), + {ok, 200, Headers, Client3} = cowboy_client:response(Client2), + {<<"server">>, <<"Serenity">>} = lists:keyfind(<<"server">>, 1, Headers), + {ok, <<"http_handler">>, _} = cowboy_client:response_body(Client3). -<HTML><HEAD><meta http-equiv=\"content-type\" content=\"text/html;charset=utf-8\"> -<TITLE>302 Moved</TITLE></HEAD><BODY> -<H1>302 Moved</H1> -The document has moved -<A HREF=\"http://www.google.co.il/\">here</A>. -</BODY></HTML>", - {Packet, 400} = raw_req(Packet, Config). +onrequest_reply(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/?reply=1", Config), Client), + {ok, 200, Headers, Client3} = cowboy_client:response(Client2), + {<<"server">>, <<"Cowboy">>} = lists:keyfind(<<"server">>, 1, Headers), + {ok, <<"replied!">>, _} = cowboy_client:response_body(Client3). + +%% Hook for the above onrequest tests. +onrequest_hook(Req) -> + case cowboy_http_req:qs_val(<<"reply">>, Req) of + {undefined, Req2} -> + {ok, Req3} = cowboy_http_req:set_resp_header( + 'Server', <<"Serenity">>, Req2), + Req3; + {_, Req2} -> + {ok, Req3} = cowboy_http_req:reply( + 200, [], <<"replied!">>, Req2), + Req3 + end. -stream_body_set_resp(Config) -> - {Packet, 200} = raw_resp( - "GET /stream_body/set_resp HTTP/1.1\r\n" - "Host: localhost\r\nConnection: close\r\n\r\n", Config), - {_Start, _Length} = binary:match(Packet, <<"stream_body_set_resp">>). +pipeline(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/", Config), Client), + {ok, Client3} = cowboy_client:request(<<"GET">>, + build_url("/", Config), Client2), + {ok, Client4} = cowboy_client:request(<<"GET">>, + build_url("/", Config), Client3), + {ok, Client5} = cowboy_client:request(<<"GET">>, + build_url("/", Config), Client4), + {ok, Client6} = cowboy_client:request(<<"GET">>, + build_url("/", Config), [{<<"connection">>, <<"close">>}], Client5), + {ok, 200, _, Client7} = cowboy_client:response(Client6), + {ok, 200, _, Client8} = cowboy_client:response(Client7), + {ok, 200, _, Client9} = cowboy_client:response(Client8), + {ok, 200, _, Client10} = cowboy_client:response(Client9), + {ok, 200, _, Client11} = cowboy_client:response(Client10), + {error, closed} = cowboy_client:response(Client11). -static_mimetypes_function(Config) -> - TestURL = build_url("/static_mimetypes_function/test.html", Config), - {ok, {{"HTTP/1.1", 200, "OK"}, Headers1, "test.html\n"}} = - httpc:request(TestURL), - "text/html" = ?config("content-type", Headers1). +rest_keepalive(Config) -> + Client = ?config(client, Config), + URL = build_url("/simple", Config), + ok = rest_keepalive_loop(Client, URL, 10). -handler_errors(Config) -> - Request = fun(Case) -> - raw_resp(["GET /handler_errors?case=", Case, " HTTP/1.1\r\n", - "Host: localhost\r\n\r\n"], Config) end, +rest_keepalive_loop(_, _, 0) -> + ok; +rest_keepalive_loop(Client, URL, N) -> + Headers = [{<<"connection">>, <<"keep-alive">>}], + {ok, Client2} = cowboy_client:request(<<"GET">>, URL, Headers, Client), + {ok, 200, RespHeaders, Client3} = cowboy_client:response(Client2), + {<<"connection">>, <<"keep-alive">>} + = lists:keyfind(<<"connection">>, 1, RespHeaders), + rest_keepalive_loop(Client3, URL, N - 1). - {_Packet1, 500} = Request("init_before_reply"), +rest_keepalive_post(Config) -> + Client = ?config(client, Config), + ok = rest_keepalive_post_loop(Config, Client, forbidden_post, 10). - {Packet2, 200} = Request("init_after_reply"), - nomatch = binary:match(Packet2, <<"HTTP/1.1 500">>), +rest_keepalive_post_loop(_, _, _, 0) -> + ok; +rest_keepalive_post_loop(Config, Client, simple_post, N) -> + Headers = [ + {<<"connection">>, <<"keep-alive">>}, + {<<"content-type">>, <<"text/plain">>} + ], + {ok, Client2} = cowboy_client:request(<<"POST">>, + build_url("/simple_post", Config), Headers, "12345", Client), + {ok, 303, RespHeaders, Client3} = cowboy_client:response(Client2), + {<<"connection">>, <<"keep-alive">>} + = lists:keyfind(<<"connection">>, 1, RespHeaders), + rest_keepalive_post_loop(Config, Client3, forbidden_post, N - 1); +rest_keepalive_post_loop(Config, Client, forbidden_post, N) -> + Headers = [ + {<<"connection">>, <<"keep-alive">>}, + {<<"content-type">>, <<"text/plain">>} + ], + {ok, Client2} = cowboy_client:request(<<"POST">>, + build_url("/forbidden_post", Config), Headers, "12345", Client), + {ok, 403, RespHeaders, Client3} = cowboy_client:response(Client2), + {<<"connection">>, <<"keep-alive">>} + = lists:keyfind(<<"connection">>, 1, RespHeaders), + rest_keepalive_post_loop(Config, Client3, simple_post, N - 1). - {Packet3, 200} = Request("init_reply_handle_error"), - nomatch = binary:match(Packet3, <<"HTTP/1.1 500">>), +rest_nodelete(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"DELETE">>, + build_url("/nodelete", Config), Client), + {ok, 500, _, _} = cowboy_client:response(Client2). + +rest_resource_get_etag(Config, Type) -> + rest_resource_get_etag(Config, Type, []). + +rest_resource_get_etag(Config, Type, Headers) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/resetags?type=" ++ Type, Config), Headers, Client), + {ok, Status, RespHeaders, _} = cowboy_client:response(Client2), + case lists:keyfind(<<"etag">>, 1, RespHeaders) of + false -> {Status, false}; + {<<"etag">>, ETag} -> {Status, ETag} + end. - {_Packet4, 500} = Request("handle_before_reply"), +rest_resource_etags(Config) -> + Tests = [ + {200, <<"W/\"etag-header-value\"">>, "tuple-weak"}, + {200, <<"\"etag-header-value\"">>, "tuple-strong"}, + {200, <<"W/\"etag-header-value\"">>, "binary-weak-quoted"}, + {200, <<"\"etag-header-value\"">>, "binary-strong-quoted"}, + {500, false, "binary-strong-unquoted"}, + {500, false, "binary-weak-unquoted"} + ], + _ = [{Status, ETag, Type} = begin + {Ret, RespETag} = rest_resource_get_etag(Config, Type), + {Ret, RespETag, Type} + end || {Status, ETag, Type} <- Tests]. - {Packet5, 200} = Request("handle_after_reply"), - nomatch = binary:match(Packet5, <<"HTTP/1.1 500">>), +rest_resource_etags_if_none_match(Config) -> + Tests = [ + {304, <<"W/\"etag-header-value\"">>, "tuple-weak"}, + {304, <<"\"etag-header-value\"">>, "tuple-strong"}, + {304, <<"W/\"etag-header-value\"">>, "binary-weak-quoted"}, + {304, <<"\"etag-header-value\"">>, "binary-strong-quoted"} + ], + _ = [{Status, Type} = begin + {Ret, _} = rest_resource_get_etag(Config, Type, + [{<<"if-none-match">>, ETag}]), + {Ret, Type} + end || {Status, ETag, Type} <- Tests]. - {Packet6, 200} = raw_resp([ - "GET / HTTP/1.1\r\n", - "Host: localhost\r\n", - "Connection: keep-alive\r\n\r\n", - "GET /handler_errors?case=handle_after_reply\r\n", - "Host: localhost\r\n\r\n"], Config), - nomatch = binary:match(Packet6, <<"HTTP/1.1 500">>), +set_resp_body(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/set_resp/body", Config), Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {ok, <<"A flameless dance does not equal a cycle">>, _} + = cowboy_client:response_body(Client3). - {Packet7, 200} = raw_resp([ - "GET / HTTP/1.1\r\n", - "Host: localhost\r\n", - "Connection: keep-alive\r\n\r\n", - "GET /handler_errors?case=handle_before_reply HTTP/1.1\r\n", - "Host: localhost\r\n\r\n"], Config), - {{_, _}, _} = {binary:match(Packet7, <<"HTTP/1.1 500">>), Packet7}, +set_resp_header(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/set_resp/header", Config), Client), + {ok, 200, Headers, _} = cowboy_client:response(Client2), + {<<"vary">>, <<"Accept">>} = lists:keyfind(<<"vary">>, 1, Headers), + {<<"set-cookie">>, _} = lists:keyfind(<<"set-cookie">>, 1, Headers). - done. +set_resp_overwrite(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/set_resp/overwrite", Config), Client), + {ok, 200, Headers, _} = cowboy_client:response(Client2), + {<<"server">>, <<"DesireDrive/1.0">>} + = lists:keyfind(<<"server">>, 1, Headers). static_attribute_etag(Config) -> - TestURL = build_url("/static_attribute_etag/test.html", Config), - {ok, {{"HTTP/1.1", 200, "OK"}, Headers1, "test.html\n"}} = - httpc:request(TestURL), - false = ?config("etag", Headers1) =:= undefined, - {ok, {{"HTTP/1.1", 200, "OK"}, Headers2, "test.html\n"}} = - httpc:request(TestURL), - true = ?config("etag", Headers1) =:= ?config("etag", Headers2). + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/static_attribute_etag/test.html", Config), Client), + {ok, Client3} = cowboy_client:request(<<"GET">>, + build_url("/static_attribute_etag/test.html", Config), Client2), + {ok, 200, Headers1, Client4} = cowboy_client:response(Client3), + {ok, 200, Headers2, _} = cowboy_client:response(Client4), + {<<"etag">>, ETag1} = lists:keyfind(<<"etag">>, 1, Headers1), + {<<"etag">>, ETag2} = lists:keyfind(<<"etag">>, 1, Headers2), + false = ETag1 =:= undefined, + ETag1 = ETag2. static_function_etag(Config) -> - TestURL = build_url("/static_function_etag/test.html", Config), - {ok, {{"HTTP/1.1", 200, "OK"}, Headers1, "test.html\n"}} = - httpc:request(TestURL), - false = ?config("etag", Headers1) =:= undefined, - {ok, {{"HTTP/1.1", 200, "OK"}, Headers2, "test.html\n"}} = - httpc:request(TestURL), - true = ?config("etag", Headers1) =:= ?config("etag", Headers2). - + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/static_function_etag/test.html", Config), Client), + {ok, Client3} = cowboy_client:request(<<"GET">>, + build_url("/static_function_etag/test.html", Config), Client2), + {ok, 200, Headers1, Client4} = cowboy_client:response(Client3), + {ok, 200, Headers2, _} = cowboy_client:response(Client4), + {<<"etag">>, ETag1} = lists:keyfind(<<"etag">>, 1, Headers1), + {<<"etag">>, ETag2} = lists:keyfind(<<"etag">>, 1, Headers2), + false = ETag1 =:= undefined, + ETag1 = ETag2. + +%% Callback function for generating the ETag for the above test. static_function_etag(Arguments, etag_data) -> {_, Filepath} = lists:keyfind(filepath, 1, Arguments), {_, _Filesize} = lists:keyfind(filesize, 1, Arguments), @@ -533,260 +737,68 @@ static_function_etag(Arguments, etag_data) -> [Checksum|_] = string:tokens(os:cmd(ChecksumCommand), " "), {strong, iolist_to_binary(Checksum)}. -te_identity(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - Body = list_to_binary(io_lib:format("~p", [lists:seq(1, 100)])), - StrLen = integer_to_list(byte_size(Body)), - ok = gen_tcp:send(Socket, ["GET /echo/body HTTP/1.1\r\n" - "Host: localhost\r\nConnection: close\r\n" - "Content-Length: ", StrLen, "\r\n\r\n", Body]), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_, _} = binary:match(Data, Body). +static_mimetypes_function(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/static_mimetypes_function/test.html", Config), Client), + {ok, 200, Headers, _} = cowboy_client:response(Client2), + {<<"content-type">>, <<"text/html">>} + = lists:keyfind(<<"content-type">>, 1, Headers). + +static_test_file(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/static/test_file", Config), Client), + {ok, 200, Headers, _} = cowboy_client:response(Client2), + {<<"content-type">>, <<"application/octet-stream">>} + = lists:keyfind(<<"content-type">>, 1, Headers). + +static_test_file_css(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/static/test_file.css", Config), Client), + {ok, 200, Headers, _} = cowboy_client:response(Client2), + {<<"content-type">>, <<"text/css">>} + = lists:keyfind(<<"content-type">>, 1, Headers). + +stream_body_set_resp(Config) -> + Client = ?config(client, Config), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/stream_body/set_resp", Config), Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {ok, <<"stream_body_set_resp">>, _} + = cowboy_client:response_body(Client3). te_chunked(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), + Client = ?config(client, Config), Body = list_to_binary(io_lib:format("~p", [lists:seq(1, 100)])), Chunks = body_to_chunks(50, Body, []), - ok = gen_tcp:send(Socket, ["GET /echo/body HTTP/1.1\r\n" - "Host: localhost\r\nConnection: close\r\n" - "Transfer-Encoding: chunked\r\n\r\n", Chunks]), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_, _} = binary:match(Data, Body). + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/echo/body", Config), + [{<<"transfer-encoding">>, <<"chunked">>}], + Chunks, Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {ok, Body, _} = cowboy_client:response_body(Client3). te_chunked_delayed(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), + Client = ?config(client, Config), Body = list_to_binary(io_lib:format("~p", [lists:seq(1, 100)])), Chunks = body_to_chunks(50, Body, []), - ok = gen_tcp:send(Socket, ["GET /echo/body HTTP/1.1\r\n" - "Host: localhost\r\nConnection: close\r\n" - "Transfer-Encoding: chunked\r\n\r\n"]), - _ = [begin ok = gen_tcp:send(Socket, Chunk), ok = timer:sleep(10) end - || Chunk <- Chunks], - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_, _} = binary:match(Data, Body). + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/echo/body", Config), + [{<<"transfer-encoding">>, <<"chunked">>}], Client), + {ok, Transport, Socket} = cowboy_client:transport(Client2), + _ = [begin + ok = Transport:send(Socket, Chunk), + ok = timer:sleep(10) + end || Chunk <- Chunks], + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {ok, Body, _} = cowboy_client:response_body(Client3). -body_to_chunks(_, <<>>, Acc) -> - lists:reverse([<<"0\r\n\r\n">>|Acc]); -body_to_chunks(ChunkSize, Body, Acc) -> - BodySize = byte_size(Body), - ChunkSize2 = case BodySize < ChunkSize of - true -> BodySize; - false -> ChunkSize - end, - << Chunk:ChunkSize2/binary, Rest/binary >> = Body, - ChunkSizeBin = list_to_binary(integer_to_list(ChunkSize2, 16)), - body_to_chunks(ChunkSize, Rest, - [<< ChunkSizeBin/binary, "\r\n", Chunk/binary, "\r\n" >>|Acc]). - -%% http and https. - -build_url(Path, Config) -> - {scheme, Scheme} = lists:keyfind(scheme, 1, Config), - {port, Port} = lists:keyfind(port, 1, Config), - Scheme ++ "://localhost:" ++ integer_to_list(Port) ++ Path. - -http_200(Config) -> - {ok, {{"HTTP/1.1", 200, "OK"}, _Headers, "http_handler"}} = - httpc:request(build_url("/", Config)). - -http_404(Config) -> - {ok, {{"HTTP/1.1", 404, "Not Found"}, _Headers, _Body}} = - httpc:request(build_url("/not/found", Config)). - -file_200(Config) -> - {ok, {{"HTTP/1.1", 200, "OK"}, Headers, "test_file\n"}} = - httpc:request(build_url("/static/test_file", Config)), - "application/octet-stream" = ?config("content-type", Headers), - - {ok, {{"HTTP/1.1", 200, "OK"}, Headers1, "test_file.css\n"}} = - httpc:request(build_url("/static/test_file.css", Config)), - "text/css" = ?config("content-type", Headers1). - -file_403(Config) -> - {ok, {{"HTTP/1.1", 403, "Forbidden"}, _Headers, _Body}} = - httpc:request(build_url("/static/test_noread", Config)). - -dir_403(Config) -> - {ok, {{"HTTP/1.1", 403, "Forbidden"}, _Headers, _Body}} = - httpc:request(build_url("/static/test_dir", Config)), - {ok, {{"HTTP/1.1", 403, "Forbidden"}, _Headers, _Body}} = - httpc:request(build_url("/static/test_dir/", Config)). - -file_404(Config) -> - {ok, {{"HTTP/1.1", 404, "Not Found"}, _Headers, _Body}} = - httpc:request(build_url("/static/not_found", Config)). - -file_400(Config) -> - {ok, {{"HTTP/1.1", 400, "Bad Request"}, _Headers, _Body}} = - httpc:request(build_url("/static/%2f", Config)), - {ok, {{"HTTP/1.1", 400, "Bad Request"}, _Headers1, _Body1}} = - httpc:request(build_url("/static/%2e", Config)), - {ok, {{"HTTP/1.1", 400, "Bad Request"}, _Headers2, _Body2}} = - httpc:request(build_url("/static/%2e%2e", Config)). - -%% misc. - -http_10_hostless(Config) -> - Packet = "GET / HTTP/1.0\r\n\r\n", - {Packet, 200} = raw_req(Packet, Config). - -http_10_chunkless(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - Packet = "GET /chunked_response HTTP/1.0\r\nContent-Length: 0\r\n\r\n", - ok = gen_tcp:send(Socket, Packet), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - nomatch = binary:match(Data, <<"Transfer-Encoding">>), - {_, _} = binary:match(Data, <<"chunked_handler\r\nworks fine!">>), - ok = gen_tcp:close(Socket). - -%% rest. - -rest_simple(Config) -> - Packet = "GET /simple HTTP/1.1\r\nHost: localhost\r\n\r\n", - {Packet, 200} = raw_req(Packet, Config). - -rest_keepalive(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = rest_keepalive_loop(Socket, 100), - ok = gen_tcp:close(Socket). - -rest_keepalive_loop(_Socket, 0) -> - ok; -rest_keepalive_loop(Socket, N) -> - ok = gen_tcp:send(Socket, "GET /simple HTTP/1.1\r\n" - "Host: localhost\r\nConnection: keep-alive\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {0, 12} = binary:match(Data, <<"HTTP/1.1 200">>), - nomatch = binary:match(Data, <<"Connection: close">>), - rest_keepalive_loop(Socket, N - 1). - -rest_keepalive_post(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = rest_keepalive_post_loop(Socket, 10, forbidden_post), - ok = gen_tcp:close(Socket). - -rest_keepalive_post_loop(_Socket, 0, _) -> - ok; -rest_keepalive_post_loop(Socket, N, simple_post) -> - ok = gen_tcp:send(Socket, "POST /simple_post HTTP/1.1\r\n" - "Host: localhost\r\nConnection: keep-alive\r\n" - "Content-Length: 5\r\nContent-Type: text/plain\r\n\r\n12345"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {0, 12} = binary:match(Data, <<"HTTP/1.1 303">>), - nomatch = binary:match(Data, <<"Connection: close">>), - rest_keepalive_post_loop(Socket, N - 1, forbidden_post); -rest_keepalive_post_loop(Socket, N, forbidden_post) -> - ok = gen_tcp:send(Socket, "POST /forbidden_post HTTP/1.1\r\n" - "Host: localhost\r\nConnection: keep-alive\r\n" - "Content-Length: 5\r\nContent-Type: text/plain\r\n\r\n12345"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {0, 12} = binary:match(Data, <<"HTTP/1.1 403">>), - nomatch = binary:match(Data, <<"Connection: close">>), - rest_keepalive_post_loop(Socket, N - 1, simple_post). - -rest_nodelete(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - Request = "DELETE /nodelete HTTP/1.1\r\nHost: localhost\r\n\r\n", - ok = gen_tcp:send(Socket, Request), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {0, 12} = binary:match(Data, <<"HTTP/1.1 500">>), - ok = gen_tcp:close(Socket). - -rest_resource_etags(Config) -> - %% The Etag header should be set to the return value of generate_etag/2. - fun() -> - %% Correct return values from generate_etag/2. - {Packet1, 200} = raw_resp([ - "GET /resetags?type=tuple-weak HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", "\r\n"], Config), - {_,_} = binary:match(Packet1, <<"ETag: W/\"etag-header-value\"\r\n">>), - {Packet2, 200} = raw_resp([ - "GET /resetags?type=tuple-strong HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", "\r\n"], Config), - {_,_} = binary:match(Packet2, <<"ETag: \"etag-header-value\"\r\n">>), - %% Backwards compatible return values from generate_etag/2. - {Packet3, 200} = raw_resp([ - "GET /resetags?type=binary-weak-quoted HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", "\r\n"], Config), - {_,_} = binary:match(Packet3, <<"ETag: W/\"etag-header-value\"\r\n">>), - {Packet4, 200} = raw_resp([ - "GET /resetags?type=binary-strong-quoted HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", "\r\n"], Config), - {_,_} = binary:match(Packet4, <<"ETag: \"etag-header-value\"\r\n">>), - %% Invalid return values from generate_etag/2. - {_Packet5, 500} = raw_resp([ - "GET /resetags?type=binary-strong-unquoted HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", "\r\n"], Config), - {_Packet6, 500} = raw_resp([ - "GET /resetags?type=binary-weak-unquoted HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", "\r\n"], Config) - end(), - - %% The return value of generate_etag/2 should match the request header. - fun() -> - %% Correct return values from generate_etag/2. - {_Packet1, 304} = raw_resp([ - "GET /resetags?type=tuple-weak HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", - "If-None-Match: W/\"etag-header-value\"\r\n", "\r\n"], Config), - {_Packet2, 304} = raw_resp([ - "GET /resetags?type=tuple-strong HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", - "If-None-Match: \"etag-header-value\"\r\n", "\r\n"], Config), - %% Backwards compatible return values from generate_etag/2. - {_Packet3, 304} = raw_resp([ - "GET /resetags?type=binary-weak-quoted HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", - "If-None-Match: W/\"etag-header-value\"\r\n", "\r\n"], Config), - {_Packet4, 304} = raw_resp([ - "GET /resetags?type=binary-strong-quoted HTTP/1.1\r\n", - "Host: localhost\r\n", "Connection: close\r\n", - "If-None-Match: \"etag-header-value\"\r\n", "\r\n"], Config) - end(). - -onrequest(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, "GET / HTTP/1.1\r\nHost: localhost\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_, _} = binary:match(Data, <<"Server: Serenity">>), - {_, _} = binary:match(Data, <<"http_handler">>), - gen_tcp:close(Socket). - -onrequest_reply(Config) -> - {port, Port} = lists:keyfind(port, 1, Config), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), - ok = gen_tcp:send(Socket, "GET /?reply=1 HTTP/1.1\r\nHost: localhost\r\n\r\n"), - {ok, Data} = gen_tcp:recv(Socket, 0, 6000), - {_, _} = binary:match(Data, <<"Server: Cowboy">>), - nomatch = binary:match(Data, <<"http_handler">>), - {_, _} = binary:match(Data, <<"replied!">>), - gen_tcp:close(Socket). - -onrequest_hook(Req) -> - case cowboy_http_req:qs_val(<<"reply">>, Req) of - {undefined, Req2} -> - {ok, Req3} = cowboy_http_req:set_resp_header( - 'Server', <<"Serenity">>, Req2), - Req3; - {_, Req2} -> - {ok, Req3} = cowboy_http_req:reply( - 200, [], <<"replied!">>, Req2), - Req3 - end. +te_identity(Config) -> + Client = ?config(client, Config), + Body = list_to_binary(io_lib:format("~p", [lists:seq(1, 100)])), + {ok, Client2} = cowboy_client:request(<<"GET">>, + build_url("/echo/body", Config), [], Body, Client), + {ok, 200, _, Client3} = cowboy_client:response(Client2), + {ok, Body, _} = cowboy_client:response_body(Client3). diff --git a/test/http_handler_long_polling.erl b/test/http_handler_long_polling.erl index e838619..8d08265 100644 --- a/test/http_handler_long_polling.erl +++ b/test/http_handler_long_polling.erl @@ -6,7 +6,7 @@ init({_Transport, http}, Req, _Opts) -> erlang:send_after(500, self(), timeout), - {loop, Req, 9, 5000, hibernate}. + {loop, Req, 5, 5000, hibernate}. handle(_Req, _State) -> exit(badarg). |