linux_cgroups_relative_hugetlb.go

package main

import (
	"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 main() {
	page := "2MB"
	var pageSize uint64 = 2 * 1024 * 1024 // 2MB in bytes
	limit := 100 * pageSize
	g := util.GetDefaultGenerator()
	g.SetLinuxCgroupsPath(cgroups.RelCgroupPath)
	g.AddLinuxResourcesHugepageLimit(page, limit)
	err := util.RuntimeOutsideValidate(g, func(config *rspec.Spec, state *rspec.State) error {
		t := tap.New()
		t.Header(0)

		cg, err := cgroups.FindCgroup()
		t.Ok((err == nil), "find hugetlb cgroup")
		if err != nil {
			t.Diagnostic(err.Error())
			t.AutoPlan()
			return nil
		}

		lhd, err := cg.GetHugepageLimitData(state.Pid, config.Linux.CgroupsPath)
		t.Ok((err == nil), "get hugetlb cgroup data")
		if err != nil {
			t.Diagnostic(err.Error())
			t.AutoPlan()
			return nil
		}

		found := false
		for _, lhl := range lhd {
			if lhl.Pagesize == page {
				found = true
				t.Ok(lhl.Limit == limit, "hugepage limit is set correctly")
				t.Diagnosticf("expect: %d, actual: %d", limit, lhl.Limit)
			}
		}
		t.Ok(found, "hugepage limit found")

		t.AutoPlan()
		return nil
	})

	if err != nil {
		util.Fatal(err)
	}
}