]>
Commit | Line | Data |
---|---|---|
bbccc6f3 AD |
1 | /* |
2 | * create_inode.c --- create an inode | |
3 | * | |
4 | * Copyright (C) 2014 Robert Yang <liezhi.yang@windriver.com> | |
5 | * | |
6 | * %Begin-Header% | |
7 | * This file may be redistributed under the terms of the GNU library | |
8 | * General Public License, version 2. | |
9 | * %End-Header% | |
10 | */ | |
11 | ||
76f13234 DW |
12 | #define _FILE_OFFSET_BITS 64 |
13 | #define _LARGEFILE64_SOURCE 1 | |
14 | #define _GNU_SOURCE 1 | |
15 | ||
16 | #include "config.h" | |
8f8d8a57 | 17 | #include <time.h> |
76f13234 | 18 | #include <sys/types.h> |
8f8d8a57 | 19 | #include <unistd.h> |
bbccc6f3 | 20 | #include <limits.h> /* for PATH_MAX */ |
c84da2ee RB |
21 | #ifdef HAVE_ATTR_XATTR_H |
22 | #include <attr/xattr.h> | |
23 | #endif | |
76f13234 DW |
24 | #include <sys/ioctl.h> |
25 | #include <ext2fs/ext2fs.h> | |
26 | #include <ext2fs/ext2_types.h> | |
27 | #include <ext2fs/fiemap.h> | |
8f8d8a57 | 28 | |
0d4deba2 | 29 | #include "create_inode.h" |
99ceb8ec | 30 | #include "support/nls-enable.h" |
0d4deba2 | 31 | |
ce600dc4 | 32 | /* 64KiB is the minimium blksize to best minimize system call overhead. */ |
76f13234 | 33 | #define COPY_FILE_BUFLEN 65536 |
ce600dc4 | 34 | |
a433db04 DW |
35 | static int ext2_file_type(unsigned int mode) |
36 | { | |
37 | if (LINUX_S_ISREG(mode)) | |
38 | return EXT2_FT_REG_FILE; | |
39 | ||
40 | if (LINUX_S_ISDIR(mode)) | |
41 | return EXT2_FT_DIR; | |
42 | ||
43 | if (LINUX_S_ISCHR(mode)) | |
44 | return EXT2_FT_CHRDEV; | |
45 | ||
46 | if (LINUX_S_ISBLK(mode)) | |
47 | return EXT2_FT_BLKDEV; | |
48 | ||
49 | if (LINUX_S_ISLNK(mode)) | |
50 | return EXT2_FT_SYMLINK; | |
51 | ||
52 | if (LINUX_S_ISFIFO(mode)) | |
53 | return EXT2_FT_FIFO; | |
54 | ||
55 | if (LINUX_S_ISSOCK(mode)) | |
56 | return EXT2_FT_SOCK; | |
57 | ||
58 | return 0; | |
59 | } | |
60 | ||
f84894bc | 61 | /* Link an inode number to a directory */ |
a3111e80 DW |
62 | static errcode_t add_link(ext2_filsys fs, ext2_ino_t parent_ino, |
63 | ext2_ino_t ino, const char *name) | |
f84894bc RY |
64 | { |
65 | struct ext2_inode inode; | |
66 | errcode_t retval; | |
67 | ||
a3111e80 | 68 | retval = ext2fs_read_inode(fs, ino, &inode); |
f84894bc | 69 | if (retval) { |
b04af4fe | 70 | com_err(__func__, retval, _("while reading inode %u"), ino); |
f84894bc RY |
71 | return retval; |
72 | } | |
73 | ||
a433db04 DW |
74 | retval = ext2fs_link(fs, parent_ino, name, ino, |
75 | ext2_file_type(inode.i_mode)); | |
f84894bc | 76 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 77 | retval = ext2fs_expand_dir(fs, parent_ino); |
f84894bc | 78 | if (retval) { |
b04af4fe DW |
79 | com_err(__func__, retval, |
80 | _("while expanding directory")); | |
f84894bc RY |
81 | return retval; |
82 | } | |
a433db04 DW |
83 | retval = ext2fs_link(fs, parent_ino, name, ino, |
84 | ext2_file_type(inode.i_mode)); | |
f84894bc RY |
85 | } |
86 | if (retval) { | |
b04af4fe | 87 | com_err(__func__, retval, _("while linking \"%s\""), name); |
f84894bc RY |
88 | return retval; |
89 | } | |
90 | ||
91 | inode.i_links_count++; | |
92 | ||
a3111e80 | 93 | retval = ext2fs_write_inode(fs, ino, &inode); |
f84894bc | 94 | if (retval) |
b04af4fe | 95 | com_err(__func__, retval, _("while writing inode %u"), ino); |
f84894bc RY |
96 | |
97 | return retval; | |
98 | } | |
99 | ||
b6960654 | 100 | /* Set the uid, gid, mode and time for the inode */ |
25f291c9 TT |
101 | static errcode_t set_inode_extra(ext2_filsys fs, ext2_ino_t ino, |
102 | struct stat *st) | |
b6960654 RY |
103 | { |
104 | errcode_t retval; | |
105 | struct ext2_inode inode; | |
106 | ||
a3111e80 | 107 | retval = ext2fs_read_inode(fs, ino, &inode); |
b6960654 | 108 | if (retval) { |
b04af4fe | 109 | com_err(__func__, retval, _("while reading inode %u"), ino); |
b6960654 RY |
110 | return retval; |
111 | } | |
112 | ||
b04af4fe DW |
113 | inode.i_uid = st->st_uid; |
114 | inode.i_gid = st->st_gid; | |
115 | inode.i_mode |= st->st_mode; | |
116 | inode.i_atime = st->st_atime; | |
117 | inode.i_mtime = st->st_mtime; | |
118 | inode.i_ctime = st->st_ctime; | |
b6960654 | 119 | |
a3111e80 | 120 | retval = ext2fs_write_inode(fs, ino, &inode); |
3aec816f | 121 | if (retval) |
b04af4fe | 122 | com_err(__func__, retval, _("while writing inode %u"), ino); |
3aec816f | 123 | return retval; |
b6960654 RY |
124 | } |
125 | ||
25f291c9 | 126 | #ifdef HAVE_LLISTXATTR |
b04af4fe DW |
127 | static errcode_t set_inode_xattr(ext2_filsys fs, ext2_ino_t ino, |
128 | const char *filename) | |
c84da2ee | 129 | { |
c84da2ee | 130 | errcode_t retval, close_retval; |
c84da2ee RB |
131 | struct ext2_xattr_handle *handle; |
132 | ssize_t size, value_size; | |
76f13234 | 133 | char *list = NULL; |
c84da2ee RB |
134 | int i; |
135 | ||
136 | size = llistxattr(filename, NULL, 0); | |
137 | if (size == -1) { | |
b04af4fe DW |
138 | retval = errno; |
139 | com_err(__func__, retval, _("while listing attributes of \"%s\""), | |
140 | filename); | |
141 | return retval; | |
c84da2ee RB |
142 | } else if (size == 0) { |
143 | return 0; | |
144 | } | |
145 | ||
146 | retval = ext2fs_xattrs_open(fs, ino, &handle); | |
147 | if (retval) { | |
148 | if (retval == EXT2_ET_MISSING_EA_FEATURE) | |
149 | return 0; | |
b04af4fe | 150 | com_err(__func__, retval, _("while opening inode %u"), ino); |
c84da2ee RB |
151 | return retval; |
152 | } | |
153 | ||
154 | retval = ext2fs_get_mem(size, &list); | |
155 | if (retval) { | |
b04af4fe | 156 | com_err(__func__, retval, _("while allocating memory")); |
c84da2ee RB |
157 | goto out; |
158 | } | |
159 | ||
160 | size = llistxattr(filename, list, size); | |
161 | if (size == -1) { | |
c84da2ee | 162 | retval = errno; |
b04af4fe DW |
163 | com_err(__func__, retval, _("while listing attributes of \"%s\""), |
164 | filename); | |
c84da2ee RB |
165 | goto out; |
166 | } | |
167 | ||
168 | for (i = 0; i < size; i += strlen(&list[i]) + 1) { | |
169 | const char *name = &list[i]; | |
170 | char *value; | |
171 | ||
344c043b | 172 | value_size = lgetxattr(filename, name, NULL, 0); |
c84da2ee | 173 | if (value_size == -1) { |
c84da2ee | 174 | retval = errno; |
b04af4fe DW |
175 | com_err(__func__, retval, |
176 | _("while reading attribute \"%s\" of \"%s\""), | |
177 | name, filename); | |
c84da2ee RB |
178 | break; |
179 | } | |
180 | ||
181 | retval = ext2fs_get_mem(value_size, &value); | |
182 | if (retval) { | |
b04af4fe | 183 | com_err(__func__, retval, _("while allocating memory")); |
c84da2ee RB |
184 | break; |
185 | } | |
186 | ||
344c043b | 187 | value_size = lgetxattr(filename, name, value, value_size); |
c84da2ee RB |
188 | if (value_size == -1) { |
189 | ext2fs_free_mem(&value); | |
c84da2ee | 190 | retval = errno; |
b04af4fe DW |
191 | com_err(__func__, retval, |
192 | _("while reading attribute \"%s\" of \"%s\""), | |
193 | name, filename); | |
c84da2ee RB |
194 | break; |
195 | } | |
196 | ||
197 | retval = ext2fs_xattr_set(handle, name, value, value_size); | |
198 | ext2fs_free_mem(&value); | |
199 | if (retval) { | |
200 | com_err(__func__, retval, | |
b04af4fe DW |
201 | _("while writing attribute \"%s\" to inode %u"), |
202 | name, ino); | |
c84da2ee RB |
203 | break; |
204 | } | |
205 | ||
206 | } | |
207 | out: | |
208 | ext2fs_free_mem(&list); | |
209 | close_retval = ext2fs_xattrs_close(&handle); | |
210 | if (close_retval) { | |
b04af4fe | 211 | com_err(__func__, retval, _("while closing inode %u"), ino); |
c84da2ee RB |
212 | retval = retval ? retval : close_retval; |
213 | } | |
214 | return retval; | |
25f291c9 TT |
215 | return 0; |
216 | } | |
c84da2ee | 217 | #else /* HAVE_LLISTXATTR */ |
25f291c9 TT |
218 | static errcode_t set_inode_xattr(ext2_filsys fs EXT2FS_ATTR((unused)), |
219 | ext2_ino_t ino EXT2FS_ATTR((unused)), | |
220 | const char *filename EXT2FS_ATTR((unused))) | |
221 | { | |
c84da2ee | 222 | return 0; |
c84da2ee | 223 | } |
25f291c9 | 224 | #endif /* HAVE_LLISTXATTR */ |
c84da2ee | 225 | |
c61e0068 | 226 | /* Make a special files (block and character devices), fifo's, and sockets */ |
a3111e80 DW |
227 | errcode_t do_mknod_internal(ext2_filsys fs, ext2_ino_t cwd, const char *name, |
228 | struct stat *st) | |
0d4deba2 | 229 | { |
b2346118 | 230 | ext2_ino_t ino; |
3cbb51f9 | 231 | errcode_t retval; |
b2346118 | 232 | struct ext2_inode inode; |
3cbb51f9 | 233 | unsigned long devmajor, devminor, mode; |
b2346118 RY |
234 | int filetype; |
235 | ||
236 | switch(st->st_mode & S_IFMT) { | |
9c891f7e DW |
237 | case S_IFCHR: |
238 | mode = LINUX_S_IFCHR; | |
239 | filetype = EXT2_FT_CHRDEV; | |
240 | break; | |
241 | case S_IFBLK: | |
242 | mode = LINUX_S_IFBLK; | |
243 | filetype = EXT2_FT_BLKDEV; | |
244 | break; | |
245 | case S_IFIFO: | |
246 | mode = LINUX_S_IFIFO; | |
247 | filetype = EXT2_FT_FIFO; | |
248 | break; | |
c61e0068 DW |
249 | case S_IFSOCK: |
250 | mode = LINUX_S_IFSOCK; | |
251 | filetype = EXT2_FT_SOCK; | |
ec3a42b1 | 252 | break; |
3aec816f | 253 | default: |
ec3a42b1 | 254 | return EXT2_ET_INVALID_ARGUMENT; |
b2346118 RY |
255 | } |
256 | ||
a3111e80 | 257 | retval = ext2fs_new_inode(fs, cwd, 010755, 0, &ino); |
b2346118 | 258 | if (retval) { |
b04af4fe DW |
259 | com_err(__func__, retval, _("while allocating inode \"%s\""), |
260 | name); | |
b2346118 RY |
261 | return retval; |
262 | } | |
263 | ||
264 | #ifdef DEBUGFS | |
265 | printf("Allocated inode: %u\n", ino); | |
266 | #endif | |
a3111e80 | 267 | retval = ext2fs_link(fs, cwd, name, ino, filetype); |
b2346118 | 268 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 269 | retval = ext2fs_expand_dir(fs, cwd); |
b2346118 | 270 | if (retval) { |
b04af4fe DW |
271 | com_err(__func__, retval, |
272 | _("while expanding directory")); | |
b2346118 RY |
273 | return retval; |
274 | } | |
a3111e80 | 275 | retval = ext2fs_link(fs, cwd, name, ino, filetype); |
b2346118 RY |
276 | } |
277 | if (retval) { | |
b04af4fe | 278 | com_err(name, retval, _("while creating inode \"%s\""), name); |
ec3a42b1 | 279 | return retval; |
b2346118 | 280 | } |
a3111e80 | 281 | if (ext2fs_test_inode_bitmap2(fs->inode_map, ino)) |
b2346118 | 282 | com_err(__func__, 0, "Warning: inode already set"); |
a3111e80 | 283 | ext2fs_inode_alloc_stats2(fs, ino, +1, 0); |
b2346118 RY |
284 | memset(&inode, 0, sizeof(inode)); |
285 | inode.i_mode = mode; | |
286 | inode.i_atime = inode.i_ctime = inode.i_mtime = | |
a3111e80 | 287 | fs->now ? fs->now : time(0); |
b2346118 | 288 | |
3cbb51f9 AD |
289 | if (filetype != S_IFIFO) { |
290 | devmajor = major(st->st_rdev); | |
291 | devminor = minor(st->st_rdev); | |
292 | ||
293 | if ((devmajor < 256) && (devminor < 256)) { | |
294 | inode.i_block[0] = devmajor * 256 + devminor; | |
295 | inode.i_block[1] = 0; | |
296 | } else { | |
297 | inode.i_block[0] = 0; | |
298 | inode.i_block[1] = (devminor & 0xff) | (devmajor << 8) | | |
299 | ((devminor & ~0xff) << 12); | |
300 | } | |
b2346118 RY |
301 | } |
302 | inode.i_links_count = 1; | |
303 | ||
a3111e80 | 304 | retval = ext2fs_write_new_inode(fs, ino, &inode); |
b2346118 | 305 | if (retval) |
b04af4fe | 306 | com_err(__func__, retval, _("while writing inode %u"), ino); |
b2346118 RY |
307 | |
308 | return retval; | |
0d4deba2 RY |
309 | } |
310 | ||
311 | /* Make a symlink name -> target */ | |
a3111e80 DW |
312 | errcode_t do_symlink_internal(ext2_filsys fs, ext2_ino_t cwd, const char *name, |
313 | char *target, ext2_ino_t root) | |
0d4deba2 | 314 | { |
b25ddc5e RY |
315 | char *cp; |
316 | ext2_ino_t parent_ino; | |
317 | errcode_t retval; | |
b25ddc5e RY |
318 | |
319 | cp = strrchr(name, '/'); | |
320 | if (cp) { | |
321 | *cp = 0; | |
a3111e80 | 322 | retval = ext2fs_namei(fs, root, cwd, name, &parent_ino); |
8f8d8a57 | 323 | if (retval) { |
b25ddc5e RY |
324 | com_err(name, retval, 0); |
325 | return retval; | |
326 | } | |
327 | name = cp+1; | |
328 | } else | |
329 | parent_ino = cwd; | |
330 | ||
a3111e80 | 331 | retval = ext2fs_symlink(fs, parent_ino, 0, name, target); |
b25ddc5e | 332 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 333 | retval = ext2fs_expand_dir(fs, parent_ino); |
b25ddc5e | 334 | if (retval) { |
9c891f7e | 335 | com_err("do_symlink_internal", retval, |
b04af4fe | 336 | _("while expanding directory")); |
b25ddc5e RY |
337 | return retval; |
338 | } | |
b04af4fe | 339 | retval = ext2fs_symlink(fs, parent_ino, 0, name, target); |
b25ddc5e | 340 | } |
3aec816f | 341 | if (retval) |
b04af4fe DW |
342 | com_err("ext2fs_symlink", retval, |
343 | _("while creating symlink \"%s\""), name); | |
3aec816f | 344 | return retval; |
0d4deba2 RY |
345 | } |
346 | ||
347 | /* Make a directory in the fs */ | |
a3111e80 | 348 | errcode_t do_mkdir_internal(ext2_filsys fs, ext2_ino_t cwd, const char *name, |
25f291c9 | 349 | ext2_ino_t root) |
0d4deba2 | 350 | { |
3068f03f | 351 | char *cp; |
c4c9bc59 | 352 | ext2_ino_t parent_ino; |
3068f03f | 353 | errcode_t retval; |
3068f03f RY |
354 | |
355 | ||
356 | cp = strrchr(name, '/'); | |
357 | if (cp) { | |
358 | *cp = 0; | |
a3111e80 | 359 | retval = ext2fs_namei(fs, root, cwd, name, &parent_ino); |
8f8d8a57 | 360 | if (retval) { |
b04af4fe DW |
361 | com_err(name, retval, _("while looking up \"%s\""), |
362 | name); | |
3068f03f RY |
363 | return retval; |
364 | } | |
365 | name = cp+1; | |
366 | } else | |
367 | parent_ino = cwd; | |
368 | ||
a3111e80 | 369 | retval = ext2fs_mkdir(fs, parent_ino, 0, name); |
3068f03f | 370 | if (retval == EXT2_ET_DIR_NO_SPACE) { |
a3111e80 | 371 | retval = ext2fs_expand_dir(fs, parent_ino); |
3068f03f | 372 | if (retval) { |
b04af4fe DW |
373 | com_err(__func__, retval, |
374 | _("while expanding directory")); | |
3068f03f RY |
375 | return retval; |
376 | } | |
b04af4fe | 377 | retval = ext2fs_mkdir(fs, parent_ino, 0, name); |
3068f03f | 378 | } |
3aec816f | 379 | if (retval) |
b04af4fe DW |
380 | com_err("ext2fs_mkdir", retval, |
381 | _("while creating directory \"%s\""), name); | |
3aec816f | 382 | return retval; |
0d4deba2 RY |
383 | } |
384 | ||
76f13234 | 385 | #if !defined HAVE_PREAD64 && !defined HAVE_PREAD |
f7134a9e | 386 | static ssize_t my_pread(int fd, void *buf, size_t count, off_t offset) |
ce600dc4 | 387 | { |
76f13234 DW |
388 | if (lseek(fd, offset, SEEK_SET) < 0) |
389 | return 0; | |
ce600dc4 | 390 | |
76f13234 DW |
391 | return read(fd, buf, count); |
392 | } | |
393 | #endif /* !defined HAVE_PREAD64 && !defined HAVE_PREAD */ | |
ce600dc4 | 394 | |
76f13234 DW |
395 | static errcode_t copy_file_range(ext2_filsys fs, int fd, ext2_file_t e2_file, |
396 | off_t start, off_t end, char *buf, | |
397 | char *zerobuf) | |
398 | { | |
399 | off_t off, bpos; | |
400 | ssize_t got, blen; | |
401 | unsigned int written; | |
402 | char *ptr; | |
403 | errcode_t err = 0; | |
404 | ||
405 | for (off = start; off < end; off += COPY_FILE_BUFLEN) { | |
406 | #ifdef HAVE_PREAD64 | |
407 | got = pread64(fd, buf, COPY_FILE_BUFLEN, off); | |
408 | #elif HAVE_PREAD | |
409 | got = pread(fd, buf, COPY_FILE_BUFLEN, off); | |
410 | #else | |
411 | got = my_pread(fd, buf, COPY_FILE_BUFLEN, off); | |
412 | #endif | |
ce600dc4 | 413 | if (got < 0) { |
76f13234 | 414 | err = errno; |
ce600dc4 RY |
415 | goto fail; |
416 | } | |
76f13234 DW |
417 | for (bpos = 0, ptr = buf; bpos < got; bpos += fs->blocksize) { |
418 | blen = fs->blocksize; | |
419 | if (blen > got - bpos) | |
420 | blen = got - bpos; | |
421 | if (memcmp(ptr, zerobuf, blen) == 0) { | |
422 | ptr += blen; | |
423 | continue; | |
424 | } | |
425 | err = ext2fs_file_lseek(e2_file, off + bpos, | |
426 | EXT2_SEEK_SET, NULL); | |
427 | if (err) | |
428 | goto fail; | |
429 | while (blen > 0) { | |
430 | err = ext2fs_file_write(e2_file, ptr, blen, | |
431 | &written); | |
432 | if (err) | |
433 | goto fail; | |
434 | if (written == 0) { | |
435 | err = EIO; | |
ce600dc4 | 436 | goto fail; |
76f13234 DW |
437 | } |
438 | blen -= written; | |
439 | ptr += written; | |
ce600dc4 RY |
440 | } |
441 | } | |
76f13234 DW |
442 | } |
443 | fail: | |
444 | return err; | |
445 | } | |
ce600dc4 | 446 | |
76f13234 DW |
447 | static errcode_t try_lseek_copy(ext2_filsys fs, int fd, struct stat *statbuf, |
448 | ext2_file_t e2_file, char *buf, char *zerobuf) | |
449 | { | |
450 | #if defined(SEEK_DATA) && defined(SEEK_HOLE) | |
451 | off_t data = 0, hole; | |
452 | off_t data_blk, hole_blk; | |
453 | errcode_t err; | |
454 | ||
455 | /* Try to use SEEK_DATA and SEEK_HOLE */ | |
456 | while (data < statbuf->st_size) { | |
457 | data = lseek(fd, data, SEEK_DATA); | |
458 | if (data < 0) { | |
459 | if (errno == ENXIO) | |
460 | break; | |
461 | return EXT2_ET_UNIMPLEMENTED; | |
ce600dc4 | 462 | } |
76f13234 DW |
463 | hole = lseek(fd, data, SEEK_HOLE); |
464 | if (hole < 0) | |
465 | return EXT2_ET_UNIMPLEMENTED; | |
466 | ||
467 | data_blk = data & ~(fs->blocksize - 1); | |
468 | hole_blk = (hole + (fs->blocksize - 1)) & ~(fs->blocksize - 1); | |
469 | err = copy_file_range(fs, fd, e2_file, data_blk, hole_blk, buf, | |
470 | zerobuf); | |
471 | if (err) | |
472 | return err; | |
473 | ||
474 | data = hole; | |
ce600dc4 | 475 | } |
ce600dc4 | 476 | |
76f13234 DW |
477 | return err; |
478 | #else | |
479 | return EXT2_ET_UNIMPLEMENTED; | |
480 | #endif /* SEEK_DATA and SEEK_HOLE */ | |
481 | } | |
482 | ||
483 | static errcode_t try_fiemap_copy(ext2_filsys fs, int fd, ext2_file_t e2_file, | |
484 | char *buf, char *zerobuf) | |
485 | { | |
486 | #if defined(FS_IOC_FIEMAP) | |
487 | #define EXTENT_MAX_COUNT 512 | |
488 | struct fiemap *fiemap_buf; | |
489 | struct fiemap_extent *ext_buf, *ext; | |
490 | int ext_buf_size, fie_buf_size; | |
491 | off_t pos = 0; | |
492 | unsigned int i; | |
493 | errcode_t err; | |
494 | ||
495 | ext_buf_size = EXTENT_MAX_COUNT * sizeof(struct fiemap_extent); | |
496 | fie_buf_size = sizeof(struct fiemap) + ext_buf_size; | |
497 | ||
498 | err = ext2fs_get_memzero(fie_buf_size, &fiemap_buf); | |
499 | if (err) | |
500 | return err; | |
501 | ||
502 | ext_buf = fiemap_buf->fm_extents; | |
503 | memset(fiemap_buf, 0, fie_buf_size); | |
504 | fiemap_buf->fm_length = FIEMAP_MAX_OFFSET; | |
505 | fiemap_buf->fm_flags |= FIEMAP_FLAG_SYNC; | |
506 | fiemap_buf->fm_extent_count = EXTENT_MAX_COUNT; | |
507 | ||
508 | do { | |
509 | fiemap_buf->fm_start = pos; | |
510 | memset(ext_buf, 0, ext_buf_size); | |
511 | err = ioctl(fd, FS_IOC_FIEMAP, fiemap_buf); | |
512 | if (err < 0 && (errno == EOPNOTSUPP || errno == ENOTTY)) { | |
513 | err = EXT2_ET_UNIMPLEMENTED; | |
514 | goto out; | |
515 | } else if (err < 0 || fiemap_buf->fm_mapped_extents == 0) { | |
516 | err = errno; | |
517 | goto out; | |
518 | } | |
519 | for (i = 0, ext = ext_buf; i < fiemap_buf->fm_mapped_extents; | |
520 | i++, ext++) { | |
521 | err = copy_file_range(fs, fd, e2_file, ext->fe_logical, | |
522 | ext->fe_logical + ext->fe_length, | |
523 | buf, zerobuf); | |
524 | if (err) | |
525 | goto out; | |
526 | } | |
527 | ||
528 | ext--; | |
529 | /* Record file's logical offset this time */ | |
530 | pos = ext->fe_logical + ext->fe_length; | |
531 | /* | |
532 | * If fm_extents array has been filled and | |
533 | * there are extents left, continue to cycle. | |
534 | */ | |
535 | } while (fiemap_buf->fm_mapped_extents == EXTENT_MAX_COUNT && | |
536 | !(ext->fe_flags & FIEMAP_EXTENT_LAST)); | |
537 | out: | |
538 | ext2fs_free_mem(&fiemap_buf); | |
539 | return err; | |
540 | #else | |
541 | return EXT2_ET_UNIMPLEMENTED; | |
542 | #endif /* FS_IOC_FIEMAP */ | |
543 | } | |
544 | ||
545 | static errcode_t copy_file(ext2_filsys fs, int fd, struct stat *statbuf, | |
546 | ext2_ino_t ino) | |
547 | { | |
548 | ext2_file_t e2_file; | |
549 | char *buf = NULL, *zerobuf = NULL; | |
550 | errcode_t err, close_err; | |
551 | ||
552 | err = ext2fs_file_open(fs, ino, EXT2_FILE_WRITE, &e2_file); | |
553 | if (err) | |
554 | return err; | |
555 | ||
556 | err = ext2fs_get_mem(COPY_FILE_BUFLEN, &buf); | |
557 | if (err) | |
558 | goto out; | |
559 | ||
560 | err = ext2fs_get_memzero(fs->blocksize, &zerobuf); | |
561 | if (err) | |
562 | goto out; | |
563 | ||
564 | err = try_lseek_copy(fs, fd, statbuf, e2_file, buf, zerobuf); | |
565 | if (err != EXT2_ET_UNIMPLEMENTED) | |
566 | goto out; | |
567 | ||
568 | err = try_fiemap_copy(fs, fd, e2_file, buf, zerobuf); | |
569 | if (err != EXT2_ET_UNIMPLEMENTED) | |
570 | goto out; | |
571 | ||
572 | err = copy_file_range(fs, fd, e2_file, 0, statbuf->st_size, buf, | |
573 | zerobuf); | |
574 | out: | |
575 | ext2fs_free_mem(&zerobuf); | |
6bb88459 | 576 | ext2fs_free_mem(&buf); |
76f13234 DW |
577 | close_err = ext2fs_file_close(e2_file); |
578 | if (err == 0) | |
579 | err = close_err; | |
580 | return err; | |
ce600dc4 RY |
581 | } |
582 | ||
b99888a0 | 583 | static int is_hardlink(struct hdlinks_s *hdlinks, dev_t dev, ino_t ino) |
f84894bc RY |
584 | { |
585 | int i; | |
586 | ||
b99888a0 DW |
587 | for (i = 0; i < hdlinks->count; i++) { |
588 | if (hdlinks->hdl[i].src_dev == dev && | |
589 | hdlinks->hdl[i].src_ino == ino) | |
f84894bc RY |
590 | return i; |
591 | } | |
592 | return -1; | |
593 | } | |
594 | ||
0d4deba2 | 595 | /* Copy the native file to the fs */ |
a3111e80 DW |
596 | errcode_t do_write_internal(ext2_filsys fs, ext2_ino_t cwd, const char *src, |
597 | const char *dest, ext2_ino_t root) | |
0d4deba2 | 598 | { |
ce600dc4 RY |
599 | int fd; |
600 | struct stat statbuf; | |
601 | ext2_ino_t newfile; | |
602 | errcode_t retval; | |
603 | struct ext2_inode inode; | |
ce600dc4 | 604 | |
ec3a42b1 | 605 | fd = ext2fs_open_file(src, O_RDONLY, 0); |
ce600dc4 | 606 | if (fd < 0) { |
b04af4fe DW |
607 | retval = errno; |
608 | com_err(__func__, retval, _("while opening \"%s\" to copy"), | |
609 | src); | |
610 | return retval; | |
ce600dc4 RY |
611 | } |
612 | if (fstat(fd, &statbuf) < 0) { | |
b04af4fe DW |
613 | retval = errno; |
614 | goto out; | |
ce600dc4 RY |
615 | } |
616 | ||
a3111e80 | 617 | retval = ext2fs_namei(fs, root, cwd, dest, &newfile); |
ce600dc4 | 618 | if (retval == 0) { |
b04af4fe DW |
619 | retval = EXT2_ET_FILE_EXISTS; |
620 | goto out; | |
ce600dc4 RY |
621 | } |
622 | ||
a3111e80 | 623 | retval = ext2fs_new_inode(fs, cwd, 010755, 0, &newfile); |
b04af4fe DW |
624 | if (retval) |
625 | goto out; | |
ce600dc4 RY |
626 | #ifdef DEBUGFS |
627 | printf("Allocated inode: %u\n", newfile); | |
628 | #endif | |
a3111e80 | 629 | retval = ext2fs_link(fs, cwd, dest, newfile, |
ce600dc4 RY |
630 | EXT2_FT_REG_FILE); |
631 | if (retval == EXT2_ET_DIR_NO_SPACE) { | |
a3111e80 | 632 | retval = ext2fs_expand_dir(fs, cwd); |
b04af4fe DW |
633 | if (retval) |
634 | goto out; | |
a3111e80 | 635 | retval = ext2fs_link(fs, cwd, dest, newfile, |
ce600dc4 RY |
636 | EXT2_FT_REG_FILE); |
637 | } | |
b04af4fe DW |
638 | if (retval) |
639 | goto out; | |
a3111e80 | 640 | if (ext2fs_test_inode_bitmap2(fs->inode_map, newfile)) |
ce600dc4 | 641 | com_err(__func__, 0, "Warning: inode already set"); |
a3111e80 | 642 | ext2fs_inode_alloc_stats2(fs, newfile, +1, 0); |
ce600dc4 RY |
643 | memset(&inode, 0, sizeof(inode)); |
644 | inode.i_mode = (statbuf.st_mode & ~LINUX_S_IFMT) | LINUX_S_IFREG; | |
645 | inode.i_atime = inode.i_ctime = inode.i_mtime = | |
a3111e80 | 646 | fs->now ? fs->now : time(0); |
ce600dc4 | 647 | inode.i_links_count = 1; |
22302aa3 | 648 | retval = ext2fs_inode_size_set(fs, &inode, statbuf.st_size); |
b04af4fe DW |
649 | if (retval) |
650 | goto out; | |
a3111e80 | 651 | if (EXT2_HAS_INCOMPAT_FEATURE(fs->super, |
d23b1965 TT |
652 | EXT4_FEATURE_INCOMPAT_INLINE_DATA)) { |
653 | inode.i_flags |= EXT4_INLINE_DATA_FL; | |
a3111e80 | 654 | } else if (fs->super->s_feature_incompat & |
d23b1965 | 655 | EXT3_FEATURE_INCOMPAT_EXTENTS) { |
3548bb64 DW |
656 | ext2_extent_handle_t handle; |
657 | ||
658 | inode.i_flags &= ~EXT4_EXTENTS_FL; | |
659 | retval = ext2fs_extent_open2(fs, newfile, &inode, &handle); | |
660 | if (retval) | |
b04af4fe | 661 | goto out; |
3548bb64 | 662 | ext2fs_extent_free(handle); |
ce600dc4 RY |
663 | } |
664 | ||
a3111e80 | 665 | retval = ext2fs_write_new_inode(fs, newfile, &inode); |
b04af4fe DW |
666 | if (retval) |
667 | goto out; | |
d23b1965 | 668 | if (inode.i_flags & EXT4_INLINE_DATA_FL) { |
a3111e80 | 669 | retval = ext2fs_inline_data_init(fs, newfile); |
b04af4fe DW |
670 | if (retval) |
671 | goto out; | |
d23b1965 | 672 | } |
ce600dc4 | 673 | if (LINUX_S_ISREG(inode.i_mode)) { |
76f13234 | 674 | retval = copy_file(fs, fd, &statbuf, newfile); |
ce600dc4 | 675 | if (retval) |
b04af4fe | 676 | goto out; |
ce600dc4 | 677 | } |
b04af4fe | 678 | out: |
ce600dc4 | 679 | close(fd); |
117b54c3 | 680 | return retval; |
0d4deba2 RY |
681 | } |
682 | ||
683 | /* Copy files from source_dir to fs */ | |
b99888a0 DW |
684 | static errcode_t __populate_fs(ext2_filsys fs, ext2_ino_t parent_ino, |
685 | const char *source_dir, ext2_ino_t root, | |
686 | struct hdlinks_s *hdlinks) | |
0d4deba2 | 687 | { |
052064b9 RY |
688 | const char *name; |
689 | DIR *dh; | |
690 | struct dirent *dent; | |
691 | struct stat st; | |
692 | char ln_target[PATH_MAX]; | |
f84894bc | 693 | unsigned int save_inode; |
052064b9 | 694 | ext2_ino_t ino; |
1bad6f46 | 695 | errcode_t retval = 0; |
052064b9 | 696 | int read_cnt; |
f84894bc | 697 | int hdlink; |
052064b9 | 698 | |
052064b9 | 699 | if (chdir(source_dir) < 0) { |
b04af4fe DW |
700 | retval = errno; |
701 | com_err(__func__, retval, | |
9c891f7e DW |
702 | _("while changing working directory to \"%s\""), |
703 | source_dir); | |
b04af4fe | 704 | return retval; |
052064b9 RY |
705 | } |
706 | ||
707 | if (!(dh = opendir("."))) { | |
b04af4fe DW |
708 | retval = errno; |
709 | com_err(__func__, retval, | |
8f8d8a57 | 710 | _("while opening directory \"%s\""), source_dir); |
b04af4fe | 711 | return retval; |
052064b9 RY |
712 | } |
713 | ||
8f8d8a57 | 714 | while ((dent = readdir(dh))) { |
9c891f7e DW |
715 | if ((!strcmp(dent->d_name, ".")) || |
716 | (!strcmp(dent->d_name, ".."))) | |
052064b9 | 717 | continue; |
1bad6f46 | 718 | if (lstat(dent->d_name, &st)) { |
b04af4fe DW |
719 | retval = errno; |
720 | com_err(__func__, retval, _("while lstat \"%s\""), | |
1bad6f46 DW |
721 | dent->d_name); |
722 | goto out; | |
723 | } | |
052064b9 RY |
724 | name = dent->d_name; |
725 | ||
f84894bc RY |
726 | /* Check for hardlinks */ |
727 | save_inode = 0; | |
9c891f7e DW |
728 | if (!S_ISDIR(st.st_mode) && !S_ISLNK(st.st_mode) && |
729 | st.st_nlink > 1) { | |
b99888a0 | 730 | hdlink = is_hardlink(hdlinks, st.st_dev, st.st_ino); |
f84894bc | 731 | if (hdlink >= 0) { |
a3111e80 | 732 | retval = add_link(fs, parent_ino, |
b99888a0 | 733 | hdlinks->hdl[hdlink].dst_ino, |
9c891f7e | 734 | name); |
f84894bc | 735 | if (retval) { |
9c891f7e DW |
736 | com_err(__func__, retval, |
737 | "while linking %s", name); | |
1bad6f46 | 738 | goto out; |
f84894bc RY |
739 | } |
740 | continue; | |
741 | } else | |
742 | save_inode = 1; | |
743 | } | |
744 | ||
052064b9 | 745 | switch(st.st_mode & S_IFMT) { |
9c891f7e DW |
746 | case S_IFCHR: |
747 | case S_IFBLK: | |
748 | case S_IFIFO: | |
c61e0068 | 749 | case S_IFSOCK: |
a3111e80 | 750 | retval = do_mknod_internal(fs, parent_ino, name, &st); |
9c891f7e DW |
751 | if (retval) { |
752 | com_err(__func__, retval, | |
753 | _("while creating special file " | |
754 | "\"%s\""), name); | |
1bad6f46 | 755 | goto out; |
9c891f7e DW |
756 | } |
757 | break; | |
9c891f7e DW |
758 | case S_IFLNK: |
759 | read_cnt = readlink(name, ln_target, | |
1bad6f46 | 760 | sizeof(ln_target) - 1); |
9c891f7e | 761 | if (read_cnt == -1) { |
ec3a42b1 | 762 | retval = errno; |
b04af4fe DW |
763 | com_err(__func__, retval, |
764 | _("while trying to read link \"%s\""), | |
765 | name); | |
ec3a42b1 | 766 | goto out; |
9c891f7e DW |
767 | } |
768 | ln_target[read_cnt] = '\0'; | |
a3111e80 DW |
769 | retval = do_symlink_internal(fs, parent_ino, name, |
770 | ln_target, root); | |
9c891f7e DW |
771 | if (retval) { |
772 | com_err(__func__, retval, | |
773 | _("while writing symlink\"%s\""), | |
774 | name); | |
1bad6f46 | 775 | goto out; |
9c891f7e DW |
776 | } |
777 | break; | |
778 | case S_IFREG: | |
a3111e80 DW |
779 | retval = do_write_internal(fs, parent_ino, name, name, |
780 | root); | |
9c891f7e DW |
781 | if (retval) { |
782 | com_err(__func__, retval, | |
783 | _("while writing file \"%s\""), name); | |
1bad6f46 | 784 | goto out; |
9c891f7e DW |
785 | } |
786 | break; | |
787 | case S_IFDIR: | |
b04af4fe DW |
788 | /* Don't choke on /lost+found */ |
789 | if (parent_ino == EXT2_ROOT_INO && | |
790 | strcmp(name, "lost+found") == 0) | |
791 | goto find_lnf; | |
25f291c9 | 792 | retval = do_mkdir_internal(fs, parent_ino, name, |
a3111e80 | 793 | root); |
9c891f7e DW |
794 | if (retval) { |
795 | com_err(__func__, retval, | |
796 | _("while making dir \"%s\""), name); | |
1bad6f46 | 797 | goto out; |
9c891f7e | 798 | } |
b04af4fe | 799 | find_lnf: |
a3111e80 | 800 | retval = ext2fs_namei(fs, root, parent_ino, |
9c891f7e DW |
801 | name, &ino); |
802 | if (retval) { | |
803 | com_err(name, retval, 0); | |
1bad6f46 | 804 | goto out; |
9c891f7e DW |
805 | } |
806 | /* Populate the dir recursively*/ | |
b99888a0 | 807 | retval = __populate_fs(fs, ino, name, root, hdlinks); |
b04af4fe | 808 | if (retval) |
1bad6f46 | 809 | goto out; |
a3111e80 | 810 | if (chdir("..")) { |
1bad6f46 | 811 | retval = errno; |
b04af4fe DW |
812 | com_err(__func__, retval, |
813 | _("while changing directory")); | |
1bad6f46 | 814 | goto out; |
a3111e80 | 815 | } |
9c891f7e DW |
816 | break; |
817 | default: | |
818 | com_err(__func__, 0, | |
819 | _("ignoring entry \"%s\""), name); | |
052064b9 | 820 | } |
b6960654 | 821 | |
a3111e80 | 822 | retval = ext2fs_namei(fs, root, parent_ino, name, &ino); |
8f8d8a57 | 823 | if (retval) { |
b04af4fe DW |
824 | com_err(name, retval, _("while looking up \"%s\""), |
825 | name); | |
1bad6f46 | 826 | goto out; |
b6960654 RY |
827 | } |
828 | ||
25f291c9 | 829 | retval = set_inode_extra(fs, ino, &st); |
8f8d8a57 | 830 | if (retval) { |
b6960654 RY |
831 | com_err(__func__, retval, |
832 | _("while setting inode for \"%s\""), name); | |
1bad6f46 | 833 | goto out; |
b6960654 | 834 | } |
f84894bc | 835 | |
c84da2ee RB |
836 | retval = set_inode_xattr(fs, ino, name); |
837 | if (retval) { | |
838 | com_err(__func__, retval, | |
839 | _("while setting xattrs for \"%s\""), name); | |
840 | goto out; | |
841 | } | |
842 | ||
f84894bc RY |
843 | /* Save the hardlink ino */ |
844 | if (save_inode) { | |
845 | /* | |
846 | * Check whether need more memory, and we don't need | |
847 | * free() since the lifespan will be over after the fs | |
848 | * populated. | |
849 | */ | |
b99888a0 DW |
850 | if (hdlinks->count == hdlinks->size) { |
851 | void *p = realloc(hdlinks->hdl, | |
852 | (hdlinks->size + HDLINK_CNT) * | |
853 | sizeof(struct hdlink_s)); | |
854 | if (p == NULL) { | |
1bad6f46 | 855 | retval = EXT2_ET_NO_MEMORY; |
b04af4fe DW |
856 | com_err(name, retval, |
857 | _("while saving inode data")); | |
1bad6f46 | 858 | goto out; |
f84894bc | 859 | } |
b99888a0 DW |
860 | hdlinks->hdl = p; |
861 | hdlinks->size += HDLINK_CNT; | |
f84894bc | 862 | } |
b99888a0 DW |
863 | hdlinks->hdl[hdlinks->count].src_dev = st.st_dev; |
864 | hdlinks->hdl[hdlinks->count].src_ino = st.st_ino; | |
865 | hdlinks->hdl[hdlinks->count].dst_ino = ino; | |
866 | hdlinks->count++; | |
f84894bc | 867 | } |
052064b9 | 868 | } |
1bad6f46 DW |
869 | |
870 | out: | |
052064b9 RY |
871 | closedir(dh); |
872 | return retval; | |
0d4deba2 | 873 | } |
b99888a0 DW |
874 | |
875 | errcode_t populate_fs(ext2_filsys fs, ext2_ino_t parent_ino, | |
876 | const char *source_dir, ext2_ino_t root) | |
877 | { | |
878 | struct hdlinks_s hdlinks; | |
879 | errcode_t retval; | |
880 | ||
b04af4fe DW |
881 | if (!(fs->flags & EXT2_FLAG_RW)) { |
882 | com_err(__func__, 0, "Filesystem opened readonly"); | |
883 | return EROFS; | |
884 | } | |
885 | ||
b99888a0 DW |
886 | hdlinks.count = 0; |
887 | hdlinks.size = HDLINK_CNT; | |
888 | hdlinks.hdl = realloc(NULL, hdlinks.size * sizeof(struct hdlink_s)); | |
889 | if (hdlinks.hdl == NULL) { | |
b04af4fe DW |
890 | retval = errno; |
891 | com_err(__func__, retval, _("while allocating memory")); | |
892 | return retval; | |
b99888a0 DW |
893 | } |
894 | ||
895 | retval = __populate_fs(fs, parent_ino, source_dir, root, &hdlinks); | |
896 | ||
897 | free(hdlinks.hdl); | |
898 | return retval; | |
899 | } |