apue学习笔记（第十一章 线程）

本章将进一步深入理解进程，了解如何使用多个控制线程（简单得说就是线程）在单进程环境中执行多个任务。

 

 

线程概念

每个线程都包含有表示执行环境所必须的信息：线程ID、一组寄存器值、栈、调度优先级和策略、信号屏蔽字、errno变量以及线程私有数据。

一个进程的所有信息对该进程的所有线程都是共享的，包括可执行程序的代码、程序的全局内存和堆内存、栈以及文件描述符。

 

 

线程标识

每个线程都有一个线程ID，线程ID只有在它所属的进程上下文中才有意义。

可以使用下面函数来对两个线程ID进行比较

#include <pthread.h>int pthread_equal(pthread_t tid1,pthread_t tid2);

可以通过pthread_self函数获得自身的线程ID

#include <pthread.h>pthread_t pthread_self(void);

 

 

 

线程创建

#include <pthread.h>int pthread_create(pthread_t *restrict tidp,const pthread_attr_t *restrict attr,void *(*start_rtn)(void *),void *restrict arg);

当pthread_create成功返回时，新创建线程的线程ID会被设置成tidp指向的内存空间。

attr属性用于定制各种不同的线程属性。

新创建的线程从start_rtn函数的地址开始运行，该函数只有一个无类型指针参数arg。

下面程序将演示线程的创建，打印出进程ID、新线程的线程ID以及初始线程的线程ID：

 1 #include "apue.h" 2 #include <pthread.h> 3  4 pthread_t ntid; 5  6 void 7 printids(const char *s) 8 { 9     pid_t        pid;10     pthread_t    tid;11 12     pid = getpid();13     tid = pthread_self();14     printf("%s pid %lu tid %lu (0x%lx)\n", s, (unsigned long)pid,15       (unsigned long)tid, (unsigned long)tid);16 }17 18 void *19 thr_fn(void *arg)20 {21     printids("new thread: ");22     return((void *)0);23 }24 25 int26 main(void)27 {28     int        err;29 30     err = pthread_create(&ntid, NULL, thr_fn, NULL);31     if (err != 0)32         err_exit(err, "can't create thread");33     printids("main thread:");34     sleep(1);35     exit(0);36 }

View Code

线程创建时并不能保证哪个线程先会运行：是新创建的线程，还是调用线程。本程序让主线程休眠，确保新线程有机会运行。

 

 

 

线程终止

如果进程中任意线程调用了exit、_Exit或者_exit，那么整个进程就会终止。

单个线程可以通过3种方式退出，因此可以在不终止整个进程的情况下，停止它的控制流。

1 线程可以简单地从启动例程中返回，返回值的线程的退出码。

2 线程可以被同一进程中的其他线程取消。

3 线程调用pthread_exit。

#include <pthread.h>void pthread_exit(void *rval_ptr);

rval_ptr参数是一个无类型指针，进程中的其他线程也可以通过调用pthread_join函数访问到这个指针

#include <pthread.h>int pthread_join(pthread_t thread,void **rval_ptr);

调用pthread_join后，调用线程将一直阻塞，直到指定的线程退出。

如果线程简单地从它的启动例程返回，rval_ptr将包含返回码。如果线程被取消，由rval_ptr指定的内存单元就设置成PTHREAD_CANCELED。

