From 5f00ac4b2bf2d47ea24a93c362737fe904456d2e Mon Sep 17 00:00:00 2001
From: Shauren <shauren.trinity@gmail.com>
Date: Sat, 16 Dec 2023 20:02:20 +0100
Subject: Core/Misc: Minor refactors in BaseEncoding and SslSocket to prepare
 for future bnetserver changes

---
 src/server/shared/Networking/SslSocket.h | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

(limited to 'src/server')

diff --git a/src/server/shared/Networking/SslSocket.h b/src/server/shared/Networking/SslSocket.h
index 3ab76d235dd..e00b1b6b65e 100644
--- a/src/server/shared/Networking/SslSocket.h
+++ b/src/server/shared/Networking/SslSocket.h
@@ -24,11 +24,11 @@
 
 namespace boostssl = boost::asio::ssl;
 
-template<class SslContext>
+template<class SslContext, class Stream = boostssl::stream<boost::asio::ip::tcp::socket>>
 class SslSocket
 {
 public:
-    explicit SslSocket(boost::asio::ip::tcp::socket&& socket) : _socket(std::move(socket)), _sslSocket(_socket, SslContext::instance())
+    explicit SslSocket(boost::asio::ip::tcp::socket&& socket) : _sslSocket(std::move(socket), SslContext::instance())
     {
         _sslSocket.set_verify_mode(boostssl::verify_none);
     }
@@ -36,13 +36,13 @@ public:
     // adapting tcp::socket api
     void close(boost::system::error_code& error)
     {
-        _socket.close(error);
+        _sslSocket.lowest_layer().close(error);
     }
 
     void shutdown(boost::asio::socket_base::shutdown_type what, boost::system::error_code& shutdownError)
     {
         _sslSocket.shutdown(shutdownError);
-        _socket.shutdown(what, shutdownError);
+        _sslSocket.lowest_layer().shutdown(what, shutdownError);
     }
 
     template<typename MutableBufferSequence, typename ReadHandlerType>
@@ -66,12 +66,12 @@ public:
     template<typename SettableSocketOption>
     void set_option(SettableSocketOption const& option, boost::system::error_code& error)
     {
-        _socket.set_option(option, error);
+        _sslSocket.lowest_layer().set_option(option, error);
     }
 
     boost::asio::ip::tcp::socket::endpoint_type remote_endpoint() const
     {
-        return _socket.remote_endpoint();
+        return _sslSocket.lowest_layer().remote_endpoint();
     }
 
     // ssl api
@@ -81,9 +81,8 @@ public:
         _sslSocket.async_handshake(type, std::move(handler));
     }
 
-private:
-    boost::asio::ip::tcp::socket _socket;
-    boostssl::stream<boost::asio::ip::tcp::socket&> _sslSocket;
+protected:
+    Stream _sslSocket;
 };
 
 #endif // SslSocket_h__
-- 
cgit v1.2.3