Magellan Linux

Contents of /trunk/mkinitrd-magellan/busybox/archival/tar.c

Parent Directory Parent Directory | Revision Log 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)
-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 }