mirror of
https://github.com/kata-containers/cgroups-rs.git
synced 2026-08-05 02:13:23 +00:00
0c908cddf8
Setting (writing into) `memory.kmem.limit_in_bytes` is not supported in Linux kernel >= 5.16 (see kernel commit 58056f77502: "memcg, kmem: further deprecate kmem.limit_in_bytes"): https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git/commit/?id=58056f77502 If the write call in `set_kmem_limit()` returns EOPNOTSUPP, log a warning message but ignore the error (do nothing). Add a unit-test for `set_kmem_limit` in cgroups v1. Fix #81 Signed-off-by: Dov Murik <dov.murik1@il.ibm.com>
107 lines
3.2 KiB
Rust
107 lines
3.2 KiB
Rust
// Copyright (c) 2020 And Group
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0 or MIT
|
|
//
|
|
|
|
//! Integration tests about the hugetlb subsystem
|
|
use cgroups_rs::memory::{MemController, SetMemory};
|
|
use cgroups_rs::Controller;
|
|
use cgroups_rs::{Cgroup, MaxValue};
|
|
|
|
#[test]
|
|
fn test_disable_oom_killer() {
|
|
let h = cgroups_rs::hierarchies::auto();
|
|
let cg = Cgroup::new(h, String::from("test_disable_oom_killer"));
|
|
{
|
|
let mem_controller: &MemController = cg.controller_of().unwrap();
|
|
|
|
// before disable
|
|
let m = mem_controller.memory_stat();
|
|
assert!(!m.oom_control.oom_kill_disable);
|
|
|
|
// now only v1
|
|
if !mem_controller.v2() {
|
|
// disable oom killer
|
|
let r = mem_controller.disable_oom_killer();
|
|
assert!(r.is_ok());
|
|
|
|
// after disable
|
|
let m = mem_controller.memory_stat();
|
|
assert!(m.oom_control.oom_kill_disable);
|
|
}
|
|
}
|
|
cg.delete().unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn set_kmem_limit_v1() {
|
|
let h = cgroups_rs::hierarchies::auto();
|
|
if h.v2() {
|
|
return;
|
|
}
|
|
|
|
let cg = Cgroup::new(h, String::from("set_kmem_limit_v1"));
|
|
{
|
|
let mem_controller: &MemController = cg.controller_of().unwrap();
|
|
mem_controller.set_kmem_limit(1).unwrap();
|
|
}
|
|
cg.delete().unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn set_mem_v2() {
|
|
let h = cgroups_rs::hierarchies::auto();
|
|
if !h.v2() {
|
|
return;
|
|
}
|
|
|
|
let cg = Cgroup::new(h, String::from("set_mem_v2"));
|
|
{
|
|
let mem_controller: &MemController = cg.controller_of().unwrap();
|
|
|
|
// before disable
|
|
let m = mem_controller.get_mem().unwrap();
|
|
// case 1: get default value
|
|
assert_eq!(m.low, Some(MaxValue::Value(0)));
|
|
assert_eq!(m.min, Some(MaxValue::Value(0)));
|
|
assert_eq!(m.high, Some(MaxValue::Max));
|
|
assert_eq!(m.max, Some(MaxValue::Max));
|
|
|
|
// case 2: set parts
|
|
let m = SetMemory {
|
|
low: Some(MaxValue::Value(1024 * 1024 * 2)),
|
|
high: Some(MaxValue::Value(1024 * 1024 * 1024 * 2)),
|
|
min: Some(MaxValue::Value(1024 * 1024 * 3)),
|
|
max: None,
|
|
};
|
|
let r = mem_controller.set_mem(m);
|
|
assert!(r.is_ok());
|
|
|
|
let m = mem_controller.get_mem().unwrap();
|
|
// get
|
|
assert_eq!(m.low, Some(MaxValue::Value(1024 * 1024 * 2)));
|
|
assert_eq!(m.min, Some(MaxValue::Value(1024 * 1024 * 3)));
|
|
assert_eq!(m.high, Some(MaxValue::Value(1024 * 1024 * 1024 * 2)));
|
|
assert_eq!(m.max, Some(MaxValue::Max));
|
|
|
|
// case 3: set parts
|
|
let m = SetMemory {
|
|
max: Some(MaxValue::Value(1024 * 1024 * 1024 * 2)),
|
|
min: Some(MaxValue::Value(1024 * 1024 * 4)),
|
|
high: Some(MaxValue::Max),
|
|
low: None,
|
|
};
|
|
let r = mem_controller.set_mem(m);
|
|
assert!(r.is_ok());
|
|
|
|
let m = mem_controller.get_mem().unwrap();
|
|
// get
|
|
assert_eq!(m.low, Some(MaxValue::Value(1024 * 1024 * 2)));
|
|
assert_eq!(m.min, Some(MaxValue::Value(1024 * 1024 * 4)));
|
|
assert_eq!(m.max, Some(MaxValue::Value(1024 * 1024 * 1024 * 2)));
|
|
assert_eq!(m.high, Some(MaxValue::Max));
|
|
}
|
|
|
|
cg.delete().unwrap();
|
|
}
|