diff --git a/src/lib/net/SecureListenSocket.cpp b/src/lib/net/SecureListenSocket.cpp index 1d678ac64..8cccab94f 100644 --- a/src/lib/net/SecureListenSocket.cpp +++ b/src/lib/net/SecureListenSocket.cpp @@ -34,12 +34,12 @@ SecureListenSocket::SecureListenSocket( std::unique_ptr SecureListenSocket::accept() { - std::unique_ptr socket; + std::unique_ptr secureSocket; try { - socket = std::make_unique( - m_events, m_socketMultiplexer, ARCH->acceptSocket(m_socket, nullptr), m_securityLevel + secureSocket = std::make_unique( + events(), socketMultiplexer(), ARCH->acceptSocket(socket(), nullptr), m_securityLevel ); - socket->initSsl(true); + secureSocket->initSsl(true); setListeningJob(); @@ -51,20 +51,20 @@ std::unique_ptr SecureListenSocket::accept() certificateFilename = ArgParser::argsBase().m_tlsCertFile; } - if (!socket->loadCertificates(certificateFilename)) { + if (!secureSocket->loadCertificates(certificateFilename)) { return nullptr; } - socket->secureAccept(); + secureSocket->secureAccept(); - return socket; + return secureSocket; } catch (XArchNetwork &) { - if (socket) { + if (secureSocket) { setListeningJob(); } return nullptr; } catch (std::exception &ex) { - if (socket) { + if (secureSocket) { setListeningJob(); } throw ex; diff --git a/src/lib/net/TCPListenSocket.h b/src/lib/net/TCPListenSocket.h index dcb648624..c9ea7f632 100644 --- a/src/lib/net/TCPListenSocket.h +++ b/src/lib/net/TCPListenSocket.h @@ -44,10 +44,24 @@ public: protected: void setListeningJob(); - IEventQueue *m_events; - ArchSocket m_socket; - SocketMultiplexer *m_socketMultiplexer; + ArchSocket socket() const + { + return m_socket; + } + + IEventQueue *events() const + { + return m_events; + } + + SocketMultiplexer *socketMultiplexer() const + { + return m_socketMultiplexer; + } private: + ArchSocket m_socket; + IEventQueue *m_events; + SocketMultiplexer *m_socketMultiplexer; std::mutex m_mutex; };