使用信号量控制Linux线程同步

linux中两种基本的同步方法是信号量和互斥量。这两种方法很相似,而且它们可以相互通过对方来实现。


信号量概述


下面介绍用信号量进行同步。


信号量概念由荷兰科学家Dijkstra首先提出。信号量是一个特殊类型的变量,它可以被增加或者减少。但对其的关键访问被保证是原子操作,即使在一个多线程程序中也是如此。


信号量有两种类型:

(1)二进制信号量。它只有0和1两种取值。

(2)计数信号量。它可以有更大的取值范围。

如果要用信号量来保护一段代码,使其每次只能被一个执行线程运行,就要用到二进制信号量、。

如果要允许有限数目的线程执行一段指定的代码,就需要用到计数信号量。


由于计数信号量并不常用,而且它实际上仅仅是二进制信号量的一种扩展,这里之介绍二进制信号量。


信号量的相关函数

信号量函数的名字都以sem_开头。线程中使用的基本函数有4个。

注意,需要包含头文件:

[cpp]  view plain  copy
 print ?
  1. #include<semaphore.h>  

创建信号量

[cpp]  view plain  copy
 print ?
  1. int sem_init(sem_t *sem, int pshared, unsigned int value);  
函数解释:

sem_init() 初始化一个定位在 sem 的匿名信号量。value 参数指定信号量的初始值。 pshared 参数指明信号量是由进程内线程共享,还是由进程之间共享。如果 pshared 的值为 0,那么信号量将被进程内的线程共享,并且应该放置在所有线程都可见的地址上(如全局变量,或者堆上动态分配的变量)。
  如果 pshared 是非零值,那么信号量将在进程之间共享,并且应该定位共享内存区域(见 shm_open(3)、mmap(2) 和 shmget(2))。(因为通过 fork(2) 创建的孩子继承其父亲的内存映射,因此它也可以见到这个信号量。)所有可以访问共享内存区域的进程都可以使用sem_post(3)、sem_wait(3) 等等操作信号量。初始化一个已经初始的信号量其结果未定义。


在工作中,你会遇到两种信号量:二进制信号量和计数信号量。二进制信号量只有0和1两种取值,而计数信号量则有更大的取值范围。如果某个共享资源只能被一个线程访问,那么二进制信号量则是最好的打算;如果有多个线程需要访问共享资源呢,使用计数信号量则是个好的主意。

返回值
  sem_init() 成功时返回 0;错误时,返回 -1,并把 errno 设置为合适的值。
错误
  EINVAL
  value 超过 SEM_VALUE_MAX。
  ENOSYS
  pshared 非零,但系统还没有支持进程共享的信号量。


下面是控制信号量的两个函数:

信号量减一操作

[cpp]  view plain  copy
 print ?
  1. int sem_wait(sem_t * sem);  
函数说明
  sem_wait函数也是一个原子操作,它的作用是从信号量的值减去一个“1”,但它永远会先等待该信号量为一个非零值才开始做减法。也就是说,如果你对一个值为2的信号量调用sem_wait(),线程将会继续执行,这信号量的值将减到1。如果对一个值为0的信号量调用sem_wait(),这个函数就 会地等待直到有其它线程增加了这个值使它不再是0为止。如果有两个线程都在sem_wait()中等待同一个信号量变成非零值,那么当它被第三个线程增加 一个“1”时,等待线程中只有一个能够对信号量做减法并继续执行,另一个还将处于等待状态。
返回值
  所有这些函数在成功时都返回 0;错误保持信号量值没有更改,-1 被返回,并设置 errno 来指明错误。
错误
  EINTR
  这个调用被信号处理器中断,
  EINVAL

  sem 不是一个有效的信号量。


信号量加一操作

[cpp]  view plain  copy
 print ?
  1. int sem_post(sem_t * sem);   
说明
  sem_post函数的作用是给信号量的值加上一个“1”,它是一个“原子操作”---即同时对同一个信号量做加“1”操作的两个线程是不会冲突的;而同 时对同一个文件进行读、加和写操作的两个程序就有可能会引起冲突。信号量的值永远会正确地加一个“2”--因为有两个线程试图改变它。
返回值
  sem_post() 成功时返回 0;错误时,信号量的值没有更改,-1 被返回,并设置 errno 来指明错误。
