linux多線程編程詳解教程(線程通過訊號量實現通訊代碼)_linux shell

線程分類

線程按照其調度者可以分為使用者級線程和核心級線程兩種。

（1）使用者級線程
使用者級線程主要解決的是環境切換的問題，它的調度演算法和調度過程全部由使用者自行選擇決定，在運行時不需要特定的核心支援。在這裡，作業系統往往會提供一個使用者空間的線程庫，該線程庫提供了線程的建立、調度、撤銷等功能，而核心仍然僅對進程進行管理。如果一個進程中的某一個線程調用了一個阻塞的系統調用，那麼該進程包括該進程中的其他所有線程也同時被阻塞。這種使用者級線程的主要缺點是在一個進程中的多個線程的調度中無法發揮多處理器的優勢。

（2）核心級線程
這種線程允許不同進程中的線程按照同一相對優先調度方法進行調度，這樣就可以發揮多處理器的並發優勢。
現在大多數系統都採用使用者級線程與核心級線程並存的方法。一個使用者級線程可以對應一個或幾個核心級線程，也就是“一對一”或“多對一”模型。這樣既可滿足多處理機系統的需要，也可以最大限度地減少調度開銷。

Linux的線程實現是在核外進行的，核內提供的是建立進程的介面do_fork()。核心提供了兩個系統調用clone()和fork()，最終都用不同的參數調用do_fork()核內API。當然，要想實現線程，沒有核心對多進程（其實是輕量級進程）共用資料區段的支援是不行的，因此，do_fork()提供了很多參數，包括CLONE_VM（共用記憶體空間）、CLONE_FS（共用檔案系統資訊）、 CLONE_FILES（共用檔案描述符表）、CLONE_SIGHAND（共用訊號控制代碼表）和CLONE_PID（共用進程ID，僅對核內進程，即0號進程有效）。當使用fork系統調用時，核心調用do_fork()不使用任何共用屬性，進程擁有獨立的運行環境，而使用 pthread_create()來建立線程時，則最終設定了所有這些屬性來調用__clone()，而這些參數又全部傳給核內的do_fork()，從而建立的“進程”擁有共用的運行環境，只有棧是獨立的，由__clone()傳入。

Linux線程在核內是以輕量級進程的形式存在的，擁有獨立的進程表項，而所有的建立、同步、刪除等操作都在核外pthread庫中進行。pthread 庫使用一個管理線程（__pthread_manager()，每個進程獨立且唯一）來管理線程的建立和終止，為線程分配線程ID，發送線程相關的訊號（比如Cancel），而主線程（pthread_create()）的調用者則通過管道將請求資訊傳給管理線程。

主要函數說明

1．線程的建立和退出

pthread_create 線程建立函數
int pthread_create (pthread_t * thread_id,__const pthread_attr_t * __attr,void *(*__start_routine) (void *),void *__restrict __arg);

線程建立函數第一個參數為指向線程標識符的指標，第二個參數用來設定線程屬性，第三個參數是線程運行函數的起始地址，最後一個參數是運行函數的參數。這裡，我們的函數thread 不需要參數，所以最後一個參數設為空白指標。第二個參數我們也設為空白指標，這樣將產生預設屬性的線程。當建立線程成功時，函數返回0，若不為0 則說明建立線程失敗，常見的錯誤傳回碼為EAGAIN 和EINVAL。前者表示系統限制建立新的線程，例如線程數目過多了；後者表示第二個參數代表的線程屬性值非法。建立線程成功後，新建立的線程則運行參數三和參數四確定的函數，原來的線程則繼續運行下一行代碼。

pthread_join 函數，來等待一個線程的結束。
函數原型為：int pthread_join (pthread_t __th, void **__thread_return)
第一個參數為被等待的線程標識符，第二個參數為一個使用者定義的指標，它可以用來儲存被等待線程的傳回值。這個函數是一個線程阻塞的函數，調用它的函數將一直等待到被等待的線程結束為止，當函數返回時，被等待線程的資源被收回。線程只能被一個線程等待終止,並且應處於joinable狀態(非detached)。