下面演示如何获取已终止线程的退出码：

 1 #include "apue.h" 2 #include <pthread.h> 3  4 void * 5 thr_fn1(void *arg) 6 { 7     printf("thread 1 returning\n"); 8     return((void *)1); 9 }10 11 void *12 thr_fn2(void *arg)13 {14     printf("thread 2 exiting\n");15     pthread_exit((void *)2);16 }17 18 int19 main(void)20 {21     int            err;22     pthread_t    tid1, tid2;23     void        *tret;24 25     err = pthread_create(&tid1, NULL, thr_fn1, NULL);26     if (err != 0)27         err_exit(err, "can't create thread 1");28     err = pthread_create(&tid2, NULL, thr_fn2, NULL);29     if (err != 0)30         err_exit(err, "can't create thread 2");31     err = pthread_join(tid1, &tret);32     if (err != 0)33         err_exit(err, "can't join with thread 1");34     printf("thread 1 exit code %ld\n", (long)tret);35     err = pthread_join(tid2, &tret);36     if (err != 0)37         err_exit(err, "can't join with thread 2");38     printf("thread 2 exit code %ld\n", (long)tret);39     exit(0);40 }

View Code

 

线程可以通过调用pthread_cancel函数来请求取消同一进程中的其他进程。

#include <pthread.h>int pthread_cancel(pthread_t tid);

pthread_cancel并不等待线程终止，它仅仅提出请求，线程可以选择忽略取消或者控制如何被取消。

线程可以安排它退出时需要调用的函数，这与进程在退出时可以用atexit函数安排退出时类似的。

如果线程是通过从它的启动例程中退出返回而终止的话，它的清理处理程序就不会被调用。

#include <pthread.h>void pthread_cleanup_push(void (*rtn)(void *),void *arg);void pthread_cleanup_pop(int execute);

如果execute参数设置为非0，则调用并删除上次pthread_cleanup_push调用建立的清理处理程序。

如果execute参数为0，则清理函数将不被调用（只删除）。

我们可以调用pthread_detach分离线程。

#include <pthread.h>int pthread_detach(pthread_t tid);

 

 

 

线程同步

当一个线程可以修改的变量，其他线程可以读取或者修改的时候，我们就需要对这些线程进行同步，确保他们在访问变量的存储内容时不会访问到无效的值。

为了解决这个问题，线程不得不使用锁，同一时间只允许一个线程访问该变量。

 

 

互斥量

可以使用pthread的互斥接口来保护数据，确保同一时间只有一个线程访问数据。

互斥量从本质上说是一把锁，在访问共享资源前对互斥量进行设置（加锁），在访问完成后释放（解锁）互斥量。

互斥变量使用pthread_mutex_t数据类型表示的。在使用之前，必须对它进行初始化，如果动态分配互斥量，在释放内存前需要调用pthread_mutex_destroy。

#include <pthread.h>int pthread_mutex_init(pthread_mutex_t *restrict mutex,const pthread_mutexattr_t *restrict attr);int pthread_mutex_destroy(pthread_mutex_t *mutex);

要用默认的属性初始化互斥量，只需把attr设为NULL，也可以把互斥量设置为常量PTHREAD_MUTEX_INITIALIZER（只适用于静态分配的互斥量）进行初始化。

互斥量有以下3种功能

#include <pthread.h>int pthread_mutex_lock(pthread_mutex_t *mutex); int pthread_mutex_trylock(pthread_mutex_t *mutex);int pthread_mutex_unlock(pthread_mutex_t *mutex);

可以使用pthread_mutex_lock对互斥量进行加锁，如果互斥量已经上锁，调用线程将阻塞直到互斥量被解锁。

可以使用pthread_mutex_unlock对互斥量解锁。

如果不希望被阻塞，可以使用pthread_mutex_trylock尝试对互斥量进行加锁。如果互斥量处于未锁住状态，则锁住互斥量，否则返回EBUSY。

 

 

 

避免死锁

如果线程试图对同一个互斥量加锁两次，那么它自身就会陷入死锁状态。

如果两个线程以相反的顺序锁住两个互斥量，也会导致死锁，两个线程都无法向前运行。

在同时需要两个互斥量时，让他们以相同的顺序加锁，这样可以避免死锁。

 

 

 

函数pthread_mutex_timedlock

与pthread_mutex_lock不同的是，pthread_mutex_timedlock允许绑定线程阻塞时间，如果超过时间值，pthread_mutex_timedlock不会对互斥量进行加锁，而是返回错误码ETIMEDOUT。

