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/>.
26 #include "extract-word.h"
28 #include "string-util.h"
30 static void test_extract_first_word(void) {
31 const char *p
, *original
;
34 p
= original
= "foobar waldo";
35 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
36 assert_se(streq(t
, "foobar"));
38 assert_se(p
== original
+ 7);
40 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
41 assert_se(streq(t
, "waldo"));
43 assert_se(isempty(p
));
45 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 0);
47 assert_se(isempty(p
));
49 p
= original
= "\"foobar\" \'waldo\'";
50 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
51 assert_se(streq(t
, "\"foobar\""));
53 assert_se(p
== original
+ 9);
55 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
56 assert_se(streq(t
, "\'waldo\'"));
58 assert_se(isempty(p
));
60 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 0);
62 assert_se(isempty(p
));
64 p
= original
= "\"foobar\" \'waldo\'";
65 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
66 assert_se(streq(t
, "foobar"));
68 assert_se(p
== original
+ 9);
70 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
71 assert_se(streq(t
, "waldo"));
73 assert_se(isempty(p
));
75 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 0);
77 assert_se(isempty(p
));
80 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 1);
81 assert_se(streq(t
, "\""));
83 assert_se(isempty(p
));
86 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) == -EINVAL
);
87 assert_se(p
== original
+ 1);
90 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 1);
91 assert_se(streq(t
, "\'"));
93 assert_se(isempty(p
));
96 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) == -EINVAL
);
97 assert_se(p
== original
+ 1);
99 p
= original
= "\'fooo";
100 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == 1);
101 assert_se(streq(t
, "\'fooo"));
103 assert_se(isempty(p
));
105 p
= original
= "\'fooo";
106 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) == -EINVAL
);
107 assert_se(p
== original
+ 5);
109 p
= original
= "\'fooo";
110 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
) > 0);
111 assert_se(streq(t
, "fooo"));
113 assert_se(isempty(p
));
115 p
= original
= "\"fooo";
116 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
) > 0);
117 assert_se(streq(t
, "fooo"));
119 assert_se(isempty(p
));
121 p
= original
= "yay\'foo\'bar";
122 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
123 assert_se(streq(t
, "yay\'foo\'bar"));
125 assert_se(isempty(p
));
127 p
= original
= "yay\'foo\'bar";
128 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
129 assert_se(streq(t
, "yayfoobar"));
131 assert_se(isempty(p
));
133 p
= original
= " foobar ";
134 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
135 assert_se(streq(t
, "foobar"));
137 assert_se(isempty(p
));
139 p
= original
= " foo\\ba\\x6ar ";
140 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) > 0);
141 assert_se(streq(t
, "foo\ba\x6ar"));
143 assert_se(isempty(p
));
145 p
= original
= " foo\\ba\\x6ar ";
146 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
147 assert_se(streq(t
, "foobax6ar"));
149 assert_se(isempty(p
));
151 p
= original
= " f\\u00f6o \"pi\\U0001F4A9le\" ";
152 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) > 0);
153 assert_se(streq(t
, "föo"));
155 assert_se(p
== original
+ 13);
157 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
) > 0);
158 assert_se(streq(t
, "pi\360\237\222\251le"));
160 assert_se(isempty(p
));
162 p
= original
= "fooo\\";
163 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_RELAX
) > 0);
164 assert_se(streq(t
, "fooo"));
166 assert_se(isempty(p
));
168 p
= original
= "fooo\\";
169 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
) > 0);
170 assert_se(streq(t
, "fooo\\"));
172 assert_se(isempty(p
));
174 p
= original
= "fooo\\";
175 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
176 assert_se(streq(t
, "fooo\\"));
178 assert_se(isempty(p
));
180 p
= original
= "fooo\\";
181 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
182 assert_se(streq(t
, "fooo\\"));
184 assert_se(isempty(p
));
186 p
= original
= "\"foo\\";
187 assert_se(extract_first_word(&p
, &t
, NULL
, 0) == -EINVAL
);
188 assert_se(p
== original
+ 5);
190 p
= original
= "\"foo\\";
191 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
) > 0);
192 assert_se(streq(t
, "foo"));
194 assert_se(isempty(p
));
196 p
= original
= "foo::bar";
197 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
198 assert_se(streq(t
, "foo"));
200 assert_se(p
== original
+ 5);
202 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
203 assert_se(streq(t
, "bar"));
205 assert_se(isempty(p
));
207 assert_se(extract_first_word(&p
, &t
, ":", 0) == 0);
209 assert_se(isempty(p
));
211 p
= original
= "foo\\:bar::waldo";
212 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
213 assert_se(streq(t
, "foo:bar"));
215 assert_se(p
== original
+ 10);
217 assert_se(extract_first_word(&p
, &t
, ":", 0) == 1);
218 assert_se(streq(t
, "waldo"));
220 assert_se(isempty(p
));
222 assert_se(extract_first_word(&p
, &t
, ":", 0) == 0);
224 assert_se(isempty(p
));
226 p
= original
= "\"foo\\";
227 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE_RELAX
) == -EINVAL
);
228 assert_se(p
== original
+ 5);
230 p
= original
= "\"foo\\";
231 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
232 assert_se(streq(t
, "foo\\"));
234 assert_se(isempty(p
));
236 p
= original
= "\"foo\\";
237 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
238 assert_se(streq(t
, "foo\\"));
240 assert_se(isempty(p
));
242 p
= original
= "fooo\\ bar quux";
243 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_RELAX
) > 0);
244 assert_se(streq(t
, "fooo bar"));
246 assert_se(p
== original
+ 10);
248 p
= original
= "fooo\\ bar quux";
249 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
) > 0);
250 assert_se(streq(t
, "fooo bar"));
252 assert_se(p
== original
+ 10);
254 p
= original
= "fooo\\ bar quux";
255 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE_RELAX
|EXTRACT_RELAX
) > 0);
256 assert_se(streq(t
, "fooo bar"));
258 assert_se(p
== original
+ 10);
260 p
= original
= "fooo\\ bar quux";
261 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) == -EINVAL
);
262 assert_se(p
== original
+ 5);
264 p
= original
= "fooo\\ bar quux";
265 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
266 assert_se(streq(t
, "fooo\\ bar"));
268 assert_se(p
== original
+ 10);
270 p
= original
= "\\w+@\\K[\\d.]+";
271 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
) == -EINVAL
);
272 assert_se(p
== original
+ 1);
274 p
= original
= "\\w+@\\K[\\d.]+";
275 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
276 assert_se(streq(t
, "\\w+@\\K[\\d.]+"));
278 assert_se(isempty(p
));
280 p
= original
= "\\w+\\b";
281 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
|EXTRACT_CUNESCAPE_RELAX
) > 0);
282 assert_se(streq(t
, "\\w+\b"));
284 assert_se(isempty(p
));
286 p
= original
= "-N ''";
287 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
288 assert_se(streq(t
, "-N"));
290 assert_se(p
== original
+ 3);
292 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_QUOTES
) > 0);
293 assert_se(streq(t
, ""));
295 assert_se(isempty(p
));
297 p
= original
= ":foo\\:bar::waldo:";
298 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
300 assert_se(streq(t
, ""));
302 assert_se(p
== original
+ 1);
304 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
305 assert_se(streq(t
, "foo:bar"));
307 assert_se(p
== original
+ 10);
309 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
311 assert_se(streq(t
, ""));
313 assert_se(p
== original
+ 11);
315 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
316 assert_se(streq(t
, "waldo"));
318 assert_se(p
== original
+ 17);
320 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 1);
321 assert_se(streq(t
, ""));
323 assert_se(p
== NULL
);
325 assert_se(extract_first_word(&p
, &t
, ":", EXTRACT_DONT_COALESCE_SEPARATORS
) == 0);
330 assert_se(extract_first_word(&p
, &t
, NULL
, 0) > 0);
331 assert_se(streq(t
, "fooxbar"));
333 assert_se(p
== NULL
);
336 assert_se(extract_first_word(&p
, &t
, NULL
, EXTRACT_RETAIN_ESCAPE
) > 0);
337 assert_se(streq(t
, "foo\\xbar"));
339 assert_se(p
== NULL
);
342 static void test_extract_first_word_and_warn(void) {
343 const char *p
, *original
;
346 p
= original
= "foobar waldo";
347 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
348 assert_se(streq(t
, "foobar"));
350 assert_se(p
== original
+ 7);
352 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
353 assert_se(streq(t
, "waldo"));
355 assert_se(isempty(p
));
357 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) == 0);
359 assert_se(isempty(p
));
361 p
= original
= "\"foobar\" \'waldo\'";
362 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) > 0);
363 assert_se(streq(t
, "foobar"));
365 assert_se(p
== original
+ 9);
367 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) > 0);
368 assert_se(streq(t
, "waldo"));
370 assert_se(isempty(p
));
372 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) == 0);
374 assert_se(isempty(p
));
377 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
378 assert_se(p
== original
+ 1);
381 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
382 assert_se(p
== original
+ 1);
384 p
= original
= "\'fooo";
385 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
386 assert_se(p
== original
+ 5);
388 p
= original
= "\'fooo";
389 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
390 assert_se(streq(t
, "fooo"));
392 assert_se(isempty(p
));
394 p
= original
= " foo\\ba\\x6ar ";
395 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
396 assert_se(streq(t
, "foo\ba\x6ar"));
398 assert_se(isempty(p
));
400 p
= original
= " foo\\ba\\x6ar ";
401 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
402 assert_se(streq(t
, "foobax6ar"));
404 assert_se(isempty(p
));
406 p
= original
= " f\\u00f6o \"pi\\U0001F4A9le\" ";
407 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
408 assert_se(streq(t
, "föo"));
410 assert_se(p
== original
+ 13);
412 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
413 assert_se(streq(t
, "pi\360\237\222\251le"));
415 assert_se(isempty(p
));
417 p
= original
= "fooo\\";
418 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
419 assert_se(streq(t
, "fooo"));
421 assert_se(isempty(p
));
423 p
= original
= "fooo\\";
424 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
425 assert_se(streq(t
, "fooo\\"));
427 assert_se(isempty(p
));
429 p
= original
= "fooo\\";
430 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
431 assert_se(streq(t
, "fooo\\"));
433 assert_se(isempty(p
));
435 p
= original
= "\"foo\\";
436 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
, NULL
, "fake", 1, original
) == -EINVAL
);
437 assert_se(p
== original
+ 5);
439 p
= original
= "\"foo\\";
440 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
441 assert_se(streq(t
, "foo"));
443 assert_se(isempty(p
));
445 p
= original
= "\"foo\\";
446 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) == -EINVAL
);
447 assert_se(p
== original
+ 5);
449 p
= original
= "\"foo\\";
450 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_QUOTES
|EXTRACT_CUNESCAPE
|EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
451 assert_se(streq(t
, "foo"));
453 assert_se(isempty(p
));
455 p
= original
= "fooo\\ bar quux";
456 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_RELAX
, NULL
, "fake", 1, original
) > 0);
457 assert_se(streq(t
, "fooo bar"));
459 assert_se(p
== original
+ 10);
461 p
= original
= "fooo\\ bar quux";
462 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, 0, NULL
, "fake", 1, original
) > 0);
463 assert_se(streq(t
, "fooo bar"));
465 assert_se(p
== original
+ 10);
467 p
= original
= "fooo\\ bar quux";
468 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
469 assert_se(streq(t
, "fooo\\ bar"));
471 assert_se(p
== original
+ 10);
473 p
= original
= "\\w+@\\K[\\d.]+";
474 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
475 assert_se(streq(t
, "\\w+@\\K[\\d.]+"));
477 assert_se(isempty(p
));
479 p
= original
= "\\w+\\b";
480 assert_se(extract_first_word_and_warn(&p
, &t
, NULL
, EXTRACT_CUNESCAPE
, NULL
, "fake", 1, original
) > 0);
481 assert_se(streq(t
, "\\w+\b"));
483 assert_se(isempty(p
));
486 static void test_extract_many_words(void) {
487 const char *p
, *original
;
490 p
= original
= "foobar waldi piep";
491 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 3);
492 assert_se(isempty(p
));
493 assert_se(streq_ptr(a
, "foobar"));
494 assert_se(streq_ptr(b
, "waldi"));
495 assert_se(streq_ptr(c
, "piep"));
500 p
= original
= "'foobar' wa\"ld\"i ";
501 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 2);
502 assert_se(isempty(p
));
503 assert_se(streq_ptr(a
, "'foobar'"));
504 assert_se(streq_ptr(b
, "wa\"ld\"i"));
505 assert_se(streq_ptr(c
, NULL
));
509 p
= original
= "'foobar' wa\"ld\"i ";
510 assert_se(extract_many_words(&p
, NULL
, EXTRACT_QUOTES
, &a
, &b
, &c
, NULL
) == 2);
511 assert_se(isempty(p
));
512 assert_se(streq_ptr(a
, "foobar"));
513 assert_se(streq_ptr(b
, "waldi"));
514 assert_se(streq_ptr(c
, NULL
));
519 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 0);
520 assert_se(isempty(p
));
521 assert_se(streq_ptr(a
, NULL
));
522 assert_se(streq_ptr(b
, NULL
));
523 assert_se(streq_ptr(c
, NULL
));
526 assert_se(extract_many_words(&p
, NULL
, 0, &a
, &b
, &c
, NULL
) == 0);
527 assert_se(isempty(p
));
528 assert_se(streq_ptr(a
, NULL
));
529 assert_se(streq_ptr(b
, NULL
));
530 assert_se(streq_ptr(c
, NULL
));
532 p
= original
= "foobar";
533 assert_se(extract_many_words(&p
, NULL
, 0, NULL
) == 0);
534 assert_se(p
== original
);
536 p
= original
= "foobar waldi";
537 assert_se(extract_many_words(&p
, NULL
, 0, &a
, NULL
) == 1);
538 assert_se(p
== original
+7);
539 assert_se(streq_ptr(a
, "foobar"));
542 p
= original
= " foobar ";
543 assert_se(extract_many_words(&p
, NULL
, 0, &a
, NULL
) == 1);
544 assert_se(isempty(p
));
545 assert_se(streq_ptr(a
, "foobar"));
549 int main(int argc
, char *argv
[]) {
550 log_parse_environment();
553 test_extract_first_word();
554 test_extract_first_word_and_warn();
555 test_extract_many_words();