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
|
package rlimit
import (
"strings"
"testing"
"github.com/cilium/ebpf/internal/linux"
"github.com/cilium/ebpf/internal/unix"
"github.com/go-quicktest/qt"
)
func TestRemoveMemlock(t *testing.T) {
var before unix.Rlimit
qt.Assert(t, qt.IsNil(unix.Prlimit(0, unix.RLIMIT_MEMLOCK, nil, &before)))
err := RemoveMemlock()
if err != nil && strings.Contains(err.Error(), "operation not permitted") {
t.Skipf("Skipping test: %v", err)
}
qt.Assert(t, qt.IsNil(err))
var after unix.Rlimit
qt.Assert(t, qt.IsNil(unix.Prlimit(0, unix.RLIMIT_MEMLOCK, nil, &after)))
// We can't use testutils here due to an import cycle.
version, err := linux.KernelVersion()
qt.Assert(t, qt.IsNil(err))
if version.Less(unsupportedMemcgAccounting.MinimumVersion) {
qt.Assert(t, qt.Equals(after.Cur, unix.RLIM_INFINITY), qt.Commentf("cur should be INFINITY"))
qt.Assert(t, qt.Equals(after.Max, unix.RLIM_INFINITY), qt.Commentf("max should be INFINITY"))
} else {
qt.Assert(t, qt.Equals(after.Cur, before.Cur), qt.Commentf("cur should be unchanged"))
qt.Assert(t, qt.Equals(after.Max, before.Max), qt.Commentf("max should be unchanged"))
}
}
|
