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
|
package main
import (
"fmt"
"runtime"
"github.com/mndrix/tap-go"
rspec "github.com/opencontainers/runtime-spec/specs-go"
"github.com/opencontainers/runtime-tools/cgroups"
"github.com/opencontainers/runtime-tools/validation/util"
)
func testHugetlbCgroups() error {
t := tap.New()
t.Header(0)
defer t.AutoPlan()
// limit =~ 100 * page size
// NOTE: on some systems, pagesize "1GB" doesn't seem to work.
// Ideally we should auto-detect the value.
cases := []struct {
page string
limit uint64
}{
{"2MB", 100 * 2 * 1024 * 1024},
{"1GB", 100 * 1024 * 1024 * 1024},
{"2MB", 100 * 2 * 1024 * 1024},
{"1GB", 100 * 1024 * 1024 * 1024},
}
for _, c := range cases {
g, err := util.GetDefaultGenerator()
if err != nil {
return err
}
g.SetLinuxCgroupsPath(cgroups.AbsCgroupPath)
g.AddLinuxResourcesHugepageLimit(c.page, c.limit)
err = util.RuntimeOutsideValidate(g, t, func(config *rspec.Spec, t *tap.T, state *rspec.State) error {
cg, err := cgroups.FindCgroup()
if err != nil {
return err
}
lhd, err := cg.GetHugepageLimitData(state.Pid, config.Linux.CgroupsPath)
if err != nil {
return err
}
for _, lhl := range lhd {
if lhl.Pagesize != c.page {
continue
}
t.Ok(lhl.Limit == c.limit, "hugepage limit is set correctly")
t.Diagnosticf("expect: %d, actual: %d", c.limit, lhl.Limit)
}
return nil
})
if err != nil {
return err
}
}
return nil
}
func testWrongHugetlb() error {
// We deliberately set the page size to a wrong value, "3MB", to see
// if the container really returns an error.
page := "3MB"
var limit uint64 = 100 * 3 * 1024 * 1024
g, err := util.GetDefaultGenerator()
if err != nil {
return err
}
t := tap.New()
t.Header(0)
defer t.AutoPlan()
g.SetLinuxCgroupsPath(cgroups.AbsCgroupPath)
g.AddLinuxResourcesHugepageLimit(page, limit)
err = util.RuntimeOutsideValidate(g, t, func(config *rspec.Spec, t *tap.T, state *rspec.State) error {
return nil
})
t.Ok(err != nil, "hugepage invalid pagesize results in an errror")
if err == nil {
t.Diagnosticf("expect: err != nil, actual: err == nil")
}
return err
}
func main() {
if "linux" != runtime.GOOS {
util.Fatal(fmt.Errorf("linux-specific cgroup test"))
}
if err := testHugetlbCgroups(); err != nil {
util.Fatal(err)
}
if err := testWrongHugetlb(); err == nil {
util.Fatal(err)
}
}
|