pthread_exit 函數
一個線程的結束有兩種途徑，一種是線程啟動並執行函數結束了，調用它的線程也就結束了；
另一種方式是通過函數pthread_exit 來實現。它的函數原型為：void pthread_exit (void *__retval)唯一的參數是函數的傳回碼，只要pthread_join 中的第二個參數thread_return 不是NULL，這個值將被傳遞給thread_return。最後要說明的是，一個線程不能被多個線程等待，否則第一個接收到訊號的線程成功返回，其餘調用pthread_join 的線程則返回錯誤碼ESRCH。

2．線程屬性

pthread_create函數的第二個參數線程的屬性。將該值設為NULL，也就是採用預設屬性，線程的多項屬性都是可以更改的。這些屬性主要包括綁定屬性、分離屬性、堆棧地址、堆棧大小、優先順序。其中系統預設的屬性為非綁定、非分離、預設1M 的堆棧、與父進程同樣層級的優先順序。下面首先對綁定屬性和分離屬性的基本概念進行講解。

綁定屬性：Linux中採用“一對一”的線程機制，也就是一個使用者線程對應一個核心線程。綁定屬性就是指一個使用者線程固定地分配給一個核心線程，因為CPU時間片的調度是面向核心線程 （也就是輕量級進程）的，因此具有綁定屬性的線程可以保證在需要的時候總有一個核心線程與之對應。而與之相對的非綁定屬性就是指使用者線程和核心線程的關係不是始終固定的，而是由系統來控制分配的。

分離屬性：分離屬性是用來決定一個線程以什麼樣的方式來終止自己。在非分離情況下，當一個線程結束時，它所佔用的系統資源並沒有被釋放，也就是沒有真正的終止。只有當pthread_join()函數返回時，建立的線程才能釋放自己佔用的系統資源。而在分離屬性情況下，一個線程結束時立即釋放它所佔有的系統資源。
這裡要注意的一點是，如果設定一個線程的分離屬性，而這個線程運行又非常快，那麼它很可能在pthread_create 函數返回之前就終止了，它終止以後就可能將線程號和系統資源移交給其他的線程使用，這時調用pthread_create 的線程就得到了錯誤的線程號。

設定綁定屬性：

int pthread_attr_init(pthread_attr_t *attr)
int pthread_attr_setscope(pthread_attr_t *attr, int scope)
int pthread_attr_getscope(pthread_attr_t *tattr, int *scope)
scope：PTHREAD_SCOPE_SYSTEM：綁定，此線程與系統中所有的線程競爭 PTHREAD_SCOPE_PROCESS：非綁定，此線程與進程中的其他線程競爭

設定分離屬性：

int pthread_attr_setdetachstate(pthread_attr_t *attr, int detachstate)
int pthread_attr_getdetachstate(const pthread_attr_t *tattr,int *detachstate)
detachstate PTHREAD_CREATE_DETACHED：分離 PTHREAD _CREATE_JOINABLE：非分離

設定調度策略：

int pthread_attr_setschedpolicy(pthread_attr_t * tattr, int policy)
int pthread_attr_getschedpolicy(pthread_attr_t * tattr, int *policy)
policy SCHED_FIFO：先入先出 SCHED_RR：迴圈 SCHED_OTHER：實現定義的方法

設定優先權：

int pthread_attr_setschedparam (pthread_attr_t *attr, struct sched_param *param)
int pthread_attr_getschedparam (pthread_attr_t *attr, struct sched_param *param)

3．線程存取控制

1）互斥鎖（mutex）
通過鎖機制實現線程間的同步。同一時刻只允許一個線程執行一個關鍵區段的代碼。

1 int pthread_mutex_init(pthread_mutex_t *mutex,const pthread_mutex_attr_t *mutexattr);
2 int pthread_mutex_lock(pthread_mutex_t *mutex);
3 int pthread_mutex_unlock(pthread_mutex_t *mutex);
4 int pthread_mutex_destroy(pthread_mutex_t *mutex);

(1)先初始化鎖init()或靜態賦值pthread_mutex_t mutex=PTHREAD_MUTEX_INITIALIER
(2)加鎖,lock,trylock,lock阻塞等待鎖,trylock立即返回EBUSY
(3)解鎖,unlock需滿足是加鎖狀態,且由加鎖線程解鎖
(4)清除鎖,destroy(此時鎖必需unlock,否則返回EBUSY)

