【Linux】信号量

在Linux系统中，使用信号量通常分为以下几个步骤：

（1）创建信号量或获得系统已存在的信号量——semget()。

（2）初始化信号量——semctl()的SETVAL操作。当时使用二维信号量时，通常将信号量初始化为1。

（3）对信号量进行PV操作——semop()。

（4）删除信号量——semctl()的IPC_RMID操作。

semget()

头文件

#include <sys/types.h>

#include <ipc.h>

#include <sys/sem.h>

函数原型：int semget(key_t key, int nsems, int semflg);

key：信号量的键值

nsems：需要创建的信号量数目，通常取值1

semflg：同open()函数的权限位

成功：信号量标识符

出错：-1

semctl()

头文件

#include <sys/types.h>

#include <sys/ipc.h>

#include <sys/sem.h>

函数原型：int semctl(int semid, int semnum, int cmd, union semun arg);

semid：semget()返回的信号量标识符

semnum：信号量编号，通常取值为0

cmd：指定对信号量的操作

        IPC_STAT：获得该信号量的semid_ds结构，并存放在第四个参数arg结构变量的buf域指向的semid_ds结构中

        IPC_SETVAL：将信号量值设置为arg的val值

        IPC_GETVAL：返回信号量当前值

        IPC_RMID：从系统中删除信号量

arg：是union struct结构

        自行定义：

union semun
{
    int val;
    struct semid_ds *buf;
    unsigned short *array;
}

 成功：IPC_GETVAL返回信号量当前值，其他操作返回0

出错：-1

semop()

头文件

#include <sys/types.h>

#include <sys/ipc.h>

#include <sys/sem.h>

函数原型：int semop(int semid, struct sembuf *sops, size_t nsops);

semid：semget()返回的信号量标识符

sops：指向信号量操作数组，一个数组包括以下成员

struct sembuf
{
    short sem_num; //信号量编号，使用单个信号量时，通常取值为0
    short sem_op; //信号量操作：取值为-1则表示P操作，取值+1则表示V操作
    short sem_flg; //通常设置为SEM_UNDO。这样在进程没释放信号量时，系统自动释放该进程中未释放的信号量
}

 nsops：操作数组sops中的操作个数（元素数目），通常取值为1

成功：信号量标识符

出错：-1

信号量相关函数调用复杂，将它们封装成二维单个信号量的几个基本函数

#include "sem_com.h"

//信号量初始化（赋值）函数
int init_sem(int sem_id, int init_value)
{
    union semun sem_union;
    sem_union.val=init_value; //init_value为初始值
    if(semctl(sem_id, 0, SETVAL, sem_union)==-1)
    {
        perror("init sem");
        return -1;
    }
}

//从系统中删除信号量的函数
int del_sem(int sem_id)
{
    union semun sem_union;
    if(semctl(sem_id, 0, IPC_RMID, sem_union)==-1)
    {
        perror("del sem");
        return -1;
    }
}

//P操作函数
int sem_p(int sem_id)
{
    struct sembuf sem_b;
    sem_b.sem_num=0; //单个信号量的编号为0
    sem_b.sem_op=-1; //表示P操作
    sem_b.sem_flg=SEM_UNDO; //系统自动释放将会在系统中残留的信号量
    if(semop(sem_id, &sem_b, 1)==-1)
    {
        perror("P op");
        return -1;
    }
    return 0;
}

//V操作函数
int sem_v(int sem_id)
{
    struct sembuf sem_b;
    sem_b.sem_num=0; //单个信号量的编号为0
    sem_b.sem_op=1; //表示V操作
    sem_b.sem_flg=SEM_UNDO; //系统自动释放将会在系统中残留的信号量
    if(semop(sem_id, &sem_b, 1)==-1)
    {
        perror("V op");
        return -1;
    }
    return 0;
}

示例：创造一个子进程，接下来使用信号量来控制两个进程（父子进程）之间的执行顺序

#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#define DELAY_TIME 3
int main()
{
    int sem_id;
    sem_id=semget(ftok(".",'a'), 1, 0666|IPC_CREAT); //创建一个信号量
    int init_sem(sem_id,0);

    pid_t pid=fork();
    if(pid<0)
    {
        perror("fork");
        exit(-1);
    }
    else if(pid==0)
    {
        printf("child process will wait for some seconds...\n");
        sleep(DELAY_TIME);
        printf("The returned value is %d in the child process(PID=%d)\n",pid,getpid());
        sem_v(sem_id);
    }
    else
    {
        sem_p(sem_id);
        printf("The returned value is %d in the father process(PID=%d)\n",pid,getpid());
        sem_v(sem_id);
        del_sem(sem_id);
    }
    exit(0);
}