Test the N command
[oweals/busybox.git] / 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->verboseFlag == 2)   /* If the archive goes to stdout, verbose to stderr */
312                         vbFd = stderr;
313                 fprintf(vbFd, "%s\n", header.name);
314         }
315
316         return (TRUE);
317 }
318
319 # if defined CONFIG_FEATURE_TAR_EXCLUDE
320 static inline int exclude_file(const llist_t *excluded_files, const char *file)
321 {
322         if (excluded_files == NULL) {
323                 return 0;
324         }
325
326         while (excluded_files) {
327                 if (excluded_files->data[0] == '/') {
328                         if (fnmatch(excluded_files->data, file,
329                                                 FNM_PATHNAME | FNM_LEADING_DIR) == 0)
330                                 return 1;
331                 } else {
332                         const char *p;
333
334                         for (p = file; p[0] != '\0'; p++) {
335                                 if ((p == file || p[-1] == '/') && p[0] != '/' &&
336                                         fnmatch(excluded_files->data, p,
337                                                         FNM_PATHNAME | FNM_LEADING_DIR) == 0)
338                                         return 1;
339                         }
340                 }
341                 excluded_files = excluded_files->link;
342         }
343
344         return 0;
345 }
346 #endif
347
348 static int writeFileToTarball(const char *fileName, struct stat *statbuf,
349                                                           void *userData)
350 {
351         struct TarBallInfo *tbInfo = (struct TarBallInfo *) userData;
352         const char *header_name;
353
354         /*
355            ** Check to see if we are dealing with a hard link.
356            ** If so -
357            ** Treat the first occurance of a given dev/inode as a file while
358            ** treating any additional occurances as hard links.  This is done
359            ** by adding the file information to the HardLinkInfo linked list.
360          */
361         tbInfo->hlInfo = NULL;
362         if (statbuf->st_nlink > 1) {
363                 tbInfo->hlInfo = findHardLinkInfo(tbInfo->hlInfoHead, statbuf);
364                 if (tbInfo->hlInfo == NULL)
365                         addHardLinkInfo(&tbInfo->hlInfoHead, statbuf, fileName);
366         }
367
368         /* It is against the rules to archive a socket */
369         if (S_ISSOCK(statbuf->st_mode)) {
370                 bb_error_msg("%s: socket ignored", fileName);
371                 return (TRUE);
372         }
373
374         /* It is a bad idea to store the archive we are in the process of creating,
375          * so check the device and inode to be sure that this particular file isn't
376          * the new tarball */
377         if (tbInfo->statBuf.st_dev == statbuf->st_dev &&
378                 tbInfo->statBuf.st_ino == statbuf->st_ino) {
379                 bb_error_msg("%s: file is the archive; skipping", fileName);
380                 return (TRUE);
381         }
382
383         header_name = fileName;
384         while (header_name[0] == '/') {
385                 static int alreadyWarned = FALSE;
386
387                 if (alreadyWarned == FALSE) {
388                         bb_error_msg("Removing leading '/' from member names");
389                         alreadyWarned = TRUE;
390                 }
391                 header_name++;
392         }
393
394         if (strlen(fileName) >= NAME_SIZE) {
395                 bb_error_msg(bb_msg_name_longer_than_foo, NAME_SIZE);
396                 return (TRUE);
397         }
398
399         if (header_name[0] == '\0')
400                 return TRUE;
401
402 # if defined CONFIG_FEATURE_TAR_EXCLUDE
403         if (exclude_file(tbInfo->excludeList, header_name)) {
404                 return SKIP;
405         }
406 # endif                                                 /* CONFIG_FEATURE_TAR_EXCLUDE */
407
408         if (writeTarHeader(tbInfo, header_name, fileName, statbuf) == FALSE) {
409                 return (FALSE);
410         }
411
412         /* Now, if the file is a regular file, copy it out to the tarball */
413         if ((tbInfo->hlInfo == NULL)
414                 && (S_ISREG(statbuf->st_mode))) {
415                 int inputFileFd;
416                 char buffer[BUFSIZ];
417                 ssize_t size = 0, readSize = 0;
418
419                 /* open the file we want to archive, and make sure all is well */
420                 if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
421                         bb_perror_msg("%s: Cannot open", fileName);
422                         return (FALSE);
423                 }
424
425                 /* write the file to the archive */
426                 while ((size = bb_full_read(inputFileFd, buffer, sizeof(buffer))) > 0) {
427                         if (bb_full_write(tbInfo->tarFd, buffer, size) != size) {
428                                 /* Output file seems to have a problem */
429                                 bb_error_msg(bb_msg_io_error, fileName);
430                                 return (FALSE);
431                         }
432                         readSize += size;
433                 }
434                 if (size == -1) {
435                         bb_error_msg(bb_msg_io_error, fileName);
436                         return (FALSE);
437                 }
438                 /* Pad the file up to the tar block size */
439                 for (; (readSize % TAR_BLOCK_SIZE) != 0; readSize++) {
440                         write(tbInfo->tarFd, "\0", 1);
441                 }
442                 close(inputFileFd);
443         }
444
445         return (TRUE);
446 }
447
448 static inline int writeTarFile(const char *tarName, const int verboseFlag,
449                                                            const llist_t *include, const llist_t *exclude, const int gzip)
450 {
451 #ifdef CONFIG_FEATURE_TAR_GZIP
452         int gzipDataPipe[2] = { -1, -1 };
453         int gzipStatusPipe[2] = { -1, -1 };
454         pid_t gzipPid = 0;
455         volatile int vfork_exec_errno = 0;
456 #endif
457
458         int errorFlag = FALSE;
459         ssize_t size;
460         struct TarBallInfo tbInfo;
461
462         tbInfo.hlInfoHead = NULL;
463
464         /* Make sure there is at least one file to tar up.  */
465         if (include == NULL) {
466                 bb_error_msg_and_die("Cowardly refusing to create an empty archive");
467         }
468
469         /* Open the tar file for writing.  */
470         if (tarName == NULL || (tarName[0] == '-' && tarName[1] == '\0')) {
471                 tbInfo.tarFd = fileno(stdout);
472                 tbInfo.verboseFlag = verboseFlag ? 2 : 0;
473         } else {
474                 unlink(tarName);
475                 tbInfo.tarFd = open(tarName, O_WRONLY | O_CREAT | O_EXCL, 0644);
476                 tbInfo.verboseFlag = verboseFlag ? 1 : 0;
477         }
478
479         if (tbInfo.tarFd < 0) {
480                 bb_perror_msg("%s: Cannot open", tarName);
481                 freeHardLinkInfo(&tbInfo.hlInfoHead);
482                 return (FALSE);
483         }
484
485         /* Store the stat info for the tarball's file, so
486          * can avoid including the tarball into itself....  */
487         if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0)
488                 bb_error_msg_and_die(bb_msg_io_error, tarName);
489
490 #ifdef CONFIG_FEATURE_TAR_GZIP
491         if (gzip) {
492                 if (pipe(gzipDataPipe) < 0 || pipe(gzipStatusPipe) < 0) {
493                         bb_perror_msg_and_die("Failed to create gzip pipe");
494                 }
495
496                 signal(SIGPIPE, SIG_IGN);       /* we only want EPIPE on errors */
497
498 # if __GNUC__
499                         /* Avoid vfork clobbering */
500                         (void) &include;
501                         (void) &errorFlag;
502 # endif
503  
504                 gzipPid = vfork();
505
506                 if (gzipPid == 0) {
507                         dup2(gzipDataPipe[0], 0);
508                         close(gzipDataPipe[1]);
509
510                         if (tbInfo.tarFd != 1);
511                         dup2(tbInfo.tarFd, 1);
512
513                         close(gzipStatusPipe[0]);
514                         fcntl(gzipStatusPipe[1], F_SETFD, FD_CLOEXEC);  /* close on exec shows sucess */
515
516                         execl("/bin/gzip", "gzip", "-f", 0);
517                         vfork_exec_errno = errno;
518
519                         close(gzipStatusPipe[1]);
520                         exit(-1);
521                 } else if (gzipPid > 0) {
522                         close(gzipDataPipe[0]);
523                         close(gzipStatusPipe[1]);
524
525                         while (1) {
526                                 char buf;
527
528                                 int n = bb_full_read(gzipStatusPipe[0], &buf, 1);
529
530                                 if (n == 0 && vfork_exec_errno != 0) {
531                                         errno = vfork_exec_errno;
532                                         bb_perror_msg_and_die("Could not exec gzip process");
533                                 } else if ((n < 0) && (errno == EAGAIN || errno == EINTR))
534                                         continue;       /* try it again */
535                                 break;
536                         }
537                         close(gzipStatusPipe[0]);
538
539                         tbInfo.tarFd = gzipDataPipe[1];
540                 } else {
541                         bb_perror_msg_and_die("Failed to vfork gzip process");
542                 }
543         }
544 #endif
545
546         tbInfo.excludeList = exclude;
547
548         /* Read the directory/files and iterate over them one at a time */
549         while (include) {
550                 if (!recursive_action(include->data, TRUE, FALSE, FALSE,
551                                                           writeFileToTarball, writeFileToTarball,
552                                                           (void *) &tbInfo)) {
553                         errorFlag = TRUE;
554                 }
555                 include = include->link;
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                 bb_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 #ifdef CONFIG_FEATURE_TAR_EXCLUDE
586 static llist_t *append_file_list_to_list(llist_t *list)
587 {
588         FILE *src_stream;
589         llist_t *cur = list;
590         llist_t *tmp;
591         char *line;
592         llist_t *newlist = NULL;
593
594         while(cur) {
595                 src_stream = bb_xfopen(cur->data, "r");
596                 tmp = cur;
597                 cur = cur->link;
598                 free(tmp);
599         while((line = bb_get_chomped_line_from_file(src_stream)) != NULL) {
600                         newlist = llist_add_to(newlist, line);
601         }
602         fclose(src_stream);
603         }
604         return newlist;
605 }
606 #endif
607
608
609 static const char tar_options[]="ctxjT:X:C:f:Opvz";
610 #define CTX_CREATE      1
611 #define CTX_TEST        2
612 #define CTX_EXTRACT     4
613 #define TAR_OPT_BZIP2    8
614 #define TAR_OPT_INCLUDE  16
615 #define TAR_OPT_EXCLUDE  32
616 #define TAR_OPT_BASEDIR  64
617 #define TAR_OPT_ARNAME   128
618 #define TAR_OPT_2STDOUT  256
619 #define TAR_OPT_P        512
620 #define TAR_OPT_VERBOSE  1024
621 #define TAR_OPT_GZIP     2048
622
623 int tar_main(int argc, char **argv)
624 {
625         char (*get_header_ptr)(archive_handle_t *) = get_header_tar;
626         archive_handle_t *tar_handle;
627         int opt;
628         char *base_dir = NULL;
629         const char *tar_filename = "-";
630         unsigned char ctx_flag = 0;
631
632         if (argc < 2) {
633                 bb_show_usage();
634         }
635
636         /* Prepend '-' to the first argument if required */
637         if (argv[1][0] != '-') {
638                 char *tmp;
639
640                 bb_xasprintf(&tmp, "-%s", argv[1]);
641                 argv[1] = tmp;
642         }
643
644         /* Initialise default values */
645         tar_handle = init_handle();
646         tar_handle->flags = ARCHIVE_CREATE_LEADING_DIRS | ARCHIVE_PRESERVE_DATE;
647
648         bb_opt_complementaly = "c~tx:t~cx:x~ct:X*";
649         opt = bb_getopt_ulflags(argc, argv, tar_options,
650                                 NULL,   /* T: arg is ignored by default
651                                             a list is an include list */
652                                 &(tar_handle->reject),
653                                 &base_dir,      /* Change to dir <optarg> */
654                                 &tar_filename); /* archive filename */
655         /* Check one and only one context option was given */
656         if(opt & 0x80000000UL)
657                 bb_show_usage();
658         ctx_flag = opt & (CTX_CREATE | CTX_TEST | CTX_EXTRACT);
659         if(ctx_flag & CTX_TEST) {
660                         if ((tar_handle->action_header == header_list) || 
661                                 (tar_handle->action_header == header_verbose_list)) {
662                                 tar_handle->action_header = header_verbose_list;
663                         } else {
664                                 tar_handle->action_header = header_list;
665                         }
666         }
667         if(ctx_flag & CTX_EXTRACT) {
668                 if (tar_handle->action_data != data_extract_to_stdout)
669                                 tar_handle->action_data = data_extract_all;
670                         }
671         if(opt & TAR_OPT_2STDOUT) {
672                 /* To stdout */
673                         tar_handle->action_data = data_extract_to_stdout;
674         }
675         if(opt & TAR_OPT_VERBOSE) {
676                         if ((tar_handle->action_header == header_list) || 
677                                 (tar_handle->action_header == header_verbose_list)) 
678                         {
679                                 tar_handle->action_header = header_verbose_list;
680                         } else {
681                                 tar_handle->action_header = header_list;
682                         }
683         }
684 #ifdef CONFIG_FEATURE_TAR_GZIP
685         if(opt & TAR_OPT_GZIP) {
686                         get_header_ptr = get_header_tar_gz;
687         }
688 #endif
689 #ifdef CONFIG_FEATURE_TAR_BZIP2
690         if(opt & TAR_OPT_BZIP2) {
691                         get_header_ptr = get_header_tar_bz2;
692         }
693 #endif
694 #ifdef CONFIG_FEATURE_TAR_EXCLUDE
695         if(opt & TAR_OPT_EXCLUDE) {
696                 tar_handle->reject = append_file_list_to_list(tar_handle->reject);
697         }
698 #endif
699         /* Check if we are reading from stdin */
700         if ((argv[optind]) && (*argv[optind] == '-')) {
701                 /* Default is to read from stdin, so just skip to next arg */
702                 optind++;
703         }
704
705         /* Setup an array of filenames to work with */
706         /* TODO: This is the same as in ar, seperate function ? */
707         while (optind < argc) {
708                 tar_handle->accept = llist_add_to(tar_handle->accept, argv[optind]);
709                 optind++;
710         }
711
712         if ((tar_handle->accept) || (tar_handle->reject)) {
713                 tar_handle->filter = filter_accept_reject_list;
714         }
715
716 #ifdef CONFIG_FEATURE_TAR_CREATE
717         /* create an archive */
718         if (ctx_flag == CTX_CREATE) {
719                 int verboseFlag = FALSE;
720                 int gzipFlag = FALSE;
721
722 # ifdef CONFIG_FEATURE_TAR_GZIP
723                 if (get_header_ptr == get_header_tar_gz) {
724                         gzipFlag = TRUE;
725                 }
726 # endif /* CONFIG_FEATURE_TAR_GZIP */
727 # ifdef CONFIG_FEATURE_TAR_BZIP2
728                 if (get_header_ptr == get_header_tar_bz2) {
729                         bb_error_msg_and_die("Creating bzip2 compressed archives is not currently supported.");
730                 }
731 # endif /* CONFIG_FEATURE_TAR_BZIP2 */
732
733                 if ((tar_handle->action_header == header_list) || 
734                                 (tar_handle->action_header == header_verbose_list)) {
735                         verboseFlag = TRUE;
736                 }
737                 writeTarFile(tar_filename, verboseFlag, tar_handle->accept,
738                         tar_handle->reject, gzipFlag);
739         } else 
740 #endif /* CONFIG_FEATURE_TAR_CREATE */
741         {
742                 if ((tar_filename[0] == '-') && (tar_filename[1] == '\0')) {
743                         tar_handle->src_fd = fileno(stdin);
744                         tar_handle->seek = seek_by_char;
745                 } else {
746                         tar_handle->src_fd = bb_xopen(tar_filename, O_RDONLY);
747                 }
748
749                 if ((base_dir) && (chdir(base_dir))) {
750                         bb_perror_msg_and_die("Couldnt chdir");
751                 }
752
753                 while (get_header_ptr(tar_handle) == EXIT_SUCCESS);
754
755                 /* Ckeck that every file that should have been extracted was */
756                 while (tar_handle->accept) {
757                         if (find_list_entry(tar_handle->reject, tar_handle->accept->data) == NULL) {
758                                 if (find_list_entry(tar_handle->passed, tar_handle->accept->data) == NULL) {
759                                         bb_error_msg_and_die("%s: Not found in archive\n", tar_handle->accept->data);
760                                 }
761                         }
762                         tar_handle->accept = tar_handle->accept->link;
763                 }
764         }
765
766 #ifdef CONFIG_FEATURE_CLEAN_UP
767         if (tar_handle->src_fd != fileno(stdin)) {
768                 close(tar_handle->src_fd);
769         }
770 #endif /* CONFIG_FEATURE_CLEAN_UP */
771
772         return(EXIT_SUCCESS);
773 }