自定义用户态线程库中Mutex锁上下文切换异常问题排查

你遇到的核心问题是：当锁被占用时直接切换到main_context，但既没有维护等待锁的线程队列，也没有让主线程在锁释放后调度这些等待线程。这导致当前线程的执行流程被强制中断，主线程继续执行自身代码，而等待锁的线程无法被唤醒回到断点（uthread_mutex_lock之后的debug、f()、g()）继续执行。

1. 给Mutex结构体添加等待队列 #

首先扩展Mutex的结构，增加一个队列来保存等待该锁的线程ID，用于后续唤醒：

#define MAX_THREADS 16

typedef struct {
    int locked;
    int wait_queue[MAX_THREADS];
    int wait_count;
} uthread_mutex_t;

2. 修改uthread_mutex_lock逻辑 #

不再直接切换到主线程，而是将当前线程加入等待队列后，调用线程库的调度函数让出CPU，让其他就绪线程执行：

void uthread_mutex_lock(uthread_mutex_t *mutex) {
    // 原子操作尝试获取锁
    while (__sync_lock_test_and_set(&mutex->locked, 1)) {
        // 将当前线程加入等待队列（需保证队列操作的原子性，可临时用自旋锁）
        __sync_lock_test_and_set(&mutex->wait_count, mutex->wait_count); // 原子读取
        if (mutex->wait_count < MAX_THREADS) {
            mutex->wait_queue[mutex->wait_count++] = curr_thread_id;
        }
        // 让出CPU，调度其他就绪线程
        uthread_yield();
        // 被唤醒后重新进入循环检查锁状态
    }
}

3. 修改uthread_mutex_unlock逻辑 #

释放锁时检查等待队列，唤醒第一个等待的线程，将其标记为就绪状态：

void uthread_mutex_unlock(uthread_mutex_t *mutex) {
    // 原子释放锁
    __sync_lock_release(&mutex->locked);
    
    if (mutex->wait_count > 0) {
        // 取出队列头部的等待线程
        int wake_thread = mutex->wait_queue[0];
        // 调整队列
        for (int i = 0; i < mutex->wait_count - 1; i++) {
            mutex->wait_queue[i] = mutex->wait_queue[i+1];
        }
        mutex->wait_count--;
        // 将该线程标记为就绪，等待调度器调度
        threads[wake_thread]->state = READY;
    }
}

4. 修复调度器逻辑 #

• 不要让uthread_mutex_lock直接调用swapcontext切换到main_context，而是统一由uthread_yield()或调度器函数管理上下文切换。
• 主线程应作为普通线程纳入调度体系，当没有其他就绪线程时，调度器才切换回主线程执行。

5. 确保上下文切换的正确性 #

检查线程库中swapcontext的调用逻辑，确保等待锁的线程被挂起时，完整保存了当前执行上下文（包括uthread_mutex_lock循环的断点），这样被唤醒后能从断点处继续执行，进而执行后续的debug、f()、g()代码。

• 等待队列的操作需要保证原子性，避免多个线程同时修改队列导致数据混乱，可使用临时的自旋锁或原子操作保护队列读写。
• 测试时需验证：当锁被释放后，等待的线程能被立即唤醒，回到uthread_mutex_lock之后的代码继续执行，而不是停留在主线程。

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