mod_posix: Some (perhaps temporary) changes to re-lock the pidfile after truncating...
[prosody.git] / plugins / mod_presence.lua
index 20afc2da934f9af7a956bf57d7c41b767dc30b06..8e6ef85a35ab97ab928f6a7e7a4dcbd6d9c5c492 100644 (file)
@@ -24,20 +24,6 @@ local rostermanager = require "core.rostermanager";
 local sessionmanager = require "core.sessionmanager";
 local offlinemanager = require "core.offlinemanager";
 
-local _core_route_stanza = core_route_stanza;
-local core_route_stanza;
-function core_route_stanza(origin, stanza)
-       if stanza.attr.type ~= nil and stanza.attr.type ~= "unavailable" and stanza.attr.type ~= "error" then
-               local node, host = jid_split(stanza.attr.to);
-               host = hosts[host];
-               if node and host and host.type == "local" then
-                       handle_inbound_presence_subscriptions_and_probes(origin, stanza, jid_bare(stanza.attr.from), jid_bare(stanza.attr.to));
-                       return;
-               end
-       end
-       _core_route_stanza(origin, stanza);
-end
-
 local function select_top_resources(user)
        local priority = 0;
        local recipients = {};
@@ -82,13 +68,13 @@ function handle_normal_presence(origin, stanza)
        for _, res in pairs(user and user.sessions or NULL) do -- broadcast to all resources
                if res ~= origin and res.presence then -- to resource
                        stanza.attr.to = res.full_jid;
-                       core_route_stanza(origin, stanza);
+                       core_post_stanza(origin, stanza, true);
                end
        end
        for jid, item in pairs(roster) do -- broadcast to all interested contacts
                if item.subscription == "both" or item.subscription == "from" then
                        stanza.attr.to = jid;
-                       core_route_stanza(origin, stanza);
+                       core_post_stanza(origin, stanza, true);
                end
        end
        if stanza.attr.type == nil and not origin.presence then -- initial presence
@@ -97,13 +83,13 @@ function handle_normal_presence(origin, stanza)
                for jid, item in pairs(roster) do -- probe all contacts we are subscribed to
                        if item.subscription == "both" or item.subscription == "to" then
                                probe.attr.to = jid;
-                               core_route_stanza(origin, probe);
+                               core_post_stanza(origin, probe, true);
                        end
                end
                for _, res in pairs(user and user.sessions or NULL) do -- broadcast from all available resources
                        if res ~= origin and res.presence then
                                res.presence.attr.to = origin.full_jid;
-                               core_route_stanza(res, res.presence);
+                               core_post_stanza(res, res.presence, true);
                                res.presence.attr.to = nil;
                        end
                end
@@ -116,7 +102,7 @@ function handle_normal_presence(origin, stanza)
                for jid, item in pairs(roster) do -- resend outgoing subscription requests
                        if item.ask then
                                request.attr.to = jid;
-                               core_route_stanza(origin, request);
+                               core_post_stanza(origin, request, true);
                        end
                end
                local offline = offlinemanager.load(node, host);
@@ -136,7 +122,7 @@ function handle_normal_presence(origin, stanza)
                if origin.directed then
                        for jid in pairs(origin.directed) do
                                stanza.attr.to = jid;
-                               core_route_stanza(origin, stanza);
+                               core_post_stanza(origin, stanza, true);
                        end
                        origin.directed = nil;
                end
