1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3
4# Kselftest frmework requirement - SKIP code is 4.
5ksft_skip=4
6
7ensure_write_succ()
8{
9	file=$1
10	content=$2
11	reason=$3
12
13	if ! echo "$content" > "$file"
14	then
15		echo "writing $content to $file failed"
16		echo "expected success because $reason"
17		exit 1
18	fi
19}
20
21ensure_write_fail()
22{
23	file=$1
24	content=$2
25	reason=$3
26
27	if (echo "$content" > "$file") 2> /dev/null
28	then
29		echo "writing $content to $file succeed ($fail_reason)"
30		echo "expected failure because $reason"
31		exit 1
32	fi
33}
34
35ensure_dir()
36{
37	dir=$1
38	to_ensure=$2
39	if [ "$to_ensure" = "exist" ] && [ ! -d "$dir" ]
40	then
41		echo "$dir dir is expected but not found"
42		exit 1
43	elif [ "$to_ensure" = "not_exist" ] && [ -d "$dir" ]
44	then
45		echo "$dir dir is not expected but found"
46		exit 1
47	fi
48}
49
50ensure_file()
51{
52	file=$1
53	to_ensure=$2
54	permission=$3
55	if [ "$to_ensure" = "exist" ]
56	then
57		if [ ! -f "$file" ]
58		then
59			echo "$file is expected but not found"
60			exit 1
61		fi
62		perm=$(stat -c "%a" "$file")
63		if [ ! "$perm" = "$permission" ]
64		then
65			echo "$file permission: expected $permission but $perm"
66			exit 1
67		fi
68	elif [ "$to_ensure" = "not_exist" ] && [ -f "$dir" ]
69	then
70		echo "$file is not expected but found"
71		exit 1
72	fi
73}
74
75test_range()
76{
77	range_dir=$1
78	ensure_dir "$range_dir" "exist"
79	ensure_file "$range_dir/min" "exist" 600
80	ensure_file "$range_dir/max" "exist" 600
81}
82
83test_tried_regions()
84{
85	tried_regions_dir=$1
86	ensure_dir "$tried_regions_dir" "exist"
87}
88
89test_stats()
90{
91	stats_dir=$1
92	ensure_dir "$stats_dir" "exist"
93	for f in nr_tried sz_tried nr_applied sz_applied qt_exceeds
94	do
95		ensure_file "$stats_dir/$f" "exist" "400"
96	done
97}
98
99test_filter()
100{
101	filter_dir=$1
102	ensure_file "$filter_dir/type" "exist" "600"
103	ensure_write_succ "$filter_dir/type" "anon" "valid input"
104	ensure_write_succ "$filter_dir/type" "memcg" "valid input"
105	ensure_write_fail "$filter_dir/type" "foo" "invalid input"
106	ensure_file "$filter_dir/matching" "exist" "600"
107	ensure_file "$filter_dir/memcg_path" "exist" "600"
108}
109
110test_filters()
111{
112	filters_dir=$1
113	ensure_dir "$filters_dir" "exist"
114	ensure_file "$filters_dir/nr_filters" "exist" "600"
115	ensure_write_succ  "$filters_dir/nr_filters" "1" "valid input"
116	test_filter "$filters_dir/0"
117
118	ensure_write_succ  "$filters_dir/nr_filters" "2" "valid input"
119	test_filter "$filters_dir/0"
120	test_filter "$filters_dir/1"
121
122	ensure_write_succ "$filters_dir/nr_filters" "0" "valid input"
123	ensure_dir "$filters_dir/0" "not_exist"
124	ensure_dir "$filters_dir/1" "not_exist"
125}
126
127test_watermarks()
128{
129	watermarks_dir=$1
130	ensure_dir "$watermarks_dir" "exist"
131	ensure_file "$watermarks_dir/metric" "exist" "600"
132	ensure_file "$watermarks_dir/interval_us" "exist" "600"
133	ensure_file "$watermarks_dir/high" "exist" "600"
134	ensure_file "$watermarks_dir/mid" "exist" "600"
135	ensure_file "$watermarks_dir/low" "exist" "600"
136}
137
138test_weights()
139{
140	weights_dir=$1
141	ensure_dir "$weights_dir" "exist"
142	ensure_file "$weights_dir/sz_permil" "exist" "600"
143	ensure_file "$weights_dir/nr_accesses_permil" "exist" "600"
144	ensure_file "$weights_dir/age_permil" "exist" "600"
145}
146
147test_quotas()
148{
149	quotas_dir=$1
150	ensure_dir "$quotas_dir" "exist"
151	ensure_file "$quotas_dir/ms" "exist" 600
152	ensure_file "$quotas_dir/bytes" "exist" 600
153	ensure_file "$quotas_dir/reset_interval_ms" "exist" 600
154	test_weights "$quotas_dir/weights"
155}
156
157test_access_pattern()
158{
159	access_pattern_dir=$1
160	ensure_dir "$access_pattern_dir" "exist"
161	test_range "$access_pattern_dir/age"
162	test_range "$access_pattern_dir/nr_accesses"
163	test_range "$access_pattern_dir/sz"
164}
165
166test_scheme()
167{
168	scheme_dir=$1
169	ensure_dir "$scheme_dir" "exist"
170	ensure_file "$scheme_dir/action" "exist" "600"
171	test_access_pattern "$scheme_dir/access_pattern"
172	test_quotas "$scheme_dir/quotas"
173	test_watermarks "$scheme_dir/watermarks"
174	test_filters "$scheme_dir/filters"
175	test_stats "$scheme_dir/stats"
176	test_tried_regions "$scheme_dir/tried_regions"
177}
178
179test_schemes()
180{
181	schemes_dir=$1
182	ensure_dir "$schemes_dir" "exist"
183	ensure_file "$schemes_dir/nr_schemes" "exist" 600
184
185	ensure_write_succ  "$schemes_dir/nr_schemes" "1" "valid input"
186	test_scheme "$schemes_dir/0"
187
188	ensure_write_succ  "$schemes_dir/nr_schemes" "2" "valid input"
189	test_scheme "$schemes_dir/0"
190	test_scheme "$schemes_dir/1"
191
192	ensure_write_succ "$schemes_dir/nr_schemes" "0" "valid input"
193	ensure_dir "$schemes_dir/0" "not_exist"
194	ensure_dir "$schemes_dir/1" "not_exist"
195}
196
197test_region()
198{
199	region_dir=$1
200	ensure_dir "$region_dir" "exist"
201	ensure_file "$region_dir/start" "exist" 600
202	ensure_file "$region_dir/end" "exist" 600
203}
204
205test_regions()
206{
207	regions_dir=$1
208	ensure_dir "$regions_dir" "exist"
209	ensure_file "$regions_dir/nr_regions" "exist" 600
210
211	ensure_write_succ  "$regions_dir/nr_regions" "1" "valid input"
212	test_region "$regions_dir/0"
213
214	ensure_write_succ  "$regions_dir/nr_regions" "2" "valid input"
215	test_region "$regions_dir/0"
216	test_region "$regions_dir/1"
217
218	ensure_write_succ "$regions_dir/nr_regions" "0" "valid input"
219	ensure_dir "$regions_dir/0" "not_exist"
220	ensure_dir "$regions_dir/1" "not_exist"
221}
222
223test_target()
224{
225	target_dir=$1
226	ensure_dir "$target_dir" "exist"
227	ensure_file "$target_dir/pid_target" "exist" "600"
228	test_regions "$target_dir/regions"
229}
230
231test_targets()
232{
233	targets_dir=$1
234	ensure_dir "$targets_dir" "exist"
235	ensure_file "$targets_dir/nr_targets" "exist" 600
236
237	ensure_write_succ  "$targets_dir/nr_targets" "1" "valid input"
238	test_target "$targets_dir/0"
239
240	ensure_write_succ  "$targets_dir/nr_targets" "2" "valid input"
241	test_target "$targets_dir/0"
242	test_target "$targets_dir/1"
243
244	ensure_write_succ "$targets_dir/nr_targets" "0" "valid input"
245	ensure_dir "$targets_dir/0" "not_exist"
246	ensure_dir "$targets_dir/1" "not_exist"
247}
248
249test_intervals()
250{
251	intervals_dir=$1
252	ensure_dir "$intervals_dir" "exist"
253	ensure_file "$intervals_dir/aggr_us" "exist" "600"
254	ensure_file "$intervals_dir/sample_us" "exist" "600"
255	ensure_file "$intervals_dir/update_us" "exist" "600"
256}
257
258test_monitoring_attrs()
259{
260	monitoring_attrs_dir=$1
261	ensure_dir "$monitoring_attrs_dir" "exist"
262	test_intervals "$monitoring_attrs_dir/intervals"
263	test_range "$monitoring_attrs_dir/nr_regions"
264}
265
266test_context()
267{
268	context_dir=$1
269	ensure_dir "$context_dir" "exist"
270	ensure_file "$context_dir/avail_operations" "exit" 400
271	ensure_file "$context_dir/operations" "exist" 600
272	test_monitoring_attrs "$context_dir/monitoring_attrs"
273	test_targets "$context_dir/targets"
274	test_schemes "$context_dir/schemes"
275}
276
277test_contexts()
278{
279	contexts_dir=$1
280	ensure_dir "$contexts_dir" "exist"
281	ensure_file "$contexts_dir/nr_contexts" "exist" 600
282
283	ensure_write_succ  "$contexts_dir/nr_contexts" "1" "valid input"
284	test_context "$contexts_dir/0"
285
286	ensure_write_fail "$contexts_dir/nr_contexts" "2" "only 0/1 are supported"
287	test_context "$contexts_dir/0"
288
289	ensure_write_succ "$contexts_dir/nr_contexts" "0" "valid input"
290	ensure_dir "$contexts_dir/0" "not_exist"
291}
292
293test_kdamond()
294{
295	kdamond_dir=$1
296	ensure_dir "$kdamond_dir" "exist"
297	ensure_file "$kdamond_dir/state" "exist" "600"
298	ensure_file "$kdamond_dir/pid" "exist" 400
299	test_contexts "$kdamond_dir/contexts"
300}
301
302test_kdamonds()
303{
304	kdamonds_dir=$1
305	ensure_dir "$kdamonds_dir" "exist"
306
307	ensure_file "$kdamonds_dir/nr_kdamonds" "exist" "600"
308
309	ensure_write_succ  "$kdamonds_dir/nr_kdamonds" "1" "valid input"
310	test_kdamond "$kdamonds_dir/0"
311
312	ensure_write_succ  "$kdamonds_dir/nr_kdamonds" "2" "valid input"
313	test_kdamond "$kdamonds_dir/0"
314	test_kdamond "$kdamonds_dir/1"
315
316	ensure_write_succ "$kdamonds_dir/nr_kdamonds" "0" "valid input"
317	ensure_dir "$kdamonds_dir/0" "not_exist"
318	ensure_dir "$kdamonds_dir/1" "not_exist"
319}
320
321test_damon_sysfs()
322{
323	damon_sysfs=$1
324	if [ ! -d "$damon_sysfs" ]
325	then
326		echo "$damon_sysfs not found"
327		exit $ksft_skip
328	fi
329
330	test_kdamonds "$damon_sysfs/kdamonds"
331}
332
333check_dependencies()
334{
335	if [ $EUID -ne 0 ]
336	then
337		echo "Run as root"
338		exit $ksft_skip
339	fi
340}
341
342check_dependencies
343test_damon_sysfs "/sys/kernel/mm/damon/admin"
344