mirror of
https://github.com/ninenines/cowboy.git
synced 2025-07-14 20:30:23 +00:00
Add case for handling infinity for idle/request_timeout
Currently cowboy assumes that idle_timeout or request_timeout is a number and always starts timers. Similar situation takes place in case of preface_timeout for http2. This commit adds case for handling infinity as a timeout, allowing to not start mentioned timers.
This commit is contained in:
parent
f9092126fa
commit
2131709328
4 changed files with 79 additions and 7 deletions
|
@ -222,8 +222,10 @@ set_timeout(State0=#state{opts=Opts, streams=Streams}) ->
|
||||||
[] -> {request_timeout, 5000};
|
[] -> {request_timeout, 5000};
|
||||||
_ -> {idle_timeout, 60000}
|
_ -> {idle_timeout, 60000}
|
||||||
end,
|
end,
|
||||||
Timeout = maps:get(Name, Opts, Default),
|
TimerRef = case maps:get(Name, Opts, Default) of
|
||||||
TimerRef = erlang:start_timer(Timeout, self(), Name),
|
infinity -> undefined;
|
||||||
|
Timeout -> erlang:start_timer(Timeout, self(), Name)
|
||||||
|
end,
|
||||||
State#state{timer=TimerRef}.
|
State#state{timer=TimerRef}.
|
||||||
|
|
||||||
cancel_timeout(State=#state{timer=TimerRef}) ->
|
cancel_timeout(State=#state{timer=TimerRef}) ->
|
||||||
|
|
|
@ -122,7 +122,7 @@
|
||||||
%% is established and continues normally. An exception is when a HEADERS
|
%% is established and continues normally. An exception is when a HEADERS
|
||||||
%% frame is sent followed by CONTINUATION frames: no other frame can be
|
%% frame is sent followed by CONTINUATION frames: no other frame can be
|
||||||
%% sent in between.
|
%% sent in between.
|
||||||
parse_state = undefined :: {preface, sequence, reference()}
|
parse_state = undefined :: {preface, sequence, undefined | reference()}
|
||||||
| {preface, settings, reference()}
|
| {preface, settings, reference()}
|
||||||
| normal
|
| normal
|
||||||
| {continuation, cowboy_stream:streamid(), cowboy_stream:fin(), binary()},
|
| {continuation, cowboy_stream:streamid(), cowboy_stream:fin(), binary()},
|
||||||
|
@ -203,8 +203,10 @@ preface(#state{socket=Socket, transport=Transport, next_settings=Settings}) ->
|
||||||
Transport:send(Socket, cow_http2:settings(Settings)).
|
Transport:send(Socket, cow_http2:settings(Settings)).
|
||||||
|
|
||||||
preface_timeout(Opts) ->
|
preface_timeout(Opts) ->
|
||||||
PrefaceTimeout = maps:get(preface_timeout, Opts, 5000),
|
case maps:get(preface_timeout, Opts, 5000) of
|
||||||
erlang:start_timer(PrefaceTimeout, self(), preface_timeout).
|
infinity -> undefined;
|
||||||
|
PrefaceTimeout -> erlang:start_timer(PrefaceTimeout, self(), preface_timeout)
|
||||||
|
end.
|
||||||
|
|
||||||
%% @todo Add the timeout for last time since we heard of connection.
|
%% @todo Add the timeout for last time since we heard of connection.
|
||||||
before_loop(State, Buffer) ->
|
before_loop(State, Buffer) ->
|
||||||
|
@ -305,7 +307,10 @@ parse(State=#state{local_settings=#{max_frame_size := MaxFrameSize},
|
||||||
end.
|
end.
|
||||||
|
|
||||||
parse_settings_preface(State, Frame={settings, _}, Rest, TRef) ->
|
parse_settings_preface(State, Frame={settings, _}, Rest, TRef) ->
|
||||||
_ = erlang:cancel_timer(TRef, [{async, true}, {info, false}]),
|
ok = case TRef of
|
||||||
|
undefined -> ok;
|
||||||
|
_ -> erlang:cancel_timer(TRef, [{async, true}, {info, false}])
|
||||||
|
end,
|
||||||
parse(frame(State#state{parse_state=normal}, Frame), Rest);
|
parse(frame(State#state{parse_state=normal}, Frame), Rest);
|
||||||
parse_settings_preface(State, _, _, _) ->
|
parse_settings_preface(State, _, _, _) ->
|
||||||
terminate(State, {connection_error, protocol_error,
|
terminate(State, {connection_error, protocol_error,
|
||||||
|
|
|
@ -59,6 +59,23 @@ inactivity_timeout(Config) ->
|
||||||
{ok, << _:24, 7:8, _:72, 2:32 >>} = gen_tcp:recv(Socket, 17, 1000),
|
{ok, << _:24, 7:8, _:72, 2:32 >>} = gen_tcp:recv(Socket, 17, 1000),
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
preface_timeout_infinity(Config) ->
|
||||||
|
doc("Ensure infinity for preface_timeout is accepted"),
|
||||||
|
ProtoOpts = #{
|
||||||
|
env => #{dispatch => cowboy_router:compile(init_routes(Config))},
|
||||||
|
preface_timeout => infinity
|
||||||
|
},
|
||||||
|
{ok, Pid} = cowboy:start_clear(preface_timeout_infinity, [{port, 0}], ProtoOpts),
|
||||||
|
Ref = erlang:monitor(process, Pid),
|
||||||
|
Port = ranch:get_port(preface_timeout_infinity),
|
||||||
|
{ok, _} = do_handshake([{port, Port}|Config]),
|
||||||
|
receive
|
||||||
|
{'DOWN', Ref, process, Pid, Reason} ->
|
||||||
|
error(Reason)
|
||||||
|
after 1000 ->
|
||||||
|
cowboy:stop_listener(preface_timeout_infinity)
|
||||||
|
end.
|
||||||
|
|
||||||
resp_iolist_body(Config) ->
|
resp_iolist_body(Config) ->
|
||||||
doc("Regression test when response bodies are iolists that "
|
doc("Regression test when response bodies are iolists that "
|
||||||
"include improper lists, empty lists and empty binaries. "
|
"include improper lists, empty lists and empty binaries. "
|
||||||
|
|
|
@ -35,13 +35,16 @@ all() ->
|
||||||
{group, http_compress},
|
{group, http_compress},
|
||||||
{group, https_compress},
|
{group, https_compress},
|
||||||
{group, parse_host},
|
{group, parse_host},
|
||||||
|
{group, request_timeout_infinity},
|
||||||
|
{group, idle_timeout_infinity},
|
||||||
{group, set_env},
|
{group, set_env},
|
||||||
{group, router_compile}
|
{group, router_compile}
|
||||||
].
|
].
|
||||||
|
|
||||||
groups() ->
|
groups() ->
|
||||||
Tests = ct_helper:all(?MODULE) -- [
|
Tests = ct_helper:all(?MODULE) -- [
|
||||||
parse_host, set_env_dispatch, path_allow_colon
|
parse_host, set_env_dispatch, path_allow_colon,
|
||||||
|
request_timeout_infinity, idle_timeout_infinity
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
{http, [], Tests}, %% @todo parallel
|
{http, [], Tests}, %% @todo parallel
|
||||||
|
@ -51,6 +54,12 @@ groups() ->
|
||||||
{parse_host, [], [
|
{parse_host, [], [
|
||||||
parse_host
|
parse_host
|
||||||
]},
|
]},
|
||||||
|
{request_timeout_infinity, [], [
|
||||||
|
request_timeout_infinity
|
||||||
|
]},
|
||||||
|
{idle_timeout_infinity, [], [
|
||||||
|
idle_timeout_infinity
|
||||||
|
]},
|
||||||
{set_env, [], [
|
{set_env, [], [
|
||||||
set_env_dispatch
|
set_env_dispatch
|
||||||
]},
|
]},
|
||||||
|
@ -84,6 +93,22 @@ init_per_group(parse_host, Config) ->
|
||||||
}),
|
}),
|
||||||
Port = ranch:get_port(parse_host),
|
Port = ranch:get_port(parse_host),
|
||||||
[{type, tcp}, {protocol, http}, {port, Port}, {opts, []}|Config];
|
[{type, tcp}, {protocol, http}, {port, Port}, {opts, []}|Config];
|
||||||
|
init_per_group(request_timeout_infinity, Config) ->
|
||||||
|
Ref = request_timeout_infinity,
|
||||||
|
{ok, Pid} = cowboy:start_clear(Ref, [{port, 0}], #{
|
||||||
|
env => #{dispatch => init_dispatch(Config)},
|
||||||
|
request_timeout => infinity
|
||||||
|
}),
|
||||||
|
Port = ranch:get_port(request_timeout_infinity),
|
||||||
|
[{pid, Pid}, {ref, Ref}, {type, tcp}, {protocol, http}, {port, Port}, {opts, []}|Config];
|
||||||
|
init_per_group(idle_timeout_infinity, Config) ->
|
||||||
|
Ref = idle_timeout_infinity,
|
||||||
|
{ok, Pid} = cowboy:start_clear(Ref, [{port, 0}], #{
|
||||||
|
env => #{dispatch => init_dispatch(Config)},
|
||||||
|
idle_timeout => infinity
|
||||||
|
}),
|
||||||
|
Port = ranch:get_port(idle_timeout_infinity),
|
||||||
|
[{pid, Pid}, {ref, Ref}, {type, tcp}, {protocol, http}, {port, Port}, {opts, []}|Config];
|
||||||
init_per_group(set_env, Config) ->
|
init_per_group(set_env, Config) ->
|
||||||
{ok, _} = cowboy:start_clear(set_env, [{port, 0}], #{
|
{ok, _} = cowboy:start_clear(set_env, [{port, 0}], #{
|
||||||
env => #{dispatch => []}
|
env => #{dispatch => []}
|
||||||
|
@ -746,3 +771,26 @@ te_identity(Config) ->
|
||||||
{response, nofin, 200, _} = gun:await(ConnPid, Ref),
|
{response, nofin, 200, _} = gun:await(ConnPid, Ref),
|
||||||
{ok, Body} = gun:await_body(ConnPid, Ref),
|
{ok, Body} = gun:await_body(ConnPid, Ref),
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
request_timeout_infinity(Config) ->
|
||||||
|
Pid = config(pid, Config),
|
||||||
|
Ref = erlang:monitor(process, Pid),
|
||||||
|
_ = gun_open(Config),
|
||||||
|
receive
|
||||||
|
{'DOWN', Ref, process, Pid, Reason} ->
|
||||||
|
error(Reason)
|
||||||
|
after 1000 ->
|
||||||
|
ok
|
||||||
|
end.
|
||||||
|
|
||||||
|
idle_timeout_infinity(Config) ->
|
||||||
|
Pid = config(pid, Config),
|
||||||
|
Ref = erlang:monitor(process, Pid),
|
||||||
|
ConnPid = gun_open(Config),
|
||||||
|
gun:post(ConnPid, "/echo/body", [], <<"TEST">>),
|
||||||
|
receive
|
||||||
|
{'DOWN', Ref, process, Pid, Reason} ->
|
||||||
|
error(Reason)
|
||||||
|
after 1000 ->
|
||||||
|
ok
|
||||||
|
end.
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue