HPCC-26873 Use ISocket::shutdown when terminating read sockets

Signed-off-by: Gavin Halliday <[email protected]>

common/thorhelper/persistent.cpp

@@ -218,8 +218,7 @@ class CPersistentHandler : implements IPersistentHandler, implements ISocketSele
             if(m_doNotReuseList.getValue(epstr.str()) != nullptr)
             {
                 PERSILOG(PersistentLogLevel::PLogNormal, "PERSISTENT: socket %d's target endpoint %s is in DoNotReuseList, will not add it.", sock->OShandle(), epstr.str());
-                sock->shutdown();
-                sock->close();
+                shutdownAndCloseNoThrow(sock);
                 return;
             }
         }

esp/platform/espplugin.cpp

@@ -112,8 +112,7 @@ int CEspServiceThread::run()
 
    DBGLOG("Closing Socket(%d)", m_socket->OShandle());
 
-   m_socket->shutdown();
-   m_socket->close();
+   shutdownAndCloseNoThrow(m_socket);
 
    Release();
    return 0;

esp/test/httptest/httptest.cpp

@@ -886,10 +886,8 @@ int COneServerHttpProxyThread::start()
         if(httptest_tracelevel > 5)
             fprintf(m_ofile, ">>sent the response back to %s:%d:\n", peername, port);
 
-        socket2->shutdown();
-        socket2->close();
-        m_client->shutdown();
-        m_client->close();
+        shutdownAndCloseNoThrow(socket2);
+        shutdownAndCloseNoThrow(m_client);
     }
     catch(IException *excpt)
     {
@@ -1092,10 +1090,8 @@ int CHttpProxyThread::run()
             //m_client->write(respbuf.str(), respbuf.length());
             fflush(m_ofile);
 
-            m_remotesocket->shutdown();
-            m_remotesocket->close();
-            m_client->shutdown();
-            m_client->close();
+            shutdownAndCloseNoThrow(m_remotesocket);
+            shutdownAndCloseNoThrow(m_client);
         }
     }
     catch(IException *excpt)

esp/tools/soapplus/http.cpp

@@ -1826,8 +1826,7 @@ int HttpClient::sendRequest(StringBuffer& req, IFileIO* request_output, IFileIO*
 
         if(!isPersist || iter >= numReq - 1)
         {
-            socket->shutdown();
-            socket->close();
+            shutdownAndCloseNoThrow(socket);
         }
 
         unsigned end1 = msTick();

fs/dafsclient/rmtclient.cpp

@@ -56,24 +56,7 @@ static ISocket *dummyReadWrite::timeoutreadsock = null; // used to trigger
 
 void cleanupDaFsSocket(ISocket *sock)
 {
-    if (!sock)
-        return;
-    try
-    {
-        sock->shutdown();
-    }
-    catch (IException *e)
-    {
-        e->Release();
-    }
-    try
-    {
-        sock->close();
-    }
-    catch (IException *e)
-    {
-        e->Release();
-    }
+    shutdownAndCloseNoThrow(sock);
 }
 
 

roxie/ccd/ccdprotocol.cpp

@@ -274,24 +274,7 @@ class ProtocolSocketListener : public ProtocolListener
 
     virtual void cleanupSocket(ISocket *sock)
     {
-        if (!sock)
-            return;
-        try
-        {
-            sock->shutdown();
-        }
-        catch (IException *e)
-        {
-            e->Release();
-        }
-        try
-        {
-            sock->close();
-        }
-        catch (IException *e)
-        {
-            e->Release();
-        }
+        shutdownAndCloseNoThrow(sock);
     }
 
     virtual int run()

roxie/ccd/ccdqueue.cpp

@@ -2824,7 +2824,7 @@ class RoxieSocketQueueManager : public RoxieReceiverBase
         if (running)
         {
             running = false;
-            multicastSocket->close();
+            shutdownAndCloseNoThrow(multicastSocket);
         }
         RoxieReceiverBase::stop();
     }

roxie/udplib/udptrr.cpp

@@ -142,7 +142,7 @@ class CReceiveManager : implements IReceiveManager, public CInterface
         {
             if (flowSocket)
             {
-                flowSocket->close();
+                shutdownAndCloseNoThrow(flowSocket);
                 flowSocket->Release();
             }
         }
@@ -438,8 +438,7 @@ class CReceiveManager : implements IReceiveManager, public CInterface
         ~receive_receive_flow() 
         {
             running = false;
-            if (flow_socket) 
-                flow_socket->close();
+            shutdownAndCloseNoThrow(flow_socket);
             join();
         }
 
