Make protect safer against exceptions in a Pipe value.

Restore commit 7bb846e, it went lost during refactoring.
pcapriotti · Jul 8, 2012 · fc467df · fc467df
1 parent 0152395
commit fc467df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Control/Pipe/Internal.hs b/Control/Pipe/Internal.hs
@@ -85,7 +85,7 @@ finallyP p w = go p
     go (Await k j h w') = Await (go . k) (go . j) (go . h) (w ++ w')
 
 protectP :: Monad m => Finalizer m -> Pipe m a b u r -> Pipe m a b u r
-protectP w = go
+protectP w p = M Masked (return $! go p) (`Throw` w)
   where
     go (Pure r w') = Pure r (w ++ w')
     go (Await k h j w') = Await k h j w'