aboutsummaryrefslogtreecommitdiff
path: root/lib/util
diff options
context:
space:
mode:
Diffstat (limited to 'lib/util')
-rw-r--r--lib/util/src/dir_tree_iterator.c20
-rw-r--r--lib/util/test/dir_tree_iterator.c51
2 files changed, 69 insertions, 2 deletions
diff --git a/lib/util/src/dir_tree_iterator.c b/lib/util/src/dir_tree_iterator.c
index 775039d..55fbabb 100644
--- a/lib/util/src/dir_tree_iterator.c
+++ b/lib/util/src/dir_tree_iterator.c
@@ -120,6 +120,23 @@ static dir_entry_t *expand_path(const dir_tree_iterator_t *it, dir_entry_t *ent)
return ent;
}
+static void apply_changes(const dir_tree_iterator_t *it, dir_entry_t *ent)
+{
+ if (!(it->cfg.flags & DIR_SCAN_KEEP_TIME))
+ ent->mtime = it->cfg.def_mtime;
+
+ if (!(it->cfg.flags & DIR_SCAN_KEEP_UID))
+ ent->uid = it->cfg.def_uid;
+
+ if (!(it->cfg.flags & DIR_SCAN_KEEP_GID))
+ ent->gid = it->cfg.def_gid;
+
+ if (!(it->cfg.flags & DIR_SCAN_KEEP_MODE)) {
+ ent->mode &= ~(07777);
+ ent->mode |= it->cfg.def_mode & 07777;
+ }
+}
+
/*****************************************************************************/
static void destroy(sqfs_object_t *obj)
@@ -176,8 +193,7 @@ retry:
return it->state;
}
- if (!(it->cfg.flags & DIR_SCAN_KEEP_TIME))
- ent->mtime = it->cfg.def_mtime;
+ apply_changes(it, ent);
if (S_ISDIR(ent->mode)) {
if (!(it->cfg.flags & DIR_SCAN_NO_RECURSION)) {
diff --git a/lib/util/test/dir_tree_iterator.c b/lib/util/test/dir_tree_iterator.c
index 7c1aa88..a45cc30 100644
--- a/lib/util/test/dir_tree_iterator.c
+++ b/lib/util/test/dir_tree_iterator.c
@@ -29,6 +29,9 @@ int main(int argc, char **argv)
(void)argc; (void)argv;
memset(&cfg, 0, sizeof(cfg));
+ cfg.def_mtime = 1337;
+ cfg.def_uid = 42;
+ cfg.def_gid = 23;
dir = dir_tree_iterator_create(TEST_PATH, &cfg);
TEST_NOT_NULL(dir);
@@ -54,36 +57,84 @@ int main(int argc, char **argv)
TEST_STR_EQUAL(ent[0]->name, "dira");
TEST_ASSERT(S_ISDIR(ent[0]->mode));
+ TEST_EQUAL_UI(ent[0]->mtime, 1337);
+ TEST_EQUAL_UI(ent[0]->uid, 42);
+ TEST_EQUAL_UI(ent[0]->gid, 23);
TEST_STR_EQUAL(ent[1]->name, "dira/file_a0");
TEST_ASSERT(S_ISREG(ent[1]->mode));
+ TEST_EQUAL_UI(ent[1]->mtime, 1337);
+ TEST_EQUAL_UI(ent[1]->uid, 42);
+ TEST_EQUAL_UI(ent[1]->gid, 23);
TEST_STR_EQUAL(ent[2]->name, "dira/file_a1");
TEST_ASSERT(S_ISREG(ent[2]->mode));
+ TEST_EQUAL_UI(ent[2]->mtime, 1337);
+ TEST_EQUAL_UI(ent[2]->uid, 42);
+ TEST_EQUAL_UI(ent[2]->gid, 23);
TEST_STR_EQUAL(ent[3]->name, "dira/file_a2");
TEST_ASSERT(S_ISREG(ent[3]->mode));
+ TEST_EQUAL_UI(ent[3]->mtime, 1337);
+ TEST_EQUAL_UI(ent[3]->uid, 42);
+ TEST_EQUAL_UI(ent[3]->gid, 23);
TEST_STR_EQUAL(ent[4]->name, "dirb");
TEST_ASSERT(S_ISDIR(ent[4]->mode));
+ TEST_EQUAL_UI(ent[4]->mtime, 1337);
+ TEST_EQUAL_UI(ent[4]->uid, 42);
+ TEST_EQUAL_UI(ent[4]->gid, 23);
TEST_STR_EQUAL(ent[5]->name, "dirb/dirx");
TEST_ASSERT(S_ISDIR(ent[5]->mode));
+ TEST_EQUAL_UI(ent[5]->mtime, 1337);
+ TEST_EQUAL_UI(ent[5]->uid, 42);
+ TEST_EQUAL_UI(ent[5]->gid, 23);
TEST_STR_EQUAL(ent[6]->name, "dirb/dirx/file_x0");
TEST_ASSERT(S_ISREG(ent[6]->mode));
+ TEST_EQUAL_UI(ent[6]->mtime, 1337);
+ TEST_EQUAL_UI(ent[6]->uid, 42);
+ TEST_EQUAL_UI(ent[6]->gid, 23);
TEST_STR_EQUAL(ent[7]->name, "dirb/dirx/file_x1");
TEST_ASSERT(S_ISREG(ent[7]->mode));
+ TEST_EQUAL_UI(ent[7]->mtime, 1337);
+ TEST_EQUAL_UI(ent[7]->uid, 42);
+ TEST_EQUAL_UI(ent[7]->gid, 23);
TEST_STR_EQUAL(ent[8]->name, "dirb/dirx/file_x2");
TEST_ASSERT(S_ISREG(ent[8]->mode));
+ TEST_EQUAL_UI(ent[8]->mtime, 1337);
+ TEST_EQUAL_UI(ent[8]->uid, 42);
+ TEST_EQUAL_UI(ent[8]->gid, 23);
TEST_STR_EQUAL(ent[9]->name, "dirb/file_b0");
TEST_ASSERT(S_ISREG(ent[9]->mode));
+ TEST_EQUAL_UI(ent[9]->mtime, 1337);
+ TEST_EQUAL_UI(ent[9]->uid, 42);
+ TEST_EQUAL_UI(ent[9]->gid, 23);
TEST_STR_EQUAL(ent[10]->name, "dirb/file_b1");
TEST_ASSERT(S_ISREG(ent[10]->mode));
+ TEST_EQUAL_UI(ent[10]->mtime, 1337);
+ TEST_EQUAL_UI(ent[10]->uid, 42);
+ TEST_EQUAL_UI(ent[10]->gid, 23);
TEST_STR_EQUAL(ent[11]->name, "dirb/file_b2");
TEST_ASSERT(S_ISREG(ent[11]->mode));
+ TEST_EQUAL_UI(ent[11]->mtime, 1337);
+ TEST_EQUAL_UI(ent[11]->uid, 42);
+ TEST_EQUAL_UI(ent[11]->gid, 23);
TEST_STR_EQUAL(ent[12]->name, "dirc");
TEST_ASSERT(S_ISDIR(ent[12]->mode));
+ TEST_EQUAL_UI(ent[12]->mtime, 1337);
+ TEST_EQUAL_UI(ent[12]->uid, 42);
+ TEST_EQUAL_UI(ent[12]->gid, 23);
TEST_STR_EQUAL(ent[13]->name, "dirc/file_c0");
TEST_ASSERT(S_ISREG(ent[13]->mode));
+ TEST_EQUAL_UI(ent[13]->mtime, 1337);
+ TEST_EQUAL_UI(ent[13]->uid, 42);
+ TEST_EQUAL_UI(ent[13]->gid, 23);
TEST_STR_EQUAL(ent[14]->name, "dirc/file_c1");
TEST_ASSERT(S_ISREG(ent[14]->mode));
+ TEST_EQUAL_UI(ent[14]->mtime, 1337);
+ TEST_EQUAL_UI(ent[14]->uid, 42);
+ TEST_EQUAL_UI(ent[14]->gid, 23);
TEST_STR_EQUAL(ent[15]->name, "dirc/file_c2");
TEST_ASSERT(S_ISREG(ent[15]->mode));
+ TEST_EQUAL_UI(ent[15]->mtime, 1337);
+ TEST_EQUAL_UI(ent[15]->uid, 42);
+ TEST_EQUAL_UI(ent[15]->gid, 23);
for (i = 0; i < 16; ++i)
free(ent[i]);