技術標籤：# C/C++

執行緒同步

當兩個執行緒同時執行時，不可避免同時操作同一個變數或者檔案等，所以需要一組機制來確保它們能正確的執行。
訊號量分為二值訊號量與計數訊號量；

用訊號量進行同步

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

sem_t bin_sem;

#define WORK_SIZE 1024
char
 
 work_area[WORK_SIZE];

void *thread_function(void *arg)
{
    sem_wait(&bin_sem);
    while (strncmp("end", work_area, 3) != 0) {
        printf("You input %ld characters\n", strlen(work_area) - 1);
        sem_wait(&bin_sem);
    }
    pthread_exit(NULL);
}

int main()
{
 

    int res;
    pthread_t a_thread;
    void *thread_result;

    res = sem_init(&bin_sem, 0, 0); /* 初始化訊號量，並且設定初始值為0 */
    if (res != 0) {
        perror("Semaphore initialization failed");
        exit(EXIT_FAILURE);
    }

    res = pthread_create(&a_thread, NULL, thread_function, NULL
 
); /* 建立新執行緒 */
    if (res != 0) {
        perror("Thread creation failed");
        exit(EXIT_FAILURE);
    }

    printf("Input some text, Enter 'end' to finish\n");
    while(strncmp("end", work_area, 3) != 0) {               /* 當工作區內不是以end開頭的字串時...*/
        fgets(work_area, WORK_SIZE, stdin);                  /* 從標準輸入獲取輸入到worl_area */
        sem_post(&bin_sem);                                  /* 訊號量+1 */
    }

    printf("\nWaiting for thread to finish...\n");
    res = pthread_join(a_thread, &thread_result);         /* 等待執行緒結束 */
    if (res != 0) {
        perror("Thread join failed");
        exit(EXIT_FAILURE);
    }
    printf("Thread joined\n");
    sem_destroy(&bin_sem);                               /* 銷燬訊號量 */
    exit(EXIT_SUCCESS);
}