主程序退出时C硬件控制库如何设置清理函数？含JNI场景

我编写了一个控制硬件设备的小型C库，该库使用pthreads启动一个负责与硬件设备交互的线程。在部分使用场景中（非全部），这个库会被Java应用通过JNI调用。当Java应用调用System.exit()时，所有线程都会被强制终止，导致库无法确保硬件设备处于"空闲"状态。

我想知道是否可以为该库设置一个在主程序退出或线程终止时自动调用的清理函数？概念上类似于atexit，但我知道不能从共享库中调用atexit。

下面是几种适配不同场景的可行实现方案：

1. 线程级清理：pthread_cleanup_push/pthread_cleanup_pop #

给硬件交互线程注册清理函数，当线程处于可取消状态下被终止时（比如收到pthread_cancel信号），会自动执行清理逻辑。

• 示例代码：

void hardware_cleanup(void *arg) {
    // 执行硬件空闲化操作：关闭端口、重置设备状态等
    hardware_device_t *dev = (hardware_device_t*)arg;
    dev_reset_to_idle(dev);
}

void *hardware_thread_func(void *arg) {
    hardware_device_t *dev = (hardware_device_t*)arg;
    // 注册清理函数
    pthread_cleanup_push(hardware_cleanup, dev);

    // 线程主逻辑：循环与硬件交互
    while (dev->running) {
        dev_interact(dev);
        // 主动检查取消请求，触发清理的取消点
        pthread_testcancel();
    }

    // 弹出清理函数，参数1表示主动执行清理
    pthread_cleanup_pop(1);
    return NULL;
}

注意：Java调用System.exit()时，JVM会异步终止线程，需确保清理函数是异步安全的。可设置线程取消类型为PTHREAD_CANCEL_DEFERRED，让取消请求仅在指定取消点触发。

2. JVM关闭钩子（JNI场景专属） #

如果库主要通过JNI调用，可在JNI初始化阶段给Java虚拟机注册关闭钩子，JVM退出前会触发C层的清理逻辑。

• 示例代码（JNI部分）：

static hardware_device_t *g_dev = NULL;

void JNI_OnLoad(JavaVM *vm, void *reserved) {
    // 初始化硬件设备并启动线程
    g_dev = dev_init();
    start_hardware_thread(g_dev);

    JNIEnv *env;
    if ((*vm)->GetEnv(vm, (void**)&env, JNI_VERSION_1_6) != JNI_OK) {
        return JNI_ERR;
    }

    // 获取Runtime类并注册关闭钩子
    jclass runtime_cls = (*env)->FindClass(env, "java/lang/Runtime");
    jmethodID get_runtime_mid = (*env)->GetStaticMethodID(env, runtime_cls, "getRuntime", "()Ljava/lang/Runtime;");
    jobject runtime_obj = (*env)->CallStaticObjectMethod(env, runtime_cls, get_runtime_mid);

    jclass thread_cls = (*env)->FindClass(env, "java/lang/Thread");
    jclass runnable_cls = (*env)->FindClass(env, "java/lang/Runnable");
    jobject runnable_obj = (*env)->AllocObject(env, runnable_cls);

    // 绑定Runnable的run方法到C清理函数
    JNINativeMethod nm = {
        "run", "()V", (void*)jni_hardware_cleanup
    };
    (*env)->RegisterNatives(env, runnable_cls, &nm, 1);

    jobject shutdown_thread = (*env)->NewObject(env, thread_cls, 
        (*env)->GetMethodID(env, thread_cls, "<init>", "(Ljava/lang/Runnable;)V"), 
        runnable_obj);
    (*env)->CallVoidMethod(env, runtime_obj, 
        (*env)->GetMethodID(env, runtime_cls, "addShutdownHook", "(Ljava/lang/Thread;)V"), 
        shutdown_thread);

    return JNI_VERSION_1_6;
}

void jni_hardware_cleanup(JNIEnv *env, jobject obj) {
    // 停止线程并清理硬件
    stop_hardware_thread(g_dev);
    dev_reset_to_idle(g_dev);
    dev_cleanup(g_dev);
    g_dev = NULL;
}

该方案能确保JVM正常退出（包括调用System.exit()）时主动触发清理，是JNI场景下最可靠的方式。

3. 共享库析构函数（Linux平台） #

在Linux系统中，可给函数添加__attribute__((destructor))属性，共享库被卸载时（进程退出或调用dlclose）会自动执行该函数。

• 示例代码：

__attribute__((destructor))
void library_cleanup() {
    if (g_dev != NULL && g_dev->running) {
        stop_hardware_thread(g_dev);
        dev_reset_to_idle(g_dev);
        dev_cleanup(g_dev);
    }
}

注意：进程被强制终止（如SIGKILL）时析构函数不会执行；JNI场景中JVM可能不会主动卸载共享库，需结合其他机制使用。

4. 显式清理API #

给库暴露一个显式的清理函数，要求调用方（包括Java/JNI）在退出前主动调用：

void hardware_lib_cleanup(hardware_device_t *dev) {
    if (dev == NULL) return;
    // 停止线程
    dev->running = 0;
    pthread_join(dev->thread, NULL);
    // 重置硬件到空闲状态
    dev_reset_to_idle(dev);
    // 释放资源
    dev_free(dev);
}

在JNI场景中，Java层需在退出前（如finally块中）调用该函数的JNI包装方法。虽然依赖调用方配合，但这是最可控的方式。

内容的提问来源于stack exchange，提问作者Grodriguez