@@ -646,10 +645,8 @@ class CReceiveManager : implements IReceiveManager, public CInterface
             DBGLOG("Total data packets seen = %u OOO(%u) Requests(%u) Permits(%u)", dataPacketsReceived.load(), packetsOOO.load(), flowRequestsReceived.load(), flowRequestsSent.load());
 
             running = false;
-            if (receive_socket)
-                receive_socket->close();
-            if (selfFlowSocket)
-                selfFlowSocket->close();
+            shutdownAndCloseNoThrow(receive_socket);
+            shutdownAndCloseNoThrow(selfFlowSocket);
             join();
             ::Release(receive_socket);
             ::Release(selfFlowSocket);

roxie/udplib/udptrs.cpp

@@ -792,8 +792,7 @@ class CSendManager : implements ISendManager, public CInterface
         ~send_receive_flow() 
         {
             running = false;
-            if (flow_socket)
-                flow_socket->shutdownNoThrow();
+            shutdownAndCloseNoThrow(flow_socket);
             join();
         }
 

system/hrpc/hrpcsock.cpp

@@ -369,7 +369,7 @@ class HRPCsockettransport: implements IHRPCtransport, public CInterface
         CriticalBlock critblock(critsect);
         if (datasock ) {
             try {
-                datasock->shutdown();
+                datasock->shutdownNoThrow();
                 datasock->close();
             }
             catch (IJSOCK_Exception *e)

system/jlib/jsocket.cpp

@@ -5545,15 +5545,7 @@ CSingletonSocketConnection::CSingletonSocketConnection(SocketEndpoint &_ep)
 
 CSingletonSocketConnection::~CSingletonSocketConnection()
 {
-    try {
-        if (sock)
-            sock->close();
-    }
-    catch (IException *e) {
-        if (e->errorCode()!=JSOCKERR_graceful_close)
-            EXCLOG(e,"CSingletonSocketConnection close");
-        e->Release();
-    }
+    shutdownAndCloseNoThrow(sock);
 }
 
 void CSingletonSocketConnection::set_keep_alive(bool keepalive)
@@ -7106,5 +7098,21 @@ IAllowListHandler *createAllowListHandler(AllowListPopulateFunction populateFunc
     return new CAllowListHandler(populateFunc, roleFormatFunc);
 }
 
+extern jlib_decl void shutdownAndCloseNoThrow(ISocket * optSocket)
+{
+    if (!optSocket)
+        return;
+
+    optSocket->shutdownNoThrow();
+    try
+    {
+        optSocket->close();
+    }
+    catch (IException * e)
+    {
+        e->Release();
+    }
+}
+
 static_assert(sizeof(IpAddress) == 16, "check size of IpAddress");
 static_assert(sizeof(SocketEndpoint) == 20, "check size of SocketEndpoint");

system/jlib/jsocket.hpp

@@ -730,5 +730,7 @@ class jlib_decl CSingletonSocketConnection: implements IConversation, public CIn
     virtual void cancel();
 };
 
+extern jlib_decl void shutdownAndCloseNoThrow(ISocket * optSocket);     // Safely shutdown and close a socket without throwing an exception.
+
 #endif
 

system/mp/mpcomm.cpp

@@ -2081,7 +2081,7 @@ void CMPConnectThread::checkSelfDestruct(void *p,size32_t sz)
     PROGLOG("MP Self destruct invoked");
     try {
         if (listensock) {
-            listensock->close();
+            shutdownAndCloseNoThrow(listensock);
             listensock->Release();
             listensock=NULL;
         }

system/security/securesocket/securesocket.cpp

@@ -2059,15 +2059,7 @@ class CSingletonSecureSocketConnection: public CSingletonSocketConnection
 
     virtual ~CSingletonSecureSocketConnection()
     {
-        try {
-            if (sock)
-                sock->close();
-        }
-        catch (IException *e) {
-            if (e->errorCode()!=JSOCKERR_graceful_close)
-                EXCLOG(e,"CSingletonSocketConnection close");
-            e->Release();
-        }
+        shutdownAndCloseNoThrow(sock);
     }
 
     bool connect(unsigned timeoutms) override

thorlcr/msort/tsorts1.cpp

@@ -420,17 +420,7 @@ protected: friend class CSortMerge;
             }
             e->Release();
         }
-        try {
-            server->close();
-        }
-        catch (IJSOCK_Exception *e)
-        {
-            if (e->errorCode()!=JSOCKERR_cancel_accept) {
-                PrintExceptionLog(e,"CSortTransferServerThread closing");
-                // Ignore for now
-            }
-            e->Release();
-        }
+        shutdownAndCloseNoThrow(server);
         subjoin();
         DBGLOG("CSortTransferServerThread finished");
         return 0;