Linux学习笔记之IPC 对象之信号量集

目的：配合共享内存完成进程间通信；

使用框架：
key ==>申请信号量集 ==》pv操作 ===》卸载删除信号量集
semget() semop() semctl()

1、申请信号量
#include <sys/sem.h>

int semget(key_t key, int nsems, int semflg);
功能：该函数可以使用特定的key向内核提出信号量集使用申请；
参数：key 唯一键值
nsems 要申请的信号量个数；
semflg 申请信号量的方式，IPC_CREAT IPC_EXCL
返回值：成功 semid 信号量集id
失败 -1；

2、pv操作：
p 操作： 申请资源 sem_wait() ===>sem = sem -1;
v 操作： 释放资源 sem_post() ===>sem = sem +1;

int semop(int semid, struct sembuf sops, unsigned nsops);
功能：该函数可以完成信号量集中指定信号量的pv操作；
参数： semid 信号量集id
sops 信号量集针对pv操作组织的结构体：如下
struct sembuf
{
unsigned short sem_num; / semaphore number /
short sem_op; / semaphore operation /
//sem_op = 1 表示v操作， 释放资源
sem_op = -1表示p操作， 申请资源
sem_op = 0 默认值，阻塞等待
short sem_flg; / operation flags */
}
nsops 操作信号量的个数；
返回值：成功 0
失败 -1；

3、删除信号量：
int semctl(int semid, int semnum, int cmd, …);
功能：该函数可以修改信号量集合中指定信号量的属性，也可以删除信号量；
参数：semid 要删除的信号量集id
semnum 要删除的信号量集中的信号量编号
cmd IPC_RMID 删除信号量的宏
… 可变长参数，取决与cmd 的宏，一般犹如下两个宏：
cmd == SETVAL 则 可变长部分可以传入一个初始值
cmd == GETVAL 则 可变长部分可以不传数据，函数执行
的返回值就是获取值
返回值：成功 0
失败 -1；

//信号灯写
#include <stdio.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <errno.h>
#include <string.h>
#include <sys/sem.h>
#define N 32
struct  shmbuf
{
     char buf[N];
};
union semun
{
     int val;
};
int main()
{

      key_t key;
      
      if((key=ftok(".",'m'))<0)
      {
            perror("ftok error\n");
            return -1;
      }
      
      int shmid;
      struct shmbuf *shm;
      if((shmid=shmget(key,512,IPC_CREAT|IPC_EXCL|0664))<0)
      {
           if(errno!=EEXIST)
           {
                 perror("shmget error\n");
                 return -1;
           }
           else
           {
                 shmid=shmget(key,512,0664);
           }
      }
      if((shm=shmat(shmid,NULL,0))!=NULL)
      {
           printf("shm=%p\n",shm);
      }

      int semid;
      union semun semun;
      struct sembuf sembuf;
      if((semid=semget(key,2,IPC_CREAT|IPC_EXCL|0664))<0)
      {
           if(errno!=EEXIST)
           {
                 perror("semget erroe\n");
                 return -1;
           }
           else
           {
                 semid=semget(key,2,0664);
           }
      }
      else
      {
           //初始化信号量的值
           semun.val=0;
           semctl(semid,0,SETVAL,semun);
           
           semun.val=1;
           semctl(semid,1,SETVAL,semun);           
      }
      while(1)
      {
           //申请信号量
           sembuf.sem_num=1;
           sembuf.sem_op=-1;
           sembuf.sem_flg=0;
           semop(semid,&sembuf,1);
           
           fgets(shm->buf,N,stdin);
           shm->buf[strlen(shm->buf)-1]='\0';
           
           //释放信号量
           sembuf.sem_num=0;
           sembuf.sem_op=1;
           sembuf.sem_flg=0;
           semop(semid,&sembuf,1);  
           
           //删除信号量
           if(strncmp(shm->buf,"quit",4)==0)
           {
                shmdt(shm);
                break;
           }         
      }
      
      return 0;
}

//信号灯读
#include <stdio.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <errno.h>
#include <string.h>
#include <sys/sem.h>
#define N 32

struct  shmbuf
{
     char buf[N];
};
union semun
{
     int val;
};
int main()
{

      key_t key;
      
      if((key=ftok(".",'m'))<0)
      {
            perror("ftok error\n");
            return -1;
      }
      
      int shmid;
      struct shmbuf *shm;
      if((shmid=shmget(key,512,IPC_CREAT|IPC_EXCL|0664))<0)
      {
           if(errno!=EEXIST)
           {
                 perror("shmget error\n");
                 return -1;
           }
           else
           {
                 shmid=shmget(key,512,0664);
           }
      }
      if((shm=shmat(shmid,NULL,0))!=NULL)
      {
           printf("shm=%p\n",shm);
      }
      int semid;
      union semun semun;
      struct sembuf sembuf;      
      if((semid=semget(key,2,IPC_CREAT|IPC_EXCL|0664))<0)
      {
           if(errno!=EEXIST)
           {
                 perror("semget erroe\n");
                 return -1;
           }
           else
           {
                 semid=semget(key,2,0664);
           }
      }
      else
      {
           //初始化信号量的值
           semun.val=0;
           semctl(semid,0,SETVAL,semun);
           
           semun.val=1;
           semctl(semid,1,SETVAL,semun);           
      } 
      while(1)
      {
           //申请信号量
           sembuf.sem_num=0;
           sembuf.sem_op=-1;
           sembuf.sem_flg=0;
           semop(semid,&sembuf,0);
           
           if(strncmp(shm->buf,"quit",4)==0)
           {
                goto err;
           } 
                       
           printf("buf:%s\n",shm->buf);
           
           //释放信号量
           sembuf.sem_num=1;
           sembuf.sem_op=1;
           sembuf.sem_flg=0;
           semop(semid,&sembuf,1);             
      }    
      
      return 0;
err:
     shmdt(shm);  
     shmctl(shmid,IPC_RMID,NULL);
     shmctl(semid,0,IPC_RMID);
     shmctl(semid,1,IPC_RMID);    
}