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
|
package sys
import (
"runtime"
"testing"
"unsafe"
"github.com/cilium/ebpf/internal/unix"
"github.com/go-quicktest/qt"
)
func TestSigset(t *testing.T) {
const maxSignal = unix.Signal(unsafe.Sizeof(unix.Sigset_t{}) * 8)
// Type-infer a sigset word. This is a typed uint of 32 or 64 bits depending
// on the target architecture, so we can't use an untyped uint.
zero := unix.Sigset_t{}.Val[0]
words := len(unix.Sigset_t{}.Val)
var want, got unix.Sigset_t
// Flip the first bit of the first word.
if err := sigsetAdd(&got, 1); err != nil {
t.Fatal(err)
}
want.Val[0] = 1
if want != got {
t.Fatalf("expected first word to be 0x%x, got: 0x%x", want, got)
}
// And the last bit of the last word.
if err := sigsetAdd(&got, maxSignal); err != nil {
t.Fatal(err)
}
want.Val[words-1] = ^(^zero >> 1)
if want != got {
t.Fatalf("expected last word to be 0x%x, got: 0x%x", want, got)
}
if err := sigsetAdd(&got, maxSignal+1); err == nil {
t.Fatal("expected out-of-bounds add to be rejected")
}
if err := sigsetAdd(&got, -1); err == nil {
t.Fatal("expected negative signal to be rejected")
}
}
func TestProfilerSignal(t *testing.T) {
// Additional goroutine lock to make the PthreadSigmask below execute on the
// same OS thread as the functions under test. UnlockOSThread needs to be
// called as many times as LockOSThread to unlock the goroutine.
runtime.LockOSThread()
defer runtime.UnlockOSThread()
var old unix.Sigset_t
if err := unix.PthreadSigmask(0, nil, &old); err != nil {
t.Fatal("get sigmask:", err)
}
maskProfilerSignal()
var have unix.Sigset_t
if err := unix.PthreadSigmask(0, nil, &have); err != nil {
t.Fatal("get sigmask:", err)
}
want := have
qt.Assert(t, qt.IsNil(sigsetAdd(&want, unix.SIGPROF)))
qt.Assert(t, qt.Equals(have, want))
unmaskProfilerSignal()
if err := unix.PthreadSigmask(0, nil, &have); err != nil {
t.Fatal("get sigmask:", err)
}
qt.Assert(t, qt.Equals(have, old))
}
|
