Merge pull request #18 from Yuras/wip/yuras/mem_leak

ndmitchell · Sep 2, 2024 · e151873 · e151873
2 parents f6f4ce3 + 85259ba
commit e151873
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Text/XML/Hexml.hs b/src/Text/XML/Hexml.hs
@@ -11,6 +11,7 @@ module Text.XML.Hexml(
     ) where
 
 import Control.Applicative
+import Control.Exception
 import Control.Monad
 import Data.Int
 import Foreign.C
@@ -84,7 +85,7 @@ touchBS = touchForeignPtr . fst3 . BS.toForeignPtr
 parse :: BS.ByteString -> Either BS.ByteString Node
 parse src = do
     let src0 = src <> BS.singleton '\0'
-    unsafePerformIO $ BS.unsafeUseAsCStringLen src0 $ \(str, len) -> do
+    unsafePerformIO $ BS.unsafeUseAsCStringLen src0 $ \(str, len) -> mask_ $ do
         doc <- hexml_document_parse str (fromIntegral len - 1)
         err <- hexml_document_error doc
         if err /= nullPtr then do