new CAPS processing
This commit is contained in:
parent
8e5297f4c3
commit
c13940e1de
|
@ -40,6 +40,11 @@
|
||||||
send_element/2,
|
send_element/2,
|
||||||
socket_type/0,
|
socket_type/0,
|
||||||
get_presence/1,
|
get_presence/1,
|
||||||
|
get_aux_field/2,
|
||||||
|
set_aux_field/3,
|
||||||
|
del_aux_field/2,
|
||||||
|
get_subscription/2,
|
||||||
|
broadcast/4,
|
||||||
get_subscribed/1]).
|
get_subscribed/1]).
|
||||||
|
|
||||||
%% gen_fsm callbacks
|
%% gen_fsm callbacks
|
||||||
|
@ -96,6 +101,7 @@
|
||||||
conn = unknown,
|
conn = unknown,
|
||||||
auth_module = unknown,
|
auth_module = unknown,
|
||||||
ip,
|
ip,
|
||||||
|
aux_fields = [],
|
||||||
lang}).
|
lang}).
|
||||||
|
|
||||||
%-define(DBGFSM, true).
|
%-define(DBGFSM, true).
|
||||||
|
@ -154,6 +160,39 @@ socket_type() ->
|
||||||
get_presence(FsmRef) ->
|
get_presence(FsmRef) ->
|
||||||
?GEN_FSM:sync_send_all_state_event(FsmRef, {get_presence}, 1000).
|
?GEN_FSM:sync_send_all_state_event(FsmRef, {get_presence}, 1000).
|
||||||
|
|
||||||
|
get_aux_field(Key, #state{aux_fields = Opts}) ->
|
||||||
|
case lists:keysearch(Key, 1, Opts) of
|
||||||
|
{value, {_, Val}} ->
|
||||||
|
{ok, Val};
|
||||||
|
_ ->
|
||||||
|
error
|
||||||
|
end.
|
||||||
|
|
||||||
|
set_aux_field(Key, Val, #state{aux_fields = Opts} = State) ->
|
||||||
|
Opts1 = lists:keydelete(Key, 1, Opts),
|
||||||
|
State#state{aux_fields = [{Key, Val}|Opts1]}.
|
||||||
|
|
||||||
|
del_aux_field(Key, #state{aux_fields = Opts} = State) ->
|
||||||
|
Opts1 = lists:keydelete(Key, 1, Opts),
|
||||||
|
State#state{aux_fields = Opts1}.
|
||||||
|
|
||||||
|
get_subscription(From = #jid{}, StateData) ->
|
||||||
|
get_subscription(jlib:jid_tolower(From), StateData);
|
||||||
|
get_subscription(LFrom, StateData) ->
|
||||||
|
LBFrom = setelement(3, LFrom, ""),
|
||||||
|
F = ?SETS:is_element(LFrom, StateData#state.pres_f) orelse
|
||||||
|
?SETS:is_element(LBFrom, StateData#state.pres_f),
|
||||||
|
T = ?SETS:is_element(LFrom, StateData#state.pres_t) orelse
|
||||||
|
?SETS:is_element(LBFrom, StateData#state.pres_t),
|
||||||
|
if F and T -> both;
|
||||||
|
F -> from;
|
||||||
|
T -> to;
|
||||||
|
true -> none
|
||||||
|
end.
|
||||||
|
|
||||||
|
broadcast(FsmRef, Type, From, Packet) ->
|
||||||
|
FsmRef ! {broadcast, Type, From, Packet}.
|
||||||
|
|
||||||
stop(FsmRef) ->
|
stop(FsmRef) ->
|
||||||
?GEN_FSM:send_event(FsmRef, closed).
|
?GEN_FSM:send_event(FsmRef, closed).
|
||||||
|
|
||||||
|
@ -1103,35 +1142,39 @@ handle_info({route, From, To, Packet}, StateName, StateData) ->
|
||||||
{Pass, NewAttrs, NewState} =
|
{Pass, NewAttrs, NewState} =
|
||||||
case Name of
|
case Name of
|
||||||
"presence" ->
|
"presence" ->
|
||||||
|
State = ejabberd_hooks:run_fold(
|
||||||
|
c2s_presence_in, StateData#state.server,
|
||||||
|
StateData,
|
||||||
|
[{From, To, Packet}]),
|
||||||
case xml:get_attr_s("type", Attrs) of
|
case xml:get_attr_s("type", Attrs) of
|
||||||
"probe" ->
|
"probe" ->
|
||||||
LFrom = jlib:jid_tolower(From),
|
LFrom = jlib:jid_tolower(From),
|
||||||
LBFrom = jlib:jid_remove_resource(LFrom),
|
LBFrom = jlib:jid_remove_resource(LFrom),
|
||||||
NewStateData =
|
NewStateData =
|
||||||
case ?SETS:is_element(
|
case ?SETS:is_element(
|
||||||
LFrom, StateData#state.pres_a) orelse
|
LFrom, State#state.pres_a) orelse
|
||||||
?SETS:is_element(
|
?SETS:is_element(
|
||||||
LBFrom, StateData#state.pres_a) of
|
LBFrom, State#state.pres_a) of
|
||||||
true ->
|
true ->
|
||||||
StateData;
|
State;
|
||||||
false ->
|
false ->
|
||||||
case ?SETS:is_element(
|
case ?SETS:is_element(
|
||||||
LFrom, StateData#state.pres_f) of
|
LFrom, State#state.pres_f) of
|
||||||
true ->
|
true ->
|
||||||
A = ?SETS:add_element(
|
A = ?SETS:add_element(
|
||||||
LFrom,
|
LFrom,
|
||||||
StateData#state.pres_a),
|
State#state.pres_a),
|
||||||
StateData#state{pres_a = A};
|
State#state{pres_a = A};
|
||||||
false ->
|
false ->
|
||||||
case ?SETS:is_element(
|
case ?SETS:is_element(
|
||||||
LBFrom, StateData#state.pres_f) of
|
LBFrom, State#state.pres_f) of
|
||||||
true ->
|
true ->
|
||||||
A = ?SETS:add_element(
|
A = ?SETS:add_element(
|
||||||
LBFrom,
|
LBFrom,
|
||||||
StateData#state.pres_a),
|
State#state.pres_a),
|
||||||
StateData#state{pres_a = A};
|
State#state{pres_a = A};
|
||||||
false ->
|
false ->
|
||||||
StateData
|
State
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end,
|
end,
|
||||||
|
@ -1139,66 +1182,66 @@ handle_info({route, From, To, Packet}, StateName, StateData) ->
|
||||||
{false, Attrs, NewStateData};
|
{false, Attrs, NewStateData};
|
||||||
"error" ->
|
"error" ->
|
||||||
NewA = remove_element(jlib:jid_tolower(From),
|
NewA = remove_element(jlib:jid_tolower(From),
|
||||||
StateData#state.pres_a),
|
State#state.pres_a),
|
||||||
{true, Attrs, StateData#state{pres_a = NewA}};
|
{true, Attrs, State#state{pres_a = NewA}};
|
||||||
"invisible" ->
|
"invisible" ->
|
||||||
Attrs1 = lists:keydelete("type", 1, Attrs),
|
Attrs1 = lists:keydelete("type", 1, Attrs),
|
||||||
{true, [{"type", "unavailable"} | Attrs1], StateData};
|
{true, [{"type", "unavailable"} | Attrs1], State};
|
||||||
"subscribe" ->
|
"subscribe" ->
|
||||||
SRes = is_privacy_allow(From, To, Packet, StateData#state.privacy_list),
|
SRes = is_privacy_allow(From, To, Packet, State#state.privacy_list),
|
||||||
{SRes, Attrs, StateData};
|
{SRes, Attrs, State};
|
||||||
"subscribed" ->
|
"subscribed" ->
|
||||||
SRes = is_privacy_allow(From, To, Packet, StateData#state.privacy_list),
|
SRes = is_privacy_allow(From, To, Packet, State#state.privacy_list),
|
||||||
{SRes, Attrs, StateData};
|
{SRes, Attrs, State};
|
||||||
"unsubscribe" ->
|
"unsubscribe" ->
|
||||||
SRes = is_privacy_allow(From, To, Packet, StateData#state.privacy_list),
|
SRes = is_privacy_allow(From, To, Packet, State#state.privacy_list),
|
||||||
{SRes, Attrs, StateData};
|
{SRes, Attrs, State};
|
||||||
"unsubscribed" ->
|
"unsubscribed" ->
|
||||||
SRes = is_privacy_allow(From, To, Packet, StateData#state.privacy_list),
|
SRes = is_privacy_allow(From, To, Packet, State#state.privacy_list),
|
||||||
{SRes, Attrs, StateData};
|
{SRes, Attrs, State};
|
||||||
_ ->
|
_ ->
|
||||||
case ejabberd_hooks:run_fold(
|
case ejabberd_hooks:run_fold(
|
||||||
privacy_check_packet, StateData#state.server,
|
privacy_check_packet, State#state.server,
|
||||||
allow,
|
allow,
|
||||||
[StateData#state.user,
|
[State#state.user,
|
||||||
StateData#state.server,
|
State#state.server,
|
||||||
StateData#state.privacy_list,
|
State#state.privacy_list,
|
||||||
{From, To, Packet},
|
{From, To, Packet},
|
||||||
in]) of
|
in]) of
|
||||||
allow ->
|
allow ->
|
||||||
LFrom = jlib:jid_tolower(From),
|
LFrom = jlib:jid_tolower(From),
|
||||||
LBFrom = jlib:jid_remove_resource(LFrom),
|
LBFrom = jlib:jid_remove_resource(LFrom),
|
||||||
case ?SETS:is_element(
|
case ?SETS:is_element(
|
||||||
LFrom, StateData#state.pres_a) orelse
|
LFrom, State#state.pres_a) orelse
|
||||||
?SETS:is_element(
|
?SETS:is_element(
|
||||||
LBFrom, StateData#state.pres_a) of
|
LBFrom, State#state.pres_a) of
|
||||||
true ->
|
true ->
|
||||||
{true, Attrs, StateData};
|
{true, Attrs, State};
|
||||||
false ->
|
false ->
|
||||||
case ?SETS:is_element(
|
case ?SETS:is_element(
|
||||||
LFrom, StateData#state.pres_f) of
|
LFrom, State#state.pres_f) of
|
||||||
true ->
|
true ->
|
||||||
A = ?SETS:add_element(
|
A = ?SETS:add_element(
|
||||||
LFrom,
|
LFrom,
|
||||||
StateData#state.pres_a),
|
State#state.pres_a),
|
||||||
{true, Attrs,
|
{true, Attrs,
|
||||||
StateData#state{pres_a = A}};
|
State#state{pres_a = A}};
|
||||||
false ->
|
false ->
|
||||||
case ?SETS:is_element(
|
case ?SETS:is_element(
|
||||||
LBFrom, StateData#state.pres_f) of
|
LBFrom, State#state.pres_f) of
|
||||||
true ->
|
true ->
|
||||||
A = ?SETS:add_element(
|
A = ?SETS:add_element(
|
||||||
LBFrom,
|
LBFrom,
|
||||||
StateData#state.pres_a),
|
State#state.pres_a),
|
||||||
{true, Attrs,
|
{true, Attrs,
|
||||||
StateData#state{pres_a = A}};
|
State#state{pres_a = A}};
|
||||||
false ->
|
false ->
|
||||||
{true, Attrs, StateData}
|
{true, Attrs, State}
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end;
|
end;
|
||||||
deny ->
|
deny ->
|
||||||
{false, Attrs, StateData}
|
{false, Attrs, State}
|
||||||
end
|
end
|
||||||
end;
|
end;
|
||||||
"broadcast" ->
|
"broadcast" ->
|
||||||
|
@ -1360,6 +1403,17 @@ handle_info({force_update_presence, LUser}, StateName,
|
||||||
StateData
|
StateData
|
||||||
end,
|
end,
|
||||||
{next_state, StateName, NewStateData};
|
{next_state, StateName, NewStateData};
|
||||||
|
handle_info({broadcast, Type, From, Packet}, StateName, StateData) ->
|
||||||
|
Recipients = ejabberd_hooks:run_fold(
|
||||||
|
c2s_broadcast_recipients, StateData#state.server,
|
||||||
|
[],
|
||||||
|
[StateData, Type, From, Packet]),
|
||||||
|
lists:foreach(
|
||||||
|
fun(USR) ->
|
||||||
|
ejabberd_router:route(
|
||||||
|
From, jlib:make_jid(USR), Packet)
|
||||||
|
end, lists:usort(Recipients)),
|
||||||
|
fsm_next_state(StateName, StateData);
|
||||||
handle_info(Info, StateName, StateData) ->
|
handle_info(Info, StateName, StateData) ->
|
||||||
?ERROR_MSG("Unexpected info: ~p", [Info]),
|
?ERROR_MSG("Unexpected info: ~p", [Info]),
|
||||||
fsm_next_state(StateName, StateData).
|
fsm_next_state(StateName, StateData).
|
||||||
|
|
105
src/mod_caps.erl
105
src/mod_caps.erl
|
@ -53,7 +53,10 @@
|
||||||
]).
|
]).
|
||||||
|
|
||||||
%% hook handlers
|
%% hook handlers
|
||||||
-export([user_send_packet/3]).
|
-export([user_send_packet/3,
|
||||||
|
user_receive_packet/4,
|
||||||
|
c2s_presence_in/2,
|
||||||
|
c2s_broadcast_recipients/5]).
|
||||||
|
|
||||||
-include("ejabberd.hrl").
|
-include("ejabberd.hrl").
|
||||||
-include("jlib.hrl").
|
-include("jlib.hrl").
|
||||||
|
@ -159,6 +162,22 @@ user_send_packet(#jid{luser = User, lserver = Server} = From,
|
||||||
user_send_packet(_From, _To, _Packet) ->
|
user_send_packet(_From, _To, _Packet) ->
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
user_receive_packet(#jid{lserver = Server}, From, _To,
|
||||||
|
{xmlelement, "presence", Attrs, Els}) ->
|
||||||
|
Type = xml:get_attr_s("type", Attrs),
|
||||||
|
if Type == ""; Type == "available" ->
|
||||||
|
case read_caps(Els) of
|
||||||
|
nothing ->
|
||||||
|
ok;
|
||||||
|
#caps{version = Version, exts = Exts} = Caps ->
|
||||||
|
feature_request(Server, From, Caps, [Version | Exts])
|
||||||
|
end;
|
||||||
|
true ->
|
||||||
|
ok
|
||||||
|
end;
|
||||||
|
user_receive_packet(_JID, _From, _To, _Packet) ->
|
||||||
|
ok.
|
||||||
|
|
||||||
caps_stream_features(Acc, MyHost) ->
|
caps_stream_features(Acc, MyHost) ->
|
||||||
case make_my_disco_hash(MyHost) of
|
case make_my_disco_hash(MyHost) of
|
||||||
"" ->
|
"" ->
|
||||||
|
@ -194,6 +213,65 @@ disco_info(_Acc, Host, Module, ?EJABBERD_URI ++ "#" ++ [_|_], Lang) ->
|
||||||
disco_info(Acc, _Host, _Module, _Node, _Lang) ->
|
disco_info(Acc, _Host, _Module, _Node, _Lang) ->
|
||||||
Acc.
|
Acc.
|
||||||
|
|
||||||
|
c2s_presence_in(C2SState, {From, To, {_, _, Attrs, Els}}) ->
|
||||||
|
Type = xml:get_attr_s("type", Attrs),
|
||||||
|
Subscription = ejabberd_c2s:get_subscription(From, C2SState),
|
||||||
|
Insert = ((Type == "") or (Type == "available"))
|
||||||
|
and ((Subscription == both) or (Subscription == to)),
|
||||||
|
Delete = (Type == "unavailable") or (Type == "error") or (Type == "invisible"),
|
||||||
|
if Insert or Delete ->
|
||||||
|
LFrom = jlib:jid_tolower(From),
|
||||||
|
Rs = case ejabberd_c2s:get_aux_field(caps_resources, C2SState) of
|
||||||
|
{ok, Rs1} ->
|
||||||
|
Rs1;
|
||||||
|
error ->
|
||||||
|
gb_trees:empty()
|
||||||
|
end,
|
||||||
|
Caps = read_caps(Els),
|
||||||
|
{FromUnavail, NewRs} =
|
||||||
|
case Caps of
|
||||||
|
nothing when Insert == true ->
|
||||||
|
{false, Rs};
|
||||||
|
_ when Insert == true ->
|
||||||
|
case gb_trees:lookup(LFrom, Rs) of
|
||||||
|
none ->
|
||||||
|
{true, gb_trees:insert(LFrom, Caps, Rs)};
|
||||||
|
_ ->
|
||||||
|
{false, gb_trees:update(LFrom, Caps, Rs)}
|
||||||
|
end;
|
||||||
|
_ ->
|
||||||
|
{false, gb_trees:delete_any(LFrom, Rs)}
|
||||||
|
end,
|
||||||
|
if FromUnavail ->
|
||||||
|
ejabberd_hooks:run(caps_user_available, To#jid.lserver,
|
||||||
|
[From, To, get_features(Caps)]);
|
||||||
|
true ->
|
||||||
|
ok
|
||||||
|
end,
|
||||||
|
ejabberd_c2s:set_aux_field(caps_resources, NewRs, C2SState);
|
||||||
|
true ->
|
||||||
|
C2SState
|
||||||
|
end.
|
||||||
|
|
||||||
|
c2s_broadcast_recipients(InAcc, C2SState, {pep_message, Feature},
|
||||||
|
_From, _Packet) ->
|
||||||
|
case ejabberd_c2s:get_aux_field(caps_resources, C2SState) of
|
||||||
|
{ok, Rs} ->
|
||||||
|
gb_trees_fold(
|
||||||
|
fun(USR, Caps, Acc) ->
|
||||||
|
case lists:member(Feature, get_features(Caps)) of
|
||||||
|
true ->
|
||||||
|
[USR|Acc];
|
||||||
|
false ->
|
||||||
|
Acc
|
||||||
|
end
|
||||||
|
end, InAcc, Rs);
|
||||||
|
_ ->
|
||||||
|
InAcc
|
||||||
|
end;
|
||||||
|
c2s_broadcast_recipients(Acc, _, _, _, _) ->
|
||||||
|
Acc.
|
||||||
|
|
||||||
%%====================================================================
|
%%====================================================================
|
||||||
%% gen_server callbacks
|
%% gen_server callbacks
|
||||||
%%====================================================================
|
%%====================================================================
|
||||||
|
@ -214,8 +292,14 @@ init([Host, Opts]) ->
|
||||||
MaxSize = gen_mod:get_opt(cache_size, Opts, 1000),
|
MaxSize = gen_mod:get_opt(cache_size, Opts, 1000),
|
||||||
LifeTime = gen_mod:get_opt(cache_life_time, Opts, timer:hours(24) div 1000),
|
LifeTime = gen_mod:get_opt(cache_life_time, Opts, timer:hours(24) div 1000),
|
||||||
cache_tab:new(caps_features, [{max_size, MaxSize}, {life_time, LifeTime}]),
|
cache_tab:new(caps_features, [{max_size, MaxSize}, {life_time, LifeTime}]),
|
||||||
|
ejabberd_hooks:add(c2s_presence_in, Host,
|
||||||
|
?MODULE, c2s_presence_in, 75),
|
||||||
|
ejabberd_hooks:add(c2s_broadcast_recipients, Host,
|
||||||
|
?MODULE, c2s_broadcast_recipients, 75),
|
||||||
ejabberd_hooks:add(user_send_packet, Host,
|
ejabberd_hooks:add(user_send_packet, Host,
|
||||||
?MODULE, user_send_packet, 75),
|
?MODULE, user_send_packet, 75),
|
||||||
|
ejabberd_hooks:add(user_receive_packet, Host,
|
||||||
|
?MODULE, user_receive_packet, 75),
|
||||||
ejabberd_hooks:add(c2s_stream_features, Host,
|
ejabberd_hooks:add(c2s_stream_features, Host,
|
||||||
?MODULE, caps_stream_features, 75),
|
?MODULE, caps_stream_features, 75),
|
||||||
ejabberd_hooks:add(s2s_stream_features, Host,
|
ejabberd_hooks:add(s2s_stream_features, Host,
|
||||||
|
@ -241,8 +325,14 @@ handle_info(_Info, State) ->
|
||||||
|
|
||||||
terminate(_Reason, State) ->
|
terminate(_Reason, State) ->
|
||||||
Host = State#state.host,
|
Host = State#state.host,
|
||||||
|
ejabberd_hooks:delete(c2s_presence_in, Host,
|
||||||
|
?MODULE, c2s_presence_in, 75),
|
||||||
|
ejabberd_hooks:delete(c2s_broadcast_recipients, Host,
|
||||||
|
?MODULE, c2s_broadcast_recipients, 75),
|
||||||
ejabberd_hooks:delete(user_send_packet, Host,
|
ejabberd_hooks:delete(user_send_packet, Host,
|
||||||
?MODULE, user_send_packet, 75),
|
?MODULE, user_send_packet, 75),
|
||||||
|
ejabberd_hooks:delete(user_receive_packet, Host,
|
||||||
|
?MODULE, user_receive_packet, 75),
|
||||||
ejabberd_hooks:delete(c2s_stream_features, Host,
|
ejabberd_hooks:delete(c2s_stream_features, Host,
|
||||||
?MODULE, caps_stream_features, 75),
|
?MODULE, caps_stream_features, 75),
|
||||||
ejabberd_hooks:delete(s2s_stream_features, Host,
|
ejabberd_hooks:delete(s2s_stream_features, Host,
|
||||||
|
@ -506,3 +596,16 @@ concat_xdata_fields(Fields) ->
|
||||||
Acc
|
Acc
|
||||||
end, ["", []], Fields),
|
end, ["", []], Fields),
|
||||||
[Form, $<, lists:sort(Res)].
|
[Form, $<, lists:sort(Res)].
|
||||||
|
|
||||||
|
gb_trees_fold(F, Acc, Tree) ->
|
||||||
|
Iter = gb_trees:iterator(Tree),
|
||||||
|
gb_trees_fold_iter(F, Acc, Iter).
|
||||||
|
|
||||||
|
gb_trees_fold_iter(F, Acc, Iter) ->
|
||||||
|
case gb_trees:next(Iter) of
|
||||||
|
{Key, Val, NewIter} ->
|
||||||
|
NewAcc = F(Key, Val, Acc),
|
||||||
|
gb_trees_fold_iter(F, NewAcc, NewIter);
|
||||||
|
_ ->
|
||||||
|
Acc
|
||||||
|
end.
|
||||||
|
|
Loading…
Reference in New Issue