diff --git a/sql/lite.sql b/sql/lite.sql index ddfb91ff0..1d057407f 100644 --- a/sql/lite.sql +++ b/sql/lite.sql @@ -369,3 +369,15 @@ CREATE TABLE carboncopy ( CREATE UNIQUE INDEX i_carboncopy_ur ON carboncopy (username, resource); CREATE INDEX i_carboncopy_user ON carboncopy (username); + +CREATE TABLE proxy65 ( + sid text NOT NULL, + pid_t text NOT NULL, + pid_i text NOT NULL, + node_t text NOT NULL, + node_i text NOT NULL, + jid_i text NOT NULL +); + +CREATE UNIQUE INDEX i_proxy65_sid ON proxy65 (sid); +CREATE INDEX i_proxy65_jid ON proxy65 (jid_i); diff --git a/sql/mysql.sql b/sql/mysql.sql index e09c39be5..c591cb761 100644 --- a/sql/mysql.sql +++ b/sql/mysql.sql @@ -385,3 +385,15 @@ CREATE TABLE carboncopy ( CREATE UNIQUE INDEX i_carboncopy_ur ON carboncopy (username(75), resource(75)); CREATE INDEX i_carboncopy_user ON carboncopy (username(75)); + +CREATE TABLE proxy65 ( + sid text NOT NULL, + pid_t text NOT NULL, + pid_i text NOT NULL, + node_t text NOT NULL, + node_i text NOT NULL, + jid_i text NOT NULL +) ENGINE=InnoDB CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci; + +CREATE UNIQUE INDEX i_proxy65_sid ON proxy65 (sid(191)); +CREATE INDEX i_proxy65_jid ON proxy65 (jid_i(191)); diff --git a/sql/pg.sql b/sql/pg.sql index 796391c74..539c1263a 100644 --- a/sql/pg.sql +++ b/sql/pg.sql @@ -389,3 +389,15 @@ CREATE TABLE carboncopy ( CREATE UNIQUE INDEX i_carboncopy_ur ON carboncopy USING btree (username, resource); CREATE INDEX i_carboncopy_user ON carboncopy USING btree (username); + +CREATE TABLE proxy65 ( + sid text NOT NULL, + pid_t text NOT NULL, + pid_i text NOT NULL, + node_t text NOT NULL, + node_i text NOT NULL, + jid_i text NOT NULL +); + +CREATE UNIQUE INDEX i_proxy65_sid ON proxy65 USING btree (sid); +CREATE INDEX i_proxy65_jid ON proxy65 USING btree (jid_i); diff --git a/src/mod_proxy65_sql.erl b/src/mod_proxy65_sql.erl new file mode 100644 index 000000000..5007bd090 --- /dev/null +++ b/src/mod_proxy65_sql.erl @@ -0,0 +1,142 @@ +%%%------------------------------------------------------------------- +%%% @author Evgeny Khramtsov +%%% Created : 30 Mar 2017 by Evgeny Khramtsov +%%% +%%% +%%% ejabberd, Copyright (C) 2002-2017 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(mod_proxy65_sql). +-behaviour(mod_proxy65). + +-compile([{parse_transform, ejabberd_sql_pt}]). + +%% API +-export([init/0, register_stream/2, unregister_stream/1, activate_stream/4]). + +-include("ejabberd.hrl"). +-include("logger.hrl"). +-include("ejabberd_sql_pt.hrl"). + +%%%=================================================================== +%%% API +%%%=================================================================== +init() -> + NodeS = erlang:atom_to_binary(node(), latin1), + ?INFO_MSG("Cleaning SQL 'proxy65' table...", []), + case ejabberd_sql:sql_query( + ?MYNAME, + ?SQL("delete from proxy65 where " + "node_i=%(NodeS)s or node_t=%(NodeS)s")) of + {updated, _} -> + ok; + Err -> + ?ERROR_MSG("failed to clean 'proxy65' table: ~p", [Err]), + Err + end. + +register_stream(SID, Pid) -> + PidS = aux:encode_pid(Pid), + NodeS = erlang:atom_to_binary(node(Pid), latin1), + F = fun() -> + case ejabberd_sql:sql_query_t( + ?SQL("update proxy65 set pid_i=%(PidS)s, " + "node_i=%(NodeS)s where sid=%(SID)s")) of + {updated, 1} -> + ok; + _ -> + ejabberd_sql:sql_query_t( + ?SQL("insert into proxy65" + "(sid, pid_t, node_t, pid_i, node_i, jid_i) " + "values (%(SID)s, %(PidS)s, %(NodeS)s, '', '', '')")) + end + end, + case ejabberd_sql:sql_transaction(?MYNAME, F) of + {atomic, _} -> + ok; + {aborted, Reason} -> + ?ERROR_MSG("failed to register stream: ~p", [Reason]), + {error, Reason} + end. + +unregister_stream(SID) -> + F = fun() -> + ejabberd_sql:sql_query_t( + ?SQL("delete from proxy65 where sid=%(SID)s")) + end, + case ejabberd_sql:sql_transaction(?MYNAME, F) of + {atomic, _} -> + ok; + {aborted, Reason} -> + ?ERROR_MSG("failed to unregister stream: ~p", [Reason]), + {error, Reason} + end. + +activate_stream(SID, IJID, MaxConnections, _Node) -> + F = fun() -> + case ejabberd_sql:sql_query_t( + ?SQL("select @(pid_t)s, @(node_t)s, @(pid_i)s, " + "@(node_i)s, @(jid_i)s from proxy65 where " + "sid=%(SID)s")) of + {selected, [{TPidS, TNodeS, IPidS, INodeS, <<"">>}]} + when IPidS /= <<"">> -> + try {aux:decode_pid(TPidS, TNodeS), + aux:decode_pid(IPidS, INodeS)} of + {TPid, IPid} -> + case ejabberd_sql:sql_query_t( + ?SQL("update proxy65 set jid_i=%(IJID)s " + "where sid=%(SID)s")) of + {updated, 1} when is_integer(MaxConnections) -> + case ejabberd_sql:sql_query_t( + ?SQL("select @(count(*))d from proxy65 " + "where jid_i=%(IJID)s")) of + {selected, [{Num}]} when Num > MaxConnections -> + exit({limit, IPid, TPid}); + {selected, _} -> + {ok, IPid, TPid}; + Err -> + exit(Err) + end; + {updated, _} -> + {ok, IPid, TPid}; + Err -> + exit(Err) + end + catch _:{bad_node, _} -> + {error, notfound} + end; + {selected, [{_, _, _, _, JID}]} when JID /= <<"">> -> + {error, conflict}; + {selected, _} -> + {error, notfound}; + Err -> + exit(Err) + end + end, + case ejabberd_sql:sql_transaction(?MYNAME, F) of + {atomic, Result} -> + Result; + {aborted, {limit, _, _} = Limit} -> + {error, Limit}; + {aborted, Reason} -> + ?ERROR_MSG("failed to activate bytestream: ~p", [Reason]), + {error, Reason} + end. + +%%%=================================================================== +%%% Internal functions +%%%===================================================================