mirror of
https://github.com/opencontainers/runc.git
synced 2026-07-11 06:03:57 +08:00
Kir Kolyshkin
3f65946756
A cgroup manager's Set method sets cgroup resources, but historically it was accepting configs.Cgroups. Refactor it to accept resources only. This is an improvement from the API point of view, as the method can not change cgroup configuration (such as path to the cgroup etc), it can only set (modify) its resources/limits. This also lays the foundation for complicated resource updates, as now Set has two sets of resources -- the one that was previously specified during cgroup manager creation (or the previous Set), and the one passed in the argument, so it could deduce the difference between these. This is a long term goal though. Signed-off-by: Kir Kolyshkin <kolyshkin@gmail.com>
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
// +build linux
|
|
|
|
package fs2
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/opencontainers/runc/libcontainer/cgroups"
|
|
"github.com/opencontainers/runc/libcontainer/cgroups/fscommon"
|
|
"github.com/opencontainers/runc/libcontainer/configs"
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func isPidsSet(r *configs.Resources) bool {
|
|
return r.PidsLimit != 0
|
|
}
|
|
|
|
func setPids(dirPath string, r *configs.Resources) error {
|
|
if !isPidsSet(r) {
|
|
return nil
|
|
}
|
|
if val := numToStr(r.PidsLimit); val != "" {
|
|
if err := fscommon.WriteFile(dirPath, "pids.max", val); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func statPidsFromCgroupProcs(dirPath string, stats *cgroups.Stats) error {
|
|
// if the controller is not enabled, let's read PIDS from cgroups.procs
|
|
// (or threads if cgroup.threads is enabled)
|
|
contents, err := fscommon.ReadFile(dirPath, "cgroup.procs")
|
|
if errors.Is(err, unix.ENOTSUP) {
|
|
contents, err = fscommon.ReadFile(dirPath, "cgroup.threads")
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pids := strings.Count(contents, "\n")
|
|
stats.PidsStats.Current = uint64(pids)
|
|
stats.PidsStats.Limit = 0
|
|
return nil
|
|
}
|
|
|
|
func statPids(dirPath string, stats *cgroups.Stats) error {
|
|
current, err := fscommon.GetCgroupParamUint(dirPath, "pids.current")
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return statPidsFromCgroupProcs(dirPath, stats)
|
|
}
|
|
return errors.Wrap(err, "failed to parse pids.current")
|
|
}
|
|
|
|
maxString, err := fscommon.GetCgroupParamString(dirPath, "pids.max")
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to parse pids.max")
|
|
}
|
|
|
|
// Default if pids.max == "max" is 0 -- which represents "no limit".
|
|
var max uint64
|
|
if maxString != "max" {
|
|
max, err = fscommon.ParseUint(maxString, 10, 64)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to parse pids.max - unable to parse %q as a uint from Cgroup file %q",
|
|
maxString, filepath.Join(dirPath, "pids.max"))
|
|
}
|
|
}
|
|
|
|
stats.PidsStats.Current = current
|
|
stats.PidsStats.Limit = max
|
|
return nil
|
|
}
|