基于ucontext.h的C语言线程切换器上下文切换失效问题排查

问题核心分析 #

你的代码存在两个关键错误：

1. 初始pause()函数中，swapcontext(t->context, t->context)传入同一个上下文参数，完全无法完成线程切换——swapcontext的作用是将当前上下文保存到第一个参数，然后切换到第二个参数指向的上下文，传相同参数等于无操作。
2. 后续尝试的switch()函数逻辑错误：getcontext保存当前上下文后，setcontext又切回同一个上下文，导致无限循环；同时setcontext会恢复getcontext时的栈状态，isSwap会被重置为0，陷入死循环，且从未真正切换到其他线程的上下文。

此外，你的线程管理逻辑缺少对当前运行线程的跟踪，无法正确保存当前线程的上下文并将其放回队列。

正确实现方案 #

基础结构定义 #

首先需要明确线程节点结构、队列管理函数，以及跟踪当前运行线程的全局指针：

#include <ucontext.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>

#define STACK_SIZE 16384

// 线程节点结构：包含上下文、ID、栈和队列指针
typedef struct ThreadNode {
    ucontext_t context;
    int thread_id;
    struct ThreadNode* next;
    char stack[STACK_SIZE]; // 线程私有栈，必须持续存在
} ThreadNode;

ThreadNode* Q = NULL;          // 待运行线程队列
ThreadNode* current_thread = NULL; // 当前运行线程指针

// 队列队尾添加节点
void QManagerPush(ThreadNode* node) {
    if (!Q) {
        Q = node;
        node->next = NULL;
        return;
    }
    ThreadNode* tmp = Q;
    while (tmp->next) tmp = tmp->next;
    tmp->next = node;
    node->next = NULL;
}

// 队列弹出队首节点
ThreadNode* QManagerPop() {
    if (!Q) return NULL;
    ThreadNode* tmp = Q;
    Q = Q->next;
    tmp->next = NULL;
    return tmp;
}

核心切换逻辑 #

实现正确的线程切换函数，以及符合队列调度逻辑的pause()：

// 保存当前线程上下文，切换到目标线程上下文
static void thread_switch(ucontext_t* from, ucontext_t* to) {
    if (swapcontext(from, to) == -1) {
        perror("swapcontext failed");
        exit(EXIT_FAILURE);
    }
}

void pause() {
    // 1. 取出下一个待运行线程
    ThreadNode* next_thread = QManagerPop();
    if (!next_thread) return; // 无待运行线程，直接返回

    // 2. 保存当前线程上下文，切换到下一个线程
    ThreadNode* prev_thread = current_thread;
    current_thread = next_thread;
    thread_switch(&prev_thread->context, &next_thread->context);

    // 3. 当切换回当前线程时，将其放回队列尾部
    QManagerPush(prev_thread);
}

线程初始化与入口函数 #

// 线程入口函数：执行任务并主动触发切换
static void thread_func() {
    for (int i = 0; i < 3; i++) {
        printf("Thread %d: iteration %d\n", current_thread->thread_id, i);
        pause(); // 主动暂停，切换到下一个线程
    }
    printf("Thread %d finished\n", current_thread->thread_id);
}

// 创建并初始化线程节点
ThreadNode* create_thread(int id) {
    ThreadNode* node = malloc(sizeof(ThreadNode));
    if (!node) {
        perror("malloc failed");
        exit(EXIT_FAILURE);
    }
    node->thread_id = id;

    // 初始化上下文
    if (getcontext(&node->context) == -1) {
        perror("getcontext failed");
        free(node);
        exit(EXIT_FAILURE);
    }

    // 配置线程栈
    node->context.uc_stack.ss_sp = node->stack;
    node->context.uc_stack.ss_size = STACK_SIZE;
    node->context.uc_stack.ss_flags = 0;
    node->context.uc_link = NULL; // 线程结束后回到NULL（可设为主线程上下文）

    // 设置线程入口函数
    makecontext(&node->context, thread_func, 0);

    return node;
}

测试主函数 #

int main() {
    // 创建两个测试线程
    ThreadNode* t1 = create_thread(1);
    ThreadNode* t2 = create_thread(2);

    // 将线程加入待运行队列
    QManagerPush(t1);
    QManagerPush(t2);

    // 启动第一个线程
    current_thread = QManagerPop();
    setcontext(&current_thread->context);

    // 清理资源
    free(t1);
    free(t2);
    return 0;
}

关键注意事项 #

• swapcontext的正确用法：必须传入两个不同的ucontext_t指针，分别对应"保存当前上下文"和"切换到目标上下文"。
• 线程栈的生命周期：线程栈必须在整个线程运行期间保持有效，不能使用局部栈变量（会被提前释放），示例中使用节点内的数组作为栈，确保生命周期与线程一致。
• 当前线程跟踪：必须维护current_thread指针，才能正确保存当前线程的上下文并将其放回队列，实现轮转调度。
• 错误处理：getcontext、swapcontext等系统调用可能失败，必须添加错误处理逻辑避免崩溃。

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