Linux下多线程编程中信号量介绍及简单使用

在Linux中有两种方法用于处理线程同步：信号量和互斥量。

线程的信号量是一种特殊的变量，它可以被增加或减少，但对其的关键访问被保证是原子操作。如果一个程序中有多个线程试图改变一个信号量的值，系统将保证所有的操作都将依次进行。信号量一般常用于保护一段代码，使其每次只被一个执行线程运行。信号量是用来调协线程对共享资源的访问的。

通过使用信号量可以很好的完成线程同步。两个线程同时监视同一个信号量。A线程增加信号量的值，B线程减少信号量的值。当A线程增加信号量大于0时，B线程的等待信号量就会触发，每触发一次将信号量减1，直到将信号量减为0，B线程继续等待A线程增加信号量。

信号量和互斥锁(mutex)的区别：互斥锁只允许一个线程进入临界区，而信号量允许多个线程同时进入临界区。

信号量用在多线程多任务同步的，一个线程完成了某一个动作就通过信号量告诉别的线程，别的线程再进行某些动作。而互斥锁是用在多线程多任务互斥的，一个线程占用了某一个资源，那么别的线程就无法访问，直到这个线程unlock，其他的线程才开始可以利用这个资源。比如对全局变量的访问，有时要加锁，操作完了，在解锁。有的时候锁和信号量会同时使用的。

信号量：只要信号量的value大于0，其他线程就可以sem_wait成功，成功后信号量的value减1。若value值不大于0，则sem_wait使得线程阻塞，直到sem_post释放后value值加1,但是sem_wait返回之前还是会将此value值减1.

如果信号量的值大于0表示可用的资源数，小于0表示阻塞的线程数。

互斥锁: 只要被锁住，其他任何线程都不可以访问被保护的资源，也就是说，信号量不一定是锁定某一个资源，而是流程上的概念，比如：有A,B两个线程，B线程要等A线程完成某一任务以后再进行自己下面的步骤，这个任务并不一定是锁定某一资源，还可以是进行一些计算或者数据处理之类。而线程互斥量则是“锁住某一资源”的概念，在锁定期间内，其他线程无法对被保护的数据进行操作。在有些情况下两者可以互换。

信号量是一个特殊类型的变量，它可以被增加或减少，但对它的访问都会被保证是原子操作，即使在一个多线程程序中也是如此。也就是说，如果一个程序中有两个或多个线程试图改变一个信号量的值，系统将保证所有的操作都将依次进行。如果换成普通的变量，来自同一个程序中的不同线程的冲突操作将会导致不确定的操作。

两种信号量：二进制信号量和计数信号量。二进制信号量只有0和1两种取值，而计数信号量则有更大的取值范围。如果某个共享资源只能被一个线程访问，那么二进制信号量则是最好的打算；如果有多个线程需要访问共享资源呢，使用计数信号量则是个好的主意。

互斥锁只有0，1两中状态，适合于线程对共享资源的独占访问，很多时候每个资源可以同时被有限的线程访问，此时互斥锁将无法满足；条件变量同步也同样存在这种问题。信号量实际是一种非负整型计数器，可以很好的控制线程之间资源访问，互斥锁能实现的功能，信号量同样可以。

信号量控制资源共享主要是PV原语操作， PV原语是对整数计数器信号量sem的操作。一次P操作使sem减一，而一次V操作使sem加一。进程（或线程）根据信号量的值来判断是否对公共资源具有访问权限。当信号量sem的值大于等于零时，该进程（或线程）具有公共资源的访问权限；相反，当信号量sem的值小于零时，该进程（或线程）就将阻塞直到信号量sem的值大于等于0为止。

信号量的函数都以sem_开头，线程中使用的基本信号量函数有4个，它们都声明在头文件semaphore.h中。

sem_init(sem_t*sem, int pshared, unsigned int value)：该函数用于创建信号量。初始化一个定位在sem的匿名信号量。value参数指定信号量的初始值。pshared参数指明信号量是由进程内线程共享，还是由进程之间共享。如果pshared的值为0，那么信号量将被进程内的线程共享，并且应该放置在所有线程都可见的地址上(如全局变量，或者堆上动态分配的变量)。

sem_wait(sem_t*sem)：该函数用于以原子操作的方式将信号量的值减1。(原子操作就是，如果两个线程企图同时给一个信号量加1或减1，它们之间不会互相干扰。)但它永远会先等待该信号量为一个非零值才开始做减法。也就是说，如果你对一个值为2的信号量调用sem_wait(),线程将会继续执行，这信号量的值将减到1。如果对一个值为0的信号量调用sem_wait()，这个函数就会等待直到有其它线程增加了这个值使它不再是0为止。如果有两个线程都在sem_wait()中等待同一个信号量变成非零值，那么当它被第三个线程增加一个“1”时，等待线程中只有一个能够对信号量做减法并继续执行，另一个还将处于等待状态。被用来阻塞当前线程直到信号量sem的值大于0，解除阻塞后将sem的值减一，表明公共资源经使用后减少。

