perf_socket.h

Go to the documentation of this file.

// Copyright (C) 2012-2019 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#ifndef PERF_SOCKET_H
#define PERF_SOCKET_H

#include <perfdhcp/command_options.h>

#include <dhcp/pkt4.h>
#include <dhcp/pkt6.h>
#include <dhcp/socket_info.h>
#include <dhcp/iface_mgr.h>

namespace isc {
namespace perfdhcp {

class BasePerfSocket : public dhcp::SocketInfo {
public:
    uint16_t ifindex_;

    BasePerfSocket() :
        SocketInfo(asiolink::IOAddress("127.0.0.1"), 0, 0),
        ifindex_(0) {}

    virtual ~BasePerfSocket() = default;

    virtual dhcp::Pkt4Ptr receive4(uint32_t timeout_sec, uint32_t timeout_usec) = 0;

    virtual dhcp::Pkt6Ptr receive6(uint32_t timeout_sec, uint32_t timeout_usec) = 0;

    virtual bool send(const dhcp::Pkt4Ptr& pkt) = 0;

    virtual bool send(const dhcp::Pkt6Ptr& pkt) = 0;

    virtual dhcp::IfacePtr getIface() = 0;
};

class PerfSocket : public BasePerfSocket {
public:
    PerfSocket(CommandOptions& options);

    virtual ~PerfSocket();

    virtual dhcp::Pkt4Ptr receive4(uint32_t timeout_sec, uint32_t timeout_usec) override;

    virtual dhcp::Pkt6Ptr receive6(uint32_t timeout_sec, uint32_t timeout_usec) override;

    virtual bool send(const dhcp::Pkt4Ptr& pkt) override;

    virtual bool send(const dhcp::Pkt6Ptr& pkt) override;

    virtual dhcp::IfacePtr getIface() override;

protected:
    void initSocketData();

    int openSocket(CommandOptions& options) const;
};

}
}

#endif /* PERF_SOCKET_H */