PresenceRouter.java 9.89 KB
Newer Older
1 2 3 4 5
/**
 * $RCSfile: PresenceRouter.java,v $
 * $Revision: 3138 $
 * $Date: 2005-12-01 02:13:26 -0300 (Thu, 01 Dec 2005) $
 *
6
 * Copyright (C) 2007 Jive Software. All rights reserved.
7 8 9 10 11
 *
 * This software is published under the terms of the GNU Public License (GPL),
 * a copy of which is included in this distribution.
 */

12
package org.jivesoftware.openfire;
13

14
import org.jivesoftware.openfire.container.BasicModule;
15
import org.jivesoftware.openfire.entitycaps.EntityCapabilitiesManager;
16 17 18 19 20 21
import org.jivesoftware.openfire.handler.PresenceSubscribeHandler;
import org.jivesoftware.openfire.handler.PresenceUpdateHandler;
import org.jivesoftware.openfire.interceptor.InterceptorManager;
import org.jivesoftware.openfire.interceptor.PacketRejectedException;
import org.jivesoftware.openfire.session.ClientSession;
import org.jivesoftware.openfire.session.Session;
22
import org.jivesoftware.openfire.user.RemotePresenceEventDispatcher;
Gaston Dombiak's avatar
Gaston Dombiak committed
23 24 25
import org.jivesoftware.util.LocaleUtils;
import org.jivesoftware.util.Log;
import org.xmpp.packet.*;
26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42

/**
 * <p>Route presence packets throughout the server.</p>
 * <p>Routing is based on the recipient and sender addresses. The typical
 * packet will often be routed twice, once from the sender to some internal
 * server component for handling or processing, and then back to the router
 * to be delivered to it's final destination.</p>
 *
 * @author Iain Shigeoka
 */
public class PresenceRouter extends BasicModule {

    private RoutingTable routingTable;
    private PresenceUpdateHandler updateHandler;
    private PresenceSubscribeHandler subscribeHandler;
    private PresenceManager presenceManager;
    private SessionManager sessionManager;
43
    private EntityCapabilitiesManager entityCapsManager;
44
    private MulticastRouter multicastRouter;
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
    private String serverName;

    /**
     * Constructs a presence router.
     */
    public PresenceRouter() {
        super("XMPP Presence Router");
    }

    /**
     * Routes presence packets.
     *
     * @param packet the packet to route.
     * @throws NullPointerException if the packet is null.
     */
    public void route(Presence packet) {
        if (packet == null) {
            throw new NullPointerException();
        }
64
        ClientSession session = sessionManager.getSession(packet.getFrom());
65 66 67
        try {
            // Invoke the interceptors before we process the read packet
            InterceptorManager.getInstance().invokeInterceptors(packet, session, true, false);
68
            if (session == null || session.getStatus() != Session.STATUS_CONNECTED) {
69 70 71 72 73 74 75 76 77 78
                handle(packet);
            }
            else {
                packet.setTo(session.getAddress());
                packet.setFrom((JID)null);
                packet.setError(PacketError.Condition.not_authorized);
                session.process(packet);
            }
            // Invoke the interceptors after we have processed the read packet
            InterceptorManager.getInstance().invokeInterceptors(packet, session, true, true);
79
        }
80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
        catch (PacketRejectedException e) {
            if (session != null) {
                // An interceptor rejected this packet so answer a not_allowed error
                Presence reply = new Presence();
                reply.setID(packet.getID());
                reply.setTo(session.getAddress());
                reply.setFrom(packet.getTo());
                reply.setError(PacketError.Condition.not_allowed);
                session.process(reply);
                // Check if a message notifying the rejection should be sent
                if (e.getRejectionMessage() != null && e.getRejectionMessage().trim().length() > 0) {
                    // A message for the rejection will be sent to the sender of the rejected packet
                    Message notification = new Message();
                    notification.setTo(session.getAddress());
                    notification.setFrom(packet.getTo());
                    notification.setBody(e.getRejectionMessage());
                    session.process(notification);
                }
            }
99 100 101 102 103
        }
    }

