TlsSocket.h

Go to the documentation of this file.

// Copyright (c) Fraunhofer MEVIS, Germany. All rights reserved.
// **InsertLicense** code
 
#pragma once
 
#include <asio/TcpSocket.h>
 
#include <ThirdPartyWarningsDisable.h>
#include <boost/asio/ssl.hpp>
#include <utility>
#include <mutex>
#include <ThirdPartyWarningsRestore.h>
 
 
namespace asio {
 
  using TlsContext = boost::asio::ssl::context;
  using TlsStream = boost::asio::ssl::stream<boost::asio::ip::tcp::socket&>;
 
  class ASYNCHRONOUSIO_EXPORT TlsSocket : public TcpSocket
  {
  public:
 
    using HandshakeCompletionHandler = std::function<void(const ErrorCode&)>;
 
    TlsSocket(std::shared_ptr<Context> io_context, std::shared_ptr<TlsContext> tls_context);
 
    ~TlsSocket() override;
 
    void async_handshake(TlsStream::handshake_type handshake_type, const HandshakeCompletionHandler& handler);
 
    bool is_valid() const override;
 
    void async_read(void* buffer, const std::size_t buffer_size, const CompletionHandler& handler) override;
 
    void async_write(const void* buffer, const std::size_t buffer_size, const CompletionHandler& handler) override;
 
    void close() override;
 
  private:
 
    void shutdown_tls();
 
    std::shared_ptr<TlsContext> tls_context;
 
    TlsStream tls_stream;
 
    bool handshake_complete;
 
    mutable std::mutex mutex;
  };
 
}