Use GC-aware allocation instead of malloc(3) where appropriate.

1 files changed, 10 insertions, 0 deletions

diff --git a/MLKForm.m b/MLKForm.m
index 674181b..c7ee3f5 100644
--- a/MLKForm.m
+++ b/MLKForm.m
@@ -26,6 +26,10 @@
 #import <Foundation/NSSet.h>
 #import <Foundation/NSString.h>
 
+#ifdef __OBJC_GC__
+#import <Foundation/NSZone.h>
+#endif
+
 #include <stdlib.h>
 
 #define MAKE_FORM(OBJECT)            \
@@ -489,7 +493,13 @@
   argtypes = [[[_tail cdr] cdr] car];
 
   _argc = [argtypes length];
+
+#ifdef __OBJC_GC__
+  _argumentTypes = NSAllocateCollectable (_argc * sizeof (MLKForeignType), NSScannedOption);
+#else
   _argumentTypes = malloc (_argc * sizeof (MLKForeignType));
+#endif
+
   while (argtypes)
     {
       _argumentTypes[i] = MLKForeignTypeWithTypeDesignator ([argtypes car]);