Catch errors when calling inet:peername in ejabberd_http

This commit is contained in:
Paweł Chmielowski 2015-08-17 15:50:02 +02:00
parent 6d16029716
commit f89db30d77
1 changed files with 10 additions and 2 deletions

View File

@ -160,7 +160,11 @@ init({SockMod, Socket}, Opts) ->
default_host = DefaultHost,
options = Opts,
request_handlers = RequestHandlers},
receive_headers(State).
try receive_headers(State) of
V -> V
catch
{error, _} -> State
end.
become_controller(_Pid) -> ok.
@ -427,13 +431,17 @@ process_request(#state{request_method = Method,
false ->
make_bad_request(State);
{LPath, LQuery, Data} ->
{ok, IPHere} =
PeerName =
case SockMod of
gen_tcp ->
inet:peername(Socket);
_ ->
SockMod:peername(Socket)
end,
IPHere = case PeerName of
{ok, V} -> V;
{error, _} = E -> throw(E)
end,
XFF = proplists:get_value('X-Forwarded-For', RequestHeaders, []),
IP = analyze_ip_xff(IPHere, XFF, Host),
Request = #request{method = Method,