receiver.h

Go to the documentation of this file.

// Copyright (C) 2018-2021 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 PERFDHCP_RECEIVER_H
#define PERFDHCP_RECEIVER_H
 
#include <perfdhcp/perf_socket.h>
#include <perfdhcp/command_options.h>
 
#include <dhcp/pkt.h>
 
#include <queue>
#include <thread>
#include <mutex>
#include <atomic>
 
namespace isc {
namespace perfdhcp {
 
class Receiver {
private:
    std::atomic_flag run_flag_;
 
    std::unique_ptr<std::thread> recv_thread_;
 
    std::queue<dhcp::PktPtr> pkt_queue_;
 
    std::mutex pkt_queue_mutex_;
 
    BasePerfSocket &socket_;
 
    bool single_threaded_;
 
    uint8_t ip_version_;
 
public:
    Receiver(BasePerfSocket &socket, bool single_threaded, uint8_t ip_version) :
        socket_(socket),
        single_threaded_(single_threaded),
        ip_version_(ip_version) {
        run_flag_.clear();
    }
 
    ~Receiver();
 
    void start();
 
    void stop();
 
    dhcp::PktPtr getPkt();
 
private:
    void run();
 
    void receivePackets();
 
    dhcp::PktPtr readPktFromSocket();
};
 
}
}
 
#endif /* PERFDHCP_RECEIVER_H */