aboutsummaryrefslogtreecommitdiff
path: root/bin/gensquashfs/src
diff options
context:
space:
mode:
authorDavid Oberhollenzer <david.oberhollenzer@sigma-star.at>2023-04-10 16:36:12 +0200
committerDavid Oberhollenzer <david.oberhollenzer@sigma-star.at>2023-04-12 20:41:34 +0200
commit72674fc880fa3a5e121e07f9dcdf6f79437a3011 (patch)
treebceed63ef051bffc68808ffd19df03ab17085a2e /bin/gensquashfs/src
parent58c9982e75c1b23b9c3ff9aad8295035866a67f0 (diff)
Split out Windows directory iteration code to a dir_iterator_t type
Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>
Diffstat (limited to 'bin/gensquashfs/src')
-rw-r--r--bin/gensquashfs/src/fstree_from_dir.c178
-rw-r--r--bin/gensquashfs/src/mkfs.h1
2 files changed, 49 insertions, 130 deletions
diff --git a/bin/gensquashfs/src/fstree_from_dir.c b/bin/gensquashfs/src/fstree_from_dir.c
index d117435..2ece45a 100644
--- a/bin/gensquashfs/src/fstree_from_dir.c
+++ b/bin/gensquashfs/src/fstree_from_dir.c
@@ -13,48 +13,28 @@
#include <errno.h>
#if defined(_WIN32) || defined(__WINDOWS__)
-#define UNIX_EPOCH_ON_W32 11644473600UL
-#define W32_TICS_PER_SEC 10000000UL
-
-static sqfs_u32 w32time_to_sqfs_time(const FILETIME *ft)
-{
- sqfs_u64 w32ts;
-
- w32ts = ft->dwHighDateTime;
- w32ts <<= 32UL;
- w32ts |= ft->dwLowDateTime;
-
- w32ts /= W32_TICS_PER_SEC;
-
- if (w32ts <= UNIX_EPOCH_ON_W32)
- return 0;
-
- w32ts -= UNIX_EPOCH_ON_W32;
-
- return (w32ts < 0x0FFFFFFFFUL) ? w32ts : 0xFFFFFFFF;
-}
-
static int add_node(fstree_t *fs, tree_node_t *root,
scan_node_callback cb, void *user,
unsigned int flags,
- const LPWIN32_FIND_DATAW entry)
+ const dir_entry_t *entry)
{
+ size_t length = strlen(entry->name);
tree_node_t *n;
- DWORD length;
- if (entry->cFileName[0] == '.') {
- if (entry->cFileName[1] == '\0')
+ if (entry->name[0] == '.') {
+ if (length == 1)
return 0;
- if (entry->cFileName[1] == '.' && entry->cFileName[2] == '\0')
+ if (entry->name[1] == '.' && length == 2)
return 0;
}
- length = WideCharToMultiByte(CP_UTF8, 0, entry->cFileName,
- -1, NULL, 0, NULL, NULL);
- if (length <= 0) {
- w32_perror("converting path to UTF-8");
- return -1;
+ if (S_ISDIR(entry->mode)) {
+ if (flags & DIR_SCAN_NO_DIR)
+ return 0;
+ } else {
+ if (flags & DIR_SCAN_NO_FILE)
+ return 0;
}
n = calloc(1, sizeof(*n) + length + 1);
@@ -63,25 +43,9 @@ static int add_node(fstree_t *fs, tree_node_t *root,
return -1;
}
+ n->mode = entry->mode;
n->name = (char *)n->payload;
- WideCharToMultiByte(CP_UTF8, 0, entry->cFileName, -1,
- n->name, length + 1, NULL, NULL);
-
- if (entry->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
- if (flags & DIR_SCAN_NO_DIR) {
- free(n);
- return 0;
- }
-
- n->mode = S_IFDIR | 0755;
- } else {
- if (flags & DIR_SCAN_NO_FILE) {
- free(n);
- return 0;
- }
-
- n->mode = S_IFREG | 0644;
- }
+ memcpy(n->name, entry->name, length);
if (cb != NULL) {
int ret = cb(user, fs, n);
@@ -93,7 +57,13 @@ static int add_node(fstree_t *fs, tree_node_t *root,
}
if (flags & DIR_SCAN_KEEP_TIME) {
- n->mod_time = w32time_to_sqfs_time(&(entry->ftLastWriteTime));
+ if (entry->mtime < 0) {
+ n->mod_time = 0;
+ } else if (entry->mtime > 0x0FFFFFFFFLL) {
+ n->mod_time = 0xFFFFFFFF;
+ } else {
+ n->mod_time = entry->mtime;
+ }
} else {
n->mod_time = fs->defaults.mtime;
}
@@ -103,81 +73,49 @@ static int add_node(fstree_t *fs, tree_node_t *root,
}
static int scan_dir(fstree_t *fs, tree_node_t *root,
- const char *path, const WCHAR *wpath,
+ const char *path,
scan_node_callback cb, void *user,
unsigned int flags)
{
- WIN32_FIND_DATAW entry;
- HANDLE dirhnd;
+ dir_iterator_t *it = dir_iterator_create(path);
- dirhnd = FindFirstFileW(wpath, &entry);
+ if (it == NULL)
+ return -1;
- if (dirhnd == INVALID_HANDLE_VALUE)
- goto fail_perror;
+ for (;;) {
+ dir_entry_t *ent;
+ int ret;
- do {
- if (add_node(fs, root, cb, user, flags, &entry))
- goto fail;
- } while (FindNextFileW(dirhnd, &entry));
+ ret = it->next(it, &ent);
+ if (ret > 0)
+ break;
+ if (ret < 0) {
+ sqfs_perror(path, "reading directory entry", ret);
+ sqfs_drop(it);
+ return -1;
+ }
- if (GetLastError() != ERROR_NO_MORE_FILES)
- goto fail_perror;
+ ret = add_node(fs, root, cb, user, flags, ent);
+ free(ent);
+ if (ret != 0) {
+ sqfs_drop(it);
+ return -1;
+ }
+ }
- FindClose(dirhnd);
+ sqfs_drop(it);
return 0;
-fail_perror:
- w32_perror(path);
-fail:
- if (dirhnd != INVALID_HANDLE_VALUE)
- FindClose(dirhnd);
- return -1;
}
int fstree_from_dir(fstree_t *fs, tree_node_t *root,
const char *path, scan_node_callback cb,
void *user, unsigned int flags)
{
- WCHAR *wpath = NULL, *new = NULL;
- size_t len, newlen;
tree_node_t *n;
- /* path -> to_wchar(path) + L"\*" */
- wpath = path_to_windows(path);
- if (wpath == NULL) {
- fprintf(stderr, "%s: allocation failure.\n", path);
+ if (scan_dir(fs, root, path, cb, user, flags))
return -1;
- }
-
- for (len = 0; wpath[len] != '\0'; ++len)
- ;
-
- newlen = len + 1;
-
- if (len > 0 && wpath[len - 1] != '\\')
- newlen += 1;
-
- new = realloc(wpath, sizeof(wpath[0]) * (newlen + 1));
- if (new == NULL) {
- fprintf(stderr, "%s: allocation failure.\n", path);
- goto fail;
- }
-
- wpath = new;
- if (len > 0 && wpath[len - 1] != '\\')
- wpath[len++] = '\\';
-
- wpath[len++] = '*';
- wpath[len++] = '\0';
-
- /* scan directory contents */
- if (scan_dir(fs, root, path, wpath, cb, user, flags))
- goto fail;
-
- free(wpath);
- wpath = NULL;
-
- /* recursion step */
if (flags & DIR_SCAN_NO_RECURSION)
return 0;
@@ -190,9 +128,6 @@ int fstree_from_dir(fstree_t *fs, tree_node_t *root,
}
return 0;
-fail:
- free(wpath);
- return -1;
}
int fstree_from_subdir(fstree_t *fs, tree_node_t *root,
@@ -200,8 +135,7 @@ int fstree_from_subdir(fstree_t *fs, tree_node_t *root,
scan_node_callback cb, void *user,
unsigned int flags)
{
- size_t len, plen, slen;
- WCHAR *wpath = NULL;
+ size_t plen, slen;
char *temp = NULL;
tree_node_t *n;
@@ -211,9 +145,7 @@ int fstree_from_subdir(fstree_t *fs, tree_node_t *root,
if (slen == 0)
return fstree_from_dir(fs, root, path, cb, user, flags);
- len = plen + 1 + slen + 2;
-
- temp = calloc(1, len + 1);
+ temp = calloc(1, plen + 1 + slen + 1);
if (temp == NULL) {
fprintf(stderr, "%s/%s: allocation failure.\n", path, subdir);
return -1;
@@ -222,29 +154,16 @@ int fstree_from_subdir(fstree_t *fs, tree_node_t *root,
memcpy(temp, path, plen);
temp[plen] = '/';
memcpy(temp + plen + 1, subdir, slen);
- temp[plen + 1 + slen ] = '/';
- temp[plen + 1 + slen + 1] = '*';
- temp[plen + 1 + slen + 2] = '\0';
-
- wpath = path_to_windows(temp);
- if (wpath == NULL) {
- fprintf(stderr, "%s: allocation failure.\n", temp);
- goto fail;
- }
+ temp[plen + 1 + slen] = '\0';
- if (scan_dir(fs, root, temp, wpath, cb, user, flags))
+ if (scan_dir(fs, root, temp, cb, user, flags))
goto fail;
- free(wpath);
- wpath = NULL;
-
if (flags & DIR_SCAN_NO_RECURSION) {
free(temp);
return 0;
}
- temp[plen + 1 + slen] = '\0';
-
for (n = root->data.dir.children; n != NULL; n = n->next) {
if (!S_ISDIR(n->mode))
continue;
@@ -257,7 +176,6 @@ int fstree_from_subdir(fstree_t *fs, tree_node_t *root,
return 0;
fail:
free(temp);
- free(wpath);
return -1;
}
diff --git a/bin/gensquashfs/src/mkfs.h b/bin/gensquashfs/src/mkfs.h
index 516ab87..6b69465 100644
--- a/bin/gensquashfs/src/mkfs.h
+++ b/bin/gensquashfs/src/mkfs.h
@@ -12,6 +12,7 @@
#include "common.h"
#include "fstree.h"
+#include "util/dir_iterator.h"
#include "util/util.h"
#include "io/file.h"