archival/tar.c

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