libunwind: Unwind across c18n compartment boundaries on RISC-V

Author: dpgao

libunwind/include/__libunwind_config.h

@@ -188,8 +188,8 @@
 #   define RISCV_FLEN 0
 #  endif
 #  ifdef __CHERI_PURE_CAPABILITY__
-#   define _LIBUNWIND_CONTEXT_SIZE 96
-#   define _LIBUNWIND_CURSOR_SIZE 120
+#   define _LIBUNWIND_CONTEXT_SIZE 98
+#   define _LIBUNWIND_CURSOR_SIZE 122
 #  else
 #   define _LIBUNWIND_CONTEXT_SIZE (32 * (__riscv_xlen + RISCV_FLEN) / 64)
 #   if __riscv_xlen == 32

libunwind/src/CompartmentInfo.hpp

@@ -26,7 +26,6 @@ struct CompartmentInfo {
 
   static int unwindIfAtBoundary(R &registers) {
 #ifdef _LIBUNWIND_HAS_CHERI_LIB_C18N
-#ifdef _LIBUNWIND_TARGET_AARCH64
     struct dl_c18n_compart_state state;
     pint_t pc = registers.getIP();
     pint_t tf = registers.getTrustedStack();
@@ -51,13 +50,23 @@ struct CompartmentInfo {
     CHERI_DBG("C18N: SET IP: %#p\n", state.pc);
 
     for (size_t i = 0; i < sizeof(state.regs) / sizeof(*state.regs); ++i) {
-      registers.setCapabilityRegister(UNW_ARM64_C19 + i, (pint_t)state.regs[i]);
-      CHERI_DBG("C18N: SET REGISTER: %lu (%s): %#p\n",
-                UNW_ARM64_C19 + i,
-                registers.getRegisterName(UNW_ARM64_C19 + i),
+      int reg_num;
+#ifdef _LIBUNWIND_TARGET_AARCH64
+      reg_num = UNW_ARM64_C19 + i;
+#elif defined(_LIBUNWIND_TARGET_RISCV)
+      static constexpr int callee_saved[] = {
+        UNW_RISCV_X9, UNW_RISCV_X18, UNW_RISCV_X19, UNW_RISCV_X20, UNW_RISCV_X21,
+        UNW_RISCV_X22, UNW_RISCV_X23, UNW_RISCV_X24, UNW_RISCV_X25, UNW_RISCV_X26,
+        UNW_RISCV_X27
+      };
+      reg_num = callee_saved[i];
+#endif
+      registers.setCapabilityRegister(reg_num, (pint_t)state.regs[i]);
+      CHERI_DBG("C18N: SET REGISTER: %d (%s): %#p\n",
+                reg_num,
+                registers.getRegisterName(reg_num),
                 state.regs[i]);
     }
-#endif
 #endif // _LIBUNWIND_HAS_CHERI_LIB_C18N
     return UNW_STEP_SUCCESS;
   }

libunwind/src/Registers.hpp

@@ -4552,10 +4552,21 @@ class _LIBUNWIND_HIDDEN Registers_riscv {
   void        setSP(reg_t value) { _registers[2] = value; }
   reg_t       getIP() const { return _registers[0]; }
   void        setIP(reg_t value) { _registers[0] = value; }
+  reg_t       getFP() const { return _registers[8]; }
+  void        setFP(reg_t value) { _registers[8] = value; }
+#ifdef __CHERI_PURE_CAPABILITY__
+  reg_t       getTrustedStack() const { return _registers[32]; }
+  void        setTrustedStack(reg_t value) { _registers[32] = value; }
+#endif
 
 private:
   // _registers[0] holds the pc
+#ifdef __CHERI_PURE_CAPABILITY__
+  // _registers[32] holds the trusted stack pointer
+  reg_t _registers[33];
+#else
   reg_t _registers[32];
+#endif
 # if defined(__riscv_flen)
   fp_t _floats[32];
 # endif
@@ -4566,8 +4577,8 @@ inline Registers_riscv::Registers_riscv(const void *registers) {
                 "riscv registers do not fit into unw_context_t");
   memcpy(&_registers, registers, sizeof(_registers));
 # ifdef __CHERI_PURE_CAPABILITY__
-  static_assert(sizeof(_registers) == 0x200,
-                "expected float registers to be at offset 512");
+  static_assert(sizeof(_registers) == 0x210,
+                "expected float registers to be at offset 528");
 # elif __riscv_xlen == 32
   static_assert(sizeof(_registers) == 0x80,
                 "expected float registers to be at offset 128");

libunwind/src/UnwindRegistersRestore.S

@@ -1330,6 +1330,18 @@ DEFINE_LIBUNWIND_FUNCTION(_ZN9libunwind15Registers_sparc6jumptoEv)
 //
   .p2align 2
 DEFINE_LIBUNWIND_FUNCTION(_ZN9libunwind15Registers_riscv6jumptoEv)
+#ifdef _LIBUNWIND_HAS_CHERI_LIB_C18N
+  // Preserve the argument in a callee-saved register instead of pushing it onto
+  // the stack because stack unwinding will switch the stack.
+  cmv      cs0, ca0
+  // Pass the target untrusted stack pointer
+  lc       ca0, (__SIZEOF_CHERI_CAPABILITY__ * 2)(cs0)
+  // Pass the target trusted stack pointer
+  lc       ca1, (__SIZEOF_CHERI_CAPABILITY__ * 32)(cs0)
+  call     dl_c18n_unwind_trusted_stack
+  cmv      ca0, cs0
+#endif
+
 # if defined(__riscv_flen)
   .irp i,FROM_0_TO_31
     restore_fpr \i, a0

libunwind/src/UnwindRegistersSave.S

@@ -1273,6 +1273,19 @@ DEFINE_LIBUNWIND_FUNCTION(__unw_getcontext)
   .endr
 # endif
 
+#ifdef _LIBUNWIND_HAS_CHERI_LIB_C18N
+  // Store the trusted stack pointer
+  caddi     csp, csp, -32
+  sc        ca0, 0(csp)
+  sc        cra, 16(csp)
+  cmv       ca0, cra
+  call      dl_c18n_get_trusted_stack
+  lc        ca1, 0(csp)
+  lc        cra, 16(csp)
+  caddi     csp, csp, 32
+  sc        ca0, (__SIZEOF_CHERI_CAPABILITY__ * 32)(ca1)
+#endif
+
   li     a0, 0  // return UNW_ESUCCESS
   ret          // jump to ra
 END_LIBUNWIND_FUNCTION(__unw_getcontext)

libunwind/src/assembly.h

@@ -36,7 +36,7 @@
 #define SEPARATOR %%
 #elif defined(__riscv)
 #ifdef __CHERI_PURE_CAPABILITY__
-# define RISCV_FOFFSET (__SIZEOF_CHERI_CAPABILITY__ * 32)
+# define RISCV_FOFFSET (__SIZEOF_CHERI_CAPABILITY__ * 33)
 #else
 # define RISCV_ISIZE (__riscv_xlen / 8)
 # define RISCV_FOFFSET (RISCV_ISIZE * 32)