archival/tar.c

   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  *  This is free software under the GNU General Public License.
  23  *
  24  * This program is free software; you can redistribute it and/or modify
  25  * it under the terms of the GNU General Public License as published by
  26  * the Free Software Foundation; either version 2 of the License, or
  27  * (at your option) any later version.
  28  *
  29  * This program is distributed in the hope that it will be useful,
  30  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  31  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  32  * General Public License for more details.
  33  *
  34  * You should have received a copy of the GNU General Public License
  35  * along with this program; if not, write to the Free Software
  36  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  37  *
  38  */
  39
  40 #include <fcntl.h>
  41 #include <getopt.h>
  42 #include <search.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <unistd.h>
  46 #include <fnmatch.h>
  47 #include <string.h>
  48 #include <errno.h>
  49 #include <signal.h>
  50 #include <sys/wait.h>
  51 #include <sys/socket.h>
  52 #include <sys/sysmacros.h>     /* major() and minor() */
  53 #include "unarchive.h"
  54 #include "busybox.h"
  55
  56 #ifdef CONFIG_FEATURE_TAR_CREATE
  57
  58 /* Tar file constants  */
  59 # define TAR_MAGIC          "ustar"     /* ustar and a null */
  60 # define TAR_VERSION        "  "        /* Be compatable with GNU tar format */
  61
  62 static const int TAR_BLOCK_SIZE = 512;
  63 static const int TAR_MAGIC_LEN = 6;
  64 static const int TAR_VERSION_LEN = 2;
  65
  66 /* POSIX tar Header Block, from POSIX 1003.1-1990  */
  67 enum { NAME_SIZE = 100 };       /* because gcc won't let me use 'static const int' */
  68 struct TarHeader {              /* byte offset */
  69         char name[NAME_SIZE];   /*   0-99 */
  70         char mode[8];           /* 100-107 */
  71         char uid[8];            /* 108-115 */
  72         char gid[8];            /* 116-123 */
  73         char size[12];          /* 124-135 */
  74         char mtime[12];         /* 136-147 */
  75         char chksum[8];         /* 148-155 */
  76         char typeflag;          /* 156-156 */
  77         char linkname[NAME_SIZE];       /* 157-256 */
  78         char magic[6];          /* 257-262 */
  79         char version[2];        /* 263-264 */
  80         char uname[32];         /* 265-296 */
  81         char gname[32];         /* 297-328 */
  82         char devmajor[8];       /* 329-336 */
  83         char devminor[8];       /* 337-344 */
  84         char prefix[155];       /* 345-499 */
  85         char padding[12];       /* 500-512 (pad to exactly the TAR_BLOCK_SIZE) */
  86 };
  87 typedef struct TarHeader TarHeader;
  88
  89 /*
  90 ** writeTarFile(),  writeFileToTarball(), and writeTarHeader() are
  91 ** the only functions that deal with the HardLinkInfo structure.
  92 ** Even these functions use the xxxHardLinkInfo() functions.
  93 */
  94 typedef struct HardLinkInfo HardLinkInfo;
  95 struct HardLinkInfo {
  96         HardLinkInfo *next;     /* Next entry in list */
  97         dev_t dev;                      /* Device number */
  98         ino_t ino;                      /* Inode number */
  99         short linkCount;        /* (Hard) Link Count */
 100         char name[1];           /* Start of filename (must be last) */
 101 };
 102
 103 /* Some info to be carried along when creating a new tarball */
 104 struct TarBallInfo {
 105         char *fileName;         /* File name of the tarball */
 106         int tarFd;                      /* Open-for-write file descriptor
 107                                                    for the tarball */
 108         struct stat statBuf;    /* Stat info for the tarball, letting
 109                                                            us know the inode and device that the
 110                                                            tarball lives, so we can avoid trying
 111                                                            to include the tarball into itself */
 112         int verboseFlag;        /* Whether to print extra stuff or not */
 113         const llist_t *excludeList;     /* List of files to not include */
 114         HardLinkInfo *hlInfoHead;       /* Hard Link Tracking Information */
 115         HardLinkInfo *hlInfo;   /* Hard Link Info for the current file */
 116 };
 117 typedef struct TarBallInfo TarBallInfo;
 118
 119 /* A nice enum with all the possible tar file content types */
 120 enum TarFileType {
 121         REGTYPE = '0',          /* regular file */
 122         REGTYPE0 = '\0',        /* regular file (ancient bug compat) */
 123         LNKTYPE = '1',          /* hard link */
 124         SYMTYPE = '2',          /* symbolic link */
 125         CHRTYPE = '3',          /* character special */
 126         BLKTYPE = '4',          /* block special */
 127         DIRTYPE = '5',          /* directory */
 128         FIFOTYPE = '6',         /* FIFO special */
 129         CONTTYPE = '7',         /* reserved */
 130         GNULONGLINK = 'K',      /* GNU long (>100 chars) link name */
 131         GNULONGNAME = 'L',      /* GNU long (>100 chars) file name */
 132 };
 133 typedef enum TarFileType TarFileType;
 134
 135 /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */
 136 static inline void addHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr,
 137                                         struct stat *statbuf,
 138                                         const char *name)
 139 {
 140         /* Note: hlInfoHeadPtr can never be NULL! */
 141         HardLinkInfo *hlInfo;
 142
 143         hlInfo = (HardLinkInfo *) xmalloc(sizeof(HardLinkInfo) + strlen(name));
 144         hlInfo->next = *hlInfoHeadPtr;
 145         *hlInfoHeadPtr = hlInfo;
 146         hlInfo->dev = statbuf->st_dev;
 147         hlInfo->ino = statbuf->st_ino;
 148         hlInfo->linkCount = statbuf->st_nlink;
 149         strcpy(hlInfo->name, name);
 150 }
 151
 152 static void freeHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr)
 153 {
 154         HardLinkInfo *hlInfo = NULL;
 155         HardLinkInfo *hlInfoNext = NULL;
 156
 157         if (hlInfoHeadPtr) {
 158                 hlInfo = *hlInfoHeadPtr;
 159                 while (hlInfo) {
 160                         hlInfoNext = hlInfo->next;
 161                         free(hlInfo);
 162                         hlInfo = hlInfoNext;
 163                 }
 164                 *hlInfoHeadPtr = NULL;
 165         }
 166         return;
 167 }
 168
 169 /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */
 170 static inline HardLinkInfo *findHardLinkInfo(HardLinkInfo * hlInfo, struct stat *statbuf)
 171 {
 172         while (hlInfo) {
 173                 if ((statbuf->st_ino == hlInfo->ino) && (statbuf->st_dev == hlInfo->dev))
 174                         break;
 175                 hlInfo = hlInfo->next;
 176         }
 177         return (hlInfo);
 178 }
 179
 180 /* Put an octal string into the specified buffer.
 181  * The number is zero and space padded and possibly null padded.
 182  * Returns TRUE if successful.  */
 183 static int putOctal(char *cp, int len, long value)
 184 {
 185         int tempLength;
 186         char tempBuffer[32];
 187         char *tempString = tempBuffer;
 188
 189         /* Create a string of the specified length with an initial space,
 190          * leading zeroes and the octal number, and a trailing null.  */
 191         sprintf(tempString, "%0*lo", len - 1, value);
 192
 193         /* If the string is too large, suppress the leading space.  */
 194         tempLength = strlen(tempString) + 1;
 195         if (tempLength > len) {
 196                 tempLength--;
 197                 tempString++;
 198         }
 199
 200         /* If the string is still too large, suppress the trailing null.  */
 201         if (tempLength > len)
 202                 tempLength--;
 203
 204         /* If the string is still too large, fail.  */
 205         if (tempLength > len)
 206                 return FALSE;
 207
 208         /* Copy the string to the field.  */
 209         memcpy(cp, tempString, len);
 210
 211         return TRUE;
 212 }
 213
 214 /* Write out a tar header for the specified file/directory/whatever */
 215 static inline int writeTarHeader(struct TarBallInfo *tbInfo,
 216                                                                  const char *header_name,
 217                                                                  const char *real_name, struct stat *statbuf)
 218 {
 219         long chksum = 0;
 220         struct TarHeader header;
 221         const unsigned char *cp = (const unsigned char *) &header;
 222         ssize_t size = sizeof(struct TarHeader);
 223
 224         memset(&header, 0, size);
 225
 226         strncpy(header.name, header_name, sizeof(header.name));
 227
 228         putOctal(header.mode, sizeof(header.mode), statbuf->st_mode);
 229         putOctal(header.uid, sizeof(header.uid), statbuf->st_uid);
 230         putOctal(header.gid, sizeof(header.gid), statbuf->st_gid);
 231         putOctal(header.size, sizeof(header.size), 0);  /* Regular file size is handled later */
 232         putOctal(header.mtime, sizeof(header.mtime), statbuf->st_mtime);
 233         strncpy(header.magic, TAR_MAGIC TAR_VERSION,
 234                         TAR_MAGIC_LEN + TAR_VERSION_LEN);
 235
 236         /* Enter the user and group names (default to root if it fails) */
 237         if (my_getpwuid(header.uname, statbuf->st_uid, sizeof(header.uname)) == NULL)
 238                 strcpy(header.uname, "root");
 239         if (my_getgrgid(header.gname, statbuf->st_gid, sizeof(header.gname)) == NULL)
 240                 strcpy(header.gname, "root");
 241
 242         if (tbInfo->hlInfo) {
 243                 /* This is a hard link */
 244                 header.typeflag = LNKTYPE;
 245                 strncpy(header.linkname, tbInfo->hlInfo->name,
 246                                 sizeof(header.linkname));
 247         } else if (S_ISLNK(statbuf->st_mode)) {
 248                 char *lpath = xreadlink(real_name);
 249
 250                 if (!lpath)             /* Already printed err msg inside xreadlink() */
 251                         return (FALSE);
 252                 header.typeflag = SYMTYPE;
 253                 strncpy(header.linkname, lpath, sizeof(header.linkname));
 254                 free(lpath);
 255         } else if (S_ISDIR(statbuf->st_mode)) {
 256                 header.typeflag = DIRTYPE;
 257                 strncat(header.name, "/", sizeof(header.name));
 258         } else if (S_ISCHR(statbuf->st_mode)) {
 259                 header.typeflag = CHRTYPE;
 260                 putOctal(header.devmajor, sizeof(header.devmajor),
 261                                  major(statbuf->st_rdev));
 262                 putOctal(header.devminor, sizeof(header.devminor),
 263                                  minor(statbuf->st_rdev));
 264         } else if (S_ISBLK(statbuf->st_mode)) {
 265                 header.typeflag = BLKTYPE;
 266                 putOctal(header.devmajor, sizeof(header.devmajor),
 267                                  major(statbuf->st_rdev));
 268                 putOctal(header.devminor, sizeof(header.devminor),
 269                                  minor(statbuf->st_rdev));
 270         } else if (S_ISFIFO(statbuf->st_mode)) {
 271                 header.typeflag = FIFOTYPE;
 272         } else if (S_ISREG(statbuf->st_mode)) {
 273                 header.typeflag = REGTYPE;
 274                 putOctal(header.size, sizeof(header.size), statbuf->st_size);
 275         } else {
 276                 bb_error_msg("%s: Unknown file type", real_name);
 277                 return (FALSE);
 278         }
 279
 280         /* Calculate and store the checksum (i.e., the sum of all of the bytes of
 281          * the header).  The checksum field must be filled with blanks for the
 282          * calculation.  The checksum field is formatted differently from the
 283          * other fields: it has [6] digits, a null, then a space -- rather than
 284          * digits, followed by a null like the other fields... */
 285         memset(header.chksum, ' ', sizeof(header.chksum));
 286         cp = (const unsigned char *) &header;
 287         while (size-- > 0)
 288                 chksum += *cp++;
 289         putOctal(header.chksum, 7, chksum);
 290
 291         /* Now write the header out to disk */
 292         if ((size =
 293                  bb_full_write(tbInfo->tarFd, (char *) &header,
 294                                         sizeof(struct TarHeader))) < 0) {
 295                 bb_error_msg(bb_msg_io_error, real_name);
 296                 return (FALSE);
 297         }
 298         /* Pad the header up to the tar block size */
 299         for (; size < TAR_BLOCK_SIZE; size++) {
 300                 write(tbInfo->tarFd, "\0", 1);
 301         }
 302         /* Now do the verbose thing (or not) */
 303
 304         if (tbInfo->verboseFlag) {
 305                 FILE *vbFd = stdout;
 306
 307                 if (tbInfo->tarFd == STDOUT_FILENO)     /* If the archive goes to stdout, verbose to stderr */
 308                         vbFd = stderr;
 309
 310                 fprintf(vbFd, "%s\n", header.name);
 311         }
 312
 313         return (TRUE);
 314 }
 315
 316 # ifdef CONFIG_FEATURE_TAR_FROM
 317 static inline int exclude_file(const llist_t *excluded_files, const char *file)
 318 {
 319         while (excluded_files) {
 320                 if (excluded_files->data[0] == '/') {
 321                         if (fnmatch(excluded_files->data, file,
 322                                                 FNM_PATHNAME | FNM_LEADING_DIR) == 0)
 323                                 return 1;
 324                 } else {
 325                         const char *p;
 326
 327                         for (p = file; p[0] != '\0'; p++) {
 328                                 if ((p == file || p[-1] == '/') && p[0] != '/' &&
 329                                         fnmatch(excluded_files->data, p,
 330                                                         FNM_PATHNAME | FNM_LEADING_DIR) == 0)
 331                                         return 1;
 332                         }
 333                 }
 334                 excluded_files = excluded_files->link;
 335         }
 336
 337         return 0;
 338 }
 339 # endif
 340
 341 static int writeFileToTarball(const char *fileName, struct stat *statbuf,
 342                                                           void *userData)
 343 {
 344         struct TarBallInfo *tbInfo = (struct TarBallInfo *) userData;
 345         const char *header_name;
 346
 347         /*
 348            ** Check to see if we are dealing with a hard link.
 349            ** If so -
 350            ** Treat the first occurance of a given dev/inode as a file while
 351            ** treating any additional occurances as hard links.  This is done
 352            ** by adding the file information to the HardLinkInfo linked list.
 353          */
 354         tbInfo->hlInfo = NULL;
 355         if (statbuf->st_nlink > 1) {
 356                 tbInfo->hlInfo = findHardLinkInfo(tbInfo->hlInfoHead, statbuf);
 357                 if (tbInfo->hlInfo == NULL)
 358                         addHardLinkInfo(&tbInfo->hlInfoHead, statbuf, fileName);
 359         }
 360
 361         /* It is against the rules to archive a socket */
 362         if (S_ISSOCK(statbuf->st_mode)) {
 363                 bb_error_msg("%s: socket ignored", fileName);
 364                 return (TRUE);
 365         }
 366
 367         /* It is a bad idea to store the archive we are in the process of creating,
 368          * so check the device and inode to be sure that this particular file isn't
 369          * the new tarball */
 370         if (tbInfo->statBuf.st_dev == statbuf->st_dev &&
 371                 tbInfo->statBuf.st_ino == statbuf->st_ino) {
 372                 bb_error_msg("%s: file is the archive; skipping", fileName);
 373                 return (TRUE);
 374         }
 375
 376         header_name = fileName;
 377         while (header_name[0] == '/') {
 378                 static int alreadyWarned = FALSE;
 379
 380                 if (alreadyWarned == FALSE) {
 381                         bb_error_msg("Removing leading '/' from member names");
 382                         alreadyWarned = TRUE;
 383                 }
 384                 header_name++;
 385         }
 386
 387         if (strlen(fileName) >= NAME_SIZE) {
 388                 bb_error_msg(bb_msg_name_longer_than_foo, NAME_SIZE);
 389                 return (TRUE);
 390         }
 391
 392         if (header_name[0] == '\0')
 393                 return TRUE;
 394
 395 # ifdef CONFIG_FEATURE_TAR_FROM
 396         if (exclude_file(tbInfo->excludeList, header_name)) {
 397                 return SKIP;
 398         }
 399 # endif                                                 /* CONFIG_FEATURE_TAR_FROM */
 400
 401         if (writeTarHeader(tbInfo, header_name, fileName, statbuf) == FALSE) {
 402                 return (FALSE);
 403         }
 404
 405         /* Now, if the file is a regular file, copy it out to the tarball */
 406         if ((tbInfo->hlInfo == NULL)
 407                 && (S_ISREG(statbuf->st_mode))) {
 408                 int inputFileFd;
 409                 ssize_t readSize = 0;
 410
 411                 /* open the file we want to archive, and make sure all is well */
 412                 if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
 413                         bb_perror_msg("%s: Cannot open", fileName);
 414                         return (FALSE);
 415                 }
 416
 417                 /* write the file to the archive */
 418                 readSize = bb_copyfd_eof(inputFileFd, tbInfo->tarFd);
 419
 420                 /* Pad the file up to the tar block size */
 421                 for (; (readSize % TAR_BLOCK_SIZE) != 0; readSize++) {
 422                         write(tbInfo->tarFd, "\0", 1);
 423                 }
 424                 close(inputFileFd);
 425         }
 426
 427         return (TRUE);
 428 }
 429
 430 static inline int writeTarFile(const int tar_fd, const int verboseFlag,
 431         const unsigned long dereferenceFlag, const llist_t *include,
 432         const llist_t *exclude, const int gzip)
 433 {
 434 #ifdef CONFIG_FEATURE_TAR_GZIP
 435         int gzipDataPipe[2] = { -1, -1 };
 436         int gzipStatusPipe[2] = { -1, -1 };
 437         pid_t gzipPid = 0;
 438         volatile int vfork_exec_errno = 0;
 439 #endif
 440
 441         int errorFlag = FALSE;
 442         ssize_t size;
 443         struct TarBallInfo tbInfo;
 444
 445         tbInfo.hlInfoHead = NULL;
 446
 447         fchmod(tar_fd, 0644);
 448         tbInfo.tarFd = tar_fd;
 449         tbInfo.verboseFlag = verboseFlag;
 450
 451         /* Store the stat info for the tarball's file, so
 452          * can avoid including the tarball into itself....  */
 453         if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0)
 454                 bb_perror_msg_and_die("Couldnt stat tar file");
 455
 456 #ifdef CONFIG_FEATURE_TAR_GZIP
 457         if (gzip) {
 458                 if (pipe(gzipDataPipe) < 0 || pipe(gzipStatusPipe) < 0) {
 459                         bb_perror_msg_and_die("Failed to create gzip pipe");
 460                 }
 461
 462                 signal(SIGPIPE, SIG_IGN);       /* we only want EPIPE on errors */
 463
 464 # if __GNUC__
 465                         /* Avoid vfork clobbering */
 466                         (void) &include;
 467                         (void) &errorFlag;
 468 # endif
 469
 470                 gzipPid = vfork();
 471
 472                 if (gzipPid == 0) {
 473                         dup2(gzipDataPipe[0], 0);
 474                         close(gzipDataPipe[1]);
 475
 476                         if (tbInfo.tarFd != 1)
 477                                 dup2(tbInfo.tarFd, 1);
 478
 479                         close(gzipStatusPipe[0]);
 480                         fcntl(gzipStatusPipe[1], F_SETFD, FD_CLOEXEC);  /* close on exec shows sucess */
 481
 482                         execl("/bin/gzip", "gzip", "-f", 0);
 483                         vfork_exec_errno = errno;
 484
 485                         close(gzipStatusPipe[1]);
 486                         exit(-1);
 487                 } else if (gzipPid > 0) {
 488                         close(gzipDataPipe[0]);
 489                         close(gzipStatusPipe[1]);
 490
 491                         while (1) {
 492                                 char buf;
 493
 494                                 int n = bb_full_read(gzipStatusPipe[0], &buf, 1);
 495
 496                                 if (n == 0 && vfork_exec_errno != 0) {
 497                                         errno = vfork_exec_errno;
 498                                         bb_perror_msg_and_die("Could not exec gzip process");
 499                                 } else if ((n < 0) && (errno == EAGAIN || errno == EINTR))
 500                                         continue;       /* try it again */
 501                                 break;
 502                         }
 503                         close(gzipStatusPipe[0]);
 504
 505                         tbInfo.tarFd = gzipDataPipe[1];
 506                 } else {
 507                         bb_perror_msg_and_die("Failed to vfork gzip process");
 508                 }
 509         }
 510 #endif
 511
 512         tbInfo.excludeList = exclude;
 513
 514         /* Read the directory/files and iterate over them one at a time */
 515         while (include) {
 516                 if (!recursive_action(include->data, TRUE, dereferenceFlag, FALSE,
 517                                                           writeFileToTarball, writeFileToTarball,
 518                                                           (void *) &tbInfo)) {
 519                         errorFlag = TRUE;
 520                 }
 521                 include = include->link;
 522         }
 523         /* Write two empty blocks to the end of the archive */
 524         for (size = 0; size < (2 * TAR_BLOCK_SIZE); size++) {
 525                 write(tbInfo.tarFd, "\0", 1);
 526         }
 527
 528         /* To be pedantically correct, we would check if the tarball
 529          * is smaller than 20 tar blocks, and pad it if it was smaller,
 530          * but that isn't necessary for GNU tar interoperability, and
 531          * so is considered a waste of space */
 532
 533         /* Hang up the tools, close up shop, head home */
 534         close(tbInfo.tarFd);
 535         if (errorFlag)
 536                 bb_error_msg("Error exit delayed from previous errors");
 537
 538         freeHardLinkInfo(&tbInfo.hlInfoHead);
 539
 540 #ifdef CONFIG_FEATURE_TAR_GZIP
 541         if (gzip && gzipPid) {
 542                 if (waitpid(gzipPid, NULL, 0) == -1)
 543                         printf("Couldnt wait ?");
 544         }
 545 #endif
 546
 547         return !errorFlag;
 548 }
 549 #endif                                                  /* tar_create */
 550
 551 #ifdef CONFIG_FEATURE_TAR_FROM
 552 static llist_t *append_file_list_to_list(llist_t *list)
 553 {
 554         FILE *src_stream;
 555         llist_t *cur = list;
 556         llist_t *tmp;
 557         char *line;
 558         llist_t *newlist = NULL;
 559
 560         while(cur) {
 561                 src_stream = bb_xfopen(cur->data, "r");
 562                 tmp = cur;
 563                 cur = cur->link;
 564                 free(tmp);
 565         while((line = bb_get_chomped_line_from_file(src_stream)) != NULL) {
 566                         newlist = llist_add_to(newlist, line);
 567         }
 568         fclose(src_stream);
 569         }
 570         return newlist;
 571 }
 572 #endif
 573
 574 #ifdef CONFIG_FEATURE_TAR_COMPRESS
 575 static char get_header_tar_Z(archive_handle_t *archive_handle)
 576 {
 577         /* Cant lseek over pipe's */
 578         archive_handle->seek = seek_by_char;
 579
 580         /* do the decompression, and cleanup */
 581         if ((bb_xread_char(archive_handle->src_fd) != 0x1f) || (bb_xread_char(archive_handle->src_fd) != 0x9d)) {
 582                 bb_error_msg_and_die("Invalid magic");
 583         }
 584
 585         archive_handle->src_fd = open_transformer(archive_handle->src_fd, uncompress);
 586         archive_handle->offset = 0;
 587         while (get_header_tar(archive_handle) == EXIT_SUCCESS);
 588
 589         /* Can only do one file at a time */
 590         return(EXIT_FAILURE);
 591 }
 592 #endif
 593
 594 #define CTX_TEST        (1 << 0)
 595 #define CTX_EXTRACT     (1 << 1)
 596 #define TAR_OPT_BASEDIR  (1 << 2)
 597 #define TAR_OPT_TARNAME  (1 << 3)
 598 #define TAR_OPT_2STDOUT  (1 << 4)
 599 #define TAR_OPT_P        (1 << 5)
 600 #define TAR_OPT_VERBOSE  (1 << 6)
 601 #define TAR_OPT_KEEP_OLD        (1 << 7)
 602
 603 #ifdef CONFIG_FEATURE_TAR_CREATE
 604 # define CTX_CREATE     (1 << 8)
 605 # define TAR_OPT_DEREFERNCE (1 << 9)
 606 # define TAR_OPT_STR_CREATE     "ch"
 607 # define TAR_OPT_FLAG_CREATE    2
 608 #else
 609 //# define CTX_CREATE   0
 610 # define TAR_OPT_STR_CREATE     ""
 611 # define TAR_OPT_FLAG_CREATE    0
 612 #endif
 613
 614 #ifdef CONFIG_FEATURE_TAR_BZIP2
 615 # define TAR_OPT_BZIP2  (1 << (8 + TAR_OPT_FLAG_CREATE))
 616 # define TAR_OPT_STR_BZIP2      "j"
 617 # define TAR_OPT_FLAG_BZIP2     1
 618 #else
 619 # define TAR_OPT_STR_BZIP2      ""
 620 # define TAR_OPT_FLAG_BZIP2     0
 621 #endif
 622
 623 #ifdef CONFIG_FEATURE_TAR_FROM
 624 # define TAR_OPT_INCLUDE_FROM   (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2))
 625 # define TAR_OPT_EXCLUDE_FROM   (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2 + 1))
 626 # define TAR_OPT_STR_FROM       "T:X:"
 627 # define TAR_OPT_FLAG_FROM      2
 628 #else
 629 # define TAR_OPT_STR_FROM       ""
 630 # define TAR_OPT_FLAG_FROM      0
 631 #endif
 632
 633 #ifdef CONFIG_FEATURE_TAR_GZIP
 634 # define TAR_OPT_GZIP   (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2 + TAR_OPT_FLAG_FROM))
 635 # define TAR_OPT_STR_GZIP       "z"
 636 # define TAR_OPT_FLAG_GZIP      1
 637 #else
 638 # define TAR_OPT_STR_GZIP       ""
 639 # define TAR_OPT_FLAG_GZIP      0
 640 #endif
 641
 642 #ifdef CONFIG_FEATURE_TAR_COMPRESS
 643 # define TAR_OPT_UNCOMPRESS     (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2 + TAR_OPT_FLAG_FROM + TAR_OPT_FLAG_GZIP))
 644 # define TAR_OPT_STR_COMPRESS   "Z"
 645 #else
 646 # define TAR_OPT_STR_COMPRESS   ""
 647 #endif
 648
 649 static const char tar_options[]="txC:f:Opvk" \
 650         TAR_OPT_STR_CREATE \
 651         TAR_OPT_STR_BZIP2 \
 652         TAR_OPT_STR_FROM \
 653         TAR_OPT_STR_GZIP \
 654         TAR_OPT_STR_COMPRESS;
 655
 656 #ifdef CONFIG_FEATURE_TAR_LONG_OPTIONS
 657 static const struct option tar_long_options[] = {
 658         { "list",                       0,      NULL,   't' },
 659         { "extract",            0,      NULL,   'x' },
 660         { "directory",          1,      NULL,   'C' },
 661         { "file",                       1,      NULL,   'f'     },
 662         { "to-stdout",          0,      NULL,   'O' },
 663         { "same-permissions",   0,      NULL,   'p' },
 664         { "verbose",            0,      NULL,   'v' },
 665         { "keep-old",           0,      NULL,   'k' },
 666 # ifdef CONFIG_FEATURE_TAR_CREATE
 667         { "create",                     0,      NULL,   'c' },
 668         { "dereference",        0,      NULL,   'h' },
 669 # endif
 670 # ifdef CONFIG_FEATURE_TAR_BZIP2
 671         { "bzip2",                      0,      NULL,   'j' },
 672 # endif
 673 # ifdef CONFIG_FEATURE_TAR_FROM
 674         { "files-from",         1,      NULL,   'T' },
 675         { "exclude-from",       1,      NULL,   'X' },
 676 # endif
 677 # ifdef CONFIG_FEATURE_TAR_GZIP
 678         { "gzip",                       0,      NULL,   'z' },
 679 # endif
 680 # ifdef CONFIG_FEATURE_TAR_COMPRESS
 681         { "compress",           0,      NULL,   'Z' },
 682 # endif
 683         { 0,                 0, 0, 0 }
 684 };
 685 #endif
 686
 687 int tar_main(int argc, char **argv)
 688 {
 689         char (*get_header_ptr)(archive_handle_t *) = get_header_tar;
 690         archive_handle_t *tar_handle;
 691         char *base_dir = NULL;
 692         const char *tar_filename = "-";
 693         unsigned long opt;
 694         unsigned long ctx_flag = 0;
 695
 696         if (argc < 2) {
 697                 bb_show_usage();
 698         }
 699
 700         /* Prepend '-' to the first argument if required */
 701         if (argv[1][0] != '-') {
 702                 char *tmp;
 703
 704                 bb_xasprintf(&tmp, "-%s", argv[1]);
 705                 argv[1] = tmp;
 706         }
 707
 708         /* Initialise default values */
 709         tar_handle = init_handle();
 710         tar_handle->flags = ARCHIVE_CREATE_LEADING_DIRS | ARCHIVE_PRESERVE_DATE | ARCHIVE_EXTRACT_UNCONDITIONAL;
 711
 712         bb_opt_complementaly = "c~tx:t~cx:x~ct:X*:T*";
 713 #ifdef CONFIG_FEATURE_TAR_LONG_OPTIONS
 714         bb_applet_long_options = tar_long_options;
 715 #endif
 716
 717         opt = bb_getopt_ulflags(argc, argv, tar_options,
 718                                 &base_dir,      /* Change to dir <optarg> */
 719                                 &tar_filename /* archive filename */
 720 #ifdef CONFIG_FEATURE_TAR_FROM
 721                                 , &(tar_handle->accept),
 722                                 &(tar_handle->reject)
 723 #endif
 724                                 );
 725
 726         /* Check one and only one context option was given */
 727         if(opt & BB_GETOPT_ERROR) {
 728                 bb_show_usage();
 729         }
 730 #ifdef CONFIG_FEATURE_TAR_CREATE
 731         ctx_flag = opt & (CTX_CREATE | CTX_TEST | CTX_EXTRACT);
 732 #else
 733         ctx_flag = opt & (CTX_TEST | CTX_EXTRACT);
 734 #endif
 735         if (ctx_flag == 0) {
 736                 bb_show_usage();
 737         }
 738         if(ctx_flag & CTX_TEST) {
 739                 if ((tar_handle->action_header == header_list) ||
 740                         (tar_handle->action_header == header_verbose_list)) {
 741                         tar_handle->action_header = header_verbose_list;
 742                 } else {
 743                         tar_handle->action_header = header_list;
 744                 }
 745         }
 746         if(ctx_flag & CTX_EXTRACT) {
 747                 if (tar_handle->action_data != data_extract_to_stdout)
 748                         tar_handle->action_data = data_extract_all;
 749                 }
 750         if(opt & TAR_OPT_2STDOUT) {
 751                 /* To stdout */
 752                 tar_handle->action_data = data_extract_to_stdout;
 753         }
 754         if(opt & TAR_OPT_VERBOSE) {
 755                 if ((tar_handle->action_header == header_list) ||
 756                         (tar_handle->action_header == header_verbose_list))
 757                 {
 758                 tar_handle->action_header = header_verbose_list;
 759                 } else {
 760                         tar_handle->action_header = header_list;
 761                 }
 762         }
 763         if (opt & TAR_OPT_KEEP_OLD) {
 764                 tar_handle->flags &= ~ARCHIVE_EXTRACT_UNCONDITIONAL;
 765         }
 766
 767 #ifdef CONFIG_FEATURE_TAR_GZIP
 768         if(opt & TAR_OPT_GZIP) {
 769                 get_header_ptr = get_header_tar_gz;
 770         }
 771 #endif
 772 #ifdef CONFIG_FEATURE_TAR_BZIP2
 773         if(opt & TAR_OPT_BZIP2) {
 774                 get_header_ptr = get_header_tar_bz2;
 775         }
 776 #endif
 777 #ifdef CONFIG_FEATURE_TAR_COMPRESS
 778         if(opt & TAR_OPT_UNCOMPRESS) {
 779                 get_header_ptr = get_header_tar_Z;
 780         }
 781 #endif
 782 #ifdef CONFIG_FEATURE_TAR_FROM
 783         if(opt & TAR_OPT_EXCLUDE_FROM) {
 784                 tar_handle->reject = append_file_list_to_list(tar_handle->reject);
 785         }
 786         if(opt & TAR_OPT_INCLUDE_FROM) {
 787                 tar_handle->accept = append_file_list_to_list(tar_handle->accept);
 788         }
 789 #endif
 790
 791         /* Check if we are reading from stdin */
 792         if ((argv[optind]) && (*argv[optind] == '-')) {
 793                 /* Default is to read from stdin, so just skip to next arg */
 794                 optind++;
 795         }
 796
 797         /* Setup an array of filenames to work with */
 798         /* TODO: This is the same as in ar, separate function ? */
 799         while (optind < argc) {
 800                 char *filename_ptr = last_char_is(argv[optind], '/');
 801                 if (filename_ptr) {
 802                         *filename_ptr = '\0';
 803                 }
 804                 tar_handle->accept = llist_add_to(tar_handle->accept, argv[optind]);
 805                 optind++;
 806         }
 807
 808         if ((tar_handle->accept) || (tar_handle->reject)) {
 809                 tar_handle->filter = filter_accept_reject_list;
 810         }
 811
 812         /* Open the tar file */
 813         {
 814                 FILE *tar_stream;
 815                 int flags;
 816
 817 #ifdef CONFIG_FEATURE_TAR_CREATE
 818                 if (opt & CTX_CREATE) {
 819                         /* Make sure there is at least one file to tar up.  */
 820                         if (tar_handle->accept == NULL) {
 821                                 bb_error_msg_and_die("Cowardly refusing to create an empty archive");
 822                         }
 823                         tar_stream = stdout;
 824                         flags = O_WRONLY | O_CREAT | O_EXCL;
 825                         unlink(tar_filename);
 826                 } else
 827 #endif
 828                 {
 829                         tar_stream = stdin;
 830                         flags = O_RDONLY;
 831                 }
 832
 833                 if ((tar_filename[0] == '-') && (tar_filename[1] == '\0')) {
 834                         tar_handle->src_fd = fileno(tar_stream);
 835                         tar_handle->seek = seek_by_char;
 836                 } else {
 837                         tar_handle->src_fd = bb_xopen(tar_filename, flags);
 838                 }
 839         }
 840
 841         if ((base_dir) && (chdir(base_dir))) {
 842                 bb_perror_msg_and_die("Couldnt chdir to %s", base_dir);
 843         }
 844
 845 #ifdef CONFIG_FEATURE_TAR_CREATE
 846         /* create an archive */
 847         if (opt & CTX_CREATE) {
 848                 int verboseFlag = FALSE;
 849                 int gzipFlag = FALSE;
 850
 851 # ifdef CONFIG_FEATURE_TAR_GZIP
 852                 if (get_header_ptr == get_header_tar_gz) {
 853                         gzipFlag = TRUE;
 854                 }
 855 # endif /* CONFIG_FEATURE_TAR_GZIP */
 856 # ifdef CONFIG_FEATURE_TAR_BZIP2
 857                 if (get_header_ptr == get_header_tar_bz2) {
 858                         bb_error_msg_and_die("Creating bzip2 compressed archives is not currently supported.");
 859                 }
 860 # endif /* CONFIG_FEATURE_TAR_BZIP2 */
 861
 862                 if ((tar_handle->action_header == header_list) ||
 863                                 (tar_handle->action_header == header_verbose_list)) {
 864                         verboseFlag = TRUE;
 865                 }
 866                 writeTarFile(tar_handle->src_fd, verboseFlag, opt & TAR_OPT_DEREFERNCE, tar_handle->accept,
 867                         tar_handle->reject, gzipFlag);
 868         } else
 869 #endif /* CONFIG_FEATURE_TAR_CREATE */
 870         {
 871                 while (get_header_ptr(tar_handle) == EXIT_SUCCESS);
 872
 873                 /* Ckeck that every file that should have been extracted was */
 874                 while (tar_handle->accept) {
 875                         if (find_list_entry(tar_handle->reject, tar_handle->accept->data) == NULL) {
 876                                 if (find_list_entry(tar_handle->passed, tar_handle->accept->data) == NULL) {
 877                                         bb_error_msg_and_die("%s: Not found in archive\n", tar_handle->accept->data);
 878                                 }
 879                         }
 880                         tar_handle->accept = tar_handle->accept->link;
 881                 }
 882         }
 883
 884 #ifdef CONFIG_FEATURE_CLEAN_UP
 885         if (tar_handle->src_fd != STDIN_FILENO) {
 886                 close(tar_handle->src_fd);
 887         }
 888 #endif /* CONFIG_FEATURE_CLEAN_UP */
 889
 890         return(EXIT_SUCCESS);
 891 }