@@ -170,7 +156,7 @@ function send_presence_of_available_resources(user, host, jid, recipient_session
                                if pres then
                                        if stanza then pres = stanza; pres.attr.from = session.full_jid; end
                                        pres.attr.to = jid;
-                                       core_route_stanza(session, pres);
+                                       core_post_stanza(session, pres, true);
                                        pres.attr.to = nil;
                                        count = count + 1;
                                end
@@ -183,24 +169,27 @@ end
 
 function handle_outbound_presence_subscriptions_and_probes(origin, stanza, from_bare, to_bare)
        local node, host = jid_split(from_bare);
-       if to_bare == origin.username.."@"..origin.host then return; end -- No self contacts
+       if to_bare == from_bare then return; end -- No self contacts
        local st_from, st_to = stanza.attr.from, stanza.attr.to;
        stanza.attr.from, stanza.attr.to = from_bare, to_bare;
        log("debug", "outbound presence "..stanza.attr.type.." from "..from_bare.." for "..to_bare);
-       if stanza.attr.type == "subscribe" then
+       if stanza.attr.type == "probe" then
+               stanza.attr.from, stanza.attr.to = st_from, st_to;
+               return;
+       elseif stanza.attr.type == "subscribe" then
                -- 1. route stanza
                -- 2. roster push (subscription = none, ask = subscribe)
                if rostermanager.set_contact_pending_out(node, host, to_bare) then
                        rostermanager.roster_push(node, host, to_bare);
                end -- else file error
-               core_route_stanza(origin, stanza);
+               core_post_stanza(origin, stanza);
        elseif stanza.attr.type == "unsubscribe" then
                -- 1. route stanza
                -- 2. roster push (subscription = none or from)
                if rostermanager.unsubscribe(node, host, to_bare) then
                        rostermanager.roster_push(node, host, to_bare); -- FIXME do roster push when roster has in fact not changed?
                end -- else file error
-               core_route_stanza(origin, stanza);
+               core_post_stanza(origin, stanza);
        elseif stanza.attr.type == "subscribed" then
                -- 1. route stanza
                -- 2. roster_push ()
@@ -208,7 +197,7 @@ function handle_outbound_presence_subscriptions_and_probes(origin, stanza, from_
                if rostermanager.subscribed(node, host, to_bare) then
                        rostermanager.roster_push(node, host, to_bare);
                end
-               core_route_stanza(origin, stanza);
+               core_post_stanza(origin, stanza);
                send_presence_of_available_resources(node, host, to_bare, origin);
        elseif stanza.attr.type == "unsubscribed" then
                -- 1. route stanza
@@ -216,9 +205,10 @@ function handle_outbound_presence_subscriptions_and_probes(origin, stanza, from_
                if rostermanager.unsubscribed(node, host, to_bare) then
                        rostermanager.roster_push(node, host, to_bare);
                end
-               core_route_stanza(origin, stanza);
+               core_post_stanza(origin, stanza);
        end
        stanza.attr.from, stanza.attr.to = st_from, st_to;
+       return true;
 end
 
 function handle_inbound_presence_subscriptions_and_probes(origin, stanza, from_bare, to_bare)
@@ -231,20 +221,20 @@ function handle_inbound_presence_subscriptions_and_probes(origin, stanza, from_b
                local result, err = rostermanager.is_contact_subscribed(node, host, from_bare);
                if result then
                        if 0 == send_presence_of_available_resources(node, host, st_from, origin) then
-                               core_route_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="unavailable"})); -- TODO send last activity
+                               core_post_stanza(hosts[host], st.presence({from=to_bare, to=st_from, type="unavailable"}), true); -- TODO send last activity
                        end
                elseif not err then
-                       core_route_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="unsubscribed"}));
+                       core_post_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="unsubscribed"}), true);
                end
        elseif stanza.attr.type == "subscribe" then
                if rostermanager.is_contact_subscribed(node, host, from_bare) then
-                       core_route_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="subscribed"})); -- already subscribed
+                       core_post_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="subscribed"}), true); -- already subscribed
                        -- Sending presence is not clearly stated in the RFC, but it seems appropriate
                        if 0 == send_presence_of_available_resources(node, host, from_bare, origin) then
-                               core_route_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="unavailable"})); -- TODO send last activity
+                               core_post_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="unavailable"}), true); -- TODO send last activity
                        end
                else
-                       core_route_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="unavailable"})); -- acknowledging receipt
+                       core_post_stanza(hosts[host], st.presence({from=to_bare, to=from_bare, type="unavailable"}), true); -- acknowledging receipt
                        if not rostermanager.is_contact_pending_in(node, host, from_bare) then
                                if rostermanager.set_contact_pending_in(node, host, from_bare) then
                                        sessionmanager.send_to_available_resources(node, host, stanza);
