能否在RISC-V中实现原子释放-获取操作：原子释放一锁并立即获取另一锁？

RISC-V基础指令集里没有直接支持双内存位置原子交换的指令，但可以通过LR/SC（加载保留/存储条件）循环结合原子指令，在软件层面实现近似的原子“释放旧锁+获取新锁”操作，以下是具体方案和注意事项：

为什么原生指令无法直接实现 #

• 所有RISC-V AMO（原子内存操作）指令（比如amoswap.w.aq）仅针对单个内存地址操作，只能完成寄存器与单内存位置的原子交换，不支持两个内存地址间的直接原子交互。
• 基础LR/SC机制同样只监控单个内存地址的修改，无法覆盖多内存位置的事务性操作，RISC-V的多地址事务内存（TM）属于可选扩展（RV-TM），并非所有硬件都支持。

基于LR/SC的软件实现方案 #

核心思路是通过循环重试，确保“释放旧锁→获取新锁”的操作序列在无干扰的情况下完成；若中途被其他核心打断，则回滚状态并重试。以下是示例汇编代码：

# 输入参数：
# a0 = 旧锁的内存地址（当前已持有该锁）
# a1 = 新锁的内存地址
# a2 = 锁的解锁值（通常为0）
# a3 = 锁的加锁值（通常为1）
atomic_lock_swap:
    lr.w    t0, (a0)          # 加载并保留旧锁的当前值
    bne     t0, a3, exit      # 若旧锁未被当前核心持有，直接退出（异常分支）
    
    # 尝试原子释放旧锁
    sc.w    t1, a2, (a0)      # 将解锁值写入旧锁，若保留未被打断则成功（t1=0）
    bne     t1, zero, atomic_lock_swap # SC失败，重试整个流程
    
    # 尝试原子获取新锁
    amoswap.w.aq t2, a3, (a1) # 用amoswap原子交换获取新锁，aq标记保证获取的内存顺序
    bne     t2, a2, rollback  # 若新锁已被持有，执行回滚
    
    fence rw,rw               # 内存屏障，确保释放旧锁的操作对所有核心可见
    j       exit

rollback:
    # 回滚旧锁：重新加锁，避免错误释放
    amoswap.w.rl t3, a3, (a0)
    j       atomic_lock_swap  # 回滚后重试整个流程

exit:
    ret

代码逻辑说明 #

1. LR/SC保护旧锁释放：通过lr.w保留旧锁地址，sc.w原子写入解锁值，确保释放操作不会被打断；若SC失败，说明其他核心修改了旧锁，直接重试。
2. 原子获取新锁：用amoswap.w.aq原子获取新锁，aq标记保证后续操作不会重排序到获取锁之前。
3. 失败回滚：若获取新锁失败，必须重新将旧锁加回，避免旧锁被错误释放导致并发问题。
4. 内存屏障：fence rw,rw确保释放旧锁的写操作全局可见后，再执行后续逻辑，避免内存重排序引发的一致性问题。

关键注意事项 #

• 软件原子性而非硬件原子性：这种方案是通过循环重试实现的“最终原子性”，并非硬件级的无中断操作，极端情况下可能会有多次重试，但能保证结果的正确性。
• 死锁风险：若多个核心同时尝试交换相同的两组锁，可能出现死锁。解决方法是为锁定义全局排序规则（比如按内存地址从小到大获取），避免循环等待。
• 可选扩展支持：如果硬件支持RISC-V的RV-TM事务内存扩展，可以直接用事务指令包裹释放和获取操作，实现真正的多内存原子事务，但该扩展并非所有RISC-V设备都具备。

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