如何在GCC中为RISC-V特权级陷阱处理器实现独立栈切换

核心方案：内联汇编嵌入栈切换逻辑 #

标准GCC的interrupt属性会自动完成上下文保存/恢复、返回指令生成，但不包含栈切换。我们可以通过在C陷阱处理函数的首尾插入内联汇编，手动完成栈切换，同时保留interrupt属性的自动生成逻辑，这个方法同时适用于机器模式和监督模式。

实现步骤与代码示例 #

1. 提前准备陷阱栈与寄存器初始化

需要为陷阱模式分配独立栈，并将栈顶地址存入对应模式的scratch寄存器（机器模式用mscratch，监督模式用sscratch），后续通过该寄存器完成栈切换。

2. 机器模式陷阱处理器实现

#include <stdint.h>

// 定义机器模式陷阱栈（16字节对齐，大小按需调整）
uint8_t m_trap_stack[4096] __attribute__((aligned(16)));
#define M_TRAP_STACK_TOP ((uintptr_t)m_trap_stack + sizeof(m_trap_stack))

// 声明带机器模式中断属性的处理函数
void m_trap_handler(void) __attribute__((interrupt("machine")));

void m_trap_handler(void) {
    // 内联汇编：原子交换sp与mscratch，切换到陷阱栈
    // 执行后mscratch存储原sp，sp变为陷阱栈顶
    __asm__ volatile (
        "csrrw sp, mscratch, sp\n"
        :
        :
        : "memory" // 告知编译器内存状态已变更，防止优化打乱顺序
    );

    // --------------------------
    // 这里编写你的陷阱处理逻辑
    uintptr_t mcause;
    __asm__ volatile ("csrr %0, mcause" : "=r"(mcause));
    // 例如：判断陷阱类型、处理异常/中断等
    // --------------------------

    // 内联汇编：恢复原sp（再次交换sp与mscratch）
    __asm__ volatile (
        "csrrw sp, mscratch, sp\n"
        :
        :
        : "memory"
    );

    // GCC的interrupt属性会自动生成：上下文恢复 + mret返回指令
}

// 初始化机器模式陷阱环境
void m_trap_init(void) {
    // 将陷阱栈顶写入mscratch
    __asm__ volatile ("csrw mscratch, %0" : : "r"(M_TRAP_STACK_TOP));
    // 设置mtvec为陷阱处理函数地址（直接模式）
    __asm__ volatile ("csrw mtvec, %0" : : "r"(m_trap_handler));
}

3. 监督模式陷阱处理器实现

仅需替换对应模式的寄存器和属性参数，逻辑完全一致：

#include <stdint.h>

uint8_t s_trap_stack[4096] __attribute__((aligned(16)));
#define S_TRAP_STACK_TOP ((uintptr_t)s_trap_stack + sizeof(s_trap_stack))

void s_trap_handler(void) __attribute__((interrupt("supervisor")));

void s_trap_handler(void) {
    // 切换到监督模式陷阱栈
    __asm__ volatile (
        "csrrw sp, sscratch, sp\n"
        :
        :
        : "memory"
    );

    // 监督模式陷阱处理逻辑
    uintptr_t scause;
    __asm__ volatile ("csrr %0, scause" : "=r"(scause));
    // ... 处理代码 ...

    // 恢复原sp
    __asm__ volatile (
        "csrrw sp, sscratch, sp\n"
        :
        :
        : "memory"
    );

    // GCC自动生成：上下文恢复 + sret返回指令
}

void s_trap_init(void) {
    __asm__ volatile ("csrw sscratch, %0" : : "r"(S_TRAP_STACK_TOP));
    __asm__ volatile ("csrw stvec, %0" : : "r"(s_trap_handler));
}

关键说明 #

• 栈切换时机：内联汇编放在函数最开头，确保在GCC自动保存上下文前完成栈切换；恢复操作放在处理逻辑末尾，确保上下文恢复前切回原栈。
• 原子性保障：使用csrrw（原子读-写控制寄存器）指令完成sp与scratch寄存器的交换，避免中断嵌套时的栈混乱。
• 兼容性：此方案基于标准RISC-V指令集和GCC通用属性，不依赖SiFive等厂商的专属扩展，同时支持机器/监督模式。

关于SiFive专属属性的补充 #

SiFive的interrupt("SiFive-CLIC-stack-swap")确实能自动完成栈切换，但仅支持机器模式，且属于厂商定制扩展，无法在通用GCC环境下使用。上述内联汇编方案是更通用的替代方案。

内容的提问来源于stack exchange，提问作者Eugene Sh.