PresenceRouterImpl.java 4.39 KB
Newer Older
Matt Tucker's avatar
Matt Tucker committed
1 2 3 4 5
/**
 * $RCSfile$
 * $Revision$
 * $Date$
 *
Matt Tucker's avatar
Matt Tucker committed
6
 * Copyright (C) 2004 Jive Software. All rights reserved.
Matt Tucker's avatar
Matt Tucker committed
7
 *
Matt Tucker's avatar
Matt Tucker committed
8 9
 * This software is published under the terms of the GNU Public License (GPL),
 * a copy of which is included in this distribution.
Matt Tucker's avatar
Matt Tucker committed
10
 */
Matt Tucker's avatar
Matt Tucker committed
11

Matt Tucker's avatar
Matt Tucker committed
12 13 14 15 16 17 18 19 20
package org.jivesoftware.messenger.spi;

import org.jivesoftware.messenger.container.BasicModule;
import org.jivesoftware.util.LocaleUtils;
import org.jivesoftware.util.Log;
import org.jivesoftware.messenger.*;
import org.jivesoftware.messenger.auth.UnauthorizedException;
import org.jivesoftware.messenger.handler.PresenceSubscribeHandler;
import org.jivesoftware.messenger.handler.PresenceUpdateHandler;
Matt Tucker's avatar
Matt Tucker committed
21 22 23
import org.xmpp.packet.Presence;
import org.xmpp.packet.JID;
import org.xmpp.packet.PacketError;
Matt Tucker's avatar
Matt Tucker committed
24 25 26 27 28 29 30 31

/**
 * Generic presence routing base class.
 *
 * @author Iain Shigeoka
 */
public class PresenceRouterImpl extends BasicModule implements PresenceRouter {

32 33 34 35
    private RoutingTable routingTable;
    private PresenceUpdateHandler updateHandler;
    private PresenceSubscribeHandler subscribeHandler;
    private SessionManager sessionManager;
Matt Tucker's avatar
Matt Tucker committed
36 37 38 39 40 41 42 43 44 45 46 47

    /**
     * Create a packet router.
     */
    public PresenceRouterImpl() {
        super("XMPP Presence Router");
    }

    public void route(Presence packet) {
        if (packet == null) {
            throw new NullPointerException();
        }
48
        Session session = sessionManager.getSession(packet.getFrom());
Matt Tucker's avatar
Matt Tucker committed
49
        if (session == null || session.getStatus() == Session.STATUS_AUTHENTICATED) {
Matt Tucker's avatar
Matt Tucker committed
50 51 52
            handle(packet);
        }
        else {
Matt Tucker's avatar
Matt Tucker committed
53 54 55
            packet.setTo(session.getAddress());
            packet.setFrom((JID)null);
            packet.setError(PacketError.Condition.not_authorized);
Matt Tucker's avatar
Matt Tucker committed
56
            try {
Matt Tucker's avatar
Matt Tucker committed
57
                session.process(packet);
Matt Tucker's avatar
Matt Tucker committed
58 59 60 61 62 63 64 65
            }
            catch (UnauthorizedException ue) {
                Log.error(ue);
            }
        }
    }

    private void handle(Presence packet) {
Matt Tucker's avatar
Matt Tucker committed
66
        JID recipientJID = packet.getTo();
Matt Tucker's avatar
Matt Tucker committed
67
        try {
Matt Tucker's avatar
Matt Tucker committed
68
            Presence.Type type = packet.getType();
Matt Tucker's avatar
Matt Tucker committed
69
            // Presence updates (null is 'available')
Matt Tucker's avatar
Matt Tucker committed
70 71
            if (type == null || Presence.Type.unavailable == type) {
                // check for local server target
Matt Tucker's avatar
Matt Tucker committed
72
                if (recipientJID == null
Matt Tucker's avatar
Matt Tucker committed
73 74 75
                        || recipientJID.getDomain() == null
                        || "".equals(recipientJID.getDomain())
                        || (recipientJID.getNode() == null && recipientJID.getResource() == null)) {
Matt Tucker's avatar
Matt Tucker committed
76 77 78 79 80 81 82 83 84 85 86 87

                    updateHandler.process(packet);
                }
                else {
                    // The user sent a directed presence to an entity
                    ChannelHandler handler = routingTable.getRoute(recipientJID);
                    handler.process(packet);
                    // Notify the PresenceUpdateHandler of the directed presence
                    updateHandler.directedPresenceSent(packet, handler);
                }

            }
Matt Tucker's avatar
Matt Tucker committed
88 89 90 91 92
            else if (Presence.Type.subscribe == type // presence subscriptions
                    || Presence.Type.unsubscribe == type
                    || Presence.Type.subscribed == type
                    || Presence.Type.unsubscribed == type)
            {
Matt Tucker's avatar
Matt Tucker committed
93 94 95 96 97 98 99 100 101 102 103 104 105 106
                subscribeHandler.process(packet);
            }
            else {
                // It's an unknown or ERROR type, just deliver it because there's nothing else to do with it
                routingTable.getRoute(recipientJID).process(packet);
            }

        }
        catch (NoSuchRouteException e) {
            // Do nothing, presence to unreachable routes are dropped
        }
        catch (Exception e) {
            Log.error(LocaleUtils.getLocalizedString("admin.error.routing"), e);
            try {
107
                Session session = sessionManager.getSession(packet.getFrom());
Matt Tucker's avatar
Matt Tucker committed
108 109 110 111 112 113 114 115 116 117 118 119 120
                if (session != null) {
                    Connection conn = session.getConnection();
                    if (conn != null) {
                        conn.close();
                    }
                }
            }
            catch (UnauthorizedException e1) {
                // do nothing
            }
        }
    }

121 122 123 124 125 126
    public void initialize(XMPPServer server) {
        super.initialize(server);
        routingTable = server.getRoutingTable();
        updateHandler = server.getPresenceUpdateHandler();
        subscribeHandler = server.getPresenceSubscribeHandler();
        sessionManager = server.getSessionManager();
Matt Tucker's avatar
Matt Tucker committed
127 128
    }
}