Fix extsample to cleanup all memory used for cnex CI heap debug testing

gitlarryf · Apr 11, 2024 · 9f90e14 · 9f90e14
1 parent 2e6b024
commit 9f90e14
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/lib/extsample/extsample.c b/lib/extsample/extsample.c
@@ -1,7 +1,11 @@
+#include <assert.h>
 #include <stdlib.h>
+#include <string.h>
+
 
 #include "neonext.h"
 
+const char *neon = "Neon!";
 const struct Ne_MethodTable *Ne;
 struct Ne_Cell *g_cb;
 
@@ -161,14 +165,22 @@ Ne_EXPORT int Ne_raiseException(struct Ne_Cell *retval, struct Ne_ParameterList
 
 Ne_EXPORT int Ne_allocHandle(struct Ne_Cell *retval, struct Ne_ParameterList *in_params, struct Ne_ParameterList *out_params)
 {
-    void *p = malloc(1);
+    void *p = malloc(sizeof(neon));
+    memcpy(p, neon, sizeof(neon));
     Ne->cell_set_pointer(retval, p);
     return Ne_SUCCESS;
 }
 
 Ne_EXPORT int Ne_freeHandle(struct Ne_Cell *retval, struct Ne_ParameterList *in_params, struct Ne_ParameterList *out_params)
 {
-    void *p = Ne->cell_get_pointer(Ne->parameterlist_get_cell(in_params, 0));
+    char *p = Ne->cell_get_pointer(Ne->parameterlist_get_cell(in_params, 0));
+    assert(memcmp(p, neon, sizeof(neon)) == 0);
     free(p);
     return Ne_SUCCESS;
 }
+
+Ne_EXPORT int Ne_shutdownExtension(struct Ne_Cell *retval, struct Ne_ParameterList *in_params, struct Ne_ParameterList *out_params)
+{
+    Ne->cell_free(g_cb);
+    return Ne_SUCCESS;
+}
diff --git a/lib/extsample/extsample.neon b/lib/extsample/extsample.neon
@@ -46,3 +46,5 @@ EXPORT DECLARE EXTENSION FUNCTION raiseException()
 
 EXPORT DECLARE EXTENSION FUNCTION allocHandle(): Handle
 EXPORT DECLARE EXTENSION FUNCTION freeHandle(h: Handle)
+
+EXPORT DECLARE EXTENSION FUNCTION shutdownExtension()
diff --git a/lib/extsample/t/extsample-test.neon b/lib/extsample/t/extsample-test.neon
@@ -53,3 +53,5 @@ END TRY
 LET h: extsample.Handle := extsample.allocHandle()
 TESTCASE h <> NIL
 extsample.freeHandle(h)
+
+extsample.shutdownExtension()