Linux 的信号量

一.信号量的概念

  信号量（信号灯）本质上是一个计数器，用于协调多个进程（但不包括父子进程）对共享数据对象的读/写。它不以传输数据为目的，主要是用来保护共享资源（共享内存、消息队列、socket连接池、数据库连接池等），保证共享资源在一个时刻只有一个进程独享。

  就像公共厕所一样，它是共享资源，它有一个信号灯，提示人们它现在的状态，有没有人在使用。这样就保证了测试一次只能让一个人使用。这也说明了信号灯本质上是一个计数器，厕所里面有人，就显示有人，没人就显示为无人。

  信号量是一个特殊的变量，只允许进程对它进行等待信号和发送信号操作。

最简单的信号量是取值0和1的二元信号量，这是信号量最常见的形式。

  通用信号量（可以取多个正整数值）和信号量集方面的知识比较复杂，应用场景也比较少。本文只介绍二元信号量。

二.相关的函数

1.包含的头文件

  Linux中提供了一组函数用于操作信号量，程序中需要包含以下头文件：

#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>

2.semget 函数

  semaphore——信号量
  get——获得

2.1.函数的声明及参数

  semget函数用来获取或创建信号量，它的原型如下：

int semget(key_t key, int nsems, int semflg);

  1）参数key是信号量的键值，typedef unsigned int key_t，是信号量在系统中的编号，不同信号量的编号不能相同，这一点由程序员保证。key用十六进制表示比较好。
  这个key值是针对系统而言的，政府安装了一批公共厕所门，在政府里的文件的编号为 0x5005。

  2）参数nsems是创建信号量集中信号量的个数，该参数只在创建信号量集时有效，这里固定填1。
  为公共厕所装门，一共装了多少扇门。

  3）参数sem_flags是一组标志，如果希望信号量不存在时创建一个新的信号量，可以和值IPC_CREAT做按位或操作。如果没有设置IPC_CREAT标志并且信号量不存在，就会返错误（errno的值为2，No such file or directory）。

  4）如果semget函数成功，返回信号量集的标识；失败返回-1，错误原因存于error中。
  公共厕所的门安装好了，其实就是厕所设置好了，要告诉百姓厕所门在哪，说是不可能 0x5005 吧，这样谁会知道？所以给它起个名字，叫做牛逼路9527号公共厕所门。这个也是它的标识。

2.2. 示例

  1）获取键值为0x5000的信号量，如果该信号量不存在，就创建它，代码如下：

int semid=semget(0x5000,1,0640|IPC_CREAT);

  2）获取键值为0x5000的信号量，如果该信号量不存在，返回-1，errno的值被设置为2，代码如下：

int semid= semget(0x5000,1,0640)；

3.semctl 函数

3.1 函数的声明及参数

  该函数用来控制信号量（常用于设置信号量的初始值和销毁信号量），它的原型如下：

int semctl(int semid, int sem_num, int command, ...);

  1）参数semid是由semget函数返回的信号量标识。
  这个是控制函数，首先我要知道去控制谁吧，所以第一个参数是semget函数返回的信号量标识。

  2）参数sem_num是信号量集数组上的下标，表示某一个信号量，填0。
  就好比公共厕所，多个坑位（多个共享资源），那么就需要多个信号量（门）。每个信号量有在信号集合里面有序号，从0开始。

  3）参数cmd是对信号量操作的命令种类，

常用的有以下两个：

  IPC_RMID：销毁信号量，不需要第四个参数；

  SETVAL：初始化信号量的值（信号量成功创建后，需要设置初始值），这个值由第四个参数决定。第四参数是一个自定义的共同体，如下：

// 用于信号灯操作的共同体。
  union semun
  {
    int val;
    struct semid_ds *buf;
    unsigned short *arry;
  };

  这个就好比说，厕所建好了，一开始设置为没有人使用的状态。。

3.2 示例

  1）销毁信号量。

 semctl(semid,0,IPC_RMID);

  2）初始化信号量的值为1，信号量可用。

union semun sem_union;
  sem_union.val = 1;
  semctl(semid,0,SETVAL,sem_union);

  semid——是哪里的信号量要设置初始值（哪里的门），

  0——序号为0的信号量要设置初始值

  sem_union.val = 1;——初始值为1

4.semop函数

4.1 函数的声明及参数

  （1）该函数有两个功能：
1）等待信号量的值变为1，如果等待成功，立即把信号量的值置为0，这个过程也称之为等待锁；
  这就像，平时生活中排队上厕所的例子。那些坑位就是共享资源，其实上面的出租车例子可以改成公共厕所。我们会发现，当有人在里面上厕所，关了门之后，那个门的把手附近就会变成红色。这就相当信号量的值为0，我们想要上厕所就要等里面的人出来（信号量的值为1）。然后我们再进去把门给锁上，门把手附近会变成红色（把信号量的值置为0）

2）把信号量的值置为1，这个过程也称之为释放锁。
  就相当于我上完了厕所，出来了让别人使用。

  （2）函数声明：

int semop(int semid, struct sembuf *sops, unsigned nsops);

  1）参数semid是由semget函数返回的信号量标识。

  2）参数nsops是操作信号量的个数，即sops结构变量的个数，设置它的为1（只对一个信号量的操作）。

  3）参数sops是一个结构体，如下：

