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
|
package psx
import (
"runtime"
"sync"
"syscall"
"testing"
)
func TestSyscall3(t *testing.T) {
want := syscall.Getpid()
if got, _, err := Syscall3(syscall.SYS_GETPID, 0, 0, 0); err != 0 {
t.Errorf("failed to get PID via libpsx: %v", err)
} else if int(got) != want {
t.Errorf("pid mismatch: got=%d want=%d", got, want)
}
if got, _, err := Syscall3(syscall.SYS_CAPGET, 0, 0, 0); err != 14 {
t.Errorf("malformed capget returned %d: %v (want 14: %v)", err, err, syscall.Errno(14))
} else if ^got != 0 {
t.Errorf("malformed capget did not return -1, got=%d", got)
}
}
func TestSyscall6(t *testing.T) {
want := syscall.Getpid()
if got, _, err := Syscall6(syscall.SYS_GETPID, 0, 0, 0, 0, 0, 0); err != 0 {
t.Errorf("failed to get PID via libpsx: %v", err)
} else if int(got) != want {
t.Errorf("pid mismatch: got=%d want=%d", got, want)
}
if got, _, err := Syscall6(syscall.SYS_CAPGET, 0, 0, 0, 0, 0, 0); err != 14 {
t.Errorf("malformed capget errno %d: %v (want 14: %v)", err, err, syscall.Errno(14))
} else if ^got != 0 {
t.Errorf("malformed capget did not return -1, got=%d", got)
}
}
// killAThread locks the goroutine to a thread and exits. This has the
// effect of making the go runtime terminate the thread.
func killAThread(c <-chan struct{}) {
runtime.LockOSThread()
<-c
}
// Test state is mirrored as expected.
func TestShared(t *testing.T) {
const prGetKeepCaps = 7
const prSetKeepCaps = 8
var wg sync.WaitGroup
newTracker := func() chan<- uintptr {
ch := make(chan uintptr)
go func() {
runtime.LockOSThread()
defer wg.Done()
tid := syscall.Gettid()
for {
if _, ok := <-ch; !ok {
break
}
val, ok := <-ch
if !ok {
break
}
got, _, e := Syscall3(syscall.SYS_PRCTL, prGetKeepCaps, 0, 0)
if e != 0 {
t.Fatalf("[%d] psx:prctl(GET_KEEPCAPS) ?= %d failed: %v", tid, val, syscall.Errno(e))
}
if got != val {
t.Errorf("[%d] bad keepcaps value: got=%d, want=%d", tid, got, val)
}
if _, ok := <-ch; !ok {
break
}
}
}()
return ch
}
var tracked []chan<- uintptr
for i := 0; i <= 10; i++ {
val := uintptr(i & 1)
if _, _, e := Syscall3(syscall.SYS_PRCTL, prSetKeepCaps, val, 0); e != 0 {
t.Fatalf("[%d] psx:prctl(SET_KEEPCAPS, %d) failed: %v", i, i&1, syscall.Errno(e))
}
wg.Add(1)
tracked = append(tracked, newTracker())
for _, ch := range tracked {
ch <- 2 // start serialization.
ch <- val // definitely written after change.
ch <- 3 // end serialization.
}
}
for _, ch := range tracked {
close(ch)
}
wg.Wait()
}
// Test to confirm no regression against:
//
// https://github.com/golang/go/issues/42494
func TestThreadChurn(t *testing.T) {
const prSetKeepCaps = 8
for j := 0; j < 4; j++ {
kill := (j & 1) != 0
sysc := (j & 2) != 0
t.Logf("[%d] testing kill=%v, sysc=%v", j, kill, sysc)
for i := 50; i > 0; i-- {
if kill {
c := make(chan struct{})
go killAThread(c)
close(c)
}
if sysc {
if _, _, e := Syscall3(syscall.SYS_PRCTL, prSetKeepCaps, uintptr(i&1), 0); e != 0 {
t.Fatalf("[%d] psx:prctl(SET_KEEPCAPS, %d) failed: %v", i, i&1, syscall.Errno(e))
}
}
}
t.Logf("[%d] PASSED kill=%v, sysc=%v", j, kill, sysc)
}
}
|
