2006-09-05 06:26:28 +02:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
%%% File : ejabberd_socket.erl
|
|
|
|
%%% Author : Alexey Shchepin <alexey@process-one.net>
|
|
|
|
%%% Purpose : Socket with zlib and TLS support library
|
2007-12-24 12:41:41 +01:00
|
|
|
%%% Created : 23 Aug 2006 by Alexey Shchepin <alexey@process-one.net>
|
|
|
|
%%%
|
|
|
|
%%%
|
2013-03-27 10:36:48 +01:00
|
|
|
%%% ejabberd, Copyright (C) 2002-2013 ProcessOne
|
2007-12-24 12:41:41 +01:00
|
|
|
%%%
|
|
|
|
%%% This program is free software; you can redistribute it and/or
|
|
|
|
%%% modify it under the terms of the GNU General Public License as
|
|
|
|
%%% published by the Free Software Foundation; either version 2 of the
|
|
|
|
%%% License, or (at your option) any later version.
|
|
|
|
%%%
|
|
|
|
%%% This program is distributed in the hope that it will be useful,
|
|
|
|
%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
%%% General Public License for more details.
|
2009-01-12 15:44:42 +01:00
|
|
|
%%%
|
2007-12-24 12:41:41 +01:00
|
|
|
%%% You should have received a copy of the GNU General Public License
|
|
|
|
%%% along with this program; if not, write to the Free Software
|
|
|
|
%%% Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
|
|
|
|
%%% 02111-1307 USA
|
|
|
|
%%%
|
2006-09-05 06:26:28 +02:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
|
|
|
|
-module(ejabberd_socket).
|
2012-09-11 15:45:59 +02:00
|
|
|
|
2006-09-05 06:26:28 +02:00
|
|
|
-author('alexey@process-one.net').
|
|
|
|
|
|
|
|
%% API
|
2012-09-11 15:45:59 +02:00
|
|
|
-export([init/0, start/4, connect/3, connect/4, starttls/2,
|
|
|
|
starttls/3, compress/1, compress/2, reset_stream/1,
|
|
|
|
send/2, send_xml/2, change_shaper/2, monitor/1,
|
|
|
|
get_sockmod/1, get_peer_certificate/1, get_conn_type/1,
|
|
|
|
get_verify_result/1, close/1, change_controller/2,
|
|
|
|
change_socket/2, sockname/1, peername/1, is_remote_receiver/1]).
|
2006-09-05 06:26:28 +02:00
|
|
|
|
2009-10-19 11:36:23 +02:00
|
|
|
-include("ejabberd.hrl").
|
2012-09-11 15:45:59 +02:00
|
|
|
-include("jlib.hrl").
|
2009-10-19 11:36:23 +02:00
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
-type sockmod() :: ejabberd_http_poll | ejabberd_bosh |
|
|
|
|
ejabberd_http_bind | ejabberd_http_bindjson |
|
|
|
|
ejabberd_http_ws | ejabberd_http_wsjson |
|
|
|
|
gen_tcp | tls | ejabberd_zlib.
|
|
|
|
-type receiver() :: pid () | atom().
|
|
|
|
-type socket() :: pid() | inet:socket() |
|
|
|
|
tls:tls_socket() |
|
|
|
|
ejabberd_zlib:zlib_socket() |
|
|
|
|
ejabberd_bosh:bosh_socket() |
|
|
|
|
ejabberd_http_ws:ws_socket() |
|
|
|
|
ejabberd_http_poll:poll_socket().
|
|
|
|
|
|
|
|
-record(socket_state, {sockmod = gen_tcp :: sockmod(),
|
|
|
|
socket = self() :: socket(),
|
|
|
|
receiver = self() :: receiver()}).
|
|
|
|
|
|
|
|
-type socket_state() :: #socket_state{}.
|
|
|
|
|
|
|
|
-export_type([socket_state/0, sockmod/0]).
|
|
|
|
|
|
|
|
-spec start(atom(), sockmod(), socket(), [{atom(), any()}]) -> any().
|
|
|
|
|
|
|
|
init() -> #socket_state{}.
|
2006-09-05 06:26:28 +02:00
|
|
|
|
|
|
|
start(Module, SockMod, Socket, Opts) ->
|
2006-09-25 05:51:11 +02:00
|
|
|
case Module:socket_type() of
|
2012-09-11 15:45:59 +02:00
|
|
|
xml_stream ->
|
|
|
|
MaxStanzaSize = case lists:keysearch(max_stanza_size, 1,
|
|
|
|
Opts)
|
|
|
|
of
|
|
|
|
{value, {_, Size}} -> Size;
|
|
|
|
_ -> infinity
|
|
|
|
end,
|
|
|
|
{ReceiverMod, Receiver, RecRef} = case catch
|
|
|
|
SockMod:custom_receiver(Socket)
|
|
|
|
of
|
|
|
|
{receiver, RecMod, RecPid} ->
|
|
|
|
{RecMod, RecPid, RecMod};
|
|
|
|
_ ->
|
|
|
|
RecPid =
|
|
|
|
ejabberd_receiver:start(Socket,
|
|
|
|
SockMod,
|
|
|
|
none,
|
|
|
|
MaxStanzaSize),
|
|
|
|
{ejabberd_receiver, RecPid,
|
|
|
|
RecPid}
|
|
|
|
end,
|
|
|
|
SocketData = #socket_state{sockmod = SockMod,
|
|
|
|
socket = Socket, receiver = RecRef},
|
|
|
|
case Module:start({?MODULE, SocketData}, Opts) of
|
|
|
|
{ok, Pid} ->
|
|
|
|
case SockMod:controlling_process(Socket, Receiver) of
|
|
|
|
ok -> ok;
|
|
|
|
{error, _Reason} -> SockMod:close(Socket)
|
2009-08-31 20:37:52 +02:00
|
|
|
end,
|
2012-09-11 15:45:59 +02:00
|
|
|
ReceiverMod:become_controller(Receiver, Pid);
|
|
|
|
{error, _Reason} ->
|
|
|
|
SockMod:close(Socket),
|
|
|
|
case ReceiverMod of
|
|
|
|
ejabberd_receiver -> ReceiverMod:close(Receiver);
|
|
|
|
_ -> ok
|
|
|
|
end
|
|
|
|
end;
|
|
|
|
independent -> ok;
|
|
|
|
raw ->
|
|
|
|
case Module:start({SockMod, Socket}, Opts) of
|
|
|
|
{ok, Pid} ->
|
|
|
|
case SockMod:controlling_process(Socket, Pid) of
|
|
|
|
ok -> ok;
|
|
|
|
{error, _Reason} -> SockMod:close(Socket)
|
|
|
|
end;
|
|
|
|
{error, _Reason} -> SockMod:close(Socket)
|
|
|
|
end
|
2006-09-25 05:51:11 +02:00
|
|
|
end.
|
2006-09-05 06:26:28 +02:00
|
|
|
|
|
|
|
connect(Addr, Port, Opts) ->
|
2008-12-26 12:45:08 +01:00
|
|
|
connect(Addr, Port, Opts, infinity).
|
|
|
|
|
|
|
|
connect(Addr, Port, Opts, Timeout) ->
|
|
|
|
case gen_tcp:connect(Addr, Port, Opts, Timeout) of
|
2012-09-11 15:45:59 +02:00
|
|
|
{ok, Socket} ->
|
|
|
|
Receiver = ejabberd_receiver:start(Socket, gen_tcp,
|
|
|
|
none),
|
|
|
|
SocketData = #socket_state{sockmod = gen_tcp,
|
|
|
|
socket = Socket, receiver = Receiver},
|
|
|
|
Pid = self(),
|
|
|
|
case gen_tcp:controlling_process(Socket, Receiver) of
|
|
|
|
ok ->
|
|
|
|
ejabberd_receiver:become_controller(Receiver, Pid),
|
|
|
|
{ok, SocketData};
|
|
|
|
{error, _Reason} = Error -> gen_tcp:close(Socket), Error
|
|
|
|
end;
|
|
|
|
{error, _Reason} = Error -> Error
|
2006-09-05 06:26:28 +02:00
|
|
|
end.
|
|
|
|
|
|
|
|
starttls(SocketData, TLSOpts) ->
|
2010-07-13 12:04:30 +02:00
|
|
|
starttls(SocketData, TLSOpts, undefined).
|
2006-09-05 06:26:28 +02:00
|
|
|
|
2006-10-01 03:53:37 +02:00
|
|
|
starttls(SocketData, TLSOpts, Data) ->
|
2012-09-11 15:45:59 +02:00
|
|
|
{ok, TLSSocket} =
|
|
|
|
ejabberd_receiver:starttls(SocketData#socket_state.receiver,
|
|
|
|
TLSOpts, Data),
|
|
|
|
SocketData#socket_state{socket = TLSSocket,
|
|
|
|
sockmod = tls}.
|
2006-10-01 03:53:37 +02:00
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
compress(SocketData) -> compress(SocketData, undefined).
|
2006-09-05 06:26:28 +02:00
|
|
|
|
2006-10-01 03:53:37 +02:00
|
|
|
compress(SocketData, Data) ->
|
2012-09-11 15:45:59 +02:00
|
|
|
{ok, ZlibSocket} =
|
|
|
|
ejabberd_receiver:compress(SocketData#socket_state.receiver,
|
|
|
|
Data),
|
|
|
|
SocketData#socket_state{socket = ZlibSocket,
|
|
|
|
sockmod = ejabberd_zlib}.
|
2006-10-01 03:53:37 +02:00
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
reset_stream(SocketData)
|
|
|
|
when is_pid(SocketData#socket_state.receiver) ->
|
2009-08-31 20:37:52 +02:00
|
|
|
ejabberd_receiver:reset_stream(SocketData#socket_state.receiver);
|
2012-09-11 15:45:59 +02:00
|
|
|
reset_stream(SocketData)
|
|
|
|
when is_atom(SocketData#socket_state.receiver) ->
|
|
|
|
(SocketData#socket_state.receiver):reset_stream(SocketData#socket_state.socket).
|
2006-09-05 06:26:28 +02:00
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
change_controller(#socket_state{receiver = Recv}, Pid)
|
|
|
|
when is_pid(Recv) ->
|
2010-07-13 12:04:30 +02:00
|
|
|
ejabberd_receiver:setopts(Recv, [{active, false}]),
|
|
|
|
sync_events(Pid),
|
|
|
|
ejabberd_receiver:change_controller(Recv, Pid);
|
2012-09-11 15:45:59 +02:00
|
|
|
change_controller(#socket_state{socket = Socket,
|
|
|
|
receiver = Mod},
|
|
|
|
Pid) ->
|
2010-07-13 12:04:30 +02:00
|
|
|
Mod:setopts(Socket, [{active, false}]),
|
|
|
|
sync_events(Pid),
|
|
|
|
Mod:change_controller(Socket, Pid).
|
|
|
|
|
2011-08-12 15:48:39 +02:00
|
|
|
change_socket(SocketData, Socket) ->
|
|
|
|
SocketData#socket_state{socket = Socket}.
|
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
-spec send(socket_state(), iodata()) -> ok.
|
|
|
|
|
2006-09-05 06:26:28 +02:00
|
|
|
send(SocketData, Data) ->
|
2012-09-11 15:45:59 +02:00
|
|
|
Res = if node(SocketData#socket_state.receiver) ==
|
|
|
|
node() ->
|
|
|
|
catch
|
|
|
|
(SocketData#socket_state.sockmod):send(SocketData#socket_state.socket,
|
|
|
|
Data);
|
2010-07-13 12:04:30 +02:00
|
|
|
true ->
|
2012-09-11 15:45:59 +02:00
|
|
|
catch
|
|
|
|
ejabberd_receiver:send(SocketData#socket_state.receiver,
|
|
|
|
Data)
|
|
|
|
end,
|
2010-07-13 12:04:30 +02:00
|
|
|
case Res of
|
2012-09-11 15:45:59 +02:00
|
|
|
ok -> ok;
|
|
|
|
{error, timeout} ->
|
|
|
|
?INFO_MSG("Timeout on ~p:send",
|
|
|
|
[SocketData#socket_state.sockmod]),
|
|
|
|
exit(normal);
|
|
|
|
Error ->
|
|
|
|
?DEBUG("Error in ~p:send: ~p",
|
|
|
|
[SocketData#socket_state.sockmod, Error]),
|
|
|
|
exit(normal)
|
2009-10-19 11:36:23 +02:00
|
|
|
end.
|
2006-09-05 06:26:28 +02:00
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
-spec send_xml(socket_state(), xmlel()) -> any().
|
|
|
|
|
2009-08-31 20:37:52 +02:00
|
|
|
send_xml(SocketData, Data) ->
|
2012-09-11 15:45:59 +02:00
|
|
|
catch
|
|
|
|
(SocketData#socket_state.sockmod):send_xml(SocketData#socket_state.socket,
|
|
|
|
Data).
|
2006-09-05 06:26:28 +02:00
|
|
|
|
2009-08-31 20:37:52 +02:00
|
|
|
change_shaper(SocketData, Shaper)
|
2012-09-11 15:45:59 +02:00
|
|
|
when is_pid(SocketData#socket_state.receiver) ->
|
|
|
|
ejabberd_receiver:change_shaper(SocketData#socket_state.receiver,
|
|
|
|
Shaper);
|
2009-08-31 20:37:52 +02:00
|
|
|
change_shaper(SocketData, Shaper)
|
2012-09-11 15:45:59 +02:00
|
|
|
when is_atom(SocketData#socket_state.receiver) ->
|
|
|
|
(SocketData#socket_state.receiver):change_shaper(SocketData#socket_state.socket,
|
|
|
|
Shaper).
|
2009-08-31 20:37:52 +02:00
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
monitor(SocketData)
|
|
|
|
when is_pid(SocketData#socket_state.receiver) ->
|
|
|
|
erlang:monitor(process,
|
|
|
|
SocketData#socket_state.receiver);
|
|
|
|
monitor(SocketData)
|
|
|
|
when is_atom(SocketData#socket_state.receiver) ->
|
|
|
|
(SocketData#socket_state.receiver):monitor(SocketData#socket_state.socket).
|
2007-04-12 08:08:32 +02:00
|
|
|
|
2006-09-05 06:26:28 +02:00
|
|
|
get_sockmod(SocketData) ->
|
|
|
|
SocketData#socket_state.sockmod.
|
|
|
|
|
2006-09-05 06:36:01 +02:00
|
|
|
get_peer_certificate(SocketData) ->
|
|
|
|
tls:get_peer_certificate(SocketData#socket_state.socket).
|
|
|
|
|
|
|
|
get_verify_result(SocketData) ->
|
|
|
|
tls:get_verify_result(SocketData#socket_state.socket).
|
|
|
|
|
2006-09-05 06:26:28 +02:00
|
|
|
close(SocketData) ->
|
|
|
|
ejabberd_receiver:close(SocketData#socket_state.receiver).
|
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
sockname(#socket_state{sockmod = SockMod,
|
|
|
|
socket = Socket}) ->
|
2006-10-19 06:46:24 +02:00
|
|
|
case SockMod of
|
2012-09-11 15:45:59 +02:00
|
|
|
gen_tcp -> inet:sockname(Socket);
|
|
|
|
_ -> SockMod:sockname(Socket)
|
2006-10-19 06:46:24 +02:00
|
|
|
end.
|
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
peername(#socket_state{sockmod = SockMod,
|
|
|
|
socket = Socket}) ->
|
2006-10-19 06:46:24 +02:00
|
|
|
case SockMod of
|
2012-09-11 15:45:59 +02:00
|
|
|
gen_tcp -> inet:peername(Socket);
|
|
|
|
_ -> SockMod:peername(Socket)
|
2006-10-19 06:46:24 +02:00
|
|
|
end.
|
|
|
|
|
2012-09-11 15:45:59 +02:00
|
|
|
get_conn_type(#socket_state{sockmod = SockMod, socket = Socket}) ->
|
|
|
|
case SockMod of
|
|
|
|
gen_tcp -> c2s;
|
|
|
|
tls -> c2s_tls;
|
|
|
|
ejabberd_zlib ->
|
|
|
|
case ejabberd_zlib:get_sockmod(Socket) of
|
|
|
|
gen_tcp -> c2s_compressed;
|
|
|
|
tls -> c2s_compressed_tls
|
|
|
|
end;
|
|
|
|
ejabberd_http_poll -> http_poll;
|
|
|
|
ejabberd_http_ws -> http_ws;
|
|
|
|
ejabberd_http_bind -> http_bind;
|
|
|
|
ejabberd_bosh -> http_bind;
|
|
|
|
ejabberd_http_bindjson -> http_bindjson;
|
|
|
|
ejabberd_http_wsjson -> http_wsjson
|
|
|
|
end.
|
|
|
|
|
|
|
|
-spec is_remote_receiver(socket_state()) -> boolean().
|
|
|
|
|
|
|
|
is_remote_receiver(#socket_state{receiver = Pid}) when is_pid(Pid) ->
|
|
|
|
node(Pid) /= node();
|
|
|
|
is_remote_receiver(_) ->
|
|
|
|
false.
|
|
|
|
|
2010-07-13 12:04:30 +02:00
|
|
|
sync_events(C2SPid) ->
|
|
|
|
receive
|
2012-09-11 15:45:59 +02:00
|
|
|
{'$gen_event', El} = Event
|
|
|
|
when element(1, El) == xmlel;
|
|
|
|
element(1, El) == xmlstreamstart;
|
|
|
|
element(1, El) == xmlstreamelement;
|
|
|
|
element(1, El) == xmlstreamend;
|
|
|
|
element(1, El) == xmlstreamerror ->
|
|
|
|
C2SPid ! Event, sync_events(C2SPid);
|
|
|
|
closed -> C2SPid ! closed, sync_events(C2SPid)
|
|
|
|
after 0 -> ok
|
2010-07-13 12:04:30 +02:00
|
|
|
end.
|