mutex 分為遞迴(recursive) 和非遞迴(non-recursive)兩種，這是POSIX 的叫法，另外的名字是可重新進入(Reentrant) 與非可重新進入。這兩種mutex 作為線程間(inter-thread) 的同步工具時沒有區別，它們的惟一區別在於：同一個線程可以重複對recursive mutex 加鎖，但是不能重複對non-recursive mutex 加鎖。
首選非遞迴mutex，絕對不是為了效能，而是為了體現設計意圖。non-recursive 和recursive 的效能差別其實不大，因為少用一個計數器，前者略快一點點而已。在同一個線程裡多次對non-recursive mutex 加鎖會立刻導致死結，我認為這是它的優點，能協助我們思考代碼對鎖的期求，並且及早（在編碼階段）發現問題。毫無疑問recursive mutex 使用起來要方便一些，因為不用考慮一個線程會自己把自己給鎖死了，我猜這也是Java 和Windows 預設提供recursive mutex 的原因。（Java 語言內建的intrinsic lock 是可重新進入的，它的concurrent 庫裡提供ReentrantLock，Windows的CRITICAL_SECTION 也是可重新進入的。似乎它們都不提供輕量級的non-recursive mutex。）

2）條件變數（cond）
利用線程間共用的全域變數進行同步的一種機制。

1 int pthread_cond_init(pthread_cond_t *cond,pthread_condattr_t *cond_attr);
2 int pthread_cond_wait(pthread_cond_t *cond,pthread_mutex_t *mutex);
3 int pthread_cond_timedwait(pthread_cond_t *cond,pthread_mutex_t *mutex,const timespec *abstime);
4 int pthread_cond_destroy(pthread_cond_t *cond); 
5 int pthread_cond_signal(pthread_cond_t *cond);
6 int pthread_cond_broadcast(pthread_cond_t *cond);  //解除所有線程的阻塞

(1)初始化. init()或者pthread_cond_t cond=PTHREAD_COND_INITIALIER;屬性置為NULL
(2)等待條件成立. pthread_cond_wait,pthread_cond_timedwait.
wait()釋放鎖,並阻塞等待條件變數為真
timedwait()設定等待時間,仍未signal,返回ETIMEOUT(加鎖保證只有一個線程wait)
(3)啟用條件變數:pthread_cond_signal,pthread_cond_broadcast(啟用所有等待線程)
(4)清除條件變數:destroy; 無線程等待,否則返回EBUSY

複製代碼 代碼如下:

int pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex);
int pthread_cond_timedwait(pthread_cond_t *cond, pthread_mutex_t *mutex, const struct timespec *abstime);

這兩個函數一定要在mutex的鎖定地區內使用。

調用 pthread_cond_signal() 釋放被條件阻塞的線程時，如果沒有任何線程基於條件變數阻塞，則調用pthread_cond_signal()不起作用。而對於 Windows，當調用 SetEvent 觸發 Auto-reset 的 Event 條件時，如果沒有被條件阻塞的線程，那麼此函數仍然起作用，條件變數會處在觸發狀態。

Linux下生產者消費者問題（使用互斥鎖和條件變數）:

複製代碼 代碼如下:

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include "pthread.h"

#define BUFFER_SIZE 16

struct prodcons 
{ 
int buffer[BUFFER_SIZE]; 
pthread_mutex_t lock;  //mutex ensuring exclusive access to buffer
int readpos,writepos;  //position for reading and writing
pthread_cond_t notempty;  //signal when buffer is not empty
pthread_cond_t notfull;  //signal when buffer is not full
}; 

//initialize a buffer
void init(struct prodcons* b) 
{ 
pthread_mutex_init(&b->lock,NULL); 
pthread_cond_init(&b->notempty,NULL); 
pthread_cond_init(&b->notfull,NULL); 
b->readpos = 0; 
b->writepos = 0; 
} 

//store an integer in the buffer
void put(struct prodcons* b, int data) 
{ 
pthread_mutex_lock(&b->lock); 
//wait until buffer is not full
while((b->writepos+1)%BUFFER_SIZE == b->readpos) 
{ 
printf("wait for not full\n"); 
pthread_cond_wait(&b->notfull,&b->lock); 
}
b->buffer[b->writepos] = data; 
b->writepos++;
b->writepos %= BUFFER_SIZE;
pthread_cond_signal(&b->notempty); //signal buffer is not empty
pthread_mutex_unlock(&b->lock); 
}

