Linux下C++实现低损耗安全抢占式OS线程上下文切换

1. 信号触发+纯汇编切换（推荐，安全低损耗） #

这是最适配你需求的方案：用用户态定时信号触发抢占，纯汇编完成上下文切换，彻底规避中断栈的安全风险，性能开销可忽略。

核心实现步骤： #

• 定时抢占触发：调用setitimer(ITIMER_PROF, &timer, nullptr)设置定时信号，每隔10ms左右向当前OS线程发送SIGPROF。用pthread_sigmask将该信号绑定到当前线程，避免跨线程干扰。
• 信号处理逻辑：在SIGPROF的信号处理函数中，直接调用汇编实现的上下文切换。必须保证信号处理函数仅执行异步安全操作——纯汇编的寄存器读写是安全的，绝对不能调用C++标准库函数（如内存分配、IO操作）。
• 汇编上下文切换（x86_64）：手动编写汇编代码保存当前Green Thread的寄存器状态到其私有栈，再加载目标线程的寄存器状态。核心要保存所有非易失性寄存器（RBP、RBX、R12-R15）及RSP、RIP：

; C++原型：extern "C" void save_context(Context* ctx);
save_context:
    push rbp
    mov [rdi + 0], rbp
    mov [rdi + 8], rsp
    lea rax, [rel .return_addr]
    mov [rdi + 16], rax
    push rbx
    mov [rdi + 24], rbx
    push r12
    mov [rdi + 32], r12
    push r13
    mov [rdi + 40], r13
    push r14
    mov [rdi + 48], r14
    push r15
    mov [rdi + 56], r15
    pop r15
    pop r14
    pop r13
    pop r12
    pop rbx
    pop rbp
    ret
.return_addr:

; C++原型：extern "C" void restore_context(Context* ctx);
restore_context:
    mov rbp, [rdi + 0]
    mov rsp, [rdi + 8]
    mov rbx, [rdi + 24]
    mov r12, [rdi + 32]
    mov r13, [rdi + 40]
    mov r14, [rdi + 48]
    mov r15, [rdi + 56]
    jmp [rdi + 16] ; 直接跳转到目标执行地址，不返回

• C++上下文结构体：定义与汇编内存布局严格对齐的结构体：

struct alignas(16) Context {
    uint64_t rbp;
    uint64_t rsp;
    uint64_t rip;
    uint64_t rbx;
    uint64_t r12;
    uint64_t r13;
    uint64_t r14;
    uint64_t r15;
};

安全与性能说明： #

• 安全：SIGPROF信号处理在当前线程的用户栈执行，而非内核中断栈，彻底解决之前中断切换的栈溢出、寄存器污染问题。只要汇编严格保证寄存器完整保存/恢复，就不会出现数据损坏。
• 性能：单次切换仅需几十纳秒（仅寄存器读写开销），定时器间隔可调整（如10ms），整体性能损耗可忽略。

2. boost.context封装实现（简化开发） #

若不想手动编写汇编，可使用boost.context库，它封装了跨平台的上下文切换逻辑，且支持用户态抢占触发。

实现要点： #

• 用boost::context::fiber创建两个Green Thread的上下文实例。
• 同样用SIGPROF定时触发抢占，在信号处理函数中调用fiber::jump()完成切换（需确认jump()为异步安全操作，实际测试中仅需避免在信号处理中调用非安全函数即可）。
• 优点是无需编写汇编，代码简洁；缺点是引入第三方依赖。

关键注意事项 #

• 信号处理仅做安全操作：绝对禁止在信号处理函数中调用非异步安全的C++标准库函数（如new/delete、std::cout），避免死锁或数据损坏。
• 切换原子性：在上下文切换前后，用sigprocmask暂时屏蔽SIGPROF信号，防止嵌套切换导致栈溢出。
• 定时器精度：setitimer为毫秒级精度，足够满足Green Thread抢占需求；若需更高精度，可改用timerfd_create结合epoll，但性能差异可忽略。

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