1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
// Copyright 2018 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// +build amd64
package testutil
import (
"reflect"
"syscall"
)
// SetTestTarget sets the rip appropriately.
func SetTestTarget(regs *syscall.PtraceRegs, fn func()) {
regs.Rip = uint64(reflect.ValueOf(fn).Pointer())
}
// SetTouchTarget sets rax appropriately.
func SetTouchTarget(regs *syscall.PtraceRegs, target *uintptr) {
if target != nil {
regs.Rax = uint64(reflect.ValueOf(target).Pointer())
} else {
regs.Rax = 0
}
}
// RewindSyscall rewinds a syscall RIP.
func RewindSyscall(regs *syscall.PtraceRegs) {
regs.Rip -= 2
}
// SetTestRegs initializes registers to known values.
func SetTestRegs(regs *syscall.PtraceRegs) {
regs.R15 = 0x15
regs.R14 = 0x14
regs.R13 = 0x13
regs.R12 = 0x12
regs.Rbp = 0xb9
regs.Rbx = 0xb4
regs.R11 = 0x11
regs.R10 = 0x10
regs.R9 = 0x09
regs.R8 = 0x08
regs.Rax = 0x44
regs.Rcx = 0xc4
regs.Rdx = 0xd4
regs.Rsi = 0x51
regs.Rdi = 0xd1
regs.Rsp = 0x59
}
// CheckTestRegs checks that registers were twiddled per TwiddleRegs.
func CheckTestRegs(regs *syscall.PtraceRegs, full bool) (err error) {
if need := ^uint64(0x15); regs.R15 != need {
err = addRegisterMismatch(err, "R15", regs.R15, need)
}
if need := ^uint64(0x14); regs.R14 != need {
err = addRegisterMismatch(err, "R14", regs.R14, need)
}
if need := ^uint64(0x13); regs.R13 != need {
err = addRegisterMismatch(err, "R13", regs.R13, need)
}
if need := ^uint64(0x12); regs.R12 != need {
err = addRegisterMismatch(err, "R12", regs.R12, need)
}
if need := ^uint64(0xb9); regs.Rbp != need {
err = addRegisterMismatch(err, "Rbp", regs.Rbp, need)
}
if need := ^uint64(0xb4); regs.Rbx != need {
err = addRegisterMismatch(err, "Rbx", regs.Rbx, need)
}
if need := ^uint64(0x10); regs.R10 != need {
err = addRegisterMismatch(err, "R10", regs.R10, need)
}
if need := ^uint64(0x09); regs.R9 != need {
err = addRegisterMismatch(err, "R9", regs.R9, need)
}
if need := ^uint64(0x08); regs.R8 != need {
err = addRegisterMismatch(err, "R8", regs.R8, need)
}
if need := ^uint64(0x44); regs.Rax != need {
err = addRegisterMismatch(err, "Rax", regs.Rax, need)
}
if need := ^uint64(0xd4); regs.Rdx != need {
err = addRegisterMismatch(err, "Rdx", regs.Rdx, need)
}
if need := ^uint64(0x51); regs.Rsi != need {
err = addRegisterMismatch(err, "Rsi", regs.Rsi, need)
}
if need := ^uint64(0xd1); regs.Rdi != need {
err = addRegisterMismatch(err, "Rdi", regs.Rdi, need)
}
if need := ^uint64(0x59); regs.Rsp != need {
err = addRegisterMismatch(err, "Rsp", regs.Rsp, need)
}
// Rcx & R11 are ignored if !full is set.
if need := ^uint64(0x11); full && regs.R11 != need {
err = addRegisterMismatch(err, "R11", regs.R11, need)
}
if need := ^uint64(0xc4); full && regs.Rcx != need {
err = addRegisterMismatch(err, "Rcx", regs.Rcx, need)
}
return
}
var fsData uint64 = 0x55
var gsData uint64 = 0x85
// SetTestSegments initializes segments to known values.
func SetTestSegments(regs *syscall.PtraceRegs) {
regs.Fs_base = uint64(reflect.ValueOf(&fsData).Pointer())
regs.Gs_base = uint64(reflect.ValueOf(&gsData).Pointer())
}
// CheckTestSegments checks that registers were twiddled per TwiddleSegments.
func CheckTestSegments(regs *syscall.PtraceRegs) (err error) {
if regs.Rax != fsData {
err = addRegisterMismatch(err, "Rax", regs.Rax, fsData)
}
if regs.Rbx != gsData {
err = addRegisterMismatch(err, "Rbx", regs.Rcx, gsData)
}
return
}
|
