OpenMP并行循环中数组allocate与free的段错误解决方案咨询

问题描述 #

在C语言中使用OpenMP并行循环时，循环内动态分配字符串数组并计划在循环结束时释放，但因代码错误与并行机制影响，释放时触发Segmentation Fault：释放i=0对应的数组时，int_comma_size_arr[i]指向了错误的值；尝试用sizeof(keys)/sizeof(*keys)获取数组大小也无法解决问题。

原代码 #

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <omp.h>

/* PRIVATE FUNCTIONS */

static int split(char* input, char* delim, char** result,int array_idx);
/*******************************************************************************
***        FUNCTIONS DEFINITIONS
*******************************************************************************/
int parser(){

    
    int ids_size=3;
    long int_comma_size_arr[]={3,2,4};
    char* str_arr[]={"one,two,three","for,five","six,seven,eight,nine"};
    char* delim=",";
    

    #pragma omp parallel for 
    for(int i=0; i <ids_size; i++) {

        
        char* keys[int_comma_size_arr[i]];
        split(str_arr[i], delim, keys, 0 );

        #pragma omp parallel for 
        for (int n=0; n<int_comma_size_arr[i]; n++) {
            printf(keys[n]);
        }
        

        for (int m=0;m <int_comma_size_arr[i];m++){
            free(keys[m]);
        }
    }
}

/*******************************************************************************
***        PRIVATE FUNCTIONS
*******************************************************************************/
/* 
function name: split
stores array of strings which are chunks of input string, splited with delimiter.
*/
int split(char* input, char* delim, char** result,int array_idx) {
   
    char* found= strstr(input, delim);

    if (found==""||strlen(input)==0){
 
        return 0;

    }else if(found == NULL){
    
        result[array_idx]=malloc(strlen(input)+1 * sizeof(char));
        strncpy(result[array_idx], input, strlen(input));
        *(result[array_idx] + strlen(input)) = '\0';
        return 0;

    }else{

        int length = found - input;
        result[array_idx]=malloc((length+1) * sizeof(char));
        strncpy(result[array_idx], input, length);
        *(result[array_idx] + length) = '\0';
        return split(found+strlen(delim),delim,result,array_idx+1); 
    }
}

问题分析 #

1. Split函数越界写入：

   • 错误判断found==""：strstr返回NULL表示未找到分隔符，而非空字符串地址，该条件永远不成立，导致空输入处理逻辑失效。
   • 无递归边界检查：split递归时未验证array_idx是否超过数组容量，当分割出的字符串数量超过int_comma_size_arr[i]时，会越界写入栈上的keys数组，破坏栈中其他变量（如i或int_comma_size_arr的内容），导致释放时使用错误的数组大小。
   • malloc运算符优先级错误：strlen(input)+1 * sizeof(char)实际计算为strlen(input)+1，虽因sizeof(char)=1无功能影响，但写法不规范易引发其他问题。

2. OpenMP并行不合理：

   • 嵌套并行无必要：外层已用parallel for并行处理每个i，内层再开并行会增加线程调度开销，且栈上的keys数组属于外层线程，内层并行访问可能干扰栈数据。
   • 栈变长数组（VLA）风险：char* keys[int_comma_size_arr[i]]是栈上的变长数组，栈空间有限，越界写入会直接破坏栈帧数据，篡改变量值。

3. 释放依赖被篡改的变量：
   释放循环直接依赖int_comma_size_arr[i]，但该变量可能已被栈越界操作篡改，导致释放次数错误，触发Segmentation Fault。

解决方案 #

1. 修复Split函数 #

• 移除错误的found==""判断，完善空输入处理。
• 添加数组容量参数，递归时检查索引是否越界，避免非法写入。
• 修正malloc的运算符优先级，规范内存分配写法。

2. 调整OpenMP并行策略 #

• 移除不必要的嵌套并行，改用单级并行。
• 用动态分配数组替代栈变长数组，避免栈溢出和数据篡改风险。
• 将当前循环的数组大小保存到私有变量中，释放时使用该变量，避免依赖可能被篡改的共享数组。

修复后的代码 #

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <omp.h>

/* PRIVATE FUNCTIONS */
static int split(char* input, char* delim, char** result, int max_size, int array_idx);

/*******************************************************************************
***        FUNCTIONS DEFINITIONS
*******************************************************************************/
int parser(){
    int ids_size = 3;
    long int_comma_size_arr[] = {3,2,4};
    char* str_arr[] = {"one,two,three","for,five","six,seven,eight,nine"};
    char* delim = ",";

    #pragma omp parallel for 
    for(int i = 0; i < ids_size; i++) {
        // 保存当前数组大小到私有变量，避免后续被篡改
        long curr_size = int_comma_size_arr[i];
        // 动态分配数组，替代栈VLA
        char** keys = malloc(curr_size * sizeof(char*));
        if (!keys) {
            perror("malloc keys failed");
            continue;
        }

        // 调用修复后的split函数，传入数组最大容量
        split(str_arr[i], delim, keys, curr_size, 0);

        // 单线程打印，或用parallel for但确保线程安全
        for (int n = 0; n < curr_size; n++) {
            printf("%s ", keys[n]);
        }
        printf("\n");

        // 使用私有变量curr_size进行释放
        for (int m = 0; m < curr_size; m++) {
            free(keys[m]);
        }
        // 释放动态分配的keys数组本身
        free(keys);
    }
    return 0;
}

/*******************************************************************************
***        PRIVATE FUNCTIONS
*******************************************************************************/
/* 
function name: split
stores array of strings which are chunks of input string, splited with delimiter.
返回实际分割的数量，或-1表示越界
*/
int split(char* input, char* delim, char** result, int max_size, int array_idx) {
    // 空输入处理
    if (strlen(input) == 0) {
        return 0;
    }

    char* found = strstr(input, delim);
    // 检查是否超出数组容量
    if (array_idx >= max_size) {
        fprintf(stderr, "split: array index out of bounds\n");
        return -1;
    }

    if (found == NULL) {
        // 修正malloc优先级
        result[array_idx] = malloc((strlen(input) + 1) * sizeof(char));
        if (!result[array_idx]) {
            perror("malloc failed");
            return -1;
        }
        strcpy(result[array_idx], input);
        return array_idx + 1;
    } else {
        int length = found - input;
        result[array_idx] = malloc((length + 1) * sizeof(char));
        if (!result[array_idx]) {
            perror("malloc failed");
            return -1;
        }
        strncpy(result[array_idx], input, length);
        result[array_idx][length] = '\0';
        // 递归处理剩余字符串
        return split(found + strlen(delim), delim, result, max_size, array_idx + 1);
    }
}

关键改进点 #

• 动态数组替代VLA：避免栈越界导致的变量篡改，同时支持更大的数组规模。
• 私有变量保存数组大小：释放时使用提前保存的curr_size，不再依赖可能被破坏的int_comma_size_arr[i]。
• Split函数边界检查：添加max_size参数，防止越界写入，同时完善错误处理。
• 移除嵌套并行：减少线程开销，避免栈数据的跨线程干扰。

内容的提问来源于stack exchange，提问作者zahra mahani