mirror of
https://github.com/torvalds/linux.git
synced 2024-11-29 15:41:36 +00:00
selftests/damon/_damon_sysfs: implement kdamonds start function
Extend the tests-writing-purpose DAMON sysfs control module to support the kdamonds start functionality. Link: https://lkml.kernel.org/r/20231212194810.54457-3-sj@kernel.org Signed-off-by: SeongJae Park <sj@kernel.org> Cc: Shuah Khan <shuah@kernel.org> Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
This commit is contained in:
SeongJae Park
Andrew Morton
parent
306abb63a8
commit
f5f0e5a2be
@ -1,5 +1,28 @@
|
||||
# SPDX-License-Identifier: GPL-2.0
|
||||
|
||||
import os
|
||||
|
||||
sysfs_root = '/sys/kernel/mm/damon/admin'
|
||||
|
||||
def write_file(path, string):
|
||||
"Returns error string if failed, or None otherwise"
|
||||
string = '%s' % string
|
||||
try:
|
||||
with open(path, 'w') as f:
|
||||
f.write(string)
|
||||
except Exception as e:
|
||||
return '%s' % e
|
||||
return None
|
||||
|
||||
def read_file(path):
|
||||
'''Returns the read content and error string. The read content is None if
|
||||
the reading failed'''
|
||||
try:
|
||||
with open(path, 'r') as f:
|
||||
return f.read(), None
|
||||
except Exception as e:
|
||||
return None, '%s' % e
|
||||
|
||||
class DamosAccessPattern:
|
||||
size = None
|
||||
nr_accesses = None
|
||||
@ -18,6 +41,35 @@ class DamosAccessPattern:
|
||||
if self.age == None:
|
||||
self.age = [0, 2**64 - 1]
|
||||
|
||||
def sysfs_dir(self):
|
||||
return os.path.join(self.scheme.sysfs_dir(), 'access_pattern')
|
||||
|
||||
def stage(self):
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'sz', 'min'), self.size[0])
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'sz', 'max'), self.size[1])
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(os.path.join(self.sysfs_dir(), 'nr_accesses', 'min'),
|
||||
self.nr_accesses[0])
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(os.path.join(self.sysfs_dir(), 'nr_accesses', 'max'),
|
||||
self.nr_accesses[1])
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'age', 'min'), self.age[0])
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'age', 'max'), self.age[1])
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
class Damos:
|
||||
action = None
|
||||
access_pattern = None
|
||||
@ -30,6 +82,39 @@ class Damos:
|
||||
self.access_pattern = access_pattern
|
||||
self.access_pattern.scheme = self
|
||||
|
||||
def sysfs_dir(self):
|
||||
return os.path.join(
|
||||
self.context.sysfs_dir(), 'schemes', '%d' % self.idx)
|
||||
|
||||
def stage(self):
|
||||
err = write_file(os.path.join(self.sysfs_dir(), 'action'), self.action)
|
||||
if err != None:
|
||||
return err
|
||||
err = self.access_pattern.stage()
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
# disable quotas
|
||||
err = write_file(os.path.join(self.sysfs_dir(), 'quotas', 'ms'), '0')
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'quotas', 'bytes'), '0')
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
# disable watermarks
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'watermarks', 'metric'), 'none')
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
# disable filters
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'filters', 'nr_filters'), '0')
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
class DamonTarget:
|
||||
pid = None
|
||||
# todo: Support target regions if test is made
|
||||
@ -39,6 +124,18 @@ class DamonTarget:
|
||||
def __init__(self, pid):
|
||||
self.pid = pid
|
||||
|
||||
def sysfs_dir(self):
|
||||
return os.path.join(
|
||||
self.context.sysfs_dir(), 'targets', '%d' % self.idx)
|
||||
|
||||
def stage(self):
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'regions', 'nr_regions'), '0')
|
||||
if err != None:
|
||||
return err
|
||||
return write_file(
|
||||
os.path.join(self.sysfs_dir(), 'pid_target'), self.pid)
|
||||
|
||||
class DamonAttrs:
|
||||
sample_us = None
|
||||
aggr_us = None
|
||||
@ -55,6 +152,40 @@ class DamonAttrs:
|
||||
self.min_nr_regions = min_nr_regions
|
||||
self.max_nr_regions = max_nr_regions
|
||||
|
||||
def interval_sysfs_dir(self):
|
||||
return os.path.join(self.context.sysfs_dir(), 'monitoring_attrs',
|
||||
'intervals')
|
||||
|
||||
def nr_regions_range_sysfs_dir(self):
|
||||
return os.path.join(self.context.sysfs_dir(), 'monitoring_attrs',
|
||||
'nr_regions')
|
||||
|
||||
def stage(self):
|
||||
err = write_file(os.path.join(self.interval_sysfs_dir(), 'sample_us'),
|
||||
self.sample_us)
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(os.path.join(self.interval_sysfs_dir(), 'aggr_us'),
|
||||
self.aggr_us)
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(os.path.join(self.interval_sysfs_dir(), 'update_us'),
|
||||
self.update_us)
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
err = write_file(
|
||||
os.path.join(self.nr_regions_range_sysfs_dir(), 'min'),
|
||||
self.min_nr_regions)
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
err = write_file(
|
||||
os.path.join(self.nr_regions_range_sysfs_dir(), 'max'),
|
||||
self.max_nr_regions)
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
class DamonCtx:
|
||||
ops = None
|
||||
monitoring_attrs = None
|
||||
@ -79,6 +210,46 @@ class DamonCtx:
|
||||
scheme.idx = idx
|
||||
scheme.context = self
|
||||
|
||||
def sysfs_dir(self):
|
||||
return os.path.join(self.kdamond.sysfs_dir(), 'contexts',
|
||||
'%d' % self.idx)
|
||||
|
||||
def stage(self):
|
||||
err = write_file(
|
||||
os.path.join(self.sysfs_dir(), 'operations'), self.ops)
|
||||
if err != None:
|
||||
return err
|
||||
err = self.monitoring_attrs.stage()
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
nr_targets_file = os.path.join(
|
||||
self.sysfs_dir(), 'targets', 'nr_targets')
|
||||
content, err = read_file(nr_targets_file)
|
||||
if err != None:
|
||||
return err
|
||||
if int(content) != len(self.targets):
|
||||
err = write_file(nr_targets_file, '%d' % len(self.targets))
|
||||
if err != None:
|
||||
return err
|
||||
for target in self.targets:
|
||||
err = target.stage()
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
nr_schemes_file = os.path.join(
|
||||
self.sysfs_dir(), 'schemes', 'nr_schemes')
|
||||
content, err = read_file(nr_schemes_file)
|
||||
if int(content) != len(self.schemes):
|
||||
err = write_file(nr_schemes_file, '%d' % len(self.schemes))
|
||||
if err != None:
|
||||
return err
|
||||
for scheme in self.schemes:
|
||||
err = scheme.stage()
|
||||
if err != None:
|
||||
return err
|
||||
return None
|
||||
|
||||
class Kdamond:
|
||||
state = None
|
||||
pid = None
|
||||
@ -92,6 +263,27 @@ class Kdamond:
|
||||
context.idx = idx
|
||||
context.kdamond = self
|
||||
|
||||
def sysfs_dir(self):
|
||||
return os.path.join(self.kdamonds.sysfs_dir(), '%d' % self.idx)
|
||||
|
||||
def start(self):
|
||||
nr_contexts_file = os.path.join(self.sysfs_dir(),
|
||||
'contexts', 'nr_contexts')
|
||||
content, err = read_file(nr_contexts_file)
|
||||
if err != None:
|
||||
return err
|
||||
if int(content) != len(self.contexts):
|
||||
err = write_file(nr_contexts_file, '%d' % len(self.contexts))
|
||||
if err != None:
|
||||
return err
|
||||
|
||||
for context in self.contexts:
|
||||
err = context.stage()
|
||||
if err != None:
|
||||
return err
|
||||
err = write_file(os.path.join(self.sysfs_dir(), 'state'), 'on')
|
||||
return err
|
||||
|
||||
class Kdamonds:
|
||||
kdamonds = []
|
||||
|
||||
@ -100,3 +292,17 @@ class Kdamonds:
|
||||
for idx, kdamond in enumerate(self.kdamonds):
|
||||
kdamond.idx = idx
|
||||
kdamond.kdamonds = self
|
||||
|
||||
def sysfs_dir(self):
|
||||
return os.path.join(sysfs_root, 'kdamonds')
|
||||
|
||||
def start(self):
|
||||
err = write_file(os.path.join(self.sysfs_dir(), 'nr_kdamonds'),
|
||||
'%s' % len(self.kdamonds))
|
||||
if err != None:
|
||||
return err
|
||||
for kdamond in self.kdamonds:
|
||||
err = kdamond.start()
|
||||
if err != None:
|
||||
return err
|
||||
return None
|
||||
|
||||
Loading…
Reference in New Issue
Block a user