错误
  EINVAL
  sem 不是一个有效的信号量。
  EOVERFLOW
  信号量允许的最大值将要被超过。


清理信号量

[cpp]  view plain  copy
 print ?
  1. int sem_destroy (sem_t *sem);  
 这个函数也使用一个信号量指针做参数,归还自己战胜的一切资源。在清理信号量的时候如果还有线程在等待它,用户就会收到一个错误。
        与其它的函数一样,这些函数在成功时都返回“0”。


信号量的使用

关于信号量操作的API函数都已经介绍完了,现在就通过一个小的程序来用一用这些函数,通过实际的编码,能更好的掌握如何使用这些函数。在主线程中,我们创建子线程,然后主线程等待直到有文本输入,然后调用sem_post来增加信号量的值,这样就会立刻使子线程从sem_wait的等待中返回并开始执行。线程函数在把字符串的小写字母变成大写并统计输入的字符数量之后,它再次调用sem_wait并再次被阻塞,直到主线程再次调用sem_post增加信号量的值。

接下来代码要完成这样一个功能:

  • 主线程从标准输入终端读取输入的内容;
  • 子线程1将主线程中输入的内容转换成大写。

来吧,我们就按照上面的功能来编码实现一下。功能简单,我准备这样实现:

  • 在主线程初始化信号量为0;
  • 当主线程输入完成以后,调用sem_post增加信号量的值;
  • 在子线程1中,调用sem_wait,等到信号量的值为1时,就减少信号量的值,然后执行线程中的代码
/*
 * semphore.c
 *
 */

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

sem_t sem;	//信号量对象
#define SIZE 1024
char work_area[SIZE];	// 主线程和子线程都访问的共享资源

void *threadFunc(void *arg);

int main(void)
{
	int res;
	pthread_t tid;
	void *thread_result;

	res = sem_init(&sem,0,0);	//信号量初始化为0
	if(res != 0)
	{
		perror("Semaphore init failed");
		exit(1);
	}

	res = pthread_create(&tid,NULL,threadFunc,NULL);
	if(res != 0)
	{
		perror("Thread create failed");
		exit(1);
	}

	printf("Input some text.Enter 'end' to finish\n");
	while(strncmp(work_area,"end",3) != 0)	//输入没有结束
	{
		fgets(work_area, SIZE, stdin);
		sem_post(&sem);	//给信号量值加1
		int semValue = 0;
		res = sem_getvalue(&sem, &semValue); // 获得信号量当前的值
		printf("%s:Semaphore count: %d\n", __FUNCTION__, semValue);
	}

	printf("waiting for thread to finish\n");
	res = pthread_join(tid, &thread_result);
	if(res != 0)
	{
		perror("Thread join failed");
		exit(1);
	}
	printf("Thread joined.\n");
	sem_destroy(&sem);	//销毁信号量对象
	exit(0);
}

void *threadFunc(void *arg)
{
	int i;
	sem_wait(&sem);	//将信号量值减1
	int semValue = 0;
	sem_getvalue(&sem, &semValue);
	printf("%s:Semaphore count: %d\n", __FUNCTION__, semValue);

	while(strncmp(work_area, "end", 3) != 0)
	{
		for(i=0;work_area[i]!='\0';i++)
		{
			if(work_area[i]>='a' && work_area[i]<='z')
				work_area[i] = work_area[i] - 32;
		}
		printf("After convtered: %s\n", work_area);
		sem_wait(&sem);	// 再次阻塞,等待输入
	}

	pthread_exit(NULL);
}<strong>
</strong>

运行结果:
localhost@ubuntu:~/Practice/pthread$ ./a.out 
Input some text.Enter 'end' to finish
xiajun
main:Semaphore count: 1
threadFunc:Semaphore count: 0
After convtered: XIAJUN

wehfoukj
main:Semaphore count: 1
After convtered: WEHFOUKJ

end
main:Semaphore count: 1
waiting for thread to finish
Thread joined.

