1 /* SPDX-License-Identifier: LGPL-2.1+ */
3 This file is part of systemd.
5 Copyright 2010 Lennart Poettering
6 Copyright 2013 Thomas H.P. Andersen
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
25 #include "extract-word.h"
27 #include "string-util.h"
29 static void test_extract_first_word(void) {
30 const char *p
, *original
;
33 p
= original
= "foobar waldo";
34 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
35 assert_se(streq(t
, "foobar"));
37 assert_se(p
== original
+ 7);
39 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
40 assert_se(streq(t
, "waldo"));
42 assert_se(isempty(p
));
44 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 0);
46 assert_se(isempty(p
));
48 p
= original
= "\"foobar\" \'waldo\'";
49 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
50 assert_se(streq(t
, "\"foobar\""));
52 assert_se(p
== original
+ 9);
54 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
55 assert_se(streq(t
, "\'waldo\'"));
57 assert_se(isempty(p
));
59 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 0);
61 assert_se(isempty(p
));
63 p
= original
= "\"foobar\" \'waldo\'";
64 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
65 assert_se(streq(t
, "foobar"));
67 assert_se(p
== original
+ 9);
69 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
70 assert_se(streq(t
, "waldo"));
72 assert_se(isempty(p
));
74 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 0);
76 assert_se(isempty(p
));
79 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 1);
80 assert_se(streq(t
, "\""));
82 assert_se(isempty(p
));
85 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) == -EINVAL
);
86 assert_se(p
== original
+ 1);
89 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 1);
90 assert_se(streq(t
, "\'"));
92 assert_se(isempty(p
));
95 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) == -EINVAL
);
96 assert_se(p
== original
+ 1);
98 p
= original
= "\'fooo";
99 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 1);
100 assert_se(streq(t
, "\'fooo"));
102 assert_se(isempty(p
));
104 p
= original
= "\'fooo";
105 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) == -EINVAL
);
106 assert_se(p
== original
+ 5);
108 p
= original
= "\'fooo";
109 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
) > 0);
110 assert_se(streq(t
, "fooo"));
112 assert_se(isempty(p
));
114 p
= original
= "\"fooo";
115 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
) > 0);
116 assert_se(streq(t
, "fooo"));
118 assert_se(isempty(p
));
120 p
= original
= "yay\'foo\'bar";
121 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
122 assert_se(streq(t
, "yay\'foo\'bar"));
124 assert_se(isempty(p
));
126 p
= original
= "yay\'foo\'bar";
127 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
128 assert_se(streq(t
, "yayfoobar"));
130 assert_se(isempty(p
));
132 p
= original
= " foobar ";
133 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
134 assert_se(streq(t
, "foobar"));
136 assert_se(isempty(p
));
138 p
= original
= " foo\\ba\\x6ar ";
139 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) > 0);
140 assert_se(streq(t
, "foo\ba\x6ar"));
142 assert_se(isempty(p
));
144 p
= original
= " foo\\ba\\x6ar ";
145 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
146 assert_se(streq(t
, "foobax6ar"));
148 assert_se(isempty(p
));
150 p
= original
= " f\\u00f6o \"pi\\U0001F4A9le\" ";
151 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) > 0);
152 assert_se(streq(t
, "föo"));
154 assert_se(p
== original
+ 13);
156 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
) > 0);
157 assert_se(streq(t
, "pi\360\237\222\251le"));
159 assert_se(isempty(p
));
161 p
= original
= "fooo\\";
162 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_RELAX
) > 0);
163 assert_se(streq(t
, "fooo"));
165 assert_se(isempty(p
));
167 p
= original
= "fooo\\";
168 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
) > 0);
169 assert_se(streq(t
, "fooo\\"));
171 assert_se(isempty(p
));
173 p
= original
= "fooo\\";
174 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
175 assert_se(streq(t
, "fooo\\"));
177 assert_se(isempty(p
));
179 p
= original
= "fooo\\";
180 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
181 assert_se(streq(t
, "fooo\\"));
183 assert_se(isempty(p
));
185 p
= original
= "\"foo\\";
186 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == -EINVAL
);
187 assert_se(p
== original
+ 5);
189 p
= original
= "\"foo\\";
190 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
) > 0);
191 assert_se(streq(t
, "foo"));
193 assert_se(isempty(p
));
195 p
= original
= "foo::bar";
196 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
197 assert_se(streq(t
, "foo"));
199 assert_se(p
== original
+ 5);
201 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
202 assert_se(streq(t
, "bar"));
204 assert_se(isempty(p
));
206 assert_se(extract_first_word(&p
, &t
, ":", 0) == 0);
208 assert_se(isempty(p
));
210 p
= original
= "foo\\:bar::waldo";
211 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
212 assert_se(streq(t
, "foo:bar"));
214 assert_se(p
== original
+ 10);
216 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
217 assert_se(streq(t
, "waldo"));
219 assert_se(isempty(p
));
221 assert_se(extract_first_word(&p
, &t
, ":", 0) == 0);
223 assert_se(isempty(p
));
225 p
= original
= "\"foo\\";
226 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE_RELAX
) == -EINVAL
);
227 assert_se(p
== original
+ 5);
229 p
= original
= "\"foo\\";
230 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
231 assert_se(streq(t
, "foo\\"));
233 assert_se(isempty(p
));
235 p
= original
= "\"foo\\";
236 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
237 assert_se(streq(t
, "foo\\"));
239 assert_se(isempty(p
));
241 p
= original
= "fooo\\ bar quux";
242 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_RELAX
) > 0);
243 assert_se(streq(t
, "fooo bar"));
245 assert_se(p
== original
+ 10);
247 p
= original
= "fooo\\ bar quux";
248 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
) > 0);
249 assert_se(streq(t
, "fooo bar"));
251 assert_se(p
== original
+ 10);
253 p
= original
= "fooo\\ bar quux";
254 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
255 assert_se(streq(t
, "fooo bar"));
257 assert_se(p
== original
+ 10);
259 p
= original
= "fooo\\ bar quux";
260 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) == -EINVAL
);
261 assert_se(p
== original
+ 5);
263 p
= original
= "fooo\\ bar quux";
264 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
265 assert_se(streq(t
, "fooo\\ bar"));
267 assert_se(p
== original
+ 10);
269 p
= original
= "\\w+@\\K[\\d.]+";
270 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) == -EINVAL
);
271 assert_se(p
== original
+ 1);
273 p
= original
= "\\w+@\\K[\\d.]+";
274 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
275 assert_se(streq(t
, "\\w+@\\K[\\d.]+"));
277 assert_se(isempty(p
));
279 p
= original
= "\\w+\\b";
280 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
281 assert_se(streq(t
, "\\w+\b"));
283 assert_se(isempty(p
));
285 p
= original
= "-N ''";
286 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
287 assert_se(streq(t
, "-N"));
289 assert_se(p
== original
+ 3);
291 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
292 assert_se(streq(t
, ""));
294 assert_se(isempty(p
));
296 p
= original
= ":foo\\:bar::waldo:";
297 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
299 assert_se(streq(t
, ""));
301 assert_se(p
== original
+ 1);
303 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
304 assert_se(streq(t
, "foo:bar"));
306 assert_se(p
== original
+ 10);
308 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
310 assert_se(streq(t
, ""));
312 assert_se(p
== original
+ 11);
314 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
315 assert_se(streq(t
, "waldo"));
317 assert_se(p
== original
+ 17);
319 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
320 assert_se(streq(t
, ""));
322 assert_se(p
== NULL
);
324 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 0);
329 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
330 assert_se(streq(t
, "fooxbar"));
332 assert_se(p
== NULL
);
335 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_RETAIN_ESCAPE
) > 0);
336 assert_se(streq(t
, "foo\\xbar"));
338 assert_se(p
== NULL
);
341 static void test_extract_first_word_and_warn(void) {
342 const char *p
, *original
;
345 p
= original
= "foobar waldo";
346 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
347 assert_se(streq(t
, "foobar"));
349 assert_se(p
== original
+ 7);
351 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
352 assert_se(streq(t
, "waldo"));
354 assert_se(isempty(p
));
356 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) == 0);
358 assert_se(isempty(p
));
360 p
= original
= "\"foobar\" \'waldo\'";
361 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) > 0);
362 assert_se(streq(t
, "foobar"));
364 assert_se(p
== original
+ 9);
366 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) > 0);
367 assert_se(streq(t
, "waldo"));
369 assert_se(isempty(p
));
371 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) == 0);
373 assert_se(isempty(p
));
376 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
377 assert_se(p
== original
+ 1);
380 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
381 assert_se(p
== original
+ 1);
383 p
= original
= "\'fooo";
384 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
385 assert_se(p
== original
+ 5);
387 p
= original
= "\'fooo";
388 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
389 assert_se(streq(t
, "fooo"));
391 assert_se(isempty(p
));
393 p
= original
= " foo\\ba\\x6ar ";
394 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
395 assert_se(streq(t
, "foo\ba\x6ar"));
397 assert_se(isempty(p
));
399 p
= original
= " foo\\ba\\x6ar ";
400 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
401 assert_se(streq(t
, "foobax6ar"));
403 assert_se(isempty(p
));
405 p
= original
= " f\\u00f6o \"pi\\U0001F4A9le\" ";
406 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
407 assert_se(streq(t
, "föo"));
409 assert_se(p
== original
+ 13);
411 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
412 assert_se(streq(t
, "pi\360\237\222\251le"));
414 assert_se(isempty(p
));
416 p
= original
= "fooo\\";
417 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
418 assert_se(streq(t
, "fooo"));
420 assert_se(isempty(p
));
422 p
= original
= "fooo\\";
423 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
424 assert_se(streq(t
, "fooo\\"));
426 assert_se(isempty(p
));
428 p
= original
= "fooo\\";
429 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
430 assert_se(streq(t
, "fooo\\"));
432 assert_se(isempty(p
));
434 p
= original
= "\"foo\\";
435 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
436 assert_se(p
== original
+ 5);
438 p
= original
= "\"foo\\";
439 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
440 assert_se(streq(t
, "foo"));
442 assert_se(isempty(p
));
444 p
= original
= "\"foo\\";
445 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) == -EINVAL
);
446 assert_se(p
== original
+ 5);
448 p
= original
= "\"foo\\";
449 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
|EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
450 assert_se(streq(t
, "foo"));
452 assert_se(isempty(p
));
454 p
= original
= "fooo\\ bar quux";
455 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
456 assert_se(streq(t
, "fooo bar"));
458 assert_se(p
== original
+ 10);
460 p
= original
= "fooo\\ bar quux";
461 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
462 assert_se(streq(t
, "fooo bar"));
464 assert_se(p
== original
+ 10);
466 p
= original
= "fooo\\ bar quux";
467 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
468 assert_se(streq(t
, "fooo\\ bar"));
470 assert_se(p
== original
+ 10);
472 p
= original
= "\\w+@\\K[\\d.]+";
473 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
474 assert_se(streq(t
, "\\w+@\\K[\\d.]+"));
476 assert_se(isempty(p
));
478 p
= original
= "\\w+\\b";
479 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
480 assert_se(streq(t
, "\\w+\b"));
482 assert_se(isempty(p
));
485 static void test_extract_many_words(void) {
486 const char *p
, *original
;
489 p
= original
= "foobar waldi piep";
490 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 3);
491 assert_se(isempty(p
));
492 assert_se(streq_ptr(a
, "foobar"));
493 assert_se(streq_ptr(b
, "waldi"));
494 assert_se(streq_ptr(c
, "piep"));
499 p
= original
= "'foobar' wa\"ld\"i ";
500 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 2);
501 assert_se(isempty(p
));
502 assert_se(streq_ptr(a
, "'foobar'"));
503 assert_se(streq_ptr(b
, "wa\"ld\"i"));
504 assert_se(streq_ptr(c
, NULL
));
508 p
= original
= "'foobar' wa\"ld\"i ";
509 assert_se(extract_many_words(&p
, NULL
, EXTRACT_QUOTES
, &a
, &b
, &c
, NULL
) == 2);
510 assert_se(isempty(p
));
511 assert_se(streq_ptr(a
, "foobar"));
512 assert_se(streq_ptr(b
, "waldi"));
513 assert_se(streq_ptr(c
, NULL
));
518 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 0);
519 assert_se(isempty(p
));
520 assert_se(streq_ptr(a
, NULL
));
521 assert_se(streq_ptr(b
, NULL
));
522 assert_se(streq_ptr(c
, NULL
));
525 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 0);
526 assert_se(isempty(p
));
527 assert_se(streq_ptr(a
, NULL
));
528 assert_se(streq_ptr(b
, NULL
));
529 assert_se(streq_ptr(c
, NULL
));
531 p
= original
= "foobar";
532 assert_se(extract_many_words(&p
, NULL
, 0, NULL
) == 0);
533 assert_se(p
== original
);
535 p
= original
= "foobar waldi";
536 assert_se(extract_many_words(&p
, NULL
, 0, &a
, NULL
) == 1);
537 assert_se(p
== original
+7);
538 assert_se(streq_ptr(a
, "foobar"));
541 p
= original
= " foobar ";
542 assert_se(extract_many_words(&p
, NULL
, 0, &a
, NULL
) == 1);
543 assert_se(isempty(p
));
544 assert_se(streq_ptr(a
, "foobar"));
548 int main(int argc
, char *argv
[]) {
549 log_parse_environment();
552 test_extract_first_word();
553 test_extract_first_word_and_warn();
554 test_extract_many_words();