test: add custom memory management hooks and related tests

Author: sfxfs

test/mem_test.c

@@ -4,15 +4,76 @@
 #include <stdlib.h>
 #include <string.h>
 
-void setUp(void) {}
-void tearDown(void) {}
+// Memory tracking for hook tests
+static size_t custom_malloc_count = 0;
+static size_t custom_free_count = 0;
+static size_t custom_strdup_count = 0;
+
+// Custom memory functions for testing hooks
+static void* test_malloc(size_t size)
+{
+    custom_malloc_count++;
+    return malloc(size);
+}
+
+static void test_free(void* ptr)
+{
+    if (ptr != NULL) {
+        custom_free_count++;
+    }
+    free(ptr);
+}
+
+static char* test_strdup(const char* str)
+{
+    if (str == NULL) {
+        return NULL;
+    }
+    custom_strdup_count++;
+    size_t len = strlen(str) + 1;
+    char* dup = test_malloc(len);
+    if (dup != NULL) {
+        memcpy(dup, str, len);
+    }
+    return dup;
+}
+
+void setUp(void) {
+    // Reset memory counters before each test
+    custom_malloc_count = 0;
+    custom_free_count = 0;
+    custom_strdup_count = 0;
+    // Reset to default memory functions
+    mjrpc_set_memory_hooks(NULL, NULL, NULL);
+}
+
+void tearDown(void) {
+    // Ensure we reset to default memory functions after each test
+    mjrpc_set_memory_hooks(NULL, NULL, NULL);
+}
 
 // 用于测试自动扩容的空方法
 static cJSON* dummy_func(mjrpc_func_ctx_t* ctx, cJSON* params, cJSON* id)
 {
     return cJSON_CreateString("ok");
 }
 
+// 用于测试内存hooks的方法
+static cJSON* memory_test_func(mjrpc_func_ctx_t* ctx, cJSON* params, cJSON* id)
+{
+    (void)ctx;
+    (void)id;
+    
+    // 创建一个包含字符串的响应，这会触发内存分配
+    cJSON* result = cJSON_CreateObject();
+    cJSON_AddStringToObject(result, "message", "memory test successful");
+    cJSON_AddNumberToObject(result, "malloc_count", custom_malloc_count);
+    cJSON_AddNumberToObject(result, "free_count", custom_free_count);
+    cJSON_AddNumberToObject(result, "strdup_count", custom_strdup_count);
+    
+    return result;
+}
+
 void test_auto_resize(void)
 {
     size_t initial_capacity = 4;
@@ -43,9 +104,141 @@ void test_auto_resize(void)
     mjrpc_destroy_handle(h);
 }
 
