add custom allocator test

Author: ZERICO2005

test/standalone/custom_allocator/autotest.json

@@ -0,0 +1,61 @@
+{
+  "transfer_files": [
+    "bin/DEMO.8xp"
+  ],
+  "target": {
+    "name": "DEMO",
+    "isASM": true
+  },
+  "sequence": [
+    "action|launch",
+    "delay|1000",
+    "hashWait|1",
+    "key|enter",
+    "delay|400",
+    "hashWait|2",
+    "key|enter",
+    "delay|400",
+    "hashWait|3",
+    "key|enter",
+    "delay|400",
+    "hashWait|4"
+  ],
+  "hashes": {
+    "1": {
+      "description": "calloc should clear the memory from custom malloc",
+      "start": "vram_start",
+      "size": "vram_16_size",
+      "expected_CRCs": [
+        "8CC9FD0C"
+      ]
+    },
+    "2": {
+      "description": "we expect malloc to be called twice (for calloc and atexit)",
+      "start": "vram_start",
+      "size": "vram_16_size",
+      "expected_CRCs": [
+        "C1BC5A5C"
+      ]
+    },
+    "3": {
+      "description": "custom free should print a custom message",
+      "start": "vram_start",
+      "size": "vram_16_size",
+      "expected_CRCs": [
+        "BD82EEA3"
+      ]
+    },
+    "4": {
+      "description": "Exit",
+      "start": "vram_start",
+      "size": "vram_16_size",
+      "expected_CRCs": [
+        "FFAF89BA",
+        "101734A5",
+        "9DA19F44",
+        "A32840C8",
+        "349F4775"
+      ]
+    }
+  }
+}

test/standalone/custom_allocator/makefile

@@ -0,0 +1,21 @@
+# ----------------------------
+# Makefile Options
+# ----------------------------
+
+NAME = DEMO
+ICON = icon.png
+DESCRIPTION = "CE C Toolchain Demo"
+COMPRESSED = NO
+ARCHIVED = NO
+
+CFLAGS = -Wall -Wextra -Wshadow -Oz
+CXXFLAGS = -Wall -Wextra -Wshadow -Oz
+
+PREFER_OS_LIBC = NO
+PREFER_OS_CRT = NO
+HAS_PRINTF = NO
+ALLOCATOR = CUSTOM
+
+# ----------------------------
+
+include $(shell cedev-config --makefile)

test/standalone/custom_allocator/src/custom_free.s

@@ -0,0 +1,23 @@
+	.assume	adl=1
+
+	.section	.rodata._free_string
+	.local	_free_string
+_free_string:
+	.asciz	"called free"
+
+	.section	.text._free
+	.global	__custom_free
+	.type	__custom_free, @function
+__custom_free:
+	ld	hl, _free_string
+	push	hl
+	call	_puts
+	pop	hl
+.L.wait_loop:
+	call	_os_GetCSC
+	or	a, a
+	jr	z, .L.wait_loop
+	ret
+
+	.extern	_puts
+	.extern	_os_GetCSC

test/standalone/custom_allocator/src/custom_malloc.c

@@ -0,0 +1,21 @@
+#include <string.h>
+#include <stdlib.h>
+
+static unsigned char arr[4096];
+
+static unsigned char *arr_ptr = &arr[0];
+
+int malloc_calls = 0;
+
+void *_custom_malloc(size_t size) __attribute__((__malloc__));
+
+void *_custom_malloc(size_t size) {
+    if (size >= sizeof(arr)) {
+        return NULL;
+    }
+    unsigned char *ret = arr_ptr;
+    arr_ptr += size;
+    memset(ret, 0xAA, size);
+    malloc_calls++;
+    return ret;
+}

test/standalone/custom_allocator/src/main.c

@@ -0,0 +1,62 @@
+#include <ti/screen.h>
+#include <ti/getcsc.h>
+#include <ti/sprintf.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+extern int malloc_calls;
+
+void atexit_func(void) {
+    /*
+     * The custom free function is supposed to print a string if it is called,
+     * however, it is unspecified if atexit_func or free will be called first.
+     * This means that we cannot print anything in this function since we
+     * cannot guarantee the order. However, this function must be non-empty so
+     * Clang does not optimize away the call to atexit. Now some of this could
+     * be fixed with -ffreestanding, but that may also hide the assumptions
+     * that Clang makes.
+     */
+    if (*(const char*)-1 != 0) {
+        puts("0xFFFFFF should be zero");
+        while (!os_GetCSC());
+    }
+}
+
+int main(void) {
+    char buf[50];
+    os_ClrHome();
+
+    // CRT0 may use malloc/free so we cannot assume this is zero
+    int malloc_calls_before = malloc_calls;
+
+    // calloc should call the custom malloc
+    int *ptr = calloc(1, sizeof(int[3]));
+
+    if (ptr == NULL) {
+        puts("calloc failed");
+        while (!os_GetCSC());
+        return 0;
+    }
+
+    for (int n = 0; n < 3; ++n) {
+        boot_sprintf(buf, "ptr(%d) == %d", n, ptr[n]);
+        puts(buf);
+    }
+
+    while (!os_GetCSC());
+
+    if (atexit(atexit_func) != 0) {
+        puts("atexit(atexit_func) failed");
+        while (!os_GetCSC());
+        return 0;
+    }
+
+    // we expect malloc to be called twice (for calloc and atexit)
+    int malloc_call_count = malloc_calls - malloc_calls_before;
+    boot_sprintf(buf, "malloc calls: %d", malloc_call_count);
+    puts(buf);
+
+    while (!os_GetCSC());
+
+    exit(EXIT_SUCCESS);
+}