@@ -268,6 +258,7 @@ function handle_inbound_presence_subscriptions_and_probes(origin, stanza, from_b
                end
        end -- discard any other type
        stanza.attr.from, stanza.attr.to = st_from, st_to;
+       return true;
 end
 
 local outbound_presence_handler = function(data)
@@ -278,12 +269,12 @@ local outbound_presence_handler = function(data)
        if to then
                local t = stanza.attr.type;
                if t ~= nil and t ~= "unavailable" and t ~= "error" then -- check for subscriptions and probes
-                       handle_outbound_presence_subscriptions_and_probes(origin, stanza, jid_bare(stanza.attr.from), jid_bare(stanza.attr.to));
-                       return true;
+                       return handle_outbound_presence_subscriptions_and_probes(origin, stanza, jid_bare(stanza.attr.from), jid_bare(stanza.attr.to));
                end
 
                local to_bare = jid_bare(to);
-               if not(origin.roster[to_bare] and (origin.roster[to_bare].subscription == "both" or origin.roster[to_bare].subscription == "from")) then -- directed presence
+               local roster = origin.roster;
+               if roster and not(roster[to_bare] and (roster[to_bare].subscription == "both" or roster[to_bare].subscription == "from")) then -- directed presence
                        origin.directed = origin.directed or {};
                        if t then -- removing from directed presence list on sending an error or unavailable
                                origin.directed[to] = nil; -- FIXME does it make more sense to add to_bare rather than to?
@@ -306,8 +297,7 @@ module:hook("presence/bare", function(data)
        local t = stanza.attr.type;
        if to then
                if t ~= nil and t ~= "unavailable" and t ~= "error" then -- check for subscriptions and probes sent to bare JID
-                       handle_inbound_presence_subscriptions_and_probes(origin, stanza, jid_bare(stanza.attr.from), jid_bare(stanza.attr.to));
-                       return true;
+                       return handle_inbound_presence_subscriptions_and_probes(origin, stanza, jid_bare(stanza.attr.from), jid_bare(stanza.attr.to));
                end
        
                local user = bare_sessions[to];
@@ -329,8 +319,7 @@ module:hook("presence/full", function(data)
 
        local t = stanza.attr.type;
        if t ~= nil and t ~= "unavailable" and t ~= "error" then -- check for subscriptions and probes sent to full JID
-               handle_inbound_presence_subscriptions_and_probes(origin, stanza, jid_bare(stanza.attr.from), jid_bare(stanza.attr.to));
-               return true;
+               return handle_inbound_presence_subscriptions_and_probes(origin, stanza, jid_bare(stanza.attr.from), jid_bare(stanza.attr.to));
        end
 
        local session = full_sessions[stanza.attr.to];
@@ -347,10 +336,10 @@ module:hook("presence/host", function(data)
        local from_bare = jid_bare(stanza.attr.from);
        local t = stanza.attr.type;
        if t == "probe" then
-               core_route_stanza(hosts[module.host], st.presence({ from = module.host, to = from_bare, id = stanza.attr.id }));
+               core_post_stanza(hosts[module.host], st.presence({ from = module.host, to = from_bare, id = stanza.attr.id }));
        elseif t == "subscribe" then
-               core_route_stanza(hosts[module.host], st.presence({ from = module.host, to = from_bare, id = stanza.attr.id, type = "subscribed" }));
-               core_route_stanza(hosts[module.host], st.presence({ from = module.host, to = from_bare, id = stanza.attr.id }));
+               core_post_stanza(hosts[module.host], st.presence({ from = module.host, to = from_bare, id = stanza.attr.id, type = "subscribed" }));
+               core_post_stanza(hosts[module.host], st.presence({ from = module.host, to = from_bare, id = stanza.attr.id }));
        end
        return true;
 end);
@@ -369,7 +358,7 @@ module:hook("resource-unbind", function(event)
                pres:tag("status"):text("Disconnected: "..err):up();
                for jid in pairs(session.directed) do
                        pres.attr.to = jid;
-                       core_route_stanza(session, pres);
+                       core_post_stanza(session, pres, true);
                end
                session.directed = nil;
        end