diff options
Diffstat (limited to 'test/spdy_server.erl')
-rw-r--r-- | test/spdy_server.erl | 101 |
1 files changed, 101 insertions, 0 deletions
diff --git a/test/spdy_server.erl b/test/spdy_server.erl new file mode 100644 index 0000000..e56c2da --- /dev/null +++ b/test/spdy_server.erl @@ -0,0 +1,101 @@ +%% Copyright (c) 2015, 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. + +-module(spdy_server). +-behaviour(gen_server). + +%% API. +-export([start_link/0]). +-export([stop/1]). + +%% gen_server. +-export([init/1]). +-export([handle_call/3]). +-export([handle_cast/2]). +-export([handle_info/2]). +-export([terminate/2]). +-export([code_change/3]). + +-type recording() :: [tuple()]. + +-record(state, { + owner = undefined :: pid(), + recording = [] :: recording(), + state_name = listen :: listen | record, + socket = undefined :: ssl:sslsocket(), + zinf = undefined :: zlib:zstream(), + buffer = <<>> :: binary() +}). + +%% API. + +-spec start_link() -> {ok, pid()}. +start_link() -> + {ok, Pid} = gen_server:start_link(?MODULE, [self()], []), + receive {port, Pid, Port} -> + {ok, Pid, Port} + after 5000 -> + exit(timeout) + end. + +-spec stop(pid()) -> recording(). +stop(Pid) -> + gen_server:call(Pid, stop). + +%% gen_server. + +init([Owner]) -> + Opts = ct_helper:get_certs_from_ets(), + {ok, LSocket} = ssl:listen(0, [binary, {active, false}, {nodelay, true}, + {next_protocols_advertised, [<<"spdy/3.1">>, <<"spdy/3">>]}|Opts]), + {ok, {_, Port}} = ssl:sockname(LSocket), + Owner ! {port, self(), Port}, + self() ! listen, + {ok, #state{owner=Owner, socket=LSocket}}. + +handle_call(stop, {Owner, _}, State=#state{owner=Owner, recording=Recording}) -> + {stop, normal, lists:reverse(Recording), State}; +handle_call(_Request, _From, State) -> + {reply, ignored, State}. + +handle_cast(_Msg, State) -> + {noreply, State}. + +handle_info(listen, State=#state{state_name=listen, socket=LSocket}) -> + {ok, CSocket} = ssl:transport_accept(LSocket, 5000), + ok = ssl:ssl_accept(CSocket, 5000), + ok = ssl:setopts(CSocket, [{active, once}]), + {noreply, State#state{state_name=record, socket=CSocket}}; +handle_info({ssl, Socket, Data}, State=#state{state_name=record, socket=Socket, buffer=Buffer}) -> + ok = ssl:setopts(Socket, [{active, once}]), + State2 = handle_data(<< Buffer/binary, Data/binary >>, State), + {noreply, State2}; +%% @todo ssl_closed ssl_error +handle_info(_Info, State) -> + {noreply, State}. + +handle_data(Data, State=#state{recording=Recording, zinf=Zinf}) -> + case cow_spdy:split(Data) of + {true, ParsedFrame, Rest} -> + Frame = cow_spdy:parse(ParsedFrame, Zinf), + handle_data(Rest, State#state{recording=[Frame|Recording]}); + false -> + State#state{buffer=Data} + end. + +terminate(_Reason, _State) -> + ok. + +code_change(_OldVsn, State, _Extra) -> + {ok, State}. |