2.将信号灯集的函数二次封装
sem.c
#include<myhead.h>
union semun
{
int val; /* Value for SETVAL */
struct semid_ds *buf; /* Buffer for IPC_STAT, IPC_SET */
unsigned short *array; /* Array for GETALL, SETALL */
struct seminfo *__buf; /* Buffer for IPC_INFO(Linux-specific) */
};
//定义给信号灯集中的指定信号灯进行赋值
int init_semno(int semid,int semno)
{
union semun buf;
printf("请输入要给编号为%d的灯设置的值:",semno);
scanf("%d",&buf.val);
//调用控制函数
if(semctl(semid,semno,SETVAL,buf) == -1)
{
perror("semctl error");
return -1;
}
return 0;
}
//创建或打开共享内存:参数为要申请的信号灯集中包含的灯的个数,返回值信号灯集的id
int open_sem(int semcount)
{
//创建key值
key_t key = -1;
if((key = ftok("/",'s')) == -1)
{
perror("ftok error");
return -1;
}
//通过key值创建一个信号灯集
int semid = -1;
if((semid = semget(key,semcount,IPC_CREAT|IPC_EXCL|0664)) == -1)
{
//对错误码进行判断,如果错误码为EEXIST,说明信号灯集已经存在
if(errno == EEXIST)
{
//直接打开信号灯集即可
semid = semget(key ,semcount ,IPC_CREAT|0664);
return semid;
}
perror("semget error");
return -1;
}
//给信号灯集中的每个灯进行初始化操作
for(int i=0; i<semcount; i++)
{
init_semno(semid,i); //对编号为i的灯进行初始化
}
return semid;
}
//进行申请资源操作:参数为要申请的信号灯id,以及灯的编码
int P(int semid ,int semno)
{
//定义要进行操作的结构体变量
struct sembuf buf;
buf.sem_num = semno;
buf.sem_op = -1; //表示申请资源
buf.sem_flg = 0; //表示阻塞方式申请
//执行函数
if(semop(semid,&buf,1) == -1)
{
perror("P error");
return -1;
}
return 0;
}
//进行释放资源操作:参数为要申请的信号灯id,以及灯的编号
int V(int semid,int semno)
{
//定义要进行操作的结构体变量
struct sembuf buf;
buf.sem_num = semno;
buf.sem_op = 1; //表示释放资源
buf.sem_flg = 0; //表示阻塞方式释放
//执行函数
if(semop(semid,&buf,1) == -1)
{
perror("V error");
return -1;
}
return 0;
}
//删除信号灯集的操作:参数为要删除的信号灯集id
int del_sem(int semid)
{
//删除信号灯集
if(semctl(semid,0,IPC_RMID) == -1)
{
perror("delete error");
return -1;
}
return 0;
}
shmrcv.c
#include<myhead.h>
#include"sem.h"
#define PAGE_SIZE 4096 //一页大小
int main(int argc, const char *argv[])
{
//创建信号灯集
int semid = open_sem(2);
//创建key值
key_t key = -1;
if((key = ftok("/",'k')) == -1)
{
perror("ftok error");
return -1;
}
printf("key = %d\n", key);
//通过key值创建一个共享内存
int shmid = -1;
if((shmid = shmget(key ,PAGE_SIZE, IPC_CREAT|0664)) == -1)
{
perror("shmget error");
return -1;
}
printf("shmid = %d\n",shmid);
//将共享内存段的数据映射到用户空间
char *addr = shmat(shmid, NULL,0);
//参数2:让系统自动选择对其页
//参数3:对共享内存的操作为读写操作
if(addr == (void*)-1)
{
perror("shmat error");
return -1;
}
printf("addr = %p\n",addr);
//4.使用共享内存
while(1)
{
//申请1号灯的资源
P(semid,1);
printf("收到数据为:%s\n",addr);
//判断输入的内容
if(strcmp(addr,"quit")== 0)
{
break;
}
//释放0号灯的资源
V(semid,0);
}
//5.取消映射关系
if(shmdt(addr) == -1)
{
perror("shmdt error");
return -1;
}
//删除信号灯集
del_sem(semid);
return 0;
}
shmsnd.c
#include<myhead.h>
#include"sem.h"
#define PAGE_SIZE 4096 //一页大小
int main(int argc, const char *argv[])
{
//创建信号灯集
int semid = open_sem(2);
//创建key值
key_t key = -1;
if((key = ftok("/",'k')) == -1)
{
perror("ftok error");
return -1;
}
printf("key = %d\n", key);
//通过key值创建一个共享内存
int shmid = -1;
if((shmid = shmget(key ,PAGE_SIZE, IPC_CREAT|0664)) == -1)
{
perror("shmget error");
return -1;
}
printf("shmid = %d\n",shmid);
//将共享内存段的数据映射到用户空间
char *addr = shmat(shmid, NULL,0);
//参数2:让系统自动选择对其页
//参数3:对共享内存的操作为读写操作
if(addr == (void*)-1)
{
perror("shmat error");
return -1;
}
printf("addr = %p\n",addr);
//4.使用共享内存
while(1)
{
//申请0号灯的资源
P(semid,0);
printf("请输入>>");
fgets(addr,PAGE_SIZE,stdin);
addr[strlen(addr)-1]=0;
printf("数据发送成功\n");
//释放1号灯的资源
V(semid,1);
//判断输入的内容
if(strcmp(addr,"quit")== 0)
{
break;
}
if(strcmp(addr,"q")== 0)
{
break;
}
}
//5.取消映射关系
if(shmdt(addr) == -1)
{
perror("shmdt error");
return -1;
}
//删除共享函数
if(shmctl(shmid,IPC_RMID,NULL) == -1)
{
perror("shmctl error");
return -1;
}
return 0;
}