connect & accept: change the scope of masked exceptions

Limit the scope of masked exceptions only to the 'Network.Socket.connect' / 'Network.Socket.accept' call.
IntersectMBO · Feb 14, 2020 · 3806a47 · 3806a47
1 parent 880e629
commit 3806a47
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Win32-network/src/System/Win32/Async/Socket.hs b/Win32-network/src/System/Win32/Async/Socket.hs
@@ -54,7 +54,7 @@ connect :: Socket -> SockAddr -> IO ()
 connect sock addr = do
     v <- newEmptyMVar
     _ <- mask_ $ forkIOWithUnmask $ \unmask -> 
-        unmask (Socket.connect sock addr >> putMVar v Nothing)
+        (unmask (Socket.connect sock addr) >> putMVar v Nothing)
         `catch` (\(e :: IOException) -> putMVar v (Just e))
     r <- takeMVar v
     case r of
@@ -76,7 +76,7 @@ accept :: Socket -> IO (Socket, SockAddr)
 accept sock = do
     v <- newEmptyMVar
     _ <- mask_ $ forkIOWithUnmask $ \unmask -> 
-          unmask (Socket.accept sock >>= putMVar v . Right)
+          (unmask (Socket.accept sock) >>= putMVar v . Right)
           `catch` (\(e :: IOException) -> putMVar v (Left e))
     r <- takeMVar v
     case r of