tls_server_.h

Go to the documentation of this file.

/* Diagnostic Client library
* Copyright (C) 2024  Avijit Dey
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/
#ifndef DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SOCKET_TCP_TLS_SERVER_H_
#define DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SOCKET_TCP_TLS_SERVER_H_
 
// includes
#include <boost/asio.hpp>
#include <boost/asio/ssl.hpp>
#include <optional>
#include <string_view>
#include <vector>
 
#include "boost-support/message/tcp/tcp_message.h"
#include "core/include/result.h"
 
namespace boost_support {
namespace socket {
namespace tcp {
 
class TcpServerConnection final {
 public:
  enum class TcpErrorCode : std::uint8_t { kOpenFailed, kBindingFailed, kGenericError };
 
  using TcpHandlerRead = std::function<void(message::tcp::TcpMessagePtr)>;
 
  using Tcp = boost::asio::ip::tcp;
 
  using TcpSocket = Tcp::socket;
 
  using TlsStream = boost::asio::ssl::stream<TcpSocket>;
 
 public:
  TcpServerConnection(TlsStream tls_socket, TcpHandlerRead tcp_handler_read);
 
  ~TcpServerConnection() = default;
 
  TcpServerConnection(TcpServerConnection &&) noexcept = default;
  TcpServerConnection &operator=(TcpServerConnection &&) noexcept = default;
 
  TcpServerConnection(TcpServerConnection const &) = delete;
  TcpServerConnection &operator=(TcpServerConnection const &) = delete;
 
  core_type::Result<void, TcpErrorCode> Transmit(message::tcp::TcpMessageConstPtr tcp_message);
 
  bool TryReceivingMessage();
 
  core_type::Result<void, TcpErrorCode> Shutdown();
 
 private:
  TlsStream tls_socket_;
 
  TcpHandlerRead tcp_handler_read_;
 
  TlsStream::lowest_layer_type &GetNativeTcpSocket();
};
 
class TlsServerSocket final {
 public:
  using TcpHandlerRead = std::function<void(message::tcp::TcpMessagePtr)>;
 
  using TcpAcceptor = boost::asio::ip::tcp::acceptor;
 
 public:
  TlsServerSocket(std::string_view local_ip_address, std::uint16_t local_port_num);
 
  ~TlsServerSocket() = default;
 
  std::optional<TcpServerConnection> GetTcpServerConnection(TcpHandlerRead tcp_handler_read);
 
 private:
  using Tcp = boost::asio::ip::tcp;
 
  std::string local_ip_address_;
 
  std::uint16_t local_port_num_;
 
  boost::asio::io_context io_context_;
 
  boost::asio::ssl::context io_ssl_context_;
 
  TcpAcceptor tcp_acceptor_;
};
 
}  // namespace tcp
}  // namespace socket
}  // namespace boost_support
 
#endif  // DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SOCKET_TCP_TLS_SERVER_H_