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,2000 by Lineo, inc. and Erik Andersen
  13  * Copyright (C) 1999-2002 by Erik Andersen <andersee@debian.org>
  14  *
  15  * Based in part in the tar implementation in sash
  16  *  Copyright (c) 1999 by David I. Bell
  17  *  Permission is granted to use, distribute, or modify this source,
  18  *  provided that this copyright notice remains intact.
  19  *  Permission to distribute sash derived code under the GPL has been granted.
  20  *
  21  * Based in part on the tar implementation from busybox-0.28
  22  *  Copyright (C) 1995 Bruce Perens
  23  *  This is free software under the GNU General Public License.
  24  *
  25  * This program is free software; you can redistribute it and/or modify
  26  * it under the terms of the GNU General Public License as published by
  27  * the Free Software Foundation; either version 2 of the License, or
  28  * (at your option) any later version.
  29  *
  30  * This program is distributed in the hope that it will be useful,
  31  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  32  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  33  * General Public License for more details.
  34  *
  35  * You should have received a copy of the GNU General Public License
  36  * along with this program; if not, write to the Free Software
  37  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  38  *
  39  */
  40
  41 #include <fcntl.h>
  42 #include <getopt.h>
  43 #include <search.h>
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <unistd.h>
  47 #include <fnmatch.h>
  48 #include <string.h>
  49 #include <errno.h>
  50 #include <signal.h>
  51 #include <sys/wait.h>
  52 #include <sys/socket.h>
  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 # ifndef MAJOR
  63 #  define MAJOR(dev) (((dev)>>8)&0xff)
  64 #  define MINOR(dev) ((dev)&0xff)
  65 # endif
  66
  67 static const int TAR_BLOCK_SIZE = 512;
  68 static const int TAR_MAGIC_LEN = 6;
  69 static const int TAR_VERSION_LEN = 2;
  70
  71 /* POSIX tar Header Block, from POSIX 1003.1-1990  */
  72 enum { NAME_SIZE = 100 };       /* because gcc won't let me use 'static const int' */
  73 struct TarHeader {              /* byte offset */
  74         char name[NAME_SIZE];   /*   0-99 */
  75         char mode[8];           /* 100-107 */
  76         char uid[8];            /* 108-115 */
  77         char gid[8];            /* 116-123 */
  78         char size[12];          /* 124-135 */
  79         char mtime[12];         /* 136-147 */
  80         char chksum[8];         /* 148-155 */
  81         char typeflag;          /* 156-156 */
  82         char linkname[NAME_SIZE];       /* 157-256 */
  83         char magic[6];          /* 257-262 */
  84         char version[2];        /* 263-264 */
  85         char uname[32];         /* 265-296 */
  86         char gname[32];         /* 297-328 */
  87         char devmajor[8];       /* 329-336 */
  88         char devminor[8];       /* 337-344 */
  89         char prefix[155];       /* 345-499 */
  90         char padding[12];       /* 500-512 (pad to exactly the TAR_BLOCK_SIZE) */
  91 };
  92 typedef struct TarHeader TarHeader;
  93
  94 /*
  95 ** writeTarFile(),  writeFileToTarball(), and writeTarHeader() are
  96 ** the only functions that deal with the HardLinkInfo structure.
  97 ** Even these functions use the xxxHardLinkInfo() functions.
  98 */
  99 typedef struct HardLinkInfo HardLinkInfo;
 100 struct HardLinkInfo {
 101         HardLinkInfo *next;     /* Next entry in list */
 102         dev_t dev;                      /* Device number */
 103         ino_t ino;                      /* Inode number */
 104         short linkCount;        /* (Hard) Link Count */
 105         char name[1];           /* Start of filename (must be last) */
 106 };
 107
 108 /* Some info to be carried along when creating a new tarball */
 109 struct TarBallInfo {
 110         char *fileName;         /* File name of the tarball */
 111         int tarFd;                      /* Open-for-write file descriptor
 112                                                    for the tarball */
 113         struct stat statBuf;    /* Stat info for the tarball, letting
 114                                                            us know the inode and device that the
 115                                                            tarball lives, so we can avoid trying
 116                                                            to include the tarball into itself */
 117         int verboseFlag;        /* Whether to print extra stuff or not */
 118         char **excludeList;     /* List of files to not include */
 119         HardLinkInfo *hlInfoHead;       /* Hard Link Tracking Information */
 120         HardLinkInfo *hlInfo;   /* Hard Link Info for the current file */
 121 };
 122 typedef struct TarBallInfo TarBallInfo;
 123
 124 /* A nice enum with all the possible tar file content types */
 125 enum TarFileType {
 126         REGTYPE = '0',          /* regular file */
 127         REGTYPE0 = '\0',        /* regular file (ancient bug compat) */
 128         LNKTYPE = '1',          /* hard link */
 129         SYMTYPE = '2',          /* symbolic link */
 130         CHRTYPE = '3',          /* character special */
 131         BLKTYPE = '4',          /* block special */
 132         DIRTYPE = '5',          /* directory */
 133         FIFOTYPE = '6',         /* FIFO special */
 134         CONTTYPE = '7',         /* reserved */
 135         GNULONGLINK = 'K',      /* GNU long (>100 chars) link name */
 136         GNULONGNAME = 'L',      /* GNU long (>100 chars) file name */
 137 };
 138 typedef enum TarFileType TarFileType;
 139
 140 /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */
 141 static inline void addHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr, dev_t dev,
 142                                                                    ino_t ino, short linkCount,
 143                                                                    const char *name)
 144 {
 145         /* Note: hlInfoHeadPtr can never be NULL! */
 146         HardLinkInfo *hlInfo;
 147
 148         hlInfo =
 149                 (HardLinkInfo *) xmalloc(sizeof(HardLinkInfo) + strlen(name) + 1);
 150         if (hlInfo) {
 151                 hlInfo->next = *hlInfoHeadPtr;
 152                 *hlInfoHeadPtr = hlInfo;
 153                 hlInfo->dev = dev;
 154                 hlInfo->ino = ino;
 155                 hlInfo->linkCount = linkCount;
 156                 strcpy(hlInfo->name, name);
 157         }
 158         return;
 159 }
 160
 161 static void freeHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr)
 162 {
 163         HardLinkInfo *hlInfo = NULL;
 164         HardLinkInfo *hlInfoNext = NULL;
 165
 166         if (hlInfoHeadPtr) {
 167                 hlInfo = *hlInfoHeadPtr;
 168                 while (hlInfo) {
 169                         hlInfoNext = hlInfo->next;
 170                         free(hlInfo);
 171                         hlInfo = hlInfoNext;
 172                 }
 173                 *hlInfoHeadPtr = NULL;
 174         }
 175         return;
 176 }
 177
 178 /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */
 179 static inline HardLinkInfo *findHardLinkInfo(HardLinkInfo * hlInfo, dev_t dev,
 180                                                                                          ino_t ino)
 181 {
 182         while (hlInfo) {
 183                 if ((ino == hlInfo->ino) && (dev == hlInfo->dev))
 184                         break;
 185                 hlInfo = hlInfo->next;
 186         }
 187         return (hlInfo);
 188 }
 189
 190 /* Put an octal string into the specified buffer.
 191  * The number is zero and space padded and possibly null padded.
 192  * Returns TRUE if successful.  */
 193 static int putOctal(char *cp, int len, long value)
 194 {
 195         int tempLength;
 196         char tempBuffer[32];
 197         char *tempString = tempBuffer;
 198
 199         /* Create a string of the specified length with an initial space,
 200          * leading zeroes and the octal number, and a trailing null.  */
 201         sprintf(tempString, "%0*lo", len - 1, value);
 202
 203         /* If the string is too large, suppress the leading space.  */
 204         tempLength = strlen(tempString) + 1;
 205         if (tempLength > len) {
 206                 tempLength--;
 207                 tempString++;
 208         }
 209
 210         /* If the string is still too large, suppress the trailing null.  */
 211         if (tempLength > len)
 212                 tempLength--;
 213
 214         /* If the string is still too large, fail.  */
 215         if (tempLength > len)
 216                 return FALSE;
 217
 218         /* Copy the string to the field.  */
 219         memcpy(cp, tempString, len);
 220
 221         return TRUE;
 222 }
 223
 224 /* Write out a tar header for the specified file/directory/whatever */
 225 static inline int writeTarHeader(struct TarBallInfo *tbInfo,
 226                                                                  const char *header_name,
 227                                                                  const char *real_name, struct stat *statbuf)
 228 {
 229         long chksum = 0;
 230         struct TarHeader header;
 231         const unsigned char *cp = (const unsigned char *) &header;
 232         ssize_t size = sizeof(struct TarHeader);
 233
 234         memset(&header, 0, size);
 235
 236         strncpy(header.name, header_name, sizeof(header.name));
 237
 238         putOctal(header.mode, sizeof(header.mode), statbuf->st_mode);
 239         putOctal(header.uid, sizeof(header.uid), statbuf->st_uid);
 240         putOctal(header.gid, sizeof(header.gid), statbuf->st_gid);
 241         putOctal(header.size, sizeof(header.size), 0);  /* Regular file size is handled later */
 242         putOctal(header.mtime, sizeof(header.mtime), statbuf->st_mtime);
 243         strncpy(header.magic, TAR_MAGIC TAR_VERSION,
 244                         TAR_MAGIC_LEN + TAR_VERSION_LEN);
 245
 246         /* Enter the user and group names (default to root if it fails) */
 247         my_getpwuid(header.uname, statbuf->st_uid);
 248         if (!*header.uname)
 249                 strcpy(header.uname, "root");
 250         my_getgrgid(header.gname, statbuf->st_gid);
 251         if (!*header.uname)
 252                 strcpy(header.uname, "root");
 253
 254         if (tbInfo->hlInfo) {
 255                 /* This is a hard link */
 256                 header.typeflag = LNKTYPE;
 257                 strncpy(header.linkname, tbInfo->hlInfo->name,
 258                                 sizeof(header.linkname));
 259         } else if (S_ISLNK(statbuf->st_mode)) {
 260                 char *lpath = xreadlink(real_name);
 261
 262                 if (!lpath)             /* Already printed err msg inside xreadlink() */
 263                         return (FALSE);
 264                 header.typeflag = SYMTYPE;
 265                 strncpy(header.linkname, lpath, sizeof(header.linkname));
 266                 free(lpath);
 267         } else if (S_ISDIR(statbuf->st_mode)) {
 268                 header.typeflag = DIRTYPE;
 269                 strncat(header.name, "/", sizeof(header.name));
 270         } else if (S_ISCHR(statbuf->st_mode)) {
 271                 header.typeflag = CHRTYPE;
 272                 putOctal(header.devmajor, sizeof(header.devmajor),
 273                                  MAJOR(statbuf->st_rdev));
 274                 putOctal(header.devminor, sizeof(header.devminor),
 275                                  MINOR(statbuf->st_rdev));
 276         } else if (S_ISBLK(statbuf->st_mode)) {
 277                 header.typeflag = BLKTYPE;
 278                 putOctal(header.devmajor, sizeof(header.devmajor),
 279                                  MAJOR(statbuf->st_rdev));
 280                 putOctal(header.devminor, sizeof(header.devminor),
 281                                  MINOR(statbuf->st_rdev));
 282         } else if (S_ISFIFO(statbuf->st_mode)) {
 283                 header.typeflag = FIFOTYPE;
 284         } else if (S_ISREG(statbuf->st_mode)) {
 285                 header.typeflag = REGTYPE;
 286                 putOctal(header.size, sizeof(header.size), statbuf->st_size);
 287         } else {
 288                 error_msg("%s: Unknown file type", real_name);
 289                 return (FALSE);
 290         }
 291
 292         /* Calculate and store the checksum (i.e., the sum of all of the bytes of
 293          * the header).  The checksum field must be filled with blanks for the
 294          * calculation.  The checksum field is formatted differently from the
 295          * other fields: it has [6] digits, a null, then a space -- rather than
 296          * digits, followed by a null like the other fields... */
 297         memset(header.chksum, ' ', sizeof(header.chksum));
 298         cp = (const unsigned char *) &header;
 299         while (size-- > 0)
 300                 chksum += *cp++;
 301         putOctal(header.chksum, 7, chksum);
 302
 303         /* Now write the header out to disk */
 304         if ((size =
 305                  full_write(tbInfo->tarFd, (char *) &header,
 306                                         sizeof(struct TarHeader))) < 0) {
 307                 error_msg(io_error, real_name, strerror(errno));
 308                 return (FALSE);
 309         }
 310         /* Pad the header up to the tar block size */
 311         for (; size < TAR_BLOCK_SIZE; size++) {
 312                 write(tbInfo->tarFd, "\0", 1);
 313         }
 314         /* Now do the verbose thing (or not) */
 315
 316         if (tbInfo->verboseFlag) {
 317                 FILE *vbFd = stdout;
 318
 319                 if (tbInfo->verboseFlag == 2)   /* If the archive goes to stdout, verbose to stderr */
 320                         vbFd = stderr;
 321                 fprintf(vbFd, "%s\n", header.name);
 322         }
 323
 324         return (TRUE);
 325 }
 326
 327 # if defined CONFIG_FEATURE_TAR_EXCLUDE
 328 static inline int exclude_file(char **excluded_files, const char *file)
 329 {
 330         int i;
 331
 332         if (excluded_files == NULL)
 333                 return 0;
 334
 335         for (i = 0; excluded_files[i] != NULL; i++) {
 336                 if (excluded_files[i][0] == '/') {
 337                         if (fnmatch(excluded_files[i], file,
 338                                                 FNM_PATHNAME | FNM_LEADING_DIR) == 0)
 339                                 return 1;
 340                 } else {
 341                         const char *p;
 342
 343                         for (p = file; p[0] != '\0'; p++) {
 344                                 if ((p == file || p[-1] == '/') && p[0] != '/' &&
 345                                         fnmatch(excluded_files[i], p,
 346                                                         FNM_PATHNAME | FNM_LEADING_DIR) == 0)
 347                                         return 1;
 348                         }
 349                 }
 350         }
 351
 352         return 0;
 353 }
 354 #endif
 355
 356 static int writeFileToTarball(const char *fileName, struct stat *statbuf,
 357                                                           void *userData)
 358 {
 359         struct TarBallInfo *tbInfo = (struct TarBallInfo *) userData;
 360         const char *header_name;
 361
 362         /*
 363            ** Check to see if we are dealing with a hard link.
 364            ** If so -
 365            ** Treat the first occurance of a given dev/inode as a file while
 366            ** treating any additional occurances as hard links.  This is done
 367            ** by adding the file information to the HardLinkInfo linked list.
 368          */
 369         tbInfo->hlInfo = NULL;
 370         if (statbuf->st_nlink > 1) {
 371                 tbInfo->hlInfo = findHardLinkInfo(tbInfo->hlInfoHead, statbuf->st_dev,
 372                                                                                   statbuf->st_ino);
 373                 if (tbInfo->hlInfo == NULL)
 374                         addHardLinkInfo(&tbInfo->hlInfoHead, statbuf->st_dev,
 375                                                         statbuf->st_ino, statbuf->st_nlink, fileName);
 376         }
 377
 378         /* It is against the rules to archive a socket */
 379         if (S_ISSOCK(statbuf->st_mode)) {
 380                 error_msg("%s: socket ignored", fileName);
 381                 return (TRUE);
 382         }
 383
 384         /* It is a bad idea to store the archive we are in the process of creating,
 385          * so check the device and inode to be sure that this particular file isn't
 386          * the new tarball */
 387         if (tbInfo->statBuf.st_dev == statbuf->st_dev &&
 388                 tbInfo->statBuf.st_ino == statbuf->st_ino) {
 389                 error_msg("%s: file is the archive; skipping", fileName);
 390                 return (TRUE);
 391         }
 392
 393         header_name = fileName;
 394         while (header_name[0] == '/') {
 395                 static int alreadyWarned = FALSE;
 396
 397                 if (alreadyWarned == FALSE) {
 398                         error_msg("Removing leading '/' from member names");
 399                         alreadyWarned = TRUE;
 400                 }
 401                 header_name++;
 402         }
 403
 404         if (strlen(fileName) >= NAME_SIZE) {
 405                 error_msg(name_longer_than_foo, NAME_SIZE);
 406                 return (TRUE);
 407         }
 408
 409         if (header_name[0] == '\0')
 410                 return TRUE;
 411
 412 # if defined CONFIG_FEATURE_TAR_EXCLUDE
 413         if (exclude_file(tbInfo->excludeList, header_name)) {
 414                 return SKIP;
 415         }
 416 # endif                                                 /* CONFIG_FEATURE_TAR_EXCLUDE */
 417
 418         if (writeTarHeader(tbInfo, header_name, fileName, statbuf) == FALSE) {
 419                 return (FALSE);
 420         }
 421
 422         /* Now, if the file is a regular file, copy it out to the tarball */
 423         if ((tbInfo->hlInfo == NULL)
 424                 && (S_ISREG(statbuf->st_mode))) {
 425                 int inputFileFd;
 426                 char buffer[BUFSIZ];
 427                 ssize_t size = 0, readSize = 0;
 428
 429                 /* open the file we want to archive, and make sure all is well */
 430                 if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
 431                         error_msg("%s: Cannot open: %s", fileName, strerror(errno));
 432                         return (FALSE);
 433                 }
 434
 435                 /* write the file to the archive */
 436                 while ((size = full_read(inputFileFd, buffer, sizeof(buffer))) > 0) {
 437                         if (full_write(tbInfo->tarFd, buffer, size) != size) {
 438                                 /* Output file seems to have a problem */
 439                                 error_msg(io_error, fileName, strerror(errno));
 440                                 return (FALSE);
 441                         }
 442                         readSize += size;
 443                 }
 444                 if (size == -1) {
 445                         error_msg(io_error, fileName, strerror(errno));
 446                         return (FALSE);
 447                 }
 448                 /* Pad the file up to the tar block size */
 449                 for (; (readSize % TAR_BLOCK_SIZE) != 0; readSize++) {
 450                         write(tbInfo->tarFd, "\0", 1);
 451                 }
 452                 close(inputFileFd);
 453         }
 454
 455         return (TRUE);
 456 }
 457
 458 static inline int writeTarFile(const char *tarName, int verboseFlag,
 459                                                            char **argv, char **excludeList, int gzip)
 460 {
 461 #ifdef CONFIG_FEATURE_TAR_GZIP
 462         int gzipDataPipe[2] = { -1, -1 };
 463         int gzipStatusPipe[2] = { -1, -1 };
 464         pid_t gzipPid = 0;
 465 #endif
 466
 467         int errorFlag = FALSE;
 468         ssize_t size;
 469         struct TarBallInfo tbInfo;
 470
 471         tbInfo.hlInfoHead = NULL;
 472
 473         /* Make sure there is at least one file to tar up.  */
 474         if (*argv == NULL)
 475                 error_msg_and_die("Cowardly refusing to create an empty archive");
 476
 477         /* Open the tar file for writing.  */
 478         if (tarName == NULL) {
 479                 tbInfo.tarFd = fileno(stdout);
 480                 tbInfo.verboseFlag = verboseFlag ? 2 : 0;
 481         } else {
 482                 tbInfo.tarFd = open(tarName, O_WRONLY | O_CREAT | O_TRUNC, 0644);
 483                 tbInfo.verboseFlag = verboseFlag ? 1 : 0;
 484         }
 485
 486         if (tbInfo.tarFd < 0) {
 487                 perror_msg("Error opening '%s'", tarName);
 488                 freeHardLinkInfo(&tbInfo.hlInfoHead);
 489                 return (FALSE);
 490         }
 491
 492         /* Store the stat info for the tarball's file, so
 493          * can avoid including the tarball into itself....  */
 494         if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0)
 495                 error_msg_and_die(io_error, tarName, strerror(errno));
 496
 497 #ifdef CONFIG_FEATURE_TAR_GZIP
 498         if (gzip) {
 499                 if (socketpair(AF_UNIX, SOCK_STREAM, 0, gzipDataPipe) < 0
 500                         || pipe(gzipStatusPipe) < 0)
 501                         perror_msg_and_die("Failed to create gzip pipe");
 502
 503                 signal(SIGPIPE, SIG_IGN);       /* we only want EPIPE on errors */
 504
 505                 gzipPid = fork();
 506
 507                 if (gzipPid == 0) {
 508                         dup2(gzipDataPipe[0], 0);
 509                         close(gzipDataPipe[1]);
 510
 511                         if (tbInfo.tarFd != 1);
 512                         dup2(tbInfo.tarFd, 1);
 513
 514                         close(gzipStatusPipe[0]);
 515                         fcntl(gzipStatusPipe[1], F_SETFD, FD_CLOEXEC);  /* close on exec shows sucess */
 516
 517                         execl("/bin/gzip", "gzip", "-f", 0);
 518
 519                         write(gzipStatusPipe[1], "", 1);
 520                         close(gzipStatusPipe[1]);
 521
 522                         exit(-1);
 523                 } else if (gzipPid > 0) {
 524                         close(gzipDataPipe[0]);
 525                         close(gzipStatusPipe[1]);
 526
 527                         while (1) {
 528                                 char buf;
 529
 530                                 int n = read(gzipStatusPipe[0], &buf, 1);
 531
 532                                 if (n == 1)
 533                                         error_msg_and_die("Could not exec gzip process");       /* socket was not closed => error */
 534                                 else if ((n < 0) && (errno == EAGAIN || errno == EINTR))
 535                                         continue;       /* try it again */
 536                                 break;
 537                         }
 538                         close(gzipStatusPipe[0]);
 539
 540                         tbInfo.tarFd = gzipDataPipe[1];
 541                 } else {
 542                         perror_msg_and_die("Failed to fork gzip process");
 543                 }
 544         }
 545 #endif
 546
 547         tbInfo.excludeList = excludeList;
 548
 549         /* Read the directory/files and iterate over them one at a time */
 550         while (*argv != NULL) {
 551                 if (!recursive_action(*argv++, TRUE, FALSE, FALSE,
 552                                                           writeFileToTarball, writeFileToTarball,
 553                                                           (void *) &tbInfo)) {
 554                         errorFlag = TRUE;
 555                 }
 556         }
 557         /* Write two empty blocks to the end of the archive */
 558         for (size = 0; size < (2 * TAR_BLOCK_SIZE); size++) {
 559                 write(tbInfo.tarFd, "\0", 1);
 560         }
 561
 562         /* To be pedantically correct, we would check if the tarball
 563          * is smaller than 20 tar blocks, and pad it if it was smaller,
 564          * but that isn't necessary for GNU tar interoperability, and
 565          * so is considered a waste of space */
 566
 567         /* Hang up the tools, close up shop, head home */
 568         close(tbInfo.tarFd);
 569         if (errorFlag)
 570                 error_msg("Error exit delayed from previous errors");
 571
 572         freeHardLinkInfo(&tbInfo.hlInfoHead);
 573
 574 #ifdef CONFIG_FEATURE_TAR_GZIP
 575         if (gzip && gzipPid) {
 576                 if (waitpid(gzipPid, NULL, 0) == -1)
 577                         printf("Couldnt wait ?");
 578         }
 579 #endif
 580
 581         return !errorFlag;
 582 }
 583 #endif                                                  /* tar_create */
 584
 585 void append_file_to_list(const char *new_name, char ***list, int *list_count)
 586 {
 587         *list = realloc(*list, sizeof(char *) * (*list_count + 2));
 588         (*list)[*list_count] = xstrdup(new_name);
 589         (*list_count)++;
 590         (*list)[*list_count] = NULL;
 591 }
 592
 593 void append_file_list_to_list(char *filename, char ***name_list,
 594                                                           int *num_of_entries)
 595 {
 596         FILE *src_stream;
 597         char *line;
 598
 599         src_stream = xfopen(filename, "r");
 600         while ((line = get_line_from_file(src_stream)) != NULL) {
 601                 chomp(line);
 602                 append_file_to_list(line, name_list, num_of_entries);
 603                 free(line);
 604         }
 605         fclose(src_stream);
 606 }
 607
 608 int tar_main(int argc, char **argv)
 609 {
 610         enum untar_funct_e {
 611                 /* This is optional */
 612                 untar_unzip = 1,
 613                 /* Require one and only one of these */
 614                 untar_list = 2,
 615                 untar_create = 4,
 616                 untar_extract = 8
 617         };
 618
 619         FILE *src_stream = NULL;
 620         FILE *uncompressed_stream = NULL;
 621         char **include_list = NULL;
 622         char **exclude_list = NULL;
 623         char *src_filename = NULL;
 624         char *dst_prefix = NULL;
 625         int opt;
 626         unsigned short untar_funct = 0;
 627         unsigned short untar_funct_required = 0;
 628         unsigned short extract_function = 0;
 629         int include_list_count = 0;
 630
 631 #ifdef CONFIG_FEATURE_TAR_EXCLUDE
 632         int exclude_list_count = 0;
 633 #endif
 634 #ifdef CONFIG_FEATURE_TAR_GZIP
 635         int gunzip_pid;
 636         int gz_fd = 0;
 637 #endif
 638
 639         if (argc < 2) {
 640                 show_usage();
 641         }
 642
 643         /* Prepend '-' to the first argument if required */
 644         if (argv[1][0] != '-') {
 645                 char *tmp = xmalloc(strlen(argv[1]) + 2);
 646
 647                 tmp[0] = '-';
 648                 strcpy(tmp + 1, argv[1]);
 649                 argv[1] = tmp;
 650         }
 651
 652         while ((opt = getopt(argc, argv, "ctxT:X:C:f:Opvz")) != -1) {
 653                 switch (opt) {
 654
 655                         /* One and only one of these is required */
 656                 case 'c':
 657                         untar_funct_required |= untar_create;
 658                         break;
 659                 case 't':
 660                         untar_funct_required |= untar_list;
 661                         extract_function |= extract_list | extract_unconditional;
 662                         break;
 663                 case 'x':
 664                         untar_funct_required |= untar_extract;
 665                         extract_function |=
 666                                 (extract_all_to_fs | extract_unconditional |
 667                                  extract_create_leading_dirs);
 668                         break;
 669
 670                         /* These are optional */
 671                         /* Exclude or Include files listed in <filename> */
 672 #ifdef CONFIG_FEATURE_TAR_EXCLUDE
 673                 case 'X':
 674                         append_file_list_to_list(optarg, &exclude_list,
 675                                                                          &exclude_list_count);
 676                         break;
 677 #endif
 678                 case 'T':
 679                         /* by default a list is an include list */
 680                         append_file_list_to_list(optarg, &include_list,
 681                                                                          &include_list_count);
 682                         break;
 683
 684                 case 'C':               /* Change to dir <optarg> */
 685                         /* Make sure dst_prefix ends in a '/' */
 686                         dst_prefix = concat_path_file(optarg, "/");
 687                         break;
 688                 case 'f':               /* archive filename */
 689                         if (strcmp(optarg, "-") == 0) {
 690                                 src_filename = NULL;
 691                         } else {
 692                                 src_filename = xstrdup(optarg);
 693                         }
 694                         break;
 695                 case 'O':
 696                         extract_function |= extract_to_stdout;
 697                         break;
 698                 case 'p':
 699                         break;
 700                 case 'v':
 701                         extract_function |= extract_verbose_list;
 702                         break;
 703 #ifdef CONFIG_FEATURE_TAR_GZIP
 704                 case 'z':
 705                         untar_funct |= untar_unzip;
 706                         break;
 707 #endif
 708                 default:
 709                         show_usage();
 710                 }
 711         }
 712
 713         /* Make sure the valid arguments were passed */
 714         if (untar_funct_required == 0) {
 715                 error_msg_and_die("You must specify one of the `-ctx' options");
 716         }
 717         if ((untar_funct_required != untar_create) &&
 718                 (untar_funct_required != untar_extract) &&
 719                 (untar_funct_required != untar_list)) {
 720                 error_msg_and_die("You may not specify more than one `ctx' option.");
 721         }
 722         untar_funct |= untar_funct_required;
 723
 724         /* Setup an array of filenames to work with */
 725         while (optind < argc) {
 726                 append_file_to_list(argv[optind], &include_list, &include_list_count);
 727                 optind++;
 728         }
 729         if (extract_function & (extract_list | extract_all_to_fs)) {
 730                 if (dst_prefix == NULL) {
 731                         dst_prefix = xstrdup("./");
 732                 }
 733
 734                 /* Setup the source of the tar data */
 735                 if (src_filename != NULL) {
 736                         src_stream = xfopen(src_filename, "r");
 737                 } else {
 738                         src_stream = stdin;
 739                 }
 740 #ifdef CONFIG_FEATURE_TAR_GZIP
 741                 /* Get a binary tree of all the tar file headers */
 742                 if (untar_funct & untar_unzip) {
 743                         uncompressed_stream = gz_open(src_stream, &gunzip_pid);
 744                 } else
 745 #endif                                                  /* CONFIG_FEATURE_TAR_GZIP */
 746                         uncompressed_stream = src_stream;
 747
 748                 /* extract or list archive */
 749                 unarchive(uncompressed_stream, stdout, &get_header_tar,
 750                                   extract_function, dst_prefix, include_list, exclude_list);
 751                 fclose(uncompressed_stream);
 752         }
 753 #ifdef CONFIG_FEATURE_TAR_CREATE
 754         /* create an archive */
 755         else if (untar_funct & untar_create) {
 756                 int verboseFlag = FALSE;
 757                 int gzipFlag = FALSE;
 758
 759 #ifdef CONFIG_FEATURE_TAR_GZIP
 760                 if (untar_funct & untar_unzip)
 761                         gzipFlag = TRUE;
 762
 763 #endif                                                  /* CONFIG_FEATURE_TAR_GZIP */
 764                 if (extract_function & extract_verbose_list)
 765                         verboseFlag = TRUE;
 766
 767                 writeTarFile(src_filename, verboseFlag, include_list, exclude_list,
 768                                          gzipFlag);
 769         }
 770 #endif                                                  /* CONFIG_FEATURE_TAR_CREATE */
 771
 772         /* Cleanups */
 773 #ifdef CONFIG_FEATURE_TAR_GZIP
 774         if (!(untar_funct & untar_create) && (untar_funct & untar_unzip)) {
 775                 fclose(src_stream);
 776                 close(gz_fd);
 777                 gz_close(gunzip_pid);
 778         }
 779 #endif                                                  /* CONFIG_FEATURE_TAR_GZIP */
 780 #ifdef CONFIG_FEATURE_CLEAN_UP
 781         if (src_filename) {
 782                 free(src_filename);
 783         }
 784 #endif
 785         return (EXIT_SUCCESS);
 786 }