2017-11-18 17:09:20 +01:00
|
|
|
/* SPDX-License-Identifier: LGPL-2.1+ */
|
2014-06-20 04:07:05 +02:00
|
|
|
/***
|
2018-06-12 19:00:24 +02:00
|
|
|
Copyright © 2014 Michael Marineau
|
2014-06-20 04:07:05 +02:00
|
|
|
***/
|
|
|
|
|
|
|
|
#include <stdarg.h>
|
2015-10-24 22:58:24 +02:00
|
|
|
#include <stdio.h>
|
2014-06-20 04:07:05 +02:00
|
|
|
|
2015-10-27 03:01:06 +01:00
|
|
|
#include "alloc-util.h"
|
2014-06-20 04:07:05 +02:00
|
|
|
#include "conf-files.h"
|
2018-04-16 21:24:13 +02:00
|
|
|
#include "fileio.h"
|
2015-10-26 21:16:26 +01:00
|
|
|
#include "fs-util.h"
|
2014-06-20 04:07:05 +02:00
|
|
|
#include "macro.h"
|
2018-04-16 21:24:13 +02:00
|
|
|
#include "mkdir.h"
|
2015-11-11 22:54:56 +01:00
|
|
|
#include "parse-util.h"
|
2018-12-19 23:05:48 +01:00
|
|
|
#include "path-util.h"
|
2015-10-24 22:58:24 +02:00
|
|
|
#include "rm-rf.h"
|
|
|
|
#include "string-util.h"
|
2014-06-20 04:07:05 +02:00
|
|
|
#include "strv.h"
|
2018-09-13 14:31:13 +02:00
|
|
|
#include "tests.h"
|
2015-10-27 00:42:07 +01:00
|
|
|
#include "user-util.h"
|
2014-06-20 04:07:05 +02:00
|
|
|
#include "util.h"
|
|
|
|
|
|
|
|
static void setup_test_dir(char *tmp_dir, const char *files, ...) {
|
|
|
|
va_list ap;
|
|
|
|
|
2018-04-16 21:24:13 +02:00
|
|
|
assert_se(mkdtemp(tmp_dir));
|
2014-06-20 04:07:05 +02:00
|
|
|
|
|
|
|
va_start(ap, files);
|
2018-04-16 21:24:13 +02:00
|
|
|
while (files) {
|
|
|
|
_cleanup_free_ char *path;
|
|
|
|
|
|
|
|
assert_se(path = strappend(tmp_dir, files));
|
|
|
|
(void) mkdir_parents(path, 0755);
|
|
|
|
assert_se(write_string_file(path, "foobar", WRITE_STRING_FILE_CREATE) >= 0);
|
|
|
|
|
2014-06-20 04:07:05 +02:00
|
|
|
files = va_arg(ap, const char *);
|
|
|
|
}
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void test_conf_files_list(bool use_root) {
|
|
|
|
char tmp_dir[] = "/tmp/test-conf-files-XXXXXX";
|
2017-01-22 20:16:19 +01:00
|
|
|
_cleanup_strv_free_ char **found_files = NULL, **found_files2 = NULL;
|
2018-04-16 21:24:13 +02:00
|
|
|
const char *root_dir, *search_1, *search_2, *expect_a, *expect_b, *expect_c, *mask;
|
2017-01-22 20:16:19 +01:00
|
|
|
|
2018-12-19 23:05:48 +01:00
|
|
|
log_debug("/* %s(%s) */", __func__, yes_no(use_root));
|
2014-06-20 04:07:05 +02:00
|
|
|
|
|
|
|
setup_test_dir(tmp_dir,
|
|
|
|
"/dir1/a.conf",
|
|
|
|
"/dir2/a.conf",
|
|
|
|
"/dir2/b.conf",
|
2017-01-22 20:16:19 +01:00
|
|
|
"/dir2/c.foo",
|
2018-04-16 21:24:13 +02:00
|
|
|
"/dir2/d.conf",
|
2014-06-20 04:07:05 +02:00
|
|
|
NULL);
|
|
|
|
|
2018-04-16 21:24:13 +02:00
|
|
|
mask = strjoina(tmp_dir, "/dir1/d.conf");
|
|
|
|
assert_se(symlink("/dev/null", mask) >= 0);
|
|
|
|
|
2014-06-20 04:07:05 +02:00
|
|
|
if (use_root) {
|
|
|
|
root_dir = tmp_dir;
|
|
|
|
search_1 = "/dir1";
|
|
|
|
search_2 = "/dir2";
|
|
|
|
} else {
|
|
|
|
root_dir = NULL;
|
2015-02-03 02:05:59 +01:00
|
|
|
search_1 = strjoina(tmp_dir, "/dir1");
|
|
|
|
search_2 = strjoina(tmp_dir, "/dir2");
|
2014-06-20 04:07:05 +02:00
|
|
|
}
|
|
|
|
|
2015-02-03 02:05:59 +01:00
|
|
|
expect_a = strjoina(tmp_dir, "/dir1/a.conf");
|
|
|
|
expect_b = strjoina(tmp_dir, "/dir2/b.conf");
|
2017-01-22 20:16:19 +01:00
|
|
|
expect_c = strjoina(tmp_dir, "/dir2/c.foo");
|
|
|
|
|
|
|
|
log_debug("/* Check when filtered by suffix */");
|
2014-06-20 04:07:05 +02:00
|
|
|
|
2018-04-16 21:24:13 +02:00
|
|
|
assert_se(conf_files_list(&found_files, ".conf", root_dir, CONF_FILES_FILTER_MASKED, search_1, search_2, NULL) == 0);
|
2014-06-20 04:07:05 +02:00
|
|
|
strv_print(found_files);
|
|
|
|
|
|
|
|
assert_se(found_files);
|
|
|
|
assert_se(streq_ptr(found_files[0], expect_a));
|
|
|
|
assert_se(streq_ptr(found_files[1], expect_b));
|
2018-04-16 21:24:13 +02:00
|
|
|
assert_se(!found_files[2]);
|
2014-06-20 04:07:05 +02:00
|
|
|
|
2017-01-22 20:16:19 +01:00
|
|
|
log_debug("/* Check when unfiltered */");
|
2018-04-16 21:24:13 +02:00
|
|
|
assert_se(conf_files_list(&found_files2, NULL, root_dir, CONF_FILES_FILTER_MASKED, search_1, search_2, NULL) == 0);
|
2017-01-22 20:16:19 +01:00
|
|
|
strv_print(found_files2);
|
|
|
|
|
|
|
|
assert_se(found_files2);
|
|
|
|
assert_se(streq_ptr(found_files2[0], expect_a));
|
|
|
|
assert_se(streq_ptr(found_files2[1], expect_b));
|
|
|
|
assert_se(streq_ptr(found_files2[2], expect_c));
|
2018-04-16 21:24:13 +02:00
|
|
|
assert_se(!found_files2[3]);
|
2017-01-22 20:16:19 +01:00
|
|
|
|
2015-04-04 11:52:57 +02:00
|
|
|
assert_se(rm_rf(tmp_dir, REMOVE_ROOT|REMOVE_PHYSICAL) == 0);
|
2014-06-20 04:07:05 +02:00
|
|
|
}
|
|
|
|
|
2018-12-19 23:05:48 +01:00
|
|
|
static void test_conf_files_insert(const char *root) {
|
|
|
|
_cleanup_strv_free_ char **s = NULL;
|
|
|
|
|
|
|
|
log_info("/* %s root=%s */", __func__, strempty(root));
|
|
|
|
|
|
|
|
char **dirs = STRV_MAKE("/dir1", "/dir2", "/dir3");
|
|
|
|
|
|
|
|
_cleanup_free_ const char
|
|
|
|
*foo1 = prefix_root(root, "/dir1/foo.conf"),
|
|
|
|
*foo2 = prefix_root(root, "/dir2/foo.conf"),
|
|
|
|
*bar2 = prefix_root(root, "/dir2/bar.conf"),
|
|
|
|
*zzz3 = prefix_root(root, "/dir3/zzz.conf"),
|
|
|
|
*whatever = prefix_root(root, "/whatever.conf");
|
|
|
|
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir2/foo.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(foo2)));
|
|
|
|
|
|
|
|
/* The same file again, https://github.com/systemd/systemd/issues/11124 */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir2/foo.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(foo2)));
|
|
|
|
|
|
|
|
/* Lower priority → new entry is ignored */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir3/foo.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(foo2)));
|
|
|
|
|
|
|
|
/* Higher priority → new entry replaces */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir1/foo.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(foo1)));
|
|
|
|
|
|
|
|
/* Earlier basename */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir2/bar.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(bar2, foo1)));
|
|
|
|
|
|
|
|
/* Later basename */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir3/zzz.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(bar2, foo1, zzz3)));
|
|
|
|
|
|
|
|
/* All lower priority → all ignored */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir3/zzz.conf") == 0);
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir2/bar.conf") == 0);
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir3/bar.conf") == 0);
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir2/foo.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(bar2, foo1, zzz3)));
|
|
|
|
|
|
|
|
/* Two entries that don't match any of the directories, but match basename */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/dir4/zzz.conf") == 0);
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/zzz.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(bar2, foo1, zzz3)));
|
|
|
|
|
|
|
|
/* An entry that doesn't match any of the directories, no match at all */
|
|
|
|
assert_se(conf_files_insert(&s, root, dirs, "/whatever.conf") == 0);
|
|
|
|
assert_se(strv_equal(s, STRV_MAKE(bar2, foo1, whatever, zzz3)));
|
|
|
|
}
|
|
|
|
|
2014-06-20 04:07:05 +02:00
|
|
|
int main(int argc, char **argv) {
|
2018-09-13 14:31:13 +02:00
|
|
|
test_setup_logging(LOG_DEBUG);
|
2017-01-22 20:16:19 +01:00
|
|
|
|
2014-06-20 04:07:05 +02:00
|
|
|
test_conf_files_list(false);
|
|
|
|
test_conf_files_list(true);
|
2018-12-19 23:05:48 +01:00
|
|
|
test_conf_files_insert(NULL);
|
|
|
|
test_conf_files_insert("/root");
|
|
|
|
test_conf_files_insert("/root/");
|
|
|
|
|
2014-06-20 04:07:05 +02:00
|
|
|
return 0;
|
|
|
|
}
|