pkt0_interface.h

Go to the documentation of this file.

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

#ifndef vnsw_agent_contrail_pkt0_interface_hpp
#define vnsw_agent_contrail_pkt0_interface_hpp

#include <string>
#include <boost/bind.hpp>
#include <boost/function.hpp>
#include <boost/asio.hpp>

#include <pkt/vrouter_interface.h>

// pkt0 interface implementation of VrouterControlInterface
class Pkt0Interface: public VrouterControlInterface {
public:
    typedef std::vector<boost::asio::const_buffer> buffer_list;

    Pkt0Interface(const std::string &name, boost::asio::io_context *io);
    virtual ~Pkt0Interface();

    virtual void InitControlInterface();
    virtual void IoShutdownControlInterface();
    virtual void ShutdownControlInterface();

    const std::string &Name() const { return name_; }
    int Send(uint8_t *buff, uint16_t buff_len, const PacketBufferPtr &pkt);
    const unsigned char *mac_address() const { return mac_address_; }
protected:
    // Implements system specific send for Pkt0Interface
    void SendImpl(uint8_t *buff, uint16_t buff_len, const PacketBufferPtr &pkt,
                  buffer_list& buff_list);

    void AsyncRead();
    void ReadHandler(const boost::system::error_code &err, std::size_t length);
    void WriteHandler(const boost::system::error_code &error,
                      std::size_t length, uint8_t *buff);

    std::string name_;
    int tap_fd_;
    unsigned char mac_address_[ETHER_ADDR_LEN];

    boost::asio::posix::stream_descriptor input_;

    uint8_t *read_buff_;
    PktHandler *pkt_handler_;
    DISALLOW_COPY_AND_ASSIGN(Pkt0Interface);
};

class Pkt0RawInterface : public Pkt0Interface {
public:
    Pkt0RawInterface(const std::string &name, boost::asio::io_context *io);
    virtual ~Pkt0RawInterface();

    void InitControlInterface();

protected:
    DISALLOW_COPY_AND_ASSIGN(Pkt0RawInterface);
};

class Pkt0Socket : public VrouterControlInterface {
public:
    static const uint32_t kConnectTimeout = 1000; // 1 second
    static string sSocketDir;
    static string sAgentSocketPath;
    static string sVrouterSocketPath;
    static void CreateMockAgent(const string&);

    Pkt0Socket(const std::string &name,
               boost::asio::io_context *io);
    ~Pkt0Socket();

    virtual void InitControlInterface();
    virtual void IoShutdownControlInterface();
    virtual void ShutdownControlInterface();
    const std::string &Name() const { return name_; }

    int Send(uint8_t *buff, uint16_t buff_len, const PacketBufferPtr &pkt);
private:
    void AsyncRead();
    void ReadHandler(const boost::system::error_code &err, std::size_t length);
    void WriteHandler(const boost::system::error_code &error,
                      std::size_t length, PacketBufferPtr pkt, uint8_t *buff);
    void CreateUnixSocket();
    void StartConnectTimer();
    bool OnTimeout();
    bool connected_;

    boost::asio::local::datagram_protocol::socket socket_;

    boost::scoped_ptr<Timer> timer_;
    uint8_t *read_buff_;
    PktHandler *pkt_handler_;
    std::string name_;
    DISALLOW_COPY_AND_ASSIGN(Pkt0Socket);
};
#endif // vnsw_agent_contrail_pkt0_interface_hpp