分析此信号量同步程序的缺陷
但是这个程序有一点点的小问题,就是这个程序依赖接收文本输入的时间足够长,这样子线程才有足够的时间在主线程还未准备好给它更多的单词去处理和统计之前处理和统计出工作区中字符的个数。所以当我们连续快速地给它两组不同的单词去统计时,子线程就没有足够的时间支执行,但是信号量已被增加不止一次,所以字符统计线程(子线程)就会反复处理和统计字符数目,并减少信号量的值,直到它再次变成0为止。

为了更加清楚地说明上面所说的情况,修改主线程的while循环中的代码,如下:
[cpp]  view plain  copy
 print ? 在CODE上查看代码片 派生到我的代码片
  1. printf("Input some text. Enter 'end'to finish...\n");  
  2. while(strcmp("end\n", msg) != 0)  
  3. {  
  4.     if(strncmp("TEST", msg, 4) == 0)  
  5.     {  
  6.         strcpy(msg, "copy_data\n");  
  7.         sem_post(&sem);  
  8.     }  
  9.     fgets(msg, MSG_SIZE, stdin);  
  10.     //把信号量加1  
  11.     sem_post(&sem);  
  12. }  
重新编译程序,此时运行结果如下:



当我们输入TEST时,主线程向子线程提供了两个输入,一个是来自键盘的输入,一个来自主线程复数据到msg中,然后从运行结果可以看出,运行出现了异常,没有处理和统计从键盘输入TEST的字符串而却对复制的数据作了两次处理。原因如上面所述。

解决此缺陷的方法

解决方法有两个,一个就是再增加一个信号量,让主线程等到子线程处理统计完成之后再继续执行;另一个方法就是使用互斥量。
下面给出用增加一个信号量的方法,由于线程1还没有完成变换,主线程又开始了新的输入。所以,主线程必须要等到线程1完成以后才能进行新的输入,那么,我们就需要再加入一个信号量,当线程1完成转换时,增加该信号量;主线程完成输入以后,减少该信号量。再来试一试:
/*
 * semphore.c
 *
 */

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

sem_t sem;	//信号量对象
sem_t sem_add;	//控制线程1完成的信号量
#define SIZE 1024
char work_area[SIZE];	// 主线程和子线程都访问的共享资源

void *threadFunc(void *arg);

int main(void)
{
	int res;
	pthread_t tid;
	void *thread_result;

	res = sem_init(&sem,0,0);	//信号量初始化为0
	if(res != 0)
	{
		perror("Semaphore init failed");
		exit(1);
	}

	res = sem_init(&sem_add,0,0);
	if(res != 0)
	{
		perror("Semaphore init failed");
		exit(1);
	}

	res = pthread_create(&tid,NULL,threadFunc,NULL);
	if(res != 0)
	{
		perror("Thread create failed");
		exit(1);
	}

	printf("Input some text.Enter 'end' to finish\n");
	while(strncmp(work_area,"end",3) != 0)	//输入没有结束
	{
		if(strncmp("test",work_area,4) == 0)
		{
			strcpy(work_area,"copy_data\n");
			sem_post(&sem);
			sem_wait(&sem_add);	 //把sem_add的值减1,即等待子线程处理完成
		}
		fgets(work_area, SIZE, stdin);
		sem_post(&sem);	//给信号量值加1
		sem_wait(&sem_add);	 //把sem_add的值减1,即等待子线程处理完成
	}

	printf("waiting for thread to finish\n");
	res = pthread_join(tid, &thread_result);
	if(res != 0)
	{
		perror("Thread join failed");
		exit(1);
	}
	printf("Thread joined.\n");
	sem_destroy(&sem);	//销毁信号量对象
	sem_destroy(&sem_add);
	exit(0);
}

void *threadFunc(void *arg)
{
	int i;
	sem_wait(&sem);	//将信号量值减1

	while(strncmp(work_area, "end", 3) != 0)
	{
		for(i=0;work_area[i]!='\0';i++)
		{
			if(work_area[i]>='a' && work_area[i]<='z')
				work_area[i] = work_area[i] - 32;
		}
		printf("You input %d characters\n",strlen(work_area) - 1);
		printf("After convtered: %s\n", work_area);
		sem_post(&sem_add);	//把信号量加1,表明子线程处理完成
		sem_wait(&sem);	// 再次阻塞,等待输入
	}
	sem_post(&sem_add);	//把信号量加1,表明子线程处理完成
	pthread_exit(NULL);
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值