POSIX線程清理函數

POSIX清理函數的調用時機：

調用pthread_exit()時，會調用清理函數；通過return返回的線程不會調用。

被別的線程取消的時候，會調用。

pthread_cleanup_pop()參數為非零時，會調用。

1. #include <stdio.h>
2. #include <pthread.h>
3. #include <windows.h> // Sleep
5. pthread_mutex_t mtx = PTHREAD_MUTEX_INITIALIZER;
6. pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
8. struct Node
9. {
10. int number;
11. struct Node *next;
12. } *head = NULL;
14. // 清理函數
15. void cleanup_handler(void *node)
16. {
17. printf("Cleanup handler of second thread.\n");
18. free(node);
19. pthread_mutex_unlock(&mtx);
20. }
22. void* thread_func(void *arg)
23. {
24. struct Node *p = NULL;
25. // 清理函數入棧，此demo只需一個清理函數
26. pthread_cleanup_push(cleanup_handler, p);
28. while (true)
29. {
30. pthread_mutex_lock(&mtx);
31. if (head == NULL)
32. {
33. pthread_cond_wait(&cond, &mtx);
34. }
35. p = head;
36. head = head->next;
37. printf("Got %d from front of queue\n", p->number);
38. free(p);
39. pthread_mutex_unlock(&mtx);
40. }
41. /* 若從此處終止線程，則屬於正常退出，無需清理。所以，只需將清理函數出棧。故而用
42. 參數零。若是從上面的“取消點”退出，則清理函數出棧時被調用：鎖被打開，同時釋放資源。*/
43. pthread_cleanup_pop(0);
44. return 0;
45. }
47. int main(int argc, char* argv[])
48. {
49. pthread_t tid;
50. pthread_create(&tid, NULL, thread_func, NULL);
51. for (int i = 0; i < 10; i++)
52. {
53. struct Node* p = (Node *)malloc(sizeof(struct Node));
54. p->number = i;
56. // <!-- 對head操作屬於臨界區
57. pthread_mutex_lock(&mtx);
58. p->next = head;
59. head = p;
60. pthread_cond_signal(&cond);
61. pthread_mutex_unlock(&mtx);
62. // 所以要用互斥鎖保護起來--!>
64. Sleep(1);
65. }
66. printf("Main thread wants to cancel the 2nd thread.\n");
68. /* 關於pthread_cancel，有一點額外的說明，它是從外部終止子線程，子線程會在
69. 最近的取消點，退出線程。而在我們的代碼裡，最近的取消點是pthread_cond_wait()了。*/
70. pthread_cancel(tid);
71. pthread_join(tid, NULL);
72. printf("All done -- exiting\n");
73. return 0;
74. }