dudeswave2/gun/test/gun_test.erl

134 lines
4.4 KiB
Erlang

%% Copyright (c) 2018-2023, Loïc Hoguin <essen@ninenines.eu>
%%
%% 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.
-module(gun_test).
-compile(export_all).
-compile(nowarn_export_all).
%% Cowboy listeners.
init_cowboy_tcp(Ref, ProtoOpts, Config) ->
{ok, _} = cowboy:start_clear(Ref, [{port, 0}], ProtoOpts),
[{ref, Ref}, {port, ranch:get_port(Ref)}|Config].
init_cowboy_tls(Ref, ProtoOpts, Config) ->
Opts = ct_helper:get_certs_from_ets(),
{ok, _} = cowboy:start_tls(Ref,
[{verify, verify_none}, {fail_if_no_peer_cert, false}]
++ Opts ++ [{port, 0}], ProtoOpts),
[{ref, Ref}, {port, ranch:get_port(Ref)}|Config].
%% Origin server helpers.
init_origin(Transport) ->
init_origin(Transport, http).
init_origin(Transport, Protocol) ->
init_origin(Transport, Protocol, fun loop_origin/4).
init_origin(Transport, Protocol, Fun) ->
Pid = spawn_link(?MODULE, init_origin, [self(), Transport, Protocol, Fun]),
Port = receive_from(Pid),
{ok, Pid, Port}.
init_origin(Parent, Transport, Protocol, Fun)
when Transport =:= tcp; Transport =:= tcp6 ->
InetOpt = case Transport of
tcp -> inet;
tcp6 -> inet6
end,
{ok, ListenSocket} = gen_tcp:listen(0, [binary, {active, false}, InetOpt]),
{ok, {_, Port}} = inet:sockname(ListenSocket),
Parent ! {self(), Port},
{ok, ClientSocket} = gen_tcp:accept(ListenSocket, 5000),
case Protocol of
http2 -> http2_handshake(ClientSocket, gen_tcp);
_ -> ok
end,
Parent ! {self(), handshake_completed},
Fun(Parent, ListenSocket, ClientSocket, gen_tcp);
init_origin(Parent, tls, Protocol, Fun) ->
Opts0 = ct_helper:get_certs_from_ets(),
Opts1 = case Protocol of
http2 -> [{alpn_preferred_protocols, [<<"h2">>]}|Opts0];
_ -> Opts0
end,
%% sni_hosts is necessary for SNI tests to succeed.
Opts = [{sni_hosts, [{net_adm:localhost(), []}]}|Opts1],
{ok, ListenSocket} = ssl:listen(0, [binary, {active, false},
{fail_if_no_peer_cert, false}|Opts]),
{ok, {_, Port}} = ssl:sockname(ListenSocket),
Parent ! {self(), Port},
{ok, ClientSocket0} = ssl:transport_accept(ListenSocket, 5000),
{ok, ClientSocket} = ssl:handshake(ClientSocket0, 5000),
case Protocol of
http2 ->
{ok, <<"h2">>} = ssl:negotiated_protocol(ClientSocket),
http2_handshake(ClientSocket, ssl);
_ ->
ok
end,
Parent ! {self(), handshake_completed},
Fun(Parent, ListenSocket, ClientSocket, ssl).
http2_handshake(Socket, Transport) ->
%% Send a valid preface.
ok = Transport:send(Socket, cow_http2:settings(#{})),
%% Receive the fixed sequence from the preface.
Preface = <<"PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n">>,
{ok, Preface} = Transport:recv(Socket, byte_size(Preface), 5000),
%% Receive the SETTINGS from the preface.
{ok, <<Len:24>>} = Transport:recv(Socket, 3, 5000),
{ok, <<4:8, 0:40, _:Len/binary>>} = Transport:recv(Socket, 6 + Len, 5000),
%% Receive the WINDOW_UPDATE sent with the preface.
{ok, <<4:24, 8:8, 0:40, _:32>>} = Transport:recv(Socket, 13, 5000),
%% Send the SETTINGS ack.
ok = Transport:send(Socket, cow_http2:settings_ack()),
%% Receive the SETTINGS ack.
{ok, <<0:24, 4:8, 1:8, 0:32>>} = Transport:recv(Socket, 9, 5000),
ok.
loop_origin(Parent, ListenSocket, ClientSocket, ClientTransport) ->
case ClientTransport:recv(ClientSocket, 0, 5000) of
{ok, Data} ->
Parent ! {self(), Data},
loop_origin(Parent, ListenSocket, ClientSocket, ClientTransport);
{error, closed} ->
ok
end.
%% Common helpers.
receive_from(Pid) ->
receive_from(Pid, 5000).
receive_from(Pid, Timeout) ->
receive
{Pid, Msg} ->
Msg
after Timeout ->
error(timeout)
end.
receive_all_from(Pid, Timeout) ->
receive_all_from(Pid, Timeout, <<>>).
receive_all_from(Pid, Timeout, Acc) ->
try
More = receive_from(Pid, Timeout),
receive_all_from(Pid, Timeout, <<Acc/binary, More/binary>>)
catch error:timeout ->
Acc
end.