--- /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_lib("exmpp/include/exmpp.hrl").
+-include("mod_privacy.hrl").
+
+start(Host, Opts) ->
+ HostB = list_to_binary(Host),
+ IQDisc = gen_mod:get_opt(iqdisc, Opts, one_queue),
+ ejabberd_hooks:add(privacy_iq_get, HostB,
+ ?MODULE, process_iq_get, 40),
+ ejabberd_hooks:add(privacy_iq_set, HostB,
+ ?MODULE, process_iq_set, 40),
+ mod_disco:register_feature(HostB, ?NS_BLOCKING),
+ gen_iq_handler:add_iq_handler(ejabberd_sm, HostB, ?NS_BLOCKING,
+ ?MODULE, process_iq, IQDisc).
+
+stop(Host) ->
+ HostB = list_to_binary(Host),
+ ejabberd_hooks:delete(privacy_iq_get, HostB,
+ ?MODULE, process_iq_get, 40),
+ ejabberd_hooks:delete(privacy_iq_set, HostB,
+ ?MODULE, process_iq_set, 40),
+ mod_disco:unregister_feature(HostB, ?NS_BLOCKING),
+ gen_iq_handler:remove_iq_handler(ejabberd_sm, HostB, ?NS_BLOCKING).
+
+process_iq(_From, _To, IQ_Rec) ->
+ exmpp_iq:error(IQ_Rec, 'not-allowed').
+
+process_iq_get(_, From, _To, #iq{ns = ?NS_BLOCKING, payload = SubEl}, _) ->
+ case SubEl#xmlel.name == blocklist of
+ true ->
+ LUser = exmpp_jid:prep_node(From),
+ LServer = exmpp_jid:prep_domain(From),
+ process_blocklist_get(LUser, LServer);
+ false ->
+ {error, 'bad-request'}
+ end;
+
+process_iq_get(Acc, _, _, _, _) ->
+ Acc.
+
+process_iq_set(_, From, _To, #iq{ns = ?NS_BLOCKING,
+ payload = #xmlel{name = SubElName,
+ children = SubEls}}) ->
+ LUser = exmpp_jid:prep_node(From),
+ LServer = exmpp_jid:prep_domain(From),
+ case {SubElName, exmpp_xml:remove_cdata_from_list(SubEls)} of
+ {block, []} ->
+ {error, '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, '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).
+
+get_list_blocklist_jids(LUser, LServer, Name) ->
+ Tuples = gen_storage:dirty_select(LServer, privacy_list_data,
+ [{'=', user_host, {LUser, LServer}},
+ {'=', name, Name},
+ {'=', type, jid}]),
+ [Tuple#privacy_list_data.value ||
+ Tuple <- Tuples, Tuple#privacy_list_data.match_all == true].
+
+parse_blocklist_items([], JIDs) ->
+ JIDs;
+
+parse_blocklist_items([#xmlel{name = item} = El | Els], JIDs) ->
+ case exmpp_xml:get_attribute(El, <<"jid">>, false) of
+ false ->
+ %% Tolerate missing jid attribute
+ parse_blocklist_items(Els, JIDs);
+ JID1 ->
+ JID = exmpp_jid:to_binary(exmpp_jid:parse(JID1)),
+ parse_blocklist_items(Els, [JID | JIDs])
+ end;
+
+parse_blocklist_items([_ | Els], JIDs) ->
+ %% Tolerate unknown elements
+ parse_blocklist_items(Els, JIDs).
+
+process_blocklist_block(LUser, LServer, JIDs) ->
+ F =
+ fun() ->
+ case gen_storage:read(LServer, {privacy_list, {LUser, LServer}}) of
+ [] ->
+ %% No lists yet
+ %% TODO: i18n here:
+ Default = <<"Blocked contacts">>,
+ gen_storage:write(LServer,
+ #privacy_list{
+ user_host = {LUser, LServer},
+ name = Default}),
+ gen_storage:write(LServer,
+ #privacy_default_list{
+ user_host = {LUser, LServer},
+ name = Default}),
+ ok;
+ _Lists ->
+ case gen_storage:read(LServer,
+ {privacy_default_list, {LUser, LServer}}) of
+ [#privacy_default_list{name = Default}] ->
+ %% Default list exists
+ Default;
+ [] ->
+ %% No default list yet, create one
+ %% TODO: i18n here:
+ Default = <<"Blocked contacts">>,
+ gen_storage:write(LServer,
+ #privacy_list{
+ user_host = {LUser, LServer},
+ name = Default}),
+ gen_storage:write(LServer,
+ #privacy_default_list{
+ user_host = {LUser, LServer},
+ name = Default})
+ end
+ end,
+ AlreadyBlocked = get_list_blocklist_jids(LUser, LServer, Default),
+ NewItems = lists:foldr(fun(JID, Res) ->
+ case lists:member(JID, AlreadyBlocked) of
+ true ->
+ Res;
+ false ->
+ Data = #privacy_list_data{
+ user_host = {LUser, LServer},
+ name = Default,
+ type = jid,
+ value = JID,
+ action = deny,
+ order = 0,
+ match_all = true
+ },
+ gen_storage:write(LServer, Data),
+ [Data | Res]
+ end
+ end, [], JIDs),
+ {ok, Default, NewItems}
+ end,
+ case gen_storage:transaction(LServer, privacy_list_data, F) of
+ {atomic, {error, _} = Error} ->
+ Error;
+ {atomic, {ok, Default, Data}} ->
+ %% Data = gen_storage:select(LServer, privacy_list_data,
+ %% [{'=', user_host, {LUser, LServer}},
+ %% {'=', name, Default}]),
+ List = list_data_to_items(Data),
+ broadcast_list_update(LUser, LServer, Default, List),
+ broadcast_blocklist_event(LUser, LServer, {block, JIDs}),
+ {result, []};
+ Error ->
+ ?DEBUG("Error ~n~p", [Error]),
+ {error, 'internal-server-error'}
+ end.
+
+%%Copied from mod_privacy
+%% storage representation to ejabberd representation
+list_data_to_items(Data) ->
+ List =
+ lists:map(
+ fun(Data1) ->
+ #listitem{type = Data1#privacy_list_data.type,
+ value = Data1#privacy_list_data.value,
+ action = Data1#privacy_list_data.action,
+ order = Data1#privacy_list_data.order,
+ match_all = Data1#privacy_list_data.match_all,
+ match_iq = Data1#privacy_list_data.match_iq,
+ match_message = Data1#privacy_list_data.match_message,
+ match_presence_in = Data1#privacy_list_data.match_presence_in,
+ match_presence_out = Data1#privacy_list_data.match_presence_out}
+ end, Data),
+ SortedList = lists:keysort(#listitem.order, List),
+ SortedList.
+
+process_blocklist_unblock_all(LUser, LServer) ->
+ F =
+ fun() ->
+ case gen_storage:read(LServer, {privacy_list, {LUser, LServer}}) of
+ [] ->
+ %% No lists, nothing to unblock
+ ok;
+ _ ->
+ case gen_storage:read(LServer, {privacy_default_list, {LUser, LServer}}) of
+ [#privacy_default_list{name = Default}] ->
+ %% Default list, remove all deny items
+ gen_storage:delete_where(LServer, privacy_list_data,
+ [{'=', user_host, {LUser, LServer}},
+ {'=', name, Default},
+ {'=', action, deny},
+ {'=', match_all, true},
+ {'=', type, jid}]),
+ Data = gen_storage:select(LServer, privacy_list_data,
+ [{'=', user_host, {LUser, LServer}},
+ {'=', name, Default}]),
+ {ok, Default, Data};
+ [] ->
+ %% No default list, nothing to unblock
+ ok
+ end
+ end
+ end,
+ case gen_storage:transaction(LServer, privacy_list_data, F) of
+ {atomic, {error, _} = Error} ->
+ Error;
+ {atomic, ok} ->
+ {result, []};
+ {atomic, {ok, Default, Data}} ->
+ List = list_data_to_items(Data),
+ broadcast_list_update(LUser, LServer, Default, List),
+ broadcast_blocklist_event(LUser, LServer, unblock_all),
+ {result, []};
+ _ ->
+ {error, 'internal-server-error'}
+ end.
+
+process_blocklist_unblock(LUser, LServer, JIDs) ->
+ F =
+ fun() ->
+ case gen_storage:read(LServer, {privacy_list, {LUser, LServer}}) of
+ [] ->
+ %% No lists, nothing to unblock
+ ok;
+ _ ->
+ case gen_storage:read(LServer, {privacy_default_list, {LUser, LServer}}) of
+ [#privacy_default_list{name = Default}] ->
+ %% Default list, remove matching deny items
+ lists:foreach(
+ fun(JID) ->
+ gen_storage:delete_where(LServer, privacy_list_data,
+ [{'=', user_host, {LUser, LServer}},
+ {'=', name, Default},
+ {'=', action, deny},
+ {'=', match_all, true},
+ {'=', value, JID},
+ {'=', type, jid}])
+ end, JIDs),
+ Data = gen_storage:select(LServer, privacy_list_data,
+ [{'=', user_host, {LUser, LServer}},
+ {'=', name, Default}]),
+ {ok, Default, Data};
+ [] ->
+ %% No default list, nothing to unblock
+ ok
+ end
+ end
+ end,
+ case gen_storage:transaction(LServer, privacy_list_data, F) of
+ {atomic, {error, _} = Error} ->
+ Error;
+ {atomic, ok} ->
+ {result, []};
+ {atomic, {ok, Default, Data}} ->
+ List = list_data_to_items(Data),
+ broadcast_list_update(LUser, LServer, Default, List),
+ broadcast_blocklist_event(LUser, LServer, {unblock, JIDs}),
+ {result, []};
+ _ ->
+ {error, 'internal-server-error'}
+ end.
+
+broadcast_list_update(LUser, LServer, Name, List) ->
+ NeedDb = is_list_needdb(List),
+ JID = exmpp_jid:make(LUser, LServer),
+ ListString = lists:flatten(io_lib:format("~p.", [#userlist{name = Name, list = List, needdb = NeedDb}])),
+ ejabberd_router:route(
+ JID,
+ JID,
+ #xmlel{name = 'broadcast', ns = privacy_list,
+ attrs = [?XMLATTR(<<"list_name">>, Name)],
+ children = [exmpp_xml:cdata(ListString)]}).
+
+broadcast_blocklist_event(LUser, LServer, Event) ->
+ JID = exmpp_jid:make(LUser, LServer),
+ EventString = lists:flatten(io_lib:format("~p.", [Event])),
+ ejabberd_router:route(
+ JID, JID,
+ #xmlel{name = 'broadcast', ns = blocking,
+ children = [exmpp_xml:cdata(EventString)]}).
+
+process_blocklist_get(LUser, LServer) ->
+ case catch gen_storage:dirty_read(LServer, privacy_default_list, {LUser, LServer}) of
+ {'EXIT', _Reason} ->
+ {error, 'internal-server-error'};
+ [] ->
+ {result, #xmlel{name = 'blocklist', ns = ?NS_BLOCKING}};
+ [#privacy_default_list{name = Default}] ->
+ JIDs = get_list_blocklist_jids(LUser, LServer, Default),
+ Items = lists:map(
+ fun(JID) ->
+ ?DEBUG("JID: ~p",[JID]),
+ #xmlel{name = item, ns = privacy_list,
+ attrs = [?XMLATTR(<<"jid">>, JID)]}
+ end, JIDs),
+ {result,
+ #xmlel{name = 'blocklist', ns = ?NS_BLOCKING, children = Items}}
+ end.
projects / ejabberd / commitdiff