rts/linker/MachO: Introduce finalizer support

3 files changed, 32 insertions, 0 deletions

diff --git a/rts/Linker.c b/rts/Linker.c
index 7bbeb49f9c..e3bcd5d795 100644
--- a/rts/Linker.c
+++ b/rts/Linker.c
@@ -1145,6 +1145,8 @@ void freeObjectCode (ObjectCode *oc)
         ocRunFini_ELF(oc);
 #elif defined(OBJFORMAT_PEi386)
         ocRunFini_PEi386(oc);
+#elif defined(OBJFORMAT_MACHO)
+        ocRunFini_MachO(oc);
 #endif
     }
 
diff --git a/rts/linker/MachO.c b/rts/linker/MachO.c
index 3145e40e03..a9b5f0c712 100644
--- a/rts/linker/MachO.c
+++ b/rts/linker/MachO.c
@@ -1641,6 +1641,35 @@ ocRunInit_MachO ( ObjectCode *oc )
     return 1;
 }
 
+int
+ocRunFini_MachO ( ObjectCode *oc )
+{
+    if (NULL == oc->info->segCmd) {
+        barf("ocRunInit_MachO: no segment load command");
+    }
+
+    for (int i = 0; i < oc->n_sections; i++) {
+        IF_DEBUG(linker, debugBelch("ocRunFini_MachO: checking section %d\n", i));
+
+        if (oc->sections[i].kind == SECTIONKIND_FINI_ARRAY) {
+            IF_DEBUG(linker, debugBelch("ocRunFini_MachO:     running mod fini functions\n"));
+
+            void *fini_startC = oc->sections[i].start;
+            fini_t *fini = (fini_t*)fini_startC;
+            fini_t *fini_end = (fini_t*)((uint8_t*)fini_startC
+                             + oc->sections[i].info->macho_section->size);
+
+            for (int pn = 0; fini < fini_end; fini++, pn++) {
+                IF_DEBUG(linker, debugBelch("ocRunFini_MachO:     function pointer %d at %p to %p\n",
+                                            pn, (void *) fini, (void *) *fini));
+                (*fini)();
+            }
+        }
+    }
+
+    return 1;
+}
+
 /*
  * Figure out by how much to shift the entire Mach-O file in memory
  * when loading so that its single segment ends up 16-byte-aligned
diff --git a/rts/linker/MachO.h b/rts/linker/MachO.h
index f55f68ddc8..4af3532b8d 100644
--- a/rts/linker/MachO.h
+++ b/rts/linker/MachO.h
@@ -12,6 +12,7 @@ int    ocBuildSegments_MachO        ( ObjectCode* oc );
 int    ocGetNames_MachO             ( ObjectCode* oc );
 int    ocResolve_MachO              ( ObjectCode* oc );
 int    ocRunInit_MachO              ( ObjectCode* oc );
+int    ocRunFini_MachO              ( ObjectCode* oc );
 int    machoGetMisalignment         ( FILE * );
 int    ocAllocateExtras_MachO       ( ObjectCode* oc );