25
1
mirror of https://github.com/processone/ejabberd.git synced 2024-11-22 16:20:52 +01:00

Support for static_modules in mod_muc and mod_muc_log

This commit is contained in:
Badlop 2011-07-19 21:19:15 +02:00
parent 82551036c2
commit 19ab0174e4
3 changed files with 48 additions and 27 deletions

View File

@ -2629,8 +2629,8 @@ The following table lists all modules included in \ejabberd{}.
\hline \ahrefloc{modhttpbind}{\modhttpbind{}} & XMPP over Bosh service (HTTP Binding) & & \\ \hline \ahrefloc{modhttpbind}{\modhttpbind{}} & XMPP over Bosh service (HTTP Binding) & & \\
\hline \ahrefloc{modhttpfileserver}{\modhttpfileserver{}} & Small HTTP file server & & \\ \hline \ahrefloc{modhttpfileserver}{\modhttpfileserver{}} & Small HTTP file server & & \\
\hline \ahrefloc{modlast}{\modlast{}} & Last Activity (\xepref{0012}) & & \\ \hline \ahrefloc{modlast}{\modlast{}} & Last Activity (\xepref{0012}) & & \\
\hline \ahrefloc{modmuc}{\modmuc{}} & Multi-User Chat (\xepref{0045}) & - & \\ \hline \ahrefloc{modmuc}{\modmuc{}} & Multi-User Chat (\xepref{0045}) & & \\
\hline \ahrefloc{modmuclog}{\modmuclog{}} & Multi-User Chat room logging & - & \modmuc{} \\ \hline \ahrefloc{modmuclog}{\modmuclog{}} & Multi-User Chat room logging & & \modmuc{} \\
\hline \ahrefloc{modmulticast}{\modmulticast{}} & Multicast Service (\xepref{0033}) & - & \\ \hline \ahrefloc{modmulticast}{\modmulticast{}} & Multicast Service (\xepref{0033}) & - & \\
\hline \ahrefloc{modoffline}{\modoffline{}} & Offline message storage (\xepref{0160}) & & \\ \hline \ahrefloc{modoffline}{\modoffline{}} & Offline message storage (\xepref{0160}) & & \\
\hline \ahrefloc{modping}{\modping{}} & XMPP Ping and keepalives (\xepref{0199}) & & \\ \hline \ahrefloc{modping}{\modping{}} & XMPP Ping and keepalives (\xepref{0199}) & & \\

View File

@ -600,13 +600,16 @@ get_table(Host, Tab) ->
case {mnesia:dirty_read(table, {Host, Tab}), Host} of case {mnesia:dirty_read(table, {Host, Tab}), Host} of
{[T], _} -> {[T], _} ->
T; T;
{_, Host} when is_binary(Host) ->
Prefix = lists:nth(1, string:tokens(binary_to_list(Host), ".")),
get_table({global, Prefix}, Tab);
{_, {global, _Prefix}} ->
get_table(global, Tab);
{_, global} -> {_, global} ->
catch throw(error123), catch throw(error123),
Stacktrace = erlang:get_stacktrace(), Stacktrace = erlang:get_stacktrace(),
error_logger:error_msg("gen_storage: Table ~p not found on ~p~nStacktrace: ~p", [Tab, Host, Stacktrace]), error_logger:error_msg("gen_storage: Table ~p not found on ~p~nStacktrace: ~p", [Tab, Host, Stacktrace]),
exit(table_not_found); exit(table_not_found)
{_, _} ->
get_table(global, Tab)
end. end.
backend_apply(F, Host, Tab) -> backend_apply(F, Host, Tab) ->

View File

