MySQL C/C++扩展UDF调用system执行Python脚本失败求助

我按照MySQL官方文档创建用户自定义函数（UDF），期望实现my_udf通过system()调用执行Python脚本，但功能无法正常运行。已确认Python脚本本身无问题，推测问题出在system(cmd)调用环节。

我的代码如下：

#include <mysql/mysql.h>
#include <string>
#include <iostream>
#include <cstring>

extern "C" bool my_udf_init(UDF_INIT* initid, UDF_ARGS* args, char* message)
{
    if (args->arg_count != 1 || args->arg_type[0] != STRING_RESULT) {
        strcpy(message, "Expected a single string argument");
        return 1;
    }

    initid->const_item = 1;     // This UDF is deterministic
    initid->maybe_null = 1;     // This UDF can return NULL
    initid->max_length = 0;     // The result can be of any length

    return 0;
}

extern "C" char* my_udf(UDF_INIT* initid, UDF_ARGS* args, char* result,
                        unsigned long* length, char* is_null, char* error)
{
    // Run the Python script and pass the input string as a command-line parameter
    char param[] = "params to script";
    char cmd[100];
    sprintf(cmd, "python3 checker.py %s", param);
    system(cmd);

    return 0;
}

排查与修复方案 #

1. 路径与权限问题 #

• MySQL通常以mysql用户运行，该用户可能找不到python3或checker.py，必须使用绝对路径，比如/usr/bin/python3 /opt/scripts/checker.py，避免依赖环境变量。
• 确保checker.py及所在目录对mysql用户开放读、执行权限，可执行chown mysql:mysql /opt/scripts/checker.py和chmod 755 /opt/scripts/checker.py修复。

2. 参数处理与安全问题 #

• 当前代码硬编码param，未使用UDF传入的参数，应替换为args->args[0]；但直接拼接参数会引发命令注入风险，必须对特殊字符（空格、引号、$等）转义。
• 替换不安全的sprintf为snprintf，避免缓冲区溢出。

3. System调用的环境限制 #

• MySQL UDF运行在受限环境中，system()调用的环境变量与终端不同，可在命令前显式指定环境变量，比如PATH=/usr/bin:/bin /usr/bin/python3 ...。
• 捕获system()的返回值，判断执行状态：

  int ret = system(cmd);
  if (ret == -1) {
      *error = 1; // 标记执行错误
      return NULL;
  } else if (WIFEXITED(ret) && WEXITSTATUS(ret) != 0) {
      *error = 1; // 脚本返回非0错误码
      return NULL;
  }
  

4. UDF返回值合规性 #

当前代码返回0违反UDF规范，若无需返回值，需标记结果为NULL：

*is_null = 1; // 标记返回NULL
return NULL;

若需要返回脚本输出，需捕获stdout内容并写入result缓冲区（注意result的长度限制）。

5. 编译与加载验证 #

• 编译时确保链接正确：

  gcc -fPIC -shared -o my_udf.so my_udf.c $(mysql_config --cflags --libs)
  

• 加载UDF后，执行SELECT * FROM mysql.func WHERE name='my_udf';确认函数类型、返回值等配置正确。

修复后的核心代码片段 #

extern "C" char* my_udf(UDF_INIT* initid, UDF_ARGS* args, char* result,
                        unsigned long* length, char* is_null, char* error)
{
    *is_null = 1; // 默认返回NULL
    if (!args->args[0]) {
        *error = 1;
        return NULL;
    }

    // 简化版参数转义（实际需处理更多特殊字符）
    char escaped_param[256];
    snprintf(escaped_param, sizeof(escaped_param), "%s", args->args[0]);
    char* p = escaped_param;
    while (*p) {
        if (*p == '"') {
            memmove(p+1, p, strlen(p)+1);
            *p = '\\';
            p += 2;
        } else {
            p++;
        }
    }

    char cmd[512];
    int cmd_len = snprintf(cmd, sizeof(cmd), "/usr/bin/python3 /opt/scripts/checker.py \"%s\"", escaped_param);
    if (cmd_len >= sizeof(cmd)) {
        *error = 1;
        return NULL;
    }

    int ret = system(cmd);
    if (ret == -1 || (WIFEXITED(ret) && WEXITSTATUS(ret) != 0)) {
        *error = 1;
        return NULL;
    }

    // 若需返回结果，取消注释以下代码
    // *is_null = 0;
    // strcpy(result, "Execution successful");
    // *length = strlen(result);

    return NULL;
}

内容的提问来源于stack exchange，提问作者Alexander Vedmed'