xmpp_client.h

Go to the documentation of this file.

/*
 * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
 */
 
#ifndef __XMPP_CLIENT_H__
#define __XMPP_CLIENT_H__
 
#include <boost/asio/ip/tcp.hpp>
#include <boost/ptr_container/ptr_map.hpp>
#include "io/ssl_server.h"
#include "io/ssl_session.h"
#include "xmpp/xmpp_config.h"
#include "xmpp/xmpp_connection.h"
#include "xmpp/xmpp_connection_manager.h"
#include "xmpp/xmpp_init.h"
 
class LifetimeActor;
class LifetimeManager;
class XmppSession;
 
// Class to represent Xmpp Client
class XmppClient : public XmppConnectionManager {
public:
    typedef boost::asio::ip::tcp::endpoint Endpoint;
 
    explicit XmppClient(EventManager *evm);
    XmppClient(EventManager *evm, const XmppChannelConfig *config);
    virtual ~XmppClient();
 
    void Shutdown();
 
    typedef boost::function<void(XmppChannelMux *, xmps::PeerState)>
        ConnectionEventCb;
    void RegisterConnectionEvent(xmps::PeerId, ConnectionEventCb);
    void UnRegisterConnectionEvent(xmps::PeerId);
    void NotifyConnectionEvent(XmppChannelMux *, xmps::PeerState);
    size_t ConnectionEventCount() const;
 
    virtual TcpSession *CreateSession();
    virtual bool Initialize(short port) ;
 
    XmppClientConnection *CreateConnection(const XmppChannelConfig *config);
    XmppClientConnection *FindConnection(const std::string &address);
    void InsertConnection(XmppClientConnection *connection);
    void RemoveConnection(XmppClientConnection *connection);
    size_t ConnectionCount() const;
 
    XmppChannel *FindChannel(const std::string &address);
    void ConfigUpdate(const XmppConfigData *cfg);
    XmppConfigManager *xmpp_config_mgr() { return config_mgr_.get(); }
 
    LifetimeManager *lifetime_manager();
    virtual LifetimeActor *deleter();
    int SetDscpValue(uint8_t value, const char *conn_id);
    uint32_t XmppTimeOut(const char *conn_id);
    void UpdateTimeOut(uint8_t time_out, const char *conn_id);
 
protected:
    virtual SslSession *AllocSession(SslSocket *socket);
 
private:
    class DeleteActor;
    friend class XmppSessionTest;
    friend class XmppStreamMessageTest;
    friend class DeleteActor;
 
    typedef std::map<Endpoint, XmppClientConnection *> ConnectionMap;
    typedef std::map<xmps::PeerId, ConnectionEventCb> ConnectionEventCbMap;
 
    void ProcessConfigUpdate(XmppConfigManager::DiffType delta,
        const XmppChannelConfig *current, const XmppChannelConfig *future);
 
    ConnectionMap connection_map_;
    ConnectionEventCbMap connection_event_map_;
    tbb::mutex connection_event_map_mutex_;
 
    boost::scoped_ptr<XmppConfigManager> config_mgr_;
    boost::scoped_ptr<LifetimeManager> lifetime_manager_;
    boost::scoped_ptr<DeleteActor> deleter_;
 
    bool auth_enabled_;
    int tcp_hold_time_;
 
    DISALLOW_COPY_AND_ASSIGN(XmppClient);
};
 
#endif