bgp_session_manager.h

Go to the documentation of this file.

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

#ifndef SRC_BGP_BGP_SESSION_MANAGER_H_
#define SRC_BGP_BGP_SESSION_MANAGER_H_

#include <boost/asio/ip/tcp.hpp>
#include <boost/dynamic_bitset.hpp>

#include "base/queue_task.h"
#include "base/address.h"
#include "io/tcp_server.h"

class BgpPeer;
class BgpServer;
class BgpSession;
class TcpSession;

class BgpSessionManager : public TcpServer {
public:
    BgpSessionManager(EventManager *evm, BgpServer *server);
    virtual ~BgpSessionManager();

    virtual TcpSession *CreateSession();
    virtual bool Initialize(unsigned short port);
    virtual bool Initialize(unsigned short port, const IpAddress &ip);
    void Shutdown();
    void Terminate();
    bool MayDelete() const;

    void EnqueueWriteReady(BgpSession *session);

    BgpServer *server() { return server_; }

protected:
    virtual TcpSession *AllocSession(Socket *socket);
    virtual bool AcceptSession(TcpSession *session);

private:
    friend class BgpServerUnitTest;

    BgpPeer *FindPeer(Endpoint remote);
    bool ProcessSession(BgpSession *session);
    bool ProcessWriteReady(TcpSessionPtr tcp_session);
    void WorkQueueExitCallback(bool done);
    size_t GetSessionQueueSize() const;
    void SetSessionQueueDisable(bool disabled);

    BgpServer *server_;
    WorkQueue<BgpSession *> session_queue_;
    WorkQueue<TcpSessionPtr> write_ready_queue_;

    IpAddress session_ip_;

    DISALLOW_COPY_AND_ASSIGN(BgpSessionManager);
};

#endif  // SRC_BGP_BGP_SESSION_MANAGER_H_