QueryResponse: Support the IO varient of fdSocket.

1 files changed, 5 insertions, 2 deletions

diff --git a/dht/src/Network/QueryResponse.hs b/dht/src/Network/QueryResponse.hs
index c7ab59d8..9c33b911 100644
--- a/dht/src/Network/QueryResponse.hs
+++ b/dht/src/Network/QueryResponse.hs
@@ -622,8 +622,11 @@ udpTransport' bind_address = do
     , closeTransport = do
         dput XMisc $ "closeTransport for udpTransport' called. " ++ show bind_address
         tryPutMVar isClosed () -- signal awaitMessage that the transport is closed.
-#if !MIN_VERSION_network (3,1,0)
-        let withFdSocket sock f = f (fdSocket sock)
+#if MIN_VERSION_network (3,1,0)
+#elif MIN_VERSION_network(3,0,0)
+        let withFdSocket sock f = fdSocket sock >>= f >>= seq sock . return
+#else
+        let withFdSocket sock f = f (fdSocket sock) >>= seq sock . return
 #endif
         withFdSocket sock $ \fd -> do
             let sorryGHCButIAmNotFuckingClosingTheSocketYet fd = return ()