Expose the function to check if socket is open (#40)

Author: ali-asadpoor

include/comm/Connection.h

@@ -65,6 +65,7 @@ class Connection
     virtual std::string get_source() const     = 0;
     virtual short get_source_family() const    = 0;
     virtual std::string get_encryption() const = 0;
+    virtual bool is_open()                     = 0;
 
    protected:
     virtual size_t read(uint8_t* buffer, size_t length)        = 0;

src/comm/TCPConnection.cc

@@ -107,3 +107,5 @@ std::string TCPConnection::get_source() const { return _tcp_socket->print_source
 short TCPConnection::get_source_family() const { return _tcp_socket->source_family(); }
 
 std::string TCPConnection::get_encryption() const { return "none"; }
+
+bool TCPConnection::is_open() { return _tcp_socket->is_open(); }

src/comm/TCPConnection.h

@@ -55,6 +55,7 @@ class TCPConnection : public Connection
     std::string get_source() const override;
     short get_source_family() const override;
     std::string get_encryption() const override;
+    bool is_open() override;
 
    protected:
     size_t read(uint8_t* buffer, size_t length) override;

src/comm/TCPSocket.cc

@@ -9,6 +9,7 @@
 #include <sys/socket.h>
 #include <arpa/inet.h>
 #include <unistd.h>
+#include <netinet/tcp.h>
 
 #include "util/gcc_util.h"
 DISABLE_WARNING(effc++)
@@ -142,3 +143,18 @@ std::string TCPSocket::print_source()
 }
 
 short TCPSocket::source_family() { return _source_family; }
+
+bool TCPSocket::is_open()
+{
+    tcp_info socket_info;
+    socklen_t socket_info_length = sizeof(socket_info);
+    if (getsockopt(_socket_fd, SOL_TCP, TCP_INFO, &socket_info, &socket_info_length) == -1) {
+        return false;
+    }
+    if (socket_info.tcpi_state == TCP_CLOSE || socket_info.tcpi_state == TCP_CLOSE_WAIT ||
+        socket_info.tcpi_state == TCP_LAST_ACK || socket_info.tcpi_state == TCP_LISTEN ||
+        socket_info.tcpi_state == TCP_CLOSING) {
+        return false;
+    }
+    return true;
+}

src/comm/TCPSocket.h

@@ -60,6 +60,7 @@ class TCPSocket
     bool set_boolean_option(int level, int option_name, bool value);
     bool set_timeval_option(int level, int option_name, timeval value);
     void shutdown();
+    bool is_open();
 
     std::string print_source();
     short source_family();

src/comm/TLSConnection.cc

@@ -90,3 +90,5 @@ std::string TLSConnection::get_source() const { return _tls_socket->print_source
 short TLSConnection::get_source_family() const { return _tls_socket->source_family(); }
 
 std::string TLSConnection::get_encryption() const { return "tls"; }
+
+bool TLSConnection::is_open() { return _tls_socket->_tcp_socket->is_open(); }

src/comm/TLSConnection.h

@@ -25,6 +25,7 @@ class TLSConnection : public Connection
     std::string get_source() const override;
     short get_source_family() const override;
     std::string get_encryption() const override;
+    bool is_open() override;
 
    protected:
     size_t read(uint8_t* buffer, size_t length) override;

src/comm/TLSSocket.h

@@ -29,6 +29,7 @@ class TLSSocket
 
     void accept();
     void connect();
+    bool is_open();
 
     std::string print_source();
     short source_family();