Linux进程间通信-信号量

什么是信号量

信号量是用来解决进程/线程间同步或互斥的一种机制，也是一个特殊的变量，变量的值代表着当前可以利用的资源。

信号量有两个原子操作：

• P操作(Prolaag，荷兰语减少)
  • –sem：若sem<=0，阻塞等待，否则继续
• V操作(Verhoog，荷兰语增加)
  • ++sem：若sem>0，则唤醒一个在等待队列中的进程/线程

内核为每个信号量集合维护着一个semid_ds结构：

struct semid_ds {
	struct ipc_perm	sem_perm;		/* permissions .. see ipc.h */
	__kernel_time_t	sem_otime;		/* last semop time */
	__kernel_time_t	sem_ctime;		/* last change time */
	struct sem	*sem_base;		/* ptr to first semaphore in array */
	struct sem_queue *sem_pending;		/* pending operations to be processed */
	struct sem_queue **sem_pending_last;	/* last pending operation */
	struct sem_undo	*undo;			/* undo requests on this array */
	unsigned short	sem_nsems;		/* no. of semaphores in array */
};

创建和获取信号量集合

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

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

参数

• key: 关键值
• nsems: 信号量数目
  • 若创建信号量，该值大于0
  • 若获得信号量，该值等于0
• semflg:
  • 若创建信号量，则semflg为IPC_CREATE和IPC_EXCL外加权限
  • 若获得信号量，则semflg为IPC_CREATE

返回值

• 成功：0，并初始化相关的semid_ds
  • sem_perm.cuid 和 sem_perm.uid 被置为用户ID
  • sem_perm.cgid 和 sem_perm.gid 被置为组ID
  • sem_perm.mode 被置为msgflg里的权限
  • sem_nsems 被置为nsems
  • sem_otime 被置为0
  • sem_ctime 被置为当前时间
• 失败：-1，errno

操作信号量

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

int semctl(int semid, int semnum, int cmd, /*union semun sem_union*/);


/* arg for semctl system calls. 
 * 需要自己在代码中声明定义
 */
union semun {
	int val;			/* value for SETVAL */
	struct semid_ds *buf;	/* buffer for IPC_STAT & IPC_SET */
	unsigned short *array;	/* array for GETALL & SETALL */
	struct seminfo *__buf;	/* buffer for IPC_INFO */
	void *__pad;
};

参数

• semid: 信号量集合标识符
• semnum: 信号量的序号
• cmd: 要操作的命令
  • IPC_STAT: 读取一个信号量集的数据结构semid_ds，并将其存储在semun中的buf参数中。
  • IPC_SET: 设置信号量集的数据结构semid_ds中的元素ipc_perm，其值取自semun中的buf参数。
  • IPC_RMID: 将信号量集从内存中删除。
  • GETALL: 用于读取信号量集中的所有信号量的值。
  • GETNCNT: 返回正在等待资源的进程数目。
  • GETPID: 返回最后一个执行semop操作的进程的PID。
  • GETVAL: 返回信号量集中的一个单个的信号量的值。
  • GETZCNT: 返回这在等待完全空闲的资源的进程数目。
  • SETALL: 设置信号量集中的所有的信号量的值。
  • SETVAL: 设置信号量集中的一个单独的信号量的值。
• sem_union: 可选的参数，具体要看cmd的取值。

返回值

• 成功：一般情况下返回0，特殊情况下取决于cmd的值。
  • GETNCNT: semncnt
  • GETPID: sempid
  • GETVAL: semval
  • GETZCNT: semzcnt
  • IPC_INFO:
  • SEM_INFO:
  • SEM_STAT: 信号量集的ID
• 失败：-1，errno

修改信号量

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

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

/* semop system calls takes an array of these. */
struct sembuf {
	unsigned short  sem_num;	/* semaphore index in array */
	short		sem_op;		/* semaphore operation */
	short		sem_flg;	/* operation flags */
};

参数

