sem.h
#ifndef _SEM_H_
#define _SEM_H_//创建或打开共享内存:参数为要申请的信号灯集中包含的灯的个数,返回值信号灯集的id
int open_sem(int semcount);//进行申请资源操作:参数为要申请的信号灯id,以及灯的编号
int P(int semid,int semno);//进行释放资源操作:参数为要申请的信号灯id,以及灯的编号
int V(int semid,int semno);//删除信号灯集的操作:参数为要删除的信号灯集id
int del_sem(int semid);#endif
sem.c
#include<myhead.h>union semun
{int val; struct semid_ds *buf; unsigned short *array; struct seminfo *__buf;
};
//定义给信号灯集中的指定信号灯进行赋值
int init_semno(int semid,int semno)
{union semun buf;printf("请输入要给编号为%d的灯设置的值:",semno);scanf("%d",&buf.val);getchar();//调用控制函数if(semctl(semid,semno,SETVAL,buf)){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;
}
shmsnd.c
#include<myhead.h>
#include"sem.h"
#define PAGE_SIZE 4096 //一页的大小
int main(int argc, const char *argv[])
{//11.创建信号灯集int semid=open_sem(2);//1.创建key值key_t key=-1;if((key=ftok("/",'k'))==-1){perror("ftok error");return -1;}printf("key=%d\n",key);//2.通过key值创建一个共享内存int shmid=-1;if((shmid=shmget(key,PAGE_SIZE,IPC_CREAT|0664))==-1){perror("shmget error");return -1;}printf("shmid=%d\n",shmid);//3.将共享内存段的数据映射到用户空间char *addr=shmat(shmid,NULL,0);if(addr==(void *)-1){perror("shmat error");return -1;}printf("addr=%p\n",addr);//4.使用共享内存while(1){//22.申请0号灯资源P(semid,0);printf("请输入>>>");fgets(addr,PAGE_SIZE,stdin);addr[strlen(addr)-1]=0;printf("数据发送成功\n");//33.释放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;}//6.删除共享内存if(shmctl(shmid,IPC_RMID,NULL)==-1){perror("shmctl error");return -1;}return 0;
}
shmrcv.c
#include<myhead.h>
#include"sem.h"
#define PAGE_SIZE 4096 //一页的大小
int main(int argc, const char *argv[])
{//11.创建信号灯集int semid=open_sem(2);//1.创建key值key_t key=-1;if((key=ftok("/",'k'))==-1){perror("ftok error");return -1;}printf("key=%d\n",key);//2.通过key值创建一个共享内存int shmid=-1;if((shmid=shmget(key,PAGE_SIZE,IPC_CREAT|0664))==-1){perror("shmget error");return -1;}printf("shmid=%d\n",shmid);//3.将共享内存段的数据映射到用户空间char *addr=shmat(shmid,NULL,0);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;//33.释放0号灯资源V(semid,0);}//5.取消映射关系if(shmdt(addr)==-1){perror("shmdt error");return -1;}//44.删除信号灯集del_sem(semid);return 0;
}
*******************本文源码来自华清远见张鹏鹏老师*******************