//read and remove an integer from the buffer
int get(struct prodcons* b) 
{ 
int data; 
pthread_mutex_lock(&b->lock); 
//wait until buffer is not empty
while(b->writepos == b->readpos) 
{ 
printf("wait for not empty\n"); 
pthread_cond_wait(&b->notempty,&b->lock); 
}
data=b->buffer[b->readpos]; 
b->readpos++;
b->readpos %= BUFFER_SIZE;
pthread_cond_signal(&b->notfull);  //signal buffer is not full
pthread_mutex_unlock(&b->lock); 
return data;
}

#define OVER -1

struct prodcons buffer; 

void * producer(void * data) 
{ 
int n; 
for(n=0; n<50; ++n) 
{
printf("put-->%d\n",n); 
put(&buffer,n); 
} 
put(&buffer,OVER); 
printf("producer stopped\n"); 
return NULL; 
} 

void * consumer(void * data) 
{ 
int n; 
while(1) 
{ 
int d = get(&buffer); 
if(d == OVER) break; 
printf("get-->%d\n",d); 
}
printf("consumer stopped\n"); 
return NULL; 
} 

int main() 
{ 
pthread_t tha,thb; 
void * retval; 

init(&buffer); 
pthread_creare(&tha,NULL,producer,0); 
pthread_creare(&thb,NULL,consumer,0); 

pthread_join(tha,&retval); 
pthread_join(thb,&retval); 

return 0; 
}

3）訊號量
如同進程一樣，線程也可以通過訊號量來實現通訊，雖然是輕量級的。

訊號量函數的名字都以"sem_"打頭。線程使用的基本訊號量函數有四個。

複製代碼 代碼如下:

#include <semaphore.h>
int sem_init(sem_t *sem , int pshared, unsigned int value);

這是對由sem指定的訊號量進行初始化，設定好它的共用選項（linux只支援為0，即表示它是當前進程的局部訊號量），然後給它一個初始值VALUE。

兩個原子操作函數：這兩個函數都要用一個由sem_init調用初始化的訊號量對象的指標做參數。

複製代碼 代碼如下:

int sem_wait(sem_t *sem); //給訊號量減1，對一個值為0的訊號量調用sem_wait，這個函數將會等待直到有其它線程使它不再是0為止。
int sem_post(sem_t *sem); //給訊號量的值加1

int sem_destroy(sem_t *sem);

這個函數的作用是再我們用完訊號量後都它進行清理。歸還自己佔有的一切資源。

用訊號量實現生產者消費者：

這裡使用4個訊號量，其中兩個訊號量occupied和empty分別用於解決生產者和消費者線程之間的同步問題，pmut用於多個生產者之間互斥問題，cmut是用於多個消費者之間互斥問題。其中empty初始化為N（有界緩區的空間元數），occupied初始化為0，pmut和cmut初始化為1。

參考代碼：

複製代碼 代碼如下:

#define BSIZE 64

typedef struct
{
char buf[BSIZE];
sem_t occupied;
sem_t empty;
int nextin;
int nextout;
sem_t pmut;
sem_t cmut;
}buffer_t;

buffer_t buffer;

void init(buffer_t * b)
{
sem_init(&b->occupied, 0, 0);
sem_init(&b->empty,0, BSIZE);
sem_init(&b->pmut, 0, 1);
sem_init(&b->cmut, 0, 1);
b->nextin = b->nextout = 0;
}

void producer(buffer_t *b, char item)
{
sem_wait(&b->empty);
sem_wait(&b->pmut);
b->buf[b->nextin] = item;
b->nextin++;
b->nextin %= BSIZE;
sem_post(&b->pmut);
sem_post(&b->occupied);
}

char consumer(buffer_t *b)
{
char item;
sem_wait(&b->occupied);
sem_wait(&b->cmut);
item = b->buf[b->nextout];
b->nextout++;
b->nextout %= BSIZE;
sem_post(&b->cmut);
sem_post(&b->empty);
return item;
}