#include <pthread.h>#include <time.h>int pthread_mutex_timedlock(pthread_mutex_t *restrict mutex,const struct timespec *restrict tsptr);

下面给出如何用pthread_mutex_timedlock避免永久阻塞

 1 #include "apue.h" 2 #include <pthread.h> 3  4 int 5 main(void) 6 { 7     int err; 8     struct timespec tout; 9     struct tm *tmp;10     char buf[64];11     pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;12 13     pthread_mutex_lock(&lock);14     printf("mutex is locked\n");15     clock_gettime(CLOCK_REALTIME, &tout);16     tmp = localtime(&tout.tv_sec);17     strftime(buf, sizeof(buf), "%r", tmp);18     printf("current time is %s\n", buf);19     tout.tv_sec += 10;    /* 10 seconds from now */20     /* caution: this could lead to deadlock */21     err = pthread_mutex_timedlock(&lock, &tout);22     clock_gettime(CLOCK_REALTIME, &tout);23     tmp = localtime(&tout.tv_sec);24     strftime(buf, sizeof(buf), "%r", tmp);25     printf("the time is now %s\n", buf);26     if (err == 0)27         printf("mutex locked again!\n");28     else29         printf("can't lock mutex again: %s\n", strerror(err));30     exit(0);31 }

View Code

这个程序对已有的互斥量加锁，演示了pthread_mutex_timedlock是如何工作的。

 

 

读写锁

读写锁与互斥量类似，不过读写锁允许更高的并行性。

读写锁可以有3种状态：读模式下加锁状态，写模式下加锁状态，不加锁状态。

一次只有一个线程可以占有写模式的读写锁，但是多个线程可以同时占有读模式的读写锁。

1. 当读写锁是写加锁状态时，在这个锁被解锁之前，所有试图对这个所加锁的线程都会被阻塞。

2. 当读写锁是读加锁状态时，所有试图以读模式对它进行加锁的线程都可以得到访问权，但是任何希望以写模式对此进行加锁的线程都会阻塞，知道所有的线程释放它们的读锁为止。

读写锁在使用之前必须初始化，在释放他们底层的内存之前必须销毁。

#include <pthread.h>int pthread_rwlock_init(pthread_rwlock_t *restrict rwlock,const pthread_rwlockattr_t *restrict attr);int pthread_rwlock_destroy(pthread_rwlock_t *rwlock);

下面是读写锁的3种用法

#include <pthread.h>int pthread_rwlock_rdlock(pthread_rwlock_t *rwlock);int pthread_rwlock_wrlock(pthread_rwlock_t *rwlock);int pthread_rwlock_unlock(pthread_rwlock_t *rwlock);

与互斥量一样，读写锁定义了下面两个函数

#include <pthread.h>int pthread_rwlock_tryrdlock(pthread_rwlock_t *rwlock);int pthread_rwlock_tryrwrock(pthread_rwlock_t *rwlock);

 

 

 

带有超时的读写锁

与互斥量一样，有两个带有超时的速写锁加锁函数

#include <pthread.h>#include <time.h>int pthread_rwlock_timedrdlock(pthread_rwlock_t *restrict rwlock,const struct timespec *restrict tsptr);int pthread_rwlock_timedwrlock(pthread_rwlock_t *restrict rwlock,const struct timespec *restrict tsptr);

 

 

 

条件变量

在使用条件变量之前，必须对它进行初始化，在释放底层的内存空间之前，可以使用pthread_cond_destroy函数对条件变量进行反初始化

#include <pthread.h>int pthread_cond_init(pthread_cond_t *restrict cond,const pthread_condattr_t *restrict attr);int pthread_cond_destroy(pthread_cond_t *cond);

条件本身是由互斥量保护的。线程在改变条件状态之前必须首先锁住互斥量，然后调用下面函数等待条件变量为真。

