SSLSocket.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/sptk.h>
 
#include <memory>
#include <sptk5/net/SSLContext.h>
#include <sptk5/net/SSLKeys.h>
#include <sptk5/net/TCPSocket.h>
 
namespace sptk {
 
class SP_EXPORT SSLSocket : public TCPSocket
    , public std::mutex
{
public:
    size_t socketBytes() override;
 
    [[noreturn]] void throwSSLError(const String& function, int resultCode) const;
 
    explicit SSLSocket(const String& cipherList = "ALL");
 
    ~SSLSocket() override;
 
    void loadKeys(const SSLKeys& keys);
 
    void setSNIHostName(const String& sniHostName);
 
    void attach(SOCKET socketHandle, bool accept) override;
 
    void close() noexcept override;
 
    SSL* handle()
    {
        return m_ssl;
    }
 
    size_t recv(uint8_t* buffer, size_t size) override;
 
    size_t send(const uint8_t* buffer, size_t len) override;
 
protected:
    void initContextAndSocket();
 
    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;
 
    virtual String getSSLError(const std::string& function, int32_t SSLError) const;
 
private:
    SharedSSLContext m_sslContext {nullptr}; 
    SSL* m_ssl {nullptr};                    
    SSLKeys m_keys;                          
 
    String m_sniHostName; 
    String m_cipherList;  
 
    void openSocketFD(bool blockingMode, const std::chrono::milliseconds& timeout);
 
    bool tryConnect(const DateTime& timeoutAt);
};
 
} // namespace sptk