xmpp_session.h

Go to the documentation of this file.

/*
 * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
 */

#ifndef __XMPP_SESSION_H__
#define __XMPP_SESSION_H__

#include <string>
#include "base/regex.h"
#include "io/ssl_server.h"
#include "io/ssl_session.h"

class XmppServer;
class XmppConnection;
class XmppConnectionManager;
class XmppRegexMock;

class XmppSession : public SslSession {
public:
    XmppSession(XmppConnectionManager *manager, SslSocket *sock,
        bool async_ready = true);
    virtual ~XmppSession();

    void SetConnection(XmppConnection *connection);
    void ClearConnection();
    XmppConnection *Connection() { return connection_; }

    virtual void WriteReady(const boost::system::error_code &error);
    void ProcessWriteReady();

    typedef std::pair<uint64_t, uint64_t> StatsPair; // (packets, bytes)
    StatsPair Stats(unsigned int message_type) const;
    void IncStats(unsigned int message_type, uint64_t bytes);

    static const int kMaxMessageSize = 4096;
    friend class XmppRegexMock;

    virtual int GetSessionInstance() const { return task_instance_; }

    boost::system::error_code EnableTcpKeepalive(int tcp_hold_time);

protected:
    std::string jid;
    virtual void OnRead(Buffer buffer);

private:
    // eg: if tcp_hold_time is 30sec,
    //        keepalive_idle_time_ = 30/3 = 10s
    //        keepalive_interval_ = (30-10) / keepalive_probes_ = 20s/3 = 6s
    //        i.e socket empty - idle timeout (30 sec)
    //            socket tx buffer not empty - tcp user timeout (30 sec)
    static const int kSessionKeepaliveProbes = 3; // # unack probe
    typedef std::deque<Buffer> BufferQueue;

    contrail::regex tag_to_pattern(const char *);
    int MatchRegex(const contrail::regex &patt);
    bool Match(Buffer buffer, int *result, bool NewBuf);
    void SetBuf(const std::string &);
    void ReplaceBuf(const std::string &);
    bool LeftOver() const;

    XmppConnectionManager *manager_;
    XmppConnection *connection_;
    BufferQueue queue_;
    std::string begin_tag_;
    std::string buf_;
    std::string::const_iterator offset_;
    int tag_known_;
    int task_instance_;
    boost::match_results<std::string::const_iterator> res_;
    std::vector<StatsPair> stats_; // packet count
    int keepalive_idle_time_;
    int keepalive_interval_;
    int keepalive_probes_;
    int tcp_user_timeout_;
    bool stream_open_matched_;

    static const contrail::regex patt_;
    static const contrail::regex stream_patt_;
    static const contrail::regex stream_res_end_;
    static const contrail::regex whitespace_;
    static const contrail::regex stream_features_patt_;
    static const contrail::regex starttls_patt_;
    static const contrail::regex proceed_patt_;
    static const contrail::regex end_patt_;

    DISALLOW_COPY_AND_ASSIGN(XmppSession);
};

#endif // __XMPP_SESSION_H__