implementing socketToFd

haskell · Sep 4, 2019 · e7312bb · e7312bb
1 parent 685cec5
commit e7312bb
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/Network/Socket.hs b/Network/Socket.hs
@@ -139,6 +139,7 @@ module Network.Socket
     , withFdSocket
     , unsafeFdSocket
     , touchSocket
+    , socketToFd
     , fdSocket
     , mkSocket
     , socketToHandle

diff --git a/Network/Socket/Types.hsc b/Network/Socket/Types.hsc
@@ -14,6 +14,7 @@ module Network.Socket.Types (
     , withFdSocket
     , unsafeFdSocket
     , touchSocket
+    , socketToFd
     , fdSocket
     , mkSocket
     , invalidateSocket
@@ -162,6 +163,18 @@ withFdSocket (Socket ref _) f = do
   touch ref
   return r
 
+-- | Socket is closed and a duplicated file descriptor is returned.
+socketToFd :: Socket -> IO CInt
+socketToFd s = do
+    fd <- unsafeFdSocket s
+    -- FIXME: throw error no if -1
+    fd2 <- c_dup fd
+    close s
+    return fd2
+
+foreign import ccall unsafe "dup"
+   c_dup :: CInt -> IO CInt
+
 -- | Creating a socket from a file descriptor.
 mkSocket :: CInt -> IO Socket
 mkSocket fd = do