]>
Commit | Line | Data |
---|---|---|
8f8d8a57 DW |
1 | #include <time.h> |
2 | #include <unistd.h> | |
3 | ||
0d4deba2 RY |
4 | #include "create_inode.h" |
5 | ||
052064b9 RY |
6 | #if __STDC_VERSION__ < 199901L |
7 | # if __GNUC__ >= 2 | |
8 | # define __func__ __FUNCTION__ | |
9 | # else | |
10 | # define __func__ "<unknown>" | |
11 | # endif | |
12 | #endif | |
13 | ||
ce600dc4 RY |
14 | /* 64KiB is the minimium blksize to best minimize system call overhead. */ |
15 | #ifndef IO_BUFSIZE | |
16 | #define IO_BUFSIZE 64*1024 | |
17 | #endif | |
18 | ||
19 | /* Block size for `st_blocks' */ | |
20 | #ifndef S_BLKSIZE | |
21 | #define S_BLKSIZE 512 | |
22 | #endif | |
23 | ||
f84894bc | 24 | /* Link an inode number to a directory */ |
a3111e80 DW |
25 | static errcode_t add_link(ext2_filsys fs, ext2_ino_t parent_ino, |
26 | ext2_ino_t ino, const char *name) | |
f84894bc RY |
27 | { |
28 | struct ext2_inode inode; | |
29 | errcode_t retval; | |
30 | ||
a3111e80 | 31 | retval = ext2fs_read_inode(fs, ino, &inode); |
f84894bc RY |
32 | if (retval) { |
33 | com_err(__func__, retval, "while reading inode %u", ino); | |
34 | return retval; | |
35 | } | |
36 | ||
a3111e80 | 37 | retval = ext2fs_link(fs, parent_ino, name, ino, inode.i_flags); |
f84894bc | 38 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 39 | retval = ext2fs_expand_dir(fs, parent_ino); |
f84894bc RY |
40 | if (retval) { |
41 | com_err(__func__, retval, "while expanding directory"); | |
42 | return retval; | |
43 | } | |
a3111e80 | 44 | retval = ext2fs_link(fs, parent_ino, name, ino, inode.i_flags); |
f84894bc RY |
45 | } |
46 | if (retval) { | |
47 | com_err(__func__, retval, "while linking %s", name); | |
48 | return retval; | |
49 | } | |
50 | ||
51 | inode.i_links_count++; | |
52 | ||
a3111e80 | 53 | retval = ext2fs_write_inode(fs, ino, &inode); |
f84894bc RY |
54 | if (retval) |
55 | com_err(__func__, retval, "while writing inode %u", ino); | |
56 | ||
57 | return retval; | |
58 | } | |
59 | ||
b6960654 RY |
60 | /* Fill the uid, gid, mode and time for the inode */ |
61 | static void fill_inode(struct ext2_inode *inode, struct stat *st) | |
62 | { | |
63 | if (st != NULL) { | |
64 | inode->i_uid = st->st_uid; | |
65 | inode->i_gid = st->st_gid; | |
66 | inode->i_mode |= st->st_mode; | |
67 | inode->i_atime = st->st_atime; | |
68 | inode->i_mtime = st->st_mtime; | |
69 | inode->i_ctime = st->st_ctime; | |
70 | } | |
71 | } | |
72 | ||
73 | /* Set the uid, gid, mode and time for the inode */ | |
a3111e80 DW |
74 | static errcode_t set_inode_extra(ext2_filsys fs, ext2_ino_t cwd, |
75 | ext2_ino_t ino, struct stat *st) | |
b6960654 RY |
76 | { |
77 | errcode_t retval; | |
78 | struct ext2_inode inode; | |
79 | ||
a3111e80 | 80 | retval = ext2fs_read_inode(fs, ino, &inode); |
b6960654 RY |
81 | if (retval) { |
82 | com_err(__func__, retval, "while reading inode %u", ino); | |
83 | return retval; | |
84 | } | |
85 | ||
86 | fill_inode(&inode, st); | |
87 | ||
a3111e80 | 88 | retval = ext2fs_write_inode(fs, ino, &inode); |
3aec816f | 89 | if (retval) |
b6960654 | 90 | com_err(__func__, retval, "while writing inode %u", ino); |
3aec816f | 91 | return retval; |
b6960654 RY |
92 | } |
93 | ||
c61e0068 | 94 | /* Make a special files (block and character devices), fifo's, and sockets */ |
a3111e80 DW |
95 | errcode_t do_mknod_internal(ext2_filsys fs, ext2_ino_t cwd, const char *name, |
96 | struct stat *st) | |
0d4deba2 | 97 | { |
b2346118 RY |
98 | ext2_ino_t ino; |
99 | errcode_t retval; | |
100 | struct ext2_inode inode; | |
101 | unsigned long major, minor, mode; | |
102 | int filetype; | |
103 | ||
104 | switch(st->st_mode & S_IFMT) { | |
9c891f7e DW |
105 | case S_IFCHR: |
106 | mode = LINUX_S_IFCHR; | |
107 | filetype = EXT2_FT_CHRDEV; | |
108 | break; | |
109 | case S_IFBLK: | |
110 | mode = LINUX_S_IFBLK; | |
111 | filetype = EXT2_FT_BLKDEV; | |
112 | break; | |
113 | case S_IFIFO: | |
114 | mode = LINUX_S_IFIFO; | |
115 | filetype = EXT2_FT_FIFO; | |
116 | break; | |
c61e0068 DW |
117 | case S_IFSOCK: |
118 | mode = LINUX_S_IFSOCK; | |
119 | filetype = EXT2_FT_SOCK; | |
3aec816f TT |
120 | default: |
121 | abort(); | |
122 | /* NOTREACHED */ | |
b2346118 RY |
123 | } |
124 | ||
a3111e80 | 125 | if (!(fs->flags & EXT2_FLAG_RW)) { |
b2346118 | 126 | com_err(__func__, 0, "Filesystem opened read/only"); |
c61e0068 | 127 | return EROFS; |
b2346118 | 128 | } |
a3111e80 | 129 | retval = ext2fs_new_inode(fs, cwd, 010755, 0, &ino); |
b2346118 RY |
130 | if (retval) { |
131 | com_err(__func__, retval, 0); | |
132 | return retval; | |
133 | } | |
134 | ||
135 | #ifdef DEBUGFS | |
136 | printf("Allocated inode: %u\n", ino); | |
137 | #endif | |
a3111e80 | 138 | retval = ext2fs_link(fs, cwd, name, ino, filetype); |
b2346118 | 139 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 140 | retval = ext2fs_expand_dir(fs, cwd); |
b2346118 RY |
141 | if (retval) { |
142 | com_err(__func__, retval, "while expanding directory"); | |
143 | return retval; | |
144 | } | |
a3111e80 | 145 | retval = ext2fs_link(fs, cwd, name, ino, filetype); |
b2346118 RY |
146 | } |
147 | if (retval) { | |
148 | com_err(name, retval, 0); | |
149 | return -1; | |
150 | } | |
a3111e80 | 151 | if (ext2fs_test_inode_bitmap2(fs->inode_map, ino)) |
b2346118 | 152 | com_err(__func__, 0, "Warning: inode already set"); |
a3111e80 | 153 | ext2fs_inode_alloc_stats2(fs, ino, +1, 0); |
b2346118 RY |
154 | memset(&inode, 0, sizeof(inode)); |
155 | inode.i_mode = mode; | |
156 | inode.i_atime = inode.i_ctime = inode.i_mtime = | |
a3111e80 | 157 | fs->now ? fs->now : time(0); |
b2346118 RY |
158 | |
159 | major = major(st->st_rdev); | |
160 | minor = minor(st->st_rdev); | |
161 | ||
162 | if ((major < 256) && (minor < 256)) { | |
163 | inode.i_block[0] = major * 256 + minor; | |
164 | inode.i_block[1] = 0; | |
165 | } else { | |
166 | inode.i_block[0] = 0; | |
9c891f7e DW |
167 | inode.i_block[1] = (minor & 0xff) | (major << 8) | |
168 | ((minor & ~0xff) << 12); | |
b2346118 RY |
169 | } |
170 | inode.i_links_count = 1; | |
171 | ||
a3111e80 | 172 | retval = ext2fs_write_new_inode(fs, ino, &inode); |
b2346118 RY |
173 | if (retval) |
174 | com_err(__func__, retval, "while creating inode %u", ino); | |
175 | ||
176 | return retval; | |
0d4deba2 RY |
177 | } |
178 | ||
179 | /* Make a symlink name -> target */ | |
a3111e80 DW |
180 | errcode_t do_symlink_internal(ext2_filsys fs, ext2_ino_t cwd, const char *name, |
181 | char *target, ext2_ino_t root) | |
0d4deba2 | 182 | { |
b25ddc5e RY |
183 | char *cp; |
184 | ext2_ino_t parent_ino; | |
185 | errcode_t retval; | |
186 | struct ext2_inode inode; | |
187 | struct stat st; | |
188 | ||
189 | cp = strrchr(name, '/'); | |
190 | if (cp) { | |
191 | *cp = 0; | |
a3111e80 | 192 | retval = ext2fs_namei(fs, root, cwd, name, &parent_ino); |
8f8d8a57 | 193 | if (retval) { |
b25ddc5e RY |
194 | com_err(name, retval, 0); |
195 | return retval; | |
196 | } | |
197 | name = cp+1; | |
198 | } else | |
199 | parent_ino = cwd; | |
200 | ||
201 | try_again: | |
a3111e80 | 202 | retval = ext2fs_symlink(fs, parent_ino, 0, name, target); |
b25ddc5e | 203 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 204 | retval = ext2fs_expand_dir(fs, parent_ino); |
b25ddc5e | 205 | if (retval) { |
9c891f7e DW |
206 | com_err("do_symlink_internal", retval, |
207 | "while expanding directory"); | |
b25ddc5e RY |
208 | return retval; |
209 | } | |
210 | goto try_again; | |
211 | } | |
3aec816f | 212 | if (retval) |
b25ddc5e | 213 | com_err("ext2fs_symlink", retval, 0); |
3aec816f | 214 | return retval; |
0d4deba2 RY |
215 | } |
216 | ||
217 | /* Make a directory in the fs */ | |
a3111e80 DW |
218 | errcode_t do_mkdir_internal(ext2_filsys fs, ext2_ino_t cwd, const char *name, |
219 | struct stat *st, ext2_ino_t root) | |
0d4deba2 | 220 | { |
3068f03f RY |
221 | char *cp; |
222 | ext2_ino_t parent_ino, ino; | |
223 | errcode_t retval; | |
224 | struct ext2_inode inode; | |
225 | ||
226 | ||
227 | cp = strrchr(name, '/'); | |
228 | if (cp) { | |
229 | *cp = 0; | |
a3111e80 | 230 | retval = ext2fs_namei(fs, root, cwd, name, &parent_ino); |
8f8d8a57 | 231 | if (retval) { |
3068f03f RY |
232 | com_err(name, retval, 0); |
233 | return retval; | |
234 | } | |
235 | name = cp+1; | |
236 | } else | |
237 | parent_ino = cwd; | |
238 | ||
239 | try_again: | |
a3111e80 | 240 | retval = ext2fs_mkdir(fs, parent_ino, 0, name); |
3068f03f | 241 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 242 | retval = ext2fs_expand_dir(fs, parent_ino); |
3068f03f RY |
243 | if (retval) { |
244 | com_err(__func__, retval, "while expanding directory"); | |
245 | return retval; | |
246 | } | |
247 | goto try_again; | |
248 | } | |
3aec816f | 249 | if (retval) |
3068f03f | 250 | com_err("ext2fs_mkdir", retval, 0); |
3aec816f | 251 | return retval; |
0d4deba2 RY |
252 | } |
253 | ||
a3111e80 DW |
254 | static errcode_t copy_file(ext2_filsys fs, int fd, ext2_ino_t newfile, |
255 | int bufsize, int make_holes) | |
ce600dc4 RY |
256 | { |
257 | ext2_file_t e2_file; | |
6bb88459 | 258 | errcode_t retval, close_ret; |
ce600dc4 RY |
259 | int got; |
260 | unsigned int written; | |
261 | char *buf; | |
262 | char *ptr; | |
263 | char *zero_buf; | |
264 | int cmp; | |
265 | ||
a3111e80 | 266 | retval = ext2fs_file_open(fs, newfile, |
ce600dc4 RY |
267 | EXT2_FILE_WRITE, &e2_file); |
268 | if (retval) | |
269 | return retval; | |
270 | ||
271 | retval = ext2fs_get_mem(bufsize, &buf); | |
272 | if (retval) { | |
273 | com_err("copy_file", retval, "can't allocate buffer\n"); | |
6bb88459 | 274 | goto out_close; |
ce600dc4 RY |
275 | } |
276 | ||
277 | /* This is used for checking whether the whole block is zero */ | |
278 | retval = ext2fs_get_memzero(bufsize, &zero_buf); | |
279 | if (retval) { | |
6bb88459 TT |
280 | com_err("copy_file", retval, "can't allocate zero buffer\n"); |
281 | goto out_free_buf; | |
ce600dc4 RY |
282 | } |
283 | ||
284 | while (1) { | |
285 | got = read(fd, buf, bufsize); | |
286 | if (got == 0) | |
287 | break; | |
288 | if (got < 0) { | |
289 | retval = errno; | |
290 | goto fail; | |
291 | } | |
292 | ptr = buf; | |
293 | ||
294 | /* Sparse copy */ | |
295 | if (make_holes) { | |
296 | /* Check whether all is zero */ | |
297 | cmp = memcmp(ptr, zero_buf, got); | |
298 | if (cmp == 0) { | |
299 | /* The whole block is zero, make a hole */ | |
9c891f7e DW |
300 | retval = ext2fs_file_lseek(e2_file, got, |
301 | EXT2_SEEK_CUR, | |
302 | NULL); | |
ce600dc4 RY |
303 | if (retval) |
304 | goto fail; | |
305 | got = 0; | |
306 | } | |
307 | } | |
308 | ||
309 | /* Normal copy */ | |
310 | while (got > 0) { | |
311 | retval = ext2fs_file_write(e2_file, ptr, | |
312 | got, &written); | |
313 | if (retval) | |
314 | goto fail; | |
315 | ||
316 | got -= written; | |
317 | ptr += written; | |
318 | } | |
319 | } | |
ce600dc4 RY |
320 | |
321 | fail: | |
ce600dc4 | 322 | ext2fs_free_mem(&zero_buf); |
6bb88459 TT |
323 | out_free_buf: |
324 | ext2fs_free_mem(&buf); | |
325 | out_close: | |
326 | close_ret = ext2fs_file_close(e2_file); | |
327 | if (retval == 0) | |
328 | retval = close_ret; | |
ce600dc4 RY |
329 | return retval; |
330 | } | |
331 | ||
b99888a0 | 332 | static int is_hardlink(struct hdlinks_s *hdlinks, dev_t dev, ino_t ino) |
f84894bc RY |
333 | { |
334 | int i; | |
335 | ||
b99888a0 DW |
336 | for (i = 0; i < hdlinks->count; i++) { |
337 | if (hdlinks->hdl[i].src_dev == dev && | |
338 | hdlinks->hdl[i].src_ino == ino) | |
f84894bc RY |
339 | return i; |
340 | } | |
341 | return -1; | |
342 | } | |
343 | ||
0d4deba2 | 344 | /* Copy the native file to the fs */ |
a3111e80 DW |
345 | errcode_t do_write_internal(ext2_filsys fs, ext2_ino_t cwd, const char *src, |
346 | const char *dest, ext2_ino_t root) | |
0d4deba2 | 347 | { |
ce600dc4 RY |
348 | int fd; |
349 | struct stat statbuf; | |
350 | ext2_ino_t newfile; | |
351 | errcode_t retval; | |
352 | struct ext2_inode inode; | |
353 | int bufsize = IO_BUFSIZE; | |
354 | int make_holes = 0; | |
355 | ||
356 | fd = open(src, O_RDONLY); | |
357 | if (fd < 0) { | |
358 | com_err(src, errno, 0); | |
359 | return errno; | |
360 | } | |
361 | if (fstat(fd, &statbuf) < 0) { | |
362 | com_err(src, errno, 0); | |
363 | close(fd); | |
364 | return errno; | |
365 | } | |
366 | ||
a3111e80 | 367 | retval = ext2fs_namei(fs, root, cwd, dest, &newfile); |
ce600dc4 | 368 | if (retval == 0) { |
ce600dc4 | 369 | close(fd); |
117b54c3 | 370 | return EXT2_ET_FILE_EXISTS; |
ce600dc4 RY |
371 | } |
372 | ||
a3111e80 | 373 | retval = ext2fs_new_inode(fs, cwd, 010755, 0, &newfile); |
ce600dc4 RY |
374 | if (retval) { |
375 | com_err(__func__, retval, 0); | |
376 | close(fd); | |
117b54c3 | 377 | return retval; |
ce600dc4 RY |
378 | } |
379 | #ifdef DEBUGFS | |
380 | printf("Allocated inode: %u\n", newfile); | |
381 | #endif | |
a3111e80 | 382 | retval = ext2fs_link(fs, cwd, dest, newfile, |
ce600dc4 RY |
383 | EXT2_FT_REG_FILE); |
384 | if (retval == EXT2_ET_DIR_NO_SPACE) { | |
a3111e80 | 385 | retval = ext2fs_expand_dir(fs, cwd); |
ce600dc4 RY |
386 | if (retval) { |
387 | com_err(__func__, retval, "while expanding directory"); | |
388 | close(fd); | |
117b54c3 | 389 | return retval; |
ce600dc4 | 390 | } |
a3111e80 | 391 | retval = ext2fs_link(fs, cwd, dest, newfile, |
ce600dc4 RY |
392 | EXT2_FT_REG_FILE); |
393 | } | |
394 | if (retval) { | |
395 | com_err(dest, retval, 0); | |
396 | close(fd); | |
397 | return errno; | |
398 | } | |
a3111e80 | 399 | if (ext2fs_test_inode_bitmap2(fs->inode_map, newfile)) |
ce600dc4 | 400 | com_err(__func__, 0, "Warning: inode already set"); |
a3111e80 | 401 | ext2fs_inode_alloc_stats2(fs, newfile, +1, 0); |
ce600dc4 RY |
402 | memset(&inode, 0, sizeof(inode)); |
403 | inode.i_mode = (statbuf.st_mode & ~LINUX_S_IFMT) | LINUX_S_IFREG; | |
404 | inode.i_atime = inode.i_ctime = inode.i_mtime = | |
a3111e80 | 405 | fs->now ? fs->now : time(0); |
ce600dc4 RY |
406 | inode.i_links_count = 1; |
407 | inode.i_size = statbuf.st_size; | |
a3111e80 | 408 | if (EXT2_HAS_INCOMPAT_FEATURE(fs->super, |
d23b1965 TT |
409 | EXT4_FEATURE_INCOMPAT_INLINE_DATA)) { |
410 | inode.i_flags |= EXT4_INLINE_DATA_FL; | |
a3111e80 | 411 | } else if (fs->super->s_feature_incompat & |
d23b1965 | 412 | EXT3_FEATURE_INCOMPAT_EXTENTS) { |
ce600dc4 RY |
413 | int i; |
414 | struct ext3_extent_header *eh; | |
415 | ||
416 | eh = (struct ext3_extent_header *) &inode.i_block[0]; | |
417 | eh->eh_depth = 0; | |
418 | eh->eh_entries = 0; | |
419 | eh->eh_magic = ext2fs_cpu_to_le16(EXT3_EXT_MAGIC); | |
420 | i = (sizeof(inode.i_block) - sizeof(*eh)) / | |
421 | sizeof(struct ext3_extent); | |
422 | eh->eh_max = ext2fs_cpu_to_le16(i); | |
423 | inode.i_flags |= EXT4_EXTENTS_FL; | |
424 | } | |
425 | ||
a3111e80 | 426 | retval = ext2fs_write_new_inode(fs, newfile, &inode); |
8f8d8a57 | 427 | if (retval) { |
ce600dc4 RY |
428 | com_err(__func__, retval, "while creating inode %u", newfile); |
429 | close(fd); | |
117b54c3 | 430 | return retval; |
ce600dc4 | 431 | } |
d23b1965 | 432 | if (inode.i_flags & EXT4_INLINE_DATA_FL) { |
a3111e80 | 433 | retval = ext2fs_inline_data_init(fs, newfile); |
117b54c3 DW |
434 | if (retval) { |
435 | com_err("copy_file", retval, 0); | |
436 | close(fd); | |
437 | return retval; | |
438 | } | |
d23b1965 | 439 | } |
ce600dc4 RY |
440 | if (LINUX_S_ISREG(inode.i_mode)) { |
441 | if (statbuf.st_blocks < statbuf.st_size / S_BLKSIZE) { | |
442 | make_holes = 1; | |
443 | /* | |
444 | * Use I/O blocksize as buffer size when | |
445 | * copying sparse files. | |
446 | */ | |
447 | bufsize = statbuf.st_blksize; | |
448 | } | |
a3111e80 | 449 | retval = copy_file(fs, fd, newfile, bufsize, make_holes); |
ce600dc4 RY |
450 | if (retval) |
451 | com_err("copy_file", retval, 0); | |
452 | } | |
453 | close(fd); | |
454 | ||
117b54c3 | 455 | return retval; |
0d4deba2 RY |
456 | } |
457 | ||
458 | /* Copy files from source_dir to fs */ | |
b99888a0 DW |
459 | static errcode_t __populate_fs(ext2_filsys fs, ext2_ino_t parent_ino, |
460 | const char *source_dir, ext2_ino_t root, | |
461 | struct hdlinks_s *hdlinks) | |
0d4deba2 | 462 | { |
052064b9 RY |
463 | const char *name; |
464 | DIR *dh; | |
465 | struct dirent *dent; | |
466 | struct stat st; | |
467 | char ln_target[PATH_MAX]; | |
f84894bc | 468 | unsigned int save_inode; |
052064b9 | 469 | ext2_ino_t ino; |
1bad6f46 | 470 | errcode_t retval = 0; |
052064b9 | 471 | int read_cnt; |
f84894bc | 472 | int hdlink; |
052064b9 | 473 | |
052064b9 RY |
474 | if (chdir(source_dir) < 0) { |
475 | com_err(__func__, errno, | |
9c891f7e DW |
476 | _("while changing working directory to \"%s\""), |
477 | source_dir); | |
052064b9 RY |
478 | return errno; |
479 | } | |
480 | ||
481 | if (!(dh = opendir("."))) { | |
482 | com_err(__func__, errno, | |
8f8d8a57 | 483 | _("while opening directory \"%s\""), source_dir); |
052064b9 RY |
484 | return errno; |
485 | } | |
486 | ||
8f8d8a57 | 487 | while ((dent = readdir(dh))) { |
9c891f7e DW |
488 | if ((!strcmp(dent->d_name, ".")) || |
489 | (!strcmp(dent->d_name, ".."))) | |
052064b9 | 490 | continue; |
1bad6f46 DW |
491 | if (lstat(dent->d_name, &st)) { |
492 | com_err(__func__, errno, _("while lstat \"%s\""), | |
493 | dent->d_name); | |
494 | goto out; | |
495 | } | |
052064b9 RY |
496 | name = dent->d_name; |
497 | ||
f84894bc RY |
498 | /* Check for hardlinks */ |
499 | save_inode = 0; | |
9c891f7e DW |
500 | if (!S_ISDIR(st.st_mode) && !S_ISLNK(st.st_mode) && |
501 | st.st_nlink > 1) { | |
b99888a0 | 502 | hdlink = is_hardlink(hdlinks, st.st_dev, st.st_ino); |
f84894bc | 503 | if (hdlink >= 0) { |
a3111e80 | 504 | retval = add_link(fs, parent_ino, |
b99888a0 | 505 | hdlinks->hdl[hdlink].dst_ino, |
9c891f7e | 506 | name); |
f84894bc | 507 | if (retval) { |
9c891f7e DW |
508 | com_err(__func__, retval, |
509 | "while linking %s", name); | |
1bad6f46 | 510 | goto out; |
f84894bc RY |
511 | } |
512 | continue; | |
513 | } else | |
514 | save_inode = 1; | |
515 | } | |
516 | ||
052064b9 | 517 | switch(st.st_mode & S_IFMT) { |
9c891f7e DW |
518 | case S_IFCHR: |
519 | case S_IFBLK: | |
520 | case S_IFIFO: | |
c61e0068 | 521 | case S_IFSOCK: |
a3111e80 | 522 | retval = do_mknod_internal(fs, parent_ino, name, &st); |
9c891f7e DW |
523 | if (retval) { |
524 | com_err(__func__, retval, | |
525 | _("while creating special file " | |
526 | "\"%s\""), name); | |
1bad6f46 | 527 | goto out; |
9c891f7e DW |
528 | } |
529 | break; | |
9c891f7e DW |
530 | case S_IFLNK: |
531 | read_cnt = readlink(name, ln_target, | |
1bad6f46 | 532 | sizeof(ln_target) - 1); |
9c891f7e DW |
533 | if (read_cnt == -1) { |
534 | com_err(__func__, errno, | |
535 | _("while trying to readlink \"%s\""), | |
536 | name); | |
537 | return errno; | |
538 | } | |
539 | ln_target[read_cnt] = '\0'; | |
a3111e80 DW |
540 | retval = do_symlink_internal(fs, parent_ino, name, |
541 | ln_target, root); | |
9c891f7e DW |
542 | if (retval) { |
543 | com_err(__func__, retval, | |
544 | _("while writing symlink\"%s\""), | |
545 | name); | |
1bad6f46 | 546 | goto out; |
9c891f7e DW |
547 | } |
548 | break; | |
549 | case S_IFREG: | |
a3111e80 DW |
550 | retval = do_write_internal(fs, parent_ino, name, name, |
551 | root); | |
9c891f7e DW |
552 | if (retval) { |
553 | com_err(__func__, retval, | |
554 | _("while writing file \"%s\""), name); | |
1bad6f46 | 555 | goto out; |
9c891f7e DW |
556 | } |
557 | break; | |
558 | case S_IFDIR: | |
a3111e80 DW |
559 | retval = do_mkdir_internal(fs, parent_ino, name, &st, |
560 | root); | |
9c891f7e DW |
561 | if (retval) { |
562 | com_err(__func__, retval, | |
563 | _("while making dir \"%s\""), name); | |
1bad6f46 | 564 | goto out; |
9c891f7e | 565 | } |
a3111e80 | 566 | retval = ext2fs_namei(fs, root, parent_ino, |
9c891f7e DW |
567 | name, &ino); |
568 | if (retval) { | |
569 | com_err(name, retval, 0); | |
1bad6f46 | 570 | goto out; |
9c891f7e DW |
571 | } |
572 | /* Populate the dir recursively*/ | |
b99888a0 | 573 | retval = __populate_fs(fs, ino, name, root, hdlinks); |
9c891f7e DW |
574 | if (retval) { |
575 | com_err(__func__, retval, | |
576 | _("while adding dir \"%s\""), name); | |
1bad6f46 | 577 | goto out; |
9c891f7e | 578 | } |
a3111e80 | 579 | if (chdir("..")) { |
1bad6f46 DW |
580 | com_err(__func__, errno, _("during cd ..")); |
581 | retval = errno; | |
582 | goto out; | |
a3111e80 | 583 | } |
9c891f7e DW |
584 | break; |
585 | default: | |
586 | com_err(__func__, 0, | |
587 | _("ignoring entry \"%s\""), name); | |
052064b9 | 588 | } |
b6960654 | 589 | |
a3111e80 | 590 | retval = ext2fs_namei(fs, root, parent_ino, name, &ino); |
8f8d8a57 | 591 | if (retval) { |
b6960654 | 592 | com_err(name, retval, 0); |
1bad6f46 | 593 | goto out; |
b6960654 RY |
594 | } |
595 | ||
a3111e80 | 596 | retval = set_inode_extra(fs, parent_ino, ino, &st); |
8f8d8a57 | 597 | if (retval) { |
b6960654 RY |
598 | com_err(__func__, retval, |
599 | _("while setting inode for \"%s\""), name); | |
1bad6f46 | 600 | goto out; |
b6960654 | 601 | } |
f84894bc RY |
602 | |
603 | /* Save the hardlink ino */ | |
604 | if (save_inode) { | |
605 | /* | |
606 | * Check whether need more memory, and we don't need | |
607 | * free() since the lifespan will be over after the fs | |
608 | * populated. | |
609 | */ | |
b99888a0 DW |
610 | if (hdlinks->count == hdlinks->size) { |
611 | void *p = realloc(hdlinks->hdl, | |
612 | (hdlinks->size + HDLINK_CNT) * | |
613 | sizeof(struct hdlink_s)); | |
614 | if (p == NULL) { | |
615 | com_err(name, errno, | |
616 | _("Not enough memory")); | |
1bad6f46 DW |
617 | retval = EXT2_ET_NO_MEMORY; |
618 | goto out; | |
f84894bc | 619 | } |
b99888a0 DW |
620 | hdlinks->hdl = p; |
621 | hdlinks->size += HDLINK_CNT; | |
f84894bc | 622 | } |
b99888a0 DW |
623 | hdlinks->hdl[hdlinks->count].src_dev = st.st_dev; |
624 | hdlinks->hdl[hdlinks->count].src_ino = st.st_ino; | |
625 | hdlinks->hdl[hdlinks->count].dst_ino = ino; | |
626 | hdlinks->count++; | |
f84894bc | 627 | } |
052064b9 | 628 | } |
1bad6f46 DW |
629 | |
630 | out: | |
052064b9 RY |
631 | closedir(dh); |
632 | return retval; | |
0d4deba2 | 633 | } |
b99888a0 DW |
634 | |
635 | errcode_t populate_fs(ext2_filsys fs, ext2_ino_t parent_ino, | |
636 | const char *source_dir, ext2_ino_t root) | |
637 | { | |
638 | struct hdlinks_s hdlinks; | |
639 | errcode_t retval; | |
640 | ||
641 | hdlinks.count = 0; | |
642 | hdlinks.size = HDLINK_CNT; | |
643 | hdlinks.hdl = realloc(NULL, hdlinks.size * sizeof(struct hdlink_s)); | |
644 | if (hdlinks.hdl == NULL) { | |
645 | com_err(__func__, errno, "Not enough memory"); | |
646 | return errno; | |
647 | } | |
648 | ||
649 | retval = __populate_fs(fs, parent_ino, source_dir, root, &hdlinks); | |
650 | ||
651 | free(hdlinks.hdl); | |
652 | return retval; | |
653 | } |