UdpReceiver.h

Go to the documentation of this file.

// This file is part of CoreLibrary containing useful reusable utility
// classes.
//
// Copyright (C) 2014 to present, Duncan Crutchley
// Contact <dac1976github@outlook.com>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License and GNU Lesser General Public License
// for more details.
//
// You should have received a copy of the GNU General Public License
// and GNU Lesser General Public License along with this program. If
// not, see <http://www.gnu.org/licenses/>.

#ifndef UDPRECEIVER
#define UDPRECEIVER

#include <mutex>
#include "IoContextThreadGroup.h"
#include "Threads/SyncEvent.h"

namespace core_lib
{
namespace asio
{
namespace udp
{

class CORE_LIBRARY_DLL_SHARED_API UdpReceiver final
{
public:
    UdpReceiver() = delete;
    UdpReceiver(boost_iocontext_t& ioContext, uint16_t listenPort,
                const defs::check_bytes_left_to_read_t& checkBytesLeftToRead,
                const defs::message_received_handler_t& messageReceivedHandler,
                eUdpOption                              receiveOptions    = eUdpOption::broadcast,
                size_t                                  receiveBufferSize = DEFAULT_UDP_BUF_SIZE);
    UdpReceiver(uint16_t listenPort, const defs::check_bytes_left_to_read_t& checkBytesLeftToRead,
                const defs::message_received_handler_t& messageReceivedHandler,
                eUdpOption                              receiveOptions    = eUdpOption::broadcast,
                size_t                                  receiveBufferSize = DEFAULT_UDP_BUF_SIZE);
    UdpReceiver(const UdpReceiver&) = delete;
    UdpReceiver& operator=(const UdpReceiver&) = delete;
    UdpReceiver(UdpReceiver&&) = delete;
    UdpReceiver& operator=(UdpReceiver&&) = delete;
    ~UdpReceiver();
    uint16_t ListenPort() const;
    void CloseSocket();

private:
    void CreateUdpSocket(eUdpOption receiveOptions, size_t receiveBufferSize);
    void StartAsyncRead();
    void ReadComplete(const boost_sys::error_code& error, size_t bytesReceived);
    void SetClosing(bool closing);
    bool IsClosing() const;
    void ProcessCloseSocket();

private:
    mutable std::mutex m_closingMutex;
    threads::SyncEvent m_closedEvent{};
    bool m_closing{false};
    std::unique_ptr<IoContextThreadGroup> m_ioThreadGroup{};
    boost_iocontext_t::strand m_strand;
    uint16_t m_listenPort{0};
    defs::check_bytes_left_to_read_t m_checkBytesLeftToRead{};
    defs::message_received_handler_t m_messageReceivedHandler{};
    defs::char_buffer_t m_receiveBuffer{};
    defs::char_buffer_t m_messageBuffer{};
    boost_udp_t::endpoint m_senderEndpoint{};
    boost_udp_t::socket m_socket;
};

} // namespace udp
} // namespace asio
} // namespace core_lib

#endif // UDPRECEIVER