aboutsummaryrefslogtreecommitdiff
path: root/src/ejabberd_socket.erl
diff options
context:
space:
mode:
Diffstat (limited to 'src/ejabberd_socket.erl')
-rw-r--r--src/ejabberd_socket.erl240
1 files changed, 0 insertions, 240 deletions
diff --git a/src/ejabberd_socket.erl b/src/ejabberd_socket.erl
deleted file mode 100644
index 887b4a0f3..000000000
--- a/src/ejabberd_socket.erl
+++ /dev/null
@@ -1,240 +0,0 @@
-%%%----------------------------------------------------------------------
-%%% File : ejabberd_socket.erl
-%%% Author : Alexey Shchepin <alexey@process-one.net>
-%%% Purpose : Socket with zlib and TLS support library
-%%% Created : 23 Aug 2006 by Alexey Shchepin <alexey@process-one.net>
-%%%
-%%%
-%%% ejabberd, Copyright (C) 2002-2016 ProcessOne
-%%%
-%%% 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.
-%%%
-%%% 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.,
-%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
-%%%
-%%%----------------------------------------------------------------------
-
--module(ejabberd_socket).
-
--author('alexey@process-one.net').
-
-%% API
--export([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_verify_result/1,
- close/1,
- sockname/1, peername/1]).
-
--include("ejabberd.hrl").
--include("logger.hrl").
--include("jlib.hrl").
-
--type sockmod() :: ejabberd_http_bind |
- ejabberd_http_ws |
- gen_tcp | fast_tls | ezlib.
--type receiver() :: pid () | atom().
--type socket() :: pid() | inet:socket() |
- fast_tls:tls_socket() |
- ezlib:zlib_socket() |
- ejabberd_http_bind:bind_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]).
-
-
-%%====================================================================
-%% API
-%%====================================================================
--spec start(atom(), sockmod(), socket(), [{atom(), any()}]) -> any().
-
-start(Module, SockMod, Socket, Opts) ->
- case Module:socket_type() of
- 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)
- end,
- 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
- end.
-
-connect(Addr, Port, Opts) ->
- connect(Addr, Port, Opts, infinity).
-
-connect(Addr, Port, Opts, Timeout) ->
- case gen_tcp:connect(Addr, Port, Opts, Timeout) of
- {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
- end.
-
-starttls(SocketData, TLSOpts) ->
- {ok, TLSSocket} = fast_tls:tcp_to_tls(SocketData#socket_state.socket, TLSOpts),
- ejabberd_receiver:starttls(SocketData#socket_state.receiver, TLSSocket),
- SocketData#socket_state{socket = TLSSocket, sockmod = fast_tls}.
-
-starttls(SocketData, TLSOpts, Data) ->
- {ok, TLSSocket} = fast_tls:tcp_to_tls(SocketData#socket_state.socket, TLSOpts),
- ejabberd_receiver:starttls(SocketData#socket_state.receiver, TLSSocket),
- send(SocketData, Data),
- SocketData#socket_state{socket = TLSSocket, sockmod = fast_tls}.
-
-compress(SocketData) -> compress(SocketData, undefined).
-
-compress(SocketData, Data) ->
- {ok, ZlibSocket} =
- ejabberd_receiver:compress(SocketData#socket_state.receiver,
- Data),
- SocketData#socket_state{socket = ZlibSocket,
- sockmod = ezlib}.
-
-reset_stream(SocketData)
- when is_pid(SocketData#socket_state.receiver) ->
- ejabberd_receiver:reset_stream(SocketData#socket_state.receiver);
-reset_stream(SocketData)
- when is_atom(SocketData#socket_state.receiver) ->
- (SocketData#socket_state.receiver):reset_stream(SocketData#socket_state.socket).
-
--spec send(socket_state(), iodata()) -> ok.
-
-send(SocketData, Data) ->
- case catch (SocketData#socket_state.sockmod):send(
- SocketData#socket_state.socket, Data) of
- 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)
- end.
-
-%% Can only be called when in c2s StateData#state.xml_socket is true
-%% This function is used for HTTP bind
-%% sockmod=ejabberd_http_ws|ejabberd_http_bind or any custom module
--spec send_xml(socket_state(), xmlel()) -> any().
-
-send_xml(SocketData, Data) ->
- catch
- (SocketData#socket_state.sockmod):send_xml(SocketData#socket_state.socket,
- Data).
-
-change_shaper(SocketData, Shaper)
- when is_pid(SocketData#socket_state.receiver) ->
- ejabberd_receiver:change_shaper(SocketData#socket_state.receiver,
- Shaper);
-change_shaper(SocketData, Shaper)
- when is_atom(SocketData#socket_state.receiver) ->
- (SocketData#socket_state.receiver):change_shaper(SocketData#socket_state.socket,
- Shaper).
-
-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).
-
-get_sockmod(SocketData) ->
- SocketData#socket_state.sockmod.
-
-get_peer_certificate(SocketData) ->
- fast_tls:get_peer_certificate(SocketData#socket_state.socket).
-
-get_verify_result(SocketData) ->
- fast_tls:get_verify_result(SocketData#socket_state.socket).
-
-close(SocketData) ->
- ejabberd_receiver:close(SocketData#socket_state.receiver).
-
-sockname(#socket_state{sockmod = SockMod,
- socket = Socket}) ->
- case SockMod of
- gen_tcp -> inet:sockname(Socket);
- _ -> SockMod:sockname(Socket)
- end.
-
-peername(#socket_state{sockmod = SockMod,
- socket = Socket}) ->
- case SockMod of
- gen_tcp -> inet:peername(Socket);
- _ -> SockMod:peername(Socket)
- end.
-
generated by cgit v1.2.3 (git 2.25.1) at 2025-09-08 18:24:24 +0000