#include <pthread.h>int pthread_cond_wait(pthread_cond_t *restrict cond,pthread_mutex_t *restrict mutex);int pthread_cond_timedwait(pthread_cond_t *restrict cond,pthread_mutex_t *restrict mutex,const struct timespec *restrict tsptr);

调用者把锁住的互斥量传给函数，函数自动把调用线程放到等待条件的线程列表上，对互斥量解锁。pthread_cond_wati返回时，互斥量再次被锁住。

pthread_cond_timedwait则添加了一个超时值，如果超过到期时条件还是没有出现，则函数重新获取互斥量，然后返回ETIMEDOUT。

两个函数调用成功返回时，线程需要重新计算条件，因为另一个线程可能已经在运行并改变条件。

下面函数用于通知线程条件已经满足：

#include <pthread.h>int pthread_cond_signal(pthread_cond_t *cond);int pthread_cond_broadcast(pthread_cond_t *cond);

phread_cond_signal函数至少能唤醒一个等待该条件的线程，而pthread_cond_broadcast函数则能唤醒等待该条件的所有线程。

下面将结合条件变量和互斥量对线程进行同步

 1 #include <pthread.h> 2  3 struct msg { 4     struct msg *m_next; 5     /* ... more stuff here ... */ 6 }; 7  8 struct msg *workq; 9 10 pthread_cond_t qready = PTHREAD_COND_INITIALIZER;11 12 pthread_mutex_t qlock = PTHREAD_MUTEX_INITIALIZER;13 14 void15 process_msg(void)16 {17     struct msg *mp;18 19     for (;;) {20         pthread_mutex_lock(&qlock);21         while (workq == NULL)22             pthread_cond_wait(&qready, &qlock);23         mp = workq;24         workq = mp->m_next;25         pthread_mutex_unlock(&qlock);26         /* now process the message mp */27     }28 }29 30 void31 enqueue_msg(struct msg *mp)32 {33     pthread_mutex_lock(&qlock);34     mp->m_next = workq;35     workq = mp;36     pthread_mutex_unlock(&qlock);37     pthread_cond_signal(&qready);38 }

View Code

 

 

 

自旋锁

自旋锁与互斥量类似，但它不是通过休眠使进程阻塞，而是在获取锁之前一直处于忙等（自旋）阻塞状态。

自旋锁可用于以下情况：锁被持有的时间短，而且线程并不希望在重新调度上花费太多的成本。

自旋锁的接口与互斥量的接口类似，提供了以下的5个函数。

#include <pthread.h>int pthread_spin_init(pthread_spinlock_t *lock,int pshared);int pthread_spin_destroy(pthread_spinlock_t *lock);int pthread_spin_lock(pthread_spinlock_t *lock);int pthread_spin_trylock(pthread_spinlock_t *lock);int pthread_spin_unlock(pthread_spinlock_t *lock);

 

 

屏障

屏障是用户协调多个线程并行工作的同步机制。

屏障允许每个线程等待，直到有的合作线程到达某一点，然后从该点继续执行。pthread_join函数就是一种屏障，允许一个线程等待，直到另一个线程退出。

可以使用下面函数对屏障进行初始化跟反初始化

#include <pthread.h>int pthread_barrier_init(pthread_barrier_t *restrict barrier,const pthread_barrierattr_t *restrict attr,unsigned int count);int pthread_barrier_destroy(pthread_barrier_t *barrier);

count参数可以用来指定在允许所有线程继续运行之前，必须到达屏障的线程数目。

可以使用pthread_barrier_wait函数来表明，线程已经完成工作，准备等所有其他线程赶上来

#include <pthread.h>int pthread_barrier_wait(pthread_barrier_t *barrier);

调用pthread_barrier_wait的线程在屏障计数（调用pthread_barrier_init时设定）未满足条件时，会进入休眠状态。

