From b2ea37401ed41b45f5497d04ab4f837a05204f8a Mon Sep 17 00:00:00 2001
From: Bhasker Hariharan <bhaskerh@google.com>
Date: Thu, 1 Apr 2021 15:36:11 -0700
Subject: [PATCH] Internal changes

PiperOrigin-RevId: 366344805
---
 pkg/abi/linux/ptrace_amd64.go | 5 +++++
 pkg/abi/linux/ptrace_arm64.go | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/pkg/abi/linux/ptrace_amd64.go b/pkg/abi/linux/ptrace_amd64.go
index 50e22fe7e..e722971f1 100644
--- a/pkg/abi/linux/ptrace_amd64.go
+++ b/pkg/abi/linux/ptrace_amd64.go
@@ -61,3 +61,8 @@ func (p *PtraceRegs) InstructionPointer() uint64 {
 func (p *PtraceRegs) StackPointer() uint64 {
 	return p.Rsp
 }
+
+// SetStackPointer sets the stack pointer to the specified value.
+func (p *PtraceRegs) SetStackPointer(sp uint64) {
+	p.Rsp = sp
+}
diff --git a/pkg/abi/linux/ptrace_arm64.go b/pkg/abi/linux/ptrace_arm64.go
index da36811d2..3d0906565 100644
--- a/pkg/abi/linux/ptrace_arm64.go
+++ b/pkg/abi/linux/ptrace_arm64.go
@@ -38,3 +38,8 @@ func (p *PtraceRegs) InstructionPointer() uint64 {
 func (p *PtraceRegs) StackPointer() uint64 {
 	return p.Sp
 }
+
+// SetStackPointer sets the stack pointer to the specified value.
+func (p *PtraceRegs) SetStackPointer(sp uint64) {
+	p.Sp = sp
+}