aboutsummaryrefslogtreecommitdiff
path: root/src/ejabberd_local.erl
diff options
context:
space:
mode:
Diffstat (limited to 'src/ejabberd_local.erl')
-rw-r--r--src/ejabberd_local.erl353
1 files changed, 93 insertions, 260 deletions
diff --git a/src/ejabberd_local.erl b/src/ejabberd_local.erl
index 2ba943693..7c8b0489f 100644
--- a/src/ejabberd_local.erl
+++ b/src/ejabberd_local.erl
@@ -5,7 +5,7 @@
%%% Created : 30 Nov 2002 by Alexey Shchepin <alexey@process-one.net>
%%%
%%%
-%%% ejabberd, Copyright (C) 2002-2016 ProcessOne
+%%% ejabberd, Copyright (C) 2002-2019 ProcessOne
%%%
%%% This program is free software; you can redistribute it and/or
%%% modify it under the terms of the GNU General Public License as
@@ -30,36 +30,29 @@
-behaviour(gen_server).
%% API
--export([start_link/0]).
+-export([start/0, start_link/0]).
--export([route/3, route_iq/4, route_iq/5, process_iq/3,
- process_iq_reply/3, register_iq_handler/4,
- register_iq_handler/5, register_iq_response_handler/4,
- register_iq_response_handler/5, unregister_iq_handler/2,
- unregister_iq_response_handler/2, refresh_iq_handlers/0,
- bounce_resource_packet/3]).
+-export([route/1,
+ get_features/1,
+ bounce_resource_packet/1,
+ host_up/1, host_down/1]).
%% gen_server callbacks
-export([init/1, handle_call/3, handle_cast/2,
handle_info/2, terminate/2, code_change/3]).
--include("ejabberd.hrl").
--include("logger.hrl").
+%% deprecated functions: use ejabberd_router:route_iq/3,4
+-export([route_iq/2, route_iq/3]).
+-deprecated([{route_iq, 2}, {route_iq, 3}]).
--include("jlib.hrl").
+-include("logger.hrl").
+-include_lib("stdlib/include/ms_transform.hrl").
+-include("xmpp.hrl").
+-include("ejabberd_stacktrace.hrl").
+-include("translate.hrl").
-record(state, {}).
--record(iq_response, {id = <<"">> :: binary(),
- module :: atom(),
- function :: atom() | fun(),
- timer = make_ref() :: reference()}).
-
--define(IQTABLE, local_iqtable).
-
-%% This value is used in SIP and Megaco for a transaction lifetime.
--define(IQ_TIMEOUT, 32000).
-
%%====================================================================
%% API
%%====================================================================
@@ -67,192 +60,87 @@
%% Function: start_link() -> {ok,Pid} | ignore | {error,Error}
%% Description: Starts the server
%%--------------------------------------------------------------------
+start() ->
+ ChildSpec = {?MODULE, {?MODULE, start_link, []},
+ transient, 1000, worker, [?MODULE]},
+ supervisor:start_child(ejabberd_sup, ChildSpec).
+
start_link() ->
gen_server:start_link({local, ?MODULE}, ?MODULE, [],
[]).
-process_iq(From, To, Packet) ->
- IQ = jlib:iq_query_info(Packet),
- case IQ of
- #iq{xmlns = XMLNS, lang = Lang} ->
- Host = To#jid.lserver,
- case ets:lookup(?IQTABLE, {XMLNS, Host}) of
- [{_, Module, Function}] ->
- ResIQ = Module:Function(From, To, IQ),
- if ResIQ /= ignore ->
- ejabberd_router:route(To, From, jlib:iq_to_xml(ResIQ));
- true -> ok
- end;
- [{_, Module, Function, Opts}] ->
- gen_iq_handler:handle(Host, Module, Function, Opts,
- From, To, IQ);
- [] ->
- Txt = <<"No module is handling this query">>,
- Err = jlib:make_error_reply(
- Packet,
- ?ERRT_FEATURE_NOT_IMPLEMENTED(Lang, Txt)),
- ejabberd_router:route(To, From, Err)
- end;
- reply ->
- IQReply = jlib:iq_query_or_response_info(Packet),
- process_iq_reply(From, To, IQReply);
- _ ->
- Err = jlib:make_error_reply(Packet, ?ERR_BAD_REQUEST),
- ejabberd_router:route(To, From, Err),
- ok
- end.
-
-process_iq_reply(From, To, #iq{id = ID} = IQ) ->
- case get_iq_callback(ID) of
- {ok, undefined, Function} -> Function(IQ), ok;
- {ok, Module, Function} ->
- Module:Function(From, To, IQ), ok;
- _ -> nothing
- end.
-
-route(From, To, Packet) ->
- case catch do_route(From, To, Packet) of
- {'EXIT', Reason} ->
- ?ERROR_MSG("~p~nwhen processing: ~p",
- [Reason, {From, To, Packet}]);
- _ -> ok
+-spec route(stanza()) -> ok.
+route(Packet) ->
+ ?DEBUG("Local route:~n~ts", [xmpp:pp(Packet)]),
+ Type = xmpp:get_type(Packet),
+ To = xmpp:get_to(Packet),
+ if To#jid.luser /= <<"">> ->
+ ejabberd_sm:route(Packet);
+ is_record(Packet, iq), To#jid.lresource == <<"">> ->
+ gen_iq_handler:handle(?MODULE, Packet);
+ Type == result; Type == error ->
+ ok;
+ true ->
+ ejabberd_hooks:run(local_send_to_resource_hook,
+ To#jid.lserver, [Packet])
end.
-route_iq(From, To, IQ, F) ->
- route_iq(From, To, IQ, F, undefined).
-
-route_iq(From, To, #iq{type = Type} = IQ, F, Timeout)
- when is_function(F) ->
- Packet = if Type == set; Type == get ->
- ID = randoms:get_string(),
- Host = From#jid.lserver,
- register_iq_response_handler(Host, ID, undefined, F, Timeout),
- jlib:iq_to_xml(IQ#iq{id = ID});
- true ->
- jlib:iq_to_xml(IQ)
- end,
- ejabberd_router:route(From, To, Packet).
-
-register_iq_response_handler(Host, ID, Module,
- Function) ->
- register_iq_response_handler(Host, ID, Module, Function,
- undefined).
-
-register_iq_response_handler(_Host, ID, Module,
- Function, Timeout0) ->
- Timeout = case Timeout0 of
- undefined -> ?IQ_TIMEOUT;
- N when is_integer(N), N > 0 -> N
- end,
- TRef = erlang:start_timer(Timeout, ejabberd_local, ID),
- mnesia:dirty_write(#iq_response{id = ID,
- module = Module,
- function = Function,
- timer = TRef}).
-
-register_iq_handler(Host, XMLNS, Module, Fun) ->
- ejabberd_local !
- {register_iq_handler, Host, XMLNS, Module, Fun}.
-
-register_iq_handler(Host, XMLNS, Module, Fun, Opts) ->
- ejabberd_local !
- {register_iq_handler, Host, XMLNS, Module, Fun, Opts}.
-
-unregister_iq_response_handler(_Host, ID) ->
- catch get_iq_callback(ID), ok.
-
-unregister_iq_handler(Host, XMLNS) ->
- ejabberd_local ! {unregister_iq_handler, Host, XMLNS}.
-
-refresh_iq_handlers() ->
- ejabberd_local ! refresh_iq_handlers.
-
-bounce_resource_packet(From, To, Packet) ->
- Lang = fxml:get_tag_attr_s(<<"xml:lang">>, Packet),
- Txt = <<"No available resource found">>,
- Err = jlib:make_error_reply(Packet,
- ?ERRT_ITEM_NOT_FOUND(Lang, Txt)),
- ejabberd_router:route(To, From, Err),
+-spec route_iq(iq(), function()) -> ok.
+route_iq(IQ, Fun) ->
+ route_iq(IQ, Fun, undefined).
+
+-spec route_iq(iq(), function(), undefined | non_neg_integer()) -> ok.
+route_iq(IQ, Fun, Timeout) ->
+ ejabberd_router:route_iq(IQ, Fun, undefined, Timeout).
+
+-spec bounce_resource_packet(stanza()) -> ok | stop.
+bounce_resource_packet(#presence{to = #jid{lresource = <<"">>}}) ->
+ ok;
+bounce_resource_packet(#message{to = #jid{lresource = <<"">>}, type = headline}) ->
+ ok;
+bounce_resource_packet(Packet) ->
+ Lang = xmpp:get_lang(Packet),
+ Txt = ?T("No available resource found"),
+ Err = xmpp:err_item_not_found(Txt, Lang),
+ ejabberd_router:route_error(Packet, Err),
stop.
+-spec get_features(binary()) -> [binary()].
+get_features(Host) ->
+ gen_iq_handler:get_features(?MODULE, Host).
+
%%====================================================================
%% gen_server callbacks
%%====================================================================
init([]) ->
- lists:foreach(fun (Host) ->
- ejabberd_router:register_route(Host,
- Host,
- {apply, ?MODULE,
- route}),
- ejabberd_hooks:add(local_send_to_resource_hook, Host,
- ?MODULE, bounce_resource_packet,
- 100)
- end,
- ?MYHOSTS),
- catch ets:new(?IQTABLE, [named_table, public]),
+ process_flag(trap_exit, true),
+ lists:foreach(fun host_up/1, ejabberd_option:hosts()),
+ ejabberd_hooks:add(host_up, ?MODULE, host_up, 10),
+ ejabberd_hooks:add(host_down, ?MODULE, host_down, 100),
+ gen_iq_handler:start(?MODULE),
update_table(),
- mnesia:create_table(iq_response,
- [{ram_copies, [node()]},
- {attributes, record_info(fields, iq_response)}]),
- mnesia:add_table_copy(iq_response, node(), ram_copies),
{ok, #state{}}.
-handle_call(_Request, _From, State) ->
- Reply = ok, {reply, Reply, State}.
+handle_call(Request, From, State) ->
+ ?WARNING_MSG("Unexpected call from ~p: ~p", [From, Request]),
+ {noreply, State}.
-handle_cast(_Msg, State) -> {noreply, State}.
+handle_cast(Msg, State) ->
+ ?WARNING_MSG("Unexpected cast: ~p", [Msg]),
+ {noreply, State}.
-handle_info({route, From, To, Packet}, State) ->
- case catch do_route(From, To, Packet) of
- {'EXIT', Reason} ->
- ?ERROR_MSG("~p~nwhen processing: ~p",
- [Reason, {From, To, Packet}]);
- _ -> ok
- end,
- {noreply, State};
-handle_info({register_iq_handler, Host, XMLNS, Module,
- Function},
- State) ->
- ets:insert(?IQTABLE, {{XMLNS, Host}, Module, Function}),
- catch mod_disco:register_feature(Host, XMLNS),
- {noreply, State};
-handle_info({register_iq_handler, Host, XMLNS, Module,
- Function, Opts},
- State) ->
- ets:insert(?IQTABLE,
- {{XMLNS, Host}, Module, Function, Opts}),
- catch mod_disco:register_feature(Host, XMLNS),
- {noreply, State};
-handle_info({unregister_iq_handler, Host, XMLNS},
- State) ->
- case ets:lookup(?IQTABLE, {XMLNS, Host}) of
- [{_, Module, Function, Opts}] ->
- gen_iq_handler:stop_iq_handler(Module, Function, Opts);
- _ -> ok
- end,
- ets:delete(?IQTABLE, {XMLNS, Host}),
- catch mod_disco:unregister_feature(Host, XMLNS),
+handle_info({route, Packet}, State) ->
+ route(Packet),
{noreply, State};
-handle_info(refresh_iq_handlers, State) ->
- lists:foreach(fun (T) ->
- case T of
- {{XMLNS, Host}, _Module, _Function, _Opts} ->
- catch mod_disco:register_feature(Host, XMLNS);
- {{XMLNS, Host}, _Module, _Function} ->
- catch mod_disco:register_feature(Host, XMLNS);
- _ -> ok
- end
- end,
- ets:tab2list(?IQTABLE)),
- {noreply, State};
-handle_info({timeout, _TRef, ID}, State) ->
- process_iq_timeout(ID),
- {noreply, State};
-handle_info(_Info, State) ->
+handle_info(Info, State) ->
+ ?WARNING_MSG("Unexpected info: ~p", [Info]),
{noreply, State}.
terminate(_Reason, _State) ->
+ lists:foreach(fun host_down/1, ejabberd_option:hosts()),
+ ejabberd_hooks:delete(host_up, ?MODULE, host_up, 10),
+ ejabberd_hooks:delete(host_down, ?MODULE, host_down, 100),
ok.
code_change(_OldVsn, State, _Extra) ->
@@ -261,80 +149,25 @@ code_change(_OldVsn, State, _Extra) ->
%%--------------------------------------------------------------------
%%% Internal functions
%%--------------------------------------------------------------------
-do_route(From, To, Packet) ->
- ?DEBUG("local route~n\tfrom ~p~n\tto ~p~n\tpacket "
- "~P~n",
- [From, To, Packet, 8]),
- if To#jid.luser /= <<"">> ->
- ejabberd_sm:route(From, To, Packet);
- To#jid.lresource == <<"">> ->
- #xmlel{name = Name} = Packet,
- case Name of
- <<"iq">> -> process_iq(From, To, Packet);
- <<"message">> ->
- #xmlel{attrs = Attrs} = Packet,
- case fxml:get_attr_s(<<"type">>, Attrs) of
- <<"headline">> -> ok;
- <<"error">> -> ok;
- _ ->
- Err = jlib:make_error_reply(Packet,
- ?ERR_SERVICE_UNAVAILABLE),
- ejabberd_router:route(To, From, Err)
- end;
- <<"presence">> -> ok;
- _ -> ok
- end;
- true ->
- #xmlel{attrs = Attrs} = Packet,
- case fxml:get_attr_s(<<"type">>, Attrs) of
- <<"error">> -> ok;
- <<"result">> -> ok;
- _ ->
- ejabberd_hooks:run(local_send_to_resource_hook,
- To#jid.lserver, [From, To, Packet])
- end
- end.
-
+-spec update_table() -> ok.
update_table() ->
- case catch mnesia:table_info(iq_response, attributes) of
- [id, module, function] ->
- mnesia:delete_table(iq_response);
- [id, module, function, timer] ->
- ok;
- {'EXIT', _} ->
- ok
- end.
-
-get_iq_callback(ID) ->
- case mnesia:dirty_read(iq_response, ID) of
- [#iq_response{module = Module, timer = TRef,
- function = Function}] ->
- cancel_timer(TRef),
- mnesia:dirty_delete(iq_response, ID),
- {ok, Module, Function};
- _ ->
- error
- end.
-
-process_iq_timeout(ID) ->
- spawn(fun process_iq_timeout/0) ! ID.
-
-process_iq_timeout() ->
- receive
- ID ->
- case get_iq_callback(ID) of
- {ok, undefined, Function} ->
- Function(timeout);
- _ ->
- ok
- end
- after 5000 ->
- ok
- end.
+ catch mnesia:delete_table(iq_response),
+ ok.
-cancel_timer(TRef) ->
- case erlang:cancel_timer(TRef) of
- false ->
- receive {timeout, TRef, _} -> ok after 0 -> ok end;
- _ -> ok
- end.
+host_up(Host) ->
+ Owner = case whereis(?MODULE) of
+ undefined -> self();
+ Pid -> Pid
+ end,
+ ejabberd_router:register_route(Host, Host, {apply, ?MODULE, route}, Owner),
+ ejabberd_hooks:add(local_send_to_resource_hook, Host,
+ ?MODULE, bounce_resource_packet, 100).
+
+host_down(Host) ->
+ Owner = case whereis(?MODULE) of
+ undefined -> self();
+ Pid -> Pid
+ end,
+ ejabberd_router:unregister_route(Host, Owner),
+ ejabberd_hooks:delete(local_send_to_resource_hook, Host,
+ ?MODULE, bounce_resource_packet, 100).
generated by cgit v1.2.3 (git 2.25.1) at 2025-09-09 03:00:39 +0000