SocketPool.h

/*
╔══════════════════════════════════════════════════════════════════════════════╗
║                       SIMPLY POWERFUL TOOLKIT (SPTK)                         ║
╟──────────────────────────────────────────────────────────────────────────────╢
║  copyright            © 1999-2022 Alexey Parshin. All rights reserved.       ║
║  email                alexeyp@gmail.com                                      ║
╚══════════════════════════════════════════════════════════════════════════════╝
┌──────────────────────────────────────────────────────────────────────────────┐
│   This library is free software; you can redistribute it and/or modify it    │
│   under the terms of the GNU Library General Public License as published by  │
│   the Free Software Foundation; either version 2 of the License, or (at your │
│   option) any later version.                                                 │
│                                                                              │
│   This library 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 Library   │
│   General Public License for more details.                                   │
│                                                                              │
│   You should have received a copy of the GNU Library General Public License  │
│   along with this library; if not, write to the Free Software Foundation,    │
│   Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.               │
│                                                                              │
│   Please report all bugs and problems to alexeyp@gmail.com.                  │
└──────────────────────────────────────────────────────────────────────────────┘
*/
 
#pragma once
 
#include <sptk5/Exception.h>
#include <sptk5/net/BaseSocket.h>
#include <sptk5/threads/Thread.h>
 
#include <functional>
#include <map>
#include <mutex>
 
#ifdef _WIN32
 
// Windows
#include <WS2tcpip.h>
#include <WinSock2.h>
#include <Windows.h>
#include <wepoll.h>
using SocketEvent = epoll_event;
 
#else
 
#if __linux__ == 1
// Linux
#include <sys/epoll.h>
 
using SocketEvent = epoll_event;
 
#else
// BSD
#include <sys/event.h>
using SocketEvent = kevent;
 
#endif
#endif
 
namespace sptk {
 
enum class SocketEventType : uint8_t
{
    UNKNOWN,          
    HAS_DATA,         
    CONNECTION_CLOSED 
};
 
using SocketEventCallback = std::function<void(uint8_t* userData, SocketEventType eventType)>;
 
#ifdef _WIN32
#define INVALID_EPOLL nullptr
#else
#define INVALID_EPOLL INVALID_SOCKET
#endif // _WIN32
 
class SP_EXPORT SocketPool
    : public std::mutex
{
public:
    explicit SocketPool(const SocketEventCallback& eventsCallback);
 
    SocketPool(const SocketPool&) noexcept = delete;
 
    SocketPool& operator=(const SocketPool&) = delete;
 
    void open();
 
    ~SocketPool();
 
    void waitForEvents(std::chrono::milliseconds timeout) const;
 
    void close();
 
    void watchSocket(BaseSocket& socket, uint8_t* userData);
 
    void forgetSocket(BaseSocket& socket);
 
    [[nodiscard]] bool active() const;
 
private:
#ifdef _WIN32
    HANDLE m_pool {INVALID_EPOLL};
#else
    SOCKET m_pool {INVALID_EPOLL};
#endif // _WIN32
 
    SocketEventCallback m_eventsCallback;
 
    std::map<BaseSocket*, std::shared_ptr<SocketEvent>> m_socketData;
};
 
} // namespace sptk