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