MySQL UDF C函数内存异常：多记录查询输出结果错误

问题场景 #

这段自定义MySQL UDF代码在处理多行查询时出现输出异常，原本应为ringo的结果被显示为ringoe。

测试代码 #

my_bool helloworld_init(UDF_INIT *initid, UDF_ARGS *args, char *message) {
    if (args->arg_count != 1 || args->arg_type[0] != STRING_RESULT) {
        strcpy(message, "helloworld() requires one string argument");
        return 1;
    }
    return 0;
}

char *helloworld(UDF_INIT *initid, UDF_ARGS *args, char *result, unsigned long *length, char *is_null, char *error) {
    char *str = args->args[0];
    size_t len = strlen(str);
    char *ptr = malloc(len + 1);
    if (ptr == NULL) {
        strcpy(error, "helloworld: failed to allocate memory");
        return NULL;
    }
    for (int i = 0; i < len; i++) {
        ptr[i] = str[i];
    }
    ptr[len] = '\0';
    *length = len;
    return ptr;
}

void helloworld_deinit(UDF_INIT *initid) {
    free(initid->ptr);
}

查询结果 #

MariaDB [test]> select helloworld(name) from users;
+------------------+
| helloworld(name) |
+------------------+
| john             |
| paul             |
| george           |
| ringoe           |
+------------------+
4 rows in set (0.000 sec)

疑问 #

为何会出现这种异常？尝试多种内存清理方式后问题仍未解决，该如何修复？

问题原因 #

1. 内存管理完全错误：每次调用helloworld都用malloc分配新内存，但helloworld_deinit仅在UDF生命周期结束时执行一次，且代码从未给initid->ptr赋值，导致每次分配的内存都无法释放，同时deinit中的free操作是无效的野指针操作。
2. 残留内存未被覆盖：MySQL会复用UDF调用的内存缓冲区，当新结果的长度短于之前的结果时，旧缓冲区中未被覆盖的字符会残留下来，最终拼接出错误字符串（比如ringo长度为5，若之前的结果长度为6，就会多出一个e）。

修复方案 #

MySQL UDF的内存管理有明确规范，以下是两种可行的修复方式：

方式一：让MySQL自动管理动态内存 #

将动态分配的指针赋值给initid->ptr，MySQL会在每次UDF调用后自动释放该内存：

my_bool helloworld_init(UDF_INIT *initid, UDF_ARGS *args, char *message) {
    if (args->arg_count != 1 || args->arg_type[0] != STRING_RESULT) {
        strcpy(message, "helloworld() requires one string argument");
        return 1;
    }
    initid->ptr = NULL; // 初始化指针，避免野指针
    return 0;
}

char *helloworld(UDF_INIT *initid, UDF_ARGS *args, char *result, unsigned long *length, char *is_null, char *error) {
    // 处理NULL输入
    if (args->args[0] == NULL) {
        *is_null = 1;
        return NULL;
    }

    char *str = args->args[0];
    size_t len = strlen(str);
    
    // 释放上一次调用分配的内存
    if (initid->ptr != NULL) {
        free(initid->ptr);
        initid->ptr = NULL;
    }

    char *ptr = malloc(len + 1);
    if (ptr == NULL) {
        strcpy(error, "helloworld: failed to allocate memory");
        return NULL;
    }

    strcpy(ptr, str); // 替代手动循环，更简洁可靠
    *length = len;
    initid->ptr = ptr; // 将指针交给MySQL管理
    return ptr;
}

void helloworld_deinit(UDF_INIT *initid) {
    // 兜底释放，避免极端情况下的内存泄漏
    if (initid->ptr != NULL) {
        free(initid->ptr);
        initid->ptr = NULL;
    }
}

方式二：复用MySQL提供的缓冲区 #

如果字符串长度在合理范围内，直接使用MySQL传入的result缓冲区，无需手动分配内存：

my_bool helloworld_init(UDF_INIT *initid, UDF_ARGS *args, char *message) {
    if (args->arg_count != 1 || args->arg_type[0] != STRING_RESULT) {
        strcpy(message, "helloworld() requires one string argument");
        return 1;
    }
    // 设置最大缓冲区长度，适配输入字符串的最大可能长度
    initid->max_length = 255; // 根据实际业务调整
    return 0;
}

char *helloworld(UDF_INIT *initid, UDF_ARGS *args, char *result, unsigned long *length, char *is_null, char *error) {
    if (args->args[0] == NULL) {
        *is_null = 1;
        return NULL;
    }

    char *str = args->args[0];
    size_t len = strlen(str);
    
    // 安全拷贝，避免缓冲区溢出
    strncpy(result, str, len);
    result[len] = '\0';
    *length = len;
    return result;
}

void helloworld_deinit(UDF_INIT *initid) {
    // 无需手动释放，缓冲区由MySQL管理
}

额外优化建议 #

• 用strcpy/strncpy替代手动字符循环，减少出错概率；
• 在init函数中设置initid->max_length，提前告知MySQL所需的缓冲区大小；
• 增加NULL输入的处理逻辑，避免空指针访问。

内容的提问来源于stack exchange，提问作者Riccardo Vacirca