1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2020 Sean Anderson <seanga2@gmail.com>
4  */
5 
6 #include <console.h>
7 #include <log.h>
8 #include <asm/global_data.h>
9 #include <test/log.h>
10 #include <test/ut.h>
11 
12 DECLARE_GLOBAL_DATA_PTR;
13 
14 /* Test invalid options */
log_test_filter_invalid(struct unit_test_state * uts)15 static int log_test_filter_invalid(struct unit_test_state *uts)
16 {
17 	ut_asserteq(1, run_command("log filter-add -AD", 0));
18 	ut_asserteq(1, run_command("log filter-add -l1 -L1", 0));
19 	ut_asserteq(1, run_command("log filter-add -l1 -L1", 0));
20 	ut_asserteq(1, run_command("log filter-add -lfoo", 0));
21 	ut_asserteq(1, run_command("log filter-add -cfoo", 0));
22 	ut_asserteq(1, run_command("log filter-add -ccore -ccore -ccore -ccore "
23 				   "-ccore -ccore", 0));
24 
25 	return 0;
26 }
27 LOG_TEST_FLAGS(log_test_filter_invalid, UTF_CONSOLE);
28 
29 /* Test adding and removing filters */
log_test_filter(struct unit_test_state * uts)30 static int log_test_filter(struct unit_test_state *uts)
31 {
32 	bool any_found = false;
33 	bool filt1_found = false;
34 	bool filt2_found = false;
35 	char cmd[32];
36 	struct log_filter *filt;
37 	struct log_device *ldev;
38 	ulong filt1, filt2;
39 
40 #define create_filter(args, filter_num) do {\
41 	ut_assertok(run_command("log filter-add -p " args, 0)); \
42 	ut_assertok(strict_strtoul(uts->actual_str, 10, &(filter_num))); \
43 	ut_assert_console_end(); \
44 } while (0)
45 
46 	create_filter("", filt1);
47 	create_filter("-DL warning -cmmc -cspi -ffile", filt2);
48 
49 	ldev = log_device_find_by_name("console");
50 	ut_assertnonnull(ldev);
51 	list_for_each_entry(filt, &ldev->filter_head, sibling_node) {
52 		if (filt->filter_num == filt1) {
53 			filt1_found = true;
54 			ut_asserteq(0, filt->flags);
55 			ut_asserteq(LOGL_MAX, filt->level);
56 			ut_assertnull(filt->file_list);
57 		} else if (filt->filter_num == filt2) {
58 			filt2_found = true;
59 			ut_asserteq(LOGFF_HAS_CAT | LOGFF_DENY |
60 				    LOGFF_LEVEL_MIN, filt->flags);
61 			ut_asserteq(true, log_has_cat(filt->cat_list,
62 						      log_uc_cat(UCLASS_MMC)));
63 			ut_asserteq(true, log_has_cat(filt->cat_list,
64 						      log_uc_cat(UCLASS_SPI)));
65 			ut_asserteq(LOGL_WARNING, filt->level);
66 			ut_asserteq_str("file", filt->file_list);
67 		}
68 	}
69 	ut_asserteq(true, filt1_found);
70 	ut_asserteq(true, filt2_found);
71 
72 #define remove_filter(filter_num) do { \
73 	snprintf(cmd, sizeof(cmd), "log filter-remove %lu", filter_num); \
74 	ut_assertok(run_command(cmd, 0)); \
75 	ut_assert_console_end(); \
76 } while (0)
77 
78 	remove_filter(filt1);
79 	remove_filter(filt2);
80 
81 	filt1_found = false;
82 	filt2_found = false;
83 	list_for_each_entry(filt, &ldev->filter_head, sibling_node) {
84 		if (filt->filter_num == filt1)
85 			filt1_found = true;
86 		else if (filt->filter_num == filt2)
87 			filt2_found = true;
88 	}
89 	ut_asserteq(false, filt1_found);
90 	ut_asserteq(false, filt2_found);
91 
92 	create_filter("", filt1);
93 	create_filter("", filt2);
94 
95 	ut_assertok(run_command("log filter-remove -a", 0));
96 	ut_assert_console_end();
97 
98 	list_for_each_entry(filt, &ldev->filter_head, sibling_node)
99 		any_found = true;
100 	ut_asserteq(false, any_found);
101 
102 	return 0;
103 }
104 LOG_TEST_FLAGS(log_test_filter, UTF_CONSOLE);
105