net: extend Sock with a method to check whether connected

vasild · vasild · commit 5bac7e45e1d3 · 2021-03-01T17:36:17.000+01:00
This will be convenient in the I2P SAM implementation.
diff --git a/src/util/sock.cpp b/src/util/sock.cpp
@@ -250,6 +250,31 @@ std::string Sock::RecvUntilTerminator(uint8_t terminator,
     }
 }
 
+bool Sock::IsConnected(std::string& errmsg) const
+{
+    if (m_socket == INVALID_SOCKET) {
+        errmsg = "not connected";
+        return false;
+    }
+
+    char c;
+    switch (Recv(&c, sizeof(c), MSG_PEEK)) {
+    case -1: {
+        const int err = WSAGetLastError();
+        if (IOErrorIsPermanent(err)) {
+            errmsg = NetworkErrorString(err);
+            return false;
+        }
+        return true;
+    }
+    case 0:
+        errmsg = "closed";
+        return false;
+    default:
+        return true;
+    }
+}
+
 #ifdef WIN32
 std::string NetworkErrorString(int err)
 {
diff --git a/src/util/sock.h b/src/util/sock.h
@@ -143,6 +143,13 @@ class Sock
                                             std::chrono::milliseconds timeout,
                                             CThreadInterrupt& interrupt) const;
 
+    /**
+     * Check if still connected.
+     * @param[out] err The error string, if the socket has been disconnected.
+     * @return true if connected
+     */
+    virtual bool IsConnected(std::string& errmsg) const;
+
 private:
     /**
      * Contained socket. `INVALID_SOCKET` designates the object is empty.
