Allow short reads when filling compress buffer
[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,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         const llist_t *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         if (my_getpwuid(header.uname, statbuf->st_uid) == NULL)
248                 strcpy(header.uname, "root");
249         if (my_getgrgid(header.gname, statbuf->st_gid) == NULL)
250                 strcpy(header.gname, "root");
251
252         if (tbInfo->hlInfo) {
253                 /* This is a hard link */
254                 header.typeflag = LNKTYPE;
255                 strncpy(header.linkname, tbInfo->hlInfo->name,
256                                 sizeof(header.linkname));
257         } else if (S_ISLNK(statbuf->st_mode)) {
258                 char *lpath = xreadlink(real_name);
259
260                 if (!lpath)             /* Already printed err msg inside xreadlink() */
261                         return (FALSE);
262                 header.typeflag = SYMTYPE;
263                 strncpy(header.linkname, lpath, sizeof(header.linkname));
264                 free(lpath);
265         } else if (S_ISDIR(statbuf->st_mode)) {
266                 header.typeflag = DIRTYPE;
267                 strncat(header.name, "/", sizeof(header.name));
268         } else if (S_ISCHR(statbuf->st_mode)) {
269                 header.typeflag = CHRTYPE;
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_ISBLK(statbuf->st_mode)) {
275                 header.typeflag = BLKTYPE;
276                 putOctal(header.devmajor, sizeof(header.devmajor),
277                                  MAJOR(statbuf->st_rdev));
278                 putOctal(header.devminor, sizeof(header.devminor),
279                                  MINOR(statbuf->st_rdev));
280         } else if (S_ISFIFO(statbuf->st_mode)) {
281                 header.typeflag = FIFOTYPE;
282         } else if (S_ISREG(statbuf->st_mode)) {
283                 header.typeflag = REGTYPE;
284                 putOctal(header.size, sizeof(header.size), statbuf->st_size);
285         } else {
286                 error_msg("%s: Unknown file type", real_name);
287                 return (FALSE);
288         }
289
290         /* Calculate and store the checksum (i.e., the sum of all of the bytes of
291          * the header).  The checksum field must be filled with blanks for the
292          * calculation.  The checksum field is formatted differently from the
293          * other fields: it has [6] digits, a null, then a space -- rather than
294          * digits, followed by a null like the other fields... */
295         memset(header.chksum, ' ', sizeof(header.chksum));
296         cp = (const unsigned char *) &header;
297         while (size-- > 0)
298                 chksum += *cp++;
299         putOctal(header.chksum, 7, chksum);
300
301         /* Now write the header out to disk */
302         if ((size =
303                  full_write(tbInfo->tarFd, (char *) &header,
304                                         sizeof(struct TarHeader))) < 0) {
305                 error_msg(io_error, real_name);
306                 return (FALSE);
307         }
308         /* Pad the header up to the tar block size */
309         for (; size < TAR_BLOCK_SIZE; size++) {
310                 write(tbInfo->tarFd, "\0", 1);
311         }
312         /* Now do the verbose thing (or not) */
313
314         if (tbInfo->verboseFlag) {
315                 FILE *vbFd = stdout;
316
317                 if (tbInfo->verboseFlag == 2)   /* If the archive goes to stdout, verbose to stderr */
318                         vbFd = stderr;
319                 fprintf(vbFd, "%s\n", header.name);
320         }
321
322         return (TRUE);
323 }
324
325 # if defined CONFIG_FEATURE_TAR_EXCLUDE
326 static inline int exclude_file(const llist_t *excluded_files, const char *file)
327 {
328         if (excluded_files == NULL) {
329                 return 0;
330         }
331
332         while (excluded_files) {
333                 if (excluded_files->data[0] == '/') {
334                         if (fnmatch(excluded_files->data, file,
335                                                 FNM_PATHNAME | FNM_LEADING_DIR) == 0)
336                                 return 1;
337                 } else {
338                         const char *p;
339
340                         for (p = file; p[0] != '\0'; p++) {
341                                 if ((p == file || p[-1] == '/') && p[0] != '/' &&
342                                         fnmatch(excluded_files->data, p,
343                                                         FNM_PATHNAME | FNM_LEADING_DIR) == 0)
344                                         return 1;
345                         }
346                 }
347                 excluded_files = excluded_files->link;
348         }
349
350         return 0;
351 }
352 #endif
353
354 static int writeFileToTarball(const char *fileName, struct stat *statbuf,
355                                                           void *userData)
356 {
357         struct TarBallInfo *tbInfo = (struct TarBallInfo *) userData;
358         const char *header_name;
359
360         /*
361            ** Check to see if we are dealing with a hard link.
362            ** If so -
363            ** Treat the first occurance of a given dev/inode as a file while
364            ** treating any additional occurances as hard links.  This is done
365            ** by adding the file information to the HardLinkInfo linked list.
366          */
367         tbInfo->hlInfo = NULL;
368         if (statbuf->st_nlink > 1) {
369                 tbInfo->hlInfo = findHardLinkInfo(tbInfo->hlInfoHead, statbuf->st_dev,
370                                                                                   statbuf->st_ino);
371                 if (tbInfo->hlInfo == NULL)
372                         addHardLinkInfo(&tbInfo->hlInfoHead, statbuf->st_dev,
373                                                         statbuf->st_ino, statbuf->st_nlink, fileName);
374         }
375
376         /* It is against the rules to archive a socket */
377         if (S_ISSOCK(statbuf->st_mode)) {
378                 error_msg("%s: socket ignored", fileName);
379                 return (TRUE);
380         }
381
382         /* It is a bad idea to store the archive we are in the process of creating,
383          * so check the device and inode to be sure that this particular file isn't
384          * the new tarball */
385         if (tbInfo->statBuf.st_dev == statbuf->st_dev &&
386                 tbInfo->statBuf.st_ino == statbuf->st_ino) {
387                 error_msg("%s: file is the archive; skipping", fileName);
388                 return (TRUE);
389         }
390
391         header_name = fileName;
392         while (header_name[0] == '/') {
393                 static int alreadyWarned = FALSE;
394
395                 if (alreadyWarned == FALSE) {
396                         error_msg("Removing leading '/' from member names");
397                         alreadyWarned = TRUE;
398                 }
399                 header_name++;
400         }
401
402         if (strlen(fileName) >= NAME_SIZE) {
403                 error_msg(name_longer_than_foo, NAME_SIZE);
404                 return (TRUE);
405         }
406
407         if (header_name[0] == '\0')
408                 return TRUE;
409
410 # if defined CONFIG_FEATURE_TAR_EXCLUDE
411         if (exclude_file(tbInfo->excludeList, header_name)) {
412                 return SKIP;
413         }
414 # endif                                                 /* CONFIG_FEATURE_TAR_EXCLUDE */
415
416         if (writeTarHeader(tbInfo, header_name, fileName, statbuf) == FALSE) {
417                 return (FALSE);
418         }
419
420         /* Now, if the file is a regular file, copy it out to the tarball */
421         if ((tbInfo->hlInfo == NULL)
422                 && (S_ISREG(statbuf->st_mode))) {
423                 int inputFileFd;
424                 char buffer[BUFSIZ];
425                 ssize_t size = 0, readSize = 0;
426
427                 /* open the file we want to archive, and make sure all is well */
428                 if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
429                         perror_msg("%s: Cannot open", fileName);
430                         return (FALSE);
431                 }
432
433                 /* write the file to the archive */
434                 while ((size = full_read(inputFileFd, buffer, sizeof(buffer))) > 0) {
435                         if (full_write(tbInfo->tarFd, buffer, size) != size) {
436                                 /* Output file seems to have a problem */
437                                 error_msg(io_error, fileName);
438                                 return (FALSE);
439                         }
440                         readSize += size;
441                 }
442                 if (size == -1) {
443                         error_msg(io_error, fileName);
444                         return (FALSE);
445                 }
446                 /* Pad the file up to the tar block size */
447                 for (; (readSize % TAR_BLOCK_SIZE) != 0; readSize++) {
448                         write(tbInfo->tarFd, "\0", 1);
449                 }
450                 close(inputFileFd);
451         }
452
453         return (TRUE);
454 }
455
456 static inline int writeTarFile(const char *tarName, const int verboseFlag,
457                                                            const llist_t *include, const llist_t *exclude, const int gzip)
458 {
459 #ifdef CONFIG_FEATURE_TAR_GZIP
460         int gzipDataPipe[2] = { -1, -1 };
461         int gzipStatusPipe[2] = { -1, -1 };
462         pid_t gzipPid = 0;
463 #endif
464
465         int errorFlag = FALSE;
466         ssize_t size;
467         struct TarBallInfo tbInfo;
468
469         tbInfo.hlInfoHead = NULL;
470
471         /* Make sure there is at least one file to tar up.  */
472         if (include == NULL) {
473                 error_msg_and_die("Cowardly refusing to create an empty archive");
474         }
475
476         /* Open the tar file for writing.  */
477         if (tarName == NULL || (tarName[0] == '-' && tarName[1] == '\0')) {
478                 tbInfo.tarFd = fileno(stdout);
479                 tbInfo.verboseFlag = verboseFlag ? 2 : 0;
480         } else {
481                 tbInfo.tarFd = open(tarName, O_WRONLY | O_CREAT | O_TRUNC, 0644);
482                 tbInfo.verboseFlag = verboseFlag ? 1 : 0;
483         }
484
485         if (tbInfo.tarFd < 0) {
486                 perror_msg("%s: Cannot open", tarName);
487                 freeHardLinkInfo(&tbInfo.hlInfoHead);
488                 return (FALSE);
489         }
490
491         /* Store the stat info for the tarball's file, so
492          * can avoid including the tarball into itself....  */
493         if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0)
494                 error_msg_and_die(io_error, tarName);
495
496 #ifdef CONFIG_FEATURE_TAR_GZIP
497         if (gzip) {
498                 if (socketpair(AF_UNIX, SOCK_STREAM, 0, gzipDataPipe) < 0
499                         || pipe(gzipStatusPipe) < 0)
500                         perror_msg_and_die("Failed to create gzip pipe");
501
502                 signal(SIGPIPE, SIG_IGN);       /* we only want EPIPE on errors */
503
504                 gzipPid = fork();
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
518                         write(gzipStatusPipe[1], "", 1);
519                         close(gzipStatusPipe[1]);
520
521                         exit(-1);
522                 } else if (gzipPid > 0) {
523                         close(gzipDataPipe[0]);
524                         close(gzipStatusPipe[1]);
525
526                         while (1) {
527                                 char buf;
528
529                                 int n = read(gzipStatusPipe[0], &buf, 1);
530
531                                 if (n == 1)
532                                         error_msg_and_die("Could not exec gzip process");       /* socket was not closed => error */
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                         perror_msg_and_die("Failed to fork 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                 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 const llist_t *append_file_list_to_list(const char *filename, const llist_t *list)
587 {
588         FILE *src_stream = xfopen(filename, "r");
589         char *line;
590         while((line = get_line_from_file(src_stream)) != NULL) {
591                 chomp(line);
592                 list = add_to_list(list, line);
593         }
594         fclose(src_stream);
595
596         return (list);
597 }
598 #endif
599
600 int tar_main(int argc, char **argv)
601 {
602 #ifdef CONFIG_FEATURE_TAR_GZIP
603         char (*get_header_ptr)(archive_handle_t *) = get_header_tar;
604 #endif
605         archive_handle_t *tar_handle;
606         int opt;
607         char *base_dir = NULL;
608         char *tar_filename = "-";
609
610 #ifdef CONFIG_FEATURE_TAR_CREATE
611         unsigned char tar_create = FALSE;
612 #endif
613
614         /* Prepend '-' to the first argument if required */
615         if (argv[1][0] != '-') {
616                 char *tmp = xmalloc(strlen(argv[1]) + 2);
617                 tmp[0] = '-';
618                 strcpy(tmp + 1, argv[1]);
619                 argv[1] = tmp;
620         }
621
622         if (argc < 2) {
623                 show_usage();
624         }
625
626         /* Initialise default values */
627         tar_handle = init_handle();
628         tar_handle->flags = ARCHIVE_CREATE_LEADING_DIRS;
629
630         while ((opt = getopt(argc, argv, "ctxT:X:C:f:Opvz")) != -1) {
631                 switch (opt) {
632                         /* One and only one of these is required */
633 #ifdef CONFIG_FEATURE_TAR_CREATE
634                 case 'c':
635                         tar_create = TRUE;
636                         break;
637 #endif
638                 case 't':
639                         if ((tar_handle->action_header == header_list) || 
640                                 (tar_handle->action_header == header_verbose_list)) {
641                                 tar_handle->action_header = header_verbose_list;
642                         } else {
643                                 tar_handle->action_header = header_list;
644                         }
645                         break;
646                 case 'x':
647                         tar_handle->action_data = data_extract_all;
648                         break;
649
650                         /* These are optional */
651                         /* Exclude or Include files listed in <filename> */
652 #ifdef CONFIG_FEATURE_TAR_EXCLUDE
653                 case 'X':
654                         tar_handle->reject =
655                                 append_file_list_to_list(optarg, tar_handle->reject);
656                         break;
657 #endif
658                 case 'T':
659                         /* by default a list is an include list */
660                         break;
661                 case 'C':               /* Change to dir <optarg> */
662                         base_dir = optarg;
663                         break;
664                 case 'f':               /* archive filename */
665                         tar_filename = optarg;
666                         break;
667                 case 'O':               /* To stdout */
668                         tar_handle->action_data = data_extract_to_stdout;
669                         break;
670                 case 'p':
671                         tar_handle->flags |= ARCHIVE_PRESERVE_DATE;
672                         break;
673                 case 'v':
674                         if ((tar_handle->action_header == header_list) || 
675                                 (tar_handle->action_header == header_verbose_list)) {
676                                 tar_handle->action_header = header_verbose_list;
677                         } else {
678                                 tar_handle->action_header = header_list;
679                         }
680                         break;
681 #ifdef CONFIG_FEATURE_TAR_GZIP
682                 case 'z':
683                         get_header_ptr = get_header_tar_gz;
684                         break;
685 #endif
686 #ifdef CONFIG_FEATURE_TAR_BZIP2
687                         /* Not enabled yet */
688                 case 'j':
689                         archive_handle->archive_action = bunzip2;
690                         break;
691 #endif
692                 default:
693                         show_usage();
694                 }
695         }
696
697         /* Check if we are reading from stdin */
698         if ((argv[optind]) && (*argv[optind] == '-')) {
699                 /* Default is to read from stdin, so just skip to next arg */
700                 optind++;
701         }
702
703         /* Setup an array of filenames to work with */
704         /* TODO: This is the same as in ar, seperate function ? */
705         while (optind < argc) {
706 #if 0
707                 char absolute_path[PATH_MAX];
708                 realpath(argv[optind], absolute_path);
709                 tar_handle->accept = add_to_list(tar_handle->accept, absolute_path);
710 #endif
711                 tar_handle->accept = add_to_list(tar_handle->accept, argv[optind]);
712                 optind++;
713
714         }
715
716         if ((tar_handle->accept) || (tar_handle->reject)) {
717                 tar_handle->filter = filter_accept_reject_list;
718         }
719
720         if ((base_dir) && (chdir(base_dir))) {
721                 perror_msg_and_die("Couldnt chdir");
722         }
723
724 #ifdef CONFIG_FEATURE_TAR_CREATE
725         /* create an archive */
726         if (tar_create == TRUE) {
727                 int verboseFlag = FALSE;
728                 int gzipFlag = FALSE;
729
730 # ifdef CONFIG_FEATURE_TAR_GZIP
731                 if (get_header_ptr == get_header_tar_gz) {
732                         gzipFlag = TRUE;
733                 }
734 # endif /* CONFIG_FEATURE_TAR_GZIP */
735
736                 if (tar_handle->action_header == header_verbose_list) {
737                         verboseFlag = TRUE;
738                 }
739                 writeTarFile(tar_filename, verboseFlag, tar_handle->accept,
740                         tar_handle->reject, gzipFlag);
741         } else 
742 #endif /* CONFIG_FEATURE_TAR_CREATE */
743         {
744                 if ((tar_filename[0] == '-') && (tar_filename[1] == '\0')) {
745                         tar_handle->src_fd = fileno(stdin);
746                 } else {
747                         tar_handle->src_fd = xopen(tar_filename, O_RDONLY);
748                 }
749 #ifdef CONFIG_FEATURE_TAR_GZIP
750                 if (get_header_ptr == get_header_tar_gz) {
751                         get_header_tar_gz(tar_handle);
752                 } else
753 #endif /* CONFIG_FEATURE_TAR_CREATE */
754
755                         while (get_header_tar(tar_handle) == EXIT_SUCCESS);
756
757                 /* Ckeck that every file that should have been extracted was */
758                 while (tar_handle->accept) {
759                         if (find_list_entry(tar_handle->reject, tar_handle->accept->data) == NULL) {
760                                 if (find_list_entry(tar_handle->passed, tar_handle->accept->data) == NULL) {
761                                         error_msg_and_die("%s: Not found in archive\n", tar_handle->accept->data);
762                                 }
763                         }
764                         tar_handle->accept = tar_handle->accept->link;
765                 }
766         }
767
768 #ifdef CONFIG_FEATURE_CLEAN_UP
769         if (tar_handle->src_fd != fileno(stdin)) {
770                 close(tar_handle->src_fd);
771         }
772 #endif /* CONFIG_FEATURE_CLEAN_UP */
773
774         return(EXIT_SUCCESS);
775 }