From 2a4ea609491b225a1ceb06da70396e93916f137a Mon Sep 17 00:00:00 2001
From: Harald Steinlechner <haraldsteinlechner@gmail.com>
Date: Thu, 23 Aug 2012 15:41:42 +0200
Subject: [PATCH] fixed linker problems (currently not portable due to absolute
 paths etc); imported some gc functions - refactoring,cleanup required ;-)

---
 hs-boehmgc.cabal     |  4 ++++
 src/Mate/GC/Boehm.hs | 34 +++++++++++++++++++++++++++++++---
 2 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/hs-boehmgc.cabal b/hs-boehmgc.cabal
index 1c4d820..dd24e6e 100644
--- a/hs-boehmgc.cabal
+++ b/hs-boehmgc.cabal
@@ -13,3 +13,7 @@ library
   build-depends:       base
   exposed-modules:     Mate.GC.Boehm
   hs-source-dirs:      src
+  extra-lib-dirs:      /home/hs/Mate/boehmgc/gc/lib/
+  extra-libraries:     gc
+  include-dirs:        /home/hs/Mate/boehmgc/gc/include
+  ld-options: -lgcblub
diff --git a/src/Mate/GC/Boehm.hs b/src/Mate/GC/Boehm.hs
index eb5711b..1e3ff5c 100644
--- a/src/Mate/GC/Boehm.hs
+++ b/src/Mate/GC/Boehm.hs
@@ -1,8 +1,11 @@
+{-# LANGUAGE ForeignFunctionInterface #-}
 {-# LANGUAGE CPP #-}
 module Mate.GC.Boehm (
      mallocBytesGC,
      unsafeFreeGC,
      getHeapSize,
+     initGC,
+     addRootGC,
   ) where
 
 #define DEBUG
@@ -11,6 +14,22 @@ import Foreign.Marshal.Alloc
 import Foreign.Ptr
 import Data.IORef
 import System.IO.Unsafe
+import Foreign.C.Types
+import Control.Monad
+
+foreign import ccall "gc/gc.h GC_malloc"
+   mallocGC :: CSize -> IO (Ptr a) 
+
+foreign import ccall "gc/gc.h GC_get_heap_size"
+   heapSizeGC :: IO CSize
+
+
+foreign import ccall "gc/gc.h GC_init"
+   initGC :: IO ()
+
+
+foreign import ccall "gc/gc.h GC_add_roots"
+   addRootGCInternal :: Ptr a -> Ptr a -> IO ()
 
 -- counts allocated memory in heap
 {-# NOINLINE globalBytesAllocated #-}
@@ -20,11 +39,17 @@ globalBytesAllocated = unsafePerformIO (newIORef 0)
 #ifdef DEBUG
 -- |Allocates size bytes in managed memory
 mallocBytesGC :: Int -> IO (Ptr a)
-mallocBytesGC size = modifyIORef globalBytesAllocated (+ size) >> mallocBytes size
+mallocBytesGC size = do
+  --print "trying alloc"
+  ptr <- modifyIORef globalBytesAllocated (+ size) >> mallocGC (fromIntegral size)
+  if ptr == nullPtr 
+   then error "alloc 0"
+   else return ptr
+
 #else
 -- |Allocates size bytes in managed memory
 mallocBytesGC :: Int -> IO (Ptr a)
-mallocBytesGC = mallocBytes
+mallocBytesGC = mallocGC . fromIntegral
 #endif
 
 -- |Explicitely deallocate an object. Not required and dangerous.
@@ -33,4 +58,7 @@ unsafeFreeGC _ = print "not implemented"
 
 -- |Returns currently allocated memory in bytes
 getHeapSize :: IO Int
-getHeapSize = readIORef globalBytesAllocated
+getHeapSize = liftM fromIntegral heapSizeGC 
+
+addRootGC :: Ptr a -> Ptr a -> IO ()
+addRootGC = addRootGCInternal 
-- 
2.25.1