]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/test/test-fileio.c
Merge pull request #27035 from DaanDeMeyer/ukify-align
[thirdparty/systemd.git] / src / test / test-fileio.c
1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
2
3 #include <fcntl.h>
4 #include <limits.h>
5 #include <stdio.h>
6 #include <unistd.h>
7
8 #include "alloc-util.h"
9 #include "ctype.h"
10 #include "env-file.h"
11 #include "env-util.h"
12 #include "errno-util.h"
13 #include "fd-util.h"
14 #include "fileio.h"
15 #include "fs-util.h"
16 #include "io-util.h"
17 #include "memfd-util.h"
18 #include "parse-util.h"
19 #include "path-util.h"
20 #include "process-util.h"
21 #include "random-util.h"
22 #include "rm-rf.h"
23 #include "socket-util.h"
24 #include "string-util.h"
25 #include "strv.h"
26 #include "tests.h"
27 #include "tmpfile-util.h"
28
29 TEST(parse_env_file) {
30 _cleanup_(unlink_tempfilep) char
31 t[] = "/tmp/test-fileio-in-XXXXXX",
32 p[] = "/tmp/test-fileio-out-XXXXXX";
33 FILE *f;
34 _cleanup_free_ char *one = NULL, *two = NULL, *three = NULL, *four = NULL, *five = NULL,
35 *six = NULL, *seven = NULL, *eight = NULL, *nine = NULL, *ten = NULL,
36 *eleven = NULL, *twelve = NULL, *thirteen = NULL;
37 _cleanup_strv_free_ char **a = NULL, **b = NULL;
38 unsigned k;
39 int r;
40
41 assert_se(fmkostemp_safe(t, "w", &f) == 0);
42 fputs("one=BAR \n"
43 "# comment\n"
44 " # comment \n"
45 " ; comment \n"
46 " two = bar \n"
47 "invalid line\n"
48 "invalid line #comment\n"
49 "three = \"333\n"
50 "xxxx\"\n"
51 "four = \'44\\\"44\'\n"
52 "five = \"55\\\"55\" \"FIVE\" cinco \n"
53 "six = seis sechs\\\n"
54 " sis\n"
55 "seven=\"sevenval\" #nocomment\n"
56 "eight=eightval #nocomment\n"
57 "export nine=nineval\n"
58 "ten=ignored\n"
59 "ten=ignored\n"
60 "ten=\n"
61 "eleven=\\value\n"
62 "twelve=\"\\value\"\n"
63 "thirteen='\\value'", f);
64
65 fflush(f);
66 fclose(f);
67
68 r = load_env_file(NULL, t, &a);
69 assert_se(r >= 0);
70
71 STRV_FOREACH(i, a)
72 log_info("Got: <%s>", *i);
73
74 assert_se(streq_ptr(a[0], "one=BAR"));
75 assert_se(streq_ptr(a[1], "two=bar"));
76 assert_se(streq_ptr(a[2], "three=333\nxxxx"));
77 assert_se(streq_ptr(a[3], "four=44\\\"44"));
78 assert_se(streq_ptr(a[4], "five=55\"55FIVEcinco"));
79 assert_se(streq_ptr(a[5], "six=seis sechs sis"));
80 assert_se(streq_ptr(a[6], "seven=sevenval#nocomment"));
81 assert_se(streq_ptr(a[7], "eight=eightval #nocomment"));
82 assert_se(streq_ptr(a[8], "export nine=nineval"));
83 assert_se(streq_ptr(a[9], "ten="));
84 assert_se(streq_ptr(a[10], "eleven=value"));
85 assert_se(streq_ptr(a[11], "twelve=\\value"));
86 assert_se(streq_ptr(a[12], "thirteen=\\value"));
87 assert_se(a[13] == NULL);
88
89 strv_env_clean(a);
90
91 k = 0;
92 STRV_FOREACH(i, b) {
93 log_info("Got2: <%s>", *i);
94 assert_se(streq(*i, a[k++]));
95 }
96
97 r = parse_env_file(
98 NULL, t,
99 "one", &one,
100 "two", &two,
101 "three", &three,
102 "four", &four,
103 "five", &five,
104 "six", &six,
105 "seven", &seven,
106 "eight", &eight,
107 "export nine", &nine,
108 "ten", &ten,
109 "eleven", &eleven,
110 "twelve", &twelve,
111 "thirteen", &thirteen);
112 assert_se(r == 0);
113
114 log_info("one=[%s]", strna(one));
115 log_info("two=[%s]", strna(two));
116 log_info("three=[%s]", strna(three));
117 log_info("four=[%s]", strna(four));
118 log_info("five=[%s]", strna(five));
119 log_info("six=[%s]", strna(six));
120 log_info("seven=[%s]", strna(seven));
121 log_info("eight=[%s]", strna(eight));
122 log_info("export nine=[%s]", strna(nine));
123 log_info("ten=[%s]", strna(nine));
124 log_info("eleven=[%s]", strna(eleven));
125 log_info("twelve=[%s]", strna(twelve));
126 log_info("thirteen=[%s]", strna(thirteen));
127
128 assert_se(streq(one, "BAR"));
129 assert_se(streq(two, "bar"));
130 assert_se(streq(three, "333\nxxxx"));
131 assert_se(streq(four, "44\\\"44"));
132 assert_se(streq(five, "55\"55FIVEcinco"));
133 assert_se(streq(six, "seis sechs sis"));
134 assert_se(streq(seven, "sevenval#nocomment"));
135 assert_se(streq(eight, "eightval #nocomment"));
136 assert_se(streq(nine, "nineval"));
137 assert_se(ten == NULL);
138 assert_se(streq(eleven, "value"));
139 assert_se(streq(twelve, "\\value"));
140 assert_se(streq(thirteen, "\\value"));
141
142 {
143 /* prepare a temporary file to write the environment to */
144 _cleanup_close_ int fd = mkostemp_safe(p);
145 assert_se(fd >= 0);
146 }
147
148 r = write_env_file(p, a);
149 assert_se(r >= 0);
150
151 r = load_env_file(NULL, p, &b);
152 assert_se(r >= 0);
153 }
154
155 static void test_one_shell_var(const char *file, const char *variable, const char *value) {
156 _cleanup_free_ char *cmd = NULL, *from_shell = NULL;
157 _cleanup_pclose_ FILE *f = NULL;
158 size_t sz;
159
160 assert_se(cmd = strjoin(". ", file, " && /bin/echo -n \"$", variable, "\""));
161 assert_se(f = popen(cmd, "re"));
162 assert_se(read_full_stream(f, &from_shell, &sz) >= 0);
163 assert_se(sz == strlen(value));
164 assert_se(streq(from_shell, value));
165 }
166
167 TEST(parse_multiline_env_file) {
168 _cleanup_(unlink_tempfilep) char
169 t[] = "/tmp/test-fileio-in-XXXXXX",
170 p[] = "/tmp/test-fileio-out-XXXXXX";
171 FILE *f;
172 _cleanup_strv_free_ char **a = NULL, **b = NULL;
173 int r;
174
175 assert_se(fmkostemp_safe(t, "w", &f) == 0);
176 fputs("one=BAR\\\n"
177 "\\ \\ \\ \\ VAR\\\n"
178 "\\\tGAR\n"
179 "#comment\n"
180 "two=\"bar\\\n"
181 " var\\\n"
182 "\tgar\"\n"
183 "#comment\n"
184 "tri=\"bar \\\n"
185 " var \\\n"
186 "\tgar \"\n", f);
187
188 assert_se(fflush_and_check(f) >= 0);
189 fclose(f);
190
191 test_one_shell_var(t, "one", "BAR VAR\tGAR");
192 test_one_shell_var(t, "two", "bar var\tgar");
193 test_one_shell_var(t, "tri", "bar var \tgar ");
194
195 r = load_env_file(NULL, t, &a);
196 assert_se(r >= 0);
197
198 STRV_FOREACH(i, a)
199 log_info("Got: <%s>", *i);
200
201 assert_se(streq_ptr(a[0], "one=BAR VAR\tGAR"));
202 assert_se(streq_ptr(a[1], "two=bar var\tgar"));
203 assert_se(streq_ptr(a[2], "tri=bar var \tgar "));
204 assert_se(a[3] == NULL);
205
206 {
207 _cleanup_close_ int fd = mkostemp_safe(p);
208 assert_se(fd >= 0);
209 }
210
211 r = write_env_file(p, a);
212 assert_se(r >= 0);
213
214 r = load_env_file(NULL, p, &b);
215 assert_se(r >= 0);
216 }
217
218 TEST(merge_env_file) {
219 _cleanup_(unlink_tempfilep) char t[] = "/tmp/test-fileio-XXXXXX";
220 _cleanup_fclose_ FILE *f = NULL;
221 _cleanup_strv_free_ char **a = NULL;
222 int r;
223
224 assert_se(fmkostemp_safe(t, "w", &f) == 0);
225 log_info("/* %s (%s) */", __func__, t);
226
227 r = write_string_stream(f,
228 "one=1 \n"
229 "twelve=${one}2\n"
230 "twentyone=2${one}\n"
231 "one=2\n"
232 "twentytwo=2${one}\n"
233 "xxx_minus_three=$xxx - 3\n"
234 "xxx=0x$one$one$one\n"
235 "yyy=${one:-fallback}\n"
236 "zzz=${one:+replacement}\n"
237 "zzzz=${foobar:-${nothing}}\n"
238 "zzzzz=${nothing:+${nothing}}\n"
239 , WRITE_STRING_FILE_AVOID_NEWLINE);
240 assert_se(r >= 0);
241
242 r = merge_env_file(&a, NULL, t);
243 assert_se(r >= 0);
244 strv_sort(a);
245
246 STRV_FOREACH(i, a)
247 log_info("Got: <%s>", *i);
248
249 assert_se(streq(a[0], "one=2"));
250 assert_se(streq(a[1], "twelve=12"));
251 assert_se(streq(a[2], "twentyone=21"));
252 assert_se(streq(a[3], "twentytwo=22"));
253 assert_se(streq(a[4], "xxx=0x222"));
254 assert_se(streq(a[5], "xxx_minus_three= - 3"));
255 assert_se(streq(a[6], "yyy=2"));
256 assert_se(streq(a[7], "zzz=replacement"));
257 assert_se(streq(a[8], "zzzz="));
258 assert_se(streq(a[9], "zzzzz="));
259 assert_se(a[10] == NULL);
260
261 r = merge_env_file(&a, NULL, t);
262 assert_se(r >= 0);
263 strv_sort(a);
264
265 STRV_FOREACH(i, a)
266 log_info("Got2: <%s>", *i);
267
268 assert_se(streq(a[0], "one=2"));
269 assert_se(streq(a[1], "twelve=12"));
270 assert_se(streq(a[2], "twentyone=21"));
271 assert_se(streq(a[3], "twentytwo=22"));
272 assert_se(streq(a[4], "xxx=0x222"));
273 assert_se(streq(a[5], "xxx_minus_three=0x222 - 3"));
274 assert_se(streq(a[6], "yyy=2"));
275 assert_se(streq(a[7], "zzz=replacement"));
276 assert_se(streq(a[8], "zzzz="));
277 assert_se(streq(a[9], "zzzzz="));
278 assert_se(a[10] == NULL);
279 }
280
281 TEST(merge_env_file_invalid) {
282 _cleanup_(unlink_tempfilep) char t[] = "/tmp/test-fileio-XXXXXX";
283 _cleanup_fclose_ FILE *f = NULL;
284 _cleanup_strv_free_ char **a = NULL;
285 int r;
286
287 assert_se(fmkostemp_safe(t, "w", &f) == 0);
288 log_info("/* %s (%s) */", __func__, t);
289
290 r = write_string_stream(f,
291 "unset one \n"
292 "unset one= \n"
293 "unset one=1 \n"
294 "one \n"
295 "one = \n"
296 "one two =\n"
297 "\x20two=\n"
298 "#comment=comment\n"
299 ";comment2=comment2\n"
300 "#\n"
301 "\n\n" /* empty line */
302 , WRITE_STRING_FILE_AVOID_NEWLINE);
303 assert_se(r >= 0);
304
305 r = merge_env_file(&a, NULL, t);
306 assert_se(r >= 0);
307
308 STRV_FOREACH(i, a)
309 log_info("Got: <%s>", *i);
310
311 assert_se(strv_isempty(a));
312 }
313
314 TEST(executable_is_script) {
315 _cleanup_(unlink_tempfilep) char t[] = "/tmp/test-fileio-XXXXXX";
316 _cleanup_fclose_ FILE *f = NULL;
317 char *command;
318 int r;
319
320 assert_se(fmkostemp_safe(t, "w", &f) == 0);
321 fputs("#! /bin/script -a -b \ngoo goo", f);
322 fflush(f);
323
324 r = executable_is_script(t, &command);
325 assert_se(r > 0);
326 assert_se(streq(command, "/bin/script"));
327 free(command);
328
329 r = executable_is_script("/bin/sh", &command);
330 assert_se(r == 0);
331
332 r = executable_is_script("/usr/bin/yum", &command);
333 assert_se(r > 0 || r == -ENOENT);
334 if (r > 0) {
335 assert_se(startswith(command, "/"));
336 free(command);
337 }
338 }
339
340 TEST(status_field) {
341 _cleanup_free_ char *p = NULL, *s = NULL, *z = NULL;
342 unsigned long long total = 0, buffers = 0;
343 int r;
344
345 r = get_proc_field("/proc/meminfo", "MemTotal", WHITESPACE, &p);
346 if (r != -ENOENT) {
347 assert_se(r == 0);
348 puts(p);
349 assert_se(safe_atollu(p, &total) == 0);
350 }
351
352 r = get_proc_field("/proc/meminfo", "Buffers", WHITESPACE, &s);
353 if (r != -ENOENT) {
354 assert_se(r == 0);
355 puts(s);
356 assert_se(safe_atollu(s, &buffers) == 0);
357 }
358
359 if (p)
360 assert_se(buffers < total);
361
362 /* Seccomp should be a good test for field full of zeros. */
363 r = get_proc_field("/proc/meminfo", "Seccomp", WHITESPACE, &z);
364 if (r != -ENOENT) {
365 assert_se(r == 0);
366 puts(z);
367 assert_se(safe_atollu(z, &buffers) == 0);
368 }
369 }
370
371 TEST(capeff) {
372 for (int pid = 0; pid < 2; pid++) {
373 _cleanup_free_ char *capeff = NULL;
374 int r, p;
375
376 r = get_process_capeff(0, &capeff);
377 log_info("capeff: '%s' (r=%d)", capeff, r);
378
379 if (IN_SET(r, -ENOENT, -EPERM))
380 return;
381
382 assert_se(r == 0);
383 assert_se(*capeff);
384 p = capeff[strspn(capeff, HEXDIGITS)];
385 assert_se(!p || isspace(p));
386 }
387 }
388
389 TEST(read_one_line_file) {
390 _cleanup_(unlink_tempfilep) char fn[] = "/tmp/test-fileio-1lf-XXXXXX";
391 int fd;
392 _cleanup_fclose_ FILE *f = NULL;
393 _cleanup_free_ char *buf, *buf2, *buf3, *buf4, *buf5;
394
395 fd = mkostemp_safe(fn);
396 assert_se(fd >= 0);
397
398 f = fdopen(fd, "we");
399 assert_se(f);
400
401 assert_se(read_one_line_file(fn, &buf) == 0);
402 assert_se(streq_ptr(buf, ""));
403 assert_se(read_one_line_file(fn, &buf2) == 0);
404 assert_se(streq_ptr(buf2, ""));
405
406 assert_se(write_string_stream(f, "x", WRITE_STRING_FILE_AVOID_NEWLINE) >= 0);
407 fflush(f);
408
409 assert_se(read_one_line_file(fn, &buf3) == 1);
410 assert_se(streq_ptr(buf3, "x"));
411
412 assert_se(write_string_stream(f, "\n", WRITE_STRING_FILE_AVOID_NEWLINE) >= 0);
413 fflush(f);
414
415 assert_se(read_one_line_file(fn, &buf4) == 2);
416 assert_se(streq_ptr(buf4, "x"));
417
418 assert_se(write_string_stream(f, "\n", WRITE_STRING_FILE_AVOID_NEWLINE) >= 0);
419 fflush(f);
420
421 assert_se(read_one_line_file(fn, &buf5) == 2);
422 assert_se(streq_ptr(buf5, "x"));
423 }
424
425 TEST(write_string_stream) {
426 _cleanup_(unlink_tempfilep) char fn[] = "/tmp/test-write_string_stream-XXXXXX";
427 _cleanup_fclose_ FILE *f = NULL;
428 int fd;
429 char buf[64];
430
431 fd = mkostemp_safe(fn);
432 assert_se(fd >= 0);
433
434 f = fdopen(fd, "r");
435 assert_se(f);
436 assert_se(write_string_stream(f, "boohoo", 0) < 0);
437 f = safe_fclose(f);
438
439 f = fopen(fn, "r+");
440 assert_se(f);
441
442 assert_se(write_string_stream(f, "boohoo", 0) == 0);
443 rewind(f);
444
445 assert_se(fgets(buf, sizeof(buf), f));
446 assert_se(streq(buf, "boohoo\n"));
447 f = safe_fclose(f);
448
449 f = fopen(fn, "w+");
450 assert_se(f);
451
452 assert_se(write_string_stream(f, "boohoo", WRITE_STRING_FILE_AVOID_NEWLINE) == 0);
453 rewind(f);
454
455 assert_se(fgets(buf, sizeof(buf), f));
456 printf(">%s<", buf);
457 assert_se(streq(buf, "boohoo"));
458 }
459
460 TEST(write_string_file) {
461 _cleanup_(unlink_tempfilep) char fn[] = "/tmp/test-write_string_file-XXXXXX";
462 char buf[64] = {};
463 _cleanup_close_ int fd;
464
465 fd = mkostemp_safe(fn);
466 assert_se(fd >= 0);
467
468 assert_se(write_string_file(fn, "boohoo", WRITE_STRING_FILE_CREATE) == 0);
469
470 assert_se(read(fd, buf, sizeof(buf)) == 7);
471 assert_se(streq(buf, "boohoo\n"));
472 }
473
474 TEST(write_string_file_no_create) {
475 _cleanup_(unlink_tempfilep) char fn[] = "/tmp/test-write_string_file_no_create-XXXXXX";
476 _cleanup_close_ int fd;
477 char buf[64] = {};
478
479 fd = mkostemp_safe(fn);
480 assert_se(fd >= 0);
481
482 assert_se(write_string_file("/a/file/which/does/not/exists/i/guess", "boohoo", 0) < 0);
483 assert_se(write_string_file(fn, "boohoo", 0) == 0);
484
485 assert_se(read(fd, buf, sizeof buf) == (ssize_t) strlen("boohoo\n"));
486 assert_se(streq(buf, "boohoo\n"));
487 }
488
489 TEST(write_string_file_verify) {
490 _cleanup_free_ char *buf = NULL, *buf2 = NULL;
491 int r;
492
493 r = read_one_line_file("/proc/version", &buf);
494 if (ERRNO_IS_PRIVILEGE(r))
495 return;
496 assert_se(r >= 0);
497 assert_se(buf2 = strjoin(buf, "\n"));
498
499 r = write_string_file("/proc/version", buf, 0);
500 assert_se(IN_SET(r, -EACCES, -EIO));
501 r = write_string_file("/proc/version", buf2, 0);
502 assert_se(IN_SET(r, -EACCES, -EIO));
503
504 assert_se(write_string_file("/proc/version", buf, WRITE_STRING_FILE_VERIFY_ON_FAILURE) == 0);
505 assert_se(write_string_file("/proc/version", buf2, WRITE_STRING_FILE_VERIFY_ON_FAILURE) == 0);
506
507 r = write_string_file("/proc/version", buf, WRITE_STRING_FILE_VERIFY_ON_FAILURE|WRITE_STRING_FILE_AVOID_NEWLINE);
508 assert_se(IN_SET(r, -EACCES, -EIO));
509 assert_se(write_string_file("/proc/version", buf2, WRITE_STRING_FILE_VERIFY_ON_FAILURE|WRITE_STRING_FILE_AVOID_NEWLINE) == 0);
510 }
511
512 static void check_file_pairs_one(char **l) {
513 assert_se(l);
514 assert_se(strv_length(l) == 14);
515
516 STRV_FOREACH_PAIR(k, v, l) {
517 assert_se(STR_IN_SET(*k, "NAME", "ID", "PRETTY_NAME", "ANSI_COLOR", "HOME_URL", "SUPPORT_URL", "BUG_REPORT_URL"));
518 printf("%s=%s\n", *k, *v);
519 assert_se(!streq(*k, "NAME") || streq(*v, "Arch Linux"));
520 assert_se(!streq(*k, "ID") || streq(*v, "arch"));
521 assert_se(!streq(*k, "PRETTY_NAME") || streq(*v, "Arch Linux"));
522 assert_se(!streq(*k, "ANSI_COLOR") || streq(*v, "0;36"));
523 assert_se(!streq(*k, "HOME_URL") || streq(*v, "https://www.archlinux.org/"));
524 assert_se(!streq(*k, "SUPPORT_URL") || streq(*v, "https://bbs.archlinux.org/"));
525 assert_se(!streq(*k, "BUG_REPORT_URL") || streq(*v, "https://bugs.archlinux.org/"));
526 }
527 }
528
529 TEST(load_env_file_pairs) {
530 _cleanup_(unlink_tempfilep) char fn[] = "/tmp/test-load_env_file_pairs-XXXXXX";
531 int fd, r;
532 _cleanup_fclose_ FILE *f = NULL;
533 _cleanup_strv_free_ char **l = NULL;
534
535 fd = mkostemp_safe(fn);
536 assert_se(fd >= 0);
537
538 r = write_string_file(fn,
539 "NAME=\"Arch Linux\"\n"
540 "ID=arch\n"
541 "PRETTY_NAME=\"Arch Linux\"\n"
542 "ANSI_COLOR=\"0;36\"\n"
543 "HOME_URL=\"https://www.archlinux.org/\"\n"
544 "SUPPORT_URL=\"https://bbs.archlinux.org/\"\n"
545 "BUG_REPORT_URL=\"https://bugs.archlinux.org/\"\n",
546 WRITE_STRING_FILE_CREATE);
547 assert_se(r == 0);
548
549 r = load_env_file_pairs_fd(fd, fn, &l);
550 assert_se(r >= 0);
551 check_file_pairs_one(l);
552 l = strv_free(l);
553
554 f = fdopen(fd, "r");
555 assert_se(f);
556
557 r = load_env_file_pairs(f, fn, &l);
558 assert_se(r >= 0);
559 check_file_pairs_one(l);
560 }
561
562 TEST(search_and_fopen) {
563 static const char* const dirs[] = {
564 "/tmp/foo/bar",
565 "/tmp",
566 NULL
567 };
568 char name[] = "/tmp/test-search_and_fopen.XXXXXX";
569 _cleanup_fclose_ FILE *f = NULL;
570 _cleanup_free_ char *p = NULL;
571 _cleanup_close_ int fd = -EBADF;
572 const char *e;
573 int r;
574
575 fd = mkostemp_safe(name);
576 assert_se(fd >= 0);
577 fd = safe_close(fd);
578
579 r = search_and_fopen(basename(name), "re", NULL, (const char**) dirs, &f, &p);
580 assert_se(r >= 0);
581 assert_se(e = path_startswith(p, "/tmp/"));
582 assert_se(streq(basename(name), e));
583 f = safe_fclose(f);
584 p = mfree(p);
585
586 r = search_and_fopen(name, "re", NULL, (const char**) dirs, &f, &p);
587 assert_se(r >= 0);
588 assert_se(path_equal(name, p));
589 f = safe_fclose(f);
590 p = mfree(p);
591
592 r = search_and_fopen(basename(name), "re", "/", (const char**) dirs, &f, &p);
593 assert_se(r >= 0);
594 assert_se(e = path_startswith(p, "/tmp/"));
595 assert_se(streq(basename(name), e));
596 f = safe_fclose(f);
597 p = mfree(p);
598
599 r = search_and_fopen("/a/file/which/does/not/exist/i/guess", "re", NULL, (const char**) dirs, &f, &p);
600 assert_se(r == -ENOENT);
601 r = search_and_fopen("afilewhichdoesnotexistiguess", "re", NULL, (const char**) dirs, &f, &p);
602 assert_se(r == -ENOENT);
603
604 r = unlink(name);
605 assert_se(r == 0);
606
607 r = search_and_fopen(basename(name), "re", NULL, (const char**) dirs, &f, &p);
608 assert_se(r == -ENOENT);
609 }
610
611 TEST(search_and_fopen_nulstr) {
612 static const char dirs[] =
613 "/tmp/foo/bar\0"
614 "/tmp\0";
615
616 _cleanup_(unlink_tempfilep) char name[] = "/tmp/test-search_and_fopen.XXXXXX";
617 _cleanup_fclose_ FILE *f = NULL;
618 _cleanup_free_ char *p = NULL;
619 _cleanup_close_ int fd = -EBADF;
620 const char *e;
621 int r;
622
623 fd = mkostemp_safe(name);
624 assert_se(fd >= 0);
625 fd = safe_close(fd);
626
627 r = search_and_fopen_nulstr(basename(name), "re", NULL, dirs, &f, &p);
628 assert_se(r >= 0);
629 assert_se(e = path_startswith(p, "/tmp/"));
630 assert_se(streq(basename(name), e));
631 f = safe_fclose(f);
632 p = mfree(p);
633
634 r = search_and_fopen_nulstr(name, "re", NULL, dirs, &f, &p);
635 assert_se(r >= 0);
636 assert_se(path_equal(name, p));
637 f = safe_fclose(f);
638 p = mfree(p);
639
640 r = search_and_fopen_nulstr("/a/file/which/does/not/exist/i/guess", "re", NULL, dirs, &f, &p);
641 assert_se(r == -ENOENT);
642 r = search_and_fopen_nulstr("afilewhichdoesnotexistiguess", "re", NULL, dirs, &f, &p);
643 assert_se(r == -ENOENT);
644
645 r = unlink(name);
646 assert_se(r == 0);
647
648 r = search_and_fopen_nulstr(basename(name), "re", NULL, dirs, &f, &p);
649 assert_se(r == -ENOENT);
650 }
651
652 TEST(writing_tmpfile) {
653 _cleanup_(unlink_tempfilep) char name[] = "/tmp/test-systemd_writing_tmpfile.XXXXXX";
654 _cleanup_free_ char *contents = NULL;
655 size_t size;
656 _cleanup_close_ int fd = -EBADF;
657 int r;
658
659 struct iovec iov[] = {
660 IOVEC_MAKE_STRING("abc\n"),
661 IOVEC_MAKE_STRING(ALPHANUMERICAL "\n"),
662 IOVEC_MAKE_STRING(""),
663 };
664
665 fd = mkostemp_safe(name);
666 printf("tmpfile: %s", name);
667
668 r = writev(fd, iov, 3);
669 assert_se(r >= 0);
670
671 r = read_full_file(name, &contents, &size);
672 assert_se(r == 0);
673 printf("contents: %s", contents);
674 assert_se(streq(contents, "abc\n" ALPHANUMERICAL "\n"));
675 }
676
677 TEST(tempfn) {
678 char *ret = NULL, *p;
679
680 assert_se(tempfn_xxxxxx("/foo/bar/waldo", NULL, &ret) >= 0);
681 assert_se(streq_ptr(ret, "/foo/bar/.#waldoXXXXXX"));
682 free(ret);
683
684 assert_se(tempfn_xxxxxx("/foo/bar/waldo", "[miau]", &ret) >= 0);
685 assert_se(streq_ptr(ret, "/foo/bar/.#[miau]waldoXXXXXX"));
686 free(ret);
687
688 assert_se(tempfn_random("/foo/bar/waldo", NULL, &ret) >= 0);
689 assert_se(p = startswith(ret, "/foo/bar/.#waldo"));
690 assert_se(strlen(p) == 16);
691 assert_se(in_charset(p, "0123456789abcdef"));
692 free(ret);
693
694 assert_se(tempfn_random("/foo/bar/waldo", "[wuff]", &ret) >= 0);
695 assert_se(p = startswith(ret, "/foo/bar/.#[wuff]waldo"));
696 assert_se(strlen(p) == 16);
697 assert_se(in_charset(p, "0123456789abcdef"));
698 free(ret);
699
700 assert_se(tempfn_random_child("/foo/bar/waldo", NULL, &ret) >= 0);
701 assert_se(p = startswith(ret, "/foo/bar/waldo/.#"));
702 assert_se(strlen(p) == 16);
703 assert_se(in_charset(p, "0123456789abcdef"));
704 free(ret);
705
706 assert_se(tempfn_random_child("/foo/bar/waldo", "[kikiriki]", &ret) >= 0);
707 assert_se(p = startswith(ret, "/foo/bar/waldo/.#[kikiriki]"));
708 assert_se(strlen(p) == 16);
709 assert_se(in_charset(p, "0123456789abcdef"));
710 free(ret);
711 }
712
713 static const char chars[] =
714 "Aąę„”\n\377";
715
716 DISABLE_WARNING_TYPE_LIMITS;
717
718 TEST(fgetc) {
719 _cleanup_fclose_ FILE *f = NULL;
720 char c;
721
722 assert_se(f = fmemopen_unlocked((void*) chars, sizeof(chars), "r"));
723
724 for (size_t i = 0; i < sizeof(chars); i++) {
725 assert_se(safe_fgetc(f, &c) == 1);
726 assert_se(c == chars[i]);
727
728 if (ungetc(c, f) == EOF) {
729 /* EOF is -1, and hence we can't push value 255 in this way – if char is signed */
730 assert_se(c == (char) EOF);
731 assert_se(CHAR_MIN == -128); /* verify that char is signed on this platform */
732 } else {
733 assert_se(safe_fgetc(f, &c) == 1);
734 assert_se(c == chars[i]);
735 }
736
737 /* But it works when we push it properly cast */
738 assert_se(ungetc((unsigned char) c, f) != EOF);
739 assert_se(safe_fgetc(f, &c) == 1);
740 assert_se(c == chars[i]);
741 }
742
743 assert_se(safe_fgetc(f, &c) == 0);
744 }
745
746 REENABLE_WARNING;
747
748 static const char buffer[] =
749 "Some test data\n"
750 "루Non-ascii chars: ąę„”\n"
751 "terminators\r\n"
752 "and even more\n\r"
753 "now the same with a NUL\n\0"
754 "and more\r\0"
755 "and even more\r\n\0"
756 "and yet even more\n\r\0"
757 "With newlines, and a NUL byte\0"
758 "\n"
759 "an empty line\n"
760 "an ignored line\n"
761 "and a very long line that is supposed to be truncated, because it is so long\n";
762
763 static void test_read_line_one_file(FILE *f) {
764 _cleanup_free_ char *line = NULL;
765
766 assert_se(read_line(f, SIZE_MAX, &line) == 15 && streq(line, "Some test data"));
767 line = mfree(line);
768
769 assert_se(read_line(f, SIZE_MAX, &line) > 0 && streq(line, "루Non-ascii chars: ąę„”"));
770 line = mfree(line);
771
772 assert_se(read_line(f, SIZE_MAX, &line) == 13 && streq(line, "terminators"));
773 line = mfree(line);
774
775 assert_se(read_line(f, SIZE_MAX, &line) == 15 && streq(line, "and even more"));
776 line = mfree(line);
777
778 assert_se(read_line(f, SIZE_MAX, &line) == 25 && streq(line, "now the same with a NUL"));
779 line = mfree(line);
780
781 assert_se(read_line(f, SIZE_MAX, &line) == 10 && streq(line, "and more"));
782 line = mfree(line);
783
784 assert_se(read_line(f, SIZE_MAX, &line) == 16 && streq(line, "and even more"));
785 line = mfree(line);
786
787 assert_se(read_line(f, SIZE_MAX, &line) == 20 && streq(line, "and yet even more"));
788 line = mfree(line);
789
790 assert_se(read_line(f, 1024, &line) == 30 && streq(line, "With newlines, and a NUL byte"));
791 line = mfree(line);
792
793 assert_se(read_line(f, 1024, &line) == 1 && streq(line, ""));
794 line = mfree(line);
795
796 assert_se(read_line(f, 1024, &line) == 14 && streq(line, "an empty line"));
797 line = mfree(line);
798
799 assert_se(read_line(f, SIZE_MAX, NULL) == 16);
800
801 assert_se(read_line(f, 16, &line) == -ENOBUFS);
802 line = mfree(line);
803
804 /* read_line() stopped when it hit the limit, that means when we continue reading we'll read at the first
805 * character after the previous limit. Let's make use of that to continue our test. */
806 assert_se(read_line(f, 1024, &line) == 62 && streq(line, "line that is supposed to be truncated, because it is so long"));
807 line = mfree(line);
808
809 assert_se(read_line(f, 1024, &line) == 0 && streq(line, ""));
810 }
811
812 TEST(read_line1) {
813 _cleanup_fclose_ FILE *f = NULL;
814
815 assert_se(f = fmemopen_unlocked((void*) buffer, sizeof(buffer), "r"));
816 test_read_line_one_file(f);
817 }
818
819 TEST(read_line2) {
820 _cleanup_(unlink_tempfilep) char name[] = "/tmp/test-fileio.XXXXXX";
821 int fd;
822 _cleanup_fclose_ FILE *f = NULL;
823
824 fd = mkostemp_safe(name);
825 assert_se(fd >= 0);
826 assert_se((size_t) write(fd, buffer, sizeof(buffer)) == sizeof(buffer));
827
828 assert_se(lseek(fd, 0, SEEK_SET) == 0);
829 assert_se(f = fdopen(fd, "r"));
830
831 test_read_line_one_file(f);
832 }
833
834 TEST(read_line3) {
835 _cleanup_fclose_ FILE *f = NULL;
836 _cleanup_free_ char *line = NULL;
837 int r;
838
839 f = fopen("/proc/uptime", "re");
840 if (!f && IN_SET(errno, ENOENT, EPERM))
841 return;
842 assert_se(f);
843
844 r = read_line(f, LINE_MAX, &line);
845 assert_se(r >= 0);
846 if (r == 0)
847 assert_se(line && isempty(line));
848 else
849 assert_se((size_t) r == strlen(line) + 1);
850 assert_se(read_line(f, LINE_MAX, NULL) == 0);
851 }
852
853 TEST(read_line4) {
854 static const struct {
855 size_t length;
856 const char *string;
857 } eof_endings[] = {
858 /* Each of these will be followed by EOF and should generate the one same single string */
859 { 3, "foo" },
860 { 4, "foo\n" },
861 { 4, "foo\r" },
862 { 4, "foo\0" },
863 { 5, "foo\n\0" },
864 { 5, "foo\r\0" },
865 { 5, "foo\r\n" },
866 { 5, "foo\n\r" },
867 { 6, "foo\r\n\0" },
868 { 6, "foo\n\r\0" },
869 };
870
871 int r;
872
873 for (size_t i = 0; i < ELEMENTSOF(eof_endings); i++) {
874 _cleanup_fclose_ FILE *f = NULL;
875 _cleanup_free_ char *s = NULL;
876
877 assert_se(f = fmemopen_unlocked((void*) eof_endings[i].string, eof_endings[i].length, "r"));
878
879 r = read_line(f, SIZE_MAX, &s);
880 assert_se((size_t) r == eof_endings[i].length);
881 assert_se(streq_ptr(s, "foo"));
882
883 assert_se(read_line(f, SIZE_MAX, NULL) == 0); /* Ensure we hit EOF */
884 }
885 }
886
887 TEST(read_nul_string) {
888 static const char test[] = "string nr. 1\0"
889 "string nr. 2\n\0"
890 "\377empty string follows\0"
891 "\0"
892 "final string\n is empty\0"
893 "\0";
894
895 _cleanup_fclose_ FILE *f = NULL;
896 _cleanup_free_ char *s = NULL;
897
898 assert_se(f = fmemopen_unlocked((void*) test, sizeof(test)-1, "r"));
899
900 assert_se(read_nul_string(f, LONG_LINE_MAX, &s) == 13 && streq_ptr(s, "string nr. 1"));
901 s = mfree(s);
902
903 assert_se(read_nul_string(f, LONG_LINE_MAX, &s) == 14 && streq_ptr(s, "string nr. 2\n"));
904 s = mfree(s);
905
906 assert_se(read_nul_string(f, LONG_LINE_MAX, &s) == 22 && streq_ptr(s, "\377empty string follows"));
907 s = mfree(s);
908
909 assert_se(read_nul_string(f, LONG_LINE_MAX, &s) == 1 && streq_ptr(s, ""));
910 s = mfree(s);
911
912 assert_se(read_nul_string(f, LONG_LINE_MAX, &s) == 23 && streq_ptr(s, "final string\n is empty"));
913 s = mfree(s);
914
915 assert_se(read_nul_string(f, LONG_LINE_MAX, &s) == 1 && streq_ptr(s, ""));
916 s = mfree(s);
917
918 assert_se(read_nul_string(f, LONG_LINE_MAX, &s) == 0 && streq_ptr(s, ""));
919 }
920
921 TEST(read_full_file_socket) {
922 _cleanup_(rm_rf_physical_and_freep) char *z = NULL;
923 _cleanup_close_ int listener = -EBADF;
924 _cleanup_free_ char *data = NULL, *clientname = NULL;
925 union sockaddr_union sa;
926 const char *j, *jj;
927 size_t size;
928 pid_t pid;
929 int r;
930
931 listener = socket(AF_UNIX, SOCK_STREAM|SOCK_CLOEXEC, 0);
932 assert_se(listener >= 0);
933
934 assert_se(mkdtemp_malloc(NULL, &z) >= 0);
935 j = strjoina(z, "/socket");
936
937 assert_se(sockaddr_un_set_path(&sa.un, j) >= 0);
938
939 assert_se(bind(listener, &sa.sa, SOCKADDR_UN_LEN(sa.un)) >= 0);
940 assert_se(listen(listener, 1) >= 0);
941
942 /* Make sure the socket doesn't fit into a struct sockaddr_un, but we can still access it */
943 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");
944 assert_se(strlen(jj) > sizeof_field(struct sockaddr_un, sun_path));
945 assert_se(rename(j, jj) >= 0);
946
947 /* Bind the *client* socket to some randomized name, to verify that this works correctly. */
948 assert_se(asprintf(&clientname, "@%" PRIx64 "/test-bindname", random_u64()) >= 0);
949
950 r = safe_fork("(server)", FORK_DEATHSIG|FORK_LOG, &pid);
951 assert_se(r >= 0);
952 if (r == 0) {
953 union sockaddr_union peer = {};
954 socklen_t peerlen = sizeof(peer);
955 _cleanup_close_ int rfd = -EBADF;
956 /* child */
957
958 rfd = accept4(listener, NULL, 0, SOCK_CLOEXEC);
959 assert_se(rfd >= 0);
960
961 assert_se(getpeername(rfd, &peer.sa, &peerlen) >= 0);
962
963 assert_se(peer.un.sun_family == AF_UNIX);
964 assert_se(peerlen > offsetof(struct sockaddr_un, sun_path));
965 assert_se(peer.un.sun_path[0] == 0);
966 assert_se(streq(peer.un.sun_path + 1, clientname + 1));
967
968 #define TEST_STR "This is a test\nreally."
969
970 assert_se(write(rfd, TEST_STR, strlen(TEST_STR)) == strlen(TEST_STR));
971 _exit(EXIT_SUCCESS);
972 }
973
974 assert_se(read_full_file_full(AT_FDCWD, jj, UINT64_MAX, SIZE_MAX, 0, NULL, &data, &size) == -ENXIO);
975 assert_se(read_full_file_full(AT_FDCWD, jj, UINT64_MAX, SIZE_MAX, READ_FULL_FILE_CONNECT_SOCKET, clientname, &data, &size) >= 0);
976 assert_se(size == strlen(TEST_STR));
977 assert_se(streq(data, TEST_STR));
978
979 assert_se(wait_for_terminate_and_check("(server)", pid, WAIT_LOG) >= 0);
980 #undef TEST_STR
981 }
982
983 TEST(read_full_file_offset_size) {
984 _cleanup_fclose_ FILE *f = NULL;
985 _cleanup_(unlink_and_freep) char *fn = NULL;
986 _cleanup_free_ char *rbuf = NULL;
987 size_t rbuf_size;
988 uint8_t buf[4711];
989
990 random_bytes(buf, sizeof(buf));
991
992 assert_se(tempfn_random_child(NULL, NULL, &fn) >= 0);
993 assert_se(f = fopen(fn, "we"));
994 assert_se(fwrite(buf, 1, sizeof(buf), f) == sizeof(buf));
995 assert_se(fflush_and_check(f) >= 0);
996
997 assert_se(read_full_file_full(AT_FDCWD, fn, UINT64_MAX, SIZE_MAX, 0, NULL, &rbuf, &rbuf_size) >= 0);
998 assert_se(rbuf_size == sizeof(buf));
999 assert_se(memcmp(buf, rbuf, rbuf_size) == 0);
1000 rbuf = mfree(rbuf);
1001
1002 assert_se(read_full_file_full(AT_FDCWD, fn, UINT64_MAX, 128, 0, NULL, &rbuf, &rbuf_size) >= 0);
1003 assert_se(rbuf_size == 128);
1004 assert_se(memcmp(buf, rbuf, rbuf_size) == 0);
1005 rbuf = mfree(rbuf);
1006
1007 assert_se(read_full_file_full(AT_FDCWD, fn, UINT64_MAX, 128, READ_FULL_FILE_FAIL_WHEN_LARGER, NULL, &rbuf, &rbuf_size) == -E2BIG);
1008 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);
1009 assert_se(read_full_file_full(AT_FDCWD, fn, UINT64_MAX, sizeof(buf), READ_FULL_FILE_FAIL_WHEN_LARGER, NULL, &rbuf, &rbuf_size) >= 0);
1010 assert_se(rbuf_size == sizeof(buf));
1011 assert_se(memcmp(buf, rbuf, rbuf_size) == 0);
1012 rbuf = mfree(rbuf);
1013
1014 assert_se(read_full_file_full(AT_FDCWD, fn, 47, 128, READ_FULL_FILE_FAIL_WHEN_LARGER, NULL, &rbuf, &rbuf_size) == -E2BIG);
1015 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);
1016 assert_se(read_full_file_full(AT_FDCWD, fn, 47, sizeof(buf)-47, READ_FULL_FILE_FAIL_WHEN_LARGER, NULL, &rbuf, &rbuf_size) >= 0);
1017 assert_se(rbuf_size == sizeof(buf)-47);
1018 assert_se(memcmp(buf+47, rbuf, rbuf_size) == 0);
1019 rbuf = mfree(rbuf);
1020
1021 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);
1022 assert_se(rbuf_size == sizeof(buf));
1023 assert_se(memcmp(buf, rbuf, rbuf_size) == 0);
1024 rbuf = mfree(rbuf);
1025
1026 assert_se(read_full_file_full(AT_FDCWD, fn, 1234, SIZE_MAX, 0, NULL, &rbuf, &rbuf_size) >= 0);
1027 assert_se(rbuf_size == sizeof(buf) - 1234);
1028 assert_se(memcmp(buf + 1234, rbuf, rbuf_size) == 0);
1029 rbuf = mfree(rbuf);
1030
1031 assert_se(read_full_file_full(AT_FDCWD, fn, 2345, 777, 0, NULL, &rbuf, &rbuf_size) >= 0);
1032 assert_se(rbuf_size == 777);
1033 assert_se(memcmp(buf + 2345, rbuf, rbuf_size) == 0);
1034 rbuf = mfree(rbuf);
1035
1036 assert_se(read_full_file_full(AT_FDCWD, fn, 4700, 20, 0, NULL, &rbuf, &rbuf_size) >= 0);
1037 assert_se(rbuf_size == 11);
1038 assert_se(memcmp(buf + 4700, rbuf, rbuf_size) == 0);
1039 rbuf = mfree(rbuf);
1040
1041 assert_se(read_full_file_full(AT_FDCWD, fn, 10000, 99, 0, NULL, &rbuf, &rbuf_size) >= 0);
1042 assert_se(rbuf_size == 0);
1043 rbuf = mfree(rbuf);
1044 }
1045
1046 static void test_read_virtual_file_one(size_t max_size) {
1047 int r;
1048
1049 log_info("/* %s (max_size=%zu) */", __func__, max_size);
1050
1051 FOREACH_STRING(filename,
1052 "/proc/1/cmdline",
1053 "/etc/nsswitch.conf",
1054 "/sys/kernel/uevent_seqnum",
1055 "/proc/kcore",
1056 "/proc/kallsyms",
1057 "/proc/self/exe",
1058 "/proc/self/pagemap") {
1059
1060 _cleanup_free_ char *buf = NULL;
1061 size_t size = 0;
1062
1063 r = read_virtual_file(filename, max_size, &buf, &size);
1064 if (r < 0) {
1065 log_info_errno(r, "read_virtual_file(\"%s\", %zu): %m", filename, max_size);
1066 assert_se(ERRNO_IS_PRIVILEGE(r) || /* /proc/kcore is not accessible to unpriv */
1067 IN_SET(r,
1068 -ENOENT, /* Some of the files might be absent */
1069 -EINVAL, /* too small reads from /proc/self/pagemap trigger EINVAL */
1070 -EFBIG)); /* /proc/kcore and /proc/self/pagemap should be too large */
1071 } else
1072 log_info("read_virtual_file(\"%s\", %zu): %s (%zu bytes)", filename, max_size, r ? "non-truncated" : "truncated", size);
1073 }
1074 }
1075
1076 TEST(test_read_virtual_file) {
1077 test_read_virtual_file_one(0);
1078 test_read_virtual_file_one(1);
1079 test_read_virtual_file_one(2);
1080 test_read_virtual_file_one(20);
1081 test_read_virtual_file_one(4096);
1082 test_read_virtual_file_one(4097);
1083 test_read_virtual_file_one(SIZE_MAX);
1084 }
1085
1086 TEST(test_fdopen_independent) {
1087 #define TEST_TEXT "this is some random test text we are going to write to a memfd"
1088 _cleanup_close_ int fd = -EBADF;
1089 _cleanup_fclose_ FILE *f = NULL;
1090 char buf[STRLEN(TEST_TEXT) + 1];
1091
1092 fd = memfd_new("fdopen_independent");
1093 if (fd < 0) {
1094 assert_se(ERRNO_IS_NOT_SUPPORTED(fd));
1095 return;
1096 }
1097
1098 assert_se(write(fd, TEST_TEXT, strlen(TEST_TEXT)) == strlen(TEST_TEXT));
1099 /* we'll leave the read offset at the end of the memfd, the fdopen_independent() descriptors should
1100 * start at the beginning anyway */
1101
1102 assert_se(fdopen_independent(fd, "re", &f) >= 0);
1103 zero(buf);
1104 assert_se(fread(buf, 1, sizeof(buf), f) == strlen(TEST_TEXT));
1105 assert_se(streq(buf, TEST_TEXT));
1106 assert_se((fcntl(fileno(f), F_GETFL) & O_ACCMODE) == O_RDONLY);
1107 assert_se(FLAGS_SET(fcntl(fileno(f), F_GETFD), FD_CLOEXEC));
1108 f = safe_fclose(f);
1109
1110 assert_se(fdopen_independent(fd, "r", &f) >= 0);
1111 zero(buf);
1112 assert_se(fread(buf, 1, sizeof(buf), f) == strlen(TEST_TEXT));
1113 assert_se(streq(buf, TEST_TEXT));
1114 assert_se((fcntl(fileno(f), F_GETFL) & O_ACCMODE) == O_RDONLY);
1115 assert_se(!FLAGS_SET(fcntl(fileno(f), F_GETFD), FD_CLOEXEC));
1116 f = safe_fclose(f);
1117
1118 assert_se(fdopen_independent(fd, "r+e", &f) >= 0);
1119 zero(buf);
1120 assert_se(fread(buf, 1, sizeof(buf), f) == strlen(TEST_TEXT));
1121 assert_se(streq(buf, TEST_TEXT));
1122 assert_se((fcntl(fileno(f), F_GETFL) & O_ACCMODE) == O_RDWR);
1123 assert_se(FLAGS_SET(fcntl(fileno(f), F_GETFD), FD_CLOEXEC));
1124 f = safe_fclose(f);
1125 }
1126
1127
1128 DEFINE_TEST_MAIN(LOG_DEBUG);