• semid: 信号量集合的ID
• sops: 操作信号量的结构体数组
  • sem_num: 要操作的信号量在集合中的序号
  • sem_op:
    • 若sem_op为正，这对应于进程释放占用的资源数。sem_op值加到信号量的值上。（V操作）
    • 若sem_op为负,这表示要获取该信号量控制的资源数。信号量值减去sem_op的绝对值。（P操作）
    • 若sem_op为0,这表示调用进程希望等待到该信号量值变成0
  • sem_flg:
    • IPC_NOWAIT: 若sem_op的绝对值大于semval，则errno被置为EAGAIN并立即返回。
    • SEM_UNDO: 进程退出时不做操作，当semop的次数达到0x7fff时便会自动退出。
• nsops: 数组中结构体的个数

返回值

• 成功：0
• 失败：-1，errno

实例

生产者和消费者问题

//sem_pro.c
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <signal.h>


int semid = 0;


union semun 
{
	int val;			/* value for SETVAL */
	struct semid_ds *buf;	/* buffer for IPC_STAT & IPC_SET */
	unsigned short *array;	/* array for GETALL & SETALL */
	struct seminfo *__buf;	/* buffer for IPC_INFO */
	void *__pad;
};


void sig_handler(int signum);


int main()
{
    // 注册信号
    signal(SIGINT, sig_handler);
    signal(SIGKILL, sig_handler);

    // 获取关键值
    key_t key = ftok(".", 1);
    if(key < 0)
    {
        perror("ftok error");
        exit(-1);
    }

    // 创建信号量
    semid = semget(key, 1, IPC_CREAT | 0666);
    if(semid < 0)
    {
        perror("semget error");
        exit(-1);
    }

    // 初始化信号量
    union semun sem_union;
    sem_union.val = 0x7fff;
    
    if(semctl(semid, 0, SETVAL, sem_union) < 0)
    {
        perror("semctl error");
        exit(-1);
    }

    // 操作信号量，生产资源
    struct sembuf sb;
    sb.sem_num = 0;
    sb.sem_op = 1;
    sb.sem_flg = 0;

    int count = 0;
    while(1)
    {
        count = semctl(semid, 0, GETVAL);
        printf("Before producing, we have %d\n", count);
        if(count >= 0x7fff)
        {
            sleep(3);
            continue;
        }
        if(semop(semid, &sb, 1) < 0)
        {
            perror("semop error");
            exit(-1);
        }
        printf("After producing, we have %d\n", semctl(semid, 0, GETVAL));
        //sleep(2);
    }

    return 0;
}


void sig_handler(int signum)
{
    if(semctl(semid, 0, IPC_RMID) < 0)
    {
        perror("semctl error");
        exit(-1);
    }
}

//sem_con.c
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <signal.h>


int semid = 0;


union semun 
{
	int val;			/* value for SETVAL */
	struct semid_ds *buf;	/* buffer for IPC_STAT & IPC_SET */
	unsigned short *array;	/* array for GETALL & SETALL */
	struct seminfo *__buf;	/* buffer for IPC_INFO */
	void *__pad;
};


void sig_handler(int signum);


int main()
{
    // 注册信号
    signal(SIGINT, sig_handler);
    signal(SIGKILL, sig_handler);

    // 获取关键值
    key_t key = ftok(".", 1);
    if(key < 0)
    {
        perror("ftok error");
        exit(-1);
    }

    // 获取信号量
    semid = semget(key, 1, IPC_CREAT | 0666);
    if(semid < 0)
    {
        perror("semget error");
        exit(-1);
    }

    // 操作信号量，消费资源
    struct sembuf sb;
    sb.sem_num = 0;
    sb.sem_op = -1;
    sb.sem_flg = 0;

    while(1)
    {
        printf("Before consuming, we have %d\n", semctl(semid, 0, GETVAL));
        if(semop(semid, &sb, 1) < 0)
        {
            perror("semop error");
            exit(-1);
        }
        printf("After consuming, we have %d\n", semctl(semid, 0, GETVAL));
        //sleep(1);
    }

    return 0;
}


void sig_handler(int signum)
{
    if(semctl(semid, 0, IPC_RMID) < 0)
    {
        perror("semctl error");
        exit(-1);
    }
}