Isolate code better for unused options, config option to enable long
[people/mcb30/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->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 # ifdef CONFIG_FEATURE_TAR_FROM
321 static inline int exclude_file(const llist_t *excluded_files, const char *file)
322 {
323         while (excluded_files) {
324                 if (excluded_files->data[0] == '/') {
325                         if (fnmatch(excluded_files->data, file,
326                                                 FNM_PATHNAME | FNM_LEADING_DIR) == 0)
327                                 return 1;
328                 } else {
329                         const char *p;
330
331                         for (p = file; p[0] != '\0'; p++) {
332                                 if ((p == file || p[-1] == '/') && p[0] != '/' &&
333                                         fnmatch(excluded_files->data, p,
334                                                         FNM_PATHNAME | FNM_LEADING_DIR) == 0)
335                                         return 1;
336                         }
337                 }
338                 excluded_files = excluded_files->link;
339         }
340
341         return 0;
342 }
343 # endif
344
345 static int writeFileToTarball(const char *fileName, struct stat *statbuf,
346                                                           void *userData)
347 {
348         struct TarBallInfo *tbInfo = (struct TarBallInfo *) userData;
349         const char *header_name;
350
351         /*
352            ** Check to see if we are dealing with a hard link.
353            ** If so -
354            ** Treat the first occurance of a given dev/inode as a file while
355            ** treating any additional occurances as hard links.  This is done
356            ** by adding the file information to the HardLinkInfo linked list.
357          */
358         tbInfo->hlInfo = NULL;
359         if (statbuf->st_nlink > 1) {
360                 tbInfo->hlInfo = findHardLinkInfo(tbInfo->hlInfoHead, statbuf);
361                 if (tbInfo->hlInfo == NULL)
362                         addHardLinkInfo(&tbInfo->hlInfoHead, statbuf, fileName);
363         }
364
365         /* It is against the rules to archive a socket */
366         if (S_ISSOCK(statbuf->st_mode)) {
367                 bb_error_msg("%s: socket ignored", fileName);
368                 return (TRUE);
369         }
370
371         /* It is a bad idea to store the archive we are in the process of creating,
372          * so check the device and inode to be sure that this particular file isn't
373          * the new tarball */
374         if (tbInfo->statBuf.st_dev == statbuf->st_dev &&
375                 tbInfo->statBuf.st_ino == statbuf->st_ino) {
376                 bb_error_msg("%s: file is the archive; skipping", fileName);
377                 return (TRUE);
378         }
379
380         header_name = fileName;
381         while (header_name[0] == '/') {
382                 static int alreadyWarned = FALSE;
383
384                 if (alreadyWarned == FALSE) {
385                         bb_error_msg("Removing leading '/' from member names");
386                         alreadyWarned = TRUE;
387                 }
388                 header_name++;
389         }
390
391         if (strlen(fileName) >= NAME_SIZE) {
392                 bb_error_msg(bb_msg_name_longer_than_foo, NAME_SIZE);
393                 return (TRUE);
394         }
395
396         if (header_name[0] == '\0')
397                 return TRUE;
398
399 # ifdef CONFIG_FEATURE_TAR_FROM
400         if (exclude_file(tbInfo->excludeList, header_name)) {
401                 return SKIP;
402         }
403 # endif                                                 /* CONFIG_FEATURE_TAR_FROM */
404
405         if (writeTarHeader(tbInfo, header_name, fileName, statbuf) == FALSE) {
406                 return (FALSE);
407         }
408
409         /* Now, if the file is a regular file, copy it out to the tarball */
410         if ((tbInfo->hlInfo == NULL)
411                 && (S_ISREG(statbuf->st_mode))) {
412                 int inputFileFd;
413                 ssize_t readSize = 0;
414
415                 /* open the file we want to archive, and make sure all is well */
416                 if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
417                         bb_perror_msg("%s: Cannot open", fileName);
418                         return (FALSE);
419                 }
420
421                 /* write the file to the archive */
422                 readSize = bb_copyfd_eof(inputFileFd, tbInfo->tarFd);
423
424                 /* Pad the file up to the tar block size */
425                 for (; (readSize % TAR_BLOCK_SIZE) != 0; readSize++) {
426                         write(tbInfo->tarFd, "\0", 1);
427                 }
428                 close(inputFileFd);
429         }
430
431         return (TRUE);
432 }
433
434 static inline int writeTarFile(const int tar_fd, const int verboseFlag,
435                                                            const llist_t *include, const llist_t *exclude, const int gzip)
436 {
437 #ifdef CONFIG_FEATURE_TAR_GZIP
438         int gzipDataPipe[2] = { -1, -1 };
439         int gzipStatusPipe[2] = { -1, -1 };
440         pid_t gzipPid = 0;
441         volatile int vfork_exec_errno = 0;
442 #endif
443
444         int errorFlag = FALSE;
445         ssize_t size;
446         struct TarBallInfo tbInfo;
447
448         tbInfo.hlInfoHead = NULL;
449
450         fchmod(tar_fd, 0644);
451         tbInfo.tarFd = tar_fd;
452         tbInfo.verboseFlag = verboseFlag;
453
454         /* Store the stat info for the tarball's file, so
455          * can avoid including the tarball into itself....  */
456         if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0)
457                 bb_perror_msg_and_die("Couldnt stat tar file");
458
459 #ifdef CONFIG_FEATURE_TAR_GZIP
460         if (gzip) {
461                 if (pipe(gzipDataPipe) < 0 || pipe(gzipStatusPipe) < 0) {
462                         bb_perror_msg_and_die("Failed to create gzip pipe");
463                 }
464
465                 signal(SIGPIPE, SIG_IGN);       /* we only want EPIPE on errors */
466
467 # if __GNUC__
468                         /* Avoid vfork clobbering */
469                         (void) &include;
470                         (void) &errorFlag;
471 # endif
472  
473                 gzipPid = vfork();
474
475                 if (gzipPid == 0) {
476                         dup2(gzipDataPipe[0], 0);
477                         close(gzipDataPipe[1]);
478
479                         if (tbInfo.tarFd != 1);
480                         dup2(tbInfo.tarFd, 1);
481
482                         close(gzipStatusPipe[0]);
483                         fcntl(gzipStatusPipe[1], F_SETFD, FD_CLOEXEC);  /* close on exec shows sucess */
484
485                         execl("/bin/gzip", "gzip", "-f", 0);
486                         vfork_exec_errno = errno;
487
488                         close(gzipStatusPipe[1]);
489                         exit(-1);
490                 } else if (gzipPid > 0) {
491                         close(gzipDataPipe[0]);
492                         close(gzipStatusPipe[1]);
493
494                         while (1) {
495                                 char buf;
496
497                                 int n = bb_full_read(gzipStatusPipe[0], &buf, 1);
498
499                                 if (n == 0 && vfork_exec_errno != 0) {
500                                         errno = vfork_exec_errno;
501                                         bb_perror_msg_and_die("Could not exec gzip process");
502                                 } else if ((n < 0) && (errno == EAGAIN || errno == EINTR))
503                                         continue;       /* try it again */
504                                 break;
505                         }
506                         close(gzipStatusPipe[0]);
507
508                         tbInfo.tarFd = gzipDataPipe[1];
509                 } else {
510                         bb_perror_msg_and_die("Failed to vfork gzip process");
511                 }
512         }
513 #endif
514
515         tbInfo.excludeList = exclude;
516
517         /* Read the directory/files and iterate over them one at a time */
518         while (include) {
519                 if (!recursive_action(include->data, TRUE, FALSE, FALSE,
520                                                           writeFileToTarball, writeFileToTarball,
521                                                           (void *) &tbInfo)) {
522                         errorFlag = TRUE;
523                 }
524                 include = include->link;
525         }
526         /* Write two empty blocks to the end of the archive */
527         for (size = 0; size < (2 * TAR_BLOCK_SIZE); size++) {
528                 write(tbInfo.tarFd, "\0", 1);
529         }
530
531         /* To be pedantically correct, we would check if the tarball
532          * is smaller than 20 tar blocks, and pad it if it was smaller,
533          * but that isn't necessary for GNU tar interoperability, and
534          * so is considered a waste of space */
535
536         /* Hang up the tools, close up shop, head home */
537         close(tbInfo.tarFd);
538         if (errorFlag)
539                 bb_error_msg("Error exit delayed from previous errors");
540
541         freeHardLinkInfo(&tbInfo.hlInfoHead);
542
543 #ifdef CONFIG_FEATURE_TAR_GZIP
544         if (gzip && gzipPid) {
545                 if (waitpid(gzipPid, NULL, 0) == -1)
546                         printf("Couldnt wait ?");
547         }
548 #endif
549
550         return !errorFlag;
551 }
552 #endif                                                  /* tar_create */
553
554 #ifdef CONFIG_FEATURE_TAR_FROM
555 static llist_t *append_file_list_to_list(llist_t *list)
556 {
557         FILE *src_stream;
558         llist_t *cur = list;
559         llist_t *tmp;
560         char *line;
561         llist_t *newlist = NULL;
562
563         while(cur) {
564                 src_stream = bb_xfopen(cur->data, "r");
565                 tmp = cur;
566                 cur = cur->link;
567                 free(tmp);
568         while((line = bb_get_chomped_line_from_file(src_stream)) != NULL) {
569                         newlist = llist_add_to(newlist, line);
570         }
571         fclose(src_stream);
572         }
573         return newlist;
574 }
575 #endif
576
577 #ifdef CONFIG_FEATURE_TAR_COMPRESS
578 static char get_header_tar_Z(archive_handle_t *archive_handle)
579 {
580         /* Cant lseek over pipe's */
581         archive_handle->seek = seek_by_char;
582
583         /* do the decompression, and cleanup */
584         if ((bb_xread_char(archive_handle->src_fd) != 0x1f) || (bb_xread_char(archive_handle->src_fd) != 0x9d)) {
585                 bb_error_msg_and_die("Invalid magic");
586         }
587
588         archive_handle->src_fd = open_transformer(archive_handle->src_fd, uncompress);
589         archive_handle->offset = 0;
590         while (get_header_tar(archive_handle) == EXIT_SUCCESS);
591
592         /* Can only do one file at a time */
593         return(EXIT_FAILURE);
594 }
595 #endif
596
597 #define CTX_TEST        (1 << 0)
598 #define CTX_EXTRACT     (1 << 1)
599 #define TAR_OPT_BASEDIR  (1 << 2)
600 #define TAR_OPT_TARNAME  (1 << 3)
601 #define TAR_OPT_2STDOUT  (1 << 4)
602 #define TAR_OPT_P        (1 << 5)
603 #define TAR_OPT_VERBOSE  (1 << 6)
604 #define TAR_OPT_KEEP_OLD        (1 << 7)
605
606 #ifdef CONFIG_FEATURE_TAR_CREATE
607 # define CTX_CREATE     (1 << 8)
608 # define TAR_OPT_STR_CREATE     "c"
609 # define TAR_OPT_FLAG_CREATE    1
610 #else
611 //# define CTX_CREATE   0
612 # define TAR_OPT_STR_CREATE     ""
613 # define TAR_OPT_FLAG_CREATE    0
614 #endif
615
616 #ifdef CONFIG_FEATURE_TAR_BZIP2
617 # define TAR_OPT_BZIP2  (1 << (8 + TAR_OPT_FLAG_CREATE)) 
618 # define TAR_OPT_STR_BZIP2      "j"
619 # define TAR_OPT_FLAG_BZIP2     1
620 #else
621 # define TAR_OPT_STR_BZIP2      ""
622 # define TAR_OPT_FLAG_BZIP2     0
623 #endif
624
625 #ifdef CONFIG_FEATURE_TAR_FROM
626 # define TAR_OPT_FROM_FILE      (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2))
627 # define TAR_OPT_EXCLUDE_FROM   (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2 + 1))
628 # define TAR_OPT_STR_FROM       "T:X:"
629 # define TAR_OPT_FLAG_FROM      2
630 #else
631 # define TAR_OPT_STR_FROM       ""
632 # define TAR_OPT_FLAG_FROM      0
633 #endif
634
635 #ifdef CONFIG_FEATURE_TAR_GZIP
636 # define TAR_OPT_GZIP   (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2 + TAR_OPT_FLAG_FROM))
637 # define TAR_OPT_STR_GZIP       "z"
638 # define TAR_OPT_FLAG_GZIP      1
639 #else
640 # define TAR_OPT_STR_GZIP       ""
641 # define TAR_OPT_FLAG_GZIP      0
642 #endif
643
644 #ifdef CONFIG_FEATURE_TAR_COMPRESS
645 # define TAR_OPT_UNCOMPRESS     (1 << (8 + TAR_OPT_FLAG_CREATE + TAR_OPT_FLAG_BZIP2 + TAR_OPT_FLAG_FROM + TAR_OPT_FLAG_GZIP))
646 # define TAR_OPT_STR_COMPRESS   "Z"
647 #else 
648 # define TAR_OPT_STR_COMPRESS   ""
649 #endif
650
651 static const char tar_options[]="txC:f:Opvk" \
652         TAR_OPT_STR_CREATE \
653         TAR_OPT_STR_BZIP2 \
654         TAR_OPT_STR_FROM \
655         TAR_OPT_STR_GZIP \
656         TAR_OPT_STR_COMPRESS;
657
658 #ifdef CONFIG_FEATURE_TAR_LONG_OPTIONS
659 static const struct option tar_long_options[] = {
660         { "list",                       0,      NULL,   't' },
661         { "extract",            0,      NULL,   'x' },
662         { "directory",          1,      NULL,   'C' },
663         { "file",                       1,      NULL,   'f'     },
664         { "to-stdout",          0,      NULL,   'O' },
665         { "same-permissions",   0,      NULL,   'p' },
666         { "verbose",            0,      NULL,   'v' },
667         { "keep-old",           0,      NULL,   'k' },
668 # ifdef CONFIG_FEATURE_TAR_CREATE
669         { "create",                     0,      NULL,   'c' },
670 # endif
671 # ifdef CONFIG_FEATURE_TAR_BZIP2
672         { "bzip2",                      0,      NULL,   'j' },
673 # endif
674 # ifdef CONFIG_FEATURE_TAR_FROM
675         { "from-file",          1,      NULL,   'T' },
676         { "exclude-from",       1,      NULL,   'X' },
677 # endif
678 # ifdef CONFIG_FEATURE_TAR_GZIP
679         { "gzip",                       0,      NULL,   'z' },
680 # endif
681 # ifdef CONFIG_FEATURE_TAR_COMPRESS
682         { "compress",           0,      NULL,   'Z' },
683 # endif
684         { 0,                 0, 0, 0 }
685 };
686 #endif
687
688 int tar_main(int argc, char **argv)
689 {
690         char (*get_header_ptr)(archive_handle_t *) = get_header_tar;
691         archive_handle_t *tar_handle;
692         char *base_dir = NULL;
693         const char *tar_filename = "-";
694         unsigned long opt;
695         unsigned long ctx_flag = 0;
696
697         if (argc < 2) {
698                 bb_show_usage();
699         }
700
701         /* Prepend '-' to the first argument if required */
702         if (argv[1][0] != '-') {
703                 char *tmp;
704
705                 bb_xasprintf(&tmp, "-%s", argv[1]);
706                 argv[1] = tmp;
707         }
708
709         /* Initialise default values */
710         tar_handle = init_handle();
711         tar_handle->flags = ARCHIVE_CREATE_LEADING_DIRS | ARCHIVE_PRESERVE_DATE | ARCHIVE_EXTRACT_UNCONDITIONAL;
712
713         bb_opt_complementaly = "c~tx:t~cx:x~ct:X*:T*";
714 #ifdef CONFIG_FEATURE_TAR_LONG_OPTIONS
715         bb_applet_long_options = tar_long_options;
716 #endif
717
718         opt = bb_getopt_ulflags(argc, argv, tar_options,
719                                 &base_dir,      /* Change to dir <optarg> */
720                                 &tar_filename /* archive filename */
721 #ifdef CONFIG_FEATURE_TAR_FROM
722                                 , NULL,
723                                 &(tar_handle->reject)
724 #endif
725                                 );
726
727         /* Check one and only one context option was given */
728         if(opt & 0x80000000UL) {
729                 bb_show_usage();
730         }
731 #ifdef CONFIG_FEATURE_TAR_CREATE
732         ctx_flag = opt & (CTX_CREATE | CTX_TEST | CTX_EXTRACT);
733 #else
734         ctx_flag = opt & (CTX_TEST | CTX_EXTRACT);
735 #endif
736         if (ctx_flag == 0) {
737                 bb_show_usage();
738         }
739         if(ctx_flag & CTX_TEST) {
740                 if ((tar_handle->action_header == header_list) || 
741                         (tar_handle->action_header == header_verbose_list)) {
742                         tar_handle->action_header = header_verbose_list;
743                 } else {
744                         tar_handle->action_header = header_list;
745                 }
746         }
747         if(ctx_flag & CTX_EXTRACT) {
748                 if (tar_handle->action_data != data_extract_to_stdout)
749                         tar_handle->action_data = data_extract_all;
750                 }
751         if(opt & TAR_OPT_2STDOUT) {
752                 /* To stdout */
753                 tar_handle->action_data = data_extract_to_stdout;
754         }
755         if(opt & TAR_OPT_VERBOSE) {
756                 if ((tar_handle->action_header == header_list) || 
757                         (tar_handle->action_header == header_verbose_list)) 
758                 {
759                 tar_handle->action_header = header_verbose_list;
760                 } else {
761                         tar_handle->action_header = header_list;
762                 }
763         }
764         if (opt & TAR_OPT_KEEP_OLD) {
765                 tar_handle->flags &= ~ARCHIVE_EXTRACT_UNCONDITIONAL;
766         }
767
768 #ifdef CONFIG_FEATURE_TAR_GZIP
769         if(opt & TAR_OPT_GZIP) {
770                 get_header_ptr = get_header_tar_gz;
771         }
772 #endif
773 #ifdef CONFIG_FEATURE_TAR_BZIP2
774         if(opt & TAR_OPT_BZIP2) {
775                 get_header_ptr = get_header_tar_bz2;
776         }
777 #endif
778 #ifdef CONFIG_FEATURE_TAR_COMPRESS
779         if(opt & TAR_OPT_UNCOMPRESS) {
780                 get_header_ptr = get_header_tar_Z;
781         }
782 #endif
783 #ifdef CONFIG_FEATURE_TAR_FROM
784         if(opt & TAR_OPT_EXCLUDE_FROM) {
785                 tar_handle->reject = append_file_list_to_list(tar_handle->reject);
786         }
787 #endif
788
789         /* Check if we are reading from stdin */
790         if ((argv[optind]) && (*argv[optind] == '-')) {
791                 /* Default is to read from stdin, so just skip to next arg */
792                 optind++;
793         }
794
795         /* Setup an array of filenames to work with */
796         /* TODO: This is the same as in ar, seperate function ? */
797         while (optind < argc) {
798                 char *filename_ptr = last_char_is(argv[optind], '/');
799                 if (filename_ptr) {
800                         *filename_ptr = '\0';
801                 }
802                 tar_handle->accept = llist_add_to(tar_handle->accept, argv[optind]);
803                 optind++;
804         }
805
806         if ((tar_handle->accept) || (tar_handle->reject)) {
807                 tar_handle->filter = filter_accept_reject_list;
808         }
809
810         /* Open the tar file */
811         {
812                 FILE *tar_stream;
813                 int flags;
814
815 #ifdef CONFIG_FEATURE_TAR_CREATE
816                 if (opt & CTX_CREATE) {
817                         /* Make sure there is at least one file to tar up.  */
818                         if (tar_handle->accept == NULL) {
819                                 bb_error_msg_and_die("Cowardly refusing to create an empty archive");
820                         }
821                         tar_stream = stdout;
822                         flags = O_WRONLY | O_CREAT | O_EXCL;
823                         unlink(tar_filename);
824                 } else
825 #endif
826                 {
827                         tar_stream = stdin;
828                         flags = O_RDONLY;
829                 }
830
831                 if ((tar_filename[0] == '-') && (tar_filename[1] == '\0')) {
832                         tar_handle->src_fd = fileno(tar_stream);
833                         tar_handle->seek = seek_by_char;
834                 } else {
835                         tar_handle->src_fd = bb_xopen(tar_filename, flags);
836                 }
837         }
838
839         if ((base_dir) && (chdir(base_dir))) {
840                 bb_perror_msg_and_die("Couldnt chdir to %s", base_dir);
841         }
842
843 #ifdef CONFIG_FEATURE_TAR_CREATE
844         /* create an archive */
845         if (opt & CTX_CREATE) {
846                 int verboseFlag = FALSE;
847                 int gzipFlag = FALSE;
848
849 # ifdef CONFIG_FEATURE_TAR_GZIP
850                 if (get_header_ptr == get_header_tar_gz) {
851                         gzipFlag = TRUE;
852                 }
853 # endif /* CONFIG_FEATURE_TAR_GZIP */
854 # ifdef CONFIG_FEATURE_TAR_BZIP2
855                 if (get_header_ptr == get_header_tar_bz2) {
856                         bb_error_msg_and_die("Creating bzip2 compressed archives is not currently supported.");
857                 }
858 # endif /* CONFIG_FEATURE_TAR_BZIP2 */
859
860                 if ((tar_handle->action_header == header_list) || 
861                                 (tar_handle->action_header == header_verbose_list)) {
862                         verboseFlag = TRUE;
863                 }
864                 writeTarFile(tar_handle->src_fd, verboseFlag, tar_handle->accept,
865                         tar_handle->reject, gzipFlag);
866         } else 
867 #endif /* CONFIG_FEATURE_TAR_CREATE */
868         {
869                 while (get_header_ptr(tar_handle) == EXIT_SUCCESS);
870
871                 /* Ckeck that every file that should have been extracted was */
872                 while (tar_handle->accept) {
873                         if (find_list_entry(tar_handle->reject, tar_handle->accept->data) == NULL) {
874                                 if (find_list_entry(tar_handle->passed, tar_handle->accept->data) == NULL) {
875                                         bb_error_msg_and_die("%s: Not found in archive\n", tar_handle->accept->data);
876                                 }
877                         }
878                         tar_handle->accept = tar_handle->accept->link;
879                 }
880         }
881
882 #ifdef CONFIG_FEATURE_CLEAN_UP
883         if (tar_handle->src_fd != fileno(stdin)) {
884                 close(tar_handle->src_fd);
885         }
886 #endif /* CONFIG_FEATURE_CLEAN_UP */
887
888         return(EXIT_SUCCESS);
889 }