TCPSocket.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/Buffer.h>
#include <sptk5/Exception.h>
#include <sptk5/Strings.h>
#include <sptk5/net/BaseSocket.h>
#include <sptk5/net/Proxy.h>
 
#ifndef _WIN32
 
#include <arpa/inet.h>
#include <fcntl.h>
#include <netdb.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/un.h>
#include <unistd.h>
 
using SOCKET = int;
using SOCKET_ADDRESS_FAMILY = sa_family_t;
 
#define INVALID_SOCKET (-1)
 
#else
#include <winsock2.h>
 
#include <windows.h>
 
using socklen_t = int;
using SOCKET_ADDRESS_FAMILY = unsigned short;
#endif
 
namespace sptk {
 
class SP_EXPORT TCPSocketReader
    : public Buffer
{
public:
    explicit TCPSocketReader(BaseSocket& socket, size_t bufferSize = 16384);
 
    void open();
 
    void close() noexcept;
 
    size_t read(uint8_t* destination, size_t sz, char delimiter, bool read_line, struct sockaddr_in* from = nullptr);
 
    size_t readLine(Buffer& dest, char delimiter);
 
    [[nodiscard]] size_t availableBytes() const;
 
    void readMoreFromSocket(int availableBytes);
 
private:
    BaseSocket& m_socket;
 
    uint32_t m_readOffset {0};
 
    [[nodiscard]] int32_t readFromSocket(sockaddr_in* from);
 
    [[nodiscard]] int32_t bufferedRead(uint8_t* destination, size_t sz, char delimiter, bool read_line,
                                       struct sockaddr_in* from = nullptr);
 
    void handleReadFromSocketError(int error);
};
 
class SP_EXPORT TCPSocket
    : public BaseSocket
{
public:
    explicit TCPSocket(SOCKET_ADDRESS_FAMILY domain = AF_INET, int32_t type = SOCK_STREAM, int32_t protocol = 0);
 
    ~TCPSocket() override;
 
    void setProxy(std::shared_ptr<Proxy> proxy);
 
    void close() noexcept override;
 
    [[nodiscard]] virtual bool accept(SOCKET& clientSocketFD, struct sockaddr_in& clientInfo, std::chrono::milliseconds timeout);
 
    size_t socketBytes() override;
 
    bool readyToRead(std::chrono::milliseconds timeout) override;
 
    size_t readLine(char* buffer, size_t size, char delimiter = '\n');
 
    size_t readLine(Buffer& buffer, char delimiter = '\n');
 
    size_t readLine(String& s, char delimiter = '\n');
 
    size_t read(uint8_t* buffer, size_t size, sockaddr_in* from = nullptr) override;
 
    size_t read(Buffer& buffer, size_t size, sockaddr_in* from = nullptr) override;
 
    size_t read(String& buffer, size_t size, sockaddr_in* from = nullptr) override;
 
    template<typename T>
    size_t read(T& value, sockaddr_in* from = nullptr)
    {
        return read((uint8_t*) &value, sizeof(T), from);
    }
 
protected:
    TCPSocketReader& reader()
    {
        return m_reader;
    }
 
    void _open(const Host& host, OpenMode openMode, bool blockingMode, std::chrono::milliseconds timeout) override;
 
    void _open(const struct sockaddr_in& address, OpenMode openMode, bool blockingMode,
               std::chrono::milliseconds timeout) override;
 
    Proxy* proxy()
    {
        return m_proxy.get();
    }
 
private:
    TCPSocketReader m_reader;       
    std::shared_ptr<Proxy> m_proxy; 
    Buffer m_stringBuffer;          
};
 
using STCPSocket = std::shared_ptr<TCPSocket>;
 
} // namespace sptk