]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/test/test-fileio.c
basic: add new merge_env_file function
[thirdparty/systemd.git] / src / test / test-fileio.c
1 /***
2 This file is part of systemd.
3
4 Copyright 2013 Lennart Poettering
5
6 systemd is free software; you can redistribute it and/or modify it
7 under the terms of the GNU Lesser General Public License as published by
8 the Free Software Foundation; either version 2.1 of the License, or
9 (at your option) any later version.
10
11 systemd is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 Lesser General Public License for more details.
15
16 You should have received a copy of the GNU Lesser General Public License
17 along with systemd; If not, see <http://www.gnu.org/licenses/>.
18 ***/
19
20 #include <fcntl.h>
21 #include <stdio.h>
22 #include <unistd.h>
23
24 #include "alloc-util.h"
25 #include "ctype.h"
26 #include "def.h"
27 #include "env-util.h"
28 #include "fd-util.h"
29 #include "fileio.h"
30 #include "io-util.h"
31 #include "parse-util.h"
32 #include "process-util.h"
33 #include "string-util.h"
34 #include "strv.h"
35 #include "util.h"
36
37 static void test_parse_env_file(void) {
38 char t[] = "/tmp/test-fileio-in-XXXXXX",
39 p[] = "/tmp/test-fileio-out-XXXXXX";
40 int fd, r;
41 FILE *f;
42 _cleanup_free_ char *one = NULL, *two = NULL, *three = NULL, *four = NULL, *five = NULL,
43 *six = NULL, *seven = NULL, *eight = NULL, *nine = NULL, *ten = NULL;
44 _cleanup_strv_free_ char **a = NULL, **b = NULL;
45 char **i;
46 unsigned k;
47
48 fd = mkostemp_safe(p);
49 assert_se(fd >= 0);
50 close(fd);
51
52 fd = mkostemp_safe(t);
53 assert_se(fd >= 0);
54
55 f = fdopen(fd, "w");
56 assert_se(f);
57
58 fputs("one=BAR \n"
59 "# comment\n"
60 " # comment \n"
61 " ; comment \n"
62 " two = bar \n"
63 "invalid line\n"
64 "invalid line #comment\n"
65 "three = \"333\n"
66 "xxxx\"\n"
67 "four = \'44\\\"44\'\n"
68 "five = \'55\\\'55\' \"FIVE\" cinco \n"
69 "six = seis sechs\\\n"
70 " sis\n"
71 "seven=\"sevenval\" #nocomment\n"
72 "eight=eightval #nocomment\n"
73 "export nine=nineval\n"
74 "ten=ignored\n"
75 "ten=ignored\n"
76 "ten=", f);
77
78 fflush(f);
79 fclose(f);
80
81 r = load_env_file(NULL, t, NULL, &a);
82 assert_se(r >= 0);
83
84 STRV_FOREACH(i, a)
85 log_info("Got: <%s>", *i);
86
87 assert_se(streq_ptr(a[0], "one=BAR"));
88 assert_se(streq_ptr(a[1], "two=bar"));
89 assert_se(streq_ptr(a[2], "three=333\nxxxx"));
90 assert_se(streq_ptr(a[3], "four=44\"44"));
91 assert_se(streq_ptr(a[4], "five=55\'55FIVEcinco"));
92 assert_se(streq_ptr(a[5], "six=seis sechs sis"));
93 assert_se(streq_ptr(a[6], "seven=sevenval#nocomment"));
94 assert_se(streq_ptr(a[7], "eight=eightval #nocomment"));
95 assert_se(streq_ptr(a[8], "export nine=nineval"));
96 assert_se(streq_ptr(a[9], "ten="));
97 assert_se(a[10] == NULL);
98
99 strv_env_clean(a);
100
101 k = 0;
102 STRV_FOREACH(i, b) {
103 log_info("Got2: <%s>", *i);
104 assert_se(streq(*i, a[k++]));
105 }
106
107 r = parse_env_file(
108 t, NULL,
109 "one", &one,
110 "two", &two,
111 "three", &three,
112 "four", &four,
113 "five", &five,
114 "six", &six,
115 "seven", &seven,
116 "eight", &eight,
117 "export nine", &nine,
118 "ten", &ten,
119 NULL);
120
121 assert_se(r >= 0);
122
123 log_info("one=[%s]", strna(one));
124 log_info("two=[%s]", strna(two));
125 log_info("three=[%s]", strna(three));
126 log_info("four=[%s]", strna(four));
127 log_info("five=[%s]", strna(five));
128 log_info("six=[%s]", strna(six));
129 log_info("seven=[%s]", strna(seven));
130 log_info("eight=[%s]", strna(eight));
131 log_info("export nine=[%s]", strna(nine));
132 log_info("ten=[%s]", strna(nine));
133
134 assert_se(streq(one, "BAR"));
135 assert_se(streq(two, "bar"));
136 assert_se(streq(three, "333\nxxxx"));
137 assert_se(streq(four, "44\"44"));
138 assert_se(streq(five, "55\'55FIVEcinco"));
139 assert_se(streq(six, "seis sechs sis"));
140 assert_se(streq(seven, "sevenval#nocomment"));
141 assert_se(streq(eight, "eightval #nocomment"));
142 assert_se(streq(nine, "nineval"));
143 assert_se(ten == NULL);
144
145 r = write_env_file(p, a);
146 assert_se(r >= 0);
147
148 r = load_env_file(NULL, p, NULL, &b);
149 assert_se(r >= 0);
150
151 unlink(t);
152 unlink(p);
153 }
154
155 static void test_parse_multiline_env_file(void) {
156 char t[] = "/tmp/test-fileio-in-XXXXXX",
157 p[] = "/tmp/test-fileio-out-XXXXXX";
158 int fd, r;
159 FILE *f;
160 _cleanup_strv_free_ char **a = NULL, **b = NULL;
161 char **i;
162
163 fd = mkostemp_safe(p);
164 assert_se(fd >= 0);
165 close(fd);
166
167 fd = mkostemp_safe(t);
168 assert_se(fd >= 0);
169
170 f = fdopen(fd, "w");
171 assert_se(f);
172
173 fputs("one=BAR\\\n"
174 " VAR\\\n"
175 "\tGAR\n"
176 "#comment\n"
177 "two=\"bar\\\n"
178 " var\\\n"
179 "\tgar\"\n"
180 "#comment\n"
181 "tri=\"bar \\\n"
182 " var \\\n"
183 "\tgar \"\n", f);
184
185 fflush(f);
186 fclose(f);
187
188 r = load_env_file(NULL, t, NULL, &a);
189 assert_se(r >= 0);
190
191 STRV_FOREACH(i, a)
192 log_info("Got: <%s>", *i);
193
194 assert_se(streq_ptr(a[0], "one=BAR VAR\tGAR"));
195 assert_se(streq_ptr(a[1], "two=bar var\tgar"));
196 assert_se(streq_ptr(a[2], "tri=bar var \tgar "));
197 assert_se(a[3] == NULL);
198
199 r = write_env_file(p, a);
200 assert_se(r >= 0);
201
202 r = load_env_file(NULL, p, NULL, &b);
203 assert_se(r >= 0);
204
205 unlink(t);
206 unlink(p);
207 }
208
209 static void test_merge_env_file(void) {
210 char t[] = "/tmp/test-fileio-XXXXXX";
211 int fd, r;
212 FILE *f;
213 _cleanup_strv_free_ char **a = NULL;
214 char **i;
215
216 fd = mkostemp_safe(t);
217 assert_se(fd >= 0);
218
219 log_info("/* %s (%s) */", __func__, t);
220
221 f = fdopen(fd, "w");
222 assert_se(f);
223
224 r = write_string_stream(f,
225 "one=1 \n"
226 "twelve=${one}2\n"
227 "twentyone=2${one}\n"
228 "one=2\n"
229 "twentytwo=2${one}\n", false);
230 assert(r >= 0);
231
232 r = merge_env_file(&a, NULL, t);
233 assert_se(r >= 0);
234 strv_sort(a);
235
236 STRV_FOREACH(i, a)
237 log_info("Got: <%s>", *i);
238
239 assert_se(streq(a[0], "one=2"));
240 assert_se(streq(a[1], "twelve=12"));
241 assert_se(streq(a[2], "twentyone=21"));
242 assert_se(streq(a[3], "twentytwo=22"));
243 assert_se(a[4] == NULL);
244
245
246 r = merge_env_file(&a, NULL, t);
247 assert_se(r >= 0);
248 strv_sort(a);
249
250 STRV_FOREACH(i, a)
251 log_info("Got2: <%s>", *i);
252
253 assert_se(streq(a[0], "one=2"));
254 assert_se(streq(a[1], "twelve=12"));
255 assert_se(streq(a[2], "twentyone=21"));
256 assert_se(streq(a[3], "twentytwo=22"));
257 assert_se(a[4] == NULL);
258 }
259
260 static void test_executable_is_script(void) {
261 char t[] = "/tmp/test-executable-XXXXXX";
262 int fd, r;
263 FILE *f;
264 char *command;
265
266 fd = mkostemp_safe(t);
267 assert_se(fd >= 0);
268
269 f = fdopen(fd, "w");
270 assert_se(f);
271
272 fputs("#! /bin/script -a -b \ngoo goo", f);
273 fflush(f);
274
275 r = executable_is_script(t, &command);
276 assert_se(r > 0);
277 assert_se(streq(command, "/bin/script"));
278 free(command);
279
280 r = executable_is_script("/bin/sh", &command);
281 assert_se(r == 0);
282
283 r = executable_is_script("/usr/bin/yum", &command);
284 assert_se(r > 0 || r == -ENOENT);
285 if (r > 0) {
286 assert_se(startswith(command, "/"));
287 free(command);
288 }
289
290 fclose(f);
291 unlink(t);
292 }
293
294 static void test_status_field(void) {
295 _cleanup_free_ char *t = NULL, *p = NULL, *s = NULL, *z = NULL;
296 unsigned long long total = 0, buffers = 0;
297 int r;
298
299 assert_se(get_proc_field("/proc/self/status", "Threads", WHITESPACE, &t) == 0);
300 puts(t);
301 assert_se(streq(t, "1"));
302
303 r = get_proc_field("/proc/meminfo", "MemTotal", WHITESPACE, &p);
304 if (r != -ENOENT) {
305 assert_se(r == 0);
306 puts(p);
307 assert_se(safe_atollu(p, &total) == 0);
308 }
309
310 r = get_proc_field("/proc/meminfo", "Buffers", WHITESPACE, &s);
311 if (r != -ENOENT) {
312 assert_se(r == 0);
313 puts(s);
314 assert_se(safe_atollu(s, &buffers) == 0);
315 }
316
317 if (p)
318 assert_se(buffers < total);
319
320 /* Seccomp should be a good test for field full of zeros. */
321 r = get_proc_field("/proc/meminfo", "Seccomp", WHITESPACE, &z);
322 if (r != -ENOENT) {
323 assert_se(r == 0);
324 puts(z);
325 assert_se(safe_atollu(z, &buffers) == 0);
326 }
327 }
328
329 static void test_capeff(void) {
330 int pid, p;
331
332 for (pid = 0; pid < 2; pid++) {
333 _cleanup_free_ char *capeff = NULL;
334 int r;
335
336 r = get_process_capeff(0, &capeff);
337 log_info("capeff: '%s' (r=%d)", capeff, r);
338
339 if (r == -ENOENT || r == -EPERM)
340 return;
341
342 assert_se(r == 0);
343 assert_se(*capeff);
344 p = capeff[strspn(capeff, HEXDIGITS)];
345 assert_se(!p || isspace(p));
346 }
347 }
348
349 static void test_write_string_stream(void) {
350 char fn[] = "/tmp/test-write_string_stream-XXXXXX";
351 _cleanup_fclose_ FILE *f = NULL;
352 int fd;
353 char buf[64];
354
355 fd = mkostemp_safe(fn);
356 assert_se(fd >= 0);
357
358 f = fdopen(fd, "r");
359 assert_se(f);
360 assert_se(write_string_stream(f, "boohoo", true) < 0);
361
362 f = freopen(fn, "r+", f);
363 assert_se(f);
364
365 assert_se(write_string_stream(f, "boohoo", true) == 0);
366 rewind(f);
367
368 assert_se(fgets(buf, sizeof(buf), f));
369 assert_se(streq(buf, "boohoo\n"));
370
371 f = freopen(fn, "w+", f);
372 assert_se(f);
373
374 assert_se(write_string_stream(f, "boohoo", false) == 0);
375 rewind(f);
376
377 assert_se(fgets(buf, sizeof(buf), f));
378 printf(">%s<", buf);
379 assert_se(streq(buf, "boohoo"));
380
381 unlink(fn);
382 }
383
384 static void test_write_string_file(void) {
385 char fn[] = "/tmp/test-write_string_file-XXXXXX";
386 char buf[64] = {};
387 _cleanup_close_ int fd;
388
389 fd = mkostemp_safe(fn);
390 assert_se(fd >= 0);
391
392 assert_se(write_string_file(fn, "boohoo", WRITE_STRING_FILE_CREATE) == 0);
393
394 assert_se(read(fd, buf, sizeof(buf)) == 7);
395 assert_se(streq(buf, "boohoo\n"));
396
397 unlink(fn);
398 }
399
400 static void test_write_string_file_no_create(void) {
401 char fn[] = "/tmp/test-write_string_file_no_create-XXXXXX";
402 _cleanup_close_ int fd;
403 char buf[64] = {0};
404
405 fd = mkostemp_safe(fn);
406 assert_se(fd >= 0);
407
408 assert_se(write_string_file("/a/file/which/does/not/exists/i/guess", "boohoo", 0) < 0);
409 assert_se(write_string_file(fn, "boohoo", 0) == 0);
410
411 assert_se(read(fd, buf, sizeof(buf)) == strlen("boohoo\n"));
412 assert_se(streq(buf, "boohoo\n"));
413
414 unlink(fn);
415 }
416
417 static void test_write_string_file_verify(void) {
418 _cleanup_free_ char *buf = NULL, *buf2 = NULL;
419 int r;
420
421 assert_se(read_one_line_file("/proc/cmdline", &buf) >= 0);
422 assert_se((buf2 = strjoin(buf, "\n")));
423
424 r = write_string_file("/proc/cmdline", buf, 0);
425 assert_se(r == -EACCES || r == -EIO);
426 r = write_string_file("/proc/cmdline", buf2, 0);
427 assert_se(r == -EACCES || r == -EIO);
428
429 assert_se(write_string_file("/proc/cmdline", buf, WRITE_STRING_FILE_VERIFY_ON_FAILURE) == 0);
430 assert_se(write_string_file("/proc/cmdline", buf2, WRITE_STRING_FILE_VERIFY_ON_FAILURE) == 0);
431
432 r = write_string_file("/proc/cmdline", buf, WRITE_STRING_FILE_VERIFY_ON_FAILURE|WRITE_STRING_FILE_AVOID_NEWLINE);
433 assert_se(r == -EACCES || r == -EIO);
434 assert_se(write_string_file("/proc/cmdline", buf2, WRITE_STRING_FILE_VERIFY_ON_FAILURE|WRITE_STRING_FILE_AVOID_NEWLINE) == 0);
435 }
436
437 static void test_load_env_file_pairs(void) {
438 char fn[] = "/tmp/test-load_env_file_pairs-XXXXXX";
439 int fd;
440 int r;
441 _cleanup_fclose_ FILE *f = NULL;
442 _cleanup_strv_free_ char **l = NULL;
443 char **k, **v;
444
445 fd = mkostemp_safe(fn);
446 assert_se(fd >= 0);
447
448 r = write_string_file(fn,
449 "NAME=\"Arch Linux\"\n"
450 "ID=arch\n"
451 "PRETTY_NAME=\"Arch Linux\"\n"
452 "ANSI_COLOR=\"0;36\"\n"
453 "HOME_URL=\"https://www.archlinux.org/\"\n"
454 "SUPPORT_URL=\"https://bbs.archlinux.org/\"\n"
455 "BUG_REPORT_URL=\"https://bugs.archlinux.org/\"\n",
456 WRITE_STRING_FILE_CREATE);
457 assert_se(r == 0);
458
459 f = fdopen(fd, "r");
460 assert_se(f);
461
462 r = load_env_file_pairs(f, fn, NULL, &l);
463 assert_se(r >= 0);
464
465 assert_se(strv_length(l) == 14);
466 STRV_FOREACH_PAIR(k, v, l) {
467 assert_se(STR_IN_SET(*k, "NAME", "ID", "PRETTY_NAME", "ANSI_COLOR", "HOME_URL", "SUPPORT_URL", "BUG_REPORT_URL"));
468 printf("%s=%s\n", *k, *v);
469 if (streq(*k, "NAME")) assert_se(streq(*v, "Arch Linux"));
470 if (streq(*k, "ID")) assert_se(streq(*v, "arch"));
471 if (streq(*k, "PRETTY_NAME")) assert_se(streq(*v, "Arch Linux"));
472 if (streq(*k, "ANSI_COLOR")) assert_se(streq(*v, "0;36"));
473 if (streq(*k, "HOME_URL")) assert_se(streq(*v, "https://www.archlinux.org/"));
474 if (streq(*k, "SUPPORT_URL")) assert_se(streq(*v, "https://bbs.archlinux.org/"));
475 if (streq(*k, "BUG_REPORT_URL")) assert_se(streq(*v, "https://bugs.archlinux.org/"));
476 }
477
478 unlink(fn);
479 }
480
481 static void test_search_and_fopen(void) {
482 const char *dirs[] = {"/tmp/foo/bar", "/tmp", NULL};
483 char name[] = "/tmp/test-search_and_fopen.XXXXXX";
484 int fd = -1;
485 int r;
486 FILE *f;
487
488 fd = mkostemp_safe(name);
489 assert_se(fd >= 0);
490 close(fd);
491
492 r = search_and_fopen(basename(name), "r", NULL, dirs, &f);
493 assert_se(r >= 0);
494 fclose(f);
495
496 r = search_and_fopen(name, "r", NULL, dirs, &f);
497 assert_se(r >= 0);
498 fclose(f);
499
500 r = search_and_fopen(basename(name), "r", "/", dirs, &f);
501 assert_se(r >= 0);
502 fclose(f);
503
504 r = search_and_fopen("/a/file/which/does/not/exist/i/guess", "r", NULL, dirs, &f);
505 assert_se(r < 0);
506 r = search_and_fopen("afilewhichdoesnotexistiguess", "r", NULL, dirs, &f);
507 assert_se(r < 0);
508
509 r = unlink(name);
510 assert_se(r == 0);
511
512 r = search_and_fopen(basename(name), "r", NULL, dirs, &f);
513 assert_se(r < 0);
514 }
515
516
517 static void test_search_and_fopen_nulstr(void) {
518 const char dirs[] = "/tmp/foo/bar\0/tmp\0";
519 char name[] = "/tmp/test-search_and_fopen.XXXXXX";
520 int fd = -1;
521 int r;
522 FILE *f;
523
524 fd = mkostemp_safe(name);
525 assert_se(fd >= 0);
526 close(fd);
527
528 r = search_and_fopen_nulstr(basename(name), "r", NULL, dirs, &f);
529 assert_se(r >= 0);
530 fclose(f);
531
532 r = search_and_fopen_nulstr(name, "r", NULL, dirs, &f);
533 assert_se(r >= 0);
534 fclose(f);
535
536 r = search_and_fopen_nulstr("/a/file/which/does/not/exist/i/guess", "r", NULL, dirs, &f);
537 assert_se(r < 0);
538 r = search_and_fopen_nulstr("afilewhichdoesnotexistiguess", "r", NULL, dirs, &f);
539 assert_se(r < 0);
540
541 r = unlink(name);
542 assert_se(r == 0);
543
544 r = search_and_fopen_nulstr(basename(name), "r", NULL, dirs, &f);
545 assert_se(r < 0);
546 }
547
548 static void test_writing_tmpfile(void) {
549 char name[] = "/tmp/test-systemd_writing_tmpfile.XXXXXX";
550 _cleanup_free_ char *contents = NULL;
551 size_t size;
552 int fd, r;
553 struct iovec iov[3];
554
555 IOVEC_SET_STRING(iov[0], "abc\n");
556 IOVEC_SET_STRING(iov[1], ALPHANUMERICAL "\n");
557 IOVEC_SET_STRING(iov[2], "");
558
559 fd = mkostemp_safe(name);
560 printf("tmpfile: %s", name);
561
562 r = writev(fd, iov, 3);
563 assert_se(r >= 0);
564
565 r = read_full_file(name, &contents, &size);
566 assert_se(r == 0);
567 printf("contents: %s", contents);
568 assert_se(streq(contents, "abc\n" ALPHANUMERICAL "\n"));
569
570 unlink(name);
571 }
572
573 static void test_tempfn(void) {
574 char *ret = NULL, *p;
575
576 assert_se(tempfn_xxxxxx("/foo/bar/waldo", NULL, &ret) >= 0);
577 assert_se(streq_ptr(ret, "/foo/bar/.#waldoXXXXXX"));
578 free(ret);
579
580 assert_se(tempfn_xxxxxx("/foo/bar/waldo", "[miau]", &ret) >= 0);
581 assert_se(streq_ptr(ret, "/foo/bar/.#[miau]waldoXXXXXX"));
582 free(ret);
583
584 assert_se(tempfn_random("/foo/bar/waldo", NULL, &ret) >= 0);
585 assert_se(p = startswith(ret, "/foo/bar/.#waldo"));
586 assert_se(strlen(p) == 16);
587 assert_se(in_charset(p, "0123456789abcdef"));
588 free(ret);
589
590 assert_se(tempfn_random("/foo/bar/waldo", "[wuff]", &ret) >= 0);
591 assert_se(p = startswith(ret, "/foo/bar/.#[wuff]waldo"));
592 assert_se(strlen(p) == 16);
593 assert_se(in_charset(p, "0123456789abcdef"));
594 free(ret);
595
596 assert_se(tempfn_random_child("/foo/bar/waldo", NULL, &ret) >= 0);
597 assert_se(p = startswith(ret, "/foo/bar/waldo/.#"));
598 assert_se(strlen(p) == 16);
599 assert_se(in_charset(p, "0123456789abcdef"));
600 free(ret);
601
602 assert_se(tempfn_random_child("/foo/bar/waldo", "[kikiriki]", &ret) >= 0);
603 assert_se(p = startswith(ret, "/foo/bar/waldo/.#[kikiriki]"));
604 assert_se(strlen(p) == 16);
605 assert_se(in_charset(p, "0123456789abcdef"));
606 free(ret);
607 }
608
609 int main(int argc, char *argv[]) {
610 log_set_max_level(LOG_DEBUG);
611 log_parse_environment();
612 log_open();
613
614 test_parse_env_file();
615 test_parse_multiline_env_file();
616 test_merge_env_file();
617 test_executable_is_script();
618 test_status_field();
619 test_capeff();
620 test_write_string_stream();
621 test_write_string_file();
622 test_write_string_file_no_create();
623 test_write_string_file_verify();
624 test_load_env_file_pairs();
625 test_search_and_fopen();
626 test_search_and_fopen_nulstr();
627 test_writing_tmpfile();
628 test_tempfn();
629
630 return 0;
631 }