Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include "alloc-util.h"
4 : : #include "env-util.h"
5 : : #include "log.h"
6 : : #include "macro.h"
7 : : #include "proc-cmdline.h"
8 : : #include "special.h"
9 : : #include "string-util.h"
10 : : #include "util.h"
11 : :
12 : : static int obj;
13 : :
14 : 24 : static int parse_item(const char *key, const char *value, void *data) {
15 [ - + ]: 24 : assert_se(key);
16 [ - + ]: 24 : assert_se(data == &obj);
17 : :
18 [ + - ]: 24 : log_info("kernel cmdline option <%s> = <%s>", key, strna(value));
19 : 24 : return 0;
20 : : }
21 : :
22 : 4 : static void test_proc_cmdline_parse(void) {
23 [ + - ]: 4 : log_info("/* %s */", __func__);
24 : :
25 [ - + ]: 4 : assert_se(proc_cmdline_parse(parse_item, &obj, PROC_CMDLINE_STRIP_RD_PREFIX) >= 0);
26 : 4 : }
27 : :
28 : 4 : static void test_proc_cmdline_override(void) {
29 [ + - ]: 4 : log_info("/* %s */", __func__);
30 : :
31 [ - + ]: 4 : assert_se(putenv((char*) "SYSTEMD_PROC_CMDLINE=foo_bar=quux wuff-piep=tuet zumm some_arg_with_space='foo bar' and_one_more=\"zzz aaa\"") == 0);
32 : :
33 : : /* Test if the override works */
34 : 4 : _cleanup_free_ char *line = NULL, *value = NULL;
35 [ - + ]: 4 : assert_se(proc_cmdline(&line) >= 0);
36 : :
37 : : /* Test if parsing makes uses of the override */
38 [ - + ]: 4 : assert_se(streq(line, "foo_bar=quux wuff-piep=tuet zumm some_arg_with_space='foo bar' and_one_more=\"zzz aaa\""));
39 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("foo_bar", 0, &value) > 0 && streq_ptr(value, "quux"));
40 : 4 : value = mfree(value);
41 : :
42 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("some_arg_with_space", 0, &value) > 0 && streq_ptr(value, "foo bar"));
43 : 4 : value = mfree(value);
44 : :
45 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("and_one_more", 0, &value) > 0 && streq_ptr(value, "zzz aaa"));
46 : 4 : value = mfree(value);
47 : 4 : }
48 : :
49 : 72 : static int parse_item_given(const char *key, const char *value, void *data) {
50 [ - + ]: 72 : assert_se(key);
51 [ - + ]: 72 : assert_se(data);
52 : :
53 : 72 : bool *strip = data;
54 : :
55 [ + - ]: 72 : log_info("%s: option <%s> = <%s>", __func__, key, strna(value));
56 [ + + ]: 72 : if (proc_cmdline_key_streq(key, "foo_bar"))
57 [ - + ]: 16 : assert_se(streq(value, "quux"));
58 [ + + ]: 56 : else if (proc_cmdline_key_streq(key, "wuff-piep"))
59 [ - + ]: 16 : assert_se(streq(value, "tuet "));
60 [ + + ]: 40 : else if (proc_cmdline_key_streq(key, "space"))
61 [ - + ]: 16 : assert_se(streq(value, "x y z"));
62 [ + + ]: 24 : else if (proc_cmdline_key_streq(key, "miepf"))
63 [ - + ]: 16 : assert_se(streq(value, "uuu"));
64 [ + - + + : 8 : else if (in_initrd() && *strip && proc_cmdline_key_streq(key, "zumm"))
+ - ]
65 [ - + ]: 4 : assert_se(!value);
66 [ + - + - : 4 : else if (in_initrd() && !*strip && proc_cmdline_key_streq(key, "rd.zumm"))
+ - ]
67 [ - + ]: 4 : assert_se(!value);
68 : : else
69 : 0 : assert_not_reached("Bad key!");
70 : :
71 : 72 : return 0;
72 : : }
73 : :
74 : 8 : static void test_proc_cmdline_given(bool flip_initrd) {
75 [ + - ]: 8 : log_info("/* %s (flip: %s) */", __func__, yes_no(flip_initrd));
76 : :
77 [ + + ]: 8 : if (flip_initrd)
78 : 4 : in_initrd_force(!in_initrd());
79 : :
80 : 8 : bool t = true, f = false;
81 [ - + ]: 8 : assert_se(proc_cmdline_parse_given("foo_bar=quux wuff-piep=\"tuet \" rd.zumm space='x y z' miepf=\"uuu\"",
82 : : parse_item_given, &t, PROC_CMDLINE_STRIP_RD_PREFIX) >= 0);
83 : :
84 [ - + ]: 8 : assert_se(proc_cmdline_parse_given("foo_bar=quux wuff-piep=\"tuet \" rd.zumm space='x y z' miepf=\"uuu\"",
85 : : parse_item_given, &f, 0) >= 0);
86 : :
87 [ + + ]: 8 : if (flip_initrd)
88 : 4 : in_initrd_force(!in_initrd());
89 : 8 : }
90 : :
91 : 4 : static void test_proc_cmdline_get_key(void) {
92 : 4 : _cleanup_free_ char *value = NULL;
93 : :
94 [ + - ]: 4 : log_info("/* %s */", __func__);
95 [ - + ]: 4 : assert_se(putenv((char*) "SYSTEMD_PROC_CMDLINE=foo_bar=quux wuff-piep=tuet zumm spaaace='ö ü ß' ticks=\"''\"\n\nkkk=uuu\n\n\n") == 0);
96 : :
97 [ - + ]: 4 : assert_se(proc_cmdline_get_key("", 0, &value) == -EINVAL);
98 [ - + ]: 4 : assert_se(proc_cmdline_get_key("abc", 0, NULL) == 0);
99 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("abc", 0, &value) == 0 && value == NULL);
100 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("abc", PROC_CMDLINE_VALUE_OPTIONAL, &value) == 0 && value == NULL);
101 : :
102 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("foo_bar", 0, &value) > 0 && streq_ptr(value, "quux"));
103 : 4 : value = mfree(value);
104 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("foo_bar", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "quux"));
105 : 4 : value = mfree(value);
106 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("foo-bar", 0, &value) > 0 && streq_ptr(value, "quux"));
107 : 4 : value = mfree(value);
108 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("foo-bar", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "quux"));
109 : 4 : value = mfree(value);
110 [ - + ]: 4 : assert_se(proc_cmdline_get_key("foo-bar", 0, NULL) == 0);
111 [ - + ]: 4 : assert_se(proc_cmdline_get_key("foo-bar", PROC_CMDLINE_VALUE_OPTIONAL, NULL) == -EINVAL);
112 : :
113 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("wuff-piep", 0, &value) > 0 && streq_ptr(value, "tuet"));
114 : 4 : value = mfree(value);
115 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("wuff-piep", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "tuet"));
116 : 4 : value = mfree(value);
117 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("wuff_piep", 0, &value) > 0 && streq_ptr(value, "tuet"));
118 : 4 : value = mfree(value);
119 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("wuff_piep", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "tuet"));
120 : 4 : value = mfree(value);
121 [ - + ]: 4 : assert_se(proc_cmdline_get_key("wuff_piep", 0, NULL) == 0);
122 [ - + ]: 4 : assert_se(proc_cmdline_get_key("wuff_piep", PROC_CMDLINE_VALUE_OPTIONAL, NULL) == -EINVAL);
123 : :
124 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("zumm", 0, &value) == 0 && value == NULL);
125 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("zumm", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && value == NULL);
126 [ - + ]: 4 : assert_se(proc_cmdline_get_key("zumm", 0, NULL) > 0);
127 : :
128 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("spaaace", 0, &value) > 0 && streq_ptr(value, "ö ü ß"));
129 : 4 : value = mfree(value);
130 : :
131 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("ticks", 0, &value) > 0 && streq_ptr(value, "''"));
132 : 4 : value = mfree(value);
133 : :
134 [ + - - + ]: 4 : assert_se(proc_cmdline_get_key("kkk", 0, &value) > 0 && streq_ptr(value, "uuu"));
135 : 4 : }
136 : :
137 : 4 : static void test_proc_cmdline_get_bool(void) {
138 : 4 : bool value = false;
139 : :
140 [ + - ]: 4 : log_info("/* %s */", __func__);
141 [ - + ]: 4 : assert_se(putenv((char*) "SYSTEMD_PROC_CMDLINE=foo_bar bar-waldo=1 x_y-z=0 quux=miep\nda=yes\nthe=1") == 0);
142 : :
143 [ - + ]: 4 : assert_se(proc_cmdline_get_bool("", &value) == -EINVAL);
144 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("abc", &value) == 0 && value == false);
145 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("foo_bar", &value) > 0 && value == true);
146 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("foo-bar", &value) > 0 && value == true);
147 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("bar-waldo", &value) > 0 && value == true);
148 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("bar_waldo", &value) > 0 && value == true);
149 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("x_y-z", &value) > 0 && value == false);
150 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("x-y-z", &value) > 0 && value == false);
151 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("x-y_z", &value) > 0 && value == false);
152 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("x_y_z", &value) > 0 && value == false);
153 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("quux", &value) == -EINVAL && value == false);
154 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("da", &value) > 0 && value == true);
155 [ + - - + ]: 4 : assert_se(proc_cmdline_get_bool("the", &value) > 0 && value == true);
156 : 4 : }
157 : :
158 : 4 : static void test_proc_cmdline_get_key_many(void) {
159 : 4 : _cleanup_free_ char *value1 = NULL, *value2 = NULL, *value3 = NULL, *value4 = NULL, *value5 = NULL, *value6 = NULL, *value7 = NULL;
160 : :
161 [ + - ]: 4 : log_info("/* %s */", __func__);
162 [ - + ]: 4 : assert_se(putenv((char*) "SYSTEMD_PROC_CMDLINE=foo_bar=quux wuff-piep=tuet zumm SPACE='one two' doubleticks=\" aaa aaa \"\n\nzummm='\n'\n") == 0);
163 : :
164 [ - + ]: 4 : assert_se(proc_cmdline_get_key_many(0,
165 : : "wuff-piep", &value3,
166 : : "foo_bar", &value1,
167 : : "idontexist", &value2,
168 : : "zumm", &value4,
169 : : "SPACE", &value5,
170 : : "doubleticks", &value6,
171 : : "zummm", &value7) == 5);
172 : :
173 [ - + ]: 4 : assert_se(streq_ptr(value1, "quux"));
174 [ - + ]: 4 : assert_se(!value2);
175 [ - + ]: 4 : assert_se(streq_ptr(value3, "tuet"));
176 [ - + ]: 4 : assert_se(!value4);
177 [ - + ]: 4 : assert_se(streq_ptr(value5, "one two"));
178 [ - + ]: 4 : assert_se(streq_ptr(value6, " aaa aaa "));
179 [ - + ]: 4 : assert_se(streq_ptr(value7, "\n"));
180 : 4 : }
181 : :
182 : 4 : static void test_proc_cmdline_key_streq(void) {
183 [ + - ]: 4 : log_info("/* %s */", __func__);
184 : :
185 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("", ""));
186 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("a", "a"));
187 [ - + ]: 4 : assert_se(!proc_cmdline_key_streq("", "a"));
188 [ - + ]: 4 : assert_se(!proc_cmdline_key_streq("a", ""));
189 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("a", "a"));
190 [ - + ]: 4 : assert_se(!proc_cmdline_key_streq("a", "b"));
191 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("x-y-z", "x-y-z"));
192 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("x-y-z", "x_y_z"));
193 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("x-y-z", "x-y_z"));
194 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("x-y-z", "x_y-z"));
195 [ - + ]: 4 : assert_se(proc_cmdline_key_streq("x_y-z", "x-y_z"));
196 [ - + ]: 4 : assert_se(!proc_cmdline_key_streq("x_y-z", "x-z_z"));
197 : 4 : }
198 : :
199 : 4 : static void test_proc_cmdline_key_startswith(void) {
200 [ + - ]: 4 : log_info("/* %s */", __func__);
201 : :
202 [ - + ]: 4 : assert_se(proc_cmdline_key_startswith("", ""));
203 [ - + ]: 4 : assert_se(proc_cmdline_key_startswith("x", ""));
204 [ - + ]: 4 : assert_se(!proc_cmdline_key_startswith("", "x"));
205 [ - + ]: 4 : assert_se(proc_cmdline_key_startswith("x", "x"));
206 [ - + ]: 4 : assert_se(!proc_cmdline_key_startswith("x", "y"));
207 [ - + ]: 4 : assert_se(!proc_cmdline_key_startswith("foo-bar", "quux"));
208 [ - + ]: 4 : assert_se(proc_cmdline_key_startswith("foo-bar", "foo"));
209 [ - + ]: 4 : assert_se(proc_cmdline_key_startswith("foo-bar", "foo-bar"));
210 [ - + ]: 4 : assert_se(proc_cmdline_key_startswith("foo-bar", "foo_bar"));
211 [ - + ]: 4 : assert_se(proc_cmdline_key_startswith("foo-bar", "foo_"));
212 [ - + ]: 4 : assert_se(!proc_cmdline_key_startswith("foo-bar", "foo_xx"));
213 : 4 : }
214 : :
215 : 4 : static void test_runlevel_to_target(void) {
216 [ + - ]: 4 : log_info("/* %s */", __func__);
217 : :
218 : 4 : in_initrd_force(false);
219 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target(NULL), NULL));
220 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("unknown-runlevel"), NULL));
221 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("rd.unknown-runlevel"), NULL));
222 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("3"), SPECIAL_MULTI_USER_TARGET));
223 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("rd.rescue"), NULL));
224 : :
225 : 4 : in_initrd_force(true);
226 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target(NULL), NULL));
227 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("unknown-runlevel"), NULL));
228 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("rd.unknown-runlevel"), NULL));
229 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("3"), NULL));
230 [ - + ]: 4 : assert_se(streq_ptr(runlevel_to_target("rd.rescue"), SPECIAL_RESCUE_TARGET));
231 : 4 : }
232 : :
233 : 4 : int main(void) {
234 : 4 : log_parse_environment();
235 : 4 : log_open();
236 : :
237 : 4 : test_proc_cmdline_parse();
238 : 4 : test_proc_cmdline_override();
239 : 4 : test_proc_cmdline_given(false);
240 : : /* Repeat the same thing, but now flip our ininitrdness */
241 : 4 : test_proc_cmdline_given(true);
242 : 4 : test_proc_cmdline_key_streq();
243 : 4 : test_proc_cmdline_key_startswith();
244 : 4 : test_proc_cmdline_get_key();
245 : 4 : test_proc_cmdline_get_bool();
246 : 4 : test_proc_cmdline_get_key_many();
247 : 4 : test_runlevel_to_target();
248 : :
249 : 4 : return 0;
250 : : }
|
