--- /dev/null
+%%%----------------------------------------------------------------------
+%%% File : mod_blocking.erl
+%%% Author : Stephan Maka
+%%% Purpose : XEP-0191: Simple Communications Blocking
+%%% Created : 24 Aug 2008 by Stephan Maka <stephan@spaceboyz.net>
+%%%
+%%%
+%%% ejabberd, Copyright (C) 2002-2011 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., 59 Temple Place, Suite 330, Boston, MA
+%%% 02111-1307 USA
+%%%
+%%%----------------------------------------------------------------------
+
+-module(mod_blocking).
+
+-behaviour(gen_mod).
+
+-export([start/2, stop/1,
+ process_iq/3,
+ process_iq_set/4,
+ process_iq_get/5]).
+
+-include("ejabberd.hrl").
+-include("jlib.hrl").
+-include("mod_privacy.hrl").
+
+start(Host, Opts) ->
+ IQDisc = gen_mod:get_opt(iqdisc, Opts, one_queue),
+ ejabberd_hooks:add(privacy_iq_get, Host,
+ ?MODULE, process_iq_get, 40),
+ ejabberd_hooks:add(privacy_iq_set, Host,
+ ?MODULE, process_iq_set, 40),
+ mod_disco:register_feature(Host, ?NS_BLOCKING),
+ gen_iq_handler:add_iq_handler(ejabberd_sm, Host, ?NS_BLOCKING,
+ ?MODULE, process_iq, IQDisc).
+
+stop(Host) ->
+ ejabberd_hooks:delete(privacy_iq_get, Host,
+ ?MODULE, process_iq_get, 40),
+ ejabberd_hooks:delete(privacy_iq_set, Host,
+ ?MODULE, process_iq_set, 40),
+ mod_disco:unregister_feature(Host, ?NS_BLOCKING),
+ gen_iq_handler:remove_iq_handler(ejabberd_sm, Host, ?NS_BLOCKING).
+
+process_iq(_From, _To, IQ) ->
+ SubEl = IQ#iq.sub_el,
+ IQ#iq{type = error, sub_el = [SubEl, ?ERR_NOT_ALLOWED]}.
+
+process_iq_get(_, From, _To,
+ #iq{xmlns = ?NS_BLOCKING,
+ sub_el = {xmlelement, "blocklist", _, _}},
+ _) ->
+ #jid{luser = LUser, lserver = LServer} = From,
+ process_blocklist_get(LUser, LServer);
+
+process_iq_get(Acc, _, _, _, _) ->
+ Acc.
+
+process_iq_set(_, From, _To, #iq{xmlns = ?NS_BLOCKING,
+ sub_el = {xmlelement, SubElName, _, SubEls}}) ->
+ #jid{luser = LUser, lserver = LServer} = From,
+ case {SubElName, xml:remove_cdata(SubEls)} of
+ {"block", []} ->
+ {error, ?ERR_BAD_REQUEST};
+ {"block", Els} ->
+ JIDs = parse_blocklist_items(Els, []),
+ process_blocklist_block(LUser, LServer, JIDs);
+ {"unblock", []} ->
+ process_blocklist_unblock_all(LUser, LServer);
+ {"unblock", Els} ->
+ JIDs = parse_blocklist_items(Els, []),
+ process_blocklist_unblock(LUser, LServer, JIDs);
+ _ ->
+ {error, ?ERR_BAD_REQUEST}
+ end;
+
+process_iq_set(Acc, _, _, _) ->
+ Acc.
+
+is_list_needdb(Items) ->
+ lists:any(
+ fun(X) ->
+ case X#listitem.type of
+ subscription -> true;
+ group -> true;
+ _ -> false
+ end
+ end, Items).
+
+list_to_blocklist_jids([], JIDs) ->
+ JIDs;
+
+list_to_blocklist_jids([#listitem{type = jid,
+ action = deny,
+ value = JID} = Item | Items], JIDs) ->
+ case Item of
+ #listitem{match_all = true} ->
+ Match = true;
+ #listitem{match_iq = true,
+ match_message = true,
+ match_presence_in = true,
+ match_presence_out = true} ->
+ Match = true;
+ _ ->
+ Match = false
+ end,
+ if
+ Match ->
+ list_to_blocklist_jids(Items, [JID | JIDs]);
+ true ->
+ list_to_blocklist_jids(Items, JIDs)
+ end;
+
+% Skip Privacy List items than cannot be mapped to Blocking items
+list_to_blocklist_jids([_ | Items], JIDs) ->
+ list_to_blocklist_jids(Items, JIDs).
+
+parse_blocklist_items([], JIDs) ->
+ JIDs;
+
+parse_blocklist_items([{xmlelement, "item", Attrs, _} | Els], JIDs) ->
+ case xml:get_attr("jid", Attrs) of
+ {value, JID1} ->
+ JID = jlib:jid_tolower(jlib:string_to_jid(JID1)),
+ parse_blocklist_items(Els, [JID | JIDs]);
+ false ->
+ % Tolerate missing jid attribute
+ parse_blocklist_items(Els, JIDs)
+ end;
+
+parse_blocklist_items([_ | Els], JIDs) ->
+ % Tolerate unknown elements
+ parse_blocklist_items(Els, JIDs).
+
+process_blocklist_block(LUser, LServer, JIDs) ->
+ F =
+ fun() ->
+ case mnesia:wread({privacy, {LUser, LServer}}) of
+ [] ->
+ % No lists yet
+ P = #privacy{us = {LUser, LServer}},
+ % TODO: i18n here:
+ NewDefault = "Blocked contacts",
+ NewLists1 = [],
+ List = [];
+ [#privacy{default = Default,
+ lists = Lists} = P] ->
+ case lists:keysearch(Default, 1, Lists) of
+ {value, {_, List}} ->
+ % Default list exists
+ NewDefault = Default,
+ NewLists1 = lists:keydelete(Default, 1, Lists);
+ false ->
+ % No default list yet, create one
+ % TODO: i18n here:
+ NewDefault = "Blocked contacts",
+ NewLists1 = Lists,
+ List = []
+ end
+ end,
+
+ AlreadyBlocked = list_to_blocklist_jids(List, []),
+ NewList =
+ lists:foldr(fun(JID, List1) ->
+ case lists:member(JID, AlreadyBlocked) of
+ true ->
+ List1;
+ false ->
+ [#listitem{type = jid,
+ value = JID,
+ action = deny,
+ order = 0,
+ match_all = true
+ } | List1]
+ end
+ end, List, JIDs),
+ NewLists = [{NewDefault, NewList} | NewLists1],
+ mnesia:write(P#privacy{default = NewDefault,
+ lists = NewLists}),
+ {ok, NewDefault, NewList}
+ end,
+ case mnesia:transaction(F) of
+ {atomic, {error, _} = Error} ->
+ Error;
+ {atomic, {ok, Default, List}} ->
+ broadcast_list_update(LUser, LServer, Default, List),
+ broadcast_blocklist_event(LUser, LServer, {block, JIDs}),
+ {result, []};
+ _ ->
+ {error, ?ERR_INTERNAL_SERVER_ERROR}
+ end.
+
+process_blocklist_unblock_all(LUser, LServer) ->
+ F =
+ fun() ->
+ case mnesia:read({privacy, {LUser, LServer}}) of
+ [] ->
+ % No lists, nothing to unblock
+ ok;
+ [#privacy{default = Default,
+ lists = Lists} = P] ->
+ case lists:keysearch(Default, 1, Lists) of
+ {value, {_, List}} ->
+ % Default list, remove all deny items
+ NewList =
+ lists:filter(
+ fun(#listitem{action = A}) ->
+ A =/= deny
+ end, List),
+
+ NewLists1 = lists:keydelete(Default, 1, Lists),
+ NewLists = [{Default, NewList} | NewLists1],
+ mnesia:write(P#privacy{lists = NewLists}),
+
+ {ok, Default, NewList};
+ false ->
+ % No default list, nothing to unblock
+ ok
+ end
+ end
+ end,
+ case mnesia:transaction(F) of
+ {atomic, {error, _} = Error} ->
+ Error;
+ {atomic, ok} ->
+ {result, []};
+ {atomic, {ok, Default, List}} ->
+ broadcast_list_update(LUser, LServer, Default, List),
+ broadcast_blocklist_event(LUser, LServer, unblock_all),
+ {result, []};
+ _ ->
+ {error, ?ERR_INTERNAL_SERVER_ERROR}
+ end.
+
+process_blocklist_unblock(LUser, LServer, JIDs) ->
+ F =
+ fun() ->
+ case mnesia:read({privacy, {LUser, LServer}}) of
+ [] ->
+ % No lists, nothing to unblock
+ ok;
+ [#privacy{default = Default,
+ lists = Lists} = P] ->
+ case lists:keysearch(Default, 1, Lists) of
+ {value, {_, List}} ->
+ % Default list, remove matching deny items
+ NewList =
+ lists:filter(
+ fun(#listitem{action = deny,
+ type = jid,
+ value = JID}) ->
+ not(lists:member(JID, JIDs));
+ (_) ->
+ true
+ end, List),
+
+ NewLists1 = lists:keydelete(Default, 1, Lists),
+ NewLists = [{Default, NewList} | NewLists1],
+ mnesia:write(P#privacy{lists = NewLists}),
+
+ {ok, Default, NewList};
+ false ->
+ % No default list, nothing to unblock
+ ok
+ end
+ end
+ end,
+ case mnesia:transaction(F) of
+ {atomic, {error, _} = Error} ->
+ Error;
+ {atomic, ok} ->
+ {result, []};
+ {atomic, {ok, Default, List}} ->
+ broadcast_list_update(LUser, LServer, Default, List),
+ broadcast_blocklist_event(LUser, LServer, {unblock, JIDs}),
+ {result, []};
+ _ ->
+ {error, ?ERR_INTERNAL_SERVER_ERROR}
+ end.
+
+broadcast_list_update(LUser, LServer, Name, List) ->
+ NeedDb = is_list_needdb(List),
+ ejabberd_router:route(
+ jlib:make_jid(LUser, LServer, ""),
+ jlib:make_jid(LUser, LServer, ""),
+ {xmlelement, "broadcast", [],
+ [{privacy_list,
+ #userlist{name = Name, list = List, needdb = NeedDb},
+ Name}]}).
+
+broadcast_blocklist_event(LUser, LServer, Event) ->
+ JID = jlib:make_jid(LUser, LServer, ""),
+ ejabberd_router:route(
+ JID, JID,
+ {xmlelement, "broadcast", [],
+ [{blocking, Event}]}).
+
+process_blocklist_get(LUser, LServer) ->
+ case catch mnesia:dirty_read(privacy, {LUser, LServer}) of
+ {'EXIT', _Reason} ->
+ {error, ?ERR_INTERNAL_SERVER_ERROR};
+ [] ->
+ {result, [{xmlelement, "blocklist", [{"xmlns", ?NS_BLOCKING}], []}]};
+ [#privacy{default = Default, lists = Lists}] ->
+ case lists:keysearch(Default, 1, Lists) of
+ {value, {_, List}} ->
+ JIDs = list_to_blocklist_jids(List, []),
+ Items = lists:map(
+ fun(JID) ->
+ ?DEBUG("JID: ~p",[JID]),
+ {xmlelement, "item",
+ [{"jid", jlib:jid_to_string(JID)}], []}
+ end, JIDs),
+ {result,
+ [{xmlelement, "blocklist", [{"xmlns", ?NS_BLOCKING}],
+ Items}]};
+ _ ->
+ {result, [{xmlelement, "blocklist", [{"xmlns", ?NS_BLOCKING}], []}]}
+ end
+ end.
projects / ejabberd / commitdiff