shaba/systemd-stable
1
1
Fork 0
mirror of https://github.com/systemd/systemd-stable.git synced 2025-01-10 01:17:44 +03:00
Code

fs-util: rewrite rmdir_parents() with path_find_last_component()

Browse Source
This commit is contained in:
Yu Watanabe 2021-09-02 13:41:15 +09:00
parent 3008a6f21c
commit 4e046c5c2f
2 changed files with 76 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/basic/fs-util.c
View File

@ -45,50 +45,39 @@ int unlink_noerrno(const char *path) {
} }
int rmdir_parents(const char *path, const char *stop) { int rmdir_parents(const char *path, const char *stop) {
size_t l; char *p;
int r = 0; int r;
assert(path); assert(path);
assert(stop); assert(stop);
l = strlen(path); if (!path_is_safe(path))
return -EINVAL;
/* Skip trailing slashes */ if (!path_is_safe(stop))
while (l > 0 && path[l-1] == '/') return -EINVAL;
l--;
while (l > 0) { p = strdupa(path);
char *t;
/* Skip last component */ for (;;) {
while (l > 0 && path[l-1] != '/') char *slash = NULL;
l--;
/* Skip trailing slashes */ /* skip the last component. */
while (l > 0 && path[l-1] == '/') r = path_find_last_component(p, /* accept_dot_dot= */ false, (const char **) &slash, NULL);
l--; if (r <= 0)
return r;
if (l <= 0) if (slash == p)
break;
t = strndup(path, l);
if (!t)
return -ENOMEM;
if (path_startswith(stop, t)) {
free(t);
return 0; return 0;
}
r = rmdir(t); assert(*slash == '/');
free(t); *slash = '\0';
if (r < 0) if (path_startswith_full(stop, p, /* accept_dot_dot= */ false))
if (errno != ENOENT) return 0;
return -errno;
if (rmdir(p) < 0 && errno != ENOENT)
return -errno;
} }
return 0;
} }
int rename_noreplace(int olddirfd, const char *oldpath, int newdirfd, const char *newpath) { int rename_noreplace(int olddirfd, const char *oldpath, int newdirfd, const char *newpath) {

55
src/test/test-fs-util.c
View File

@ -865,6 +865,60 @@ static void test_conservative_rename(void) {
assert_se(access(q, F_OK) < 0 && errno == ENOENT); assert_se(access(q, F_OK) < 0 && errno == ENOENT);
} }
static void test_rmdir_parents_one(
const char *prefix,
const char *path,
const char *stop,
int expected,
const char *test_exist,
const char *test_nonexist_subdir) {
const char *p, *s;
log_debug("/* %s(%s, %s) */", __func__, path, stop);
p = strjoina(prefix, path);
s = strjoina(prefix, stop);
if (expected >= 0)
assert_se(mkdir_parents(p, 0700) >= 0);
assert_se(rmdir_parents(p, s) == expected);
if (expected >= 0) {
const char *e, *f;
e = strjoina(prefix, test_exist);
f = strjoina(e, test_nonexist_subdir);
assert_se(access(e, F_OK) >= 0);
assert_se(access(f, F_OK) < 0);
}
}
static void test_rmdir_parents(void) {
char *temp;
log_info("/* %s */", __func__);
temp = strjoina(arg_test_dir ?: "/tmp", "/test-rmdir.XXXXXX");
assert_se(mkdtemp(temp));
test_rmdir_parents_one(temp, "/aaa/../hoge/foo", "/hoge/foo", -EINVAL, NULL, NULL);
test_rmdir_parents_one(temp, "/aaa/bbb/ccc", "/hoge/../aaa", -EINVAL, NULL, NULL);
test_rmdir_parents_one(temp, "/aaa/bbb/ccc/ddd/eee", "/aaa/bbb/ccc/ddd", 0, "/aaa/bbb/ccc/ddd", "/eee");
test_rmdir_parents_one(temp, "/aaa/bbb/ccc/ddd/eee", "/aaa/bbb/ccc", 0, "/aaa/bbb/ccc", "/ddd");
test_rmdir_parents_one(temp, "/aaa/bbb/ccc/ddd/eee", "/aaa/bbb", 0, "/aaa/bbb", "/ccc");
test_rmdir_parents_one(temp, "/aaa/bbb/ccc/ddd/eee", "/aaa", 0, "/aaa", "/bbb");
test_rmdir_parents_one(temp, "/aaa/bbb/ccc/ddd/eee", "/", 0, "/", "/aaa");
test_rmdir_parents_one(temp, "/aaa/bbb/ccc/ddd/eee", "/aaa/hoge/foo", 0, "/aaa", "/bbb");
test_rmdir_parents_one(temp, "/aaa////bbb/.//ccc//ddd/eee///./.", "///././aaa/.", 0, "/aaa", "/bbb");
assert_se(rm_rf(temp, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
}
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
test_setup_logging(LOG_INFO); test_setup_logging(LOG_INFO);
@ -883,6 +937,7 @@ int main(int argc, char *argv[]) {
test_rename_noreplace(); test_rename_noreplace();
test_chmod_and_chown(); test_chmod_and_chown();
test_conservative_rename(); test_conservative_rename();
test_rmdir_parents();
return 0; return 0;
} }