bgp_session.h

Go to the documentation of this file.

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

#ifndef SRC_BGP_BGP_SESSION_H_
#define SRC_BGP_BGP_SESSION_H_

#include <boost/scoped_ptr.hpp>

#include <string>

#include "bgp/bgp_proto.h"
#include "io/tcp_session.h"

class BgpPeer;
class BgpSessionManager;

class BgpMessageReader : public TcpMessageReader {
public:
    BgpMessageReader(TcpSession *session, ReceiveCallback callback);
    virtual ~BgpMessageReader();

protected:
    virtual int MsgLength(Buffer buffer, int offset);

    virtual const int GetHeaderLenSize() {
        return kHeaderLenSize;
    }

    virtual const int GetMaxMessageSize() {
        return kMaxMessageSize;
    }

private:
    static const int kHeaderLenSize = 18;
    static const int kMaxMessageSize = 4096;

    DISALLOW_COPY_AND_ASSIGN(BgpMessageReader);
};

class BgpSession : public TcpSession {
public:
    BgpSession(BgpSessionManager *session_mgr, Socket *socket);
    virtual ~BgpSession();

    void SendNotification(int code, int subcode,
                          const std::string &data = std::string());
    virtual int GetSessionInstance() const { return task_instance_; }
    void ProcessWriteReady();

    void set_peer(BgpPeer *peer);
    void clear_peer();
    BgpPeer *peer() { return peer_; }
    void LogNotification(int code, int subcode, const std::string &direction,
                         const std::string &peer_key,
                         const BgpProto::Notification &msg) const;

protected:
    virtual void OnRead(Buffer buffer) {
        reader_->OnRead(buffer);
    }

private:
    bool ReceiveMsg(const u_int8_t *msg, size_t size);
    virtual void WriteReady(const boost::system::error_code &error);

    BgpSessionManager *session_mgr_;
    BgpPeer *peer_;
    int task_instance_;
    boost::scoped_ptr<BgpMessageReader> reader_;

    DISALLOW_COPY_AND_ASSIGN(BgpSession);
};

#endif  // SRC_BGP_BGP_SESSION_H_