%%
%% %CopyrightBegin%
%%
%% Copyright Ericsson AB 1998-2012. All Rights Reserved.
%%
%% The contents of this file are subject to the Erlang Public License,
%% Version 1.1, (the "License"); you may not use this file except in
%% compliance with the License. You should have received a copy of the
%% Erlang Public License along with this software. If not, it can be
%% retrieved online at http://www.erlang.org/.
%%
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%% the License for the specific language governing rights and limitations
%% under the License.
%%
%% %CopyrightEnd%
%%
-module(gen_tcp_misc_SUITE).
-include_lib("test_server/include/test_server.hrl").
%-compile(export_all).
-export([all/0, suite/0,groups/0,init_per_suite/1, end_per_suite/1,
init_per_group/2,end_per_group/2,
controlling_process/1, controlling_process_self/1,
no_accept/1, close_with_pending_output/1,
data_before_close/1, iter_max_socks/1, get_status/1,
passive_sockets/1, accept_closed_by_other_process/1,
init_per_testcase/2, end_per_testcase/2,
otp_3924/1, otp_3924_sender/4, closed_socket/1,
shutdown_active/1, shutdown_passive/1, shutdown_pending/1,
default_options/1, http_bad_packet/1,
busy_send/1, busy_disconnect_passive/1, busy_disconnect_active/1,
fill_sendq/1, partial_recv_and_close/1,
partial_recv_and_close_2/1,partial_recv_and_close_3/1,so_priority/1,
% Accept tests
primitive_accept/1,multi_accept_close_listen/1,accept_timeout/1,
accept_timeouts_in_order/1,accept_timeouts_in_order2/1,
accept_timeouts_in_order3/1,accept_timeouts_mixed/1,
killing_acceptor/1,killing_multi_acceptors/1,killing_multi_acceptors2/1,
several_accepts_in_one_go/1, accept_system_limit/1,
active_once_closed/1, send_timeout/1, send_timeout_active/1,
otp_7731/1, zombie_sockets/1, otp_7816/1, otp_8102/1, wrapping_oct/1,
otp_9389/1]).
%% Internal exports.
-export([sender/3, not_owner/1, passive_sockets_server/2, priority_server/1,
oct_acceptor/1,
otp_7731_server/1, zombie_server/2, do_iter_max_socks/2]).
init_per_testcase(_Func, Config) when is_list(Config) ->
Dog = test_server:timetrap(test_server:seconds(240)),
[{watchdog, Dog}|Config].
end_per_testcase(_Func, Config) ->
Dog = ?config(watchdog, Config),
test_server:timetrap_cancel(Dog).
suite() -> [{ct_hooks,[ts_install_cth]}].
all() ->
[controlling_process, controlling_process_self, no_accept,
close_with_pending_output, data_before_close,
iter_max_socks, passive_sockets,
accept_closed_by_other_process, otp_3924, closed_socket,
shutdown_active, shutdown_passive, shutdown_pending,
default_options, http_bad_packet, busy_send,
busy_disconnect_passive, busy_disconnect_active,
fill_sendq, partial_recv_and_close,
partial_recv_and_close_2, partial_recv_and_close_3,
so_priority, primitive_accept,
multi_accept_close_listen, accept_timeout,
accept_timeouts_in_order, accept_timeouts_in_order2,
accept_timeouts_in_order3, accept_timeouts_mixed,
killing_acceptor, killing_multi_acceptors,
killing_multi_acceptors2, several_accepts_in_one_go, accept_system_limit,
active_once_closed, send_timeout, send_timeout_active, otp_7731,
wrapping_oct,
zombie_sockets, otp_7816, otp_8102, otp_9389].
groups() ->
[].
init_per_suite(Config) ->
Config.
end_per_suite(_Config) ->
ok.
init_per_group(_GroupName, Config) ->
Config.
end_per_group(_GroupName, Config) ->
Config.
default_options(doc) ->
["Tests kernel application variables inet_default_listen_options and "
"inet_default_connect_options"];
default_options(suite) ->
[];
default_options(Config) when is_list(Config) ->
%% First check the delay_send option
?line {true,true,true}=do_delay_send_1(),
?line {false,false,false}=do_delay_send_2(),
?line {true,false,false}=do_delay_send_3(),
?line {false,false,false}=do_delay_send_4(),
?line {false,false,false}=do_delay_send_5(),
?line {false,true,true}=do_delay_send_6(),
%% Now lets start some nodes with different combinations of options:
?line {true,true,true} = do_delay_on_other_node("",
fun do_delay_send_1/0),
?line {true,false,false} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"[{delay_send,true}]\"",
fun do_delay_send_2/0),
?line {false,true,true} =
do_delay_on_other_node("-kernel inet_default_listen_options "
"\"[{delay_send,true}]\"",
fun do_delay_send_2/0),
?line {true,true,true} =
do_delay_on_other_node("-kernel inet_default_listen_options "
"\"[{delay_send,true}]\"",
fun do_delay_send_3/0),
?line {true,true,true} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"[{delay_send,true}]\"",
fun do_delay_send_6/0),
?line {false,false,false} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"[{delay_send,true}]\"",
fun do_delay_send_5/0),
?line {false,true,true} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"[{delay_send,true}]\" "
"-kernel inet_default_listen_options "
"\"[{delay_send,true}]\"",
fun do_delay_send_5/0),
?line {true,false,false} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"[{delay_send,true}]\" "
"-kernel inet_default_listen_options "
"\"[{delay_send,true}]\"",
fun do_delay_send_4/0),
?line {true,true,true} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"{delay_send,true}\" "
"-kernel inet_default_listen_options "
"\"{delay_send,true}\"",
fun do_delay_send_2/0),
%% Active is to dangerous and is supressed
?line {true,true,true} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"{active,false}\" "
"-kernel inet_default_listen_options "
"\"{active,false}\"",
fun do_delay_send_7/0),
?line {true,true,true} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"[{active,false},{delay_send,true}]\" "
"-kernel inet_default_listen_options "
"\"[{active,false},{delay_send,true}]\"",
fun do_delay_send_7/0),
?line {true,true,true} =
do_delay_on_other_node("-kernel inet_default_connect_options "
"\"[{active,false},{delay_send,true}]\" "
"-kernel inet_default_listen_options "
"\"[{active,false},{delay_send,true}]\"",
fun do_delay_send_2/0),
ok.
do_delay_on_other_node(XArgs, Function) ->
Dir = filename:dirname(code:which(?MODULE)),
{ok,Node} = test_server:start_node(test_default_options_slave,slave,
[{args,"-pa " ++ Dir ++ " " ++
XArgs}]),
Res = rpc:call(Node,erlang,apply,[Function,[]]),
test_server:stop_node(Node),
Res.
do_delay_send_1() ->
{ok,LS}=gen_tcp:listen(0,[{delay_send,true}]),
{ok,{{0,0,0,0},PortNum}}=inet:sockname(LS),
{ok,S}=gen_tcp:connect("localhost",PortNum,[{delay_send,true}]),
{ok,S2}= gen_tcp:accept(LS),
{ok,[{delay_send,B1}]}=inet:getopts(S,[delay_send]),
{ok,[{delay_send,B2}]}=inet:getopts(LS,[delay_send]),
{ok,[{delay_send,B3}]}=inet:getopts(S2,[delay_send]),
gen_tcp:close(S2),
gen_tcp:close(S),
gen_tcp:close(LS),
{B1,B2,B3}.
do_delay_send_2() ->
{ok,LS}=gen_tcp:listen(0,[]),
{ok,{{0,0,0,0},PortNum}}=inet:sockname(LS),
{ok,S}=gen_tcp:connect("localhost",PortNum,[]),
{ok,S2}= gen_tcp:accept(LS),
{ok,[{delay_send,B1}]}=inet:getopts(S,[delay_send]),
{ok,[{delay_send,B2}]}=inet:getopts(LS,[delay_send]),
{ok,[{delay_send,B3}]}=inet:getopts(S2,[delay_send]),
gen_tcp:close(S2),
gen_tcp:close(S),
gen_tcp:close(LS),
{B1,B2,B3}.
do_delay_send_3() ->
{ok,LS}=gen_tcp:listen(0,[]),
{ok,{{0,0,0,0},PortNum}}=inet:sockname(LS),
{ok,S}=gen_tcp:connect("localhost",PortNum,[{delay_send,true}]),
{ok,S2}= gen_tcp:accept(LS),
{ok,[{delay_send,B1}]}=inet:getopts(S,[delay_send]),
{ok,[{delay_send,B2}]}=inet:getopts(LS,[delay_send]),
{ok,[{delay_send,B3}]}=inet:getopts(S2,[delay_send]),
gen_tcp:close(S2),
gen_tcp:close(S),
gen_tcp:close(LS),
{B1,B2,B3}.
do_delay_send_4() ->
{ok,LS}=gen_tcp:listen(0,[{delay_send,false}]),
{ok,{{0,0,0,0},PortNum}}=inet:sockname(LS),
{ok,S}=gen_tcp:connect("localhost",PortNum,[]),
{ok,S2}= gen_tcp:accept(LS),
{ok,[{delay_send,B1}]}=inet:getopts(S,[delay_send]),
{ok,[{delay_send,B2}]}=inet:getopts(LS,[delay_send]),
{ok,[{delay_send,B3}]}=inet:getopts(S2,[delay_send]),
gen_tcp:close(S2),
gen_tcp:close(S),
gen_tcp:close(LS),
{B1,B2,B3}.
do_delay_send_5() ->
{ok,LS}=gen_tcp:listen(0,[]),
{ok,{{0,0,0,0},PortNum}}=inet:sockname(LS),
{ok,S}=gen_tcp:connect("localhost",PortNum,[{delay_send,false}]),
{ok,S2}= gen_tcp:accept(LS),
{ok,[{delay_send,B1}]}=inet:getopts(S,[delay_send]),
{ok,[{delay_send,B2}]}=inet:getopts(LS,[delay_send]),
{ok,[{delay_send,B3}]}=inet:getopts(S2,[delay_send]),
gen_tcp:close(S2),
gen_tcp:close(S),
gen_tcp:close(LS),
{B1,B2,B3}.
do_delay_send_6() ->
{ok,LS}=gen_tcp:listen(0,[{delay_send,true}]),
{ok,{{0,0,0,0},PortNum}}=inet:sockname(LS),
{ok,S}=gen_tcp:connect("localhost",PortNum,[]),
{ok,S2}= gen_tcp:accept(LS),
{ok,[{delay_send,B1}]}=inet:getopts(S,[delay_send]),
{ok,[{delay_send,B2}]}=inet:getopts(LS,[delay_send]),
{ok,[{delay_send,B3}]}=inet:getopts(S2,[delay_send]),
gen_tcp:close(S2),
gen_tcp:close(S),
gen_tcp:close(LS),
{B1,B2,B3}.
do_delay_send_7() ->
{ok,LS}=gen_tcp:listen(0,[]),
{ok,{{0,0,0,0},PortNum}}=inet:sockname(LS),
{ok,S}=gen_tcp:connect("localhost",PortNum,[]),
{ok,S2}= gen_tcp:accept(LS),
{ok,[{active,B1}]}=inet:getopts(S,[active]),
{ok,[{active,B2}]}=inet:getopts(LS,[active]),
{ok,[{active,B3}]}=inet:getopts(S2,[active]),
gen_tcp:close(S2),
gen_tcp:close(S),
gen_tcp:close(LS),
{B1,B2,B3}.
controlling_process(doc) ->
["Open a listen port and change controlling_process for it",
"The result should be ok of done by the owner process,"
"Otherwise is should return {error,not_owner} or similar"];
controlling_process(suite) -> [];
controlling_process(Config) when is_list(Config) ->
{ok,S} = gen_tcp:listen(0,[]),
Pid2 = spawn(?MODULE,not_owner,[S]),
Pid2 ! {self(),2,control},
?line {error, E} = receive {2,_E} ->
_E
after 10000 -> timeout
end,
io:format("received ~p~n",[E]),
Pid = spawn(?MODULE,not_owner,[S]),
?line ok = gen_tcp:controlling_process(S,Pid),
Pid ! {self(),1,control},
?line ok = receive {1,ok} ->
ok
after 1000 -> timeout
end,
Pid ! close.
not_owner(S) ->
receive
{From,Tag,control} ->
From ! {Tag,gen_tcp:controlling_process(S,self())};
close ->
gen_tcp:close(S)
after 1000 ->
ok
end.
controlling_process_self(doc) ->
["Open a listen port and assign the controlling process to "
"it self, then exit and make sure the port is closed properly."];
controlling_process_self(Config) when is_list(Config) ->
S = self(),
process_flag(trap_exit,true),
spawn_link(fun() ->
{ok,Sock} = gen_tcp:listen(0,[]),
S ! {socket, Sock},
ok = gen_tcp:controlling_process(Sock,self()),
S ! done
end),
receive
done ->
receive
{socket,Sock} ->
process_flag(trap_exit,false),
%% Make sure the port is invalid after process crash
{error,einval} = inet:port(Sock)
end;
Msg when element(1,Msg) /= socket ->
process_flag(trap_exit,false),
exit({unknown_msg,Msg})
end.
no_accept(doc) ->
["Open a listen port and connect to it, then close the listen port ",
"without doing any accept. The connected socket should receive ",
"a tcp_closed message."];
no_accept(suite) -> [];
no_accept(Config) when is_list(Config) ->
{ok, L} = gen_tcp:listen(0, []),
{ok, {_, Port}} = inet:sockname(L),
{ok, Client} = gen_tcp:connect(localhost, Port, []),
ok = gen_tcp:close(L),
receive
{tcp_closed, Client} ->
ok
after 5000 ->
?line test_server:fail(never_closed)
end.
close_with_pending_output(doc) ->
["Send several packets to a socket and close it. All packets should arrive ",
"to the other end."];
close_with_pending_output(suite) -> [];
close_with_pending_output(Config) when is_list(Config) ->
?line {ok, L} = gen_tcp:listen(0, [binary, {active, false}]),
?line {ok, {_, Port}} = inet:sockname(L),
?line Packets = 16,
?line Total = 2048*Packets,
case start_remote(close_pending) of
{ok, Node} ->
?line {ok, Host} = inet:gethostname(),
?line spawn_link(Node, ?MODULE, sender, [Port, Packets, Host]),
?line {ok, A} = gen_tcp:accept(L),
?line case gen_tcp:recv(A, Total) of
{ok, Bin} when byte_size(Bin) == Total ->
gen_tcp:close(A),
gen_tcp:close(L);
{ok, Bin} ->
?line test_server:fail({small_packet,
byte_size(Bin)});
Error ->
?line test_server:fail({unexpected, Error})
end,
ok;
{error, no_remote_hosts} ->
{skipped,"No remote hosts"};
{error, Other} ->
?line ?t:fail({failed_to_start_slave_node, Other})
end.
sender(Port, Packets, Host) ->
X256 = lists:seq(0, 255),
X512 = [X256|X256],
X1K = [X512|X512],
Bin = list_to_binary([X1K|X1K]),
{ok, Sock} = gen_tcp:connect(Host, Port, []),
send_loop(Sock, Bin, Packets),
ok = gen_tcp:close(Sock).
send_loop(_Sock, _Data, 0) -> ok;
send_loop(Sock, Data, Left) ->
ok = gen_tcp:send(Sock, Data),
send_loop(Sock, Data, Left-1).
-define(OTP_3924_MAX_DELAY, 100).
%% Taken out of the blue, but on intra host connections
%% I expect propagation of a close to be quite fast
%% so 100 ms seems reasonable.
otp_3924(doc) ->
["Tests that a socket can be closed fast enough."];
otp_3924(suite) -> [];
otp_3924(Config) when is_list(Config) ->
MaxDelay = (case has_superfluous_schedulers() of
true -> 4;
false -> 1
end
* case {erlang:system_info(debug_compiled),
erlang:system_info(lock_checking)} of
{true, _} -> 6;
{_, true} -> 2;
_ -> 1
end * ?OTP_3924_MAX_DELAY),
otp_3924_1(MaxDelay).
otp_3924_1(MaxDelay) ->
Dog = test_server:timetrap(test_server:seconds(240)),
?line {ok, Node} = start_node(otp_3924),
?line DataLen = 100*1024,
?line Data = otp_3924_data(DataLen),
% Repeat the test a couple of times to prevent the test from passing
% by chance.
repeat(10,
fun (N) ->
?line ok = otp_3924(MaxDelay, Node, Data, DataLen, N)
end),
?line test_server:stop_node(Node),
test_server:timetrap_cancel(Dog),
ok.
otp_3924(MaxDelay, Node, Data, DataLen, N) ->
?line {ok, L} = gen_tcp:listen(0, [list, {active, false}]),
?line {ok, {_, Port}} = inet:sockname(L),
?line {ok, Host} = inet:gethostname(),
?line Sender = spawn_link(Node,
?MODULE,
otp_3924_sender,
[self(), Host, Port, Data]),
?line Data = otp_3924_receive_data(L, Sender, MaxDelay, DataLen, N),
?line ok = gen_tcp:close(L).
otp_3924_receive_data(LSock, Sender, MaxDelay, Len, N) ->
?line OP = process_flag(priority, max),
?line OTE = process_flag(trap_exit, true),
?line TimeoutRef = make_ref(),
?line Data = (catch begin
?line Sender ! start,
?line {ok, Sock} = gen_tcp:accept(LSock),
?line D = otp_3924_receive_data(Sock,
TimeoutRef,
MaxDelay,
Len,
[],
0),
?line ok = gen_tcp:close(Sock),
D
end),
?line unlink(Sender),
?line process_flag(trap_exit, OTE),
?line process_flag(priority, OP),
receive
{'EXIT', _, TimeoutRef} ->
?line test_server:fail({close_not_fast_enough,MaxDelay,N});
{'EXIT', Sender, Reason} ->
?line test_server:fail({sender_exited, Reason});
{'EXIT', _Other, Reason} ->
?line test_server:fail({linked_process_exited, Reason})
after 0 ->
case Data of
{'EXIT', {A,B}} ->
?line test_server:fail({A,B,N});
{'EXIT', Failure} ->
?line test_server:fail(Failure);
_ ->
?line Data
end
end.
otp_3924_receive_data(Sock, TimeoutRef, MaxDelay, Len, Acc, AccLen) ->
case gen_tcp:recv(Sock, 0) of
{ok, Data} ->
NewAccLen = AccLen + length(Data),
if
NewAccLen == Len ->
?line {ok, TRef} = timer:exit_after(MaxDelay,
self(),
TimeoutRef),
?line {error, closed} = gen_tcp:recv(Sock, 0),
?line timer:cancel(TRef),
?line lists:flatten([Acc, Data]);
NewAccLen > Len ->
exit({received_too_much, NewAccLen});
true ->
otp_3924_receive_data(Sock,
TimeoutRef,
MaxDelay,
Len,
[Acc, Data],
NewAccLen)
end;
{error, closed} ->
exit({premature_close, AccLen});
Error ->
exit({unexpected_error, Error})
end.
otp_3924_data(Size) ->
Block =
"This is a sequence of characters that will be repeated "
"again and again and again and again and again and ... ",
L = length(Block),
otp_3924_data(Block, [], Size div L, Size rem L).
otp_3924_data(_, Acc, 0, 0) ->
lists:flatten(Acc);
otp_3924_data(_, Acc, 0, SingleLeft) ->
otp_3924_data(false, ["."|Acc], 0, SingleLeft-1);
otp_3924_data(Block, Acc, BlockLeft, SingleLeft) ->
otp_3924_data(Block, [Block|Acc], BlockLeft-1, SingleLeft).
otp_3924_sender(Receiver, Host, Port, Data) ->
receive
start ->
{ok, Sock} = gen_tcp:connect(Host, Port, [list]),
gen_tcp:send(Sock, Data),
ok = gen_tcp:close(Sock),
unlink(Receiver)
end.
data_before_close(doc) ->
["Tests that a huge amount of data can be received before a close."];
data_before_close(Config) when is_list(Config) ->
{ok, L} = gen_tcp:listen(0, [binary]),
{ok, {_, TcpPort}} = inet:sockname(L),
Bytes = 256*1024,
spawn_link(fun() -> huge_sender(TcpPort, Bytes) end),
{ok, A} = gen_tcp:accept(L),
case count_bytes_recv(A, 0) of
{Bytes, Result} ->
io:format("Result: ~p", [Result]);
{Wrong, Result} ->
io:format("Result: ~p", [Result]),
test_server:fail({wrong_count, Wrong})
end,
ok.
count_bytes_recv(Sock, Total) ->
receive
{tcp, Sock, Bin} ->
count_bytes_recv(Sock, Total+byte_size(Bin));
Other ->
{Total, Other}
end.
huge_sender(TcpPort, Bytes) ->
{ok, Client} = gen_tcp:connect(localhost, TcpPort, []),
receive after 500 -> ok end,
gen_tcp:send(Client, make_zero_packet(Bytes)),
gen_tcp:close(Client).
make_zero_packet(0) -> [];
make_zero_packet(N) when N rem 2 == 0 ->
P = make_zero_packet(N div 2),
[P|P];
make_zero_packet(N) ->
P = make_zero_packet(N div 2),
[0, P|P].
get_status(doc) ->
["OTP-2924",
"test that the socket process does not crash when sys:get_status(Pid)",
"is called."];
get_status(suite) -> [];
get_status(Config) when is_list(Config) ->
?line {ok,{socket,Pid,_,_}} = gen_tcp:listen(5678,[]),
?line {status,Pid,_,_} = sys:get_status(Pid).
-define(RECOVER_SLEEP, 60000).
-define(RETRY_SLEEP, 15000).
iter_max_socks(doc) ->
["Open as many sockets as possible. Do this several times and check ",
"that we get the same number of sockets every time."];
iter_max_socks(Config) when is_list(Config) ->
N = 20,
%% Run on a different node in order to limit the effect if this test fails.
Dir = filename:dirname(code:which(?MODULE)),
{ok,Node} = test_server:start_node(test_iter_max_socks,slave,
[{args,"-pa " ++ Dir}]),
L = rpc:call(Node,?MODULE,do_iter_max_socks,[N, initalize]),
test_server:stop_node(Node),
io:format("Result: ~p",[L]),
all_equal(L),
{comment, "Max sockets: " ++ integer_to_list(hd(L))}.
do_iter_max_socks(0, _) ->
[];
do_iter_max_socks(N, initalize) ->
MS = max_socks(),
[MS|do_iter_max_socks(N-1, MS)];
do_iter_max_socks(N, failed) ->
MS = max_socks(),
[MS|do_iter_max_socks(N-1, failed)];
do_iter_max_socks(N, First) when is_integer(First) ->
?line MS = max_socks(),
if MS == First ->
?line [MS|do_iter_max_socks(N-1, First)];
true ->
?line io:format("Sleeping for ~p seconds...~n",
[?RETRY_SLEEP/1000]),
?line ?t:sleep(?RETRY_SLEEP),
?line io:format("Trying again...~n", []),
?line RetryMS = max_socks(),
?line if RetryMS == First ->
?line [RetryMS|do_iter_max_socks(N-1, First)];
true ->
?line [RetryMS|do_iter_max_socks(N-1, failed)]
end
end.
all_equal([]) ->
ok;
all_equal([Rule | T]) ->
all_equal(Rule, T).
all_equal(Rule, [Rule | T]) ->
all_equal(Rule, T);
all_equal(_, [_ | _]) ->
?line ?t:sleep(?RECOVER_SLEEP), % Wait a while and *hope* that we'll
% recover so other tests won't be
% affected.
?t:fail(max_socket_mismatch);
all_equal(_Rule, []) ->
ok.
max_socks() ->
?line Socks = open_socks(),
?line N = length(Socks),
?line lists:foreach(fun(S) -> ok = gen_tcp:close(S) end, Socks),
io:format("Got ~p sockets", [N]),
N.
open_socks() ->
case gen_tcp:listen(0, []) of
{ok, L} ->
{ok, {_, Port}} = inet:sockname(L),
[L| connect_accept(L, Port)];
_ ->
[]
end.
connect_accept(L, Port) ->
case gen_tcp:connect(localhost, Port, []) of
{ok, C} ->
[C| do_accept(L, Port)];
_ ->
[]
end.
do_accept(L, Port) ->
case gen_tcp:accept(L) of
{ok, A} -> [A| connect_accept(L, Port)];
_ -> []
end.
start_node(Name) ->
Pa = filename:dirname(code:which(?MODULE)),
test_server:start_node(Name, slave, [{args, "-pa " ++ Pa}]).
start_remote(Name) ->
Pa = filename:dirname(code:which(?MODULE)),
test_server:start_node(Name, slave, [{remote, true}, {args, "-pa " ++ Pa}]).
passive_sockets(doc) ->
["Tests that when 'the other side' on a passive socket closes, the connecting",
"side still can read until the end of data."];
passive_sockets(Config) when is_list(Config) ->
?line spawn_link(?MODULE, passive_sockets_server,
[[{active,false}],self()]),
?line receive
{socket,Port} -> ok
end,
?t:sleep(500),
?line case gen_tcp:connect("localhost", Port, [{active, false}]) of
{ok, Sock} ->
passive_sockets_read(Sock);
Error ->
?t:fail({"Could not connect to server", Error})
end.
%%
%% Read until we get an {error, closed}. If we get another error, this test case
%% should fail.
%%
passive_sockets_read(Sock) ->
case gen_tcp:recv(Sock, 0, 2000) of
{ok, Data} ->
io:format("Received ~p bytes~n", [length(Data)]),
passive_sockets_read(Sock);
{error, closed} ->
gen_tcp:close(Sock);
Error ->
gen_tcp:close(Sock),
?t:fail({"Did not get {error, closed} before other error", Error})
end.
passive_sockets_server(Opts, Parent) ->
?line case gen_tcp:listen(0, Opts) of
{ok, LSock} ->
{ok,{_,Port}} = inet:sockname(LSock),
Parent ! {socket,Port},
passive_sockets_server_accept(LSock);
Error ->
?t:fail({"Could not create listen socket", Error})
end.
passive_sockets_server_accept(Sock) ->
?line case gen_tcp:accept(Sock) of
{ok, Socket} ->
?t:sleep(500), % Simulate latency
passive_sockets_server_send(Socket, 5),
passive_sockets_server_accept(Sock);
Error ->
?t:fail({"Could not accept connection", Error})
end.
passive_sockets_server_send(Socket, 0) ->
io:format("Closing other end..~n", []),
gen_tcp:close(Socket);
passive_sockets_server_send(Socket, X) ->
?line Data = lists:duplicate(1024*X, $a),
?line case gen_tcp:send(Socket, Data) of
ok ->
?t:sleep(50), % Simulate some processing.
passive_sockets_server_send(Socket, X-1);
{error, _Reason} ->
?t:fail("Failed to send data")
end.
accept_closed_by_other_process(doc) ->
["Tests the return value from gen_tcp:accept when ",
"the socket is closed from an other process. (OTP-3817)"];
accept_closed_by_other_process(Config) when is_list(Config) ->
?line Parent = self(),
?line {ok, ListenSocket} = gen_tcp:listen(0, []),
?line Child =
spawn_link(
fun() ->
Parent ! {self(), gen_tcp:accept(ListenSocket)}
end),
?line receive after 1000 -> ok end,
?line ok = gen_tcp:close(ListenSocket),
?line receive
{Child, {error, closed}} ->
ok;
{Child, Other} ->
?t:fail({"Wrong result of gen_tcp:accept", Other})
end.
repeat(N, Fun) ->
repeat(N, N, Fun).
repeat(N, T, Fun) when is_integer(N), N > 0 ->
Fun(T-N),
repeat(N-1, T, Fun);
repeat(_, _, _) ->
ok.
closed_socket(suite) ->
[];
closed_socket(doc) ->
["Tests the response when using a closed socket as argument"];
closed_socket(Config) when is_list(Config) ->
?line {ok, LS1} = gen_tcp:listen(0, []),
?line erlang:yield(),
?line ok = gen_tcp:close(LS1),
%% If the following delay is uncommented, the result error values
%% below will change from {error, einval} to {error, closed} since
%% inet_db then will have noticed that the socket is closed.
%% This is a scheduling issue, i.e when the gen_server in
%% in inet_db processes the 'EXIT' message from the port,
%% the socket is unregistered.
%%
%% ?line test_server:sleep(test_server:seconds(2)),
%%
?line {error, R_send} = gen_tcp:send(LS1, "data"),
?line {error, R_recv} = gen_tcp:recv(LS1, 17),
?line {error, R_accept} = gen_tcp:accept(LS1),
?line {error, R_controlling_process} =
gen_tcp:controlling_process(LS1, self()),
%%
?line ok = io:format("R_send = ~p~n", [R_send]),
?line ok = io:format("R_recv = ~p~n", [R_recv]),
?line ok = io:format("R_accept = ~p~n", [R_accept]),
?line ok = io:format("R_controlling_process = ~p~n",
[R_controlling_process]),
ok.
%%%
%%% Test using the gen_tcp:shutdown/2 function using a sort server.
%%%
shutdown_active(Config) when is_list(Config) ->
?line shutdown_common(true).
shutdown_passive(Config) when is_list(Config) ->
?line shutdown_common(false).
shutdown_common(Active) ->
?line P = sort_server(Active),
io:format("Sort server port: ~p\n", [P]),
?line do_sort(P, []),
?line do_sort(P, ["glurf"]),
?line do_sort(P, ["abc","nisse","dum"]),
?line do_sort(P, [lists:reverse(integer_to_list(I)) || I <- lists:seq(25, 255)]),
?line do_sort(P, [lists:reverse(integer_to_list(I)) || I <- lists:seq(77, 999)]),
?line do_sort(P, [lists:reverse(integer_to_list(I)) || I <- lists:seq(25, 55)]),
?line do_sort(P, []),
?line do_sort(P, ["apa"]),
?line do_sort(P, ["kluns","gorilla"]),
?line do_sort(P, [lists:reverse(integer_to_list(I)) || I <- lists:seq(25, 1233)]),
?line do_sort(P, []),
receive
Any ->
?t:fail({unexpected_message,Any})
after 0 -> ok
end.
do_sort(P, List0) ->
List = [El++"\n" || El <- List0],
{ok,S} = gen_tcp:connect(localhost, P, [{packet,line}]),
send_lines(S, List),
gen_tcp:shutdown(S, write),
Lines = collect_lines(S, true),
io:format("~p\n", [Lines]),
Lines = lists:sort(List),
ok = gen_tcp:close(S).
sort_server(Active) ->
Opts = [{exit_on_close,false},{packet,line},{active,Active}],
?line {ok,L} = gen_tcp:listen(0, Opts),
Go = make_ref(),
?line Pid = spawn_link(fun() ->
receive Go -> sort_server_1(L, Active) end
end),
?line ok = gen_tcp:controlling_process(L, Pid),
?line Pid ! Go,
?line {ok,Port} = inet:port(L),
Port.
sort_server_1(L, Active) ->
{ok,S} = gen_tcp:accept(L),
Go = make_ref(),
Sorter = spawn(fun() -> receive Go -> sorter(S, Active) end end),
ok = gen_tcp:controlling_process(S, Sorter),
Sorter ! Go,
sort_server_1(L, Active).
sorter(S, Active) ->
Lines = collect_lines(S, Active),
send_lines(S, lists:sort(Lines)),
gen_tcp:shutdown(S, write),
gen_tcp:close(S).
collect_lines(S, true) ->
collect_lines_1(S, []);
collect_lines(S, false) ->
passive_collect_lines_1(S, []).
collect_lines_1(S, Acc) ->
receive
{tcp,S,Line} -> collect_lines_1(S, [Line|Acc]);
{tcp_closed,S} -> lists:reverse(Acc)
end.
passive_collect_lines_1(S, Acc) ->
case gen_tcp:recv(S, 0) of
{ok,Line} -> passive_collect_lines_1(S, [Line|Acc]);
{error,closed} -> lists:reverse(Acc)
end.
send_lines(S, Lines) ->
lists:foreach(fun(Line) ->
gen_tcp:send(S, Line)
end, Lines).
%%%
%%% Shutdown pending.
%%%
shutdown_pending(Config) when is_list(Config) ->
N = 512*1024+17,
io:format("~p\n", [N]),
Data = [<<N:32>>,ones(N),42],
P = a_server(),
io:format("Server port: ~p\n", [P]),
?line {ok,S} = gen_tcp:connect(localhost, P, []),
?line gen_tcp:send(S, Data),
?line gen_tcp:shutdown(S, write),
?line receive
{tcp,S,Msg} ->
io:format("~p\n", [Msg]),
?line N = list_to_integer(Msg) - 5;
Other ->
?t:fail({unexpected,Other})
end,
ok.
ones(0) -> [];
ones(1) -> [1];
ones(N) ->
Half = N div 2,
Ones = ones(Half),
case 2*Half of
N -> [Ones|Ones];
_ -> [1,Ones|Ones]
end.
a_server() ->
?line {ok,L} = gen_tcp:listen(0, [{exit_on_close,false},{active,false}]),
?line Pid = spawn_link(fun() -> a_server(L) end),
?line ok = gen_tcp:controlling_process(L, Pid),
?line {ok,Port} = inet:port(L),
Port.
a_server(L) ->
{ok,S} = gen_tcp:accept(L),
do_recv(S, []).
do_recv(S, Bs0) ->
case gen_tcp:recv(S, 0) of
{ok,B} ->
do_recv(S, [Bs0,B]);
{error,closed} ->
Bs = list_to_binary(Bs0),
gen_tcp:send(S, integer_to_list(byte_size(Bs))),
gen_tcp:close(S)
end.
%% Thanks to Luke Gorrie. Tests for a very specific problem with
%% corrupt data. The testcase will be killed by the timetrap timeout
%% if the bug is present.
http_bad_packet(Config) when is_list(Config) ->
?line {ok,L} = gen_tcp:listen(0,
[{active, false},
binary,
{reuseaddr, true},
{packet, http}]),
?line {ok,Port} = inet:port(L),
?line spawn_link(fun() -> erlang:yield(), http_bad_client(Port) end),
?line case gen_tcp:accept(L) of
{ok,S} ->
http_worker(S);
Err ->
exit({accept,Err})
end.
http_worker(S) ->
case gen_tcp:recv(S, 0, 30000) of
{ok,{http_error,Error}} ->
io:format("Http error: ~s\n", [Error]);
{ok,Data} ->
io:format("Data: ~p\n", [Data]),
http_worker(S)
end.
http_bad_client(Port) ->
{ok,S} = gen_tcp:connect("localhost", Port, [{active,false}, binary]),
ok = gen_tcp:send(S, "\r\n"),
ok = gen_tcp:close(S).
%% Fill send queue and then start receiving.
%%
busy_send(Config) when is_list(Config) ->
?line Master = self(),
?line Msg = <<"the quick brown fox jumps over a lazy dog~n">>,
?line Server =
spawn_link(fun () ->
{ok,L} = gen_tcp:listen
(0, [{active,false},binary,
{reuseaddr,true},{packet,0}]),
{ok,Port} = inet:port(L),
Master ! {self(),client,
busy_send_client(Port, Master, Msg)},
busy_send_srv(L, Master, Msg)
end),
?line io:format("~p Server~n", [Server]),
?line receive
{Server,client,Client} ->
?line io:format("~p Client~n", [Client]),
?line busy_send_loop(Server, Client, 0)
end.
busy_send_loop(Server, Client, N) ->
%% Master
%%
?line receive {Server,send} ->
busy_send_loop(Server, Client, N+1)
after 2000 ->
%% Send queue full, sender blocked
%% -> stop sender and release client
?line io:format("Send timeout, time to receive...~n", []),
?line Server ! {self(),close},
?line Client ! {self(),recv,N+1},
?line receive
{Server,send} ->
?line busy_send_2(Server, Client, N+1)
after 10000 ->
%% If this happens, see busy_send_srv
?t:fail({timeout,{server,not_send,flush([])}})
end
end.
busy_send_2(Server, Client, _N) ->
%% Master
%%
?line receive
{Server,[closed]} ->
?line receive
{Client,[0,{error,closed}]} ->
ok
end
after 10000 ->
?t:fail({timeout,{server,not_closed,flush([])}})
end.
busy_send_srv(L, Master, Msg) ->
%% Server
%% Sometimes this accept does not return, do not really know why
%% but is causes the timeout error in busy_send_loop to be
%% triggered. Only happens on OS X Leopard?!?
{ok,Socket} = gen_tcp:accept(L),
busy_send_srv_loop(Socket, Master, Msg).
busy_send_srv_loop(Socket, Master, Msg) ->
%% Server
%%
receive
{Master,close} ->
ok = gen_tcp:close(Socket),
Master ! {self(),flush([closed])}
after 0 ->
ok = gen_tcp:send(Socket, Msg),
Master ! {self(),send},
busy_send_srv_loop(Socket, Master, Msg)
end.
busy_send_client(Port, Master, Msg) ->
%% Client
%%
spawn_link(
fun () ->
{ok,Socket} = gen_tcp:connect(
"localhost", Port,
[{active,false},binary,{packet,0}]),
receive
{Master,recv, N} ->
busy_send_client_loop(Socket, Master, Msg, N)
end
end).
busy_send_client_loop(Socket, Master, Msg, N) ->
%% Client
%%
Size = byte_size(Msg),
case gen_tcp:recv(Socket, Size) of
{ok,Msg} ->
busy_send_client_loop(Socket, Master, Msg, N-1);
Other ->
Master ! {self(),flush([Other,N])}
end.
%%%
%%% Send to a socket whose other end does not read until the port gets busy.
%%% Then close the other end. The writer should get an {error,closed} error.
%%% (Passive mode.)
%%%
busy_disconnect_passive(Config) when is_list(Config) ->
MuchoData = list_to_binary(ones(64*1024)),
?line [do_busy_disconnect_passive(MuchoData) || _ <- lists:seq(1, 10)],
ok.
do_busy_disconnect_passive(MuchoData) ->
S = busy_disconnect_prepare_server([{active,false}]),
busy_disconnect_passive_send(S, MuchoData).
busy_disconnect_passive_send(S, Data) ->
?line case gen_tcp:send(S, Data) of
ok -> ?line busy_disconnect_passive_send(S, Data);
{error,closed} -> ok
end.
%%%
%%% Send to a socket whose other end does not read until the port gets busy.
%%% Then close the other end. The writer should get an {error,closed} error and
%%% a {tcp_closed,Socket} message. (Active mode.)
%%%
busy_disconnect_active(Config) when is_list(Config) ->
MuchoData = list_to_binary(ones(64*1024)),
?line [do_busy_disconnect_active(MuchoData) || _ <- lists:seq(1, 10)],
ok.
do_busy_disconnect_active(MuchoData) ->
S = busy_disconnect_prepare_server([{active,true}]),
busy_disconnect_active_send(S, MuchoData).
busy_disconnect_active_send(S, Data) ->
?line case gen_tcp:send(S, Data) of
ok -> ?line busy_disconnect_active_send(S, Data);
{error,closed} ->
receive
{tcp_closed,S} -> ok;
_Other -> ?line ?t:fail()
end
end.
busy_disconnect_prepare_server(ConnectOpts) ->
?line Sender = self(),
?line Server = spawn_link(fun() -> busy_disconnect_server(Sender) end),
receive {port,Server,Port} -> ok end,
?line {ok,S} = gen_tcp:connect(localhost, Port, ConnectOpts),
Server ! {Sender,sending},
S.
busy_disconnect_server(Sender) ->
{ok,L} = gen_tcp:listen(0, [{active,false},binary,{reuseaddr,true},{packet,0}]),
{ok,Port} = inet:port(L),
Sender ! {port,self(),Port},
{ok,S} = gen_tcp:accept(L),
receive
{Sender,sending} ->
busy_disconnect_server_wait_for_busy(Sender, S)
end.
%% Close the socket as soon as the Sender process can't send because of
%% a busy port.
busy_disconnect_server_wait_for_busy(Sender, S) ->
case process_info(Sender, status) of
{status,waiting} ->
%% We KNOW that the sender will be in state 'waiting' only
%% if the port has become busy. (Fallback solution if the
%% implementation changes: Watch Sender's reduction count;
%% when it stops changing, wait 2 seconds and then close.)
gen_tcp:close(S);
_Other ->
io:format("~p\n", [_Other]),
timer:sleep(100),
busy_disconnect_server_wait_for_busy(Sender, S)
end.
%%%
%%% Fill send queue
%%%
fill_sendq(Config) when is_list(Config) ->
?line Master = self(),
?line Server =
spawn_link(fun () ->
{ok,L} = gen_tcp:listen
(0, [{active,false},binary,
{reuseaddr,true},{packet,0}]),
{ok,Port} = inet:port(L),
Master ! {self(),client,
fill_sendq_client(Port, Master)},
fill_sendq_srv(L, Master)
end),
?line io:format("~p Server~n", [Server]),
?line receive {Server,client,Client} ->
?line io:format("~p Client~n", [Client]),
?line receive {Server,reader,Reader} ->
?line io:format("~p Reader~n", [Reader]),
?line fill_sendq_loop(Server, Client, Reader)
end
end.
fill_sendq_loop(Server, Client, Reader) ->
%% Master
%%
receive {Server,send} ->
fill_sendq_loop(Server, Client, Reader)
after 2000 ->
%% Send queue full, sender blocked -> close client.
?line io:format("Send timeout, closing Client...~n", []),
?line Client ! {self(),close},
?line receive {Server,[{error,closed}]} ->
?line io:format("Got server closed.~n"),
?line receive {Reader,[{error,closed}]} ->
?line io:format
("Got reader closed.~n"),
ok
after 3000 ->
?t:fail({timeout,{closed,reader}})
end;
{Reader,[{error,closed}]} ->
?line io:format("Got reader closed.~n"),
?line receive {Server,[{error,closed}]} ->
?line io:format("Got server closed~n"),
ok
after 3000 ->
?t:fail({timeout,{closed,server}})
end
after 3000 ->
?t:fail({timeout,{closed,[server,reader]}})
end
end.
fill_sendq_srv(L, Master) ->
%% Server
%%
case gen_tcp:accept(L) of
{ok,S} ->
Master ! {self(),reader,
spawn_link(fun () -> fill_sendq_read(S, Master) end)},
Msg = "the quick brown fox jumps over a lazy dog~n",
fill_sendq_write(S, Master, [Msg,Msg,Msg,Msg,Msg,Msg,Msg,Msg]);
Error ->
io:format("~p error: ~p.~n", [self(),Error]),
Master ! {self(),flush([Error])}
end.
fill_sendq_write(S, Master, Msg) ->
%% Server
%%
%%io:format("~p sending...~n", [self()]),
Master ! {self(),send},
case gen_tcp:send(S, Msg) of
ok ->
%%io:format("~p ok.~n", [self()]),
fill_sendq_write(S, Master, Msg);
E ->
Error = flush([E]),
io:format("~p error: ~p.~n", [self(),Error]),
Master ! {self(),Error}
end.
fill_sendq_read(S, Master) ->
%% Reader
%%
io:format("~p read infinity...~n", [self()]),
case gen_tcp:recv(S, 0, infinity) of
{ok,Data} ->
io:format("~p got: ~p.~n", [self(),Data]),
fill_sendq_read(S, Master);
E ->
Error = flush([E]),
io:format("~p error: ~p.~n", [self(),Error]),
Master ! {self(),Error}
end.
fill_sendq_client(Port, Master) ->
%% Client
%%
spawn_link(fun () ->
%% Just close on order
{ok,S} = gen_tcp:connect(
"localhost", Port,
[{active,false},binary,{packet,0}]),
receive
{Master,close} ->
ok = gen_tcp:close(S)
end
end).
%%% Try to receive more than available number of bytes from
%%% a closed socket.
%%%
partial_recv_and_close(Config) when is_list(Config) ->
?line Msg = "the quick brown fox jumps over a lazy dog 0123456789\n",
?line Len = length(Msg),
?line {ok,L} = gen_tcp:listen(0, [{active,false}]),
?line {ok,P} = inet:port(L),
?line {ok,S} = gen_tcp:connect("localhost", P, [{active,false}]),
?line {ok,A} = gen_tcp:accept(L),
?line ok = gen_tcp:send(S, Msg),
?line ok = gen_tcp:close(S),
?line {error,closed} = gen_tcp:recv(A, Len+1),
ok.
%%% Try to receive more than available number of bytes from
%%% a closed socket, this time waiting in the recv before closing.
%%%
partial_recv_and_close_2(Config) when is_list(Config) ->
?line Msg = "the quick brown fox jumps over a lazy dog 0123456789\n",
?line Len = length(Msg),
?line {ok,L} = gen_tcp:listen(0, [{active,false}]),
?line {ok,P} = inet:port(L),
?line Server = self(),
?line Client =
spawn_link(
fun () ->
receive after 2000 -> ok end,
{ok,S} = gen_tcp:connect("localhost", P, [{active,false}]),
?line ok = gen_tcp:send(S, Msg),
receive {Server,close} -> ok end,
receive after 2000 -> ok end,
?line ok = gen_tcp:close(S)
end),
?line {ok,A} = gen_tcp:accept(L),
?line Client ! {Server,close},
?line {error,closed} = gen_tcp:recv(A, Len+1),
ok.
%%% Here we tests that gen_tcp:recv/2 will return {error,closed} following
%%% a send operation of a huge amount data when the other end closed the socket.
%%%
partial_recv_and_close_3(Config) when is_list(Config) ->
[do_partial_recv_and_close_3() || _ <- lists:seq(0, 20)],
ok.
do_partial_recv_and_close_3() ->
Parent = self(),
spawn_link(fun() ->
{ok,L} = gen_tcp:listen(0, [{active,false}]),
{ok,{_,Port}} = inet:sockname(L),
Parent ! {port,Port},
{ok,S} = gen_tcp:accept(L),
gen_tcp:recv(S, 1),
gen_tcp:close(S)
end),
receive
{port,Port} -> ok
end,
?line Much = ones(8*64*1024),
?line {ok,S} = gen_tcp:connect(localhost, Port, [{active,false}]),
%% Send a lot of data (most of it will be queued). The receiver will read one byte
%% and close the connection. The write operation will fail.
?line gen_tcp:send(S, Much),
%% We should always get {error,closed} here.
?line {error,closed} = gen_tcp:recv(S, 0).
test_prio_put_get() ->
Tos = 3 bsl 5,
?line {ok,L1} = gen_tcp:listen(0, [{active,false}]),
?line ok = inet:setopts(L1,[{priority,3}]),
?line ok = inet:setopts(L1,[{tos,Tos}]),
?line {ok,[{priority,3},{tos,Tos}]} = inet:getopts(L1,[priority,tos]),
?line ok = inet:setopts(L1,[{priority,3}]), % Dont destroy each other
?line {ok,[{priority,3},{tos,Tos}]} = inet:getopts(L1,[priority,tos]),
?line ok = inet:setopts(L1,[{reuseaddr,true}]), % Dont let others destroy
?line {ok,[{priority,3},{tos,Tos}]} = inet:getopts(L1,[priority,tos]),
?line gen_tcp:close(L1),
ok.
test_prio_accept() ->
?line {ok,Sock}=gen_tcp:listen(0,[binary,{packet,0},{active,false},
{reuseaddr,true},{priority,4}]),
?line {ok,Port} = inet:port(Sock),
?line {ok,Sock2}=gen_tcp:connect("localhost",Port,[binary,{packet,0},
{active,false},
{reuseaddr,true},
{priority,4}]),
?line {ok,Sock3}=gen_tcp:accept(Sock),
?line {ok,[{priority,4}]} = inet:getopts(Sock,[priority]),
?line {ok,[{priority,4}]} = inet:getopts(Sock2,[priority]),
?line {ok,[{priority,4}]} = inet:getopts(Sock3,[priority]),
?line gen_tcp:close(Sock),
?line gen_tcp:close(Sock2),
?line gen_tcp:close(Sock3),
ok.
test_prio_accept2() ->
Tos1 = 4 bsl 5,
Tos2 = 3 bsl 5,
?line {ok,Sock}=gen_tcp:listen(0,[binary,{packet,0},{active,false},
{reuseaddr,true},{priority,4},
{tos,Tos1}]),
?line {ok,Port} = inet:port(Sock),
?line {ok,Sock2}=gen_tcp:connect("localhost",Port,[binary,{packet,0},
{active,false},
{reuseaddr,true},
{priority,4},
{tos,Tos2}]),
?line {ok,Sock3}=gen_tcp:accept(Sock),
?line {ok,[{priority,4},{tos,Tos1}]} = inet:getopts(Sock,[priority,tos]),
?line {ok,[{priority,4},{tos,Tos2}]} = inet:getopts(Sock2,[priority,tos]),
?line {ok,[{priority,4},{tos,Tos1}]} = inet:getopts(Sock3,[priority,tos]),
?line gen_tcp:close(Sock),
?line gen_tcp:close(Sock2),
?line gen_tcp:close(Sock3),
ok.
test_prio_accept3() ->
Tos1 = 4 bsl 5,
Tos2 = 3 bsl 5,
?line {ok,Sock}=gen_tcp:listen(0,[binary,{packet,0},{active,false},
{reuseaddr,true},
{tos,Tos1}]),
?line {ok,Port} = inet:port(Sock),
?line {ok,Sock2}=gen_tcp:connect("localhost",Port,[binary,{packet,0},
{active,false},
{reuseaddr,true},
{tos,Tos2}]),
?line {ok,Sock3}=gen_tcp:accept(Sock),
?line {ok,[{priority,0},{tos,Tos1}]} = inet:getopts(Sock,[priority,tos]),
?line {ok,[{priority,0},{tos,Tos2}]} = inet:getopts(Sock2,[priority,tos]),
?line {ok,[{priority,0},{tos,Tos1}]} = inet:getopts(Sock3,[priority,tos]),
?line gen_tcp:close(Sock),
?line gen_tcp:close(Sock2),
?line gen_tcp:close(Sock3),
ok.
test_prio_accept_async() ->
Tos1 = 4 bsl 5,
Tos2 = 3 bsl 5,
Ref = make_ref(),
?line spawn(?MODULE,priority_server,[{self(),Ref}]),
?line Port = receive
{Ref,P} -> P
after 5000 -> ?t:fail({error,"helper process timeout"})
end,
?line receive
after 3000 -> ok
end,
?line {ok,Sock2}=gen_tcp:connect("localhost",Port,[binary,{packet,0},
{active,false},
{reuseaddr,true},
{priority,4},
{tos,Tos2}]),
?line receive
{Ref,{ok,[{priority,4},{tos,Tos1}]}} ->
ok ;
{Ref,Error} ->
?t:fail({missmatch,Error})
after 5000 -> ?t:fail({error,"helper process timeout"})
end,
?line receive
{Ref,{ok,[{priority,4},{tos,Tos1}]}} ->
ok ;
{Ref,Error2} ->
?t:fail({missmatch,Error2})
after 5000 -> ?t:fail({error,"helper process timeout"})
end,
?line {ok,[{priority,4},{tos,Tos2}]} = inet:getopts(Sock2,[priority,tos]),
?line catch gen_tcp:close(Sock2),
ok.
priority_server({Parent,Ref}) ->
Tos1 = 4 bsl 5,
?line {ok,Sock}=gen_tcp:listen(0,[binary,{packet,0},{active,false},
{reuseaddr,true},{priority,4},
{tos,Tos1}]),
?line {ok,Port} = inet:port(Sock),
Parent ! {Ref,Port},
?line {ok,Sock3}=gen_tcp:accept(Sock),
Parent ! {Ref, inet:getopts(Sock,[priority,tos])},
Parent ! {Ref, inet:getopts(Sock3,[priority,tos])},
ok.
test_prio_fail() ->
?line {ok,L} = gen_tcp:listen(0, [{active,false}]),
?line {error,_} = inet:setopts(L,[{priority,1000}]),
% This error could only happen in linux kernels earlier than 2.6.24.4
% Privilege check is now disabled and IP_TOS can never fail (only silently
% be masked).
% ?line {error,_} = inet:setopts(L,[{tos,6 bsl 5}]),
?line gen_tcp:close(L),
ok.
test_prio_udp() ->
Tos = 3 bsl 5,
?line {ok,S} = gen_udp:open(0,[{active,false},binary,{tos, Tos},
{priority,3}]),
?line {ok,[{priority,3},{tos,Tos}]} = inet:getopts(S,[priority,tos]),
?line gen_udp:close(S),
ok.
so_priority(doc) ->
["Tests the so_priority and ip_tos options on sockets when applicable."];
so_priority(suite) ->
[];
so_priority(Config) when is_list(Config) ->
?line {ok,L} = gen_tcp:listen(0, [{active,false}]),
?line ok = inet:setopts(L,[{priority,1}]),
?line case inet:getopts(L,[priority]) of
{ok,[{priority,1}]} ->
gen_tcp:close(L),
test_prio_put_get(),
test_prio_accept(),
test_prio_accept2(),
test_prio_accept3(),
test_prio_accept_async(),
test_prio_fail(),
test_prio_udp(),
ok;
_ ->
case os:type() of
{unix,linux} ->
case os:version() of
{X,Y,_} when (X > 2) or ((X =:= 2) and (Y >= 4)) ->
?line ?t:fail({error,
"so_priority should work on this "
"OS, but does not"});
_ ->
{skip, "SO_PRIORITY not suppoorted"}
end;
_ ->
{skip, "SO_PRIORITY not suppoorted"}
end
end.
%% Accept test utilities (suites are below)
millis() ->
{A,B,C}=erlang:now(),
(A*1000000*1000)+(B*1000)+(C div 1000).
collect_accepts(Tmo) ->
A = millis(),
receive
{accepted,P,Msg} ->
[{P,Msg}] ++ collect_accepts(Tmo-(millis() - A))
after Tmo ->
[]
end.
-define(EXPECT_ACCEPTS(Pattern,Timeout),
(fun() ->
case collect_accepts(Timeout) of
Pattern ->
ok;
Other ->
{error,{unexpected,{Other,process_info(self(),messages)}}}
end
end)()).
collect_connects(Tmo) ->
A = millis(),
receive
{connected,P,Msg} ->
[{P,Msg}] ++ collect_connects(Tmo-(millis() - A))
after Tmo ->
[]
end.
-define(EXPECT_CONNECTS(Pattern,Timeout),
(fun() ->
case collect_connects(Timeout) of
Pattern ->
ok;
Other ->
{error,{unexpected,Other}}
end
end)()).
mktmofun(Tmo,Parent,LS) ->
fun() -> Parent ! {accepted,self(), catch gen_tcp:accept(LS,Tmo)} end.
%% Accept tests
primitive_accept(suite) ->
[];
primitive_accept(doc) ->
["Test singular accept"];
primitive_accept(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line {ok,PortNo}=inet:port(LS),
?line Parent = self(),
?line F = fun() -> Parent ! {accepted,self(),gen_tcp:accept(LS)} end,
?line P = spawn(F),
?line gen_tcp:connect("localhost",PortNo,[]),
?line receive
{accepted,P,{ok,P0}} when is_port(P0) ->
ok;
{accepted,P,Other0} ->
{error,Other0}
after 500 ->
{error,timeout}
end.
multi_accept_close_listen(suite) ->
[];
multi_accept_close_listen(doc) ->
["Closing listen socket when multi-accepting"];
multi_accept_close_listen(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line F = fun() -> Parent ! {accepted,self(),gen_tcp:accept(LS)} end,
?line spawn(F),
?line spawn(F),
?line spawn(F),
?line spawn(F),
?line gen_tcp:close(LS),
?line ?EXPECT_ACCEPTS([{_,{error,closed}},{_,{error,closed}},
{_,{error,closed}},{_,{error,closed}}], 500).
accept_timeout(suite) ->
[];
accept_timeout(doc) ->
["Single accept with timeout"];
accept_timeout(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line F = fun() -> Parent ! {accepted,self(),gen_tcp:accept(LS,1000)} end,
?line P = spawn(F),
?line ?EXPECT_ACCEPTS([{P,{error,timeout}}],2000).
accept_timeouts_in_order(suite) ->
[];
accept_timeouts_in_order(doc) ->
["Check that multi-accept timeouts happen in the correct order"];
accept_timeouts_in_order(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line P1 = spawn(mktmofun(1000,Parent,LS)),
?line P2 = spawn(mktmofun(1200,Parent,LS)),
?line P3 = spawn(mktmofun(1300,Parent,LS)),
?line P4 = spawn(mktmofun(1400,Parent,LS)),
?line ?EXPECT_ACCEPTS([{P1,{error,timeout}},{P2,{error,timeout}},
{P3,{error,timeout}},{P4,{error,timeout}}], 2000).
accept_timeouts_in_order2(suite) ->
[];
accept_timeouts_in_order2(doc) ->
["Check that multi-accept timeouts happen in the correct order (more)"];
accept_timeouts_in_order2(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line P1 = spawn(mktmofun(1400,Parent,LS)),
?line P2 = spawn(mktmofun(1300,Parent,LS)),
?line P3 = spawn(mktmofun(1200,Parent,LS)),
?line P4 = spawn(mktmofun(1000,Parent,LS)),
?line ?EXPECT_ACCEPTS([{P4,{error,timeout}},{P3,{error,timeout}},
{P2,{error,timeout}},{P1,{error,timeout}}], 2000).
accept_timeouts_in_order3(suite) ->
[];
accept_timeouts_in_order3(doc) ->
["Check that multi-accept timeouts happen in the correct order (even more)"];
accept_timeouts_in_order3(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line P1 = spawn(mktmofun(1200,Parent,LS)),
?line P2 = spawn(mktmofun(1400,Parent,LS)),
?line P3 = spawn(mktmofun(1300,Parent,LS)),
?line P4 = spawn(mktmofun(1000,Parent,LS)),
?line ?EXPECT_ACCEPTS([{P4,{error,timeout}},{P1,{error,timeout}},
{P3,{error,timeout}},{P2,{error,timeout}}], 2000).
accept_timeouts_mixed(suite) ->
[];
accept_timeouts_mixed(doc) ->
["Check that multi-accept timeouts behave correctly when mixed with successful timeouts"];
accept_timeouts_mixed(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line {ok,PortNo}=inet:port(LS),
?line P1 = spawn(mktmofun(1000,Parent,LS)),
?line wait_until_accepting(P1,500),
?line P2 = spawn(mktmofun(2000,Parent,LS)),
?line wait_until_accepting(P2,500),
?line P3 = spawn(mktmofun(3000,Parent,LS)),
?line wait_until_accepting(P3,500),
?line P4 = spawn(mktmofun(4000,Parent,LS)),
?line wait_until_accepting(P4,500),
?line ok = ?EXPECT_ACCEPTS([{P1,{error,timeout}}],1500),
?line {ok,_}=gen_tcp:connect("localhost",PortNo,[]),
?line ok = ?EXPECT_ACCEPTS([{P2,{ok,Port0}}] when is_port(Port0),100),
?line ok = ?EXPECT_ACCEPTS([{P3,{error,timeout}}],2000),
?line gen_tcp:connect("localhost",PortNo,[]),
?line ?EXPECT_ACCEPTS([{P4,{ok,Port1}}] when is_port(Port1),100).
killing_acceptor(suite) ->
[];
killing_acceptor(doc) ->
["Check that single acceptor behaves as expected when killed"];
killing_acceptor(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Pid = spawn(fun() -> erlang:display({accepted,self(),gen_tcp:accept(LS)}) end),
?line receive after 100 ->
ok
end,
?line {ok,L1} = prim_inet:getstatus(LS),
?line true = lists:member(accepting, L1),
?line exit(Pid,kill),
?line receive after 100 ->
ok
end,
?line {ok,L2} = prim_inet:getstatus(LS),
?line false = lists:member(accepting, L2),
ok.
killing_multi_acceptors(suite) ->
[];
killing_multi_acceptors(doc) ->
["Check that multi acceptors behaves as expected when killed"];
killing_multi_acceptors(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line F = fun() -> Parent ! {accepted,self(),gen_tcp:accept(LS)} end,
?line F2 = mktmofun(1000,Parent,LS),
?line Pid = spawn(F),
?line Pid2 = spawn(F2),
?line receive after 100 ->
ok
end,
?line {ok,L1} = prim_inet:getstatus(LS),
?line true = lists:member(accepting, L1),
?line exit(Pid,kill),
?line receive after 100 ->
ok
end,
?line {ok,L2} = prim_inet:getstatus(LS),
?line true = lists:member(accepting, L2),
?line ok = ?EXPECT_ACCEPTS([{Pid2,{error,timeout}}],1000),
?line {ok,L3} = prim_inet:getstatus(LS),
?line false = lists:member(accepting, L3),
ok.
killing_multi_acceptors2(suite) ->
[];
killing_multi_acceptors2(doc) ->
["Check that multi acceptors behaves as expected when killed (more)"];
killing_multi_acceptors2(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line {ok,PortNo}=inet:port(LS),
?line F = fun() -> Parent ! {accepted,self(),gen_tcp:accept(LS)} end,
?line F2 = mktmofun(1000,Parent,LS),
?line Pid = spawn(F),
?line Pid2 = spawn(F),
?line receive after 100 ->
ok
end,
?line {ok,L1} = prim_inet:getstatus(LS),
?line true = lists:member(accepting, L1),
?line exit(Pid,kill),
?line receive after 100 ->
ok
end,
?line {ok,L2} = prim_inet:getstatus(LS),
?line true = lists:member(accepting, L2),
?line exit(Pid2,kill),
?line receive after 100 ->
ok
end,
?line {ok,L3} = prim_inet:getstatus(LS),
?line false = lists:member(accepting, L3),
?line Pid3 = spawn(F2),
?line receive after 100 ->
ok
end,
?line {ok,L4} = prim_inet:getstatus(LS),
?line true = lists:member(accepting, L4),
?line gen_tcp:connect("localhost",PortNo,[]),
?line ok = ?EXPECT_ACCEPTS([{Pid3,{ok,Port}}] when is_port(Port),100),
?line {ok,L5} = prim_inet:getstatus(LS),
?line false = lists:member(accepting, L5),
ok.
several_accepts_in_one_go(suite) ->
[];
several_accepts_in_one_go(doc) ->
["checks that multi-accept works when more than one accept can be "
"done at once (wb test of inet_driver)"];
several_accepts_in_one_go(Config) when is_list(Config) ->
?line {ok,LS}=gen_tcp:listen(0,[]),
?line Parent = self(),
?line {ok,PortNo}=inet:port(LS),
?line F1 = fun() -> Parent ! {accepted,self(),gen_tcp:accept(LS)} end,
?line F2 = fun() -> Parent ! {connected,self(),gen_tcp:connect("localhost",PortNo,[])} end,
?line spawn(F1),
?line spawn(F1),
?line spawn(F1),
?line spawn(F1),
?line spawn(F1),
?line spawn(F1),
?line spawn(F1),
?line spawn(F1),
?line ok = ?EXPECT_ACCEPTS([],500),
?line spawn(F2),
?line spawn(F2),
?line spawn(F2),
?line spawn(F2),
?line spawn(F2),
?line spawn(F2),
?line spawn(F2),
?line spawn(F2),
?line ok = ?EXPECT_ACCEPTS([{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}}],15000),
?line ok = ?EXPECT_CONNECTS([{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}},{_,{ok,_}}],1000),
ok.
flush(Msgs) ->
erlang:yield(),
receive Msg -> flush([Msg|Msgs])
after 0 -> lists:reverse(Msgs)
end.
wait_until_accepting(Proc,0) ->
exit({timeout_waiting_for_accepting,Proc});
wait_until_accepting(Proc,N) ->
case process_info(Proc,current_function) of
{current_function,{prim_inet,accept0,2}} ->
case process_info(Proc,status) of
{status,waiting} ->
ok;
_O1 ->
receive
after 5 ->
wait_until_accepting(Proc,N-1)
end
end;
_O2 ->
receive
after 5 ->
wait_until_accepting(Proc,N-1)
end
end.
accept_system_limit(suite) ->
[];
accept_system_limit(doc) ->
["Check that accept returns {error, system_limit} "
"(and not {error, enfile}) when running out of ports"];
accept_system_limit(Config) when is_list(Config) ->
?line {ok, LS} = gen_tcp:listen(0, []),
?line {ok, TcpPort} = inet:port(LS),
?line Connector = spawn_link(fun () -> connector(TcpPort) end),
?line ok = acceptor(LS, false, []),
?line Connector ! stop,
ok.
acceptor(LS, GotSL, A) ->
case gen_tcp:accept(LS, 1000) of
{ok, S} ->
acceptor(LS, GotSL, [S|A]);
{error, system_limit} ->
acceptor(LS, true, A);
{error, timeout} when GotSL ->
ok;
{error, timeout} ->
error
end.
connector(TcpPort) ->
ManyPorts = open_ports([]),
ConnF = fun (Port) ->
case catch gen_tcp:connect({127,0,0,1}, TcpPort, []) of
{ok, Sock} ->
Sock;
_Error ->
port_close(Port)
end
end,
R = [ConnF(Port) || Port <- lists:sublist(ManyPorts, 10)],
receive stop -> R end.
open_ports(L) ->
case catch open_port({spawn_driver, "ram_file_drv"}, []) of
Port when is_port(Port) ->
open_ports([Port|L]);
{'EXIT', {system_limit, _}} ->
{L1, L2} = lists:split(5, L),
[port_close(Port) || Port <- L1],
L2
end.
active_once_closed(suite) ->
[];
active_once_closed(doc) ->
["Check that active once and tcp_close messages behave as expected"];
active_once_closed(Config) when is_list(Config) ->
(fun() ->
?line {Loop,A} = setup_closed_ao(),
?line Loop({{error,closed},{error,econnaborted}},
fun() -> gen_tcp:send(A,"Hello") end),
?line ok = inet:setopts(A,[{active,once}]),
?line ok = receive {tcp_closed, A} -> ok after 1000 -> error end,
?line {error,einval} = inet:setopts(A,[{active,once}]),
?line ok = receive {tcp_closed, A} -> error after 1000 -> ok end
end)(),
(fun() ->
?line {Loop,A} = setup_closed_ao(),
?line Loop({{error,closed},{error,econnaborted}},
fun() -> gen_tcp:send(A,"Hello") end),
?line ok = inet:setopts(A,[{active,true}]),
?line ok = receive {tcp_closed, A} -> ok after 1000 -> error end,
?line {error,einval} = inet:setopts(A,[{active,true}]),
?line ok = receive {tcp_closed, A} -> error after 1000 -> ok end
end)(),
(fun() ->
?line {Loop,A} = setup_closed_ao(),
?line Loop({{error,closed},{error,econnaborted}},
fun() -> gen_tcp:send(A,"Hello") end),
?line ok = inet:setopts(A,[{active,true}]),
?line ok = receive {tcp_closed, A} -> ok after 1000 -> error end,
?line {error,einval} = inet:setopts(A,[{active,once}]),
?line ok = receive {tcp_closed, A} -> error after 1000 -> ok end
end)(),
(fun() ->
?line {Loop,A} = setup_closed_ao(),
?line Loop({{error,closed},{error,econnaborted}},
fun() -> gen_tcp:send(A,"Hello") end),
?line ok = inet:setopts(A,[{active,once}]),
?line ok = receive {tcp_closed, A} -> ok after 1000 -> error end,
?line {error,einval} = inet:setopts(A,[{active,true}]),
?line ok = receive {tcp_closed, A} -> error after 1000 -> ok end
end)(),
(fun() ->
?line {Loop,A} = setup_closed_ao(),
?line Loop({{error,closed},{error,econnaborted}},
fun() -> gen_tcp:send(A,"Hello") end),
?line ok = inet:setopts(A,[{active,false}]),
?line ok = receive {tcp_closed, A} -> error after 1000 -> ok end,
?line ok = inet:setopts(A,[{active,once}]),
?line ok = receive {tcp_closed, A} -> ok after 1000 -> error end
end)().
send_timeout(suite) ->
[];
send_timeout(doc) ->
["Test the send_timeout socket option"];
send_timeout(Config) when is_list(Config) ->
%% Basic
BasicFun =
fun(AutoClose) ->
?line {Loop,A,RNode} = setup_timeout_sink(1000, AutoClose),
?line {error,timeout} =
Loop(fun() ->
Res = gen_tcp:send(A,<<1:10000>>),
%%erlang:display(Res),
Res
end),
%% Check that the socket is not busy/closed...
Error = after_send_timeout(AutoClose),
?line {error,Error} = gen_tcp:send(A,<<"Hej">>),
?line test_server:stop_node(RNode)
end,
BasicFun(false),
BasicFun(true),
%% Check timeout length
?line Self = self(),
?line Pid =
spawn(fun() ->
{Loop,A,RNode} = setup_timeout_sink(1000, true),
{error,timeout} =
Loop(fun() ->
Res = gen_tcp:send(A,<<1:10000>>),
%%erlang:display(Res),
Self ! Res,
Res
end),
test_server:stop_node(RNode)
end),
?line Diff = get_max_diff(),
?line io:format("Max time for send: ~p~n",[Diff]),
?line true = (Diff > 500) and (Diff < 1500),
%% Let test_server slave die...
?line Mon = erlang:monitor(process, Pid),
?line receive {'DOWN',Mon,process,Pid,_} -> ok end,
%% Check that parallell writers do not hang forever
ParaFun =
fun(AutoClose) ->
?line {Loop,A,RNode} = setup_timeout_sink(1000, AutoClose),
SenderFun = fun() ->
{error,Error} =
Loop(fun() ->
gen_tcp:send(A, <<1:10000>>)
end),
Self ! {error,Error}
end,
?line spawn_link(SenderFun),
?line spawn_link(SenderFun),
?line receive
{error,timeout} -> ok
after 10000 ->
?line exit(timeout)
end,
NextErr = after_send_timeout(AutoClose),
?line receive
{error,NextErr} -> ok
after 10000 ->
?line exit(timeout)
end,
?line {error,NextErr} = gen_tcp:send(A,<<"Hej">>),
?line test_server:stop_node(RNode)
end,
ParaFun(false),
ParaFun(true),
ok.
mad_sender(S) ->
{_, _, USec} = now(),
case gen_tcp:send(S, integer_to_list(USec)) of
ok ->
mad_sender(S);
Err ->
Err
end.
flush() ->
receive
_X ->
%erlang:display(_X),
flush()
after 0 ->
ok
end.
send_timeout_active(suite) ->
[];
send_timeout_active(doc) ->
["Test the send_timeout socket option for active sockets"];
send_timeout_active(Config) when is_list(Config) ->
Dog = test_server:timetrap(test_server:seconds(20)),
%% Basic
BasicFun =
fun(AutoClose) ->
?line {Loop,A,RNode,C} = setup_active_timeout_sink(1, AutoClose),
inet:setopts(A, [{active, once}]),
?line Mad = spawn_link(RNode,fun() -> mad_sender(C) end),
?line {error,timeout} =
Loop(fun() ->
receive
{tcp, _Sock, _Data} ->
inet:setopts(A, [{active, once}]),
Res = gen_tcp:send(A,lists:duplicate(1000, $a)),
%erlang:display(Res),
Res;
Err ->
io:format("sock closed: ~p~n", [Err]),
Err
end
end),
unlink(Mad),
exit(Mad,kill),
?line test_server:stop_node(RNode)
end,
BasicFun(false),
flush(),
BasicFun(true),
flush(),
test_server:timetrap_cancel(Dog),
ok.
after_send_timeout(AutoClose) ->
case AutoClose of
true -> enotconn;
false -> timeout
end.
get_max_diff() ->
receive
ok ->
get_max_diff(0)
after 10000 ->
exit(timeout)
end.
get_max_diff(Max) ->
T1 = millistamp(),
receive
ok ->
Diff = millistamp() - T1,
if
Diff > Max ->
get_max_diff(Diff);
true ->
get_max_diff(Max)
end;
{error,timeout} ->
Diff = millistamp() - T1,
if
Diff > Max ->
Diff;
true ->
Max
end
after 10000 ->
exit(timeout)
end.
setup_closed_ao() ->
Dir = filename:dirname(code:which(?MODULE)),
{ok,R} = test_server:start_node(test_default_options_slave,slave,
[{args,"-pa " ++ Dir}]),
Host = list_to_atom(lists:nth(2,string:tokens(atom_to_list(node()),"@"))),
{ok, L} = gen_tcp:listen(0, [{active,false},{packet,2}]),
Fun = fun(F) ->
receive
{From,X} when is_function(X) ->
From ! {self(),X()}, F(F);
die -> ok
end
end,
Pid = rpc:call(R,erlang,spawn,[fun() -> Fun(Fun) end]),
{ok, Port} = inet:port(L),
Remote = fun(Fu) ->
Pid ! {self(), Fu},
receive {Pid,X} -> X
end
end,
{ok, C} = Remote(fun() ->
gen_tcp:connect(Host,Port,
[{active,false},{packet,2}])
end),
{ok,A} = gen_tcp:accept(L),
gen_tcp:send(A,"Hello"),
{ok, "Hello"} = Remote(fun() -> gen_tcp:recv(C,0) end),
ok = Remote(fun() -> gen_tcp:close(C) end),
Loop2 = fun(_,_,_,0) ->
{failure, timeout};
(L2,{MA,MB},F2,N) ->
case F2() of
MA -> MA;
MB -> MB;
Other -> io:format("~p~n",[Other]),
receive after 1000 -> ok end,
L2(L2,{MA,MB},F2,N-1)
end
end,
Loop = fun(Match2,F3) -> Loop2(Loop2,Match2,F3,10) end,
test_server:stop_node(R),
{Loop,A}.
setup_timeout_sink(Timeout, AutoClose) ->
?line Dir = filename:dirname(code:which(?MODULE)),
?line {ok,R} = test_server:start_node(test_default_options_slave,slave,
[{args,"-pa " ++ Dir}]),
?line Host = list_to_atom(lists:nth(2,string:tokens(atom_to_list(node()),"@"))),
?line {ok, L} = gen_tcp:listen(0, [{active,false},{packet,2},
{send_timeout,Timeout},
{send_timeout_close,AutoClose}]),
?line Fun = fun(F) ->
receive
{From,X} when is_function(X) ->
From ! {self(),X()}, F(F);
die -> ok
end
end,
?line Pid = rpc:call(R,erlang,spawn,[fun() -> Fun(Fun) end]),
?line {ok, Port} = inet:port(L),
?line Remote = fun(Fu) ->
Pid ! {self(), Fu},
receive {Pid,X} -> X
end
end,
?line {ok, C} = Remote(fun() ->
gen_tcp:connect(Host,Port,
[{active,false},{packet,2}])
end),
?line {ok,A} = gen_tcp:accept(L),
?line gen_tcp:send(A,"Hello"),
?line {ok, "Hello"} = Remote(fun() -> gen_tcp:recv(C,0) end),
?line Loop2 = fun(_,_,0) ->
{failure, timeout};
(L2,F2,N) ->
Ret = F2(),
io:format("~p~n",[Ret]),
case Ret of
ok -> receive after 1 -> ok end,
L2(L2,F2,N-1);
Other -> Other
end
end,
?line Loop = fun(F3) -> Loop2(Loop2,F3,1000) end,
{Loop,A,R}.
setup_active_timeout_sink(Timeout, AutoClose) ->
?line Dir = filename:dirname(code:which(?MODULE)),
?line {ok,R} = test_server:start_node(test_default_options_slave,slave,
[{args,"-pa " ++ Dir}]),
?line Host = list_to_atom(lists:nth(2,string:tokens(atom_to_list(node()),"@"))),
?line {ok, L} = gen_tcp:listen(0, [binary,{active,false},{packet,0},{nodelay, true},{keepalive, true},
{send_timeout,Timeout},
{send_timeout_close,AutoClose}]),
?line Fun = fun(F) ->
receive
{From,X} when is_function(X) ->
From ! {self(),X()}, F(F);
die -> ok
end
end,
?line Pid = rpc:call(R,erlang,spawn,[fun() -> Fun(Fun) end]),
?line {ok, Port} = inet:port(L),
?line Remote = fun(Fu) ->
Pid ! {self(), Fu},
receive {Pid,X} -> X
end
end,
?line {ok, C} = Remote(fun() ->
gen_tcp:connect(Host,Port,
[{active,false}])
end),
?line {ok,A} = gen_tcp:accept(L),
?line gen_tcp:send(A,"Hello"),
?line {ok, "H"++_} = Remote(fun() -> gen_tcp:recv(C,0) end),
?line Loop2 = fun(_,_,0) ->
{failure, timeout};
(L2,F2,N) ->
Ret = F2(),
io:format("~p~n",[Ret]),
case Ret of
ok -> receive after 1 -> ok end,
L2(L2,F2,N-1);
Other -> Other
end
end,
?line Loop = fun(F3) -> Loop2(Loop2,F3,1000) end,
{Loop,A,R,C}.
millistamp() ->
{Mega, Secs, Micros} = erlang:now(),
(Micros div 1000) + Secs * 1000 + Mega * 1000000000.
has_superfluous_schedulers() ->
case {erlang:system_info(schedulers),
erlang:system_info(logical_processors)} of
{S, unknown} when S > 1 -> true;
{S, P} when S > P -> true;
_ -> false
end.
otp_7731(suite) -> [];
otp_7731(doc) ->
"Leaking message from inet_drv {inet_reply,P,ok} "
"when a socket sending resumes working after a send_timeout";
otp_7731(Config) when is_list(Config) ->
?line ServerPid = spawn_link(?MODULE, otp_7731_server, [self()]),
?line receive {ServerPid, ready, PortNum} -> ok end,
?line {ok, Socket} = gen_tcp:connect("localhost", PortNum,
[binary, {active, false}, {packet, raw},
{send_timeout, 1000}]),
otp_7731_send(Socket),
io:format("Sending complete...\n",[]),
ServerPid ! {self(), recv},
receive {ServerPid, ok} -> ok end,
io:format("Client waiting for leaking messages...\n",[]),
%% Now make sure inet_drv does not leak any internal messages.
receive Msg ->
?line test_server:fail({unexpected, Msg})
after 1000 ->
ok
end,
io:format("No leaking messages. Done.\n",[]),
gen_tcp:close(Socket).
otp_7731_send(Socket) ->
Bin = <<1:10000>>,
io:format("Client sending ~p bytes...\n",[size(Bin)]),
?line case gen_tcp:send(Socket, Bin) of
ok -> otp_7731_send(Socket);
{error,timeout} -> ok
end.
otp_7731_server(ClientPid) ->
?line {ok, LSocket} = gen_tcp:listen(0, [binary, {packet, raw},
{active, false}]),
?line {ok, {_, PortNum}} = inet:sockname(LSocket),
io:format("Listening on ~w with port number ~p\n", [LSocket, PortNum]),
ClientPid ! {self(), ready, PortNum},
{ok, CSocket} = gen_tcp:accept(LSocket),
gen_tcp:close(LSocket),
io:format("Server got connection, wait for recv order...\n",[]),
receive {ClientPid, recv} -> ok end,
io:format("Server start receiving...\n",[]),
otp_7731_recv(CSocket),
ClientPid ! {self(), ok},
io:format("Server finished, closing...\n",[]),
gen_tcp:close(CSocket).
otp_7731_recv(Socket) ->
?line case gen_tcp:recv(Socket, 0, 1000) of
{ok, Bin} ->
io:format("Server received ~p bytes\n",[size(Bin)]),
otp_7731_recv(Socket);
{error,timeout} ->
io:format("Server got receive timeout\n",[]),
ok
end.
%% OTP-7615: TCP-ports hanging in CLOSING state when sending large
%% buffer followed by a recv() that returns error due to closed
%% connection.
zombie_sockets(suite) -> [];
zombie_sockets(doc) -> ["OTP-7615 Leaking closed ports."];
zombie_sockets(Config) when is_list(Config) ->
register(zombie_collector,self()),
Calls = 10,
Server = spawn_link(?MODULE, zombie_server,[self(), Calls]),
?line {Server, ready, PortNum} = receive Msg -> Msg end,
io:format("Ports before = ~p\n",[lists:sort(erlang:ports())]),
zombie_client_loop(Calls, PortNum),
Ports = lists:sort(zombie_collector(Calls,[])),
Server ! terminate,
io:format("Collected ports = ~p\n",[Ports]),
?line [] = zombies_alive(Ports, 10),
timer:sleep(1000),
ok.
zombie_client_loop(0, _) -> ok;
zombie_client_loop(N, PortNum) when is_integer(PortNum) ->
?line {ok, Socket} = gen_tcp:connect("localhost", PortNum,
[binary, {active, false}, {packet, raw}]),
?line gen_tcp:close(Socket), % to make server recv fail
zombie_client_loop(N-1, PortNum).
zombie_collector(0,Acc) ->
Acc;
zombie_collector(N,Acc) ->
receive
{closed, Socket} ->
zombie_collector(N-1,[Socket|Acc]);
E ->
{unexpected, E, Acc}
end.
zombies_alive(Ports, WaitSec) ->
Alive = lists:sort(erlang:ports()),
io:format("Alive = ~p\n",[Alive]),
Zombies = lists:filter(fun(P) -> lists:member(P, Alive) end, Ports),
case Zombies of
[] -> [];
_ ->
case WaitSec of
0 -> Zombies;
_ -> timer:sleep(1000), % Wait some more for zombies to die
zombies_alive(Zombies, WaitSec-1)
end
end.
zombie_server(Pid, Calls) ->
?line {ok, LSocket} = gen_tcp:listen(0, [binary, {packet, raw},
{active, false}, {backlog, Calls}]),
?line {ok, {_, PortNum}} = inet:sockname(LSocket),
io:format("Listening on ~w with port number ~p\n", [LSocket, PortNum]),
BigBin = list_to_binary(lists:duplicate(100*1024, 77)),
Pid ! {self(), ready, PortNum},
zombie_accept_loop(LSocket, BigBin, Calls),
?line terminate = receive Msg -> Msg end.
zombie_accept_loop(_, _, 0) ->
ok;
zombie_accept_loop(Socket, BigBin, Calls) ->
?line case gen_tcp:accept(Socket) of
{ok, NewSocket} ->
spawn_link(fun() -> zombie_serve_client(NewSocket, BigBin) end),
zombie_accept_loop(Socket, BigBin, Calls-1);
E ->
E
end.
zombie_serve_client(Socket, Bin) ->
%%io:format("Got connection on ~p\n",[Socket]),
?line gen_tcp:send(Socket, Bin),
%%io:format("Sent data, waiting for reply on ~p\n",[Socket]),
?line case gen_tcp:recv(Socket, 4) of
{error,closed} -> ok;
{error,econnaborted} -> ok % may be returned on Windows
end,
%%io:format("Closing ~p\n",[Socket]),
?line gen_tcp:close(Socket),
zombie_collector ! {closed, Socket}.
otp_7816(suite) -> [];
otp_7816(doc) ->
"Hanging send on windows when sending iolist with more than 16 binaries.";
otp_7816(Config) when is_list(Config) ->
Client = self(),
?line Server = spawn_link(fun()-> otp_7816_server(Client) end),
?line receive {Server, ready, PortNum} -> ok end,
?line {ok, Socket} = gen_tcp:connect("localhost", PortNum,
[binary, {active, false}, {packet, 4},
{send_timeout, 10}]),
%% We use the undocumented feature that sending can be resumed after
%% a send_timeout without any data loss if the peer starts to receive data.
%% Unless of course the 7816-bug is in affect, in which case the write event
%% for the socket is lost on windows and not all data is sent.
[otp_7816_send(Socket,18,BinSize,Server) || BinSize <- lists:seq(1000, 2000, 123)],
io:format("Sending complete...\n",[]),
?line ok = gen_tcp:close(Socket),
Server ! {self(), closed},
?line {Server, closed} = receive M -> M end.
otp_7816_send(Socket, BinNr, BinSize, Server) ->
Data = lists:duplicate(BinNr, <<1:(BinSize*8)>>),
SentBytes = otp_7816_send_data(Socket, Data, 0) * BinNr * BinSize,
io:format("Client sent ~p bytes...\n",[SentBytes]),
Server ! {self(),recv,SentBytes},
?line {Server, ok} = receive M -> M end.
otp_7816_send_data(Socket, Data, Loops) ->
io:format("Client sending data...\n",[]),
case gen_tcp:send(Socket, Data) of
ok ->
otp_7816_send_data(Socket,Data, Loops+1);
{error,timeout} ->
Loops+1
end.
otp_7816_server(Client) ->
?line {ok, LSocket} = gen_tcp:listen(0, [binary, {packet, 4},
{active, false}]),
?line {ok, {_, PortNum}} = inet:sockname(LSocket),
io:format("Listening on ~w with port number ~p\n", [LSocket, PortNum]),
Client ! {self(), ready, PortNum},
?line {ok, CSocket} = gen_tcp:accept(LSocket),
io:format("Server got connection...\n",[]),
?line gen_tcp:close(LSocket),
otp_7816_server_loop(CSocket),
io:format("Server terminating.\n",[]).
otp_7816_server_loop(CSocket) ->
io:format("Server waiting for order...\n",[]),
receive
{Client, recv, RecvBytes} ->
io:format("Server start receiving...\n",[]),
?line ok = otp_7816_recv(CSocket, RecvBytes),
Client ! {self(), ok},
otp_7816_server_loop(CSocket);
{Client, closed} ->
?line {error, closed} = gen_tcp:recv(CSocket, 0, 1000),
Client ! {self(), closed}
end.
otp_7816_recv(_, 0) ->
io:format("Server got all.\n",[]),
ok;
otp_7816_recv(CSocket, BytesLeft) ->
?line case gen_tcp:recv(CSocket, 0, 1000) of
{ok, Bin} when byte_size(Bin) =< BytesLeft ->
io:format("Server received ~p of ~p bytes.\n",[size(Bin), BytesLeft]),
otp_7816_recv(CSocket, BytesLeft - byte_size(Bin));
{error,timeout} ->
io:format("Server got receive timeout when expecting more data\n",[]),
error
end.
otp_8102(doc) -> ["Receive a packet with a faulty packet header"];
otp_8102(suite) -> [];
otp_8102(Config) when is_list(Config) ->
?line {ok, LSocket} = gen_tcp:listen(0, []),
?line {ok, {_, PortNum}} = inet:sockname(LSocket),
io:format("Listening on ~w with port number ~p\n", [LSocket, PortNum]),
[otp_8102_do(LSocket, PortNum, otp_8102_packet(Type,Size))
|| Size <- lists:seq(-10,-1),
Type <- [4, {cdr,big}, {cdr,little}]],
gen_tcp:close(LSocket),
ok.
otp_8102_packet(4, Size) ->
{<<Size:32/big>>, 4};
otp_8102_packet({cdr,big}, Size) ->
{<<"GIOP",0,0,0,0,Size:32/big>>, cdr};
otp_8102_packet({cdr,little}, Size) ->
{<<"GIOP",0,0,1,0,Size:32/little>>, cdr}.
otp_8102_do(LSocket, PortNum, {Bin,PType}) ->
io:format("Connect with packet option ~p ...\n",[PType]),
?line {ok, RSocket} = gen_tcp:connect("localhost", PortNum, [binary,
{packet,PType},
{active,true}]),
?line {ok, SSocket} = gen_tcp:accept(LSocket),
io:format("Got connection, sending ~p...\n",[Bin]),
?line ok = gen_tcp:send(SSocket, Bin),
io:format("Sending complete...\n",[]),
?line {tcp_error,RSocket,emsgsize} = receive M -> M end,
io:format("Got error msg, ok.\n",[]),
gen_tcp:close(SSocket),
gen_tcp:close(RSocket).
otp_9389(doc) -> ["Verify packet_size handles long HTTP header lines"];
otp_9389(suite) -> [];
otp_9389(Config) when is_list(Config) ->
?line {ok, LS} = gen_tcp:listen(0, [{active,false}]),
?line {ok, {_, PortNum}} = inet:sockname(LS),
io:format("Listening on ~w with port number ~p\n", [LS, PortNum]),
OrigLinkHdr = "/" ++ string:chars($S, 8192),
_Server = spawn_link(
fun() ->
?line {ok, S} = gen_tcp:accept(LS),
?line ok = inet:setopts(S, [{packet_size, 16384}]),
?line ok = otp_9389_loop(S, OrigLinkHdr),
?line ok = gen_tcp:close(S)
end),
?line {ok, S} = gen_tcp:connect("localhost", PortNum,
[binary, {active, false}]),
Req = "GET / HTTP/1.1\r\n"
++ "Host: localhost\r\n"
++ "Link: " ++ OrigLinkHdr ++ "\r\n\r\n",
?line ok = gen_tcp:send(S, Req),
?line ok = inet:setopts(S, [{packet, http}]),
?line {ok, {http_response, {1,1}, 200, "OK"}} = gen_tcp:recv(S, 0),
?line ok = inet:setopts(S, [{packet, httph}, {packet_size, 16384}]),
?line {ok, {http_header, _, 'Content-Length', _, "0"}} = gen_tcp:recv(S, 0),
?line {ok, {http_header, _, "Link", _, LinkHdr}} = gen_tcp:recv(S, 0),
?line true = (LinkHdr == OrigLinkHdr),
ok = gen_tcp:close(S),
ok = gen_tcp:close(LS),
ok.
otp_9389_loop(S, OrigLinkHdr) ->
?line ok = inet:setopts(S, [{active,once},{packet,http}]),
receive
{http, S, {http_request, 'GET', _, _}} ->
?line ok = otp_9389_loop(S, OrigLinkHdr, undefined)
after
3000 ->
?line error({timeout,request_line})
end.
otp_9389_loop(S, OrigLinkHdr, ok) ->
?line Resp = "HTTP/1.1 200 OK\r\nContent-length: 0\r\n" ++
"Link: " ++ OrigLinkHdr ++ "\r\n\r\n",
?line ok = gen_tcp:send(S, Resp);
otp_9389_loop(S, OrigLinkHdr, State) ->
?line ok = inet:setopts(S, [{active,once}, {packet,httph}]),
receive
{http, S, http_eoh} ->
?line otp_9389_loop(S, OrigLinkHdr, ok);
{http, S, {http_header, _, "Link", _, LinkHdr}} ->
?line LinkHdr = OrigLinkHdr,
?line otp_9389_loop(S, OrigLinkHdr, State);
{http, S, {http_header, _, _Hdr, _, _Val}} ->
?line otp_9389_loop(S, OrigLinkHdr, State);
{http, S, {http_error, Err}} ->
?line error({error, Err})
after
3000 ->
?line error({timeout,header})
end.
wrapping_oct(doc) ->
"Check that 64bit octet counters work.";
wrapping_oct(suite) ->
[];
wrapping_oct(Config) when is_list(Config) ->
Dog = test_server:timetrap(test_server:seconds(600)),
{ok,Sock} = gen_tcp:listen(0,[{active,false},{mode,binary}]),
{ok,Port} = inet:port(Sock),
spawn_link(?MODULE,oct_acceptor,[Sock]),
Res = oct_datapump(Port,16#1FFFFFFFF),
gen_tcp:close(Sock),
test_server:timetrap_cancel(Dog),
ok = Res,
ok.
oct_datapump(Port,N) ->
{ok,Sock} = gen_tcp:connect("localhost",Port,
[{active,false},{mode,binary}]),
oct_pump(Sock,N,binary:copy(<<$a:8>>,100000),0).
oct_pump(S,N,_,_) when N =< 0 ->
gen_tcp:close(S),
ok;
oct_pump(S,N,Bin,Last) ->
case gen_tcp:send(S,Bin) of
ok ->
{ok,Stat}=inet:getstat(S),
{_,R}=lists:keyfind(send_oct,1,Stat),
case (R < Last) of
true ->
io:format("ERROR (output) ~p < ~p~n",[R,Last]),
output_counter_error;
false ->
oct_pump(S,N-byte_size(Bin),Bin,R)
end;
_ ->
input_counter_error
end.
oct_acceptor(Sock) ->
{ok,Data} = gen_tcp:accept(Sock),
oct_aloop(Data,0,0).
oct_aloop(S,X,Times) ->
case gen_tcp:recv(S,0) of
{ok,_} ->
{ok,Stat}=inet:getstat(S),
{_,R}=lists:keyfind(recv_oct,1,Stat),
case (R < X) of
true ->
io:format("ERROR ~p < ~p~n",[R,X]),
gen_tcp:close(S),
input_counter_error;
false ->
case Times rem 16#FFFFF of
0 ->
io:format("Read: ~p~n",[R]);
_ ->
ok
end,
oct_aloop(S,R,Times+1)
end;
_ ->
gen_tcp:close(S),
closed
end.