如果该线程是最后一个调用pthread_barrier_wait的线程，就满足了屏障计数，所有的线程都被唤醒。

下面给出在一个任务上合作的多个线程之间如何用屏障进行同步

  1 #include "apue.h"  2 #include <pthread.h>  3 #include <limits.h>  4 #include <sys/time.h>  5   6 #define NTHR   8                /* number of threads */  7 #define NUMNUM 8000000L            /* number of numbers to sort */  8 #define TNUM   (NUMNUM/NTHR)    /* number to sort per thread */  9  10 long nums[NUMNUM]; 11 long snums[NUMNUM]; 12  13 pthread_barrier_t b; 14  15 #ifdef SOLARIS 16 #define heapsort qsort 17 #else 18 extern int heapsort(void *, size_t, size_t, 19                     int (*)(const void *, const void *)); 20 #endif 21  22 /* 23  * Compare two long integers (helper function for heapsort) 24  */ 25 int 26 complong(const void *arg1, const void *arg2) 27 { 28     long l1 = *(long *)arg1; 29     long l2 = *(long *)arg2; 30  31     if (l1 == l2) 32         return 0; 33     else if (l1 < l2) 34         return -1; 35     else 36         return 1; 37 } 38  39 /* 40  * Worker thread to sort a portion of the set of numbers. 41  */ 42 void * 43 thr_fn(void *arg) 44 { 45     long    idx = (long)arg; 46  47     heapsort(&nums[idx], TNUM, sizeof(long), complong); 48     pthread_barrier_wait(&b); 49  50     /* 51      * Go off and perform more work ... 52      */ 53     return((void *)0); 54 } 55  56 /* 57  * Merge the results of the individual sorted ranges. 58  */ 59 void 60 merge() 61 { 62     long    idx[NTHR]; 63     long    i, minidx, sidx, num; 64  65     for (i = 0; i < NTHR; i++) 66         idx[i] = i * TNUM; 67     for (sidx = 0; sidx < NUMNUM; sidx++) { 68         num = LONG_MAX; 69         for (i = 0; i < NTHR; i++) { 70             if ((idx[i] < (i+1)*TNUM) && (nums[idx[i]] < num)) { 71                 num = nums[idx[i]]; 72                 minidx = i; 73             } 74         } 75         snums[sidx] = nums[idx[minidx]]; 76         idx[minidx]++; 77     } 78 } 79  80 int 81 main() 82 { 83     unsigned long    i; 84     struct timeval    start, end; 85     long long        startusec, endusec; 86     double            elapsed; 87     int                err; 88     pthread_t        tid; 89  90     /* 91      * Create the initial set of numbers to sort. 92      */ 93     srandom(1); 94     for (i = 0; i < NUMNUM; i++) 95         nums[i] = random(); 96  97     /* 98      * Create 8 threads to sort the numbers. 99      */100     gettimeofday(&start, NULL);101     pthread_barrier_init(&b, NULL, NTHR+1);102     for (i = 0; i < NTHR; i++) {103         err = pthread_create(&tid, NULL, thr_fn, (void *)(i * TNUM));104         if (err != 0)105             err_exit(err, "can't create thread");106     }107     pthread_barrier_wait(&b);108     merge();109     gettimeofday(&end, NULL);110 111     /*112      * Print the sorted list.113      */114     startusec = start.tv_sec * 1000000 + start.tv_usec;115     endusec = end.tv_sec * 1000000 + end.tv_usec;116     elapsed = (double)(endusec - startusec) / 1000000.0;117     printf("sort took %.4f seconds\n", elapsed);118     for (i = 0; i < NUMNUM; i++)119         printf("%ld\n", snums[i]);120     exit(0);121 }

View Code

在这个实例中，使用8个线程分解了800万个数的排序工作。每个线程用堆排序算法对100万个数进行排序，然后主线程调用一个函数对这些结果进行合并。