sem_post(sem_t*sem)：该函数用于以原子操作的方式将信号量的值加1。用来增加信号量的值当有线程阻塞在这个信号量上时，调用这个函数会使其中的一个线程不再阻塞，选择机制同样是由线程的调度策略决定的。它信号量的值加1同时发出信号来唤醒等待的线程。

sem_destroy：该函数用于对用完的信号量的清理。用来释放信号量sem。

下面是从其他文章中copy的测试代码，详细内容介绍可以参考对应的reference：

test_thread_sem.cpp：

// reference: https://software.intel.com/zh-cn/blogs/2011/12/02/linux-3
#include <iostream>
#include <pthread.h>
#include <semaphore.h>
#include <string.h>namespace {int g_Flag = 0;
sem_t sem_mutex; // 用于互斥
sem_t sem_syn; // 用于同步void *thread1(void *arg)
{fprintf(stdout, "Enter thread1\n");fprintf(stdout, "thread1 id: %u, g_Flag: %d\n", (unsigned int)pthread_self(), g_Flag);if (sem_wait(&sem_mutex) != 0) {fprintf(stderr, "pthread1 sem_mutex fail\n");return nullptr;}if (g_Flag == 2)sem_post(&sem_syn);g_Flag = 1;if (sem_post(&sem_mutex) != 0) {fprintf(stderr, "pthread1 sem_post fail\n");return nullptr;}fprintf(stdout, "thread1 id: %u, g_Flag: %d\n", (unsigned int)pthread_self(), g_Flag);fprintf(stdout, "Leave thread1\n");pthread_t tid = pthread_self();fprintf(stdout, "thread1 tid = %u\n", tid);pthread_join(tid, nullptr);fprintf(stdout, "\n");return nullptr;
}void *thread2(void *arg)
{fprintf(stdout, "Enter thread2\n");fprintf(stdout, "thread2 id: %u , g_Flag: %d\n", (unsigned int)pthread_self(), g_Flag);if (sem_wait(&sem_mutex) != 0) {fprintf(stderr, "thread2 sem_wait fail\n");return nullptr;}if (g_Flag == 1)sem_post(&sem_syn);g_Flag = 2;if (sem_post(&sem_mutex) != 0) {fprintf(stderr, "thread2 sem_post fail\n");return nullptr;}fprintf(stdout, "thread2 id: %u , g_Flag: %d\n", (unsigned int)pthread_self(), g_Flag);fprintf(stdout, "Leave thread2\n");pthread_t tid = pthread_self();fprintf(stdout, "thread2 tid = %u\n", tid);pthread_join(tid, nullptr);fprintf(stdout, "\n");return nullptr;
}} // namespaceint main()
{pthread_t tid1, tid2;sem_init(&sem_mutex, 0, 1);sem_init(&sem_syn, 0, 0);fprintf(stdout, "Inter main!\n");int ret = pthread_create(&tid2, nullptr, thread2, nullptr);if (ret != 0) {fprintf(stderr, "%s, %d\n", __func__, strerror(ret));return -1;}ret = pthread_create(&tid1, nullptr, thread1, nullptr);if (ret != 0) {fprintf(stderr, "%s, %d\n", __func__, strerror(ret));return -1;}fprintf(stdout, "Leave main!\n\n");sem_wait(&sem_syn); // 同步等待，阻塞return 0;
}

test_thread_sem1.cpp：

// reference: http://man7.org/linux/man-pages/man3/sem_wait.3.html#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <semaphore.h>
#include <time.h>
#include <assert.h>
#include <errno.h>
#include <signal.h>namespace {sem_t sem;#define handle_error(msg) \do { perror(msg); exit(EXIT_FAILURE); } while (0)void handler(int sig)
{write(STDOUT_FILENO, "sem_post() from handler\n", 24);if (sem_post(&sem) == -1) {write(STDERR_FILENO, "sem_post() failed\n", 18);_exit(EXIT_FAILURE);}
}} // namespaceint main(int argc, char *argv[])
{struct sigaction sa;struct timespec ts;int s;if (argc != 3) {fprintf(stderr, "Usage: %s <alarm-secs> <wait-secs>\n", argv[0]);exit(EXIT_FAILURE);}if (sem_init(&sem, 0, 0) == -1)handle_error("sem_init");/* Establish SIGALRM handler; set alarm timer using argv[1] */sa.sa_handler = handler;sigemptyset(&sa.sa_mask);sa.sa_flags = 0;if (sigaction(SIGALRM, &sa, nullptr) == -1)handle_error("sigaction");alarm(atoi(argv[1]));/* Calculate relative interval as current time plusnumber of seconds given argv[2] */if (clock_gettime(CLOCK_REALTIME, &ts) == -1)handle_error("clock_gettime");ts.tv_sec += atoi(argv[2]);printf("main() about to call sem_timedwait()\n");while ((s = sem_timedwait(&sem, &ts)) == -1 && errno == EINTR)continue;       /* Restart if interrupted by handler *//* Check what happened */if (s == -1) {if (errno == ETIMEDOUT)printf("sem_timedwait() timed out\n");elseperror("sem_timedwait");} elseprintf("sem_timedwait() succeeded\n");exit((s == 0) ? EXIT_SUCCESS : EXIT_FAILURE);
}// ./test_thread_sem1 2 3
// ./test_thread_sem1 2 1

