X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=rts%2FWeak.c;fp=rts%2FWeak.c;h=a50a72e59cd4bf56152ea6e766b6e6c7a7d201e1;hb=6c095bfa3c8c81b52ad92853acd326453d320d7b;hp=e97ff57e9a7806b7d71ef912800eec9578685392;hpb=d4a17c3a253d02c2ebf2315e71a29cb740278977;p=ghc-hetmet.git

diff --git a/rts/Weak.c b/rts/Weak.c
index e97ff57..a50a72e 100644
--- a/rts/Weak.c
+++ b/rts/Weak.c
@@ -17,8 +17,38 @@
 #include "RtsAPI.h"
 #include "Trace.h"
 
+// ForeignPtrs with C finalizers rely on weak pointers inside weak_ptr_list
+// to always be in the same order.
+
 StgWeak *weak_ptr_list;
 
+void
+runCFinalizer(StgVoid *fn, StgVoid *ptr, StgVoid *env, StgWord flag)
+{
+    if (flag)
+	((void (*)(void *, void *))fn)(env, ptr);
+    else
+	((void (*)(void *))fn)(ptr);
+}
+
+void
+runAllCFinalizers(StgWeak *list)
+{
+    StgWeak *w;
+
+    for (w = list; w; w = w->link) {
+	StgArrWords *farr;
+
+	farr = (StgArrWords *)UNTAG_CLOSURE(w->cfinalizer);
+
+	if ((StgClosure *)farr != &stg_NO_FINALIZER_closure)
+	    runCFinalizer((StgVoid *)farr->payload[0],
+	                  (StgVoid *)farr->payload[1],
+	                  (StgVoid *)farr->payload[2],
+	                  farr->payload[3]);
+    }
+}
+
 /*
  * scheduleFinalizers() is called on the list of weak pointers found
  * to be dead after a garbage collection.  It overwrites each object
@@ -45,6 +75,7 @@ scheduleFinalizers(Capability *cap, StgWeak *list)
     // count number of finalizers, and kill all the weak pointers first...
     n = 0;
     for (w = list; w; w = w->link) { 
+	StgArrWords *farr;
 
 	// Better not be a DEAD_WEAK at this stage; the garbage
 	// collector removes DEAD_WEAKs from the weak pointer list.
@@ -54,6 +85,14 @@ scheduleFinalizers(Capability *cap, StgWeak *list)
 	    n++;
 	}
 
+	farr = (StgArrWords *)UNTAG_CLOSURE(w->cfinalizer);
+
+	if ((StgClosure *)farr != &stg_NO_FINALIZER_closure)
+	    runCFinalizer((StgVoid *)farr->payload[0],
+	                  (StgVoid *)farr->payload[1],
+	                  (StgVoid *)farr->payload[2],
+	                  farr->payload[3]);
+
 #ifdef PROFILING
         // A weak pointer is inherently used, so we do not need to call
         // LDV_recordDead().