tls_client_.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_CLIENT_H_
#define DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SOCKET_TCP_TLS_CLIENT_H_
// includes
#include <boost/asio.hpp>
#include <boost/asio/ssl.hpp>
#include <string>
#include <string_view>
#include <thread>
 
#include "boost-support/message/tcp/tcp_message.h"
#include "core/include/result.h"
 
namespace boost_support {
namespace socket {
namespace tcp {
 
class TlsClientSocket final {
 public:
  enum class TlsErrorCode : std::uint8_t {
    kOpenFailed,
    kBindingFailed,
    kConnectFailed,
    kTlsHandshakeFailed,
    kGenericError
  };
 
  using TcpHandlerRead = std::function<void(message::tcp::TcpMessagePtr)>;
 
 public:
  TlsClientSocket(std::string_view local_ip_address, std::uint16_t local_port_num,
                  TcpHandlerRead tcp_handler_read, std::string_view ca_certification_path);
 
  ~TlsClientSocket();
 
  core_type::Result<void, TlsErrorCode> Open();
 
  core_type::Result<void, TlsErrorCode> ConnectToHost(std::string_view host_ip_address,
                                                      std::uint16_t host_port_num);
 
  core_type::Result<void, TlsErrorCode> DisconnectFromHost();
 
  core_type::Result<void, TlsErrorCode> Transmit(message::tcp::TcpMessageConstPtr tcp_message);
 
  core_type::Result<void, TlsErrorCode> Destroy();
 
 private:
  using Tcp = boost::asio::ip::tcp;
 
  using TcpSocket = Tcp::socket;
 
  using TlsStream = boost::asio::ssl::stream<TcpSocket>;
 
  using TcpIpAddress = boost::asio::ip::address;
 
  using TcpErrorCodeType = boost::system::error_code;
 
  std::string local_ip_address_;
 
  std::uint16_t local_port_num_;
 
  boost::asio::io_context io_context_;
 
  boost::asio::ssl::context io_ssl_context_;
 
  TlsStream tls_socket_;
 
  std::atomic_bool exit_request_;
 
  std::atomic_bool running_;
 
  std::condition_variable cond_var_;
 
  std::mutex mutex_;
 
  std::thread thread_;
 
  TcpHandlerRead tcp_handler_read_;
 
 private:
  TlsStream::lowest_layer_type& GetNativeTcpSocket();
 
  void HandleMessage();
};
}  // namespace tcp
}  // namespace socket
}  // namespace boost_support
#endif  // DIAG_CLIENT_LIB_LIB_BOOST_SUPPORT_SOCKET_TCP_TLS_CLIENT_H_