1#!/usr/bin/env python3 2# SPDX-License-Identifier: GPL-2.0 3 4import json 5import os 6import subprocess 7 8import _damon_sysfs 9 10def dump_damon_status_dict(pid): 11 try: 12 subprocess.check_output(['which', 'drgn'], stderr=subprocess.DEVNULL) 13 except: 14 return None, 'drgn not found' 15 file_dir = os.path.dirname(os.path.abspath(__file__)) 16 dump_script = os.path.join(file_dir, 'drgn_dump_damon_status.py') 17 rc = subprocess.call(['drgn', dump_script, pid, 'damon_dump_output'], 18 stderr=subprocess.DEVNULL) 19 if rc != 0: 20 return None, 'drgn fail' 21 try: 22 with open('damon_dump_output', 'r') as f: 23 return json.load(f), None 24 except Exception as e: 25 return None, 'json.load fail (%s)' % e 26 27def fail(expectation, status): 28 print('unexpected %s' % expectation) 29 print(json.dumps(status, indent=4)) 30 exit(1) 31 32def assert_true(condition, expectation, status): 33 if condition is not True: 34 fail(expectation, status) 35 36def assert_watermarks_committed(watermarks, dump): 37 wmark_metric_val = { 38 'none': 0, 39 'free_mem_rate': 1, 40 } 41 assert_true(dump['metric'] == wmark_metric_val[watermarks.metric], 42 'metric', dump) 43 assert_true(dump['interval'] == watermarks.interval, 'interval', dump) 44 assert_true(dump['high'] == watermarks.high, 'high', dump) 45 assert_true(dump['mid'] == watermarks.mid, 'mid', dump) 46 assert_true(dump['low'] == watermarks.low, 'low', dump) 47 48def assert_quota_goal_committed(qgoal, dump): 49 metric_val = { 50 'user_input': 0, 51 'some_mem_psi_us': 1, 52 'node_mem_used_bp': 2, 53 'node_mem_free_bp': 3, 54 } 55 assert_true(dump['metric'] == metric_val[qgoal.metric], 'metric', dump) 56 assert_true(dump['target_value'] == qgoal.target_value, 'target_value', 57 dump) 58 if qgoal.metric == 'user_input': 59 assert_true(dump['current_value'] == qgoal.current_value, 60 'current_value', dump) 61 assert_true(dump['nid'] == qgoal.nid, 'nid', dump) 62 63def assert_quota_committed(quota, dump): 64 assert_true(dump['reset_interval'] == quota.reset_interval_ms, 65 'reset_interval', dump) 66 assert_true(dump['ms'] == quota.ms, 'ms', dump) 67 assert_true(dump['sz'] == quota.sz, 'sz', dump) 68 for idx, qgoal in enumerate(quota.goals): 69 assert_quota_goal_committed(qgoal, dump['goals'][idx]) 70 assert_true(dump['weight_sz'] == quota.weight_sz_permil, 'weight_sz', dump) 71 assert_true(dump['weight_nr_accesses'] == quota.weight_nr_accesses_permil, 72 'weight_nr_accesses', dump) 73 assert_true( 74 dump['weight_age'] == quota.weight_age_permil, 'weight_age', dump) 75 76def main(): 77 kdamonds = _damon_sysfs.Kdamonds( 78 [_damon_sysfs.Kdamond( 79 contexts=[_damon_sysfs.DamonCtx( 80 targets=[_damon_sysfs.DamonTarget(pid=-1)], 81 schemes=[_damon_sysfs.Damos()], 82 )])]) 83 err = kdamonds.start() 84 if err is not None: 85 print('kdamond start failed: %s' % err) 86 exit(1) 87 88 status, err = dump_damon_status_dict(kdamonds.kdamonds[0].pid) 89 if err is not None: 90 print(err) 91 kdamonds.stop() 92 exit(1) 93 94 if len(status['contexts']) != 1: 95 fail('number of contexts', status) 96 97 ctx = status['contexts'][0] 98 attrs = ctx['attrs'] 99 if attrs['sample_interval'] != 5000: 100 fail('sample interval', status) 101 if attrs['aggr_interval'] != 100000: 102 fail('aggr interval', status) 103 if attrs['ops_update_interval'] != 1000000: 104 fail('ops updte interval', status) 105 106 if attrs['intervals_goal'] != { 107 'access_bp': 0, 'aggrs': 0, 108 'min_sample_us': 0, 'max_sample_us': 0}: 109 fail('intervals goal') 110 111 if attrs['min_nr_regions'] != 10: 112 fail('min_nr_regions') 113 if attrs['max_nr_regions'] != 1000: 114 fail('max_nr_regions') 115 116 if ctx['adaptive_targets'] != [ 117 { 'pid': 0, 'nr_regions': 0, 'regions_list': []}]: 118 fail('adaptive targets', status) 119 120 if len(ctx['schemes']) != 1: 121 fail('number of schemes', status) 122 123 scheme = ctx['schemes'][0] 124 if scheme['pattern'] != { 125 'min_sz_region': 0, 126 'max_sz_region': 2**64 - 1, 127 'min_nr_accesses': 0, 128 'max_nr_accesses': 2**32 - 1, 129 'min_age_region': 0, 130 'max_age_region': 2**32 - 1, 131 }: 132 fail('damos pattern', status) 133 if scheme['action'] != 9: # stat 134 fail('damos action', status) 135 if scheme['apply_interval_us'] != 0: 136 fail('damos apply interval', status) 137 if scheme['target_nid'] != -1: 138 fail('damos target nid', status) 139 140 migrate_dests = scheme['migrate_dests'] 141 if migrate_dests['nr_dests'] != 0: 142 fail('nr_dests', status) 143 if migrate_dests['node_id_arr'] != []: 144 fail('node_id_arr', status) 145 if migrate_dests['weight_arr'] != []: 146 fail('weight_arr', status) 147 148 assert_quota_committed(_damon_sysfs.DamosQuota(), scheme['quota']) 149 assert_watermarks_committed(_damon_sysfs.DamosWatermarks(), 150 scheme['wmarks']) 151 152 kdamonds.stop() 153 154if __name__ == '__main__': 155 main() 156