linux代码实操——信号量

信号量描述

信号量是一个特殊的变量，一般取正数值。它的值代表允许访问的资源数目，获取资源时，需要对信号量的值进行原子减一，该操作被称为 P 操作。当信号量值为 0 时，代表没有资源可用，P 操作会阻塞。释放资源时，需要对信号量的值进行原子加一，该操作被称为 V 操作。信号量主要用来同步进程。信号量的值如果只取 0,1，将其称为二值信号量。如果信号量的值大于 1，则称之为计数信号量。

临界资源：同一时刻，只允许被一个进程或线程访问的资源

临界区：访问临界资源的代码段

 

信号量使用

我们结合代码进行操作信号量的接口介绍：

1. #include <sys/sem.h>
2. #include <sys/types.h>
3. #include <sys/ipc.h>
4. /*
5. semget()创建或者获取已存在的信号量
6. semget()成功返回信号量的 ID， 失败返回-1
7. key：两个进程使用相同的 key 值，就可以使用同一个信号量
8. nsems：内核维护的是一个信号量集，在新建信号量时，其指定信号量集中信号
量的个数
9. semflg 可选： IPC_CREAT IPC_EXCL
10. */
11. int semget(key_t key, int nsems, int semflg);
12.
13. /*
14. semop()对信号量进行改变，做 P 操作或者 V 操作
15. semop()成功返回 0，失败返回-1
16. struct sembuf
17. {
18. unsigned short sem_num; //指定信号量集中的信号量下标
19. short sem_op; //其值为-1，代表 P 操作，其值为 1，代表 V 操作
20. short sem_flg; //SEM_UNDO
21. };
22. */
23. int semop(int semid, struct sembuf *sops, unsigned nsops);
24.
25. /*
26. semctl()控制信号量
27. semctl()成功返回 0，失败返回-1
28. cmd 选项： SETVAL IPC_RMID
29.
30. union semun
31. {
32. int val;
33. struct semid_ds *buf;
34. unsigned short *array;
35. struct seminfo *_buf;
36. };
37. */
38. int semctl(int semid, int semnum, int cmd, ...);

例题：进程 a 和进程 b 模拟访问打印机，进程 a 输出第一个字符‘a’表示开始使用打印 机，输出第二个字符‘a’表示结束使用，b 进程操作与 a 进程相同。（由于打印机同一时刻 只能被一个进程使用，所以输出结果不应该出现 abab），如图所示：

 

 

          封装信号量的接口：

sem.h 的代码如下：

 

1. #include <stdio.h>
2. #include <stdlib.h>
3. #include <unistd.h>
4. #include <string.h>
5. #include <assert.h>
6. #include <sys/sem.h>
7.
8. union semun{
9. int val;
10. };
11.
12. void sem_init();//创建/或者已存在的信号量
13. void sem_p();//p 减一
14. void sem_v();//v 加一
15. void sem_destroy();//销毁

sem.c 的代码如下：

1. #include "sem.h"
2.
3. static int semid = -1;
4.
5. void sem_init()//创建/或者已存在的信号量
6. {
7. semid = semget((key_t)1234,1,IPC_CREAT|IPC_EXCL|0600);
8. if ( semid == -1 )//已存在
9. {
10. semid = semget((key_t)1234,1,0600);
11. }
12. else
13. {
14. union semun a;
15.
16. a.val = 1;//信号量的初始值
17. if ( semctl(semid,0,SETVAL,a) == -1 )
18. {
19. perror("semctl error");
20. }
21. }
22. }
23.
24. void sem_p()//p 减一
25. {
26. struct sembuf buf;
27. buf.sem_num = 0;
28. buf.sem_op = -1;//p
29. buf.sem_flg = SEM_UNDO;
30.
31. if ( semop(semid,&buf,1) == -1 )
32. {
33. perror("semop p error");
34. }
35. }
36.
37. void sem_v()//v 加一
38. {
39. struct sembuf buf;
40. buf.sem_num = 0;
41. buf.sem_op = 1;//v
42. buf.sem_flg = SEM_UNDO;
43.
44. if ( semop(semid,&buf,1) == -1 )
45. {
46. perror("semop v error");
47. }
48. }
49.
50. void sem_destroy()//销毁
51. {
52. if ( semctl(semid,0,IPC_RMID) == -1 )
53. {
54. perror("semctl del error");
55. }
56. }

a.c 的代码如下：

1. #include <stdlib.h>
2. #include <unistd.h>
3. #include <string.h>
4. #include <assert.h>
5.
6. int main()
7. {
8. sem_init();
9. int i = 0;
10. for( ;i < 10; i++ )
11. {
12. sem_p();
13. printf("a");
14. fflush(stdout);
15. int n = rand() % 3;
16. sleep(n);
17. printf("a");
18. fflush(stdout);
19. sem_v();
20.
21. n = rand() % 3;
22. sleep(n);
23. }
24. }

b.c 的代码如下：

1. #include <stdlib.h>
2. #include <unistd.h>
3. #include <string.h>
4. #include <assert.h>
5.
6. int main()
7. {
8. sem_init();
9. int i = 0;
10. for( ;i < 10; i++ )
11. {
12. sem_p();
13. printf("b");
14. fflush(stdout);
15. int n = rand() % 3;
16. sleep(n);
17. printf("b");
18. fflush(stdout);
19. sem_v();
20.
21. n = rand() % 3;
22. sleep(n);
23. }
24. }

运行结果如下图所示，输出结果只截了部分：