我想做一个接收一些参数的ac程序。第一个参数是一个文件,其他参数是单词。程序将为每个单词创建一个线程,并计算文件中出现的次数。我用了一个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;
}
答案 0 :(得分:1)
您的代码需要一些修复:
malloc()
。实际问题:迭代传递给程序的所有参数并访问最后一个参数。
这个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 1 你可以声明一个线程数组而不是malloc()
一个,这没有错,但如果可以,你应该避免它。
1 <子> Variable Length Arrays 子>