diff --git a/src/mod_private.erl b/src/mod_private.erl index f09c6100e..e127c2029 100644 --- a/src/mod_private.erl +++ b/src/mod_private.erl @@ -60,12 +60,15 @@ start(Host, Opts) -> end, ejabberd_hooks:add(remove_user, Host, ?MODULE, remove_user, 50), + gen_iq_handler:add_iq_handler(ejabberd_local, Host, + ?NS_PRIVATE, ?MODULE, process_sm_iq, IQDisc), gen_iq_handler:add_iq_handler(ejabberd_sm, Host, ?NS_PRIVATE, ?MODULE, process_sm_iq, IQDisc). stop(Host) -> ejabberd_hooks:delete(remove_user, Host, ?MODULE, remove_user, 50), + gen_iq_handler:remove_iq_handler(ejabberd_local, Host, ?NS_PRIVATE), gen_iq_handler:remove_iq_handler(ejabberd_sm, Host, ?NS_PRIVATE). diff --git a/src/mod_roster.erl b/src/mod_roster.erl index 605e8e367..2f5d771ce 100644 --- a/src/mod_roster.erl +++ b/src/mod_roster.erl @@ -100,6 +100,8 @@ start(Host, Opts) -> webadmin_page, 50), ejabberd_hooks:add(webadmin_user, Host, ?MODULE, webadmin_user, 50), + gen_iq_handler:add_iq_handler(ejabberd_local, Host, + ?NS_ROSTER, ?MODULE, process_iq, IQDisc), gen_iq_handler:add_iq_handler(ejabberd_sm, Host, ?NS_ROSTER, ?MODULE, process_iq, IQDisc). @@ -126,6 +128,7 @@ stop(Host) -> webadmin_page, 50), ejabberd_hooks:delete(webadmin_user, Host, ?MODULE, webadmin_user, 50), + gen_iq_handler:remove_iq_handler(ejabberd_local, Host, ?NS_ROSTER), gen_iq_handler:remove_iq_handler(ejabberd_sm, Host, ?NS_ROSTER).