【Linux】线程同步之信号量（Semaphore）

概述

信号量概念

信号量(Semaphore)，有时被称为信号灯，是在多线程环境下使用的一种设施，是可以用来保证两个或多个关键代码段不被并发调用。在进入一个关键代码段之前，线程必须获取一个信号量；一旦该关键代码段完成了，那么该线程必须释放信号量。其它想进入该关键代码段的线程必须等待直到第一个线程释放信号量。

信号量分类

信号量分为有名信号量和内存信号量。

有名信号量通常用于进程之间的信号量同步，而无名信号量一般用于线程之间的信号量同步

函数定义

头文件

#include <semaphore.h>

有名信号量

Posix提供了基于名称的信号量接口，这些信号量由一个name参数标识，它通常指代文件系统中的某个文件，其接口定义如下：

sem_open

• 打开（或创建）有名信号量

sem_t  *sem_open(const char *name, int oflag,..,/*mode_t mode,unsigned int value*/)

参数说明：

• name：信号量的外部名字;
• oflag：选择创建或打开一个现有的信号量;
• mode：权限位;
• value ：信号量初始值;

返回值

• 成功：指向信号量的指针
• 失败：SEM_FAILED

对于oflag的说明

• oflag参数能是0、O_CREAT（创建一个信号量）或O_CREAT|O_EXCL(如果没有指定的信号量就创建）。
• 如果指定了O_CREAT，那么第三个和第四个参数是需要的；
  • 其中mode参数指定权限位;value参数;
  • 指定信号量的初始值，通常用来指定共享资源的数量。该初始不能超过SEM_VALUE_MAX，这个常值必须低于为32767;
• 如果指定了O_CREAT（而没有指定O_EXCL），那么只有所需的信号量尚未存在时才初始化他。
  • 所需信号量已存在条件下指定O_CREAT不是个错误。该标志的意思仅仅是“如果所需信号量尚未存在，那就创建并初始化他”;
  • 不过所需信号量等已存在条件下指定O_CREAT|O_EXCL却是个错误。
• sem_open返回指向sem_t信号量的指针，该结构里记录着当前共享资源的数目。

sem_close

• 关闭有名信号量

int sem_close(sem_t *sem);

说明：

• 一个进程终止时，内核还对其上仍然打开着的所有有名信号量自动执行这样的信号量关闭操作。不论该进程是自愿终止的还是非自愿终止的，这种自动关闭都会发生。
• 但应注意的是关闭一个信号量并没有将他从系统中删除。这就是说，Posix有名信号量至少是随内核持续的：即使当前没有进程打开着某个信号量，他的值仍然保持。

sem_unlink

• 从系统中删除信号量

int sem_unlink(count char *name);

返回值

• 成功：返回0；
• 失败：返回-1

说明：

• 每个信号量有一个引用计数器记录当前的打开次数，sem_unlink必须等待这个数为0时才能把name所指的信号量从文件系统中删除。也就是要等待最后一个sem_close发生。

内存信号量

Posix提供了基于内存的信号量接口，它们由应用程序分配信号量的内存空间，然后由系统初始化它们的值

sem_init

• 初始化信号量

int sem_init(sem_t *sem, int pshared, unsigned int value);

参数说明：

• sem ：信号量地址。
• pshared ：0 – 线程同步 1 – 进程同步。
• value ：最多有几个 线程/进程 操作共享数据。

返回值

• 成功：返回 0；
• 失败：返回-1，并设置错误号；

sem_open和sem_init的区别：

• 创建有名信号量必须指定一个与信号量相关链的文件名称，这个name通常是文件系统中的某个文件，基于内存的信号量不需要指定名称；
• 有名信号量sem 是由sem_open分配内存并初始化成value值，基于内存的信号量是由应用程序分配内存，有sem_init初始化成为value值。如果shared为1，则分配的信号量应该在共享内存中；
• sem_open不需要类似shared的参数，因为有名信号量总是可以在不同进程间共享的，而基于内存的信号量通过shared参数来决定是进程内还是进程间共享，并且必须指定相应的内存；
• 基于内存的信号量不使用任何类似于O_CREAT标志的东西，也就是说，sem_init总是初始化信号量的值，因此，对于一个给定的信号量，我们必须小心保证只调用sem_init一次，对于一个已经初始化过的信号量调用sem_init,结果是未定义的；
• 内存信号量通过sem_destroy删除信号量，有名信号量通过sem_unlink删除；

sem_destroy

• 销毁信号量

int sem_destroy(sem_t *sem);

通用接口

sem_wait

• 监听信号量

int sem_wait(sem_t *sem);
/* 获取信号量后，该接口返回，否则接口一直阻塞 */

 int sem_trywait(sem_t *sem);
/* 尝试获取信号量，无论是否获取成功，该接口都立刻返回*/

int sem_timedwait (sem_t *__restrict __sem,  const struct timespec *__restrict __abstime);
/* 在指定时间内尝试获取信号量，无论是否获取成功，超时后接口返回该接口都立刻返回*/

函数返回值

• 成功：返回 0；
• 失败：返回-1，并设置错误号；

说明：

• 我们能用sem_wait来申请共享资源；
• sem_wait函数能测试所指定信号量的值，如果该值大于0，那就将信号量减1并即时返回。共享资源申请成功。如果该值等于0，调用线程就被进入睡眠状态，直到该值变为大于0，这时再将他减1，函数随后返回。
• sem_wait操作必须是原子的。
• sem_wait和sem_trywait的差别是：当所指定信号量的值已是0时，后者并不将调用线程投入睡眠。相反，他返回一个EAGAIN错误。

sem_post

• 释放信号量

int sem_post(sem_t *sem); 

说明：

当一个线程使用完某个信号量时，他应该调用sem_post来告诉系统申请的资源已用完。本函数和sem_wait函数的功能正好相反，他把所指定的信号量的值加1，然后唤醒正在等待该信号量值变为正数的任意线程。

sem_getvalue

• 获取信号量的值

int sem_getvalue (sem_t *__restrict __sem, int *__restrict __sval)

说明：

该函数返回当前可用信号量的值，通过sval输出参数返回。

代码示例

内存信号量

#include <pthread.h>
#include <semaphore.h>
#include <stdio.h>
#include <unistd.h>

#define MAXNUM 2
sem_t semPtr;
pthread_t a_thread, b_thread, c_thread;
int g_phreadNum = 1;

void *func1(void *arg) {
    sem_wait(&semPtr);
    printf("a_thread get a semaphore \n");
    sleep(5);
    sem_post(&semPtr);
    printf("a_thread release semaphore \n");
    pthread_join(a_thread, NULL);
}

void *func2(void *arg) {
    sem_wait(&semPtr);
    printf("b_thread get a semaphore \n");
    sleep(5);
    sem_post(&semPtr);
    printf("b_thread release semaphore \n");
    pthread_join(b_thread, NULL);
}

void *func3(void *arg) {
    sem_wait(&semPtr);
    printf("c_thread get a semaphore \n");
    sleep(5);
    sem_post(&semPtr);
    printf("c_thread release semaphore \n");
    pthread_join(c_thread, NULL);
}

int main() {
    int taskNum;

    // 创建2个信号量
    sem_init(&semPtr, 0, MAXNUM);
    //线程1获取1个信号量，5秒后释放
    pthread_create(&a_thread, NULL, func1, NULL);
    //线程2获取1个信号量，5秒后释放
    pthread_create(&b_thread, NULL, func2, NULL);

    sleep(1);
    //线程3获取信号量，只有线程1或者线程2释放后，才能获取到
    pthread_create(&c_thread, NULL, func3, NULL);
    sleep(10);
    //销毁信号量
    sem_destroy(&semPtr);
    return 0;
}

有名信号量

#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

int main(int argc, char **argv) {
    sem_t *sem;

    if (argc != 2) {
        printf("please input a file name !\n");
        exit(1);
    }

    sem = sem_open(argv[1], O_CREAT, 0644, 1);
    /* 信号量使用请参考无名信号量种的使用方式 */
    sleep(1);
    sem_close(sem);
    sem_unlink(argv[1]);
    exit(0);
}