ServerConnection.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/net/TCPSocket.h>
#include <sptk5/threads/Runable.h>
#include <sptk5/threads/Thread.h>
#include <functional>
 
#ifndef _WIN32
 
#include <netinet/in.h>
 
#endif
 
namespace sptk {
 
class TCPServer;
 
class SP_EXPORT ServerConnection
    : public Runable
{
    friend class TCPServer;
 
public:
    enum class Type
    {
        TCP,
        SSL
    };
 
    using Function = std::function<void(const Runable& task, TCPSocket& socket, const String& address)>;
 
    ServerConnection(TCPServer& server, SOCKET connectionSocket, Type type, const sockaddr_in* connectionAddress,
                     const String& taskName, const ServerConnection::Function& connectionFunction = {});
 
    TCPSocket& socket() const;
 
    TCPServer& server() const;
 
    String address() const
    {
        return m_address;
    }
 
    size_t serial() const
    {
        return m_serial;
    }
 
protected:
    void setSocket(const STCPSocket& socket);
 
    void parseAddress(const sockaddr_in* connectionAddress);
 
    void run() override
    {
        if (m_connectionFunction)
        {
            m_connectionFunction(*this, socket(), address());
        }
    }
 
private:
    mutable std::mutex m_mutex;
    TCPServer& m_server;                             
    STCPSocket m_socket;                             
    String m_address;                                
    size_t m_serial {0};                             
    Type m_type;                                     
    ServerConnection::Function m_connectionFunction; 
 
    static size_t nextSerial();
};
 
using SServerConnection = std::shared_ptr<ServerConnection>;
 
} // namespace sptk