mirror of
https://github.com/opencontainers/runc.git
synced 2026-07-11 14:13:58 +08:00
Kir Kolyshkin
771903608c
It has been pointed out that some controllers can not accept multiple lines of output at once. In particular, io.max can only set one device at a time. Practically, the only multi-line resource values we can get come from unified.* -- let's write those line by line. Add a test case. Reported-by: Tao Shen <shentaoskyking@gmail.com> Signed-off-by: Kir Kolyshkin <kolyshkin@gmail.com>
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package cgroups
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/opencontainers/runc/internal/testutil"
|
|
)
|
|
|
|
func TestWriteCgroupFileHandlesInterrupt(t *testing.T) {
|
|
testutil.SkipOnCentOS(t, "Flaky (see #3418)", 7)
|
|
|
|
const (
|
|
memoryCgroupMount = "/sys/fs/cgroup/memory"
|
|
memoryLimit = "memory.limit_in_bytes"
|
|
)
|
|
if _, err := os.Stat(memoryCgroupMount); err != nil {
|
|
// most probably cgroupv2
|
|
t.Skip(err)
|
|
}
|
|
|
|
cgroupName := fmt.Sprintf("test-eint-%d", time.Now().Nanosecond())
|
|
cgroupPath := filepath.Join(memoryCgroupMount, cgroupName)
|
|
if err := os.MkdirAll(cgroupPath, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(cgroupPath)
|
|
|
|
if _, err := os.Stat(filepath.Join(cgroupPath, memoryLimit)); err != nil {
|
|
// either cgroupv2, or memory controller is not available
|
|
t.Skip(err)
|
|
}
|
|
|
|
for i := 0; i < 100000; i++ {
|
|
limit := 1024*1024 + i
|
|
if err := WriteFile(cgroupPath, memoryLimit, strconv.Itoa(limit)); err != nil {
|
|
t.Fatalf("Failed to write %d on attempt %d: %+v", limit, i, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestOpenat2(t *testing.T) {
|
|
if !IsCgroup2UnifiedMode() {
|
|
// The reason is many test cases below test opening files from
|
|
// the top-level directory, where cgroup v1 has no files.
|
|
t.Skip("test requires cgroup v2")
|
|
}
|
|
|
|
// Make sure we test openat2, not its fallback.
|
|
openFallback = func(_ string, _ int, _ os.FileMode) (*os.File, error) {
|
|
return nil, errors.New("fallback")
|
|
}
|
|
defer func() { openFallback = openAndCheck }()
|
|
|
|
for _, tc := range []struct{ dir, file string }{
|
|
{"/sys/fs/cgroup", "cgroup.controllers"},
|
|
{"/sys/fs/cgroup", "/cgroup.controllers"},
|
|
{"/sys/fs/cgroup/", "cgroup.controllers"},
|
|
{"/sys/fs/cgroup/", "/cgroup.controllers"},
|
|
{"/", "/sys/fs/cgroup/cgroup.controllers"},
|
|
{"/", "sys/fs/cgroup/cgroup.controllers"},
|
|
{"/sys/fs/cgroup/cgroup.controllers", ""},
|
|
} {
|
|
fd, err := OpenFile(tc.dir, tc.file, os.O_RDONLY)
|
|
if err != nil {
|
|
t.Errorf("case %+v: %v", tc, err)
|
|
}
|
|
fd.Close()
|
|
}
|
|
}
|
|
|
|
func BenchmarkWriteFile(b *testing.B) {
|
|
TestMode = true
|
|
defer func() { TestMode = false }()
|
|
|
|
dir := b.TempDir()
|
|
tc := []string{
|
|
"one",
|
|
"one\ntwo\nthree",
|
|
"10:200 foo=bar boo=far\n300:1200 something=other\ndefault 45000\n",
|
|
"\n\n\n\n\n\n\n\n",
|
|
}
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
for _, val := range tc {
|
|
if err := WriteFileByLine(dir, "file", val); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
}
|