249 lines
7.8 KiB
Erlang
249 lines
7.8 KiB
Erlang
%% @author Bob Ippolito <bob@mochimedia.com>
|
|
%% @copyright 2007 Mochi Media, Inc.
|
|
|
|
%% @doc MochiWeb socket server.
|
|
|
|
-module(mochiweb_socket_server).
|
|
-author('bob@mochimedia.com').
|
|
-behaviour(gen_server).
|
|
|
|
-export([start/1, stop/1]).
|
|
-export([init/1, handle_call/3, handle_cast/2, terminate/2, code_change/3,
|
|
handle_info/2]).
|
|
-export([get/2]).
|
|
|
|
-export([acceptor_loop/1]).
|
|
|
|
-record(mochiweb_socket_server,
|
|
{port,
|
|
loop,
|
|
name=undefined,
|
|
max=2048,
|
|
ip=any,
|
|
listen=null,
|
|
acceptor=null,
|
|
backlog=30}).
|
|
|
|
start(State=#mochiweb_socket_server{}) ->
|
|
start_server(State);
|
|
start(Options) ->
|
|
start(parse_options(Options)).
|
|
|
|
get(Name, Property) ->
|
|
gen_server:call(Name, {get, Property}).
|
|
|
|
stop(Name) when is_atom(Name) ->
|
|
gen_server:cast(Name, stop);
|
|
stop(Pid) when is_pid(Pid) ->
|
|
gen_server:cast(Pid, stop);
|
|
stop({local, Name}) ->
|
|
stop(Name);
|
|
stop({global, Name}) ->
|
|
stop(Name);
|
|
stop(Options) ->
|
|
State = parse_options(Options),
|
|
stop(State#mochiweb_socket_server.name).
|
|
|
|
%% Internal API
|
|
|
|
parse_options(Options) ->
|
|
parse_options(Options, #mochiweb_socket_server{}).
|
|
|
|
parse_options([], State) ->
|
|
State;
|
|
parse_options([{name, L} | Rest], State) when is_list(L) ->
|
|
Name = {local, list_to_atom(L)},
|
|
parse_options(Rest, State#mochiweb_socket_server{name=Name});
|
|
parse_options([{name, A} | Rest], State) when is_atom(A) ->
|
|
Name = {local, A},
|
|
parse_options(Rest, State#mochiweb_socket_server{name=Name});
|
|
parse_options([{name, Name} | Rest], State) ->
|
|
parse_options(Rest, State#mochiweb_socket_server{name=Name});
|
|
parse_options([{port, L} | Rest], State) when is_list(L) ->
|
|
Port = list_to_integer(L),
|
|
parse_options(Rest, State#mochiweb_socket_server{port=Port});
|
|
parse_options([{port, Port} | Rest], State) ->
|
|
parse_options(Rest, State#mochiweb_socket_server{port=Port});
|
|
parse_options([{ip, Ip} | Rest], State) ->
|
|
ParsedIp = case Ip of
|
|
any ->
|
|
any;
|
|
Ip when is_tuple(Ip) ->
|
|
Ip;
|
|
Ip when is_list(Ip) ->
|
|
{ok, IpTuple} = inet_parse:address(Ip),
|
|
IpTuple
|
|
end,
|
|
parse_options(Rest, State#mochiweb_socket_server{ip=ParsedIp});
|
|
parse_options([{loop, Loop} | Rest], State) ->
|
|
parse_options(Rest, State#mochiweb_socket_server{loop=Loop});
|
|
parse_options([{backlog, Backlog} | Rest], State) ->
|
|
parse_options(Rest, State#mochiweb_socket_server{backlog=Backlog});
|
|
parse_options([{max, Max} | Rest], State) ->
|
|
MaxInt = case Max of
|
|
Max when is_list(Max) ->
|
|
list_to_integer(Max);
|
|
Max when is_integer(Max) ->
|
|
Max
|
|
end,
|
|
parse_options(Rest, State#mochiweb_socket_server{max=MaxInt}).
|
|
|
|
start_server(State=#mochiweb_socket_server{name=Name}) ->
|
|
case Name of
|
|
undefined ->
|
|
gen_server:start_link(?MODULE, State, []);
|
|
_ ->
|
|
gen_server:start_link(Name, ?MODULE, State, [])
|
|
end.
|
|
|
|
ipv6_supported() ->
|
|
case (catch inet:getaddr("localhost", inet6)) of
|
|
{ok, _Addr} ->
|
|
true;
|
|
{error, _} ->
|
|
false
|
|
end.
|
|
|
|
init(State=#mochiweb_socket_server{ip=Ip, port=Port, backlog=Backlog}) ->
|
|
process_flag(trap_exit, true),
|
|
BaseOpts = [binary,
|
|
{reuseaddr, true},
|
|
{packet, 0},
|
|
{backlog, Backlog},
|
|
{recbuf, 8192},
|
|
{active, false},
|
|
{nodelay, true}],
|
|
Opts = case Ip of
|
|
any ->
|
|
case ipv6_supported() of % IPv4, and IPv6 if supported
|
|
true -> [inet, inet6 | BaseOpts];
|
|
_ -> BaseOpts
|
|
end;
|
|
{_, _, _, _} -> % IPv4
|
|
[inet, {ip, Ip} | BaseOpts];
|
|
{_, _, _, _, _, _, _, _} -> % IPv6
|
|
[inet6, {ip, Ip} | BaseOpts]
|
|
end,
|
|
case gen_tcp_listen(Port, Opts, State) of
|
|
{stop, eacces} ->
|
|
case Port < 1024 of
|
|
true ->
|
|
case fdsrv:start() of
|
|
{ok, _} ->
|
|
case fdsrv:bind_socket(tcp, Port) of
|
|
{ok, Fd} ->
|
|
gen_tcp_listen(Port, [{fd, Fd} | Opts], State);
|
|
_ ->
|
|
{stop, fdsrv_bind_failed}
|
|
end;
|
|
_ ->
|
|
{stop, fdsrv_start_failed}
|
|
end;
|
|
false ->
|
|
{stop, eacces}
|
|
end;
|
|
Other ->
|
|
Other
|
|
end.
|
|
|
|
gen_tcp_listen(Port, Opts, State) ->
|
|
case gen_tcp:listen(Port, Opts) of
|
|
{ok, Listen} ->
|
|
{ok, ListenPort} = inet:port(Listen),
|
|
{ok, new_acceptor(State#mochiweb_socket_server{listen=Listen,
|
|
port=ListenPort})};
|
|
{error, Reason} ->
|
|
{stop, Reason}
|
|
end.
|
|
|
|
new_acceptor(State=#mochiweb_socket_server{max=0}) ->
|
|
io:format("Not accepting new connections~n"),
|
|
State#mochiweb_socket_server{acceptor=null};
|
|
new_acceptor(State=#mochiweb_socket_server{listen=Listen,loop=Loop}) ->
|
|
Pid = proc_lib:spawn_link(?MODULE, acceptor_loop,
|
|
[{self(), Listen, Loop}]),
|
|
State#mochiweb_socket_server{acceptor=Pid}.
|
|
|
|
call_loop({M, F}, Socket) ->
|
|
M:F(Socket);
|
|
call_loop(Loop, Socket) ->
|
|
Loop(Socket).
|
|
|
|
acceptor_loop({Server, Listen, Loop}) ->
|
|
case catch gen_tcp:accept(Listen) of
|
|
{ok, Socket} ->
|
|
gen_server:cast(Server, {accepted, self()}),
|
|
call_loop(Loop, Socket);
|
|
{error, closed} ->
|
|
exit({error, closed});
|
|
Other ->
|
|
error_logger:error_report(
|
|
[{application, mochiweb},
|
|
"Accept failed error",
|
|
lists:flatten(io_lib:format("~p", [Other]))]),
|
|
exit({error, accept_failed})
|
|
end.
|
|
|
|
|
|
do_get(port, #mochiweb_socket_server{port=Port}) ->
|
|
Port.
|
|
|
|
handle_call({get, Property}, _From, State) ->
|
|
Res = do_get(Property, State),
|
|
{reply, Res, State};
|
|
handle_call(_Message, _From, State) ->
|
|
Res = error,
|
|
{reply, Res, State}.
|
|
|
|
handle_cast({accepted, Pid},
|
|
State=#mochiweb_socket_server{acceptor=Pid, max=Max}) ->
|
|
% io:format("accepted ~p~n", [Pid]),
|
|
State1 = State#mochiweb_socket_server{max=Max - 1},
|
|
{noreply, new_acceptor(State1)};
|
|
handle_cast(stop, State) ->
|
|
{stop, normal, State}.
|
|
|
|
terminate(_Reason, #mochiweb_socket_server{listen=Listen, port=Port}) ->
|
|
gen_tcp:close(Listen),
|
|
case Port < 1024 of
|
|
true ->
|
|
catch fdsrv:stop(),
|
|
ok;
|
|
false ->
|
|
ok
|
|
end.
|
|
|
|
code_change(_OldVsn, State, _Extra) ->
|
|
State.
|
|
|
|
handle_info({'EXIT', Pid, normal},
|
|
State=#mochiweb_socket_server{acceptor=Pid}) ->
|
|
% io:format("normal acceptor down~n"),
|
|
{noreply, new_acceptor(State)};
|
|
handle_info({'EXIT', Pid, Reason},
|
|
State=#mochiweb_socket_server{acceptor=Pid}) ->
|
|
error_logger:error_report({?MODULE, ?LINE,
|
|
{acceptor_error, Reason}}),
|
|
timer:sleep(100),
|
|
{noreply, new_acceptor(State)};
|
|
handle_info({'EXIT', _LoopPid, Reason},
|
|
State=#mochiweb_socket_server{acceptor=Pid, max=Max}) ->
|
|
case Reason of
|
|
normal ->
|
|
ok;
|
|
_ ->
|
|
error_logger:error_report({?MODULE, ?LINE,
|
|
{child_error, Reason}})
|
|
end,
|
|
State1 = State#mochiweb_socket_server{max=Max + 1},
|
|
State2 = case Pid of
|
|
null ->
|
|
new_acceptor(State1);
|
|
_ ->
|
|
State1
|
|
end,
|
|
{noreply, State2};
|
|
handle_info(Info, State) ->
|
|
error_logger:info_report([{'INFO', Info}, {'State', State}]),
|
|
{noreply, State}.
|