udp_listener.h

#pragma once
 
#include <boost/asio.hpp>
 
typedef std::function<void(const char*, const size_t&)> ReceiveCallback;
 
class UdpListener
{
   public:
    UdpListener(boost::asio::io_service& io_service, const std::string& ip_address,
                unsigned short port, const std::string& interface, bool multicast,
                ReceiveCallback receive_callback);
 
    UdpListener(boost::asio::io_service& io_service, const unsigned short port,
                ReceiveCallback receive_callback);
 
    virtual ~UdpListener();
 
    void close();
 
   private:
    void handleDataReception(const boost::system::error_code& error,
                             std::size_t bytes_transferred);
 
    void startListen();
 
    void setupMulticast(const boost::asio::ip::address& ip_address,
                        const std::string& listen_interface);
 
    // The maximum buffer length for the raw data received from the network
    static constexpr unsigned int MAX_BUFFER_LENGTH = 9000;
 
    // Whether this listener should continue running
    bool running_;
 
    // The raw data received from the network
    std::array<char, MAX_BUFFER_LENGTH> raw_received_data_;
 
    // A UDP socket to receive data on
    boost::asio::ip::udp::socket socket_;
 
    // The endpoint for the sender
    boost::asio::ip::udp::endpoint sender_endpoint_;
 
    // Callback once a new message is received
    ReceiveCallback receive_callback_;
};