1 /* SPDX-License-Identifier: LGPL-2.1+ */
7 #include "alloc-util.h"
14 #include "parse-util.h"
15 #include "process-util.h"
16 #include "string-util.h"
19 #include "tmpfile-util.h"
22 static void test_parse_env_file(void) {
23 _cleanup_(unlink_tempfilep
) char
24 t
[] = "/tmp/test-fileio-in-XXXXXX",
25 p
[] = "/tmp/test-fileio-out-XXXXXX";
27 _cleanup_free_
char *one
= NULL
, *two
= NULL
, *three
= NULL
, *four
= NULL
, *five
= NULL
,
28 *six
= NULL
, *seven
= NULL
, *eight
= NULL
, *nine
= NULL
, *ten
= NULL
;
29 _cleanup_strv_free_
char **a
= NULL
, **b
= NULL
;
34 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
41 "invalid line #comment\n"
44 "four = \'44\\\"44\'\n"
45 "five = \'55\\\'55\' \"FIVE\" cinco \n"
46 "six = seis sechs\\\n"
48 "seven=\"sevenval\" #nocomment\n"
49 "eight=eightval #nocomment\n"
50 "export nine=nineval\n"
58 r
= load_env_file(NULL
, t
, &a
);
62 log_info("Got: <%s>", *i
);
64 assert_se(streq_ptr(a
[0], "one=BAR"));
65 assert_se(streq_ptr(a
[1], "two=bar"));
66 assert_se(streq_ptr(a
[2], "three=333\nxxxx"));
67 assert_se(streq_ptr(a
[3], "four=44\"44"));
68 assert_se(streq_ptr(a
[4], "five=55\'55FIVEcinco"));
69 assert_se(streq_ptr(a
[5], "six=seis sechs sis"));
70 assert_se(streq_ptr(a
[6], "seven=sevenval#nocomment"));
71 assert_se(streq_ptr(a
[7], "eight=eightval #nocomment"));
72 assert_se(streq_ptr(a
[8], "export nine=nineval"));
73 assert_se(streq_ptr(a
[9], "ten="));
74 assert_se(a
[10] == NULL
);
80 log_info("Got2: <%s>", *i
);
81 assert_se(streq(*i
, a
[k
++]));
99 log_info("one=[%s]", strna(one
));
100 log_info("two=[%s]", strna(two
));
101 log_info("three=[%s]", strna(three
));
102 log_info("four=[%s]", strna(four
));
103 log_info("five=[%s]", strna(five
));
104 log_info("six=[%s]", strna(six
));
105 log_info("seven=[%s]", strna(seven
));
106 log_info("eight=[%s]", strna(eight
));
107 log_info("export nine=[%s]", strna(nine
));
108 log_info("ten=[%s]", strna(nine
));
110 assert_se(streq(one
, "BAR"));
111 assert_se(streq(two
, "bar"));
112 assert_se(streq(three
, "333\nxxxx"));
113 assert_se(streq(four
, "44\"44"));
114 assert_se(streq(five
, "55\'55FIVEcinco"));
115 assert_se(streq(six
, "seis sechs sis"));
116 assert_se(streq(seven
, "sevenval#nocomment"));
117 assert_se(streq(eight
, "eightval #nocomment"));
118 assert_se(streq(nine
, "nineval"));
119 assert_se(ten
== NULL
);
122 /* prepare a temporary file to write the environment to */
123 _cleanup_close_
int fd
= mkostemp_safe(p
);
127 r
= write_env_file(p
, a
);
130 r
= load_env_file(NULL
, p
, &b
);
134 static void test_parse_multiline_env_file(void) {
135 _cleanup_(unlink_tempfilep
) char
136 t
[] = "/tmp/test-fileio-in-XXXXXX",
137 p
[] = "/tmp/test-fileio-out-XXXXXX";
139 _cleanup_strv_free_
char **a
= NULL
, **b
= NULL
;
143 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
159 r
= load_env_file(NULL
, t
, &a
);
163 log_info("Got: <%s>", *i
);
165 assert_se(streq_ptr(a
[0], "one=BAR VAR\tGAR"));
166 assert_se(streq_ptr(a
[1], "two=bar var\tgar"));
167 assert_se(streq_ptr(a
[2], "tri=bar var \tgar "));
168 assert_se(a
[3] == NULL
);
171 _cleanup_close_
int fd
= mkostemp_safe(p
);
175 r
= write_env_file(p
, a
);
178 r
= load_env_file(NULL
, p
, &b
);
182 static void test_merge_env_file(void) {
183 _cleanup_(unlink_tempfilep
) char t
[] = "/tmp/test-fileio-XXXXXX";
184 _cleanup_fclose_
FILE *f
= NULL
;
185 _cleanup_strv_free_
char **a
= NULL
;
189 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
190 log_info("/* %s (%s) */", __func__
, t
);
192 r
= write_string_stream(f
,
195 "twentyone=2${one}\n"
197 "twentytwo=2${one}\n"
198 "xxx_minus_three=$xxx - 3\n"
199 "xxx=0x$one$one$one\n"
200 "yyy=${one:-fallback}\n"
201 "zzz=${one:+replacement}\n"
202 "zzzz=${foobar:-${nothing}}\n"
203 "zzzzz=${nothing:+${nothing}}\n"
204 , WRITE_STRING_FILE_AVOID_NEWLINE
);
207 r
= merge_env_file(&a
, NULL
, t
);
212 log_info("Got: <%s>", *i
);
214 assert_se(streq(a
[0], "one=2"));
215 assert_se(streq(a
[1], "twelve=12"));
216 assert_se(streq(a
[2], "twentyone=21"));
217 assert_se(streq(a
[3], "twentytwo=22"));
218 assert_se(streq(a
[4], "xxx=0x222"));
219 assert_se(streq(a
[5], "xxx_minus_three= - 3"));
220 assert_se(streq(a
[6], "yyy=2"));
221 assert_se(streq(a
[7], "zzz=replacement"));
222 assert_se(streq(a
[8], "zzzz="));
223 assert_se(streq(a
[9], "zzzzz="));
224 assert_se(a
[10] == NULL
);
226 r
= merge_env_file(&a
, NULL
, t
);
231 log_info("Got2: <%s>", *i
);
233 assert_se(streq(a
[0], "one=2"));
234 assert_se(streq(a
[1], "twelve=12"));
235 assert_se(streq(a
[2], "twentyone=21"));
236 assert_se(streq(a
[3], "twentytwo=22"));
237 assert_se(streq(a
[4], "xxx=0x222"));
238 assert_se(streq(a
[5], "xxx_minus_three=0x222 - 3"));
239 assert_se(streq(a
[6], "yyy=2"));
240 assert_se(streq(a
[7], "zzz=replacement"));
241 assert_se(streq(a
[8], "zzzz="));
242 assert_se(streq(a
[9], "zzzzz="));
243 assert_se(a
[10] == NULL
);
246 static void test_merge_env_file_invalid(void) {
247 _cleanup_(unlink_tempfilep
) char t
[] = "/tmp/test-fileio-XXXXXX";
248 _cleanup_fclose_
FILE *f
= NULL
;
249 _cleanup_strv_free_
char **a
= NULL
;
253 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
254 log_info("/* %s (%s) */", __func__
, t
);
256 r
= write_string_stream(f
,
265 ";comment2=comment2\n"
267 "\n\n" /* empty line */
268 , WRITE_STRING_FILE_AVOID_NEWLINE
);
271 r
= merge_env_file(&a
, NULL
, t
);
275 log_info("Got: <%s>", *i
);
277 assert_se(strv_isempty(a
));
280 static void test_executable_is_script(void) {
281 _cleanup_(unlink_tempfilep
) char t
[] = "/tmp/test-fileio-XXXXXX";
282 _cleanup_fclose_
FILE *f
= NULL
;
286 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
287 fputs("#! /bin/script -a -b \ngoo goo", f
);
290 r
= executable_is_script(t
, &command
);
292 assert_se(streq(command
, "/bin/script"));
295 r
= executable_is_script("/bin/sh", &command
);
298 r
= executable_is_script("/usr/bin/yum", &command
);
299 assert_se(r
> 0 || r
== -ENOENT
);
301 assert_se(startswith(command
, "/"));
306 static void test_status_field(void) {
307 _cleanup_free_
char *t
= NULL
, *p
= NULL
, *s
= NULL
, *z
= NULL
;
308 unsigned long long total
= 0, buffers
= 0;
311 assert_se(get_proc_field("/proc/self/status", "Threads", WHITESPACE
, &t
) == 0);
313 assert_se(streq(t
, "1"));
315 r
= get_proc_field("/proc/meminfo", "MemTotal", WHITESPACE
, &p
);
319 assert_se(safe_atollu(p
, &total
) == 0);
322 r
= get_proc_field("/proc/meminfo", "Buffers", WHITESPACE
, &s
);
326 assert_se(safe_atollu(s
, &buffers
) == 0);
330 assert_se(buffers
< total
);
332 /* Seccomp should be a good test for field full of zeros. */
333 r
= get_proc_field("/proc/meminfo", "Seccomp", WHITESPACE
, &z
);
337 assert_se(safe_atollu(z
, &buffers
) == 0);
341 static void test_capeff(void) {
344 for (pid
= 0; pid
< 2; pid
++) {
345 _cleanup_free_
char *capeff
= NULL
;
348 r
= get_process_capeff(0, &capeff
);
349 log_info("capeff: '%s' (r=%d)", capeff
, r
);
351 if (IN_SET(r
, -ENOENT
, -EPERM
))
356 p
= capeff
[strspn(capeff
, HEXDIGITS
)];
357 assert_se(!p
|| isspace(p
));
361 static void test_write_string_stream(void) {
362 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-write_string_stream-XXXXXX";
363 _cleanup_fclose_
FILE *f
= NULL
;
367 fd
= mkostemp_safe(fn
);
372 assert_se(write_string_stream(f
, "boohoo", 0) < 0);
378 assert_se(write_string_stream(f
, "boohoo", 0) == 0);
381 assert_se(fgets(buf
, sizeof(buf
), f
));
382 assert_se(streq(buf
, "boohoo\n"));
388 assert_se(write_string_stream(f
, "boohoo", WRITE_STRING_FILE_AVOID_NEWLINE
) == 0);
391 assert_se(fgets(buf
, sizeof(buf
), f
));
393 assert_se(streq(buf
, "boohoo"));
396 static void test_write_string_file(void) {
397 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-write_string_file-XXXXXX";
399 _cleanup_close_
int fd
;
401 fd
= mkostemp_safe(fn
);
404 assert_se(write_string_file(fn
, "boohoo", WRITE_STRING_FILE_CREATE
) == 0);
406 assert_se(read(fd
, buf
, sizeof(buf
)) == 7);
407 assert_se(streq(buf
, "boohoo\n"));
410 static void test_write_string_file_no_create(void) {
411 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-write_string_file_no_create-XXXXXX";
412 _cleanup_close_
int fd
;
415 fd
= mkostemp_safe(fn
);
418 assert_se(write_string_file("/a/file/which/does/not/exists/i/guess", "boohoo", 0) < 0);
419 assert_se(write_string_file(fn
, "boohoo", 0) == 0);
421 assert_se(read(fd
, buf
, sizeof(buf
)) == STRLEN("boohoo\n"));
422 assert_se(streq(buf
, "boohoo\n"));
425 static void test_write_string_file_verify(void) {
426 _cleanup_free_
char *buf
= NULL
, *buf2
= NULL
;
429 assert_se(read_one_line_file("/proc/cmdline", &buf
) >= 0);
430 assert_se(buf2
= strjoin(buf
, "\n"));
432 r
= write_string_file("/proc/cmdline", buf
, 0);
433 assert_se(IN_SET(r
, -EACCES
, -EIO
));
434 r
= write_string_file("/proc/cmdline", buf2
, 0);
435 assert_se(IN_SET(r
, -EACCES
, -EIO
));
437 assert_se(write_string_file("/proc/cmdline", buf
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
) == 0);
438 assert_se(write_string_file("/proc/cmdline", buf2
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
) == 0);
440 r
= write_string_file("/proc/cmdline", buf
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
|WRITE_STRING_FILE_AVOID_NEWLINE
);
441 assert_se(IN_SET(r
, -EACCES
, -EIO
));
442 assert_se(write_string_file("/proc/cmdline", buf2
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
|WRITE_STRING_FILE_AVOID_NEWLINE
) == 0);
445 static void test_load_env_file_pairs(void) {
446 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-load_env_file_pairs-XXXXXX";
448 _cleanup_fclose_
FILE *f
= NULL
;
449 _cleanup_strv_free_
char **l
= NULL
;
452 fd
= mkostemp_safe(fn
);
455 r
= write_string_file(fn
,
456 "NAME=\"Arch Linux\"\n"
458 "PRETTY_NAME=\"Arch Linux\"\n"
459 "ANSI_COLOR=\"0;36\"\n"
460 "HOME_URL=\"https://www.archlinux.org/\"\n"
461 "SUPPORT_URL=\"https://bbs.archlinux.org/\"\n"
462 "BUG_REPORT_URL=\"https://bugs.archlinux.org/\"\n",
463 WRITE_STRING_FILE_CREATE
);
469 r
= load_env_file_pairs(f
, fn
, &l
);
472 assert_se(strv_length(l
) == 14);
473 STRV_FOREACH_PAIR(k
, v
, l
) {
474 assert_se(STR_IN_SET(*k
, "NAME", "ID", "PRETTY_NAME", "ANSI_COLOR", "HOME_URL", "SUPPORT_URL", "BUG_REPORT_URL"));
475 printf("%s=%s\n", *k
, *v
);
476 if (streq(*k
, "NAME")) assert_se(streq(*v
, "Arch Linux"));
477 if (streq(*k
, "ID")) assert_se(streq(*v
, "arch"));
478 if (streq(*k
, "PRETTY_NAME")) assert_se(streq(*v
, "Arch Linux"));
479 if (streq(*k
, "ANSI_COLOR")) assert_se(streq(*v
, "0;36"));
480 if (streq(*k
, "HOME_URL")) assert_se(streq(*v
, "https://www.archlinux.org/"));
481 if (streq(*k
, "SUPPORT_URL")) assert_se(streq(*v
, "https://bbs.archlinux.org/"));
482 if (streq(*k
, "BUG_REPORT_URL")) assert_se(streq(*v
, "https://bugs.archlinux.org/"));
486 static void test_search_and_fopen(void) {
487 const char *dirs
[] = {"/tmp/foo/bar", "/tmp", NULL
};
489 char name
[] = "/tmp/test-search_and_fopen.XXXXXX";
493 fd
= mkostemp_safe(name
);
497 r
= search_and_fopen(basename(name
), "r", NULL
, dirs
, &f
);
501 r
= search_and_fopen(name
, "r", NULL
, dirs
, &f
);
505 r
= search_and_fopen(basename(name
), "r", "/", dirs
, &f
);
509 r
= search_and_fopen("/a/file/which/does/not/exist/i/guess", "r", NULL
, dirs
, &f
);
511 r
= search_and_fopen("afilewhichdoesnotexistiguess", "r", NULL
, dirs
, &f
);
517 r
= search_and_fopen(basename(name
), "r", NULL
, dirs
, &f
);
521 static void test_search_and_fopen_nulstr(void) {
522 const char dirs
[] = "/tmp/foo/bar\0/tmp\0";
524 _cleanup_(unlink_tempfilep
) char name
[] = "/tmp/test-search_and_fopen.XXXXXX";
528 fd
= mkostemp_safe(name
);
532 r
= search_and_fopen_nulstr(basename(name
), "r", NULL
, dirs
, &f
);
536 r
= search_and_fopen_nulstr(name
, "r", NULL
, dirs
, &f
);
540 r
= search_and_fopen_nulstr("/a/file/which/does/not/exist/i/guess", "r", NULL
, dirs
, &f
);
542 r
= search_and_fopen_nulstr("afilewhichdoesnotexistiguess", "r", NULL
, dirs
, &f
);
548 r
= search_and_fopen_nulstr(basename(name
), "r", NULL
, dirs
, &f
);
552 static void test_writing_tmpfile(void) {
553 _cleanup_(unlink_tempfilep
) char name
[] = "/tmp/test-systemd_writing_tmpfile.XXXXXX";
554 _cleanup_free_
char *contents
= NULL
;
556 _cleanup_close_
int fd
= -1;
560 iov
[0] = IOVEC_MAKE_STRING("abc\n");
561 iov
[1] = IOVEC_MAKE_STRING(ALPHANUMERICAL
"\n");
562 iov
[2] = IOVEC_MAKE_STRING("");
564 fd
= mkostemp_safe(name
);
565 printf("tmpfile: %s", name
);
567 r
= writev(fd
, iov
, 3);
570 r
= read_full_file(name
, &contents
, &size
);
572 printf("contents: %s", contents
);
573 assert_se(streq(contents
, "abc\n" ALPHANUMERICAL
"\n"));
576 static void test_tempfn(void) {
577 char *ret
= NULL
, *p
;
579 assert_se(tempfn_xxxxxx("/foo/bar/waldo", NULL
, &ret
) >= 0);
580 assert_se(streq_ptr(ret
, "/foo/bar/.#waldoXXXXXX"));
583 assert_se(tempfn_xxxxxx("/foo/bar/waldo", "[miau]", &ret
) >= 0);
584 assert_se(streq_ptr(ret
, "/foo/bar/.#[miau]waldoXXXXXX"));
587 assert_se(tempfn_random("/foo/bar/waldo", NULL
, &ret
) >= 0);
588 assert_se(p
= startswith(ret
, "/foo/bar/.#waldo"));
589 assert_se(strlen(p
) == 16);
590 assert_se(in_charset(p
, "0123456789abcdef"));
593 assert_se(tempfn_random("/foo/bar/waldo", "[wuff]", &ret
) >= 0);
594 assert_se(p
= startswith(ret
, "/foo/bar/.#[wuff]waldo"));
595 assert_se(strlen(p
) == 16);
596 assert_se(in_charset(p
, "0123456789abcdef"));
599 assert_se(tempfn_random_child("/foo/bar/waldo", NULL
, &ret
) >= 0);
600 assert_se(p
= startswith(ret
, "/foo/bar/waldo/.#"));
601 assert_se(strlen(p
) == 16);
602 assert_se(in_charset(p
, "0123456789abcdef"));
605 assert_se(tempfn_random_child("/foo/bar/waldo", "[kikiriki]", &ret
) >= 0);
606 assert_se(p
= startswith(ret
, "/foo/bar/waldo/.#[kikiriki]"));
607 assert_se(strlen(p
) == 16);
608 assert_se(in_charset(p
, "0123456789abcdef"));
612 static const char buffer
[] =
614 "With newlines, and a NUL byte\0"
618 "and a very long line that is supposed to be truncated, because it is so long\n";
620 static void test_read_line_one_file(FILE *f
) {
621 _cleanup_free_
char *line
= NULL
;
623 assert_se(read_line(f
, (size_t) -1, &line
) == 15 && streq(line
, "Some test data"));
626 assert_se(read_line(f
, 1024, &line
) == 30 && streq(line
, "With newlines, and a NUL byte"));
629 assert_se(read_line(f
, 1024, &line
) == 1 && streq(line
, ""));
632 assert_se(read_line(f
, 1024, &line
) == 14 && streq(line
, "an empty line"));
635 assert_se(read_line(f
, (size_t) -1, NULL
) == 16);
637 assert_se(read_line(f
, 16, &line
) == -ENOBUFS
);
640 /* read_line() stopped when it hit the limit, that means when we continue reading we'll read at the first
641 * character after the previous limit. Let's make use of tha to continue our test. */
642 assert_se(read_line(f
, 1024, &line
) == 61 && streq(line
, "line that is supposed to be truncated, because it is so long"));
645 assert_se(read_line(f
, 1024, &line
) == 1 && streq(line
, ""));
648 assert_se(read_line(f
, 1024, &line
) == 0 && streq(line
, ""));
651 static void test_read_line(void) {
652 _cleanup_fclose_
FILE *f
= NULL
;
654 f
= fmemopen((void*) buffer
, sizeof(buffer
), "re");
657 test_read_line_one_file(f
);
660 static void test_read_line2(void) {
661 _cleanup_(unlink_tempfilep
) char name
[] = "/tmp/test-fileio.XXXXXX";
663 _cleanup_fclose_
FILE *f
= NULL
;
665 fd
= mkostemp_safe(name
);
667 assert_se((size_t) write(fd
, buffer
, sizeof(buffer
)) == sizeof(buffer
));
669 assert_se(lseek(fd
, 0, SEEK_SET
) == 0);
670 assert_se(f
= fdopen(fd
, "r"));
672 test_read_line_one_file(f
);
675 static void test_read_line3(void) {
676 _cleanup_fclose_
FILE *f
= NULL
;
677 _cleanup_free_
char *line
= NULL
;
680 f
= fopen("/proc/cmdline", "re");
681 if (!f
&& IN_SET(errno
, ENOENT
, EPERM
))
685 r
= read_line(f
, LINE_MAX
, &line
);
686 assert_se((size_t) r
== strlen(line
) + 1);
687 assert_se(read_line(f
, LINE_MAX
, NULL
) == 0);
690 int main(int argc
, char *argv
[]) {
691 test_setup_logging(LOG_DEBUG
);
693 test_parse_env_file();
694 test_parse_multiline_env_file();
695 test_merge_env_file();
696 test_merge_env_file_invalid();
697 test_executable_is_script();
700 test_write_string_stream();
701 test_write_string_file();
702 test_write_string_file_no_create();
703 test_write_string_file_verify();
704 test_load_env_file_pairs();
705 test_search_and_fopen();
706 test_search_and_fopen_nulstr();
707 test_writing_tmpfile();