struct sembuf
{
  short sem_num;   // 信号量集的个数，单个信号量设置为0。
  short sem_op;    // 信号量在本次操作中需要改变的数据：-1-等待操作；1-发送操作。
  short sem_flg;   // 把此标志设置为SEM_UNDO，操作系统将跟踪这个信号量。
                   // 如果当前进程退出时没有释放信号量，操作系统将释放信号量，避免资源被死锁。
};

4.2 示例

  1）等待信号量的值变为1，如果等待成功，立即把信号量的值置为0；

struct sembuf sem_b;
  sem_b.sem_num = 0;
  sem_b.sem_op = -1;
  sem_b.sem_flg = SEM_UNDO;
  semop(sem_id, &sem_b, 1);

  2）把信号量的值置为1。

struct sembuf sem_b;
  sem_b.sem_num = 0;
  sem_b.sem_op = 1;
  sem_b.sem_flg = SEM_UNDO;
  semop(sem_id, &sem_b, 1);

三.示例程序

1. 示例程序(book259.cpp)

  为了便于理解，我把信号量的操作封装成CSEM类，称之为信号灯，类似互斥锁，包括初始化信号灯、等待信号灯、挂出信号灯和销毁信号灯。

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/ipc.h>
#include <sys/sem.h>
 
class CSEM
{
private:
  union semun  // 用于信号灯操作的共同体。
  {
    int val;
    struct semid_ds *buf;
    unsigned short *arry;
  };
 
  int  sem_id;  // 信号灯描述符。
public:
  bool init(key_t key); // 如果信号灯已存在，获取信号灯；如果信号灯不存在，则创建信号灯并初始化。
  bool wait();          // 等待信号灯挂出。
  bool post();          // 挂出信号灯。
  bool destroy();       // 销毁信号灯。
};
 
int main(int argc, char *argv[])
{
   CSEM sem;
 
   // 初始信号灯。
   if (sem.init(0x5000)==false) { printf("sem.init failed.\n"); return -1; }
   printf("sem.init ok\n");
  
   // 等待信信号挂出，等待成功后，将持有锁。
   if (sem.wait()==false) { printf("sem.wait failed.\n"); return -1; }
   printf("sem.wait ok\n");
 
   sleep(50);  // 在sleep的过程中，运行其它的book259程序将等待锁。
  
   // 挂出信号灯，释放锁。
   if (sem.post()==false) { printf("sem.post failed.\n"); return -1; }
   printf("sem.post ok\n");
  
   // 销毁信号灯。
   // if (sem.destroy()==false) { printf("sem.destroy failed.\n"); return -1; }
   // printf("sem.destroy ok\n");
}
 
bool CSEM::init(key_t key)
{
  // 获取信号灯。
  if ( (sem_id=semget(key,1,0640)) == -1)
  {
    // 如果信号灯不存在，创建它。
    if (errno==2)
    {
      if ( (sem_id=semget(key,1,0640|IPC_CREAT)) == -1) { perror("init 1 semget()"); return false; }
 
      // 信号灯创建成功后，还需要把它初始化成可用的状态。
      union semun sem_union;
      sem_union.val = 1;
      if (semctl(sem_id,0,SETVAL,sem_union) <  0) { perror("init semctl()"); return false; }
    }
    else
    { perror("init 2 semget()"); return false; }
  }
 
  return true;
}
 
bool CSEM::destroy()
{
  if (semctl(sem_id,0,IPC_RMID) == -1) { perror("destroy semctl()"); return false; }
 
  return true;
}
 
bool CSEM::wait()
{
  struct sembuf sem_b;
  sem_b.sem_num = 0;
  sem_b.sem_op = -1;
  sem_b.sem_flg = SEM_UNDO;
  if (semop(sem_id, &sem_b, 1) == -1) { perror("wait semop()"); return false; }
   
  return true;
}
 
bool CSEM::post()
{
  struct sembuf sem_b;
  sem_b.sem_num = 0;
  sem_b.sem_op = 1;  
  sem_b.sem_flg = SEM_UNDO;
  if (semop(sem_id, &sem_b, 1) == -1) { perror("post semop()"); return false; }
 
  return true;
}

2.代码解析

2.1 初始化信号灯函数：bool init(key_t key)

  1.先尝试获取信号灯——
semget(key,1,0640)

  2.如果信号灯不存在，就创建信号灯——sem_id=semget(key,1,0640|IPC_CREAT)

  3.信号灯创建成功后，将信号灯初始化成可用的状态——semctl(sem_id,0,SETVAL,sem_union)

2.2 等待信号灯挂出函数 bool wait();

3.测试

  第一步：运行book259程序，它会创建键值为5000的信号灯，并持有锁，如下：

  这就相当于我进去上厕所，把门给锁上了。

  第二步：立即再运行一个book259程序，它会获取键值为5000的信号灯，并等待锁，如下：

  我还在厕所里面，来的人在外面等待

  第三步，当第一次运行的book259程序sleep完50秒之后，释放锁，第二个运行book259的程序将获得锁；

  我出厕所了

  在外面等的人进去了。

  第四步，可以启动更多的book259程序，它们将排队等待锁。

四.其他的操作命令

1.查看系统的信号量

  用 ipcs -s 可以查看系统的信号量，内容有键值（key），信号量编号（semid），创建者（owner），权限（perms），信号量数（nsems）。

2.手工删除信号量

  用ipcrm sem 信号量编号，可以手工删除信号量，如下：