c - C 程序的段错误

Question

我想做一个接收一些参数的c程序。第一个参数是一个文件，其他参数是单词。该程序将为每个单词创建一个线程，并计算文件中出现的次数。我使用了shell脚本(prts.sh)来计算文件中单词的出现次数。我编写了代码，但是当我尝试运行它时，出现段错误。我通过结构发送文件和参数。我认为问题所在当我尝试访问在线程函数中发送的结构元素时。这是我到目前为止编写的代码:

#include<pthread.h>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#define N 1000
int sum=0;
int n;

typedef struct {
char arg1[N];
char arg2[N];
} Mesaj;

pthread_mutex_t m=PTHREAD_MUTEX_INITIALIZER;
void* func(void *p){
    Mesaj *msg_func=(Mesaj*)p;
    char cmd[N];
    char result[N];
    FILE *fp;
    pthread_mutex_lock(&m);
    sprintf(cmd,"/home/alexdamian/prts.sh %s %s",msg_func->arg1,msg_func->arg2);  
    fp = popen(cmd, "r");
    fgets(result, N, fp);
    pclose(fp); 
    int n=atoi((char*)result);
    sum=sum+n;
    pthread_mutex_unlock(&m);
    free(msg_func);
    return NULL;
}
int main(int argc,char *argv[]){

pthread_t *th=malloc(argc*sizeof(pthread_t));


int i;
for(i=1;i<argc;i++){

    Mesaj *msg=malloc(sizeof(Mesaj));
    strcpy(msg->arg1,argv[i]);
    strcpy(msg->arg2,argv[i+1]);    

    pthread_create(th+i,NULL,func,msg);
}

for(i=0;i<argc;i++){
    pthread_join(*(th+i),NULL);
}
printf("the sum is %d\n",sum);
free(th);
return 0;

}

Answer 1

您的代码需要进行一些修复:

1. 风格和格式，代码必须美观、赏心悦目。
2. 不需要时不要使用 malloc()。
3. 可以使用索引表示法时，请勿使用指针算术表示法。
4. 始终检查函数的返回值

实际问题:您迭代传递给程序的所有参数并访问最后一个参数之外的 1 个参数。

这个for循环:

int i;
for(i=1;i<argc;i++){

    Mesaj *msg=malloc(sizeof(Mesaj));
    strcpy(msg->arg1,argv[i]);
    strcpy(msg->arg2,argv[i+1]);    

    pthread_create(th+i,NULL,func,msg);
}

应该是:

int i;
for (i = 1 ; i < argc - 1 ; i++)
{    
    Mesaj *msg;

    msg = malloc(sizeof(*msg));
    if (msg == NULL) /* do anything except dereferencing `msg' */
        continue;
    strcpy(msg->arg1, argv[i]);
    strcpy(msg->arg2, argv[i + 1]);    

    pthread_create(&th[i], NULL, func, msg);
}

我修复了您的程序以显示一些您可以改进的地方:

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

#define N 1000

typedef struct
{
    char arguments[2][N];
    int *data;
} Mesaj;
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;

void *
function(void *data)
{
    Mesaj      *message;
    int        *value;
    char        cmd[N];
    char        result[N];
    FILE       *pipe;
    const char *format;

    message = (Mesaj*) data;
    value   = (int *) message->data;
    format  = "/home/iharob/prts.sh %s %s";

    snprintf(cmd, sizeof(cmd), format, 
        message->arguments[0], message->arguments[1]);
    pipe = popen(cmd, "r");
    if (pipe != NULL)
    {
        fgets(result, sizeof(result), pipe);
        pclose(pipe);

        /* This section should be protected with mutex */
        pthread_mutex_lock(&mutex);
        *value = *value + strtol(result, NULL, 10);
        pthread_mutex_unlock(&mutex);
    }
    else
        fprintf(stderr, "cannot execute the command...\n");

    free(message);

    return NULL;
}

int main(int argc,char *argv[])
{
    pthread_t  thread[argc];
    int        i;
    int        value;

    value = 0;
    for (i = 1 ; i < argc - 1 ; i++)
    {
        Mesaj *message;

        message = malloc(sizeof(*message));
        if (message != NULL)
        {
            message->data = &value;

            strcpy(message->arguments[0], argv[i]);
            strcpy(message->arguments[1], argv[i + 1]);

            pthread_create(&thread[i], NULL, function, message);
        }
    }

    for (i = 1 ; i < argc - 1 ; i++)
        pthread_join(thread[i], NULL);
    printf("The sum is %d\n", value);

    return 0;
}

例如，您在此程序中不需要单个全局变量，您可以在堆栈中创建变量并将其作为消息的一部分传递，并且您只需要 malloc()消息，因此它在线程启动时存在，使用 VLA¹ 您可以声明一个线程数组，而不是 malloc() 一个，这并没有错，但您应该如果可以的话，避免它。

<小时/>

1<子> Variable Length Arrays