pthread_setcancelstate概述

随着多线程编程的普及，如何管理线程的状态和组织线程成为编程难点之一。而pthread_setcancelstate函数的作用是设置线程被取消的状态。

一、pthread_setcancelstate的基本用法

#include <pthread.h>

int pthread_setcancelstate(int state, int *oldstate);

其中state参数为期望的线程取消状态，可选值为：

• PTHREAD_CANCEL_ENABLE：允许线程被取消
• PTHREAD_CANCEL_DISABLE：禁止线程被取消

oldstate参数为之前的线程取消状态，如果此参数为非NULL，则之前的状态会被保存在oldstate所指向的内存中。

二、pthread_setcancelstate的注意事项

1、pthread_setcancelstate函数只会影响当前线程的取消状态，不会影响其他线程的取消状态。

2、当线程处于被取消状态时，若需要清理资源等工作，可以通过在线程入口函数中调用pthread_cleanup_push和pthread_cleanup_pop实现。

void cleanup_func(void *arg) {
    // 资源清理等工作
}

void *thread_func(void *arg) {
    pthread_cleanup_push(cleanup_func, NULL);
    // 线程主逻辑
    pthread_cleanup_pop(0);
}

其中cleanup_func是需要调用的清理函数，arg为传入清理函数的参数。

三、pthread_setcancelstate的适用场景

当线程需要长时间运行或需要处理耗时工作时，建议设置为允许被取消状态，以防止其它线程长时间等待它的完成。而当数据不稳定或处理不可重复时，建议采用禁止被取消状态，以保证程序不发生不可预期的错误。

四、pthread_setcancelstate的错误处理

1、如果state参数不是PTHREAD_CANCEL_ENABLE或PTHREAD_CANCEL_DISABLE，函数会返回EINVAL错误。

2、如果oldstate参数是NULL，函数会忽略之前的状态，不会返回错误。

3、如果调用pthread_setcancelstate函数失败，可以通过errno全局变量获取错误号。

五、示例代码

#include <stdio.h>
#include <pthread.h>

volatile int is_running = 1;

void *thread_func(void *arg) {
    printf("Thread started\n");
    int oldstate;
    pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &oldstate);
    
    while (is_running) {
        // 长时间运行的逻辑
    }
    
    printf("Thread finished\n");
    pthread_cleanup_push(free, arg);
    pthread_cleanup_pop(0);
}

int main() {
    pthread_t tid;
    if (pthread_create(&tid, NULL, thread_func, NULL) < 0) {
        printf("Failed to create thread\n");
        return -1;
    }
    
    printf("Waiting for thread to finish\n");
    sleep(1);
    
    pthread_cancel(tid);
    is_running = 0;
    
    if (pthread_join(tid, NULL) < 0) {
        printf("Failed to join thread\n");
        return -1;
    }
    
    printf("Thread joined\n");
    return 0;
}

该代码演示了如何使用pthread_setcancelstate函数设置线程的取消状态。其中，在主线程中通过pthread_cancel函数取消线程，并通过设置is_running变量为0让线程退出。