From f9b06df544c8134b5982b76f2d24aa93289f6d71 Mon Sep 17 00:00:00 2001 From: auth Date: Sat, 11 Jul 2020 17:09:27 +0200 Subject: Added blacklist implementation on server. Overall code cleanup and optimization. --- server/src/client/client.h | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'server/src/client/client.h') diff --git a/server/src/client/client.h b/server/src/client/client.h index 141ac67..a3d558d 100644 --- a/server/src/client/client.h +++ b/server/src/client/client.h @@ -24,8 +24,10 @@ class client { void cleanup() { close(m_socket); - SSL_shutdown(m_ssl); - SSL_free(m_ssl); + if (m_ssl) { + SSL_shutdown(m_ssl); + SSL_free(m_ssl); + } m_socket = -1; } @@ -63,8 +65,9 @@ class client { int& get_socket() { return m_socket; } auto& get_ip() { return m_ip; } - auto& get_session() { return m_session_id; } - operator bool() const { return m_ssl && m_socket > 0; } + operator bool() const { return m_socket > 0; } + auto &operator()() { return m_session_id; } + }; }; // namespace tcp \ No newline at end of file -- cgit v1.2.3