优化这段代码//为消息发送程序 #include<stdio.h> #include<stdlib.h> #include<string.h> #include<sys/stat.h> #include<fcntl.h> #include<pthread.h> #include<semaphore.h> #include<sys/types.h> #include<unistd.h> #include<sys/ipc.h> #include<sys/shm.h> static const char * MUTEX_NAME = "mutex_shm"; static const char * FULL_NAME = "full_shm"; #define INPUT_SIZE 1024 //输入的最大长度 #define KEY_NUM 8848 void P(sem_t *semPtr){ sem_wait(semPtr); } void V(sem_t *semPtr){ sem_post(semPtr); } int main(int argc, char** argv){ key_t key = KEY_NUM; //为共享内存段命名 char input[INPUT_SIZE]; char reply[INPUT_SIZE]; int shmid; char* shmptr; //创建共享内存 shmid = shmget(key, INPUT_SIZE, IPC_CREAT | 0666); if(shmid < 0) { perror("Receiver: Shmget Error"); exit(EXIT_FAILURE); } //启动对该共享内存的访问,并把共享内存连接到当前进程的地址空间 shmptr = shmat(shmid, NULL, 0); sem_t* mutex = sem_open(MUTEX_NAME,O_CREAT); //共享内存只能同时一个程序访问 sem_t* full = sem_open(FULL_NAME,O_CREAT); //共享内存的消息数量 printf("请输入一串字符:"); scanf("%s",input); P(mutex); strcpy(shmptr,input); V(mutex); V(full); printf("消息已发送给receiver!\n"); //把共享内存从当前进程中分离 if(shmdt(shmptr) == -1){ fprintf(stderr, "shmdt failed\n"); exit(EXIT_FAILURE); } return 0; }
时间: 2024-03-26 08:42:05 浏览: 163
C语言头文件 STORAGE
这段代码可以进行以下优化:
1. 在创建信号量时,应该设置初始值,否则可能会出现死锁的情况。
2. 在使用信号量时,应该先判断是否成功获取信号量,否则可能会出现程序崩溃的情况。
3. 应该检查共享内存的连接是否成功。
4. 在结束程序之前,应该关闭信号量和共享内存。
下面是优化后的代码:
```
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/stat.h>
#include<fcntl.h>
#include<pthread.h>
#include<semaphore.h>
#include<sys/types.h>
#include<unistd.h>
#include<sys/ipc.h>
#include<sys/shm.h>
static const char * MUTEX_NAME = "mutex_shm";
static const char * FULL_NAME = "full_shm";
#define INPUT_SIZE 1024 //输入的最大长度
#define KEY_NUM 8848
void P(sem_t *semPtr){
if(sem_wait(semPtr) != 0){
perror("P: sem_wait() failed");
exit(EXIT_FAILURE);
}
}
void V(sem_t *semPtr){
if(sem_post(semPtr) != 0){
perror("V: sem_post() failed");
exit(EXIT_FAILURE);
}
}
int main(int argc, char** argv){
key_t key = KEY_NUM; //为共享内存段命名
char input[INPUT_SIZE];
int shmid;
char* shmptr;
sem_t* mutex;
sem_t* full;
//创建共享内存
shmid = shmget(key, INPUT_SIZE, IPC_CREAT | 0666);
if(shmid < 0)
{
perror("Receiver: Shmget Error");
exit(EXIT_FAILURE);
}
//启动对该共享内存的访问,并把共享内存连接到当前进程的地址空间
shmptr = shmat(shmid, NULL, 0);
if(shmptr == (char*)-1){
perror("shmat failed");
exit(EXIT_FAILURE);
}
//创建信号量
mutex = sem_open(MUTEX_NAME, O_CREAT, 0666, 1);
if(mutex == SEM_FAILED){
perror("sem_open(mutex) failed");
exit(EXIT_FAILURE);
}
full = sem_open(FULL_NAME, O_CREAT, 0666, 0);
if(full == SEM_FAILED){
perror("sem_open(full) failed");
exit(EXIT_FAILURE);
}
printf("请输入一串字符:");
scanf("%s",input);
//获取信号量
P(mutex);
strcpy(shmptr,input);
V(mutex);
V(full);
printf("消息已发送给receiver!\n");
//关闭信号量和共享内存
sem_close(mutex);
sem_close(full);
shmctl(shmid, IPC_RMID, NULL);
return 0;
}
```
阅读全文