Implement ptrace emulation of NT_ARM_PACA_KEYS/NT_ARM_PACG_KEYS

Author: rocallahan

src/Task.cc

@@ -827,6 +827,20 @@ void Task::on_syscall_exit_arch(int syscallno, const Registers& regs) {
                   2*sizeof(ARM64Arch::hw_bp));
               break;
             }
+            case NT_ARM_PACA_KEYS: {
+              auto set = ptrace_get_regs_set<Arch>(
+                  this, regs, sizeof(ARM64Arch::user_pac_address_keys));
+              struct iovec vec = { (void*)set.data(), sizeof(ARM64Arch::user_pac_address_keys) };
+              fallible_ptrace(PTRACE_SETREGSET, NT_ARM_PACA_KEYS, &vec);
+              break;
+            }
+            case NT_ARM_PACG_KEYS: {
+              auto set = ptrace_get_regs_set<Arch>(
+                  this, regs, sizeof(ARM64Arch::user_pac_generic_keys));
+              struct iovec vec = { (void*)set.data(), sizeof(ARM64Arch::user_pac_generic_keys) };
+              fallible_ptrace(PTRACE_SETREGSET, NT_ARM_PACG_KEYS, &vec);
+              break;
+            }
             case NT_X86_XSTATE: {
               if (auto extra_regs = tracee->extra_regs_fallible()) {
                 switch (extra_regs->format()) {

src/record_syscall.cc

@@ -2950,6 +2950,27 @@ static Switchable prepare_ptrace(RecordTask* t,
           }
           break;
         }
+        case NT_ARM_PACA_KEYS:
+        case NT_ARM_PACG_KEYS: {
+          if (Arch::arch() != aarch64) {
+            syscall_state.expect_errno = EINVAL;
+            emulate = false;
+            break;
+          }
+          RecordTask* tracee = verify_ptrace_target(t, syscall_state, tid);
+          if (tracee) {
+            bool ok = true;
+            vector<uint8_t> keys = tracee->pac_keys(&ok);
+            ASSERT(tracee, ok);
+            if ((int)t->regs().arg3() == NT_ARM_PACA_KEYS) {
+              keys.erase(keys.begin() + sizeof(ARM64Arch::user_pac_address_keys), keys.end());
+            } else {
+              keys.erase(keys.begin(), keys.begin() + sizeof(ARM64Arch::user_pac_address_keys));
+            }
+            ptrace_get_reg_set<Arch>(t, syscall_state, keys);
+          }
+          break;
+        }
         case NT_X86_XSTATE: {
           if (!Arch::is_x86ish()) {
             syscall_state.expect_errno = EINVAL;
@@ -3028,6 +3049,25 @@ static Switchable prepare_ptrace(RecordTask* t,
           }
           break;
         }
+        case NT_ARM_PACA_KEYS:
+        case NT_ARM_PACG_KEYS: {
+          if (Arch::arch() != aarch64) {
+            syscall_state.expect_errno = EINVAL;
+            emulate = false;
+            break;
+          }
+          RecordTask* tracee = verify_ptrace_target(t, syscall_state, tid);
+          if (tracee) {
+            if ((int)t->regs().arg3() == NT_ARM_PACA_KEYS) {
+              ptrace_verify_set_reg_set<Arch>(
+                t, sizeof(ARM64Arch::user_pac_address_keys), syscall_state);
+            } else {
+              ptrace_verify_set_reg_set<Arch>(
+                t, sizeof(ARM64Arch::user_pac_generic_keys), syscall_state);
+            }
+          }
+          break;
+        }
         case NT_X86_XSTATE: {
           if (!Arch::is_x86ish()) {
             syscall_state.expect_errno = EINVAL;