Catch abrupt SSL connection termination

snapframework · Jan 31, 2012 · e9f7988 · e9f7988 · gregorycollins · Jan 31, 2012
1 parent 1b2dfd3
commit e9f7988
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/Snap/Internal/Http/Server/TLS.hs b/src/Snap/Internal/Http/Server/TLS.hs
@@ -175,10 +175,13 @@ send tickleTimeout _ (NetworkSession _ aSSL sz) bs = go bs
 
 ------------------------------------------------------------------------------
 recv :: IO b -> NetworkSession -> IO (Maybe ByteString)
-recv _ (NetworkSession _ aSSL recvLen) = do
+recv _ (NetworkSession _ aSSL recvLen) = handle termination $ do
     b <- SSL.read ssl recvLen
     return $! if S.null b then Nothing else Just b
   where
     ssl = unsafeCoerce aSSL
 
+    termination :: ConnectionAbruptlyTerminated -> IO (Maybe ByteString)
+    termination = const $ return Nothing
+
 #endif