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