#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); 
}