static char *TCPCommand = NULL; static pthread_mutex_t mutex = P_MUTEX_INITIALIZER; int SetTCPCommand(char *command) { pthread_mutex_lock(&mutex); // 加锁 if (TCPCommand != NULL) { free(TCPCommand); } TCPCommand = malloc(strlen(command) + 1); if (TCPCommand == NULL) { printf("Failed to allocate memory for TCP command\n"); pthread_mutex_unlock(&mutex); // 解锁 return 0; } strcpy(TCPCommand, command); printf("set: %s\n", TCPCommand); pthread_mutex_unlock(&mutex); // 解锁 return 1; } int GetTCPCommand(char **command) { pthread_mutex_lock(&mutex); // 加锁 if (TCPCommand == NULL || strlen(TCPCommand) == 0) { printf("TCP command is empty\n"); pthread_mutex_unlock(&mutex); // 解锁 return 0; } *command = malloc(strlen(TCPCommand) + 1); if (*command == NULL) { printf("Failed to allocate memory for command\n"); pthread_mutex_unlock(&mutex); // 解锁 return 0; } strcpy(*command, TCPCommand); printf("get: %s\n", *command); free(TCPCommand); TCPCommand = NULL; pthread_mutex_unlock(&mutex); // 解锁 return 1; }根据这个帮我优化代码,因为我是在另一个进程一直执行GetTCPCommand,所以他会先于SetTCPCommand执行
时间: 2024-04-15 13:28:20 浏览: 109
线程加锁程序,pthread-mutex-t
根据你的描述,如果另一个进程在不断执行`GetTCPCommand`,而且在调用`SetTCPCommand`之前执行了`GetTCPCommand`,那么可能会出现问题。这是因为在获取到TCPCommand之后,你立即将其释放,并将其置为NULL,这样在另一个进程中再次调用`GetTCPCommand`时就会返回空值。
为了解决这个问题,你可以使用条件变量来实现等待和通知机制。以下是修改后的代码:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>
static char *TCPCommand = NULL;
static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
static pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
int SetTCPCommand(char *command) {
pthread_mutex_lock(&mutex); // 加锁
if (TCPCommand != NULL) {
free(TCPCommand);
}
TCPCommand = malloc(strlen(command) + 1);
if (TCPCommand == NULL) {
printf("Failed to allocate memory for TCP command\n");
pthread_mutex_unlock(&mutex); // 解锁
return 0;
}
strcpy(TCPCommand, command);
printf("set: %s\n", TCPCommand);
pthread_cond_signal(&cond); // 发送信号通知等待的线程
pthread_mutex_unlock(&mutex); // 解锁
return 1;
}
int GetTCPCommand(char **command) {
pthread_mutex_lock(&mutex); // 加锁
while (TCPCommand == NULL || strlen(TCPCommand) == 0) {
// 等待条件满足
pthread_cond_wait(&cond, &mutex);
}
*command = malloc(strlen(TCPCommand) + 1);
if (*command == NULL) {
printf("Failed to allocate memory for command\n");
pthread_mutex_unlock(&mutex); // 解锁
return 0;
}
strcpy(*command, TCPCommand);
printf("get: %s\n", *command);
free(TCPCommand);
TCPCommand = NULL;
pthread_mutex_unlock(&mutex); // 解锁
return 1;
}
```
在修改后的代码中,我们引入了一个条件变量`cond`来实现等待和通知机制。在`GetTCPCommand`函数中,如果TCPCommand为空,则调用`pthread_cond_wait`等待条件满足。而在`SetTCPCommand`函数中,我们在设置完TCPCommand并释放锁之前,调用`pthread_cond_signal`发送信号通知等待的线程。
这样,在另一个进程一直执行`GetTCPCommand`时,如果在调用`SetTCPCommand`之前执行了`GetTCPCommand`,它会等待直到有新的TCPCommand被设置,并且不会错过通知。
希望这次能满足你的需求!如果你还有其他问题,请随时提问。
阅读全文