linux中两种基本的同步方法是信号量和互斥量。这两种方法很相似,而且它们可以相互通过对方来实现。
信号量概述
下面介绍用信号量进行同步。
信号量概念由荷兰科学家Dijkstra首先提出。信号量是一个特殊类型的变量,它可以被增加或者减少。但对其的关键访问被保证是原子操作,即使在一个多线程程序中也是如此。
信号量有两种类型:
(1)二进制信号量。它只有0和1两种取值。
(2)计数信号量。它可以有更大的取值范围。
如果要用信号量来保护一段代码,使其每次只能被一个执行线程运行,就要用到二进制信号量、。
如果要允许有限数目的线程执行一段指定的代码,就需要用到计数信号量。
由于计数信号量并不常用,而且它实际上仅仅是二进制信号量的一种扩展,这里之介绍二进制信号量。
信号量的相关函数
信号量函数的名字都以sem_开头。线程中使用的基本函数有4个。
注意,需要包含头文件:
- #include<semaphore.h>
创建信号量
- 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 非零,但系统还没有支持进程共享的信号量。
下面是控制信号量的两个函数:
信号量减一操作
- 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 不是一个有效的信号量。
信号量加一操作
- int sem_post(sem_t * sem);
sem_post函数的作用是给信号量的值加上一个“1”,它是一个“原子操作”---即同时对同一个信号量做加“1”操作的两个线程是不会冲突的;而同 时对同一个文件进行读、加和写操作的两个程序就有可能会引起冲突。信号量的值永远会正确地加一个“2”--因为有两个线程试图改变它。
返回值
sem_post() 成功时返回 0;错误时,信号量的值没有更改,-1 被返回,并设置 errno 来指明错误。
错误
EINVAL
sem 不是一个有效的信号量。
EOVERFLOW
信号量允许的最大值将要被超过。
清理信号量
- 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>
运行结果:
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.
- printf("Input some text. Enter 'end'to finish...\n");
- while(strcmp("end\n", msg) != 0)
- {
- if(strncmp("TEST", msg, 4) == 0)
- {
- strcpy(msg, "copy_data\n");
- sem_post(&sem);
- }
- fgets(msg, MSG_SIZE, stdin);
- //把信号量加1
- sem_post(&sem);
- }
/*
* 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);
}