test_thread_sem2.cpp：

// reference: https://mahaveerdarade.wordpress.com/2013/09/16/semaphores-in-linux-sem_wait-sem_post-code-examples-in-c/
#include <stdlib.h>
#include <pthread.h> 
#include <stdio.h>
#include <semaphore.h>
#include <unistd.h>namespace {int cnt = 0;
int a[] = {1,2,3,4,5,6,7,8,9};
char arr[] = {'a','b','c','d','e','f','g','h','j'};
sem_t s1;void* pc(void* arg)
{int i = 0;while (i < 9) {fprintf(stdout, "---- Line: %d\n", __LINE__);sem_wait(&s1);while (cnt == 0) {//fprintf(stdout, "---- Line: %d\n", __LINE__);	// 注意：打开此条语句对结果的影响sem_post(&s1);}fprintf(stdout, "%c\n", arr[i++]);sleep(1);cnt=0;sem_post(&s1);}return nullptr;
}void* pi(void* arg)
{int i = 0;while (i < 9) {sleep(2);		fprintf(stdout, "++++ Line: %d\n", __LINE__);		sem_wait(&s1);while (cnt == 1) {sem_post(&s1);}fprintf(stdout, "%d\n", a[i++]);sleep(1);cnt = 1;sem_post(&s1);}return nullptr;
}} // namespaceint main()
{pthread_t t1,t2;sem_init(&s1, 0, 1);pthread_create(&t1, nullptr, pc, nullptr);pthread_create(&t2, nullptr, pi, nullptr);pthread_join(t1, nullptr);pthread_join(t2, nullptr);sem_destroy(&s1);return 0;
}

test_thread_sem3.cpp：

// reference: http://www.amparo.net/ce155/sem-ex.html
/* Includes */
#include <unistd.h>     /* Symbolic Constants */
#include <sys/types.h>  /* Primitive System Data Types */ 
#include <errno.h>      /* Errors */
#include <stdio.h>      /* Input/Output */
#include <stdlib.h>     /* General Utilities */
#include <pthread.h>    /* POSIX Threads */
#include <string.h>     /* String handling */
#include <semaphore.h>  /* Semaphore */namespace {/* global vars */
/* semaphores are declared global so they can be accessed in main() and in thread routine,here, the semaphore is used as a mutex */
sem_t mutex;
int counter = 0; /* shared variable *//* prototype for thread routine */
void* handler(void* ptr)
{int x; x = *((int *)ptr);printf("Thread %d: Waiting to enter critical region...\n", x);sem_wait(&mutex);       /* down semaphore *//* START CRITICAL REGION */printf("Thread %d: Now in critical region...\n", x);printf("Thread %d: Counter Value: %d\n", x, counter);printf("Thread %d: Incrementing Counter...\n", x);counter++;printf("Thread %d: New Counter Value: %d\n", x, counter);printf("Thread %d: Exiting critical region...\n", x);/* END CRITICAL REGION */    sem_post(&mutex);       /* up semaphore */pthread_exit(0); /* exit thread */
}} // namespaceint main()
{int i[2];pthread_t thread_a;pthread_t thread_b;i[0] = 0; /* argument to threads */i[1] = 1;sem_init(&mutex, 0, 1);      /* initialize mutex to 1 - binary semaphore *//* second param = 0 - semaphore is local *//* Note: you can check if thread has been successfully created by checking return value ofpthread_create */                                 pthread_create(&thread_a, NULL, handler, (void*)&i[0]);pthread_create(&thread_b, NULL, handler, (void*)&i[1]);pthread_join(thread_a, NULL);pthread_join(thread_b, NULL);sem_destroy(&mutex); /* destroy semaphore *//* exit */  exit(0);
} /* main() */

GitHub： https://github.com/fengbingchun/Linux_Code_Test