+void test_memory_hooks_set_and_reset(void)
+{
+    // 测试设置自定义内存hooks
+    int ret = mjrpc_set_memory_hooks(test_malloc, test_free, test_strdup);
+    TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, ret);
+    
+    // 测试重置为默认内存函数
+    ret = mjrpc_set_memory_hooks(NULL, NULL, NULL);
+    TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, ret);
+}
+
+void test_memory_hooks_invalid_params(void)
+{
+    // 测试无效参数组合 - 只设置部分函数应该失败
+    int ret = mjrpc_set_memory_hooks(test_malloc, NULL, NULL);
+    TEST_ASSERT_NOT_EQUAL(MJRPC_RET_OK, ret);
+    
+    ret = mjrpc_set_memory_hooks(NULL, test_free, NULL);
+    TEST_ASSERT_NOT_EQUAL(MJRPC_RET_OK, ret);
+    
+    ret = mjrpc_set_memory_hooks(NULL, NULL, test_strdup);
+    TEST_ASSERT_NOT_EQUAL(MJRPC_RET_OK, ret);
+    
+    ret = mjrpc_set_memory_hooks(test_malloc, test_free, NULL);
+    TEST_ASSERT_NOT_EQUAL(MJRPC_RET_OK, ret);
+}
+
+void test_memory_hooks_functionality(void)
+{
+    // 设置自定义内存hooks
+    int ret = mjrpc_set_memory_hooks(test_malloc, test_free, test_strdup);
+    TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, ret);
+    
+    // 创建handle - 这应该会触发内存分配
+    size_t malloc_before = custom_malloc_count;
+    size_t strdup_before = custom_strdup_count;
+    
+    mjrpc_handle_t* h = mjrpc_create_handle(4);
+    TEST_ASSERT_NOT_NULL(h);
+    
+    // 验证创建handle时使用了自定义内存函数
+    TEST_ASSERT_GREATER_THAN(malloc_before, custom_malloc_count);
+    
+    // 添加方法 - 这应该会触发strdup
+    ret = mjrpc_add_method(h, memory_test_func, "memory_test", NULL);
+    TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, ret);
+    TEST_ASSERT_GREATER_THAN(strdup_before, custom_strdup_count);
+    
+    // 处理请求 - 验证hooks正常工作
+    cJSON* id = cJSON_CreateNumber(1);
+    cJSON* req = mjrpc_request_cjson("memory_test", NULL, id);
+    TEST_ASSERT_NOT_NULL(req);
+    
+    int code = -1;
+    cJSON* resp = mjrpc_process_cjson(h, req, &code);
+    TEST_ASSERT_NOT_NULL(resp);
+    TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, code);
+    
+    // 验证响应内容
+    cJSON* result = cJSON_GetObjectItem(resp, "result");
+    TEST_ASSERT_NOT_NULL(result);
+    TEST_ASSERT_TRUE(cJSON_IsObject(result));
+    
+    cJSON* message = cJSON_GetObjectItem(result, "message");
+    TEST_ASSERT_TRUE(cJSON_IsString(message));
+    TEST_ASSERT_EQUAL_STRING("memory test successful", message->valuestring);
+    
+    cJSON_Delete(resp);
+    
+    // 记录销毁前的free计数
+    size_t free_before = custom_free_count;
+    
+    // 销毁handle - 这应该会触发内存释放
+    mjrpc_destroy_handle(h);
+    
+    // 验证销毁时使用了自定义free函数
+    TEST_ASSERT_GREATER_THAN(free_before, custom_free_count);
+}
+
+void test_memory_hooks_multiple_operations(void)
+{
+    // 设置自定义内存hooks
+    int ret = mjrpc_set_memory_hooks(test_malloc, test_free, test_strdup);
+    TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, ret);
+    
+    // 重置计数器
+    custom_malloc_count = 0;
+    custom_free_count = 0; 
+    custom_strdup_count = 0;
+    
+    mjrpc_handle_t* h = mjrpc_create_handle(2);
+    TEST_ASSERT_NOT_NULL(h);
+    
+    // 记录添加方法前的strdup计数
+    size_t strdup_before = custom_strdup_count;
+    
+    // 添加多个方法来测试strdup的使用
+    char method_names[3][20] = {"test1", "test2", "test3"};
+    for (int i = 0; i < 3; i++) {
+        ret = mjrpc_add_method(h, dummy_func, method_names[i], NULL);
+        TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, ret);
+    }
+    
+    // 验证至少有3次strdup调用（可能因为内部实现有额外的strdup调用）
+    TEST_ASSERT_GREATER_OR_EQUAL_size_t(3, custom_strdup_count - strdup_before);
+    
+    // 处理多个请求
+    for (int i = 0; i < 3; i++) {
+        cJSON* id = cJSON_CreateNumber(i);
+        cJSON* req = mjrpc_request_cjson(method_names[i], NULL, id);
+        
+        int code = -1;
+        cJSON* resp = mjrpc_process_cjson(h, req, &code);
+        TEST_ASSERT_NOT_NULL(resp);
+        TEST_ASSERT_EQUAL_INT(MJRPC_RET_OK, code);
+        
+        cJSON_Delete(resp);
+    }
+    
+    // 验证malloc和free被调用
+    TEST_ASSERT_GREATER_THAN_size_t(0, custom_malloc_count);
+    
+    mjrpc_destroy_handle(h);
+    
+    // 验证free被调用用于清理
+    TEST_ASSERT_GREATER_THAN_size_t(0, custom_free_count);
+}
+
 int main(void)
 {
     UNITY_BEGIN();
     RUN_TEST(test_auto_resize);
+    RUN_TEST(test_memory_hooks_set_and_reset);
+    RUN_TEST(test_memory_hooks_invalid_params);
+    RUN_TEST(test_memory_hooks_functionality);
+    RUN_TEST(test_memory_hooks_multiple_operations);
     return UNITY_END();
 }