本文转自:http://linux.chinaitlab.com/c/825371.html
#include<sys/types.h>
#include<linux/sem.h>
#include<linux/shm.h>
#include<unistd.h>
#include<stdio.h>
#include<errno.h>
#include<time.h>
#define MAXSHM 5 //定义缓冲区数组的下标变量个数
/* 定义3个信号量的内部标识 */
int fullid;
int emptyid;
int mutexid;
/* 主函数 */
int main()
{
/* 定义2个共享内存的ID */
int arrayid;
int getid;
/* 定义共享内存虚拟地址 */
int *array;
int *get;
/* 创建共享内存 */
arrayid=shmget(IPC_PRIVATE,sizeof(int) *MAXSHN,IPC_CREAT|0666);
getid=shmget(IPC_PRIVATE,sizeof(int),IPC_CREAT|0666);
/* 初始化共享内存 */
array=(int *) shmat(arrayid,0,0);
get=(int *) shmat(getid,0,0);
*get=0;
/* 定义信号量数据结构 */
struct sembuf P,V;
union semun arg;
/* 创建信号量 */
fullid=semget(IPC_PRIVATE,1,IPC_CREAT|0666);
emptyid=semget(IPC_PRIVATE,1,IPC_CREAT|0666);
mutexid=semget(IPC_PRIVATE,1,IPC_CREAT|0666);
/*初始化信号量 */
arg.val=0; //初始时缓冲区中无数据
if(semctl(fullid,0,SETVAL,arg)==-1)
perror(“semctl setval error”);
arg.val=MAXSHM; //初始时缓冲区中有5个空闲的数组元素
if(semctl(emptyid,0,SETVAL,arg)==-1)
perror(“semctl setval error”);
arg.val=1; //初始时互斥信号为1,允许一个进程进入
if(semctl(mutexid,0,SETVAL,arg)==-1)
perror(“semctl setval error”);
/* 初始化 P V操作 */
P.sem_num=0;
P.sem_op=-1;
P.sem_flg=SEM_UNDO;
V.sem_num=0;
V.sem_op=1;
V.sem_flg=SEM_UNDO;
/* 生产者进程 */
if(fork()==0)
{
int i=0;
int set=0;
while(i<10)
{
semop(emptyid,&P,1); //对 emptyid执行P操作
semop(mutexid,&P,1); //对 mutexid执行 P操作
array[set%MAXSHM]=i+1;
printf(“Producer put number %d to No.%d/n”,array[set%MAXSHM],set%MAXSHM);
set++; //写计数加1
semop(mutexid,&V,1); //对mutexid执行 V 操作
semop(fullid,&V,1); //对fullid执行 V 操作
i++;
}
sleep(3); //SLEEP 3秒,等待消费者进程执行完毕
printf(“Poducer if over/n”);
exit(0);
}
else
{
/* 消费者A进程 */
if(fork()==0)
{
while(1)
{
if(*get==10)
break;
semop(fullid,&P,1); //对fullid执行 P 操作
semop(mutexid,&P,1); //对mutexid执行 P 操作
printf(“The ConsumerA get number from No.%d/n”,(*get)%MAXSHM);
(*get)++; //读计数加1
semop(mutexid,&V,1); //对mutexid执行 V 操作
semop(emptyid,&V,1); //对fullid执行 V 操作
sleep(1);
}
printf(“ConsunerA is over/n”);
exit(0);
}
else
{
/*消费者B进程 */
if(fork()==0)
{
while(1)
{
if(*get==10)
break;
semop(fullid,&P,1); //对fullid执行 P 操作
semop(mutexid,&P,1); //对mutexid执行 P 操作
printf(“The ConsumerA get number from No.%d/n”,(*get)%MAXSHM);
(*get)++; //读计数加1
semop(mutexid,&V,1); //对mutexid执行 V 操作
semop(emptyid,&V,1); //对emptyid执行 V 操作
sleep(1);
}
printf(“ConsunerB is over/n”);
exit(0);
}
}
}
/* 父进程返回后回收3个子进程 */
wait(0);
wait(0);
wait(0);
/* 断开并撤消2个共享内存 */
shmdt(array);
shmctl(arrayid,IPC_RMID,0);
shmctl(get);
shmctl(getid,IPC_RMID,0);
/* 撤消3个信号量集 */
semctl(emptyid,IPC_RMID,0);
semctl(fullid,IPC_RMID,0);
semctl(mutexid,IPC_RMID,0);
exit(0);
}