@ -128,12 +128,14 @@ start_link(Host, Opts) ->
Proc = gen_mod:get_module_proc(Host, ?PROCNAME), Proc = gen_mod:get_module_proc(Host, ?PROCNAME),
gen_server:start_link({local, Proc}, ?MODULE, [Host, Opts], []). gen_server:start_link({local, Proc}, ?MODULE, [Host, Opts], []).
start(Host, Opts) -> start(Host, Opts) when is_list(Host) ->
start_supervisor(Host), start(list_to_binary(Host), Opts);
Proc = gen_mod:get_module_proc(Host, ?PROCNAME), start(HostB, Opts) ->
start_supervisor(HostB),
Proc = gen_mod:get_module_proc(HostB, ?PROCNAME),
ChildSpec = ChildSpec =
{Proc, {Proc,
{?MODULE, start_link, [Host, Opts]}, {?MODULE, start_link, [HostB, Opts]},
temporary, temporary,
1000, 1000,
worker, worker,
@ -154,7 +156,7 @@ stop(Host) ->
%% So the message sending must be catched %% So the message sending must be catched
room_destroyed(Host, Room, Pid, ServerHost) when is_binary(Host), room_destroyed(Host, Room, Pid, ServerHost) when is_binary(Host),
is_binary(Room) -> is_binary(Room) ->
catch gen_mod:get_module_proc(ServerHost, ?PROCNAME) ! catch gen_mod:get_module_proc_existing(ServerHost, ?PROCNAME) !
{room_destroyed, {Room, Host}, Pid}, {room_destroyed, {Room, Host}, Pid},
ok. ok.
@ -162,7 +164,7 @@ room_destroyed(Host, Room, Pid, ServerHost) when is_binary(Host),
%% If Opts = default, the default room options are used. %% If Opts = default, the default room options are used.
%% Else use the passed options as defined in mod_muc_room. %% Else use the passed options as defined in mod_muc_room.
create_room(Host, Name, From, Nick, Opts) -> create_room(Host, Name, From, Nick, Opts) ->
Proc = gen_mod:get_module_proc(Host, ?PROCNAME), Proc = gen_mod:get_module_proc_existing(Host, ?PROCNAME),
RoomHost = gen_mod:get_module_opt_host(Host, ?MODULE, "conference.@HOST@"), RoomHost = gen_mod:get_module_opt_host(Host, ?MODULE, "conference.@HOST@"),
Node = ejabberd_cluster:get_node({Name, RoomHost}), Node = ejabberd_cluster:get_node({Name, RoomHost}),
gen_server:call({Proc, Node}, {create, Name, From, Nick, Opts}). gen_server:call({Proc, Node}, {create, Name, From, Nick, Opts}).
@ -255,7 +257,7 @@ forget_room(Host, Name) when is_binary(Host), is_binary(Name) ->
end, end,
gen_storage:transaction(Host, muc_room_opt, F). gen_storage:transaction(Host, muc_room_opt, F).
process_iq_disco_items(Host, From, To, #iq{lang = Lang} = IQ) when is_binary(Host) -> process_iq_disco_items(Host, From, To, #iq{lang = Lang} = IQ) ->
Rsm = jlib:rsm_decode(IQ), Rsm = jlib:rsm_decode(IQ),
Res = exmpp_iq:result(IQ, #xmlel{ns = ?NS_DISCO_ITEMS, Res = exmpp_iq:result(IQ, #xmlel{ns = ?NS_DISCO_ITEMS,
name = 'query', name = 'query',
@ -308,8 +310,8 @@ migrate(After) ->
%% Description: Initiates the server %% Description: Initiates the server
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
init([Host, Opts]) -> init([Host, Opts]) ->
MyHost_L = gen_mod:expand_host_name(Host, Opts, "conference"), MyHostStr = gen_mod:expand_host_name(Host, Opts, "conference"),
MyHost = list_to_binary(MyHost_L), MyHost = l2b(MyHostStr),
Backend = gen_mod:get_opt(backend, Opts, mnesia), Backend = gen_mod:get_opt(backend, Opts, mnesia),
gen_storage:create_table(Backend, MyHost, muc_room_opt, gen_storage:create_table(Backend, MyHost, muc_room_opt,
[{disc_copies, [node()]}, [{disc_copies, [node()]},
@ -343,7 +345,7 @@ init([Host, Opts]) ->
gen_storage:add_table_copy(MyHost, muc_online_room, node(), ram_copies), gen_storage:add_table_copy(MyHost, muc_online_room, node(), ram_copies),
catch ets:new(muc_online_users, [bag, named_table, public, {keypos, 2}]), catch ets:new(muc_online_users, [bag, named_table, public, {keypos, 2}]),
gen_storage:add_table_index(MyHost, muc_registered, nick), gen_storage:add_table_index(MyHost, muc_registered, nick),
update_tables(MyHost_L, Backend), update_tables(MyHost, Backend),
Access = gen_mod:get_opt(access, Opts, all), Access = gen_mod:get_opt(access, Opts, all),
AccessCreate = gen_mod:get_opt(access_create, Opts, all), AccessCreate = gen_mod:get_opt(access_create, Opts, all),
AccessAdmin = gen_mod:get_opt(access_admin, Opts, none), AccessAdmin = gen_mod:get_opt(access_admin, Opts, none),
@ -351,7 +353,7 @@ init([Host, Opts]) ->
HistorySize = gen_mod:get_opt(history_size, Opts, 20), HistorySize = gen_mod:get_opt(history_size, Opts, 20),
DefRoomOpts = gen_mod:get_opt(default_room_options, Opts, []), DefRoomOpts = gen_mod:get_opt(default_room_options, Opts, []),
RoomShaper = gen_mod:get_opt(room_shaper, Opts, none), RoomShaper = gen_mod:get_opt(room_shaper, Opts, none),
ejabberd_router:register_route(MyHost_L), ejabberd_router:register_route(MyHostStr),
ejabberd_hooks:add(node_hash_update, ?MODULE, migrate, 100), ejabberd_hooks:add(node_hash_update, ?MODULE, migrate, 100),
load_permanent_rooms(MyHost, Host, load_permanent_rooms(MyHost, Host,
{Access, AccessCreate, AccessAdmin, AccessPersistent}, {Access, AccessCreate, AccessAdmin, AccessPersistent},
@ -413,12 +415,13 @@ handle_cast(_Msg, State) ->
%% Description: Handling all non call/cast messages %% Description: Handling all non call/cast messages
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
handle_info({route, From, To, Packet}, handle_info({route, From, To, Packet},
#state{host = Host, #state{host = HostOrGlobal,
server_host = ServerHost, server_host = ServerHost,
access = Access, access = Access,
default_room_opts = DefRoomOpts, default_room_opts = DefRoomOpts,
history_size = HistorySize, history_size = HistorySize,
room_shaper = RoomShaper} = State) -> room_shaper = RoomShaper} = State) ->
Host = exmpp_jid:domain(To),
US = {exmpp_jid:prep_node(To), exmpp_jid:prep_domain(To)}, US = {exmpp_jid:prep_node(To), exmpp_jid:prep_domain(To)},
case ejabberd_cluster:get_node(US) of case ejabberd_cluster:get_node(US) of
Node when Node == node() -> Node when Node == node() ->
@ -430,7 +433,7 @@ handle_info({route, From, To, Packet},
ok ok
end; end;
Node -> Node ->
Proc = gen_mod:get_module_proc(ServerHost, ?PROCNAME), Proc = gen_mod:get_module_proc_existing(ServerHost, ?PROCNAME),
{Proc, Node} ! {route, From, To, Packet} {Proc, Node} ! {route, From, To, Packet}
end, end,
{noreply, State}; {noreply, State};
@ -500,6 +503,9 @@ do_route(Host, ServerHost, Access, HistorySize, RoomShaper,
ejabberd_router:route_error(To, From, Err, Packet) ejabberd_router:route_error(To, From, Err, Packet)
end. end.
l2b(global) -> global;
l2b(String) when is_list(String) -> list_to_binary(String);
l2b(Other) -> Other.
do_route1(Host, ServerHost, Access, HistorySize, RoomShaper, do_route1(Host, ServerHost, Access, HistorySize, RoomShaper,
From, To, Packet, DefRoomOpts) -> From, To, Packet, DefRoomOpts) ->
@ -516,7 +522,7 @@ do_route1(Host, ServerHost, Access, HistorySize, RoomShaper,
case exmpp_iq:xmlel_to_iq(Packet) of case exmpp_iq:xmlel_to_iq(Packet) of
#iq{type = get, ns = ?NS_DISCO_INFO = XMLNS, #iq{type = get, ns = ?NS_DISCO_INFO = XMLNS,
payload = _SubEl, lang = Lang} = IQ -> payload = _SubEl, lang = Lang} = IQ ->
ServerHostB = list_to_binary(ServerHost), ServerHostB = l2b(ServerHost),
Info = ejabberd_hooks:run_fold( Info = ejabberd_hooks:run_fold(
disco_info, ServerHostB, [], disco_info, ServerHostB, [],
[ServerHost, ?MODULE, <<>>, ""]), [ServerHost, ?MODULE, <<>>, ""]),
@ -661,10 +667,20 @@ check_user_can_create_room(ServerHost, AccessCreate, From, RoomID) ->
end. end.
load_permanent_rooms(Host, ServerHost, Access, HistorySize, RoomShaper) -> load_permanent_rooms({global, Prefix}, global, Access, HistorySize, RoomShaper) ->
timer:sleep(2000), %% Wait for ejabberd_hosts to start
lists:foreach(
fun(ServerHost) ->
Host = list_to_binary(Prefix++"."++ServerHost),
load_permanent_rooms(Host, global, Access, HistorySize, RoomShaper)
end,
?MYHOSTS);
load_permanent_rooms(Host, ServerHost1, Access, HistorySize, RoomShaper) ->
F = fun() -> F = fun() ->
Rs = gen_storage:select(Host, muc_room_opt, Rs = gen_storage:select(Host, muc_room_opt,
[{'=', name_host, {'_', Host}}]), [{'=', opt, persistent},
{'=', name_host, {'_', Host}}]),
Names = lists:foldl( Names = lists:foldl(
fun(#muc_room_opt{name_host = {Room, _}}, Names) -> fun(#muc_room_opt{name_host = {Room, _}}, Names) ->
sets:add_element(Room, Names) sets:add_element(Room, Names)
@ -678,7 +694,7 @@ load_permanent_rooms(Host, ServerHost, Access, HistorySize, RoomShaper) ->
Opts = restore_room_internal(Host, Room), Opts = restore_room_internal(Host, Room),
{ok, Pid} = mod_muc_room:start( {ok, Pid} = mod_muc_room:start(
Host, Host,
ServerHost, ServerHost1,
Access, Access,
Room, Room,
HistorySize, HistorySize,
@ -1029,10 +1045,12 @@ get_vh_rooms(Host) when is_binary(Host) ->
%%%%% Database migration from ejabberd 2.1.x %%%%% Database migration from ejabberd 2.1.x
%%%%% %%%%%
update_tables(Host, mnesia) -> update_tables(global, Storage) ->
HostB = list_to_binary(Host), [update_tables(HostB, Storage) || HostB <- ejabberd_hosts:get_hosts(ejabberd)];
update_tables(HostB, mnesia) ->
gen_storage_migration:migrate_mnesia( gen_storage_migration:migrate_mnesia(
Host, muc_registered, HostB, muc_registered,
[{muc_registered, [us_host, nick], [{muc_registered, [us_host, nick],
fun({muc_registered, {{UserS, ServerS}, HostS}, NickS}) -> fun({muc_registered, {{UserS, ServerS}, HostS}, NickS}) ->
#muc_registered{user_host = {exmpp_jid:make(UserS, ServerS), #muc_registered{user_host = {exmpp_jid:make(UserS, ServerS),
@ -1040,7 +1058,7 @@ update_tables(Host, mnesia) ->
nick = list_to_binary(NickS)} nick = list_to_binary(NickS)}
end}]), end}]),
gen_storage_migration:migrate_mnesia( gen_storage_migration:migrate_mnesia(
Host, muc_room_opt, HostB, muc_room_opt,
[{muc_room, [name_host, opts], [{muc_room, [name_host, opts],
fun({muc_room, {NameString, HostString}, Options}) -> fun({muc_room, {NameString, HostString}, Options}) ->
NameHost = {list_to_binary(NameString), NameHost = {list_to_binary(NameString),