using the same logic for sendAll and sendAllTo

Network/Socket/ByteString/IO.hsc

@@ -126,10 +126,13 @@ sendAllTo :: SocketAddress sa =>
           -> sa    -- ^ Recipient address
           -> IO ()
 sendAllTo _ "" _  = return ()
-sendAllTo s xs sa = do
-    sent <- sendTo s xs sa
-    waitWhen0 sent s
-    when (sent >= 0) $ sendAllTo s (B.drop sent xs) sa
+sendAllTo s bs0 sa = loop bs0
+  where
+    loop bs = do
+        -- "send" throws an exception.
+        sent <- sendTo s bs sa
+        waitWhen0 sent s
+        when (sent /= B.length bs) $ loop $ B.drop sent bs
 
 -- | Send data to the socket.  The socket must be in a connected
 -- state.  The data is sent as if the parts have been concatenated.

Network/Socket/ByteString/Lazy/Posix.hs

@@ -51,8 +51,10 @@ sendAll
     -> L.ByteString -- ^ Data to send
     -> IO ()
 sendAll _ "" = return ()
-sendAll s bs = do
-    -- "send" throws an exception.
-    sent <- send s bs
-    waitWhen0 (fromIntegral sent) s
-    when (sent /= L.length bs) $ sendAll s $ L.drop sent bs
+sendAll s bs0 = loop bs0
+  where
+    loop bs = do
+        -- "send" throws an exception.
+        sent <- send s bs
+        waitWhen0 (fromIntegral sent) s
+        when (sent /= L.length bs) $ loop $ L.drop sent bs

Network/Socket/ByteString/Lazy/Windows.hs

@@ -29,8 +29,10 @@ sendAll
     -> L.ByteString -- ^ Data to send
     -> IO ()
 sendAll _ "" = return ()
-sendAll s bs = do
-    -- "send" throws an exception.
-    sent <- send s bs
-    waitWhen0 (fromIntegral sent) s
-    when (sent /= L.length bs) $ sendAll s $ L.drop sent bs
+sendAll s bs0 = loop bs0
+  where
+    loop bs = do
+        -- "send" throws an exception.
+        sent <- send s bs
+        waitWhen0 (fromIntegral sent) s
+        when (sent /= L.length bs) $ loop $ L.drop sent bs