1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
* dirscan.c
*
* Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
*/
#include "mkfs.h"
#ifdef _WIN32
int fstree_from_dir(fstree_t *fs, const char *path, unsigned int flags)
{
(void)fs; (void)path; (void)flags;
fputs("Packing a directory is not supported on Windows.\n", stderr);
return -1;
}
#else
static int populate_dir(int dir_fd, fstree_t *fs, tree_node_t *root,
dev_t devstart, unsigned int flags)
{
char *extra = NULL;
struct dirent *ent;
struct stat sb;
tree_node_t *n;
int childfd;
DIR *dir;
dir = fdopendir(dir_fd);
if (dir == NULL) {
perror("fdopendir");
close(dir_fd);
return -1;
}
/* XXX: fdopendir can dup and close dir_fd internally
and still be compliant with the spec. */
dir_fd = dirfd(dir);
for (;;) {
errno = 0;
ent = readdir(dir);
if (ent == NULL) {
if (errno) {
perror("readdir");
goto fail;
}
break;
}
if (!strcmp(ent->d_name, "..") || !strcmp(ent->d_name, "."))
continue;
if (fstatat(dir_fd, ent->d_name, &sb, AT_SYMLINK_NOFOLLOW)) {
perror(ent->d_name);
goto fail;
}
if ((flags & DIR_SCAN_ONE_FILESYSTEM) && sb.st_dev != devstart)
continue;
if (S_ISLNK(sb.st_mode)) {
extra = calloc(1, sb.st_size + 1);
if (extra == NULL)
goto fail_rdlink;
if (readlinkat(dir_fd, ent->d_name,
extra, sb.st_size) < 0) {
goto fail_rdlink;
}
extra[sb.st_size] = '\0';
}
if (!(flags & DIR_SCAN_KEEP_TIME))
sb.st_mtime = fs->defaults.st_mtime;
n = fstree_mknode(root, ent->d_name, strlen(ent->d_name),
extra, &sb);
if (n == NULL) {
perror("creating tree node");
goto fail;
}
free(extra);
extra = NULL;
if (S_ISDIR(n->mode)) {
childfd = openat(dir_fd, n->name, O_DIRECTORY |
O_RDONLY | O_CLOEXEC);
if (childfd < 0) {
perror(n->name);
goto fail;
}
if (populate_dir(childfd, fs, n, devstart, flags))
goto fail;
}
}
closedir(dir);
return 0;
fail_rdlink:
perror("readlink");
fail:
closedir(dir);
free(extra);
return -1;
}
int fstree_from_dir(fstree_t *fs, const char *path, unsigned int flags)
{
struct stat sb;
int fd;
fd = open(path, O_DIRECTORY | O_RDONLY | O_CLOEXEC);
if (fd < 0) {
perror(path);
return -1;
}
if (fstat(fd, &sb)) {
perror(path);
close(fd);
return -1;
}
return populate_dir(fd, fs, fs->root, sb.st_dev, flags);
}
#endif
|
