perf_socket.h

Go to the documentation of this file.

// Copyright (C) 2012-2023 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:
    unsigned int 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 */