Changed bb_regcomp to xregcomp and #if 0'ed out destroy_cmd_strs in sed.c
[oweals/busybox.git] / archival / tar.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini tar implementation for busybox 
4  *
5  * Note, that as of BusyBox-0.43, tar has been completely rewritten from the
6  * ground up.  It still has remnents of the old code lying about, but it is
7  * very different now (i.e. cleaner, less global variables, etc)
8  *
9  * Copyright (C) 2000 by Lineo, inc.
10  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
11  *
12  * Based in part in the tar implementation in sash
13  *  Copyright (c) 1999 by David I. Bell
14  *  Permission is granted to use, distribute, or modify this source,
15  *  provided that this copyright notice remains intact.
16  *  Permission to distribute sash derived code under the GPL has been granted.
17  *
18  * Based in part on the tar implementation from busybox-0.28
19  *  Copyright (C) 1995 Bruce Perens
20  *  This is free software under the GNU General Public License.
21  *
22  * This program is free software; you can redistribute it and/or modify
23  * it under the terms of the GNU General Public License as published by
24  * the Free Software Foundation; either version 2 of the License, or
25  * (at your option) any later version.
26  *
27  * This program is distributed in the hope that it will be useful,
28  * but WITHOUT ANY WARRANTY; without even the implied warranty of
29  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
30  * General Public License for more details.
31  *
32  * You should have received a copy of the GNU General Public License
33  * along with this program; if not, write to the Free Software
34  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
35  *
36  */
37
38
39 #include "internal.h"
40 #define BB_DECLARE_EXTERN
41 #define bb_need_io_error
42 #include "messages.c"
43 #include <stdio.h>
44 #include <dirent.h>
45 #include <errno.h>
46 #include <fcntl.h>
47 #include <signal.h>
48 #include <time.h>
49 #include <utime.h>
50 #include <sys/types.h>
51 #include <sys/sysmacros.h>
52
53
54 static const char tar_usage[] =
55 #ifdef BB_FEATURE_TAR_CREATE
56         "tar -[cxtvO] "
57 #else
58         "tar -[xtvO] "
59 #endif
60 #if defined BB_FEATURE_TAR_EXCLUDE
61         "[--exclude File] "
62 #endif
63         "[-f tarFile] [FILE] ...\n"
64 #ifndef BB_FEATURE_TRIVIAL_HELP
65         "\nCreate, extract, or list files from a tar file.  Note that\n"
66         "this version of tar treats hard links as separate files.\n\n"
67         "Main operation mode:\n"
68 #ifdef BB_FEATURE_TAR_CREATE
69         "\tc\t\tcreate\n"
70 #endif
71         "\tx\t\textract\n"
72         "\tt\t\tlist\n"
73         "\nFile selection:\n"
74         "\tf\t\tname of tarfile or \"-\" for stdin\n"
75         "\tO\t\textract to stdout\n"
76 #if defined BB_FEATURE_TAR_EXCLUDE
77         "\t--exclude\tfile to exclude\n"
78 #endif
79         "\nInformative output:\n"
80         "\tv\t\tverbosely list files processed\n"
81 #endif
82         ;
83
84 /* Tar file constants  */
85 #ifndef MAJOR
86 #define MAJOR(dev) (((dev)>>8)&0xff)
87 #define MINOR(dev) ((dev)&0xff)
88 #endif
89
90
91 /* POSIX tar Header Block, from POSIX 1003.1-1990  */
92 struct TarHeader
93 {
94                                 /* byte offset */
95         char name[100];               /*   0-99 */
96         char mode[8];                 /* 100-107 */
97         char uid[8];                  /* 108-115 */
98         char gid[8];                  /* 116-123 */
99         char size[12];                /* 124-135 */
100         char mtime[12];               /* 136-147 */
101         char chksum[8];               /* 148-155 */
102         char typeflag;                /* 156-156 */
103         char linkname[100];           /* 157-256 */
104         char magic[6];                /* 257-262 */
105         char version[2];              /* 263-264 */
106         char uname[32];               /* 265-296 */
107         char gname[32];               /* 297-328 */
108         char devmajor[8];             /* 329-336 */
109         char devminor[8];             /* 337-344 */
110         char prefix[155];             /* 345-499 */
111         char padding[12];             /* 500-512 (pad to exactly the TAR_BLOCK_SIZE) */
112 };
113 typedef struct TarHeader TarHeader;
114
115
116 /* A few useful constants */
117 #define TAR_MAGIC          "ustar"        /* ustar and a null */
118 #define TAR_VERSION        "  "           /* Be compatable with GNU tar format */
119 #define TAR_MAGIC_LEN       6
120 #define TAR_VERSION_LEN     2
121 #define TAR_BLOCK_SIZE      512
122
123 /* A nice enum with all the possible tar file content types */
124 enum TarFileType 
125 {
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 };
136 typedef enum TarFileType TarFileType;
137
138 /* This struct ignores magic, non-numeric user name, 
139  * non-numeric group name, and the checksum, since
140  * these are all ignored by BusyBox tar. */ 
141 struct TarInfo
142 {
143         int              tarFd;          /* An open file descriptor for reading from the tarball */
144         char *           name;           /* File name */
145         mode_t           mode;           /* Unix mode, including device bits. */
146         uid_t            uid;            /* Numeric UID */
147         gid_t            gid;            /* Numeric GID */
148         size_t           size;           /* Size of file */
149         time_t           mtime;          /* Last-modified time */
150         enum TarFileType type;           /* Regular, directory, link, etc */
151         char *           linkname;       /* Name for symbolic and hard links */
152         long             devmajor;       /* Major number for special device */
153         long             devminor;       /* Minor number for special device */
154 };
155 typedef struct TarInfo TarInfo;
156
157 /* Local procedures to restore files from a tar file.  */
158 static int readTarFile(const char* tarName, int extractFlag, int listFlag, 
159                 int tostdoutFlag, int verboseFlag, char** excludeList);
160
161
162
163 #ifdef BB_FEATURE_TAR_CREATE
164 /* Local procedures to save files into a tar file.  */
165 static int writeTarFile(const char* tarName, int tostdoutFlag, 
166                 int verboseFlag, int argc, char **argv, char** excludeList);
167 #endif
168
169
170 extern int tar_main(int argc, char **argv)
171 {
172         char** excludeList=NULL;
173 #if defined BB_FEATURE_TAR_EXCLUDE
174         int excludeListSize=0;
175 #endif
176         const char *tarName="-";
177         int listFlag     = FALSE;
178         int extractFlag  = FALSE;
179         int createFlag   = FALSE;
180         int verboseFlag  = FALSE;
181         int tostdoutFlag = FALSE;
182         int stopIt;
183
184         if (argc <= 1)
185                 usage(tar_usage);
186
187         /* Parse any options */
188         while (--argc > 0 && strspn(*(++argv), "-cxt") >0 ) {
189                 stopIt=FALSE;
190                 while (stopIt==FALSE && *argv && **argv) {
191                         switch (**argv) {
192                                 case 'f':
193                                         if (--argc == 0) {
194                                                 fatalError( "Option requires an argument: No file specified\n");
195                                         }
196                                         if (*tarName != '-')
197                                                 fatalError( "Only one 'f' option allowed\n");
198                                         tarName = *(++argv);
199                                         if (tarName == NULL)
200                                                 fatalError( "Option requires an argument: No file specified\n");
201                                         if (!strcmp(tarName, "-"))
202                                                 tostdoutFlag = TRUE;
203                                         stopIt=TRUE;
204                                         break;
205
206                                 case 't':
207                                         if (extractFlag == TRUE || createFlag == TRUE)
208                                                 goto flagError;
209                                         listFlag = TRUE;
210                                         break;
211
212                                 case 'x':
213                                         if (listFlag == TRUE || createFlag == TRUE)
214                                                 goto flagError;
215                                         extractFlag = TRUE;
216                                         break;
217                                 case 'c':
218                                         if (extractFlag == TRUE || listFlag == TRUE)
219                                                 goto flagError;
220                                         createFlag = TRUE;
221                                         break;
222
223                                 case 'v':
224                                         verboseFlag = TRUE;
225                                         break;
226
227                                 case 'O':
228                                         tostdoutFlag = TRUE;
229                                         tarName = "-";
230                                         break;
231                                 case '-':
232 #if defined BB_FEATURE_TAR_EXCLUDE
233                                         if (strcmp(*argv, "-exclude")==0) {
234                                                 if (--argc == 0) {
235                                                         fatalError( "Option requires an argument: No file specified\n");
236                                                 }
237                                                 excludeList=realloc( excludeList, sizeof(char**) * (excludeListSize+2));
238                                                 excludeList[excludeListSize] = *(++argv);
239                                                 /* Remove leading "/"s */
240                                                 if (*excludeList[excludeListSize] =='/') {
241                                                         excludeList[excludeListSize] = (excludeList[excludeListSize])+1;
242                                                 }
243                                                 if (excludeList[excludeListSize++] == NULL)
244                                                         fatalError( "Option requires an argument: No file specified\n");
245                                                 /* Tack a NULL onto the end of the list */
246                                                 excludeList[excludeListSize] = NULL;
247                                                 stopIt=TRUE;
248                                                 break;
249                                         }
250 #endif
251                                         if (strcmp(*argv, "-help")==0) {
252                                                 usage(tar_usage);
253                                         }
254                                         break;
255
256                                 default:
257                                         fatalError( "Unknown tar flag '%c'\n" 
258                                                         "Try `tar --help' for more information\n", **argv);
259                         }
260                         ++(*argv);
261                 }
262         }
263
264         /* 
265          * Do the correct type of action supplying the rest of the
266          * command line arguments as the list of files to process.
267          */
268         if (createFlag == TRUE) {
269 #ifndef BB_FEATURE_TAR_CREATE
270                 fatalError( "This version of tar was not compiled with tar creation support.\n");
271 #else
272                 exit(writeTarFile(tarName, tostdoutFlag, verboseFlag, argc, argv, excludeList));
273 #endif
274         }
275         if (listFlag == TRUE || extractFlag == TRUE) {
276                 exit(readTarFile(tarName, extractFlag, listFlag, tostdoutFlag, verboseFlag, excludeList));
277         }
278
279   flagError:
280         fatalError( "Exactly one of 'c', 'x' or 't' must be specified\n");
281 }
282                                         
283 static void
284 fixUpPermissions(TarInfo *header)
285 {
286         struct utimbuf t;
287         /* Now set permissions etc for the new file */
288         chown(header->name, header->uid, header->gid);
289         chmod(header->name, header->mode);
290         /* Reset the time */
291         t.actime = time(0);
292         t.modtime = header->mtime;
293         utime(header->name, &t);
294 }
295                                 
296 static int
297 tarExtractRegularFile(TarInfo *header, int extractFlag, int tostdoutFlag)
298 {
299         size_t  writeSize;
300         size_t  readSize;
301         size_t  actualWriteSz;
302         char    buffer[BUFSIZ];
303         size_t  size = header->size;
304         int outFd=fileno(stdout);
305
306         /* Open the file to be written, if a file is supposed to be written */
307         if (extractFlag==TRUE && tostdoutFlag==FALSE) {
308                 /* Create the path to the file, just in case it isn't there...
309                  * This should not screw up path permissions or anything. */
310                 createPath(header->name, 0777);
311                 if ((outFd=open(header->name, O_CREAT|O_TRUNC|O_WRONLY, 
312                                                 header->mode & ~S_IFMT)) < 0) {
313                         errorMsg(io_error, header->name, strerror(errno)); 
314                         return( FALSE);
315                 }
316         }
317
318         /* Write out the file, if we are supposed to be doing that */
319         while ( size > 0 ) {
320                 actualWriteSz=0;
321                 if ( size > sizeof(buffer) )
322                         writeSize = readSize = sizeof(buffer);
323                 else {
324                         int mod = size % 512;
325                         if ( mod != 0 )
326                                 readSize = size + (512 - mod);
327                         else
328                                 readSize = size;
329                         writeSize = size;
330                 }
331                 if ( (readSize = fullRead(header->tarFd, buffer, readSize)) <= 0 ) {
332                         /* Tarball seems to have a problem */
333                         errorMsg("Unexpected EOF in archive\n"); 
334                         return( FALSE);
335                 }
336                 if ( readSize < writeSize )
337                         writeSize = readSize;
338
339                 /* Write out the file, if we are supposed to be doing that */
340                 if (extractFlag==TRUE) {
341
342                         if ((actualWriteSz=fullWrite(outFd, buffer, writeSize)) != writeSize ) {
343                                 /* Output file seems to have a problem */
344                                 errorMsg(io_error, header->name, strerror(errno)); 
345                                 return( FALSE);
346                         }
347                 } else {
348                         actualWriteSz=writeSize;
349                 }
350
351                 size -= actualWriteSz;
352         }
353
354         /* Now we are done writing the file out, so try 
355          * and fix up the permissions and whatnot */
356         if (extractFlag==TRUE && tostdoutFlag==FALSE) {
357                 close(outFd);
358                 fixUpPermissions(header);
359         }
360         return( TRUE);
361 }
362
363 static int
364 tarExtractDirectory(TarInfo *header, int extractFlag, int tostdoutFlag)
365 {
366
367         if (extractFlag==FALSE || tostdoutFlag==TRUE)
368                 return( TRUE);
369
370         if (createPath(header->name, header->mode) != TRUE) {
371                 errorMsg("%s: Cannot mkdir: %s\n", 
372                                 header->name, strerror(errno)); 
373                 return( FALSE);
374         }
375         /* make the final component, just in case it was
376          * omitted by createPath() (which will skip the
377          * directory if it doesn't have a terminating '/') */
378         if (mkdir(header->name, header->mode) == 0) {
379                 fixUpPermissions(header);
380         }
381         return( TRUE);
382 }
383
384 static int
385 tarExtractHardLink(TarInfo *header, int extractFlag, int tostdoutFlag)
386 {
387         if (extractFlag==FALSE || tostdoutFlag==TRUE)
388                 return( TRUE);
389
390         if (link(header->linkname, header->name) < 0) {
391                 errorMsg("%s: Cannot create hard link to '%s': %s\n", 
392                                 header->name, header->linkname, strerror(errno)); 
393                 return( FALSE);
394         }
395
396         /* Now set permissions etc for the new directory */
397         fixUpPermissions(header);
398         return( TRUE);
399 }
400
401 static int
402 tarExtractSymLink(TarInfo *header, int extractFlag, int tostdoutFlag)
403 {
404         if (extractFlag==FALSE || tostdoutFlag==TRUE)
405                 return( TRUE);
406
407 #ifdef  S_ISLNK
408         if (symlink(header->linkname, header->name) < 0) {
409                 errorMsg("%s: Cannot create symlink to '%s': %s\n", 
410                                 header->name, header->linkname, strerror(errno)); 
411                 return( FALSE);
412         }
413         /* Try to change ownership of the symlink.
414          * If libs doesn't support that, don't bother.
415          * Changing the pointed-to-file is the Wrong Thing(tm).
416          */
417 #if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1)
418         lchown(header->name, header->uid, header->gid);
419 #endif
420
421         /* Do not change permissions or date on symlink,
422          * since it changes the pointed to file instead.  duh. */
423 #else
424         errorMsg("%s: Cannot create symlink to '%s': %s\n", 
425                         header->name, header->linkname, 
426                         "symlinks not supported"); 
427 #endif
428         return( TRUE);
429 }
430
431 static int
432 tarExtractSpecial(TarInfo *header, int extractFlag, int tostdoutFlag)
433 {
434         if (extractFlag==FALSE || tostdoutFlag==TRUE)
435                 return( TRUE);
436
437         if (S_ISCHR(header->mode) || S_ISBLK(header->mode) || S_ISSOCK(header->mode)) {
438                 if (mknod(header->name, header->mode, makedev(header->devmajor, header->devminor)) < 0) {
439                         errorMsg("%s: Cannot mknod: %s\n",
440                                 header->name, strerror(errno)); 
441                         return( FALSE);
442                 }
443         } else if (S_ISFIFO(header->mode)) {
444                 if (mkfifo(header->name, header->mode) < 0) {
445                         errorMsg("%s: Cannot mkfifo: %s\n",
446                                 header->name, strerror(errno)); 
447                         return( FALSE);
448                 }
449         }
450
451         /* Now set permissions etc for the new directory */
452         fixUpPermissions(header);
453         return( TRUE);
454 }
455
456 /* Read an octal value in a field of the specified width, with optional
457  * spaces on both sides of the number and with an optional null character
458  * at the end.  Returns -1 on an illegal format.  */
459 static long getOctal(const char *cp, int size)
460 {
461         long val = 0;
462
463         for(;(size > 0) && (*cp == ' '); cp++, size--);
464         if ((size == 0) || !isOctal(*cp))
465                 return -1;
466         for(; (size > 0) && isOctal(*cp); size--) {
467                 val = val * 8 + *cp++ - '0';
468         }
469         for (;(size > 0) && (*cp == ' '); cp++, size--);
470         if ((size > 0) && *cp)
471                 return -1;
472         return val;
473 }
474
475
476 /* Parse the tar header and fill in the nice struct with the details */
477 static int
478 readTarHeader(struct TarHeader *rawHeader, struct TarInfo *header)
479 {
480         int i;
481         long chksum, sum=0;
482         unsigned char *s = (unsigned char *)rawHeader;
483
484         header->name  = rawHeader->name;
485         /* Check for and relativify any absolute paths */
486         if ( *(header->name) == '/' ) {
487                 static int alreadyWarned=FALSE;
488
489                 while (*(header->name) == '/')
490                         ++*(header->name);
491
492                 if (alreadyWarned == FALSE) {
493                         errorMsg("Removing leading '/' from member names\n");
494                         alreadyWarned = TRUE;
495                 }
496         }
497
498         header->mode  = getOctal(rawHeader->mode, sizeof(rawHeader->mode));
499         header->uid   =  getOctal(rawHeader->uid, sizeof(rawHeader->uid));
500         header->gid   =  getOctal(rawHeader->gid, sizeof(rawHeader->gid));
501         header->size  = getOctal(rawHeader->size, sizeof(rawHeader->size));
502         header->mtime = getOctal(rawHeader->mtime, sizeof(rawHeader->mtime));
503         chksum = getOctal(rawHeader->chksum, sizeof(rawHeader->chksum));
504         header->type  = rawHeader->typeflag;
505         header->linkname  = rawHeader->linkname;
506         header->devmajor  = getOctal(rawHeader->devmajor, sizeof(rawHeader->devmajor));
507         header->devminor  = getOctal(rawHeader->devminor, sizeof(rawHeader->devminor));
508
509         /* Check the checksum */
510         for (i = sizeof(*rawHeader); i-- != 0;) {
511                 sum += *s++;
512         }
513         /* Remove the effects of the checksum field (replace 
514          * with blanks for the purposes of the checksum) */
515         s = rawHeader->chksum;
516         for (i = sizeof(rawHeader->chksum) ; i-- != 0;) {
517                 sum -= *s++;
518         }
519         sum += ' ' * sizeof(rawHeader->chksum);
520         if (sum == chksum )
521                 return ( TRUE);
522         return( FALSE);
523 }
524
525
526 /*
527  * Read a tar file and extract or list the specified files within it.
528  * If the list is empty than all files are extracted or listed.
529  */
530 static int readTarFile(const char* tarName, int extractFlag, int listFlag, 
531                 int tostdoutFlag, int verboseFlag, char** excludeList)
532 {
533         int status, tarFd=-1;
534         int errorFlag=FALSE;
535         TarHeader rawHeader;
536         TarInfo header;
537 #if defined BB_FEATURE_TAR_EXCLUDE
538         char** tmpList;
539 #endif
540
541         /* Open the tar file for reading.  */
542         if (!strcmp(tarName, "-"))
543                 tarFd = fileno(stdin);
544         else
545                 tarFd = open(tarName, O_RDONLY);
546         if (tarFd < 0) {
547                 errorMsg( "Error opening '%s': %s\n", tarName, strerror(errno));
548                 return ( FALSE);
549         }
550
551         /* Set the umask for this process so it doesn't 
552          * screw up permission setting for us later. */
553         umask(0);
554
555         /* Read the tar file, and iterate over it one file at a time */
556         while ( (status = fullRead(tarFd, (char*)&rawHeader, TAR_BLOCK_SIZE)) == TAR_BLOCK_SIZE ) {
557
558                 /* First, try to read the header */
559                 if ( readTarHeader(&rawHeader, &header) == FALSE ) {
560                         if ( *(header.name) == '\0' ) {
561                                 goto endgame;
562                         } else {
563                                 errorFlag=TRUE;
564                                 errorMsg("Bad tar header, skipping\n");
565                                 continue;
566                         }
567                 }
568                 if ( *(header.name) == '\0' )
569                                 goto endgame;
570                 header.tarFd = tarFd;
571
572 #if defined BB_FEATURE_TAR_EXCLUDE
573                 {
574                         int skipFlag=FALSE;
575                         /* Check for excluded files....  */
576                         for (tmpList=excludeList; tmpList && *tmpList; tmpList++) {
577                                 /* Do some extra hoop jumping for when directory names
578                                  * end in '/' but the entry in tmpList doesn't */
579                                 if (strncmp( *tmpList, header.name, strlen(*tmpList))==0 || (
580                                                         header.name[strlen(header.name)-1]=='/'
581                                                         && strncmp( *tmpList, header.name, 
582                                                                 MIN(strlen(header.name)-1, strlen(*tmpList)))==0)) {
583                                         /* If it is a regular file, pretend to extract it with
584                                          * the extractFlag set to FALSE, so the junk in the tarball
585                                          * is properly skipped over */
586                                         if ( header.type==REGTYPE || header.type==REGTYPE0 ) {
587                                                         tarExtractRegularFile(&header, FALSE, FALSE);
588                                         }
589                                         skipFlag=TRUE;
590                                         break;
591                                 }
592                         }
593                         /* There are not the droids you're looking for, move along */
594                         if (skipFlag==TRUE)
595                                 continue;
596                 }
597 #endif
598                 /* Special treatment if the list (-t) flag is on */
599                 if (verboseFlag == TRUE && extractFlag == FALSE) {
600                         int len, len1;
601                         char buf[35];
602                         struct tm *tm = localtime (&(header.mtime));
603
604                         len=printf("%s ", modeString(header.mode));
605                         memset(buf, 0, 8*sizeof(char));
606                         my_getpwuid(buf, header.uid);
607                         if (! *buf)
608                                 len+=printf("%d", header.uid);
609                         else
610                                 len+=printf("%s", buf);
611                         memset(buf, 0, 8*sizeof(char));
612                         my_getgrgid(buf, header.gid);
613                         if (! *buf)
614                                 len+=printf("/%-d ", header.gid);
615                         else
616                                 len+=printf("/%-s ", buf);
617
618                         if (header.type==CHRTYPE || header.type==BLKTYPE) {
619                                 len1=snprintf(buf, sizeof(buf), "%ld,%-ld ", 
620                                                 header.devmajor, header.devminor);
621                         } else {
622                                 len1=snprintf(buf, sizeof(buf), "%lu ", (long)header.size);
623                         }
624                         /* Jump through some hoops to make the columns match up */
625                         for(;(len+len1)<31;len++)
626                                 printf(" ");
627                         printf(buf);
628
629                         /* Use ISO 8610 time format */
630                         if (tm) { 
631                                 printf ("%04d-%02d-%02d %02d:%02d:%02d ", 
632                                                 tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday, 
633                                                 tm->tm_hour, tm->tm_min, tm->tm_sec);
634                         }
635                 }
636                 /* List contents if we are supposed to do that */
637                 if (verboseFlag == TRUE || listFlag == TRUE) {
638                         /* Now the normal listing */
639                         printf("%s", header.name);
640                 }
641                 if (verboseFlag == TRUE && listFlag == TRUE) {
642                         /* If this is a link, say so */
643                         if (header.type==LNKTYPE)
644                                 printf(" link to %s", header.linkname);
645                         else if (header.type==SYMTYPE)
646                                 printf(" -> %s", header.linkname);
647                 }
648                 if (verboseFlag == TRUE || listFlag == TRUE) {
649                         printf("\n");
650                 }
651
652                 /* Remove any clutter lying in our way */
653                 if (extractFlag == TRUE)        /* .. but only if we are extracting (as */
654                         unlink( header.name);   /* opposed to listing) (rob@sysgo.de)   */
655
656                 /* If we got here, we can be certain we have a legitimate 
657                  * header to work with.  So work with it.  */
658                 switch ( header.type ) {
659                         case REGTYPE:
660                         case REGTYPE0:
661                                 /* If the name ends in a '/' then assume it is
662                                  * supposed to be a directory, and fall through */
663                                 if (header.name[strlen(header.name)-1] != '/') {
664                                         if (tarExtractRegularFile(&header, extractFlag, tostdoutFlag)==FALSE)
665                                                 errorFlag=TRUE;
666                                         break;
667                                 }
668                         case DIRTYPE:
669                                 if (tarExtractDirectory( &header, extractFlag, tostdoutFlag)==FALSE)
670                                         errorFlag=TRUE;
671                                 break;
672                         case LNKTYPE:
673                                 if (tarExtractHardLink( &header, extractFlag, tostdoutFlag)==FALSE)
674                                         errorFlag=TRUE;
675                                 break;
676                         case SYMTYPE:
677                                 if (tarExtractSymLink( &header, extractFlag, tostdoutFlag)==FALSE)
678                                         errorFlag=TRUE;
679                                 break;
680                         case CHRTYPE:
681                         case BLKTYPE:
682                         case FIFOTYPE:
683                                 if (tarExtractSpecial( &header, extractFlag, tostdoutFlag)==FALSE)
684                                         errorFlag=TRUE;
685                                 break;
686                         default:
687                                 close( tarFd);
688                                 return( FALSE);
689                 }
690         }
691         close(tarFd);
692         if (status > 0) {
693                 /* Bummer - we read a partial header */
694                 errorMsg( "Error reading '%s': %s\n", tarName, strerror(errno));
695                 return ( FALSE);
696         }
697         else if (errorFlag==TRUE) {
698                 errorMsg( "Error exit delayed from previous errors\n");
699                 return( FALSE);
700         } else 
701                 return( status);
702
703         /* Stuff to do when we are done */
704 endgame:
705         close( tarFd);
706         if ( *(header.name) == '\0' ) {
707                 if (errorFlag==TRUE)
708                         errorMsg( "Error exit delayed from previous errors\n");
709                 else
710                         return( TRUE);
711         } 
712         return( FALSE);
713 }
714
715
716 #ifdef BB_FEATURE_TAR_CREATE
717
718 /* Some info to be carried along when creating a new tarball */
719 struct TarBallInfo
720 {
721         char* fileName;               /* File name of the tarball */
722         int tarFd;                    /* Open-for-write file descriptor
723                                                                          for the tarball */
724         struct stat statBuf;          /* Stat info for the tarball, letting
725                                                                          us know the inode and device that the
726                                                                          tarball lives, so we can avoid trying 
727                                                                          to include the tarball into itself */
728         int verboseFlag;              /* Whether to print extra stuff or not */
729         char** excludeList;           /* List of files to not include */
730 };
731 typedef struct TarBallInfo TarBallInfo;
732
733
734 /* Put an octal string into the specified buffer.
735  * The number is zero and space padded and possibly null padded.
736  * Returns TRUE if successful.  */ 
737 static int putOctal (char *cp, int len, long value)
738 {
739         int tempLength;
740         char tempBuffer[32];
741         char *tempString = tempBuffer;
742
743         /* Create a string of the specified length with an initial space,
744          * leading zeroes and the octal number, and a trailing null.  */
745         sprintf (tempString, "%0*lo", len - 1, value);
746
747         /* If the string is too large, suppress the leading space.  */
748         tempLength = strlen (tempString) + 1;
749         if (tempLength > len) {
750                 tempLength--;
751                 tempString++;
752         }
753
754         /* If the string is still too large, suppress the trailing null.  */
755         if (tempLength > len)
756                 tempLength--;
757
758         /* If the string is still too large, fail.  */
759         if (tempLength > len)
760                 return FALSE;
761
762         /* Copy the string to the field.  */
763         memcpy (cp, tempString, len);
764
765         return TRUE;
766 }
767
768 /* Write out a tar header for the specified file/directory/whatever */
769 static int
770 writeTarHeader(struct TarBallInfo *tbInfo, const char *fileName, struct stat *statbuf)
771 {
772         long chksum=0;
773         struct TarHeader header;
774 #if defined BB_FEATURE_TAR_EXCLUDE
775         char** tmpList;
776 #endif
777         const unsigned char *cp = (const unsigned char *) &header;
778         ssize_t size = sizeof(struct TarHeader);
779
780         memset( &header, 0, size);
781
782         if (*fileName=='/') {
783                 static int alreadyWarned=FALSE;
784                 if (alreadyWarned==FALSE) {
785                         errorMsg("Removing leading '/' from member names\n");
786                         alreadyWarned=TRUE;
787                 }
788                 strncpy(header.name, fileName+1, sizeof(header.name)); 
789         }
790         else {
791                 strncpy(header.name, fileName, sizeof(header.name)); 
792         }
793
794 #if defined BB_FEATURE_TAR_EXCLUDE
795         /* Check for excluded files....  */
796         for (tmpList=tbInfo->excludeList; tmpList && *tmpList; tmpList++) {
797                 /* Do some extra hoop jumping for when directory names
798                  * end in '/' but the entry in tmpList doesn't */
799                 if (strncmp( *tmpList, header.name, strlen(*tmpList))==0 || (
800                                         header.name[strlen(header.name)-1]=='/'
801                                         && strncmp( *tmpList, header.name, 
802                                                 MIN(strlen(header.name)-1, strlen(*tmpList)))==0)) {
803                         /* Set the mode to something that is not a regular file, thereby
804                          * faking out writeTarFile into thinking that nothing further need
805                          * be done for this file.  Yes, I know this is ugly, but it works. */
806                         statbuf->st_mode = 0;
807                         return( TRUE);
808                 }
809         }
810 #endif
811
812         putOctal(header.mode, sizeof(header.mode), statbuf->st_mode);
813         putOctal(header.uid, sizeof(header.uid), statbuf->st_uid);
814         putOctal(header.gid, sizeof(header.gid), statbuf->st_gid);
815         putOctal(header.size, sizeof(header.size), 0); /* Regular file size is handled later */
816         putOctal(header.mtime, sizeof(header.mtime), statbuf->st_mtime);
817         strncpy(header.magic, TAR_MAGIC TAR_VERSION, 
818                         TAR_MAGIC_LEN + TAR_VERSION_LEN );
819
820         /* Enter the user and group names (default to root if it fails) */
821         my_getpwuid(header.uname, statbuf->st_uid);
822         if (! *header.uname)
823                 strcpy(header.uname, "root");
824         my_getgrgid(header.gname, statbuf->st_gid);
825         if (! *header.uname)
826                 strcpy(header.uname, "root");
827
828         /* WARNING/NOTICE: I break Hard Links */
829         if (S_ISLNK(statbuf->st_mode)) {
830                 int link_size=0;
831                 char buffer[BUFSIZ];
832                 header.typeflag  = SYMTYPE;
833                 link_size = readlink(fileName, buffer, sizeof(buffer) - 1);
834                 if ( link_size < 0) {
835                         errorMsg("Error reading symlink '%s': %s\n", header.name, strerror(errno));
836                         return ( FALSE);
837                 }
838                 buffer[link_size] = '\0';
839                 strncpy(header.linkname, buffer, sizeof(header.linkname)); 
840         } else if (S_ISDIR(statbuf->st_mode)) {
841                 header.typeflag  = DIRTYPE;
842                 strncat(header.name, "/", sizeof(header.name)); 
843         } else if (S_ISCHR(statbuf->st_mode)) {
844                 header.typeflag  = CHRTYPE;
845                 putOctal(header.devmajor, sizeof(header.devmajor), MAJOR(statbuf->st_rdev));
846                 putOctal(header.devminor, sizeof(header.devminor), MINOR(statbuf->st_rdev));
847         } else if (S_ISBLK(statbuf->st_mode)) {
848                 header.typeflag  = BLKTYPE;
849                 putOctal(header.devmajor, sizeof(header.devmajor), MAJOR(statbuf->st_rdev));
850                 putOctal(header.devminor, sizeof(header.devminor), MINOR(statbuf->st_rdev));
851         } else if (S_ISFIFO(statbuf->st_mode)) {
852                 header.typeflag  = FIFOTYPE;
853         } else if (S_ISREG(statbuf->st_mode)) {
854                 header.typeflag  = REGTYPE;
855                 putOctal(header.size, sizeof(header.size), statbuf->st_size);
856         } else {
857                 errorMsg("%s: Unknown file type\n", fileName);
858                 return ( FALSE);
859         }
860
861         /* Calculate and store the checksum (i.e. the sum of all of the bytes of
862          * the header).  The checksum field must be filled with blanks for the
863          * calculation.  The checksum field is formatted differently from the
864          * other fields: it has [6] digits, a null, then a space -- rather than
865          * digits, followed by a null like the other fields... */
866         memset(header.chksum, ' ', sizeof(header.chksum));
867         cp = (const unsigned char *) &header;
868         while (size-- > 0)
869                 chksum += *cp++;
870         putOctal(header.chksum, 7, chksum);
871         
872         /* Now write the header out to disk */
873         if ((size=fullWrite(tbInfo->tarFd, (char*)&header, sizeof(struct TarHeader))) < 0) {
874                 errorMsg(io_error, fileName, strerror(errno)); 
875                 return ( FALSE);
876         }
877         /* Pad the header up to the tar block size */
878         for (; size<TAR_BLOCK_SIZE; size++) {
879                 write(tbInfo->tarFd, "\0", 1);
880         }
881         /* Now do the verbose thing (or not) */
882         if (tbInfo->verboseFlag==TRUE)
883                 fprintf(stdout, "%s\n", header.name);
884
885         return ( TRUE);
886 }
887
888
889 static int writeFileToTarball(const char *fileName, struct stat *statbuf, void* userData)
890 {
891         struct TarBallInfo *tbInfo = (struct TarBallInfo *)userData;
892
893         /* It is against the rules to archive a socket */
894         if (S_ISSOCK(statbuf->st_mode)) {
895                 errorMsg("%s: socket ignored\n", fileName);
896                 return( TRUE);
897         }
898
899         /* It is a bad idea to store the archive we are in the process of creating,
900          * so check the device and inode to be sure that this particular file isn't
901          * the new tarball */
902         if (tbInfo->statBuf.st_dev == statbuf->st_dev &&
903                         tbInfo->statBuf.st_ino == statbuf->st_ino) {
904                 errorMsg("%s: file is the archive; skipping\n", fileName);
905                 return( TRUE);
906         }
907
908         if (writeTarHeader(tbInfo, fileName, statbuf)==FALSE) {
909                 return( FALSE);
910         } 
911
912         /* Now, if the file is a regular file, copy it out to the tarball */
913         if (S_ISREG(statbuf->st_mode)) {
914                 int  inputFileFd;
915                 char buffer[BUFSIZ];
916                 ssize_t size=0, readSize=0;
917
918                 /* open the file we want to archive, and make sure all is well */
919                 if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
920                         errorMsg("%s: Cannot open: %s\n", fileName, strerror(errno));
921                         return( FALSE);
922                 }
923                 
924                 /* write the file to the archive */
925                 while ( (size = fullRead(inputFileFd, buffer, sizeof(buffer))) > 0 ) {
926                         if (fullWrite(tbInfo->tarFd, buffer, size) != size ) {
927                                 /* Output file seems to have a problem */
928                                 errorMsg(io_error, fileName, strerror(errno)); 
929                                 return( FALSE);
930                         }
931                         readSize+=size;
932                 }
933                 if (size == -1) {
934                         errorMsg(io_error, fileName, strerror(errno)); 
935                         return( FALSE);
936                 }
937                 /* Pad the file up to the tar block size */
938                 for (; (readSize%TAR_BLOCK_SIZE) != 0; readSize++) {
939                         write(tbInfo->tarFd, "\0", 1);
940                 }
941                 close( inputFileFd);
942         }
943
944         return( TRUE);
945 }
946
947 static int writeTarFile(const char* tarName, int tostdoutFlag, 
948                 int verboseFlag, int argc, char **argv, char** excludeList)
949 {
950         int tarFd=-1;
951         int errorFlag=FALSE;
952         ssize_t size;
953         struct TarBallInfo tbInfo;
954         tbInfo.verboseFlag = verboseFlag;
955
956         /* Make sure there is at least one file to tar up.  */
957         if (argc <= 0)
958                 fatalError("Cowardly refusing to create an empty archive\n");
959
960         /* Open the tar file for writing.  */
961         if (tostdoutFlag == TRUE)
962                 tbInfo.tarFd = fileno(stdout);
963         else
964                 tbInfo.tarFd = open (tarName, O_WRONLY | O_CREAT | O_TRUNC, 0644);
965         if (tbInfo.tarFd < 0) {
966                 errorMsg( "Error opening '%s': %s\n", tarName, strerror(errno));
967                 return ( FALSE);
968         }
969         tbInfo.excludeList=excludeList;
970         /* Store the stat info for the tarball's file, so
971          * can avoid including the tarball into itself....  */
972         if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0)
973                 fatalError(io_error, tarName, strerror(errno)); 
974
975         /* Set the umask for this process so it doesn't 
976          * screw up permission setting for us later. */
977         umask(0);
978
979         /* Read the directory/files and iterate over them one at a time */
980         while (argc-- > 0) {
981                 if (recursiveAction(*argv++, TRUE, FALSE, FALSE,
982                                         writeFileToTarball, writeFileToTarball, 
983                                         (void*) &tbInfo) == FALSE) {
984                         errorFlag = TRUE;
985                 }
986         }
987         /* Write two empty blocks to the end of the archive */
988         for (size=0; size<(2*TAR_BLOCK_SIZE); size++) {
989                 write(tbInfo.tarFd, "\0", 1);
990         }
991
992         /* To be pedantically correct, we would check if the tarball
993          * is smaller then 20 tar blocks, and pad it if it was smaller,
994          * but that isn't necessary for GNU tar interoperability, and
995          * so is considered a waste of space */
996
997         /* Hang up the tools, close up shop, head home */
998         close(tarFd);
999         if (errorFlag == TRUE) {
1000                 errorMsg("Error exit delayed from previous errors\n");
1001                 return(FALSE);
1002         }
1003         return( TRUE);
1004 }
1005
1006
1007 #endif
1008