1. #include <sys/types.h>
    2. #include <sys/ipc.h>
    3. #include <sys/sem.h>
    5. /*
    6.  * key：返回的 ID 与 key 有关系
    7.  * nsems：信号量的值
    8.  * semflg：创建标记
    9.  * return：成功返回信号量 ID，失败返回 -1，并设置 erron
    10.  */
    11. int semget(key_t key, int nsems, int semflg);
    13. #include <sys/types.h>
    14. #include <sys/ipc.h>
    15. #include <sys/sem.h>
    17. /*
    18.  * semid：信号量 ID
    19.  * sops：对信号量的操作
    20.  * nsops：要操作的信号数量
    21.  * return：成功返回 0，失败返回 -1，并设置 erron
    22.  */
    23. int semop(int semid, struct sembuf *sops, size_t nsops);
    25. struct sembuf {
    26.     /* 信号量的个数，除非使用多个信号量，否则设置为 0 */
    27.     unsigned short sem_num;  
    29.     /* 信号量的操作，-1 表示 p 操作，1 表示 v 操作 */
    30.     short          sem_op;   
    32.     /* 通常设置为 SEM_UNDO，使得 OS 能够跟踪信号量并在没有释放时自动释放 */
    33.     short          sem_flg;  
    34. };
    36. #include <sys/types.h>
    37. #include <sys/ipc.h>
    38. #include <sys/sem.h>
    40. /*
    41.  * semid：要设置的信号量 ID
    42.  * semnum：要设置的信号量的个数
    43.  * cmd：设置的属性
    44.  */
    45. int semctl(int semid, int semnum, int cmd, ...);
    47. union semun {
    48.     int              val;    /* Value for SETVAL */
    49.     struct semid_ds *buf;    /* Buffer for IPC_STAT, IPC_SET */
    50.     unsigned short  *array;  /* Array for GETALL, SETALL */
    51. };
    53. // 创建或获取一个信号量
    54. int sem_get(int sem_key) {
    55.     int sem_id = semget(sem_key, 1, IPC_CREAT | 0666);
    57.     if (sem_id == -1) {
    58.         printf("sem get failed.\n");
    59.         exit(-1);
    60.     } else {
    61.         printf("sem_id = %d\n", sem_id);
    62.         return sem_id;
    63.     }
    64. }
    66. // 初始化信号量
    67. int set_sem(int sem_id) {
    68.     union semun sem_union;  
    69.     sem_union.val = 1;  
    70.     if(semctl(sem_id, 0, SETVAL, sem_union) == -1) {
    71.         fprintf(stderr, "Failed to set sem\n");  
    72.         return 0;  
    73.     }
    74.     return 1;  
    75. }
    77.  void del_sem(int sem_id) {  
    78.     union semun sem_union;  
    79.     if(semctl(sem_id, 0, IPC_RMID, sem_union) == -1)  
    80.         fprintf(stderr, "Failed to delete sem, sem has been del.\n");
    81.  }
    83. // 减 1，加锁，P 操作
    84. void sem_down(int sem_id) {
    85.     if (-1 == semop(sem_id, &sem_lock, 1))
    86.         fprintf(stderr, "semaphore lock failed.\n");
    87. }
    89. // 加 1，解锁，V 操作
    90. void sem_up(int sem_id) {
    91.     if (-1 == semop(sem_id, &sem_unlock, 1))
    92.         fprintf(stderr, "semaphore unlock failed.\n");
    93. }
    95. int main(int argc, char **argv) {
    96.     int sem_id = sem_get(12);
    98.     // 第一次调用多加一个参数，第二次调用不加参数，仅在第一次调用时创建信号量
    99.     if (argc > 1 && (!set_sem(sem_id))) {
    100.         printf("set sem failed.\n");
    101.         return -1;
    102.     }
    104.     // P 操作
    105.     sem_down(sem_id);
    106.     printf("sem lock...\n");
    108.     printf("do something...\n");
    109.     sleep(10);
    111.     // V 操作
    112.     sem_up(sem_id);
    113.     printf("sem unlock...\n");
    115.     // 第二次调用后删除信号量
    116.     if (argc == 1)
    117.         del_sem(sem_id);
    119.     return 0;
    120. }