1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
7 #include "alloc-util.h"
10 #include "errno-util.h"
14 #include "iovec-util.h"
15 #include "memfd-util.h"
16 #include "parse-util.h"
17 #include "path-util.h"
18 #include "process-util.h"
19 #include "random-util.h"
21 #include "socket-util.h"
22 #include "string-util.h"
25 #include "tmpfile-util.h"
27 TEST(parse_env_file
) {
28 _cleanup_(unlink_tempfilep
) char
29 t
[] = "/tmp/test-fileio-in-XXXXXX",
30 p
[] = "/tmp/test-fileio-out-XXXXXX";
32 _cleanup_free_
char *one
= NULL
, *two
= NULL
, *three
= NULL
, *four
= NULL
, *five
= NULL
,
33 *six
= NULL
, *seven
= NULL
, *eight
= NULL
, *nine
= NULL
, *ten
= NULL
,
34 *eleven
= NULL
, *twelve
= NULL
, *thirteen
= NULL
;
35 _cleanup_strv_free_
char **a
= NULL
, **b
= NULL
;
39 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
46 "invalid line #comment\n"
49 "four = \'44\\\"44\'\n"
50 "five = \"55\\\"55\" \"FIVE\" cinco \n"
51 "six = seis sechs\\\n"
53 "seven=\"sevenval\" #nocomment\n"
54 "eight=eightval #nocomment\n"
55 "export nine=nineval\n"
60 "twelve=\"\\value\"\n"
61 "thirteen='\\value'", f
);
66 r
= load_env_file(NULL
, t
, &a
);
70 log_info("Got: <%s>", *i
);
72 ASSERT_STREQ(a
[0], "one=BAR");
73 ASSERT_STREQ(a
[1], "two=bar");
74 ASSERT_STREQ(a
[2], "three=333\nxxxx");
75 ASSERT_STREQ(a
[3], "four=44\\\"44");
76 ASSERT_STREQ(a
[4], "five=55\"55FIVEcinco");
77 ASSERT_STREQ(a
[5], "six=seis sechs sis");
78 ASSERT_STREQ(a
[6], "seven=sevenval#nocomment");
79 ASSERT_STREQ(a
[7], "eight=eightval #nocomment");
80 ASSERT_STREQ(a
[8], "export nine=nineval");
81 ASSERT_STREQ(a
[9], "ten=");
82 ASSERT_STREQ(a
[10], "eleven=value");
83 ASSERT_STREQ(a
[11], "twelve=\\value");
84 ASSERT_STREQ(a
[12], "thirteen=\\value");
91 log_info("Got2: <%s>", *i
);
92 ASSERT_STREQ(*i
, a
[k
++]);
105 "export nine", &nine
,
109 "thirteen", &thirteen
);
112 log_info("one=[%s]", strna(one
));
113 log_info("two=[%s]", strna(two
));
114 log_info("three=[%s]", strna(three
));
115 log_info("four=[%s]", strna(four
));
116 log_info("five=[%s]", strna(five
));
117 log_info("six=[%s]", strna(six
));
118 log_info("seven=[%s]", strna(seven
));
119 log_info("eight=[%s]", strna(eight
));
120 log_info("export nine=[%s]", strna(nine
));
121 log_info("ten=[%s]", strna(nine
));
122 log_info("eleven=[%s]", strna(eleven
));
123 log_info("twelve=[%s]", strna(twelve
));
124 log_info("thirteen=[%s]", strna(thirteen
));
126 ASSERT_STREQ(one
, "BAR");
127 ASSERT_STREQ(two
, "bar");
128 ASSERT_STREQ(three
, "333\nxxxx");
129 ASSERT_STREQ(four
, "44\\\"44");
130 ASSERT_STREQ(five
, "55\"55FIVEcinco");
131 ASSERT_STREQ(six
, "seis sechs sis");
132 ASSERT_STREQ(seven
, "sevenval#nocomment");
133 ASSERT_STREQ(eight
, "eightval #nocomment");
134 ASSERT_STREQ(nine
, "nineval");
136 ASSERT_STREQ(eleven
, "value");
137 ASSERT_STREQ(twelve
, "\\value");
138 ASSERT_STREQ(thirteen
, "\\value");
141 /* prepare a temporary file to write the environment to */
142 _cleanup_close_
int fd
= mkostemp_safe(p
);
146 r
= write_env_file(AT_FDCWD
, p
, /* headers= */ NULL
, a
, /* flags= */ 0);
149 r
= load_env_file(NULL
, p
, &b
);
153 static void test_one_shell_var(const char *file
, const char *variable
, const char *value
) {
154 _cleanup_free_
char *cmd
= NULL
, *from_shell
= NULL
;
155 _cleanup_pclose_
FILE *f
= NULL
;
158 assert_se(cmd
= strjoin(". ", file
, " && /bin/echo -n \"$", variable
, "\""));
159 assert_se(f
= popen(cmd
, "re"));
160 assert_se(read_full_stream(f
, &from_shell
, &sz
) >= 0);
161 assert_se(sz
== strlen(value
));
162 ASSERT_STREQ(from_shell
, value
);
165 TEST(parse_multiline_env_file
) {
166 _cleanup_(unlink_tempfilep
) char
167 t
[] = "/tmp/test-fileio-in-XXXXXX",
168 p
[] = "/tmp/test-fileio-out-XXXXXX";
170 _cleanup_strv_free_
char **a
= NULL
, **b
= NULL
;
173 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
175 "\\ \\ \\ \\ VAR\\\n"
186 assert_se(fflush_and_check(f
) >= 0);
189 test_one_shell_var(t
, "one", "BAR VAR\tGAR");
190 test_one_shell_var(t
, "two", "bar var\tgar");
191 test_one_shell_var(t
, "tri", "bar var \tgar ");
193 r
= load_env_file(NULL
, t
, &a
);
197 log_info("Got: <%s>", *i
);
199 ASSERT_STREQ(a
[0], "one=BAR VAR\tGAR");
200 ASSERT_STREQ(a
[1], "two=bar var\tgar");
201 ASSERT_STREQ(a
[2], "tri=bar var \tgar ");
205 _cleanup_close_
int fd
= mkostemp_safe(p
);
209 r
= write_env_file(AT_FDCWD
, p
, /* headers= */ NULL
, a
, /* flags= */ 0);
212 r
= load_env_file(NULL
, p
, &b
);
216 TEST(merge_env_file
) {
217 _cleanup_(unlink_tempfilep
) char t
[] = "/tmp/test-fileio-XXXXXX";
218 _cleanup_fclose_
FILE *f
= NULL
;
219 _cleanup_strv_free_
char **a
= NULL
;
222 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
223 log_info("/* %s (%s) */", __func__
, t
);
225 r
= write_string_stream(f
,
228 "twentyone=2${one}\n"
230 "twentytwo=2${one}\n"
231 "xxx_minus_three=$xxx - 3\n"
232 "xxx=0x$one$one$one\n"
233 "yyy=${one:-fallback}\n"
234 "zzz=${one:+replacement}\n"
235 "zzzz=${foobar:-${nothing}}\n"
236 "zzzzz=${nothing:+${nothing}}\n"
237 , WRITE_STRING_FILE_AVOID_NEWLINE
);
240 r
= merge_env_file(&a
, NULL
, t
);
245 log_info("Got: <%s>", *i
);
247 ASSERT_STREQ(a
[0], "one=2");
248 ASSERT_STREQ(a
[1], "twelve=12");
249 ASSERT_STREQ(a
[2], "twentyone=21");
250 ASSERT_STREQ(a
[3], "twentytwo=22");
251 ASSERT_STREQ(a
[4], "xxx=0x222");
252 ASSERT_STREQ(a
[5], "xxx_minus_three= - 3");
253 ASSERT_STREQ(a
[6], "yyy=2");
254 ASSERT_STREQ(a
[7], "zzz=replacement");
255 ASSERT_STREQ(a
[8], "zzzz=");
256 ASSERT_STREQ(a
[9], "zzzzz=");
259 r
= merge_env_file(&a
, NULL
, t
);
264 log_info("Got2: <%s>", *i
);
266 ASSERT_STREQ(a
[0], "one=2");
267 ASSERT_STREQ(a
[1], "twelve=12");
268 ASSERT_STREQ(a
[2], "twentyone=21");
269 ASSERT_STREQ(a
[3], "twentytwo=22");
270 ASSERT_STREQ(a
[4], "xxx=0x222");
271 ASSERT_STREQ(a
[5], "xxx_minus_three=0x222 - 3");
272 ASSERT_STREQ(a
[6], "yyy=2");
273 ASSERT_STREQ(a
[7], "zzz=replacement");
274 ASSERT_STREQ(a
[8], "zzzz=");
275 ASSERT_STREQ(a
[9], "zzzzz=");
279 TEST(merge_env_file_invalid
) {
280 _cleanup_(unlink_tempfilep
) char t
[] = "/tmp/test-fileio-XXXXXX";
281 _cleanup_fclose_
FILE *f
= NULL
;
282 _cleanup_strv_free_
char **a
= NULL
;
285 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
286 log_info("/* %s (%s) */", __func__
, t
);
288 r
= write_string_stream(f
,
297 ";comment2=comment2\n"
299 "\n\n" /* empty line */
300 , WRITE_STRING_FILE_AVOID_NEWLINE
);
303 r
= merge_env_file(&a
, NULL
, t
);
307 log_info("Got: <%s>", *i
);
309 assert_se(strv_isempty(a
));
312 TEST(script_get_shebang_interpreter
) {
313 _cleanup_(unlink_tempfilep
) char t
[] = "/tmp/test-fileio-XXXXXX";
314 _cleanup_fclose_
FILE *f
= NULL
;
317 assert_se(fmkostemp_safe(t
, "w", &f
) == 0);
318 fputs("#! /bin/script -a -b \ngoo goo", f
);
321 ASSERT_OK(script_get_shebang_interpreter(t
, &command
));
322 ASSERT_STREQ(command
, "/bin/script");
325 ASSERT_ERROR(script_get_shebang_interpreter("/bin/sh", NULL
), EMEDIUMTYPE
);
327 if (script_get_shebang_interpreter("/usr/bin/yum", &command
) >= 0) {
328 assert_se(startswith(command
, "/"));
334 _cleanup_free_
char *p
= NULL
, *s
= NULL
;
335 unsigned long long total
= 0, buffers
= 0;
338 r
= get_proc_field("/proc/meminfo", "MemTotal", &p
);
339 if (!IN_SET(r
, -ENOENT
, -ENOSYS
)) {
342 assert_se(safe_atollu(p
, &total
) == 0);
345 r
= get_proc_field("/proc/meminfo", "Buffers", &s
);
346 if (!IN_SET(r
, -ENOENT
, -ENOSYS
)) {
349 assert_se(safe_atollu(s
, &buffers
) == 0);
353 assert_se(buffers
< total
);
356 TEST(read_one_line_file
) {
357 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-fileio-1lf-XXXXXX";
359 _cleanup_fclose_
FILE *f
= NULL
;
360 _cleanup_free_
char *buf
, *buf2
, *buf3
, *buf4
, *buf5
;
362 fd
= mkostemp_safe(fn
);
365 f
= fdopen(fd
, "we");
368 assert_se(read_one_line_file(fn
, &buf
) == 0);
369 ASSERT_STREQ(buf
, "");
370 assert_se(read_one_line_file(fn
, &buf2
) == 0);
371 ASSERT_STREQ(buf2
, "");
373 assert_se(write_string_stream(f
, "x", WRITE_STRING_FILE_AVOID_NEWLINE
) >= 0);
376 assert_se(read_one_line_file(fn
, &buf3
) == 1);
377 ASSERT_STREQ(buf3
, "x");
379 assert_se(write_string_stream(f
, "\n", WRITE_STRING_FILE_AVOID_NEWLINE
) >= 0);
382 assert_se(read_one_line_file(fn
, &buf4
) == 2);
383 ASSERT_STREQ(buf4
, "x");
385 assert_se(write_string_stream(f
, "\n", WRITE_STRING_FILE_AVOID_NEWLINE
) >= 0);
388 assert_se(read_one_line_file(fn
, &buf5
) == 2);
389 ASSERT_STREQ(buf5
, "x");
392 TEST(write_string_stream
) {
393 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-write_string_stream-XXXXXX";
394 _cleanup_fclose_
FILE *f
= NULL
;
398 fd
= mkostemp_safe(fn
);
403 assert_se(write_string_stream(f
, "boohoo", 0) < 0);
409 assert_se(write_string_stream(f
, "boohoo", 0) == 0);
412 assert_se(fgets(buf
, sizeof(buf
), f
));
413 ASSERT_STREQ(buf
, "boohoo\n");
419 assert_se(write_string_stream(f
, "boohoo", WRITE_STRING_FILE_AVOID_NEWLINE
) == 0);
422 assert_se(fgets(buf
, sizeof(buf
), f
));
424 ASSERT_STREQ(buf
, "boohoo");
427 TEST(write_string_file
) {
428 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-write_string_file-XXXXXX";
430 _cleanup_close_
int fd
= -EBADF
;
432 fd
= mkostemp_safe(fn
);
435 assert_se(write_string_file(fn
, "boohoo", WRITE_STRING_FILE_CREATE
) == 0);
437 assert_se(read(fd
, buf
, sizeof(buf
)) == 7);
438 ASSERT_STREQ(buf
, "boohoo\n");
441 TEST(write_string_file_no_create
) {
442 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-write_string_file_no_create-XXXXXX";
443 _cleanup_close_
int fd
= -EBADF
;
446 fd
= mkostemp_safe(fn
);
449 assert_se(write_string_file("/a/file/which/does/not/exists/i/guess", "boohoo", 0) < 0);
450 assert_se(write_string_file(fn
, "boohoo", 0) == 0);
452 assert_se(read(fd
, buf
, sizeof buf
) == (ssize_t
) strlen("boohoo\n"));
453 ASSERT_STREQ(buf
, "boohoo\n");
456 TEST(write_string_file_verify
) {
457 _cleanup_free_
char *buf
= NULL
, *buf2
= NULL
;
460 r
= read_one_line_file("/proc/version", &buf
);
461 if (ERRNO_IS_NEG_PRIVILEGE(r
))
464 assert_se(buf2
= strjoin(buf
, "\n"));
466 r
= write_string_file("/proc/version", buf
, 0);
467 assert_se(IN_SET(r
, -EACCES
, -EIO
));
468 r
= write_string_file("/proc/version", buf2
, 0);
469 assert_se(IN_SET(r
, -EACCES
, -EIO
));
471 assert_se(write_string_file("/proc/version", buf
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
) == 0);
472 assert_se(write_string_file("/proc/version", buf2
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
) == 0);
474 r
= write_string_file("/proc/version", buf
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
|WRITE_STRING_FILE_AVOID_NEWLINE
);
475 assert_se(IN_SET(r
, -EACCES
, -EIO
));
476 assert_se(write_string_file("/proc/version", buf2
, WRITE_STRING_FILE_VERIFY_ON_FAILURE
|WRITE_STRING_FILE_AVOID_NEWLINE
) == 0);
479 static void check_file_pairs_one(char **l
) {
481 assert_se(strv_length(l
) == 14);
483 STRV_FOREACH_PAIR(k
, v
, l
) {
484 assert_se(STR_IN_SET(*k
, "NAME", "ID", "PRETTY_NAME", "ANSI_COLOR", "HOME_URL", "SUPPORT_URL", "BUG_REPORT_URL"));
485 printf("%s=%s\n", *k
, *v
);
486 assert_se(!streq(*k
, "NAME") || streq(*v
, "Arch Linux"));
487 assert_se(!streq(*k
, "ID") || streq(*v
, "arch"));
488 assert_se(!streq(*k
, "PRETTY_NAME") || streq(*v
, "Arch Linux"));
489 assert_se(!streq(*k
, "ANSI_COLOR") || streq(*v
, "0;36"));
490 assert_se(!streq(*k
, "HOME_URL") || streq(*v
, "https://www.archlinux.org/"));
491 assert_se(!streq(*k
, "SUPPORT_URL") || streq(*v
, "https://bbs.archlinux.org/"));
492 assert_se(!streq(*k
, "BUG_REPORT_URL") || streq(*v
, "https://bugs.archlinux.org/"));
496 TEST(load_env_file_pairs
) {
497 _cleanup_(unlink_tempfilep
) char fn
[] = "/tmp/test-load_env_file_pairs-XXXXXX";
499 _cleanup_fclose_
FILE *f
= NULL
;
500 _cleanup_strv_free_
char **l
= NULL
;
502 fd
= mkostemp_safe(fn
);
505 r
= write_string_file(fn
,
506 "NAME=\"Arch Linux\"\n"
508 "PRETTY_NAME=\"Arch Linux\"\n"
509 "ANSI_COLOR=\"0;36\"\n"
510 "HOME_URL=\"https://www.archlinux.org/\"\n"
511 "SUPPORT_URL=\"https://bbs.archlinux.org/\"\n"
512 "BUG_REPORT_URL=\"https://bugs.archlinux.org/\"\n",
513 WRITE_STRING_FILE_CREATE
);
516 r
= load_env_file_pairs_fd(fd
, fn
, &l
);
518 check_file_pairs_one(l
);
524 r
= load_env_file_pairs(f
, fn
, &l
);
526 check_file_pairs_one(l
);
529 TEST(search_and_fopen
) {
530 static const char* const dirs
[] = {
535 char name
[] = "/tmp/test-search_and_fopen.XXXXXX";
536 _cleanup_fclose_
FILE *f
= NULL
;
537 _cleanup_free_
char *p
= NULL
, *bn
= NULL
;
538 _cleanup_close_
int fd
= -EBADF
;
542 fd
= mkostemp_safe(name
);
546 ASSERT_OK(path_extract_filename(name
, &bn
));
547 ASSERT_OK(search_and_fopen(bn
, "re", NULL
, (const char**) dirs
, &f
, &p
));
548 assert_se(e
= path_startswith(p
, "/tmp/"));
553 ASSERT_OK(search_and_fopen(bn
, NULL
, NULL
, (const char**) dirs
, NULL
, &p
));
554 assert_se(e
= path_startswith(p
, "/tmp/"));
558 ASSERT_OK(search_and_fopen(name
, "re", NULL
, (const char**) dirs
, &f
, &p
));
559 assert_se(path_equal(name
, p
));
563 ASSERT_OK(search_and_fopen(name
, NULL
, NULL
, (const char**) dirs
, NULL
, &p
));
564 assert_se(path_equal(name
, p
));
567 ASSERT_OK(search_and_fopen(bn
, "re", "/", (const char**) dirs
, &f
, &p
));
568 assert_se(e
= path_startswith(p
, "/tmp/"));
573 ASSERT_OK(search_and_fopen(bn
, NULL
, "/", (const char**) dirs
, NULL
, &p
));
574 assert_se(e
= path_startswith(p
, "/tmp/"));
578 ASSERT_ERROR(search_and_fopen("/a/file/which/does/not/exist/i/guess", "re", NULL
, (const char**) dirs
, &f
, &p
), ENOENT
);
579 ASSERT_ERROR(search_and_fopen("/a/file/which/does/not/exist/i/guess", NULL
, NULL
, (const char**) dirs
, NULL
, &p
), ENOENT
);
580 ASSERT_ERROR(search_and_fopen("afilewhichdoesnotexistiguess", "re", NULL
, (const char**) dirs
, &f
, &p
), ENOENT
);
581 ASSERT_ERROR(search_and_fopen("afilewhichdoesnotexistiguess", NULL
, NULL
, (const char**) dirs
, NULL
, &p
), ENOENT
);
586 ASSERT_ERROR(search_and_fopen(bn
, "re", NULL
, (const char**) dirs
, &f
, &p
), ENOENT
);
587 ASSERT_ERROR(search_and_fopen(bn
, NULL
, NULL
, (const char**) dirs
, NULL
, &p
), ENOENT
);
590 TEST(search_and_fopen_nulstr
) {
591 static const char dirs
[] =
595 _cleanup_(unlink_tempfilep
) char name
[] = "/tmp/test-search_and_fopen.XXXXXX";
596 _cleanup_fclose_
FILE *f
= NULL
;
597 _cleanup_free_
char *p
= NULL
, *bn
= NULL
;
598 _cleanup_close_
int fd
= -EBADF
;
602 fd
= mkostemp_safe(name
);
606 ASSERT_OK(path_extract_filename(name
, &bn
));
607 ASSERT_OK(search_and_fopen_nulstr(bn
, "re", NULL
, dirs
, &f
, &p
));
608 assert_se(e
= path_startswith(p
, "/tmp/"));
613 ASSERT_OK(search_and_fopen_nulstr(name
, "re", NULL
, dirs
, &f
, &p
));
614 assert_se(path_equal(name
, p
));
618 ASSERT_ERROR(search_and_fopen_nulstr("/a/file/which/does/not/exist/i/guess", "re", NULL
, dirs
, &f
, &p
), ENOENT
);
619 ASSERT_ERROR(search_and_fopen_nulstr("afilewhichdoesnotexistiguess", "re", NULL
, dirs
, &f
, &p
), ENOENT
);
624 ASSERT_ERROR(search_and_fopen_nulstr(bn
, "re", NULL
, dirs
, &f
, &p
), ENOENT
);
627 TEST(writing_tmpfile
) {
628 _cleanup_(unlink_tempfilep
) char name
[] = "/tmp/test-systemd_writing_tmpfile.XXXXXX";
629 _cleanup_free_
char *contents
= NULL
;
631 _cleanup_close_
int fd
= -EBADF
;
634 struct iovec iov
[] = {
635 IOVEC_MAKE_STRING("abc\n"),
636 IOVEC_MAKE_STRING(ALPHANUMERICAL
"\n"),
637 IOVEC_MAKE_STRING(""),
640 fd
= mkostemp_safe(name
);
641 printf("tmpfile: %s", name
);
643 r
= writev(fd
, iov
, 3);
646 r
= read_full_file(name
, &contents
, &size
);
648 printf("contents: %s", contents
);
649 ASSERT_STREQ(contents
, "abc\n" ALPHANUMERICAL
"\n");
653 char *ret
= NULL
, *p
;
655 assert_se(tempfn_xxxxxx("/foo/bar/waldo", NULL
, &ret
) >= 0);
656 ASSERT_STREQ(ret
, "/foo/bar/.#waldoXXXXXX");
659 assert_se(tempfn_xxxxxx("/foo/bar/waldo", "[miau]", &ret
) >= 0);
660 ASSERT_STREQ(ret
, "/foo/bar/.#[miau]waldoXXXXXX");
663 assert_se(tempfn_random("/foo/bar/waldo", NULL
, &ret
) >= 0);
664 assert_se(p
= startswith(ret
, "/foo/bar/.#waldo"));
665 assert_se(strlen(p
) == 16);
666 assert_se(in_charset(p
, "0123456789abcdef"));
669 assert_se(tempfn_random("/foo/bar/waldo", "[wuff]", &ret
) >= 0);
670 assert_se(p
= startswith(ret
, "/foo/bar/.#[wuff]waldo"));
671 assert_se(strlen(p
) == 16);
672 assert_se(in_charset(p
, "0123456789abcdef"));
675 assert_se(tempfn_random_child("/foo/bar/waldo", NULL
, &ret
) >= 0);
676 assert_se(p
= startswith(ret
, "/foo/bar/waldo/.#"));
677 assert_se(strlen(p
) == 16);
678 assert_se(in_charset(p
, "0123456789abcdef"));
681 assert_se(tempfn_random_child("/foo/bar/waldo", "[kikiriki]", &ret
) >= 0);
682 assert_se(p
= startswith(ret
, "/foo/bar/waldo/.#[kikiriki]"));
683 assert_se(strlen(p
) == 16);
684 assert_se(in_charset(p
, "0123456789abcdef"));
688 static const char chars
[] =
691 DISABLE_WARNING_TYPE_LIMITS
;
694 _cleanup_fclose_
FILE *f
= NULL
;
697 assert_se(f
= fmemopen_unlocked((void*) chars
, sizeof(chars
), "r"));
699 for (size_t i
= 0; i
< sizeof(chars
); i
++) {
700 assert_se(safe_fgetc(f
, &c
) == 1);
701 assert_se(c
== chars
[i
]);
703 if (ungetc(c
, f
) == EOF
) {
704 /* EOF is -1, and hence we can't push value 255 in this way – if char is signed */
705 assert_se(c
== (char) EOF
);
706 assert_se(CHAR_MIN
== -128); /* verify that char is signed on this platform */
708 assert_se(safe_fgetc(f
, &c
) == 1);
709 assert_se(c
== chars
[i
]);
712 /* But it works when we push it properly cast */
713 assert_se(ungetc((unsigned char) c
, f
) != EOF
);
714 assert_se(safe_fgetc(f
, &c
) == 1);
715 assert_se(c
== chars
[i
]);
718 assert_se(safe_fgetc(f
, &c
) == 0);
723 static const char buffer
[] =
725 "루Non-ascii chars: ąę„”\n"
728 "now the same with a NUL\n\0"
730 "and even more\r\n\0"
731 "and yet even more\n\r\0"
732 "With newlines, and a NUL byte\0"
736 "and a very long line that is supposed to be truncated, because it is so long\n";
738 static void test_read_line_one_file(FILE *f
) {
739 _cleanup_free_
char *line
= NULL
;
741 assert_se(read_line(f
, SIZE_MAX
, &line
) == 15 && streq(line
, "Some test data"));
744 assert_se(read_line(f
, SIZE_MAX
, &line
) > 0 && streq(line
, "루Non-ascii chars: ąę„”"));
747 assert_se(read_line(f
, SIZE_MAX
, &line
) == 13 && streq(line
, "terminators"));
750 assert_se(read_line(f
, SIZE_MAX
, &line
) == 15 && streq(line
, "and even more"));
753 assert_se(read_line(f
, SIZE_MAX
, &line
) == 25 && streq(line
, "now the same with a NUL"));
756 assert_se(read_line(f
, SIZE_MAX
, &line
) == 10 && streq(line
, "and more"));
759 assert_se(read_line(f
, SIZE_MAX
, &line
) == 16 && streq(line
, "and even more"));
762 assert_se(read_line(f
, SIZE_MAX
, &line
) == 20 && streq(line
, "and yet even more"));
765 assert_se(read_line(f
, 1024, &line
) == 30 && streq(line
, "With newlines, and a NUL byte"));
768 assert_se(read_line(f
, 1024, &line
) == 1 && streq(line
, ""));
771 assert_se(read_line(f
, 1024, &line
) == 14 && streq(line
, "an empty line"));
774 assert_se(read_line(f
, SIZE_MAX
, NULL
) == 16);
776 assert_se(read_line(f
, 16, &line
) == -ENOBUFS
);
779 /* read_line() stopped when it hit the limit, that means when we continue reading we'll read at the first
780 * character after the previous limit. Let's make use of that to continue our test. */
781 assert_se(read_line(f
, 1024, &line
) == 62 && streq(line
, "line that is supposed to be truncated, because it is so long"));
784 assert_se(read_line(f
, 1024, &line
) == 0 && streq(line
, ""));
788 _cleanup_fclose_
FILE *f
= NULL
;
790 assert_se(f
= fmemopen_unlocked((void*) buffer
, sizeof(buffer
), "r"));
791 test_read_line_one_file(f
);
795 _cleanup_(unlink_tempfilep
) char name
[] = "/tmp/test-fileio.XXXXXX";
797 _cleanup_fclose_
FILE *f
= NULL
;
799 fd
= mkostemp_safe(name
);
801 assert_se((size_t) write(fd
, buffer
, sizeof(buffer
)) == sizeof(buffer
));
803 assert_se(lseek(fd
, 0, SEEK_SET
) == 0);
804 assert_se(f
= fdopen(fd
, "r"));
806 test_read_line_one_file(f
);
810 _cleanup_fclose_
FILE *f
= NULL
;
811 _cleanup_free_
char *line
= NULL
;
814 f
= fopen("/proc/uptime", "re");
815 if (!f
&& IN_SET(errno
, ENOENT
, EPERM
))
819 r
= read_line(f
, LINE_MAX
, &line
);
822 assert_se(line
&& isempty(line
));
824 assert_se((size_t) r
== strlen(line
) + 1);
825 assert_se(read_line(f
, LINE_MAX
, NULL
) == 0);
829 static const struct {
833 /* Each of these will be followed by EOF and should generate the one same single string */
848 FOREACH_ELEMENT(ending
, eof_endings
) {
849 _cleanup_fclose_
FILE *f
= NULL
;
850 _cleanup_free_
char *s
= NULL
;
852 assert_se(f
= fmemopen_unlocked((void*) ending
->string
, ending
->length
, "r"));
854 r
= read_line(f
, SIZE_MAX
, &s
);
855 assert_se((size_t) r
== ending
->length
);
856 ASSERT_STREQ(s
, "foo");
858 assert_se(read_line(f
, SIZE_MAX
, NULL
) == 0); /* Ensure we hit EOF */
862 TEST(read_nul_string
) {
863 static const char test
[] = "string nr. 1\0"
865 "\377empty string follows\0"
867 "final string\n is empty\0"
870 _cleanup_fclose_
FILE *f
= NULL
;
871 _cleanup_free_
char *s
= NULL
;
873 assert_se(f
= fmemopen_unlocked((void*) test
, sizeof(test
)-1, "r"));
875 assert_se(read_nul_string(f
, LONG_LINE_MAX
, &s
) == 13 && streq_ptr(s
, "string nr. 1"));
878 assert_se(read_nul_string(f
, LONG_LINE_MAX
, &s
) == 14 && streq_ptr(s
, "string nr. 2\n"));
881 assert_se(read_nul_string(f
, LONG_LINE_MAX
, &s
) == 22 && streq_ptr(s
, "\377empty string follows"));
884 assert_se(read_nul_string(f
, LONG_LINE_MAX
, &s
) == 1 && streq_ptr(s
, ""));
887 assert_se(read_nul_string(f
, LONG_LINE_MAX
, &s
) == 23 && streq_ptr(s
, "final string\n is empty"));
890 assert_se(read_nul_string(f
, LONG_LINE_MAX
, &s
) == 1 && streq_ptr(s
, ""));
893 assert_se(read_nul_string(f
, LONG_LINE_MAX
, &s
) == 0 && streq_ptr(s
, ""));
896 TEST(read_full_file_socket
) {
897 _cleanup_(rm_rf_physical_and_freep
) char *z
= NULL
;
898 _cleanup_close_
int listener
= -EBADF
;
899 _cleanup_free_
char *data
= NULL
, *clientname
= NULL
;
900 union sockaddr_union sa
;
906 listener
= socket(AF_UNIX
, SOCK_STREAM
|SOCK_CLOEXEC
, 0);
907 assert_se(listener
>= 0);
909 assert_se(mkdtemp_malloc(NULL
, &z
) >= 0);
910 j
= strjoina(z
, "/socket");
912 assert_se(sockaddr_un_set_path(&sa
.un
, j
) >= 0);
914 assert_se(bind(listener
, &sa
.sa
, sockaddr_un_len(&sa
.un
)) >= 0);
915 assert_se(listen(listener
, 1) >= 0);
917 /* Make sure the socket doesn't fit into a struct sockaddr_un, but we can still access it */
918 jj
= strjoina(z
, "/a_very_long_patha_very_long_patha_very_long_patha_very_long_patha_very_long_patha_very_long_patha_very_long_patha_very_long_path");
919 assert_se(strlen(jj
) > sizeof_field(struct sockaddr_un
, sun_path
));
920 assert_se(rename(j
, jj
) >= 0);
922 /* Bind the *client* socket to some randomized name, to verify that this works correctly. */
923 assert_se(asprintf(&clientname
, "@%" PRIx64
"/test-bindname", random_u64()) >= 0);
925 r
= safe_fork("(server)", FORK_DEATHSIG_SIGTERM
|FORK_LOG
, &pid
);
928 union sockaddr_union peer
= {};
929 socklen_t peerlen
= sizeof(peer
);
930 _cleanup_close_
int rfd
= -EBADF
;
933 rfd
= accept4(listener
, NULL
, NULL
, SOCK_CLOEXEC
);
936 assert_se(getpeername(rfd
, &peer
.sa
, &peerlen
) >= 0);
938 assert_se(peer
.un
.sun_family
== AF_UNIX
);
939 assert_se(peerlen
> offsetof(struct sockaddr_un
, sun_path
));
940 assert_se(peer
.un
.sun_path
[0] == 0);
941 ASSERT_STREQ(peer
.un
.sun_path
+ 1, clientname
+ 1);
943 #define TEST_STR "This is a test\nreally."
945 assert_se(write(rfd
, TEST_STR
, strlen(TEST_STR
)) == strlen(TEST_STR
));
949 assert_se(read_full_file_full(AT_FDCWD
, jj
, UINT64_MAX
, SIZE_MAX
, 0, NULL
, &data
, &size
) == -ENXIO
);
950 assert_se(read_full_file_full(AT_FDCWD
, jj
, UINT64_MAX
, SIZE_MAX
, READ_FULL_FILE_CONNECT_SOCKET
, clientname
, &data
, &size
) >= 0);
951 assert_se(size
== strlen(TEST_STR
));
952 ASSERT_STREQ(data
, TEST_STR
);
954 assert_se(wait_for_terminate_and_check("(server)", pid
, WAIT_LOG
) >= 0);
958 TEST(read_full_file_offset_size
) {
959 _cleanup_fclose_
FILE *f
= NULL
;
960 _cleanup_(unlink_and_freep
) char *fn
= NULL
;
961 _cleanup_free_
char *rbuf
= NULL
;
965 random_bytes(buf
, sizeof(buf
));
967 assert_se(tempfn_random_child(NULL
, NULL
, &fn
) >= 0);
968 assert_se(f
= fopen(fn
, "we"));
969 assert_se(fwrite(buf
, 1, sizeof(buf
), f
) == sizeof(buf
));
970 assert_se(fflush_and_check(f
) >= 0);
972 assert_se(read_full_file_full(AT_FDCWD
, fn
, UINT64_MAX
, SIZE_MAX
, 0, NULL
, &rbuf
, &rbuf_size
) >= 0);
973 assert_se(rbuf_size
== sizeof(buf
));
974 assert_se(memcmp(buf
, rbuf
, rbuf_size
) == 0);
977 assert_se(read_full_file_full(AT_FDCWD
, fn
, UINT64_MAX
, 128, 0, NULL
, &rbuf
, &rbuf_size
) >= 0);
978 assert_se(rbuf_size
== 128);
979 assert_se(memcmp(buf
, rbuf
, rbuf_size
) == 0);
982 assert_se(read_full_file_full(AT_FDCWD
, fn
, UINT64_MAX
, 128, READ_FULL_FILE_FAIL_WHEN_LARGER
, NULL
, &rbuf
, &rbuf_size
) == -E2BIG
);
983 assert_se(read_full_file_full(AT_FDCWD
, fn
, UINT64_MAX
, sizeof(buf
)-1, READ_FULL_FILE_FAIL_WHEN_LARGER
, NULL
, &rbuf
, &rbuf_size
) == -E2BIG
);
984 assert_se(read_full_file_full(AT_FDCWD
, fn
, UINT64_MAX
, sizeof(buf
), READ_FULL_FILE_FAIL_WHEN_LARGER
, NULL
, &rbuf
, &rbuf_size
) >= 0);
985 assert_se(rbuf_size
== sizeof(buf
));
986 assert_se(memcmp(buf
, rbuf
, rbuf_size
) == 0);
989 assert_se(read_full_file_full(AT_FDCWD
, fn
, 47, 128, READ_FULL_FILE_FAIL_WHEN_LARGER
, NULL
, &rbuf
, &rbuf_size
) == -E2BIG
);
990 assert_se(read_full_file_full(AT_FDCWD
, fn
, 47, sizeof(buf
)-47-1, READ_FULL_FILE_FAIL_WHEN_LARGER
, NULL
, &rbuf
, &rbuf_size
) == -E2BIG
);
991 assert_se(read_full_file_full(AT_FDCWD
, fn
, 47, sizeof(buf
)-47, READ_FULL_FILE_FAIL_WHEN_LARGER
, NULL
, &rbuf
, &rbuf_size
) >= 0);
992 assert_se(rbuf_size
== sizeof(buf
)-47);
993 assert_se(memcmp(buf
+47, rbuf
, rbuf_size
) == 0);
996 assert_se(read_full_file_full(AT_FDCWD
, fn
, UINT64_MAX
, sizeof(buf
)+1, READ_FULL_FILE_FAIL_WHEN_LARGER
, NULL
, &rbuf
, &rbuf_size
) >= 0);
997 assert_se(rbuf_size
== sizeof(buf
));
998 assert_se(memcmp(buf
, rbuf
, rbuf_size
) == 0);
1001 assert_se(read_full_file_full(AT_FDCWD
, fn
, 1234, SIZE_MAX
, 0, NULL
, &rbuf
, &rbuf_size
) >= 0);
1002 assert_se(rbuf_size
== sizeof(buf
) - 1234);
1003 assert_se(memcmp(buf
+ 1234, rbuf
, rbuf_size
) == 0);
1006 assert_se(read_full_file_full(AT_FDCWD
, fn
, 2345, 777, 0, NULL
, &rbuf
, &rbuf_size
) >= 0);
1007 assert_se(rbuf_size
== 777);
1008 assert_se(memcmp(buf
+ 2345, rbuf
, rbuf_size
) == 0);
1011 assert_se(read_full_file_full(AT_FDCWD
, fn
, 4700, 20, 0, NULL
, &rbuf
, &rbuf_size
) >= 0);
1012 assert_se(rbuf_size
== 11);
1013 assert_se(memcmp(buf
+ 4700, rbuf
, rbuf_size
) == 0);
1016 assert_se(read_full_file_full(AT_FDCWD
, fn
, 10000, 99, 0, NULL
, &rbuf
, &rbuf_size
) >= 0);
1017 assert_se(rbuf_size
== 0);
1021 static void test_read_virtual_file_one(size_t max_size
) {
1024 log_info("/* %s (max_size=%zu) */", __func__
, max_size
);
1026 FOREACH_STRING(filename
,
1028 "/etc/nsswitch.conf",
1029 "/sys/kernel/uevent_seqnum",
1033 "/proc/self/pagemap") {
1035 _cleanup_free_
char *buf
= NULL
;
1038 r
= read_virtual_file(filename
, max_size
, &buf
, &size
);
1040 log_info_errno(r
, "read_virtual_file(\"%s\", %zu): %m", filename
, max_size
);
1041 assert_se(ERRNO_IS_PRIVILEGE(r
) || /* /proc/kcore is not accessible to unpriv */
1043 -ENOENT
, /* Some of the files might be absent */
1044 -EINVAL
, /* too small reads from /proc/self/pagemap trigger EINVAL */
1045 -EFBIG
, /* /proc/kcore and /proc/self/pagemap should be too large */
1046 -EBADF
)); /* /proc/kcore is masked when we are running in docker. */
1048 log_info("read_virtual_file(\"%s\", %zu): %s (%zu bytes)", filename
, max_size
, r
? "non-truncated" : "truncated", size
);
1052 TEST(read_virtual_file
) {
1053 test_read_virtual_file_one(0);
1054 test_read_virtual_file_one(1);
1055 test_read_virtual_file_one(2);
1056 test_read_virtual_file_one(20);
1057 test_read_virtual_file_one(4096);
1058 test_read_virtual_file_one(4097);
1059 test_read_virtual_file_one(SIZE_MAX
);
1062 TEST(fdopen_independent
) {
1063 #define TEST_TEXT "this is some random test text we are going to write to a memfd"
1064 _cleanup_close_
int fd
= -EBADF
;
1065 _cleanup_fclose_
FILE *f
= NULL
;
1066 char buf
[STRLEN(TEST_TEXT
) + 1];
1068 fd
= memfd_new("fdopen_independent");
1070 assert_se(ERRNO_IS_NOT_SUPPORTED(fd
));
1074 assert_se(write(fd
, TEST_TEXT
, strlen(TEST_TEXT
)) == strlen(TEST_TEXT
));
1075 /* we'll leave the read offset at the end of the memfd, the fdopen_independent() descriptors should
1076 * start at the beginning anyway */
1078 assert_se(fdopen_independent(fd
, "re", &f
) >= 0);
1080 assert_se(fread(buf
, 1, sizeof(buf
), f
) == strlen(TEST_TEXT
));
1081 ASSERT_STREQ(buf
, TEST_TEXT
);
1082 assert_se((fcntl(fileno(f
), F_GETFL
) & O_ACCMODE_STRICT
) == O_RDONLY
);
1083 assert_se(FLAGS_SET(fcntl(fileno(f
), F_GETFD
), FD_CLOEXEC
));
1086 assert_se(fdopen_independent(fd
, "r", &f
) >= 0);
1088 assert_se(fread(buf
, 1, sizeof(buf
), f
) == strlen(TEST_TEXT
));
1089 ASSERT_STREQ(buf
, TEST_TEXT
);
1090 assert_se((fcntl(fileno(f
), F_GETFL
) & O_ACCMODE_STRICT
) == O_RDONLY
);
1091 assert_se(!FLAGS_SET(fcntl(fileno(f
), F_GETFD
), FD_CLOEXEC
));
1094 assert_se(fdopen_independent(fd
, "r+e", &f
) >= 0);
1096 assert_se(fread(buf
, 1, sizeof(buf
), f
) == strlen(TEST_TEXT
));
1097 ASSERT_STREQ(buf
, TEST_TEXT
);
1098 assert_se((fcntl(fileno(f
), F_GETFL
) & O_ACCMODE_STRICT
) == O_RDWR
);
1099 assert_se(FLAGS_SET(fcntl(fileno(f
), F_GETFD
), FD_CLOEXEC
));
1103 DEFINE_TEST_MAIN(LOG_DEBUG
);