    private void handle(Presence packet) {
        JID recipientJID = packet.getTo();
104
        JID senderJID = packet.getFrom();
105 106 107 108 109 110 111 112 113 114
        // Check if the packet was sent to the server hostname
        if (recipientJID != null && recipientJID.getNode() == null &&
                recipientJID.getResource() == null && serverName.equals(recipientJID.getDomain())) {
            if (packet.getElement().element("addresses") != null) {
                // Presence includes multicast processing instructions. Ask the multicastRouter
                // to route this packet
                multicastRouter.route(packet);
                return;
            }
        }
115
        try {
116 117 118 119 120 121 122 123
            // Presences sent between components are just routed to the component
            if (recipientJID != null && !XMPPServer.getInstance().isLocal(recipientJID) &&
                    !XMPPServer.getInstance().isLocal(senderJID)) {
                // Route the packet
                routingTable.routePacket(recipientJID, packet, false);
                return;
            }

124 125 126 127 128 129 130 131
            Presence.Type type = packet.getType();
            // Presence updates (null is 'available')
            if (type == null || Presence.Type.unavailable == type) {
                // check for local server target
                if (recipientJID == null || recipientJID.getDomain() == null ||
                        "".equals(recipientJID.getDomain()) || (recipientJID.getNode() == null &&
                        recipientJID.getResource() == null) &&
                        serverName.equals(recipientJID.getDomain())) {
132
                    entityCapsManager.process(packet);
133 134 135
                    updateHandler.process(packet);
                }
                else {
136 137 138
                    // Trigger events for presences of remote users
                    if (senderJID != null && !serverName.equals(senderJID.getDomain()) &&
                            !routingTable.hasComponentRoute(senderJID)) {
139
                        entityCapsManager.process(packet);
140 141 142 143 144 145 146 147 148 149
                        if (type == null) {
                            // Remote user has become available
                            RemotePresenceEventDispatcher.remoteUserAvailable(packet);
                        }
                        else if (type == Presence.Type.unavailable) {
                            // Remote user is now unavailable
                            RemotePresenceEventDispatcher.remoteUserUnavailable(packet);
                        }
                    }
                    
150
                    // Check that sender session is still active (let unavailable presence go through)
151
                    Session session = sessionManager.getSession(packet.getFrom());
152
                    if (session != null && session.getStatus() == Session.STATUS_CLOSED && type == null) {
153 154 155 156
                        Log.warn("Rejected available presence: " + packet + " - " + session);
                        return;
                    }

157
                    // The user sent a directed presence to an entity
158
                    // Broadcast it to all connected resources
Gaston Dombiak's avatar
Gaston Dombiak committed
159
                    for (JID jid : routingTable.getRoutes(recipientJID)) {
160
                        // Register the sent directed presence
Gaston Dombiak's avatar
Gaston Dombiak committed
161
                        updateHandler.directedPresenceSent(packet, jid, recipientJID.toString());
162
                        // Route the packet
Gaston Dombiak's avatar
Gaston Dombiak committed
163
                        routingTable.routePacket(jid, packet, false);
164 165 166 167 168 169 170 171 172 173 174 175 176
                    }
                }

            }
            else if (Presence.Type.subscribe == type // presence subscriptions
                    || Presence.Type.unsubscribe == type
                    || Presence.Type.subscribed == type
                    || Presence.Type.unsubscribed == type)
            {
                subscribeHandler.process(packet);
            }
            else if (Presence.Type.probe == type) {
                // Handle a presence probe sent by a remote server
177
                if (!XMPPServer.getInstance().isLocal(recipientJID)) {
Gaston Dombiak's avatar
Gaston Dombiak committed
178
                    routingTable.routePacket(recipientJID, packet, false);
179 180 181 182 183
                }
                else {
                    // Handle probe to a local user
                    presenceManager.handleProbe(packet);
                }
184 185 186 187
            }
            else {
                // It's an unknown or ERROR type, just deliver it because there's nothing
                // else to do with it
Gaston Dombiak's avatar
Gaston Dombiak committed
188
                routingTable.routePacket(recipientJID, packet, false);
189 190 191 192 193 194 195
            }

        }
        catch (Exception e) {
            Log.error(LocaleUtils.getLocalizedString("admin.error.routing"), e);
            Session session = sessionManager.getSession(packet.getFrom());
            if (session != null) {
196
                session.close();
197 198 199 200 201 202 203 204 205 206 207
            }
        }
    }

    public void initialize(XMPPServer server) {
        super.initialize(server);
        serverName = server.getServerInfo().getName();
        routingTable = server.getRoutingTable();
        updateHandler = server.getPresenceUpdateHandler();
        subscribeHandler = server.getPresenceSubscribeHandler();
        presenceManager = server.getPresenceManager();
208
        multicastRouter = server.getMulticastRouter();
209
        sessionManager = server.getSessionManager();
210
        entityCapsManager = EntityCapabilitiesManager.getInstance();
211
    }
Gaston Dombiak's avatar
Gaston Dombiak committed
212 213 214 215

    /**
     * Notification message indicating that a packet has failed to be routed to the receipient.
     *
216
     * @param receipient address of the entity that failed to receive the packet.
Gaston Dombiak's avatar
Gaston Dombiak committed
217 218
     * @param packet Presence packet that failed to be sent to the receipient.
     */
219
    public void routingFailed(JID receipient, Packet packet) {
Gaston Dombiak's avatar
Gaston Dombiak committed
220 221
        // presence packets are dropped silently
    }
222
}