mirror of
https://github.com/Dasharo/systemd.git
synced 2026-03-06 15:02:31 -08:00
cd93ec5ed2
When we see a "\" in COMMENT state, we change the state to COMMENT_ESCAPE. When we got a new character, we reset the state to COMMENT, but this character is not dispatched. Usually the character is NEWLINE, if so we will stay in COMMENT state until we find the next NEWLINE. fix: https://github.com/systemd/systemd/issues/27975
192 lines
7.3 KiB
C
192 lines
7.3 KiB
C
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
|
|
|
#include "env-file.h"
|
|
#include "fd-util.h"
|
|
#include "fileio.h"
|
|
#include "fs-util.h"
|
|
#include "macro.h"
|
|
#include "strv.h"
|
|
#include "tests.h"
|
|
#include "tmpfile-util.h"
|
|
|
|
/* In case of repeating keys, later entries win. */
|
|
|
|
#define env_file_1 \
|
|
"a=a\n" \
|
|
"a=b\n" \
|
|
"a=b\n" \
|
|
"a=a\n" \
|
|
"b=b\\\n" \
|
|
"c\n" \
|
|
"d= d\\\n" \
|
|
"e \\\n" \
|
|
"f \n" \
|
|
"g=g\\ \n" \
|
|
"h= ąęół\\ śćńźżµ \n" \
|
|
"i=i\\"
|
|
|
|
#define env_file_2 \
|
|
"a=a\\\n"
|
|
|
|
#define env_file_3 \
|
|
"#SPAMD_ARGS=\"-d --socketpath=/var/lib/bulwark/spamd \\\n" \
|
|
"#--nouser-config \\\n" \
|
|
"normal1=line\\\n" \
|
|
"111\n" \
|
|
";normal=ignored \\\n" \
|
|
"normal2=line222\n" \
|
|
"normal ignored \\\n"
|
|
|
|
#define env_file_4 \
|
|
"# Generated\n" \
|
|
"\n" \
|
|
"HWMON_MODULES=\"coretemp f71882fg\"\n" \
|
|
"\n" \
|
|
"# For compatibility reasons\n" \
|
|
"\n" \
|
|
"MODULE_0=coretemp\n" \
|
|
"MODULE_1=f71882fg"
|
|
|
|
#define env_file_5 \
|
|
"a=\n" \
|
|
"b="
|
|
|
|
#define env_file_6 \
|
|
"a=\\ \\n \\t \\x \\y \\' \n" \
|
|
"b= \\$' \n" \
|
|
"c= ' \\n\\t\\$\\`\\\\\n" \
|
|
"' \n" \
|
|
"d= \" \\n\\t\\$\\`\\\\\n" \
|
|
"\" \n"
|
|
|
|
TEST(load_env_file_1) {
|
|
_cleanup_(unlink_tempfilep) char name[] = "/tmp/test-load-env-file.XXXXXX";
|
|
assert_se(write_tmpfile(name, env_file_1) == 0);
|
|
|
|
_cleanup_strv_free_ char **data = NULL;
|
|
assert_se(load_env_file(NULL, name, &data) == 0);
|
|
assert_se(streq(data[0], "a=a"));
|
|
assert_se(streq(data[1], "b=bc"));
|
|
assert_se(streq(data[2], "d=de f"));
|
|
assert_se(streq(data[3], "g=g "));
|
|
assert_se(streq(data[4], "h=ąęół śćńźżµ"));
|
|
assert_se(streq(data[5], "i=i"));
|
|
assert_se(data[6] == NULL);
|
|
}
|
|
|
|
TEST(load_env_file_2) {
|
|
_cleanup_(unlink_tempfilep) char name[] = "/tmp/test-load-env-file.XXXXXX";
|
|
assert_se(write_tmpfile(name, env_file_2) == 0);
|
|
|
|
_cleanup_strv_free_ char **data = NULL;
|
|
assert_se(load_env_file(NULL, name, &data) == 0);
|
|
assert_se(streq(data[0], "a=a"));
|
|
assert_se(data[1] == NULL);
|
|
}
|
|
|
|
TEST(load_env_file_3) {
|
|
_cleanup_(unlink_tempfilep) char name[] = "/tmp/test-load-env-file.XXXXXX";
|
|
assert_se(write_tmpfile(name, env_file_3) == 0);
|
|
|
|
_cleanup_strv_free_ char **data = NULL;
|
|
assert_se(load_env_file(NULL, name, &data) == 0);
|
|
assert_se(streq(data[0], "normal1=line111"));
|
|
assert_se(streq(data[1], "normal2=line222"));
|
|
assert_se(data[2] == NULL);
|
|
}
|
|
|
|
TEST(load_env_file_4) {
|
|
_cleanup_(unlink_tempfilep) char name[] = "/tmp/test-load-env-file.XXXXXX";
|
|
assert_se(write_tmpfile(name, env_file_4) == 0);
|
|
|
|
_cleanup_strv_free_ char **data = NULL;
|
|
assert_se(load_env_file(NULL, name, &data) == 0);
|
|
assert_se(streq(data[0], "HWMON_MODULES=coretemp f71882fg"));
|
|
assert_se(streq(data[1], "MODULE_0=coretemp"));
|
|
assert_se(streq(data[2], "MODULE_1=f71882fg"));
|
|
assert_se(data[3] == NULL);
|
|
}
|
|
|
|
TEST(load_env_file_5) {
|
|
_cleanup_(unlink_tempfilep) char name[] = "/tmp/test-load-env-file.XXXXXX";
|
|
assert_se(write_tmpfile(name, env_file_5) == 0);
|
|
|
|
_cleanup_strv_free_ char **data = NULL;
|
|
assert_se(load_env_file(NULL, name, &data) == 0);
|
|
assert_se(streq(data[0], "a="));
|
|
assert_se(streq(data[1], "b="));
|
|
assert_se(data[2] == NULL);
|
|
}
|
|
|
|
TEST(load_env_file_6) {
|
|
_cleanup_(unlink_tempfilep) char name[] = "/tmp/test-load-env-file.XXXXXX";
|
|
assert_se(write_tmpfile(name, env_file_6) == 0);
|
|
|
|
_cleanup_strv_free_ char **data = NULL;
|
|
assert_se(load_env_file(NULL, name, &data) == 0);
|
|
assert_se(streq(data[0], "a= n t x y '"));
|
|
assert_se(streq(data[1], "b=$'"));
|
|
assert_se(streq(data[2], "c= \\n\\t\\$\\`\\\\\n"));
|
|
assert_se(streq(data[3], "d= \\n\\t$`\\\n"));
|
|
assert_se(data[4] == NULL);
|
|
}
|
|
|
|
TEST(load_env_file_invalid_utf8) {
|
|
/* Test out a couple of assignments where the key/value has an invalid
|
|
* UTF-8 character ("noncharacter")
|
|
*
|
|
* See: https://en.wikipedia.org/wiki/Universal_Character_Set_characters#Non-characters
|
|
*/
|
|
FOREACH_STRING(s,
|
|
"fo\ufffeo=bar",
|
|
"foo=b\uffffar",
|
|
"baz=hello world\ufffe") {
|
|
_cleanup_(unlink_tempfilep) char name[] = "/tmp/test-load-env-file.XXXXXX";
|
|
assert_se(write_tmpfile(name, s) == 0);
|
|
|
|
_cleanup_strv_free_ char **data = NULL;
|
|
assert_se(load_env_file(NULL, name, &data) == -EINVAL);
|
|
assert_se(!data);
|
|
}
|
|
}
|
|
|
|
TEST(write_and_load_env_file) {
|
|
/* Make sure that our writer, parser and the shell agree on what our env var files mean */
|
|
|
|
FOREACH_STRING(v,
|
|
"obbardc-laptop",
|
|
"obbardc\\-laptop",
|
|
"obbardc-lap\\top",
|
|
"obbardc-lap\\top",
|
|
"obbardc-lap\\\\top",
|
|
"double\"quote",
|
|
"single\'quote",
|
|
"dollar$dollar",
|
|
"newline\nnewline") {
|
|
_cleanup_(unlink_and_freep) char *p = NULL;
|
|
_cleanup_strv_free_ char **l = NULL;
|
|
_cleanup_free_ char *j = NULL, *w = NULL, *cmd = NULL, *from_shell = NULL;
|
|
_cleanup_pclose_ FILE *f = NULL;
|
|
size_t sz;
|
|
|
|
assert_se(tempfn_random_child(NULL, NULL, &p) >= 0);
|
|
|
|
assert_se(j = strjoin("TEST=", v));
|
|
assert_se(write_env_file(p, STRV_MAKE(j)) >= 0);
|
|
|
|
assert_se(cmd = strjoin(". ", p, " && /bin/echo -n \"$TEST\""));
|
|
assert_se(f = popen(cmd, "re"));
|
|
assert_se(read_full_stream(f, &from_shell, &sz) >= 0);
|
|
assert_se(sz == strlen(v));
|
|
assert_se(streq(from_shell, v));
|
|
|
|
assert_se(load_env_file(NULL, p, &l) >= 0);
|
|
assert_se(strv_equal(l, STRV_MAKE(j)));
|
|
|
|
assert_se(parse_env_file(NULL, p, "TEST", &w) >= 0);
|
|
assert_se(streq_ptr(w, v));
|
|
}
|
|
}
|
|
|
|
DEFINE_TEST_MAIN(LOG_INFO);
|