1 /* SPDX-License-Identifier: LGPL-2.1+ */
10 #include "string-util.h"
14 static void test_strv_env_delete(void) {
15 _cleanup_strv_free_
char **a
= NULL
, **b
= NULL
, **c
= NULL
, **d
= NULL
;
17 a
= strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF");
20 b
= strv_new("PIEP", "FOO");
23 c
= strv_new("SCHLUMPF");
26 d
= strv_env_delete(a
, 2, b
, c
);
29 assert_se(streq(d
[0], "WALDO=WALDO"));
30 assert_se(streq(d
[1], "WALDO="));
31 assert_se(strv_length(d
) == 2);
34 static void test_strv_env_get(void) {
37 l
= STRV_MAKE("ONE_OR_TWO=1", "THREE=3", "ONE_OR_TWO=2", "FOUR=4");
39 assert_se(streq(strv_env_get(l
, "ONE_OR_TWO"), "2"));
40 assert_se(streq(strv_env_get(l
, "THREE"), "3"));
41 assert_se(streq(strv_env_get(l
, "FOUR"), "4"));
44 static void test_strv_env_unset(void) {
45 _cleanup_strv_free_
char **l
= NULL
;
47 l
= strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES");
50 assert_se(strv_env_unset(l
, "SCHLUMPF") == l
);
52 assert_se(streq(l
[0], "PIEP"));
53 assert_se(streq(l
[1], "NANANANA=YES"));
54 assert_se(strv_length(l
) == 2);
57 static void test_strv_env_set(void) {
58 _cleanup_strv_free_
char **l
= NULL
, **r
= NULL
;
60 l
= strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES");
63 r
= strv_env_set(l
, "WALDO=WALDO");
66 assert_se(streq(r
[0], "PIEP"));
67 assert_se(streq(r
[1], "SCHLUMPF=SMURFF"));
68 assert_se(streq(r
[2], "NANANANA=YES"));
69 assert_se(streq(r
[3], "WALDO=WALDO"));
70 assert_se(strv_length(r
) == 4);
73 static void test_strv_env_merge(void) {
74 _cleanup_strv_free_
char **a
= NULL
, **b
= NULL
, **r
= NULL
;
76 a
= strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF");
79 b
= strv_new("FOO=KKK", "FOO=", "PIEP=", "SCHLUMPF=SMURFF", "NANANANA=YES");
82 r
= strv_env_merge(2, a
, b
);
84 assert_se(streq(r
[0], "FOO="));
85 assert_se(streq(r
[1], "WALDO="));
86 assert_se(streq(r
[2], "PIEP"));
87 assert_se(streq(r
[3], "SCHLUMPF=SMURFF"));
88 assert_se(streq(r
[4], "PIEP="));
89 assert_se(streq(r
[5], "NANANANA=YES"));
90 assert_se(strv_length(r
) == 6);
92 assert_se(strv_env_clean(r
) == r
);
93 assert_se(streq(r
[0], "FOO="));
94 assert_se(streq(r
[1], "WALDO="));
95 assert_se(streq(r
[2], "SCHLUMPF=SMURFF"));
96 assert_se(streq(r
[3], "PIEP="));
97 assert_se(streq(r
[4], "NANANANA=YES"));
98 assert_se(strv_length(r
) == 5);
101 static void test_env_strv_get_n(void) {
102 const char *_env
[] = {
109 char **env
= (char**) _env
;
111 assert_se(streq(strv_env_get_n(env
, "FOO__", 3, 0), "BAR BAR"));
112 assert_se(streq(strv_env_get_n(env
, "FOO__", 3, REPLACE_ENV_USE_ENVIRONMENT
), "BAR BAR"));
113 assert_se(streq(strv_env_get_n(env
, "FOO", 3, 0), "BAR BAR"));
114 assert_se(streq(strv_env_get_n(env
, "FOO", 3, REPLACE_ENV_USE_ENVIRONMENT
), "BAR BAR"));
116 assert_se(streq(strv_env_get_n(env
, "PATH__", 4, 0), "unset"));
117 assert_se(streq(strv_env_get_n(env
, "PATH", 4, 0), "unset"));
118 assert_se(streq(strv_env_get_n(env
, "PATH__", 4, REPLACE_ENV_USE_ENVIRONMENT
), "unset"));
119 assert_se(streq(strv_env_get_n(env
, "PATH", 4, REPLACE_ENV_USE_ENVIRONMENT
), "unset"));
121 env
[3] = NULL
; /* kill our $PATH */
123 assert_se(!strv_env_get_n(env
, "PATH__", 4, 0));
124 assert_se(!strv_env_get_n(env
, "PATH", 4, 0));
125 assert_se(streq_ptr(strv_env_get_n(env
, "PATH__", 4, REPLACE_ENV_USE_ENVIRONMENT
),
127 assert_se(streq_ptr(strv_env_get_n(env
, "PATH", 4, REPLACE_ENV_USE_ENVIRONMENT
),
131 static void test_replace_env(bool braceless
) {
132 const char *env
[] = {
137 _cleanup_free_
char *t
= NULL
, *s
= NULL
, *q
= NULL
, *r
= NULL
, *p
= NULL
;
138 unsigned flags
= REPLACE_ENV_ALLOW_BRACELESS
*braceless
;
140 t
= replace_env("FOO=$FOO=${FOO}", (char**) env
, flags
);
141 assert_se(streq(t
, braceless
? "FOO=BAR BAR=BAR BAR" : "FOO=$FOO=BAR BAR"));
143 s
= replace_env("BAR=$BAR=${BAR}", (char**) env
, flags
);
144 assert_se(streq(s
, braceless
? "BAR=waldo=waldo" : "BAR=$BAR=waldo"));
146 q
= replace_env("BARBAR=$BARBAR=${BARBAR}", (char**) env
, flags
);
147 assert_se(streq(q
, braceless
? "BARBAR==" : "BARBAR=$BARBAR="));
149 r
= replace_env("BAR=$BAR$BAR${BAR}${BAR}", (char**) env
, flags
);
150 assert_se(streq(r
, braceless
? "BAR=waldowaldowaldowaldo" : "BAR=$BAR$BARwaldowaldo"));
152 p
= replace_env("${BAR}$BAR$BAR", (char**) env
, flags
);
153 assert_se(streq(p
, braceless
? "waldowaldowaldo" : "waldo$BAR$BAR"));
156 static void test_replace_env2(bool extended
) {
157 const char *env
[] = {
162 _cleanup_free_
char *t
= NULL
, *s
= NULL
, *q
= NULL
, *r
= NULL
, *p
= NULL
, *x
= NULL
;
163 unsigned flags
= REPLACE_ENV_ALLOW_EXTENDED
*extended
;
165 t
= replace_env("FOO=${FOO:-${BAR}}", (char**) env
, flags
);
166 assert_se(streq(t
, extended
? "FOO=foo" : "FOO=${FOO:-bar}"));
168 s
= replace_env("BAR=${XXX:-${BAR}}", (char**) env
, flags
);
169 assert_se(streq(s
, extended
? "BAR=bar" : "BAR=${XXX:-bar}"));
171 q
= replace_env("XXX=${XXX:+${BAR}}", (char**) env
, flags
);
172 assert_se(streq(q
, extended
? "XXX=" : "XXX=${XXX:+bar}"));
174 r
= replace_env("FOO=${FOO:+${BAR}}", (char**) env
, flags
);
175 assert_se(streq(r
, extended
? "FOO=bar" : "FOO=${FOO:+bar}"));
177 p
= replace_env("FOO=${FOO:-${BAR}post}", (char**) env
, flags
);
178 assert_se(streq(p
, extended
? "FOO=foo" : "FOO=${FOO:-barpost}"));
180 x
= replace_env("XXX=${XXX:+${BAR}post}", (char**) env
, flags
);
181 assert_se(streq(x
, extended
? "XXX=" : "XXX=${XXX:+barpost}"));
184 static void test_replace_env_argv(void) {
185 const char *env
[] = {
190 const char *line
[] = {
210 _cleanup_strv_free_
char **r
= NULL
;
212 r
= replace_env_argv((char**) line
, (char**) env
);
214 assert_se(streq(r
[0], "FOO$FOO"));
215 assert_se(streq(r
[1], "FOO$FOOFOO"));
216 assert_se(streq(r
[2], "FOOBAR BAR$FOO"));
217 assert_se(streq(r
[3], "FOOBAR BAR"));
218 assert_se(streq(r
[4], "BAR BAR"));
219 assert_se(streq(r
[5], "BAR"));
220 assert_se(streq(r
[6], "BAR"));
221 assert_se(streq(r
[7], "BAR BARwaldo"));
222 assert_se(streq(r
[8], "${FOO"));
223 assert_se(streq(r
[9], "FOO$BAR BAR"));
224 assert_se(streq(r
[10], "$FOOBAR BAR"));
225 assert_se(streq(r
[11], "${FOO:-waldo}"));
226 assert_se(streq(r
[12], "${QUUX:-BAR BAR}"));
227 assert_se(streq(r
[13], "${FOO:+waldo}"));
228 assert_se(streq(r
[14], "${QUUX:+waldo}"));
229 assert_se(streq(r
[15], "${FOO:+|waldo|}}"));
230 assert_se(streq(r
[16], "${FOO:+|waldo{|}"));
231 assert_se(strv_length(r
) == 17);
234 static void test_env_clean(void) {
235 _cleanup_strv_free_
char **e
;
237 e
= strv_new("FOOBAR=WALDO",
252 "another=final one");
254 assert_se(!strv_env_is_valid(e
));
255 assert_se(strv_env_clean(e
) == e
);
256 assert_se(strv_env_is_valid(e
));
258 assert_se(streq(e
[0], "FOOBAR=WALDO"));
259 assert_se(streq(e
[1], "X="));
260 assert_se(streq(e
[2], "F=F"));
261 assert_se(streq(e
[3], "abcd=äöüß"));
262 assert_se(streq(e
[4], "xyz=xyz\n"));
263 assert_se(streq(e
[5], "another=final one"));
264 assert_se(e
[6] == NULL
);
267 static void test_env_name_is_valid(void) {
268 assert_se(env_name_is_valid("test"));
270 assert_se(!env_name_is_valid(NULL
));
271 assert_se(!env_name_is_valid(""));
272 assert_se(!env_name_is_valid("xxx\a"));
273 assert_se(!env_name_is_valid("xxx\007b"));
274 assert_se(!env_name_is_valid("\007\009"));
275 assert_se(!env_name_is_valid("5_starting_with_a_number_is_wrong"));
276 assert_se(!env_name_is_valid("#¤%&?_only_numbers_letters_and_underscore_allowed"));
279 static void test_env_value_is_valid(void) {
280 assert_se(env_value_is_valid(""));
281 assert_se(env_value_is_valid("głąb kapuściany"));
282 assert_se(env_value_is_valid("printf \"\\x1b]0;<mock-chroot>\\x07<mock-chroot>\""));
283 assert_se(env_value_is_valid("tab\tcharacter"));
284 assert_se(env_value_is_valid("new\nline"));
287 static void test_env_assignment_is_valid(void) {
288 assert_se(env_assignment_is_valid("a="));
289 assert_se(env_assignment_is_valid("b=głąb kapuściany"));
290 assert_se(env_assignment_is_valid("c=\\007\\009\\011"));
291 assert_se(env_assignment_is_valid("e=printf \"\\x1b]0;<mock-chroot>\\x07<mock-chroot>\""));
292 assert_se(env_assignment_is_valid("f=tab\tcharacter"));
293 assert_se(env_assignment_is_valid("g=new\nline"));
295 assert_se(!env_assignment_is_valid("="));
296 assert_se(!env_assignment_is_valid("a b="));
297 assert_se(!env_assignment_is_valid("a ="));
298 assert_se(!env_assignment_is_valid(" b="));
299 /* no dots or dashes: http://tldp.org/LDP/abs/html/gotchas.html */
300 assert_se(!env_assignment_is_valid("a.b="));
301 assert_se(!env_assignment_is_valid("a-b="));
302 assert_se(!env_assignment_is_valid("\007=głąb kapuściany"));
303 assert_se(!env_assignment_is_valid("c\009=\007\009\011"));
304 assert_se(!env_assignment_is_valid("głąb=printf \"\x1b]0;<mock-chroot>\x07<mock-chroot>\""));
307 int main(int argc
, char *argv
[]) {
308 test_strv_env_delete();
310 test_strv_env_unset();
312 test_strv_env_merge();
313 test_env_strv_get_n();
314 test_replace_env(false);
315 test_replace_env(true);
316 test_replace_env2(false);
317 test_replace_env2(true);
318 test_replace_env_argv();
320 test_env_name_is_valid();
321 test_env_value_is_valid();
322 test_env_assignment_is_valid();