aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tools/testing/selftests/damon/damos_quota.py
blob: 7d4c6bb2e3cd27cf42d9d08439475cccf6ca3f08 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#!/usr/bin/env python3
# SPDX-License-Identifier: GPL-2.0

import subprocess
import time

import _damon_sysfs

def main():
    # access two 10 MiB memory regions, 2 second per each
    sz_region = 10 * 1024 * 1024
    proc = subprocess.Popen(['./access_memory', '2', '%d' % sz_region, '2000'])

    # Set quota up to 1 MiB per 100 ms
    sz_quota = 1024 * 1024 # 1 MiB
    quota_reset_interval = 100 # 100 ms
    kdamonds = _damon_sysfs.Kdamonds([_damon_sysfs.Kdamond(
            contexts=[_damon_sysfs.DamonCtx(
                ops='vaddr',
                targets=[_damon_sysfs.DamonTarget(pid=proc.pid)],
                schemes=[_damon_sysfs.Damos(
                    access_pattern=_damon_sysfs.DamosAccessPattern(
                        # >= 25% access rate, >= 200ms age
                        nr_accesses=[5, 20], age=[2, 2**64 - 1]),
                    quota=_damon_sysfs.DamosQuota(
                        sz=sz_quota, reset_interval_ms=quota_reset_interval)
                    )] # schemes
                )] # contexts
            )]) # kdamonds

    err = kdamonds.start()
    if err != None:
        print('kdamond start failed: %s' % err)
        exit(1)

    wss_collected = []
    nr_quota_exceeds = 0
    while proc.poll() == None:
        time.sleep(0.1)
        err = kdamonds.kdamonds[0].update_schemes_tried_bytes()
        if err != None:
            print('tried bytes update failed: %s' % err)
            exit(1)
        err = kdamonds.kdamonds[0].update_schemes_stats()
        if err != None:
            print('stats update failed: %s' % err)
            exit(1)

        scheme = kdamonds.kdamonds[0].contexts[0].schemes[0]
        wss_collected.append(scheme.tried_bytes)
        nr_quota_exceeds = scheme.stats.qt_exceeds

    wss_collected.sort()
    for wss in wss_collected:
        if wss > sz_quota:
            print('quota is not kept: %s > %s' % (wss, sz_quota))
            print('collected samples are as below')
            print('\n'.join(['%d' % wss for wss in wss_collected]))
            exit(1)

    if nr_quota_exceeds < len(wss_collected):
        print('quota is not always exceeded: %d > %d' %
              (len(wss_collected), nr_quota_exceeds))
        exit(1)

if __name__ == '__main__':
    main()