Contents of /trunk/mkinitrd-magellan/busybox/archival/tar.c
Parent Directory | Revision Log
Revision 532 -
(show annotations)
(download)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 28905 byte(s)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 28905 byte(s)
-import if magellan mkinitrd; it is a fork of redhats mkinitrd-5.0.8 with all magellan patches and features; deprecates magellan-src/mkinitrd
1 | /* vi: set sw=4 ts=4: */ |
2 | /* |
3 | * Mini tar implementation for busybox |
4 | * |
5 | * Modified to use common extraction code used by ar, cpio, dpkg-deb, dpkg |
6 | * Glenn McGrath <bug1@iinet.net.au> |
7 | * |
8 | * Note, that as of BusyBox-0.43, tar has been completely rewritten from the |
9 | * ground up. It still has remnants of the old code lying about, but it is |
10 | * very different now (i.e., cleaner, less global variables, etc.) |
11 | * |
12 | * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org> |
13 | * |
14 | * Based in part in the tar implementation in sash |
15 | * Copyright (c) 1999 by David I. Bell |
16 | * Permission is granted to use, distribute, or modify this source, |
17 | * provided that this copyright notice remains intact. |
18 | * Permission to distribute sash derived code under the GPL has been granted. |
19 | * |
20 | * Based in part on the tar implementation from busybox-0.28 |
21 | * Copyright (C) 1995 Bruce Perens |
22 | * |
23 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
24 | */ |
25 | |
26 | #include <fnmatch.h> |
27 | #include <getopt.h> |
28 | #include "busybox.h" |
29 | #include "unarchive.h" |
30 | |
31 | #if ENABLE_FEATURE_TAR_CREATE |
32 | |
33 | /* Tar file constants */ |
34 | |
35 | #define TAR_BLOCK_SIZE 512 |
36 | |
37 | /* POSIX tar Header Block, from POSIX 1003.1-1990 */ |
38 | #define NAME_SIZE 100 |
39 | #define NAME_SIZE_STR "100" |
40 | typedef struct TarHeader TarHeader; |
41 | struct TarHeader { /* byte offset */ |
42 | char name[NAME_SIZE]; /* 0-99 */ |
43 | char mode[8]; /* 100-107 */ |
44 | char uid[8]; /* 108-115 */ |
45 | char gid[8]; /* 116-123 */ |
46 | char size[12]; /* 124-135 */ |
47 | char mtime[12]; /* 136-147 */ |
48 | char chksum[8]; /* 148-155 */ |
49 | char typeflag; /* 156-156 */ |
50 | char linkname[NAME_SIZE]; /* 157-256 */ |
51 | char magic[6]; /* 257-262 */ |
52 | char version[2]; /* 263-264 */ |
53 | char uname[32]; /* 265-296 */ |
54 | char gname[32]; /* 297-328 */ |
55 | char devmajor[8]; /* 329-336 */ |
56 | char devminor[8]; /* 337-344 */ |
57 | char prefix[155]; /* 345-499 */ |
58 | char padding[12]; /* 500-512 (pad to exactly the TAR_BLOCK_SIZE) */ |
59 | }; |
60 | |
61 | /* |
62 | ** writeTarFile(), writeFileToTarball(), and writeTarHeader() are |
63 | ** the only functions that deal with the HardLinkInfo structure. |
64 | ** Even these functions use the xxxHardLinkInfo() functions. |
65 | */ |
66 | typedef struct HardLinkInfo HardLinkInfo; |
67 | struct HardLinkInfo { |
68 | HardLinkInfo *next; /* Next entry in list */ |
69 | dev_t dev; /* Device number */ |
70 | ino_t ino; /* Inode number */ |
71 | short linkCount; /* (Hard) Link Count */ |
72 | char name[1]; /* Start of filename (must be last) */ |
73 | }; |
74 | |
75 | /* Some info to be carried along when creating a new tarball */ |
76 | typedef struct TarBallInfo TarBallInfo; |
77 | struct TarBallInfo { |
78 | int tarFd; /* Open-for-write file descriptor |
79 | for the tarball */ |
80 | struct stat statBuf; /* Stat info for the tarball, letting |
81 | us know the inode and device that the |
82 | tarball lives, so we can avoid trying |
83 | to include the tarball into itself */ |
84 | int verboseFlag; /* Whether to print extra stuff or not */ |
85 | const llist_t *excludeList; /* List of files to not include */ |
86 | HardLinkInfo *hlInfoHead; /* Hard Link Tracking Information */ |
87 | HardLinkInfo *hlInfo; /* Hard Link Info for the current file */ |
88 | }; |
89 | |
90 | /* A nice enum with all the possible tar file content types */ |
91 | enum TarFileType { |
92 | REGTYPE = '0', /* regular file */ |
93 | REGTYPE0 = '\0', /* regular file (ancient bug compat) */ |
94 | LNKTYPE = '1', /* hard link */ |
95 | SYMTYPE = '2', /* symbolic link */ |
96 | CHRTYPE = '3', /* character special */ |
97 | BLKTYPE = '4', /* block special */ |
98 | DIRTYPE = '5', /* directory */ |
99 | FIFOTYPE = '6', /* FIFO special */ |
100 | CONTTYPE = '7', /* reserved */ |
101 | GNULONGLINK = 'K', /* GNU long (>100 chars) link name */ |
102 | GNULONGNAME = 'L', /* GNU long (>100 chars) file name */ |
103 | }; |
104 | typedef enum TarFileType TarFileType; |
105 | |
106 | /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */ |
107 | static void addHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr, |
108 | struct stat *statbuf, |
109 | const char *fileName) |
110 | { |
111 | /* Note: hlInfoHeadPtr can never be NULL! */ |
112 | HardLinkInfo *hlInfo; |
113 | |
114 | hlInfo = xmalloc(sizeof(HardLinkInfo) + strlen(fileName)); |
115 | hlInfo->next = *hlInfoHeadPtr; |
116 | *hlInfoHeadPtr = hlInfo; |
117 | hlInfo->dev = statbuf->st_dev; |
118 | hlInfo->ino = statbuf->st_ino; |
119 | hlInfo->linkCount = statbuf->st_nlink; |
120 | strcpy(hlInfo->name, fileName); |
121 | } |
122 | |
123 | static void freeHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr) |
124 | { |
125 | HardLinkInfo *hlInfo; |
126 | HardLinkInfo *hlInfoNext; |
127 | |
128 | if (hlInfoHeadPtr) { |
129 | hlInfo = *hlInfoHeadPtr; |
130 | while (hlInfo) { |
131 | hlInfoNext = hlInfo->next; |
132 | free(hlInfo); |
133 | hlInfo = hlInfoNext; |
134 | } |
135 | *hlInfoHeadPtr = NULL; |
136 | } |
137 | return; |
138 | } |
139 | |
140 | /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */ |
141 | static HardLinkInfo *findHardLinkInfo(HardLinkInfo * hlInfo, struct stat *statbuf) |
142 | { |
143 | while (hlInfo) { |
144 | if ((statbuf->st_ino == hlInfo->ino) && (statbuf->st_dev == hlInfo->dev)) |
145 | break; |
146 | hlInfo = hlInfo->next; |
147 | } |
148 | return hlInfo; |
149 | } |
150 | |
151 | /* Put an octal string into the specified buffer. |
152 | * The number is zero padded and possibly null terminated. |
153 | * Stores low-order bits only if whole value does not fit. */ |
154 | static void putOctal(char *cp, int len, off_t value) |
155 | { |
156 | char tempBuffer[sizeof(off_t)*3+1]; |
157 | char *tempString = tempBuffer; |
158 | int width; |
159 | |
160 | width = sprintf(tempBuffer, "%0*"OFF_FMT"o", len, value); |
161 | tempString += (width - len); |
162 | |
163 | /* If string has leading zeroes, we can drop one */ |
164 | /* and field will have trailing '\0' */ |
165 | /* (increases chances of compat with other tars) */ |
166 | if (tempString[0] == '0') |
167 | tempString++; |
168 | |
169 | /* Copy the string to the field */ |
170 | memcpy(cp, tempString, len); |
171 | } |
172 | #define PUT_OCTAL(a, b) putOctal((a), sizeof(a), (b)) |
173 | |
174 | static void chksum_and_xwrite(int fd, struct TarHeader* hp) |
175 | { |
176 | const unsigned char *cp; |
177 | int chksum, size; |
178 | |
179 | strcpy(hp->magic, "ustar "); |
180 | |
181 | /* Calculate and store the checksum (i.e., the sum of all of the bytes of |
182 | * the header). The checksum field must be filled with blanks for the |
183 | * calculation. The checksum field is formatted differently from the |
184 | * other fields: it has 6 digits, a null, then a space -- rather than |
185 | * digits, followed by a null like the other fields... */ |
186 | memset(hp->chksum, ' ', sizeof(hp->chksum)); |
187 | cp = (const unsigned char *) hp; |
188 | chksum = 0; |
189 | size = sizeof(*hp); |
190 | do { chksum += *cp++; } while (--size); |
191 | putOctal(hp->chksum, sizeof(hp->chksum)-1, chksum); |
192 | |
193 | /* Now write the header out to disk */ |
194 | xwrite(fd, hp, sizeof(*hp)); |
195 | } |
196 | |
197 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS |
198 | static void writeLongname(int fd, int type, const char *name, int dir) |
199 | { |
200 | static const struct { |
201 | char mode[8]; /* 100-107 */ |
202 | char uid[8]; /* 108-115 */ |
203 | char gid[8]; /* 116-123 */ |
204 | char size[12]; /* 124-135 */ |
205 | char mtime[12]; /* 136-147 */ |
206 | } prefilled = { |
207 | "0000000", |
208 | "0000000", |
209 | "0000000", |
210 | "00000000000", |
211 | "00000000000", |
212 | }; |
213 | struct TarHeader header; |
214 | int size; |
215 | |
216 | dir = !!dir; /* normalize: 0/1 */ |
217 | size = strlen(name) + 1 + dir; /* GNU tar uses strlen+1 */ |
218 | /* + dir: account for possible '/' */ |
219 | |
220 | memset(&header, 0, sizeof(header)); |
221 | strcpy(header.name, "././@LongLink"); |
222 | memcpy(header.mode, prefilled.mode, sizeof(prefilled)); |
223 | PUT_OCTAL(header.size, size); |
224 | header.typeflag = type; |
225 | chksum_and_xwrite(fd, &header); |
226 | |
227 | /* Write filename[/] and pad the block. */ |
228 | /* dir=0: writes 'name<NUL>', pads */ |
229 | /* dir=1: writes 'name', writes '/<NUL>', pads */ |
230 | dir *= 2; |
231 | xwrite(fd, name, size - dir); |
232 | xwrite(fd, "/", dir); |
233 | size = (-size) & (TAR_BLOCK_SIZE-1); |
234 | memset(&header, 0, size); |
235 | xwrite(fd, &header, size); |
236 | } |
237 | #endif |
238 | |
239 | /* Write out a tar header for the specified file/directory/whatever */ |
240 | void BUG_tar_header_size(void); |
241 | static int writeTarHeader(struct TarBallInfo *tbInfo, |
242 | const char *header_name, const char *fileName, struct stat *statbuf) |
243 | { |
244 | struct TarHeader header; |
245 | |
246 | if (sizeof(header) != 512) |
247 | BUG_tar_header_size(); |
248 | |
249 | memset(&header, 0, sizeof(struct TarHeader)); |
250 | |
251 | strncpy(header.name, header_name, sizeof(header.name)); |
252 | |
253 | /* POSIX says to mask mode with 07777. */ |
254 | PUT_OCTAL(header.mode, statbuf->st_mode & 07777); |
255 | PUT_OCTAL(header.uid, statbuf->st_uid); |
256 | PUT_OCTAL(header.gid, statbuf->st_gid); |
257 | memset(header.size, '0', sizeof(header.size)-1); /* Regular file size is handled later */ |
258 | PUT_OCTAL(header.mtime, statbuf->st_mtime); |
259 | |
260 | /* Enter the user and group names */ |
261 | safe_strncpy(header.uname, get_cached_username(statbuf->st_uid), sizeof(header.uname)); |
262 | safe_strncpy(header.gname, get_cached_groupname(statbuf->st_gid), sizeof(header.gname)); |
263 | |
264 | if (tbInfo->hlInfo) { |
265 | /* This is a hard link */ |
266 | header.typeflag = LNKTYPE; |
267 | strncpy(header.linkname, tbInfo->hlInfo->name, |
268 | sizeof(header.linkname)); |
269 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS |
270 | /* Write out long linkname if needed */ |
271 | if (header.linkname[sizeof(header.linkname)-1]) |
272 | writeLongname(tbInfo->tarFd, GNULONGLINK, |
273 | tbInfo->hlInfo->name, 0); |
274 | #endif |
275 | } else if (S_ISLNK(statbuf->st_mode)) { |
276 | char *lpath = xreadlink(fileName); |
277 | if (!lpath) /* Already printed err msg inside xreadlink() */ |
278 | return FALSE; |
279 | header.typeflag = SYMTYPE; |
280 | strncpy(header.linkname, lpath, sizeof(header.linkname)); |
281 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS |
282 | /* Write out long linkname if needed */ |
283 | if (header.linkname[sizeof(header.linkname)-1]) |
284 | writeLongname(tbInfo->tarFd, GNULONGLINK, lpath, 0); |
285 | #else |
286 | /* If it is larger than 100 bytes, bail out */ |
287 | if (header.linkname[sizeof(header.linkname)-1]) { |
288 | free(lpath); |
289 | bb_error_msg("names longer than "NAME_SIZE_STR" chars not supported"); |
290 | return FALSE; |
291 | } |
292 | #endif |
293 | free(lpath); |
294 | } else if (S_ISDIR(statbuf->st_mode)) { |
295 | header.typeflag = DIRTYPE; |
296 | /* Append '/' only if there is a space for it */ |
297 | if (!header.name[sizeof(header.name)-1]) |
298 | header.name[strlen(header.name)] = '/'; |
299 | } else if (S_ISCHR(statbuf->st_mode)) { |
300 | header.typeflag = CHRTYPE; |
301 | PUT_OCTAL(header.devmajor, major(statbuf->st_rdev)); |
302 | PUT_OCTAL(header.devminor, minor(statbuf->st_rdev)); |
303 | } else if (S_ISBLK(statbuf->st_mode)) { |
304 | header.typeflag = BLKTYPE; |
305 | PUT_OCTAL(header.devmajor, major(statbuf->st_rdev)); |
306 | PUT_OCTAL(header.devminor, minor(statbuf->st_rdev)); |
307 | } else if (S_ISFIFO(statbuf->st_mode)) { |
308 | header.typeflag = FIFOTYPE; |
309 | } else if (S_ISREG(statbuf->st_mode)) { |
310 | if (sizeof(statbuf->st_size) > 4 |
311 | && statbuf->st_size > (off_t)0777777777777LL |
312 | ) { |
313 | bb_error_msg_and_die("cannot store file '%s' " |
314 | "of size %"OFF_FMT"d, aborting", |
315 | fileName, statbuf->st_size); |
316 | } |
317 | header.typeflag = REGTYPE; |
318 | PUT_OCTAL(header.size, statbuf->st_size); |
319 | } else { |
320 | bb_error_msg("%s: unknown file type", fileName); |
321 | return FALSE; |
322 | } |
323 | |
324 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS |
325 | /* Write out long name if needed */ |
326 | /* (we, like GNU tar, output long linkname *before* long name) */ |
327 | if (header.name[sizeof(header.name)-1]) |
328 | writeLongname(tbInfo->tarFd, GNULONGNAME, |
329 | header_name, S_ISDIR(statbuf->st_mode)); |
330 | #endif |
331 | |
332 | /* Now write the header out to disk */ |
333 | chksum_and_xwrite(tbInfo->tarFd, &header); |
334 | |
335 | /* Now do the verbose thing (or not) */ |
336 | if (tbInfo->verboseFlag) { |
337 | FILE *vbFd = stdout; |
338 | |
339 | if (tbInfo->tarFd == STDOUT_FILENO) /* If the archive goes to stdout, verbose to stderr */ |
340 | vbFd = stderr; |
341 | /* GNU "tar cvvf" prints "extended" listing a-la "ls -l" */ |
342 | /* We don't have such excesses here: for us "v" == "vv" */ |
343 | /* '/' is probably a GNUism */ |
344 | fprintf(vbFd, "%s%s\n", header_name, |
345 | S_ISDIR(statbuf->st_mode) ? "/" : ""); |
346 | } |
347 | |
348 | return TRUE; |
349 | } |
350 | |
351 | #if ENABLE_FEATURE_TAR_FROM |
352 | static int exclude_file(const llist_t *excluded_files, const char *file) |
353 | { |
354 | while (excluded_files) { |
355 | if (excluded_files->data[0] == '/') { |
356 | if (fnmatch(excluded_files->data, file, |
357 | FNM_PATHNAME | FNM_LEADING_DIR) == 0) |
358 | return 1; |
359 | } else { |
360 | const char *p; |
361 | |
362 | for (p = file; p[0] != '\0'; p++) { |
363 | if ((p == file || p[-1] == '/') && p[0] != '/' && |
364 | fnmatch(excluded_files->data, p, |
365 | FNM_PATHNAME | FNM_LEADING_DIR) == 0) |
366 | return 1; |
367 | } |
368 | } |
369 | excluded_files = excluded_files->link; |
370 | } |
371 | |
372 | return 0; |
373 | } |
374 | #else |
375 | #define exclude_file(excluded_files, file) 0 |
376 | #endif |
377 | |
378 | static int writeFileToTarball(const char *fileName, struct stat *statbuf, |
379 | void *userData, int depth ATTRIBUTE_UNUSED) |
380 | { |
381 | struct TarBallInfo *tbInfo = (struct TarBallInfo *) userData; |
382 | const char *header_name; |
383 | int inputFileFd = -1; |
384 | |
385 | /* |
386 | * Check to see if we are dealing with a hard link. |
387 | * If so - |
388 | * Treat the first occurance of a given dev/inode as a file while |
389 | * treating any additional occurances as hard links. This is done |
390 | * by adding the file information to the HardLinkInfo linked list. |
391 | */ |
392 | tbInfo->hlInfo = NULL; |
393 | if (statbuf->st_nlink > 1) { |
394 | tbInfo->hlInfo = findHardLinkInfo(tbInfo->hlInfoHead, statbuf); |
395 | if (tbInfo->hlInfo == NULL) |
396 | addHardLinkInfo(&tbInfo->hlInfoHead, statbuf, fileName); |
397 | } |
398 | |
399 | /* It is against the rules to archive a socket */ |
400 | if (S_ISSOCK(statbuf->st_mode)) { |
401 | bb_error_msg("%s: socket ignored", fileName); |
402 | return TRUE; |
403 | } |
404 | |
405 | /* It is a bad idea to store the archive we are in the process of creating, |
406 | * so check the device and inode to be sure that this particular file isn't |
407 | * the new tarball */ |
408 | if (tbInfo->statBuf.st_dev == statbuf->st_dev && |
409 | tbInfo->statBuf.st_ino == statbuf->st_ino) { |
410 | bb_error_msg("%s: file is the archive; skipping", fileName); |
411 | return TRUE; |
412 | } |
413 | |
414 | header_name = fileName; |
415 | while (header_name[0] == '/') { |
416 | static int alreadyWarned = FALSE; |
417 | |
418 | if (alreadyWarned == FALSE) { |
419 | bb_error_msg("removing leading '/' from member names"); |
420 | alreadyWarned = TRUE; |
421 | } |
422 | header_name++; |
423 | } |
424 | |
425 | #if !ENABLE_FEATURE_TAR_GNU_EXTENSIONS |
426 | if (strlen(fileName) >= NAME_SIZE) { |
427 | bb_error_msg("names longer than "NAME_SIZE_STR" chars not supported"); |
428 | return TRUE; |
429 | } |
430 | #endif |
431 | |
432 | if (header_name[0] == '\0') |
433 | return TRUE; |
434 | |
435 | if (exclude_file(tbInfo->excludeList, header_name)) |
436 | return SKIP; |
437 | |
438 | /* Is this a regular file? */ |
439 | if (tbInfo->hlInfo == NULL && S_ISREG(statbuf->st_mode)) { |
440 | /* open the file we want to archive, and make sure all is well */ |
441 | inputFileFd = open(fileName, O_RDONLY); |
442 | if (inputFileFd < 0) { |
443 | bb_perror_msg("%s: cannot open", fileName); |
444 | return FALSE; |
445 | } |
446 | } |
447 | |
448 | /* Add an entry to the tarball */ |
449 | if (writeTarHeader(tbInfo, header_name, fileName, statbuf) == FALSE) { |
450 | return FALSE; |
451 | } |
452 | |
453 | /* If it was a regular file, write out the body */ |
454 | if (inputFileFd >= 0) { |
455 | size_t readSize; |
456 | /* Wwrite the file to the archive. */ |
457 | /* We record size into header first, */ |
458 | /* and then write out file. If file shrinks in between, */ |
459 | /* tar will be corrupted. So we don't allow for that. */ |
460 | /* NB: GNU tar 1.16 warns and pads with zeroes */ |
461 | /* or even seeks back and updates header */ |
462 | bb_copyfd_exact_size(inputFileFd, tbInfo->tarFd, statbuf->st_size); |
463 | ////off_t readSize; |
464 | ////readSize = bb_copyfd_size(inputFileFd, tbInfo->tarFd, statbuf->st_size); |
465 | ////if (readSize != statbuf->st_size && readSize >= 0) { |
466 | //// bb_error_msg_and_die("short read from %s, aborting", fileName); |
467 | ////} |
468 | |
469 | /* Check that file did not grow in between? */ |
470 | /* if (safe_read(inputFileFd, 1) == 1) warn but continue? */ |
471 | |
472 | close(inputFileFd); |
473 | |
474 | /* Pad the file up to the tar block size */ |
475 | /* (a few tricks here in the name of code size) */ |
476 | readSize = (-(int)statbuf->st_size) & (TAR_BLOCK_SIZE-1); |
477 | memset(bb_common_bufsiz1, 0, readSize); |
478 | xwrite(tbInfo->tarFd, bb_common_bufsiz1, readSize); |
479 | } |
480 | |
481 | return TRUE; |
482 | } |
483 | |
484 | static int writeTarFile(const int tar_fd, const int verboseFlag, |
485 | const unsigned long dereferenceFlag, const llist_t *include, |
486 | const llist_t *exclude, const int gzip) |
487 | { |
488 | pid_t gzipPid = 0; |
489 | int errorFlag = FALSE; |
490 | struct TarBallInfo tbInfo; |
491 | |
492 | tbInfo.hlInfoHead = NULL; |
493 | |
494 | fchmod(tar_fd, 0644); |
495 | tbInfo.tarFd = tar_fd; |
496 | tbInfo.verboseFlag = verboseFlag; |
497 | |
498 | /* Store the stat info for the tarball's file, so |
499 | * can avoid including the tarball into itself.... */ |
500 | if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0) |
501 | bb_perror_msg_and_die("cannot stat tar file"); |
502 | |
503 | if ((ENABLE_FEATURE_TAR_GZIP || ENABLE_FEATURE_TAR_BZIP2) && gzip) { |
504 | int gzipDataPipe[2] = { -1, -1 }; |
505 | int gzipStatusPipe[2] = { -1, -1 }; |
506 | volatile int vfork_exec_errno = 0; |
507 | char *zip_exec = (gzip == 1) ? "gzip" : "bzip2"; |
508 | |
509 | if (pipe(gzipDataPipe) < 0 || pipe(gzipStatusPipe) < 0) |
510 | bb_perror_msg_and_die("pipe"); |
511 | |
512 | signal(SIGPIPE, SIG_IGN); /* we only want EPIPE on errors */ |
513 | |
514 | #if defined(__GNUC__) && __GNUC__ |
515 | /* Avoid vfork clobbering */ |
516 | (void) &include; |
517 | (void) &errorFlag; |
518 | (void) &zip_exec; |
519 | #endif |
520 | |
521 | gzipPid = vfork(); |
522 | |
523 | if (gzipPid == 0) { |
524 | dup2(gzipDataPipe[0], 0); |
525 | close(gzipDataPipe[1]); |
526 | |
527 | dup2(tbInfo.tarFd, 1); |
528 | |
529 | close(gzipStatusPipe[0]); |
530 | fcntl(gzipStatusPipe[1], F_SETFD, FD_CLOEXEC); /* close on exec shows success */ |
531 | |
532 | execlp(zip_exec, zip_exec, "-f", NULL); |
533 | vfork_exec_errno = errno; |
534 | |
535 | close(gzipStatusPipe[1]); |
536 | exit(-1); |
537 | } else if (gzipPid > 0) { |
538 | close(gzipDataPipe[0]); |
539 | close(gzipStatusPipe[1]); |
540 | |
541 | while (1) { |
542 | char buf; |
543 | |
544 | int n = full_read(gzipStatusPipe[0], &buf, 1); |
545 | |
546 | if (n == 0 && vfork_exec_errno != 0) { |
547 | errno = vfork_exec_errno; |
548 | bb_perror_msg_and_die("cannot exec %s", zip_exec); |
549 | } else if ((n < 0) && (errno == EAGAIN || errno == EINTR)) |
550 | continue; /* try it again */ |
551 | break; |
552 | } |
553 | close(gzipStatusPipe[0]); |
554 | |
555 | tbInfo.tarFd = gzipDataPipe[1]; |
556 | } else bb_perror_msg_and_die("vfork gzip"); |
557 | } |
558 | |
559 | tbInfo.excludeList = exclude; |
560 | |
561 | /* Read the directory/files and iterate over them one at a time */ |
562 | while (include) { |
563 | if (!recursive_action(include->data, TRUE, dereferenceFlag, |
564 | FALSE, writeFileToTarball, writeFileToTarball, &tbInfo, 0)) |
565 | { |
566 | errorFlag = TRUE; |
567 | } |
568 | include = include->link; |
569 | } |
570 | /* Write two empty blocks to the end of the archive */ |
571 | memset(bb_common_bufsiz1, 0, 2*TAR_BLOCK_SIZE); |
572 | xwrite(tbInfo.tarFd, bb_common_bufsiz1, 2*TAR_BLOCK_SIZE); |
573 | |
574 | /* To be pedantically correct, we would check if the tarball |
575 | * is smaller than 20 tar blocks, and pad it if it was smaller, |
576 | * but that isn't necessary for GNU tar interoperability, and |
577 | * so is considered a waste of space */ |
578 | |
579 | /* Close so the child process (if any) will exit */ |
580 | close(tbInfo.tarFd); |
581 | |
582 | /* Hang up the tools, close up shop, head home */ |
583 | if (ENABLE_FEATURE_CLEAN_UP) |
584 | freeHardLinkInfo(&tbInfo.hlInfoHead); |
585 | |
586 | if (errorFlag) |
587 | bb_error_msg("error exit delayed from previous errors"); |
588 | |
589 | if (gzipPid) { |
590 | int status; |
591 | if (waitpid(gzipPid, &status, 0) == -1) |
592 | bb_perror_msg("waitpid"); |
593 | else if (!WIFEXITED(status) || WEXITSTATUS(status)) |
594 | /* gzip was killed or has exited with nonzero! */ |
595 | errorFlag = TRUE; |
596 | } |
597 | return errorFlag; |
598 | } |
599 | #else |
600 | int writeTarFile(const int tar_fd, const int verboseFlag, |
601 | const unsigned long dereferenceFlag, const llist_t *include, |
602 | const llist_t *exclude, const int gzip); |
603 | #endif /* FEATURE_TAR_CREATE */ |
604 | |
605 | #if ENABLE_FEATURE_TAR_FROM |
606 | static llist_t *append_file_list_to_list(llist_t *list) |
607 | { |
608 | FILE *src_stream; |
609 | llist_t *cur = list; |
610 | llist_t *tmp; |
611 | char *line; |
612 | llist_t *newlist = NULL; |
613 | |
614 | while (cur) { |
615 | src_stream = xfopen(cur->data, "r"); |
616 | tmp = cur; |
617 | cur = cur->link; |
618 | free(tmp); |
619 | while ((line = xmalloc_getline(src_stream)) != NULL) { |
620 | /* kill trailing '/' unless the string is just "/" */ |
621 | char *cp = last_char_is(line, '/'); |
622 | if (cp > line) |
623 | *cp = '\0'; |
624 | llist_add_to(&newlist, line); |
625 | } |
626 | fclose(src_stream); |
627 | } |
628 | return newlist; |
629 | } |
630 | #else |
631 | #define append_file_list_to_list(x) 0 |
632 | #endif |
633 | |
634 | #if ENABLE_FEATURE_TAR_COMPRESS |
635 | static char get_header_tar_Z(archive_handle_t *archive_handle) |
636 | { |
637 | /* Can't lseek over pipes */ |
638 | archive_handle->seek = seek_by_read; |
639 | |
640 | /* do the decompression, and cleanup */ |
641 | if (xread_char(archive_handle->src_fd) != 0x1f |
642 | || xread_char(archive_handle->src_fd) != 0x9d |
643 | ) { |
644 | bb_error_msg_and_die("invalid magic"); |
645 | } |
646 | |
647 | archive_handle->src_fd = open_transformer(archive_handle->src_fd, uncompress); |
648 | archive_handle->offset = 0; |
649 | while (get_header_tar(archive_handle) == EXIT_SUCCESS) |
650 | /* nothing */; |
651 | |
652 | /* Can only do one file at a time */ |
653 | return EXIT_FAILURE; |
654 | } |
655 | #else |
656 | #define get_header_tar_Z NULL |
657 | #endif |
658 | |
659 | #ifdef CHECK_FOR_CHILD_EXITCODE |
660 | /* Looks like it isn't needed - tar detects malformed (truncated) |
661 | * archive if e.g. bunzip2 fails */ |
662 | static int child_error; |
663 | |
664 | static void handle_SIGCHLD(int status) |
665 | { |
666 | /* Actually, 'status' is a signo. We reuse it for other needs */ |
667 | |
668 | /* Wait for any child without blocking */ |
669 | if (waitpid(-1, &status, WNOHANG) < 0) |
670 | /* wait failed?! I'm confused... */ |
671 | return; |
672 | |
673 | if (WIFEXITED(status) && WEXITSTATUS(status)==0) |
674 | /* child exited with 0 */ |
675 | return; |
676 | /* Cannot happen? |
677 | if(!WIFSIGNALED(status) && !WIFEXITED(status)) return; */ |
678 | child_error = 1; |
679 | } |
680 | #endif |
681 | |
682 | enum { |
683 | OPTBIT_KEEP_OLD = 7, |
684 | USE_FEATURE_TAR_CREATE( OPTBIT_CREATE ,) |
685 | USE_FEATURE_TAR_CREATE( OPTBIT_DEREFERENCE ,) |
686 | USE_FEATURE_TAR_BZIP2( OPTBIT_BZIP2 ,) |
687 | USE_FEATURE_TAR_LZMA( OPTBIT_LZMA ,) |
688 | USE_FEATURE_TAR_FROM( OPTBIT_INCLUDE_FROM,) |
689 | USE_FEATURE_TAR_FROM( OPTBIT_EXCLUDE_FROM,) |
690 | USE_FEATURE_TAR_GZIP( OPTBIT_GZIP ,) |
691 | USE_FEATURE_TAR_COMPRESS(OPTBIT_COMPRESS ,) |
692 | OPTBIT_NOPRESERVE_OWN, |
693 | OPTBIT_NOPRESERVE_PERM, |
694 | OPT_TEST = 1 << 0, // t |
695 | OPT_EXTRACT = 1 << 1, // x |
696 | OPT_BASEDIR = 1 << 2, // C |
697 | OPT_TARNAME = 1 << 3, // f |
698 | OPT_2STDOUT = 1 << 4, // O |
699 | OPT_P = 1 << 5, // p |
700 | OPT_VERBOSE = 1 << 6, // v |
701 | OPT_KEEP_OLD = 1 << 7, // k |
702 | OPT_CREATE = USE_FEATURE_TAR_CREATE( (1<<OPTBIT_CREATE )) + 0, // c |
703 | OPT_DEREFERENCE = USE_FEATURE_TAR_CREATE( (1<<OPTBIT_DEREFERENCE )) + 0, // h |
704 | OPT_BZIP2 = USE_FEATURE_TAR_BZIP2( (1<<OPTBIT_BZIP2 )) + 0, // j |
705 | OPT_LZMA = USE_FEATURE_TAR_LZMA( (1<<OPTBIT_LZMA )) + 0, // a |
706 | OPT_INCLUDE_FROM = USE_FEATURE_TAR_FROM( (1<<OPTBIT_INCLUDE_FROM)) + 0, // T |
707 | OPT_EXCLUDE_FROM = USE_FEATURE_TAR_FROM( (1<<OPTBIT_EXCLUDE_FROM)) + 0, // X |
708 | OPT_GZIP = USE_FEATURE_TAR_GZIP( (1<<OPTBIT_GZIP )) + 0, // z |
709 | OPT_COMPRESS = USE_FEATURE_TAR_COMPRESS((1<<OPTBIT_COMPRESS )) + 0, // Z |
710 | OPT_NOPRESERVE_OWN = 1 << OPTBIT_NOPRESERVE_OWN , // no-same-owner |
711 | OPT_NOPRESERVE_PERM = 1 << OPTBIT_NOPRESERVE_PERM, // no-same-permissions |
712 | }; |
713 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS |
714 | static const struct option tar_long_options[] = { |
715 | { "list", 0, NULL, 't' }, |
716 | { "extract", 0, NULL, 'x' }, |
717 | { "directory", 1, NULL, 'C' }, |
718 | { "file", 1, NULL, 'f' }, |
719 | { "to-stdout", 0, NULL, 'O' }, |
720 | { "same-permissions", 0, NULL, 'p' }, |
721 | { "verbose", 0, NULL, 'v' }, |
722 | { "keep-old", 0, NULL, 'k' }, |
723 | # if ENABLE_FEATURE_TAR_CREATE |
724 | { "create", 0, NULL, 'c' }, |
725 | { "dereference", 0, NULL, 'h' }, |
726 | # endif |
727 | # if ENABLE_FEATURE_TAR_BZIP2 |
728 | { "bzip2", 0, NULL, 'j' }, |
729 | # endif |
730 | # if ENABLE_FEATURE_TAR_LZMA |
731 | { "lzma", 0, NULL, 'a' }, |
732 | # endif |
733 | # if ENABLE_FEATURE_TAR_FROM |
734 | { "files-from", 1, NULL, 'T' }, |
735 | { "exclude-from", 1, NULL, 'X' }, |
736 | # endif |
737 | # if ENABLE_FEATURE_TAR_GZIP |
738 | { "gzip", 0, NULL, 'z' }, |
739 | # endif |
740 | # if ENABLE_FEATURE_TAR_COMPRESS |
741 | { "compress", 0, NULL, 'Z' }, |
742 | # endif |
743 | { "no-same-owner", 0, NULL, 0xfd }, |
744 | { "no-same-permissions",0, NULL, 0xfe }, |
745 | /* --exclude takes next bit position in option mask, */ |
746 | /* therefore we have to either put it _after_ --no-same-perm */ |
747 | /* or add OPT[BIT]_EXCLUDE before OPT[BIT]_NOPRESERVE_OWN */ |
748 | # if ENABLE_FEATURE_TAR_FROM |
749 | { "exclude", 1, NULL, 0xff }, |
750 | # endif |
751 | { 0, 0, 0, 0 } |
752 | }; |
753 | #endif |
754 | |
755 | int tar_main(int argc, char **argv) |
756 | { |
757 | char (*get_header_ptr)(archive_handle_t *) = get_header_tar; |
758 | archive_handle_t *tar_handle; |
759 | char *base_dir = NULL; |
760 | const char *tar_filename = "-"; |
761 | unsigned opt; |
762 | int verboseFlag = 0; |
763 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS && ENABLE_FEATURE_TAR_FROM |
764 | llist_t *excludes = NULL; |
765 | #endif |
766 | |
767 | /* Initialise default values */ |
768 | tar_handle = init_handle(); |
769 | tar_handle->flags = ARCHIVE_CREATE_LEADING_DIRS |
770 | | ARCHIVE_PRESERVE_DATE |
771 | | ARCHIVE_EXTRACT_UNCONDITIONAL; |
772 | |
773 | /* Prepend '-' to the first argument if required */ |
774 | opt_complementary = "--:" // first arg is options |
775 | "tt:vv:" // count -t,-v |
776 | "?:" // bail out with usage instead of error return |
777 | "X::T::" // cumulative lists |
778 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS && ENABLE_FEATURE_TAR_FROM |
779 | "\xff::" // cumulative lists for --exclude |
780 | #endif |
781 | USE_FEATURE_TAR_CREATE("c:") "t:x:" // at least one of these is reqd |
782 | USE_FEATURE_TAR_CREATE("c--tx:t--cx:x--ct") // mutually exclusive |
783 | SKIP_FEATURE_TAR_CREATE("t--x:x--t"); // mutually exclusive |
784 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS |
785 | applet_long_options = tar_long_options; |
786 | #endif |
787 | opt = getopt32(argc, argv, |
788 | "txC:f:Opvk" |
789 | USE_FEATURE_TAR_CREATE( "ch" ) |
790 | USE_FEATURE_TAR_BZIP2( "j" ) |
791 | USE_FEATURE_TAR_LZMA( "a" ) |
792 | USE_FEATURE_TAR_FROM( "T:X:") |
793 | USE_FEATURE_TAR_GZIP( "z" ) |
794 | USE_FEATURE_TAR_COMPRESS("Z" ) |
795 | , &base_dir // -C dir |
796 | , &tar_filename // -f filename |
797 | USE_FEATURE_TAR_FROM(, &(tar_handle->accept)) // T |
798 | USE_FEATURE_TAR_FROM(, &(tar_handle->reject)) // X |
799 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS && ENABLE_FEATURE_TAR_FROM |
800 | , &excludes // --exclude |
801 | #endif |
802 | , &verboseFlag // combined count for -t and -v |
803 | , &verboseFlag // combined count for -t and -v |
804 | ); |
805 | |
806 | if (verboseFlag) tar_handle->action_header = header_verbose_list; |
807 | if (verboseFlag == 1) tar_handle->action_header = header_list; |
808 | |
809 | if (opt & OPT_EXTRACT) |
810 | tar_handle->action_data = data_extract_all; |
811 | |
812 | if (opt & OPT_2STDOUT) |
813 | tar_handle->action_data = data_extract_to_stdout; |
814 | |
815 | if (opt & OPT_KEEP_OLD) |
816 | tar_handle->flags &= ~ARCHIVE_EXTRACT_UNCONDITIONAL; |
817 | |
818 | if (opt & OPT_NOPRESERVE_OWN) |
819 | tar_handle->flags |= ARCHIVE_NOPRESERVE_OWN; |
820 | |
821 | if (opt & OPT_NOPRESERVE_PERM) |
822 | tar_handle->flags |= ARCHIVE_NOPRESERVE_PERM; |
823 | |
824 | if (opt & OPT_GZIP) |
825 | get_header_ptr = get_header_tar_gz; |
826 | |
827 | if (opt & OPT_BZIP2) |
828 | get_header_ptr = get_header_tar_bz2; |
829 | |
830 | if (opt & OPT_LZMA) |
831 | get_header_ptr = get_header_tar_lzma; |
832 | |
833 | if (opt & OPT_COMPRESS) |
834 | get_header_ptr = get_header_tar_Z; |
835 | |
836 | #if ENABLE_FEATURE_TAR_FROM |
837 | tar_handle->reject = append_file_list_to_list(tar_handle->reject); |
838 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS |
839 | /* Append excludes to reject */ |
840 | while (excludes) { |
841 | llist_t *next = excludes->link; |
842 | excludes->link = tar_handle->reject; |
843 | tar_handle->reject = excludes; |
844 | excludes = next; |
845 | } |
846 | #endif |
847 | tar_handle->accept = append_file_list_to_list(tar_handle->accept); |
848 | #endif |
849 | |
850 | /* Check if we are reading from stdin */ |
851 | if (argv[optind] && *argv[optind] == '-') { |
852 | /* Default is to read from stdin, so just skip to next arg */ |
853 | optind++; |
854 | } |
855 | |
856 | /* Setup an array of filenames to work with */ |
857 | /* TODO: This is the same as in ar, separate function ? */ |
858 | while (optind < argc) { |
859 | /* kill trailing '/' unless the string is just "/" */ |
860 | char *cp = last_char_is(argv[optind], '/'); |
861 | if (cp > argv[optind]) |
862 | *cp = '\0'; |
863 | llist_add_to(&tar_handle->accept, argv[optind]); |
864 | optind++; |
865 | } |
866 | tar_handle->accept = rev_llist(tar_handle->accept); |
867 | |
868 | if (tar_handle->accept || tar_handle->reject) |
869 | tar_handle->filter = filter_accept_reject_list; |
870 | |
871 | /* Open the tar file */ |
872 | { |
873 | FILE *tar_stream; |
874 | int flags; |
875 | |
876 | if (opt & OPT_CREATE) { |
877 | /* Make sure there is at least one file to tar up. */ |
878 | if (tar_handle->accept == NULL) |
879 | bb_error_msg_and_die("empty archive"); |
880 | |
881 | tar_stream = stdout; |
882 | /* Mimicking GNU tar 1.15.1: */ |
883 | flags = O_WRONLY|O_CREAT|O_TRUNC; |
884 | /* was doing unlink; open(O_WRONLY|O_CREAT|O_EXCL); why? */ |
885 | } else { |
886 | tar_stream = stdin; |
887 | flags = O_RDONLY; |
888 | } |
889 | |
890 | if (LONE_DASH(tar_filename)) { |
891 | tar_handle->src_fd = fileno(tar_stream); |
892 | tar_handle->seek = seek_by_read; |
893 | } else { |
894 | tar_handle->src_fd = xopen(tar_filename, flags); |
895 | } |
896 | } |
897 | |
898 | if (base_dir) |
899 | xchdir(base_dir); |
900 | |
901 | #ifdef CHECK_FOR_CHILD_EXITCODE |
902 | /* We need to know whether child (gzip/bzip/etc) exits abnormally */ |
903 | signal(SIGCHLD, handle_SIGCHLD); |
904 | #endif |
905 | |
906 | /* create an archive */ |
907 | if (opt & OPT_CREATE) { |
908 | int zipMode = 0; |
909 | if (ENABLE_FEATURE_TAR_GZIP && get_header_ptr == get_header_tar_gz) |
910 | zipMode = 1; |
911 | if (ENABLE_FEATURE_TAR_BZIP2 && get_header_ptr == get_header_tar_bz2) |
912 | zipMode = 2; |
913 | /* NB: writeTarFile() closes tar_handle->src_fd */ |
914 | return writeTarFile(tar_handle->src_fd, verboseFlag, opt & OPT_DEREFERENCE, |
915 | tar_handle->accept, |
916 | tar_handle->reject, zipMode); |
917 | } |
918 | |
919 | while (get_header_ptr(tar_handle) == EXIT_SUCCESS) |
920 | /* nothing */; |
921 | |
922 | /* Check that every file that should have been extracted was */ |
923 | while (tar_handle->accept) { |
924 | if (!find_list_entry(tar_handle->reject, tar_handle->accept->data) |
925 | && !find_list_entry(tar_handle->passed, tar_handle->accept->data) |
926 | ) { |
927 | bb_error_msg_and_die("%s: not found in archive", |
928 | tar_handle->accept->data); |
929 | } |
930 | tar_handle->accept = tar_handle->accept->link; |
931 | } |
932 | if (ENABLE_FEATURE_CLEAN_UP /* && tar_handle->src_fd != STDIN_FILENO */) |
933 | close(tar_handle->src_fd); |
934 | |
935 | return EXIT_SUCCESS; |
936 | } |