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 */