2002-11-24 21:36:57 +01:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
%%% File : ejabberd_sm.erl
|
|
|
|
%%% Author : Alexey Shchepin <alexey@sevcom.net>
|
|
|
|
%%% Purpose :
|
|
|
|
%%% Created : 24 Nov 2002 by Alexey Shchepin <alexey@sevcom.net>
|
|
|
|
%%% Id : $Id$
|
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
|
|
|
|
-module(ejabberd_sm).
|
|
|
|
-author('alexey@sevcom.net').
|
2002-11-30 19:46:16 +01:00
|
|
|
-vsn('$Revision$ ').
|
2002-11-24 21:36:57 +01:00
|
|
|
|
2002-12-14 21:07:26 +01:00
|
|
|
-export([start/0, init/0, open_session/2, close_session/2,
|
|
|
|
get_user_resources/1]).
|
2002-11-24 21:36:57 +01:00
|
|
|
|
|
|
|
-include_lib("mnemosyne/include/mnemosyne.hrl").
|
2002-11-30 19:46:16 +01:00
|
|
|
-include("ejabberd.hrl").
|
2002-11-24 21:36:57 +01:00
|
|
|
|
2002-12-17 21:49:45 +01:00
|
|
|
-record(session, {ur, user, node}).
|
|
|
|
-record(mysession, {ur, pid}).
|
2002-11-24 21:36:57 +01:00
|
|
|
|
|
|
|
|
|
|
|
start() ->
|
|
|
|
spawn(ejabberd_sm, init, []).
|
|
|
|
|
|
|
|
init() ->
|
|
|
|
register(ejabberd_sm, self()),
|
|
|
|
mnesia:create_table(session,[{ram_copies, [node()]},
|
|
|
|
{attributes, record_info(fields, session)}]),
|
2002-12-17 21:49:45 +01:00
|
|
|
mnesia:add_table_index(session, user),
|
2002-11-24 21:36:57 +01:00
|
|
|
mnesia:add_table_index(session, node),
|
|
|
|
mnesia:create_table(mysession,
|
|
|
|
[{ram_copies, [node()]},
|
|
|
|
{local_content, true},
|
|
|
|
{attributes, record_info(fields, mysession)}]),
|
|
|
|
mnesia:subscribe(system),
|
|
|
|
loop().
|
|
|
|
|
|
|
|
loop() ->
|
|
|
|
receive
|
|
|
|
{open_session, User, Resource, From} ->
|
|
|
|
replace_and_register_my_connection(User, Resource, From),
|
|
|
|
replace_alien_connection(User, Resource),
|
|
|
|
loop();
|
2002-11-30 19:46:16 +01:00
|
|
|
{close_session, User, Resource} ->
|
|
|
|
remove_connection(User, Resource),
|
|
|
|
loop();
|
2002-11-24 21:36:57 +01:00
|
|
|
{replace, User, Resource} ->
|
|
|
|
replace_my_connection(User, Resource),
|
|
|
|
loop();
|
|
|
|
{mnesia_system_event, {mnesia_down, Node}} ->
|
|
|
|
clean_table_from_bad_node(Node),
|
|
|
|
loop();
|
2002-11-30 19:46:16 +01:00
|
|
|
{route, From, To, Packet} ->
|
|
|
|
do_route(From, To, Packet),
|
|
|
|
loop();
|
2002-11-24 21:36:57 +01:00
|
|
|
_ ->
|
|
|
|
loop()
|
|
|
|
end.
|
|
|
|
|
|
|
|
|
|
|
|
open_session(User, Resource) ->
|
|
|
|
ejabberd_sm ! {open_session, User, Resource, self()}.
|
|
|
|
|
2002-11-30 19:46:16 +01:00
|
|
|
close_session(User, Resource) ->
|
|
|
|
ejabberd_sm ! {close_session, User, Resource}.
|
|
|
|
|
2002-11-24 21:36:57 +01:00
|
|
|
replace_alien_connection(User, Resource) ->
|
|
|
|
F = fun() ->
|
2002-12-17 21:49:45 +01:00
|
|
|
UR = {User, Resource},
|
|
|
|
Es = mnesia:read({session, UR}),
|
|
|
|
mnesia:write(#session{ur = UR, user = User, node = node()}),
|
2002-11-24 21:36:57 +01:00
|
|
|
Es
|
|
|
|
end,
|
|
|
|
case mnesia:transaction(F) of
|
|
|
|
{atomic, Rs} ->
|
|
|
|
lists:foreach(
|
|
|
|
fun(R) ->
|
|
|
|
if R#session.node /= node() ->
|
|
|
|
{ejabberd_sm, R#session.node} !
|
|
|
|
{replace, User, Resource};
|
|
|
|
true ->
|
|
|
|
ok
|
|
|
|
end
|
|
|
|
end, Rs);
|
|
|
|
_ ->
|
|
|
|
false
|
|
|
|
end.
|
|
|
|
|
|
|
|
|
|
|
|
replace_my_connection(User, Resource) ->
|
|
|
|
F = fun() ->
|
2002-12-17 21:49:45 +01:00
|
|
|
UR = {User, Resource},
|
|
|
|
Es = mnesia:read({mysession, UR}),
|
|
|
|
mnesia:delete({mysession, UR}),
|
2002-11-24 21:36:57 +01:00
|
|
|
Es
|
|
|
|
end,
|
|
|
|
case mnesia:transaction(F) of
|
|
|
|
{atomic, Rs} ->
|
|
|
|
lists:foreach(
|
|
|
|
fun(R) ->
|
2002-12-17 21:49:45 +01:00
|
|
|
R#mysession.pid ! replaced
|
2002-11-24 21:36:57 +01:00
|
|
|
end, Rs);
|
|
|
|
_ ->
|
|
|
|
false
|
|
|
|
end.
|
|
|
|
|
2002-11-30 19:46:16 +01:00
|
|
|
remove_connection(User, Resource) ->
|
|
|
|
F = fun() ->
|
2002-12-17 21:49:45 +01:00
|
|
|
UR = {User, Resource},
|
|
|
|
mnesia:delete({mysession, UR}),
|
|
|
|
mnesia:delete({session, UR})
|
2002-11-30 19:46:16 +01:00
|
|
|
end,
|
|
|
|
mnesia:transaction(F).
|
|
|
|
|
2002-11-24 21:36:57 +01:00
|
|
|
replace_and_register_my_connection(User, Resource, Pid) ->
|
|
|
|
F = fun() ->
|
2002-12-17 21:49:45 +01:00
|
|
|
UR = {User, Resource},
|
|
|
|
Es = mnesia:read({mysession, UR}),
|
|
|
|
mnesia:write(#mysession{ur = UR, pid = Pid}),
|
2002-11-24 21:36:57 +01:00
|
|
|
Es
|
|
|
|
end,
|
|
|
|
case mnesia:transaction(F) of
|
|
|
|
{atomic, Rs} ->
|
|
|
|
lists:foreach(
|
|
|
|
fun(R) ->
|
2002-12-17 21:49:45 +01:00
|
|
|
R#mysession.pid ! replaced
|
2002-11-24 21:36:57 +01:00
|
|
|
end, Rs);
|
|
|
|
_ ->
|
|
|
|
false
|
|
|
|
end.
|
|
|
|
|
|
|
|
|
|
|
|
clean_table_from_bad_node(Node) ->
|
|
|
|
F = fun() ->
|
|
|
|
Es = mnesia:index_read(session, Node, #session.node),
|
|
|
|
lists:foreach(fun(E) ->
|
|
|
|
mnesia:delete_object(session, E, write),
|
|
|
|
mnesia:delete(
|
2002-12-17 21:49:45 +01:00
|
|
|
{user_resource, E#session.ur})
|
2002-11-24 21:36:57 +01:00
|
|
|
end, Es)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F).
|
|
|
|
|
2002-11-30 19:46:16 +01:00
|
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|
|
|
|
|
|
|
do_route(From, To, Packet) ->
|
|
|
|
?DEBUG("session manager~n\tfrom ~p~n\tto ~p~n\tpacket ~P~n",
|
|
|
|
[From, To, Packet, 8]),
|
2002-12-17 21:49:45 +01:00
|
|
|
{User, Server, Resource} = To,
|
2002-11-30 19:46:16 +01:00
|
|
|
F = fun() ->
|
2002-12-17 21:49:45 +01:00
|
|
|
UR = {User, Resource},
|
|
|
|
Sess = mnesia:read({session, UR}),
|
|
|
|
case Sess of
|
2002-11-30 19:46:16 +01:00
|
|
|
[] ->
|
|
|
|
not_exists;
|
2002-12-17 21:49:45 +01:00
|
|
|
[Ses] ->
|
|
|
|
case mnesia:read({mysession, UR}) of
|
2002-11-30 19:46:16 +01:00
|
|
|
[] ->
|
2002-12-17 21:49:45 +01:00
|
|
|
{remote, Ses#session.node};
|
2002-11-30 19:46:16 +01:00
|
|
|
[El] ->
|
2002-12-17 21:49:45 +01:00
|
|
|
{local, El#mysession.pid}
|
2002-11-30 19:46:16 +01:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end,
|
2002-12-17 21:49:45 +01:00
|
|
|
{xmlelement, Name, Attrs, Els} = Packet,
|
|
|
|
case Resource of
|
|
|
|
"" ->
|
2002-11-30 19:46:16 +01:00
|
|
|
% TODO
|
2002-12-17 21:49:45 +01:00
|
|
|
case Name of
|
|
|
|
"presence" ->
|
|
|
|
lists:foreach(
|
|
|
|
fun(R) ->
|
|
|
|
if From /= {User, Server, R} ->
|
|
|
|
ejabberd_sm ! {route,
|
|
|
|
From,
|
|
|
|
{User, Server, R},
|
|
|
|
Packet};
|
|
|
|
true ->
|
|
|
|
ok
|
|
|
|
end
|
|
|
|
end, get_user_resources(User)),
|
|
|
|
ok;
|
|
|
|
"message" ->
|
|
|
|
% TODO
|
|
|
|
ok;
|
|
|
|
"iq" ->
|
|
|
|
% TODO
|
|
|
|
ok;
|
|
|
|
_ ->
|
|
|
|
ok
|
|
|
|
end;
|
|
|
|
_ ->
|
|
|
|
case mnesia:transaction(F) of
|
|
|
|
{atomic, {local, Pid}} ->
|
|
|
|
?DEBUG("sending to process ~p~n", [Pid]),
|
|
|
|
% TODO
|
|
|
|
Pid ! {route, From, To, Packet},
|
|
|
|
%NewAttrs =
|
|
|
|
% jlib:replace_from_to_attrs(jlib:jid_to_string(From),
|
|
|
|
% jlib:jid_to_string(To),
|
|
|
|
% Attrs),
|
|
|
|
%ejabberd_c2s:send_element(
|
|
|
|
% Pid, {xmlelement, Name, NewAttrs, Els}),
|
|
|
|
%?DEBUG("sended~n", []),
|
|
|
|
ok;
|
|
|
|
{atomic, {remote, Node}} ->
|
|
|
|
?DEBUG("sending to node ~p~n", [Node]),
|
|
|
|
{ejabberd_sm, Node} ! {route, From, To, Packet},
|
|
|
|
ok;
|
|
|
|
{atomic, not_exists} ->
|
|
|
|
% TODO
|
|
|
|
?DEBUG("packet droped~n", []),
|
|
|
|
ok;
|
|
|
|
{aborted, Reason} ->
|
|
|
|
?DEBUG("delivery failed: ~p~n", [Reason]),
|
|
|
|
false
|
|
|
|
end
|
2002-11-30 19:46:16 +01:00
|
|
|
end.
|
|
|
|
|
2002-12-14 21:07:26 +01:00
|
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|
|
|
|
|
|
|
get_user_resources(User) ->
|
|
|
|
F = fun() ->
|
2002-12-17 21:49:45 +01:00
|
|
|
mnemosyne:eval(query [X.ur || X <- table(session),
|
|
|
|
X.user = User]
|
2002-12-14 21:07:26 +01:00
|
|
|
end)
|
|
|
|
end,
|
|
|
|
case mnesia:transaction(F) of
|
|
|
|
{atomic, Rs} ->
|
2002-12-17 21:49:45 +01:00
|
|
|
lists:map(fun(R) -> element(2, R) end, Rs);
|
2002-12-14 21:07:26 +01:00
|
|
|
{aborted, Reason} ->
|
|
|
|
?DEBUG("delivery failed: ~p~n", [Reason]),
|
|
|
|
[]
|
|
|
|
end.
|
|
|
|
|
|
|
|
|