Annotation of /trunk/mkinitrd-magellan/busybox/archival/tar.c
Parent Directory | Revision Log
Revision 532 -
(hide annotations)
(download)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 28905 byte(s)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 28905 byte(s)
-import if magellan mkinitrd; it is a fork of redhats mkinitrd-5.0.8 with all magellan patches and features; deprecates magellan-src/mkinitrd
1 | niro | 532 | /* vi: set sw=4 ts=4: */ |
2 | /* | ||
3 | * Mini tar implementation for busybox | ||
4 | * | ||
5 | * Modified to use common extraction code used by ar, cpio, dpkg-deb, dpkg | ||
6 | * Glenn McGrath <bug1@iinet.net.au> | ||
7 | * | ||
8 | * Note, that as of BusyBox-0.43, tar has been completely rewritten from the | ||
9 | * ground up. It still has remnants 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-2004 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 | * | ||
23 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | ||
24 | */ | ||
25 | |||
26 | #include <fnmatch.h> | ||
27 | #include <getopt.h> | ||
28 | #include "busybox.h" | ||
29 | #include "unarchive.h" | ||
30 | |||
31 | #if ENABLE_FEATURE_TAR_CREATE | ||
32 | |||
33 | /* Tar file constants */ | ||
34 | |||
35 | #define TAR_BLOCK_SIZE 512 | ||
36 | |||
37 | /* POSIX tar Header Block, from POSIX 1003.1-1990 */ | ||
38 | #define NAME_SIZE 100 | ||
39 | #define NAME_SIZE_STR "100" | ||
40 | typedef struct TarHeader TarHeader; | ||
41 | struct TarHeader { /* byte offset */ | ||
42 | char name[NAME_SIZE]; /* 0-99 */ | ||
43 | char mode[8]; /* 100-107 */ | ||
44 | char uid[8]; /* 108-115 */ | ||
45 | char gid[8]; /* 116-123 */ | ||
46 | char size[12]; /* 124-135 */ | ||
47 | char mtime[12]; /* 136-147 */ | ||
48 | char chksum[8]; /* 148-155 */ | ||
49 | char typeflag; /* 156-156 */ | ||
50 | char linkname[NAME_SIZE]; /* 157-256 */ | ||
51 | char magic[6]; /* 257-262 */ | ||
52 | char version[2]; /* 263-264 */ | ||
53 | char uname[32]; /* 265-296 */ | ||
54 | char gname[32]; /* 297-328 */ | ||
55 | char devmajor[8]; /* 329-336 */ | ||
56 | char devminor[8]; /* 337-344 */ | ||
57 | char prefix[155]; /* 345-499 */ | ||
58 | char padding[12]; /* 500-512 (pad to exactly the TAR_BLOCK_SIZE) */ | ||
59 | }; | ||
60 | |||
61 | /* | ||
62 | ** writeTarFile(), writeFileToTarball(), and writeTarHeader() are | ||
63 | ** the only functions that deal with the HardLinkInfo structure. | ||
64 | ** Even these functions use the xxxHardLinkInfo() functions. | ||
65 | */ | ||
66 | typedef struct HardLinkInfo HardLinkInfo; | ||
67 | struct HardLinkInfo { | ||
68 | HardLinkInfo *next; /* Next entry in list */ | ||
69 | dev_t dev; /* Device number */ | ||
70 | ino_t ino; /* Inode number */ | ||
71 | short linkCount; /* (Hard) Link Count */ | ||
72 | char name[1]; /* Start of filename (must be last) */ | ||
73 | }; | ||
74 | |||
75 | /* Some info to be carried along when creating a new tarball */ | ||
76 | typedef struct TarBallInfo TarBallInfo; | ||
77 | struct TarBallInfo { | ||
78 | int tarFd; /* Open-for-write file descriptor | ||
79 | for the tarball */ | ||
80 | struct stat statBuf; /* Stat info for the tarball, letting | ||
81 | us know the inode and device that the | ||
82 | tarball lives, so we can avoid trying | ||
83 | to include the tarball into itself */ | ||
84 | int verboseFlag; /* Whether to print extra stuff or not */ | ||
85 | const llist_t *excludeList; /* List of files to not include */ | ||
86 | HardLinkInfo *hlInfoHead; /* Hard Link Tracking Information */ | ||
87 | HardLinkInfo *hlInfo; /* Hard Link Info for the current file */ | ||
88 | }; | ||
89 | |||
90 | /* A nice enum with all the possible tar file content types */ | ||
91 | enum TarFileType { | ||
92 | REGTYPE = '0', /* regular file */ | ||
93 | REGTYPE0 = '\0', /* regular file (ancient bug compat) */ | ||
94 | LNKTYPE = '1', /* hard link */ | ||
95 | SYMTYPE = '2', /* symbolic link */ | ||
96 | CHRTYPE = '3', /* character special */ | ||
97 | BLKTYPE = '4', /* block special */ | ||
98 | DIRTYPE = '5', /* directory */ | ||
99 | FIFOTYPE = '6', /* FIFO special */ | ||
100 | CONTTYPE = '7', /* reserved */ | ||
101 | GNULONGLINK = 'K', /* GNU long (>100 chars) link name */ | ||
102 | GNULONGNAME = 'L', /* GNU long (>100 chars) file name */ | ||
103 | }; | ||
104 | typedef enum TarFileType TarFileType; | ||
105 | |||
106 | /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */ | ||
107 | static void addHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr, | ||
108 | struct stat *statbuf, | ||
109 | const char *fileName) | ||
110 | { | ||
111 | /* Note: hlInfoHeadPtr can never be NULL! */ | ||
112 | HardLinkInfo *hlInfo; | ||
113 | |||
114 | hlInfo = xmalloc(sizeof(HardLinkInfo) + strlen(fileName)); | ||
115 | hlInfo->next = *hlInfoHeadPtr; | ||
116 | *hlInfoHeadPtr = hlInfo; | ||
117 | hlInfo->dev = statbuf->st_dev; | ||
118 | hlInfo->ino = statbuf->st_ino; | ||
119 | hlInfo->linkCount = statbuf->st_nlink; | ||
120 | strcpy(hlInfo->name, fileName); | ||
121 | } | ||
122 | |||
123 | static void freeHardLinkInfo(HardLinkInfo ** hlInfoHeadPtr) | ||
124 | { | ||
125 | HardLinkInfo *hlInfo; | ||
126 | HardLinkInfo *hlInfoNext; | ||
127 | |||
128 | if (hlInfoHeadPtr) { | ||
129 | hlInfo = *hlInfoHeadPtr; | ||
130 | while (hlInfo) { | ||
131 | hlInfoNext = hlInfo->next; | ||
132 | free(hlInfo); | ||
133 | hlInfo = hlInfoNext; | ||
134 | } | ||
135 | *hlInfoHeadPtr = NULL; | ||
136 | } | ||
137 | return; | ||
138 | } | ||
139 | |||
140 | /* Might be faster (and bigger) if the dev/ino were stored in numeric order;) */ | ||
141 | static HardLinkInfo *findHardLinkInfo(HardLinkInfo * hlInfo, struct stat *statbuf) | ||
142 | { | ||
143 | while (hlInfo) { | ||
144 | if ((statbuf->st_ino == hlInfo->ino) && (statbuf->st_dev == hlInfo->dev)) | ||
145 | break; | ||
146 | hlInfo = hlInfo->next; | ||
147 | } | ||
148 | return hlInfo; | ||
149 | } | ||
150 | |||
151 | /* Put an octal string into the specified buffer. | ||
152 | * The number is zero padded and possibly null terminated. | ||
153 | * Stores low-order bits only if whole value does not fit. */ | ||
154 | static void putOctal(char *cp, int len, off_t value) | ||
155 | { | ||
156 | char tempBuffer[sizeof(off_t)*3+1]; | ||
157 | char *tempString = tempBuffer; | ||
158 | int width; | ||
159 | |||
160 | width = sprintf(tempBuffer, "%0*"OFF_FMT"o", len, value); | ||
161 | tempString += (width - len); | ||
162 | |||
163 | /* If string has leading zeroes, we can drop one */ | ||
164 | /* and field will have trailing '\0' */ | ||
165 | /* (increases chances of compat with other tars) */ | ||
166 | if (tempString[0] == '0') | ||
167 | tempString++; | ||
168 | |||
169 | /* Copy the string to the field */ | ||
170 | memcpy(cp, tempString, len); | ||
171 | } | ||
172 | #define PUT_OCTAL(a, b) putOctal((a), sizeof(a), (b)) | ||
173 | |||
174 | static void chksum_and_xwrite(int fd, struct TarHeader* hp) | ||
175 | { | ||
176 | const unsigned char *cp; | ||
177 | int chksum, size; | ||
178 | |||
179 | strcpy(hp->magic, "ustar "); | ||
180 | |||
181 | /* Calculate and store the checksum (i.e., the sum of all of the bytes of | ||
182 | * the header). The checksum field must be filled with blanks for the | ||
183 | * calculation. The checksum field is formatted differently from the | ||
184 | * other fields: it has 6 digits, a null, then a space -- rather than | ||
185 | * digits, followed by a null like the other fields... */ | ||
186 | memset(hp->chksum, ' ', sizeof(hp->chksum)); | ||
187 | cp = (const unsigned char *) hp; | ||
188 | chksum = 0; | ||
189 | size = sizeof(*hp); | ||
190 | do { chksum += *cp++; } while (--size); | ||
191 | putOctal(hp->chksum, sizeof(hp->chksum)-1, chksum); | ||
192 | |||
193 | /* Now write the header out to disk */ | ||
194 | xwrite(fd, hp, sizeof(*hp)); | ||
195 | } | ||
196 | |||
197 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS | ||
198 | static void writeLongname(int fd, int type, const char *name, int dir) | ||
199 | { | ||
200 | static const struct { | ||
201 | char mode[8]; /* 100-107 */ | ||
202 | char uid[8]; /* 108-115 */ | ||
203 | char gid[8]; /* 116-123 */ | ||
204 | char size[12]; /* 124-135 */ | ||
205 | char mtime[12]; /* 136-147 */ | ||
206 | } prefilled = { | ||
207 | "0000000", | ||
208 | "0000000", | ||
209 | "0000000", | ||
210 | "00000000000", | ||
211 | "00000000000", | ||
212 | }; | ||
213 | struct TarHeader header; | ||
214 | int size; | ||
215 | |||
216 | dir = !!dir; /* normalize: 0/1 */ | ||
217 | size = strlen(name) + 1 + dir; /* GNU tar uses strlen+1 */ | ||
218 | /* + dir: account for possible '/' */ | ||
219 | |||
220 | memset(&header, 0, sizeof(header)); | ||
221 | strcpy(header.name, "././@LongLink"); | ||
222 | memcpy(header.mode, prefilled.mode, sizeof(prefilled)); | ||
223 | PUT_OCTAL(header.size, size); | ||
224 | header.typeflag = type; | ||
225 | chksum_and_xwrite(fd, &header); | ||
226 | |||
227 | /* Write filename[/] and pad the block. */ | ||
228 | /* dir=0: writes 'name<NUL>', pads */ | ||
229 | /* dir=1: writes 'name', writes '/<NUL>', pads */ | ||
230 | dir *= 2; | ||
231 | xwrite(fd, name, size - dir); | ||
232 | xwrite(fd, "/", dir); | ||
233 | size = (-size) & (TAR_BLOCK_SIZE-1); | ||
234 | memset(&header, 0, size); | ||
235 | xwrite(fd, &header, size); | ||
236 | } | ||
237 | #endif | ||
238 | |||
239 | /* Write out a tar header for the specified file/directory/whatever */ | ||
240 | void BUG_tar_header_size(void); | ||
241 | static int writeTarHeader(struct TarBallInfo *tbInfo, | ||
242 | const char *header_name, const char *fileName, struct stat *statbuf) | ||
243 | { | ||
244 | struct TarHeader header; | ||
245 | |||
246 | if (sizeof(header) != 512) | ||
247 | BUG_tar_header_size(); | ||
248 | |||
249 | memset(&header, 0, sizeof(struct TarHeader)); | ||
250 | |||
251 | strncpy(header.name, header_name, sizeof(header.name)); | ||
252 | |||
253 | /* POSIX says to mask mode with 07777. */ | ||
254 | PUT_OCTAL(header.mode, statbuf->st_mode & 07777); | ||
255 | PUT_OCTAL(header.uid, statbuf->st_uid); | ||
256 | PUT_OCTAL(header.gid, statbuf->st_gid); | ||
257 | memset(header.size, '0', sizeof(header.size)-1); /* Regular file size is handled later */ | ||
258 | PUT_OCTAL(header.mtime, statbuf->st_mtime); | ||
259 | |||
260 | /* Enter the user and group names */ | ||
261 | safe_strncpy(header.uname, get_cached_username(statbuf->st_uid), sizeof(header.uname)); | ||
262 | safe_strncpy(header.gname, get_cached_groupname(statbuf->st_gid), sizeof(header.gname)); | ||
263 | |||
264 | if (tbInfo->hlInfo) { | ||
265 | /* This is a hard link */ | ||
266 | header.typeflag = LNKTYPE; | ||
267 | strncpy(header.linkname, tbInfo->hlInfo->name, | ||
268 | sizeof(header.linkname)); | ||
269 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS | ||
270 | /* Write out long linkname if needed */ | ||
271 | if (header.linkname[sizeof(header.linkname)-1]) | ||
272 | writeLongname(tbInfo->tarFd, GNULONGLINK, | ||
273 | tbInfo->hlInfo->name, 0); | ||
274 | #endif | ||
275 | } else if (S_ISLNK(statbuf->st_mode)) { | ||
276 | char *lpath = xreadlink(fileName); | ||
277 | if (!lpath) /* Already printed err msg inside xreadlink() */ | ||
278 | return FALSE; | ||
279 | header.typeflag = SYMTYPE; | ||
280 | strncpy(header.linkname, lpath, sizeof(header.linkname)); | ||
281 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS | ||
282 | /* Write out long linkname if needed */ | ||
283 | if (header.linkname[sizeof(header.linkname)-1]) | ||
284 | writeLongname(tbInfo->tarFd, GNULONGLINK, lpath, 0); | ||
285 | #else | ||
286 | /* If it is larger than 100 bytes, bail out */ | ||
287 | if (header.linkname[sizeof(header.linkname)-1]) { | ||
288 | free(lpath); | ||
289 | bb_error_msg("names longer than "NAME_SIZE_STR" chars not supported"); | ||
290 | return FALSE; | ||
291 | } | ||
292 | #endif | ||
293 | free(lpath); | ||
294 | } else if (S_ISDIR(statbuf->st_mode)) { | ||
295 | header.typeflag = DIRTYPE; | ||
296 | /* Append '/' only if there is a space for it */ | ||
297 | if (!header.name[sizeof(header.name)-1]) | ||
298 | header.name[strlen(header.name)] = '/'; | ||
299 | } else if (S_ISCHR(statbuf->st_mode)) { | ||
300 | header.typeflag = CHRTYPE; | ||
301 | PUT_OCTAL(header.devmajor, major(statbuf->st_rdev)); | ||
302 | PUT_OCTAL(header.devminor, minor(statbuf->st_rdev)); | ||
303 | } else if (S_ISBLK(statbuf->st_mode)) { | ||
304 | header.typeflag = BLKTYPE; | ||
305 | PUT_OCTAL(header.devmajor, major(statbuf->st_rdev)); | ||
306 | PUT_OCTAL(header.devminor, minor(statbuf->st_rdev)); | ||
307 | } else if (S_ISFIFO(statbuf->st_mode)) { | ||
308 | header.typeflag = FIFOTYPE; | ||
309 | } else if (S_ISREG(statbuf->st_mode)) { | ||
310 | if (sizeof(statbuf->st_size) > 4 | ||
311 | && statbuf->st_size > (off_t)0777777777777LL | ||
312 | ) { | ||
313 | bb_error_msg_and_die("cannot store file '%s' " | ||
314 | "of size %"OFF_FMT"d, aborting", | ||
315 | fileName, statbuf->st_size); | ||
316 | } | ||
317 | header.typeflag = REGTYPE; | ||
318 | PUT_OCTAL(header.size, statbuf->st_size); | ||
319 | } else { | ||
320 | bb_error_msg("%s: unknown file type", fileName); | ||
321 | return FALSE; | ||
322 | } | ||
323 | |||
324 | #if ENABLE_FEATURE_TAR_GNU_EXTENSIONS | ||
325 | /* Write out long name if needed */ | ||
326 | /* (we, like GNU tar, output long linkname *before* long name) */ | ||
327 | if (header.name[sizeof(header.name)-1]) | ||
328 | writeLongname(tbInfo->tarFd, GNULONGNAME, | ||
329 | header_name, S_ISDIR(statbuf->st_mode)); | ||
330 | #endif | ||
331 | |||
332 | /* Now write the header out to disk */ | ||
333 | chksum_and_xwrite(tbInfo->tarFd, &header); | ||
334 | |||
335 | /* Now do the verbose thing (or not) */ | ||
336 | if (tbInfo->verboseFlag) { | ||
337 | FILE *vbFd = stdout; | ||
338 | |||
339 | if (tbInfo->tarFd == STDOUT_FILENO) /* If the archive goes to stdout, verbose to stderr */ | ||
340 | vbFd = stderr; | ||
341 | /* GNU "tar cvvf" prints "extended" listing a-la "ls -l" */ | ||
342 | /* We don't have such excesses here: for us "v" == "vv" */ | ||
343 | /* '/' is probably a GNUism */ | ||
344 | fprintf(vbFd, "%s%s\n", header_name, | ||
345 | S_ISDIR(statbuf->st_mode) ? "/" : ""); | ||
346 | } | ||
347 | |||
348 | return TRUE; | ||
349 | } | ||
350 | |||
351 | #if ENABLE_FEATURE_TAR_FROM | ||
352 | static int exclude_file(const llist_t *excluded_files, const char *file) | ||
353 | { | ||
354 | while (excluded_files) { | ||
355 | if (excluded_files->data[0] == '/') { | ||
356 | if (fnmatch(excluded_files->data, file, | ||
357 | FNM_PATHNAME | FNM_LEADING_DIR) == 0) | ||
358 | return 1; | ||
359 | } else { | ||
360 | const char *p; | ||
361 | |||
362 | for (p = file; p[0] != '\0'; p++) { | ||
363 | if ((p == file || p[-1] == '/') && p[0] != '/' && | ||
364 | fnmatch(excluded_files->data, p, | ||
365 | FNM_PATHNAME | FNM_LEADING_DIR) == 0) | ||
366 | return 1; | ||
367 | } | ||
368 | } | ||
369 | excluded_files = excluded_files->link; | ||
370 | } | ||
371 | |||
372 | return 0; | ||
373 | } | ||
374 | #else | ||
375 | #define exclude_file(excluded_files, file) 0 | ||
376 | #endif | ||
377 | |||
378 | static int writeFileToTarball(const char *fileName, struct stat *statbuf, | ||
379 | void *userData, int depth ATTRIBUTE_UNUSED) | ||
380 | { | ||
381 | struct TarBallInfo *tbInfo = (struct TarBallInfo *) userData; | ||
382 | const char *header_name; | ||
383 | int inputFileFd = -1; | ||
384 | |||
385 | /* | ||
386 | * Check to see if we are dealing with a hard link. | ||
387 | * If so - | ||
388 | * Treat the first occurance of a given dev/inode as a file while | ||
389 | * treating any additional occurances as hard links. This is done | ||
390 | * by adding the file information to the HardLinkInfo linked list. | ||
391 | */ | ||
392 | tbInfo->hlInfo = NULL; | ||
393 | if (statbuf->st_nlink > 1) { | ||
394 | tbInfo->hlInfo = findHardLinkInfo(tbInfo->hlInfoHead, statbuf); | ||
395 | if (tbInfo->hlInfo == NULL) | ||
396 | addHardLinkInfo(&tbInfo->hlInfoHead, statbuf, fileName); | ||
397 | } | ||
398 | |||
399 | /* It is against the rules to archive a socket */ | ||
400 | if (S_ISSOCK(statbuf->st_mode)) { | ||
401 | bb_error_msg("%s: socket ignored", fileName); | ||
402 | return TRUE; | ||
403 | } | ||
404 | |||
405 | /* It is a bad idea to store the archive we are in the process of creating, | ||
406 | * so check the device and inode to be sure that this particular file isn't | ||
407 | * the new tarball */ | ||
408 | if (tbInfo->statBuf.st_dev == statbuf->st_dev && | ||
409 | tbInfo->statBuf.st_ino == statbuf->st_ino) { | ||
410 | bb_error_msg("%s: file is the archive; skipping", fileName); | ||
411 | return TRUE; | ||
412 | } | ||
413 | |||
414 | header_name = fileName; | ||
415 | while (header_name[0] == '/') { | ||
416 | static int alreadyWarned = FALSE; | ||
417 | |||
418 | if (alreadyWarned == FALSE) { | ||
419 | bb_error_msg("removing leading '/' from member names"); | ||
420 | alreadyWarned = TRUE; | ||
421 | } | ||
422 | header_name++; | ||
423 | } | ||
424 | |||
425 | #if !ENABLE_FEATURE_TAR_GNU_EXTENSIONS | ||
426 | if (strlen(fileName) >= NAME_SIZE) { | ||
427 | bb_error_msg("names longer than "NAME_SIZE_STR" chars not supported"); | ||
428 | return TRUE; | ||
429 | } | ||
430 | #endif | ||
431 | |||
432 | if (header_name[0] == '\0') | ||
433 | return TRUE; | ||
434 | |||
435 | if (exclude_file(tbInfo->excludeList, header_name)) | ||
436 | return SKIP; | ||
437 | |||
438 | /* Is this a regular file? */ | ||
439 | if (tbInfo->hlInfo == NULL && S_ISREG(statbuf->st_mode)) { | ||
440 | /* open the file we want to archive, and make sure all is well */ | ||
441 | inputFileFd = open(fileName, O_RDONLY); | ||
442 | if (inputFileFd < 0) { | ||
443 | bb_perror_msg("%s: cannot open", fileName); | ||
444 | return FALSE; | ||
445 | } | ||
446 | } | ||
447 | |||
448 | /* Add an entry to the tarball */ | ||
449 | if (writeTarHeader(tbInfo, header_name, fileName, statbuf) == FALSE) { | ||
450 | return FALSE; | ||
451 | } | ||
452 | |||
453 | /* If it was a regular file, write out the body */ | ||
454 | if (inputFileFd >= 0) { | ||
455 | size_t readSize; | ||
456 | /* Wwrite the file to the archive. */ | ||
457 | /* We record size into header first, */ | ||
458 | /* and then write out file. If file shrinks in between, */ | ||
459 | /* tar will be corrupted. So we don't allow for that. */ | ||
460 | /* NB: GNU tar 1.16 warns and pads with zeroes */ | ||
461 | /* or even seeks back and updates header */ | ||
462 | bb_copyfd_exact_size(inputFileFd, tbInfo->tarFd, statbuf->st_size); | ||
463 | ////off_t readSize; | ||
464 | ////readSize = bb_copyfd_size(inputFileFd, tbInfo->tarFd, statbuf->st_size); | ||
465 | ////if (readSize != statbuf->st_size && readSize >= 0) { | ||
466 | //// bb_error_msg_and_die("short read from %s, aborting", fileName); | ||
467 | ////} | ||
468 | |||
469 | /* Check that file did not grow in between? */ | ||
470 | /* if (safe_read(inputFileFd, 1) == 1) warn but continue? */ | ||
471 | |||
472 | close(inputFileFd); | ||
473 | |||
474 | /* Pad the file up to the tar block size */ | ||
475 | /* (a few tricks here in the name of code size) */ | ||
476 | readSize = (-(int)statbuf->st_size) & (TAR_BLOCK_SIZE-1); | ||
477 | memset(bb_common_bufsiz1, 0, readSize); | ||
478 | xwrite(tbInfo->tarFd, bb_common_bufsiz1, readSize); | ||
479 | } | ||
480 | |||
481 | return TRUE; | ||
482 | } | ||
483 | |||
484 | static int writeTarFile(const int tar_fd, const int verboseFlag, | ||
485 | const unsigned long dereferenceFlag, const llist_t *include, | ||
486 | const llist_t *exclude, const int gzip) | ||
487 | { | ||
488 | pid_t gzipPid = 0; | ||
489 | int errorFlag = FALSE; | ||
490 | struct TarBallInfo tbInfo; | ||
491 | |||
492 | tbInfo.hlInfoHead = NULL; | ||
493 | |||
494 | fchmod(tar_fd, 0644); | ||
495 | tbInfo.tarFd = tar_fd; | ||
496 | tbInfo.verboseFlag = verboseFlag; | ||
497 | |||
498 | /* Store the stat info for the tarball's file, so | ||
499 | * can avoid including the tarball into itself.... */ | ||
500 | if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0) | ||
501 | bb_perror_msg_and_die("cannot stat tar file"); | ||
502 | |||
503 | if ((ENABLE_FEATURE_TAR_GZIP || ENABLE_FEATURE_TAR_BZIP2) && gzip) { | ||
504 | int gzipDataPipe[2] = { -1, -1 }; | ||
505 | int gzipStatusPipe[2] = { -1, -1 }; | ||
506 | volatile int vfork_exec_errno = 0; | ||
507 | char *zip_exec = (gzip == 1) ? "gzip" : "bzip2"; | ||
508 | |||
509 | if (pipe(gzipDataPipe) < 0 || pipe(gzipStatusPipe) < 0) | ||
510 | bb_perror_msg_and_die("pipe"); | ||
511 | |||
512 | signal(SIGPIPE, SIG_IGN); /* we only want EPIPE on errors */ | ||
513 | |||
514 | #if defined(__GNUC__) && __GNUC__ | ||
515 | /* Avoid vfork clobbering */ | ||
516 | (void) &include; | ||
517 | (void) &errorFlag; | ||
518 | (void) &zip_exec; | ||
519 | #endif | ||
520 | |||
521 | gzipPid = vfork(); | ||
522 | |||
523 | if (gzipPid == 0) { | ||
524 | dup2(gzipDataPipe[0], 0); | ||
525 | close(gzipDataPipe[1]); | ||
526 | |||
527 | dup2(tbInfo.tarFd, 1); | ||
528 | |||
529 | close(gzipStatusPipe[0]); | ||
530 | fcntl(gzipStatusPipe[1], F_SETFD, FD_CLOEXEC); /* close on exec shows success */ | ||
531 | |||
532 | execlp(zip_exec, zip_exec, "-f", NULL); | ||
533 | vfork_exec_errno = errno; | ||
534 | |||
535 | close(gzipStatusPipe[1]); | ||
536 | exit(-1); | ||
537 | } else if (gzipPid > 0) { | ||
538 | close(gzipDataPipe[0]); | ||
539 | close(gzipStatusPipe[1]); | ||
540 | |||
541 | while (1) { | ||
542 | char buf; | ||
543 | |||
544 | int n = full_read(gzipStatusPipe[0], &buf, 1); | ||
545 | |||
546 | if (n == 0 && vfork_exec_errno != 0) { | ||
547 | errno = vfork_exec_errno; | ||
548 | bb_perror_msg_and_die("cannot exec %s", zip_exec); | ||
549 | } else if ((n < 0) && (errno == EAGAIN || errno == EINTR)) | ||
550 | continue; /* try it again */ | ||
551 | break; | ||
552 | } | ||
553 | close(gzipStatusPipe[0]); | ||
554 | |||
555 | tbInfo.tarFd = gzipDataPipe[1]; | ||
556 | } else bb_perror_msg_and_die("vfork gzip"); | ||
557 | } | ||
558 | |||
559 | tbInfo.excludeList = exclude; | ||
560 | |||
561 | /* Read the directory/files and iterate over them one at a time */ | ||
562 | while (include) { | ||
563 | if (!recursive_action(include->data, TRUE, dereferenceFlag, | ||
564 | FALSE, writeFileToTarball, writeFileToTarball, &tbInfo, 0)) | ||
565 | { | ||
566 | errorFlag = TRUE; | ||
567 | } | ||
568 | include = include->link; | ||
569 | } | ||
570 | /* Write two empty blocks to the end of the archive */ | ||
571 | memset(bb_common_bufsiz1, 0, 2*TAR_BLOCK_SIZE); | ||
572 | xwrite(tbInfo.tarFd, bb_common_bufsiz1, 2*TAR_BLOCK_SIZE); | ||
573 | |||
574 | /* To be pedantically correct, we would check if the tarball | ||
575 | * is smaller than 20 tar blocks, and pad it if it was smaller, | ||
576 | * but that isn't necessary for GNU tar interoperability, and | ||
577 | * so is considered a waste of space */ | ||
578 | |||
579 | /* Close so the child process (if any) will exit */ | ||
580 | close(tbInfo.tarFd); | ||
581 | |||
582 | /* Hang up the tools, close up shop, head home */ | ||
583 | if (ENABLE_FEATURE_CLEAN_UP) | ||
584 | freeHardLinkInfo(&tbInfo.hlInfoHead); | ||
585 | |||
586 | if (errorFlag) | ||
587 | bb_error_msg("error exit delayed from previous errors"); | ||
588 | |||
589 | if (gzipPid) { | ||
590 | int status; | ||
591 | if (waitpid(gzipPid, &status, 0) == -1) | ||
592 | bb_perror_msg("waitpid"); | ||
593 | else if (!WIFEXITED(status) || WEXITSTATUS(status)) | ||
594 | /* gzip was killed or has exited with nonzero! */ | ||
595 | errorFlag = TRUE; | ||
596 | } | ||
597 | return errorFlag; | ||
598 | } | ||
599 | #else | ||
600 | int writeTarFile(const int tar_fd, const int verboseFlag, | ||
601 | const unsigned long dereferenceFlag, const llist_t *include, | ||
602 | const llist_t *exclude, const int gzip); | ||
603 | #endif /* FEATURE_TAR_CREATE */ | ||
604 | |||
605 | #if ENABLE_FEATURE_TAR_FROM | ||
606 | static llist_t *append_file_list_to_list(llist_t *list) | ||
607 | { | ||
608 | FILE *src_stream; | ||
609 | llist_t *cur = list; | ||
610 | llist_t *tmp; | ||
611 | char *line; | ||
612 | llist_t *newlist = NULL; | ||
613 | |||
614 | while (cur) { | ||
615 | src_stream = xfopen(cur->data, "r"); | ||
616 | tmp = cur; | ||
617 | cur = cur->link; | ||
618 | free(tmp); | ||
619 | while ((line = xmalloc_getline(src_stream)) != NULL) { | ||
620 | /* kill trailing '/' unless the string is just "/" */ | ||
621 | char *cp = last_char_is(line, '/'); | ||
622 | if (cp > line) | ||
623 | *cp = '\0'; | ||
624 | llist_add_to(&newlist, line); | ||
625 | } | ||
626 | fclose(src_stream); | ||
627 | } | ||
628 | return newlist; | ||
629 | } | ||
630 | #else | ||
631 | #define append_file_list_to_list(x) 0 | ||
632 | #endif | ||
633 | |||
634 | #if ENABLE_FEATURE_TAR_COMPRESS | ||
635 | static char get_header_tar_Z(archive_handle_t *archive_handle) | ||
636 | { | ||
637 | /* Can't lseek over pipes */ | ||
638 | archive_handle->seek = seek_by_read; | ||
639 | |||
640 | /* do the decompression, and cleanup */ | ||
641 | if (xread_char(archive_handle->src_fd) != 0x1f | ||
642 | || xread_char(archive_handle->src_fd) != 0x9d | ||
643 | ) { | ||
644 | bb_error_msg_and_die("invalid magic"); | ||
645 | } | ||
646 | |||
647 | archive_handle->src_fd = open_transformer(archive_handle->src_fd, uncompress); | ||
648 | archive_handle->offset = 0; | ||
649 | while (get_header_tar(archive_handle) == EXIT_SUCCESS) | ||
650 | /* nothing */; | ||
651 | |||
652 | /* Can only do one file at a time */ | ||
653 | return EXIT_FAILURE; | ||
654 | } | ||
655 | #else | ||
656 | #define get_header_tar_Z NULL | ||
657 | #endif | ||
658 | |||
659 | #ifdef CHECK_FOR_CHILD_EXITCODE | ||
660 | /* Looks like it isn't needed - tar detects malformed (truncated) | ||
661 | * archive if e.g. bunzip2 fails */ | ||
662 | static int child_error; | ||
663 | |||
664 | static void handle_SIGCHLD(int status) | ||
665 | { | ||
666 | /* Actually, 'status' is a signo. We reuse it for other needs */ | ||
667 | |||
668 | /* Wait for any child without blocking */ | ||
669 | if (waitpid(-1, &status, WNOHANG) < 0) | ||
670 | /* wait failed?! I'm confused... */ | ||
671 | return; | ||
672 | |||
673 | if (WIFEXITED(status) && WEXITSTATUS(status)==0) | ||
674 | /* child exited with 0 */ | ||
675 | return; | ||
676 | /* Cannot happen? | ||
677 | if(!WIFSIGNALED(status) && !WIFEXITED(status)) return; */ | ||
678 | child_error = 1; | ||
679 | } | ||
680 | #endif | ||
681 | |||
682 | enum { | ||
683 | OPTBIT_KEEP_OLD = 7, | ||
684 | USE_FEATURE_TAR_CREATE( OPTBIT_CREATE ,) | ||
685 | USE_FEATURE_TAR_CREATE( OPTBIT_DEREFERENCE ,) | ||
686 | USE_FEATURE_TAR_BZIP2( OPTBIT_BZIP2 ,) | ||
687 | USE_FEATURE_TAR_LZMA( OPTBIT_LZMA ,) | ||
688 | USE_FEATURE_TAR_FROM( OPTBIT_INCLUDE_FROM,) | ||
689 | USE_FEATURE_TAR_FROM( OPTBIT_EXCLUDE_FROM,) | ||
690 | USE_FEATURE_TAR_GZIP( OPTBIT_GZIP ,) | ||
691 | USE_FEATURE_TAR_COMPRESS(OPTBIT_COMPRESS ,) | ||
692 | OPTBIT_NOPRESERVE_OWN, | ||
693 | OPTBIT_NOPRESERVE_PERM, | ||
694 | OPT_TEST = 1 << 0, // t | ||
695 | OPT_EXTRACT = 1 << 1, // x | ||
696 | OPT_BASEDIR = 1 << 2, // C | ||
697 | OPT_TARNAME = 1 << 3, // f | ||
698 | OPT_2STDOUT = 1 << 4, // O | ||
699 | OPT_P = 1 << 5, // p | ||
700 | OPT_VERBOSE = 1 << 6, // v | ||
701 | OPT_KEEP_OLD = 1 << 7, // k | ||
702 | OPT_CREATE = USE_FEATURE_TAR_CREATE( (1<<OPTBIT_CREATE )) + 0, // c | ||
703 | OPT_DEREFERENCE = USE_FEATURE_TAR_CREATE( (1<<OPTBIT_DEREFERENCE )) + 0, // h | ||
704 | OPT_BZIP2 = USE_FEATURE_TAR_BZIP2( (1<<OPTBIT_BZIP2 )) + 0, // j | ||
705 | OPT_LZMA = USE_FEATURE_TAR_LZMA( (1<<OPTBIT_LZMA )) + 0, // a | ||
706 | OPT_INCLUDE_FROM = USE_FEATURE_TAR_FROM( (1<<OPTBIT_INCLUDE_FROM)) + 0, // T | ||
707 | OPT_EXCLUDE_FROM = USE_FEATURE_TAR_FROM( (1<<OPTBIT_EXCLUDE_FROM)) + 0, // X | ||
708 | OPT_GZIP = USE_FEATURE_TAR_GZIP( (1<<OPTBIT_GZIP )) + 0, // z | ||
709 | OPT_COMPRESS = USE_FEATURE_TAR_COMPRESS((1<<OPTBIT_COMPRESS )) + 0, // Z | ||
710 | OPT_NOPRESERVE_OWN = 1 << OPTBIT_NOPRESERVE_OWN , // no-same-owner | ||
711 | OPT_NOPRESERVE_PERM = 1 << OPTBIT_NOPRESERVE_PERM, // no-same-permissions | ||
712 | }; | ||
713 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS | ||
714 | static const struct option tar_long_options[] = { | ||
715 | { "list", 0, NULL, 't' }, | ||
716 | { "extract", 0, NULL, 'x' }, | ||
717 | { "directory", 1, NULL, 'C' }, | ||
718 | { "file", 1, NULL, 'f' }, | ||
719 | { "to-stdout", 0, NULL, 'O' }, | ||
720 | { "same-permissions", 0, NULL, 'p' }, | ||
721 | { "verbose", 0, NULL, 'v' }, | ||
722 | { "keep-old", 0, NULL, 'k' }, | ||
723 | # if ENABLE_FEATURE_TAR_CREATE | ||
724 | { "create", 0, NULL, 'c' }, | ||
725 | { "dereference", 0, NULL, 'h' }, | ||
726 | # endif | ||
727 | # if ENABLE_FEATURE_TAR_BZIP2 | ||
728 | { "bzip2", 0, NULL, 'j' }, | ||
729 | # endif | ||
730 | # if ENABLE_FEATURE_TAR_LZMA | ||
731 | { "lzma", 0, NULL, 'a' }, | ||
732 | # endif | ||
733 | # if ENABLE_FEATURE_TAR_FROM | ||
734 | { "files-from", 1, NULL, 'T' }, | ||
735 | { "exclude-from", 1, NULL, 'X' }, | ||
736 | # endif | ||
737 | # if ENABLE_FEATURE_TAR_GZIP | ||
738 | { "gzip", 0, NULL, 'z' }, | ||
739 | # endif | ||
740 | # if ENABLE_FEATURE_TAR_COMPRESS | ||
741 | { "compress", 0, NULL, 'Z' }, | ||
742 | # endif | ||
743 | { "no-same-owner", 0, NULL, 0xfd }, | ||
744 | { "no-same-permissions",0, NULL, 0xfe }, | ||
745 | /* --exclude takes next bit position in option mask, */ | ||
746 | /* therefore we have to either put it _after_ --no-same-perm */ | ||
747 | /* or add OPT[BIT]_EXCLUDE before OPT[BIT]_NOPRESERVE_OWN */ | ||
748 | # if ENABLE_FEATURE_TAR_FROM | ||
749 | { "exclude", 1, NULL, 0xff }, | ||
750 | # endif | ||
751 | { 0, 0, 0, 0 } | ||
752 | }; | ||
753 | #endif | ||
754 | |||
755 | int tar_main(int argc, char **argv) | ||
756 | { | ||
757 | char (*get_header_ptr)(archive_handle_t *) = get_header_tar; | ||
758 | archive_handle_t *tar_handle; | ||
759 | char *base_dir = NULL; | ||
760 | const char *tar_filename = "-"; | ||
761 | unsigned opt; | ||
762 | int verboseFlag = 0; | ||
763 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS && ENABLE_FEATURE_TAR_FROM | ||
764 | llist_t *excludes = NULL; | ||
765 | #endif | ||
766 | |||
767 | /* Initialise default values */ | ||
768 | tar_handle = init_handle(); | ||
769 | tar_handle->flags = ARCHIVE_CREATE_LEADING_DIRS | ||
770 | | ARCHIVE_PRESERVE_DATE | ||
771 | | ARCHIVE_EXTRACT_UNCONDITIONAL; | ||
772 | |||
773 | /* Prepend '-' to the first argument if required */ | ||
774 | opt_complementary = "--:" // first arg is options | ||
775 | "tt:vv:" // count -t,-v | ||
776 | "?:" // bail out with usage instead of error return | ||
777 | "X::T::" // cumulative lists | ||
778 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS && ENABLE_FEATURE_TAR_FROM | ||
779 | "\xff::" // cumulative lists for --exclude | ||
780 | #endif | ||
781 | USE_FEATURE_TAR_CREATE("c:") "t:x:" // at least one of these is reqd | ||
782 | USE_FEATURE_TAR_CREATE("c--tx:t--cx:x--ct") // mutually exclusive | ||
783 | SKIP_FEATURE_TAR_CREATE("t--x:x--t"); // mutually exclusive | ||
784 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS | ||
785 | applet_long_options = tar_long_options; | ||
786 | #endif | ||
787 | opt = getopt32(argc, argv, | ||
788 | "txC:f:Opvk" | ||
789 | USE_FEATURE_TAR_CREATE( "ch" ) | ||
790 | USE_FEATURE_TAR_BZIP2( "j" ) | ||
791 | USE_FEATURE_TAR_LZMA( "a" ) | ||
792 | USE_FEATURE_TAR_FROM( "T:X:") | ||
793 | USE_FEATURE_TAR_GZIP( "z" ) | ||
794 | USE_FEATURE_TAR_COMPRESS("Z" ) | ||
795 | , &base_dir // -C dir | ||
796 | , &tar_filename // -f filename | ||
797 | USE_FEATURE_TAR_FROM(, &(tar_handle->accept)) // T | ||
798 | USE_FEATURE_TAR_FROM(, &(tar_handle->reject)) // X | ||
799 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS && ENABLE_FEATURE_TAR_FROM | ||
800 | , &excludes // --exclude | ||
801 | #endif | ||
802 | , &verboseFlag // combined count for -t and -v | ||
803 | , &verboseFlag // combined count for -t and -v | ||
804 | ); | ||
805 | |||
806 | if (verboseFlag) tar_handle->action_header = header_verbose_list; | ||
807 | if (verboseFlag == 1) tar_handle->action_header = header_list; | ||
808 | |||
809 | if (opt & OPT_EXTRACT) | ||
810 | tar_handle->action_data = data_extract_all; | ||
811 | |||
812 | if (opt & OPT_2STDOUT) | ||
813 | tar_handle->action_data = data_extract_to_stdout; | ||
814 | |||
815 | if (opt & OPT_KEEP_OLD) | ||
816 | tar_handle->flags &= ~ARCHIVE_EXTRACT_UNCONDITIONAL; | ||
817 | |||
818 | if (opt & OPT_NOPRESERVE_OWN) | ||
819 | tar_handle->flags |= ARCHIVE_NOPRESERVE_OWN; | ||
820 | |||
821 | if (opt & OPT_NOPRESERVE_PERM) | ||
822 | tar_handle->flags |= ARCHIVE_NOPRESERVE_PERM; | ||
823 | |||
824 | if (opt & OPT_GZIP) | ||
825 | get_header_ptr = get_header_tar_gz; | ||
826 | |||
827 | if (opt & OPT_BZIP2) | ||
828 | get_header_ptr = get_header_tar_bz2; | ||
829 | |||
830 | if (opt & OPT_LZMA) | ||
831 | get_header_ptr = get_header_tar_lzma; | ||
832 | |||
833 | if (opt & OPT_COMPRESS) | ||
834 | get_header_ptr = get_header_tar_Z; | ||
835 | |||
836 | #if ENABLE_FEATURE_TAR_FROM | ||
837 | tar_handle->reject = append_file_list_to_list(tar_handle->reject); | ||
838 | #if ENABLE_FEATURE_TAR_LONG_OPTIONS | ||
839 | /* Append excludes to reject */ | ||
840 | while (excludes) { | ||
841 | llist_t *next = excludes->link; | ||
842 | excludes->link = tar_handle->reject; | ||
843 | tar_handle->reject = excludes; | ||
844 | excludes = next; | ||
845 | } | ||
846 | #endif | ||
847 | tar_handle->accept = append_file_list_to_list(tar_handle->accept); | ||
848 | #endif | ||
849 | |||
850 | /* Check if we are reading from stdin */ | ||
851 | if (argv[optind] && *argv[optind] == '-') { | ||
852 | /* Default is to read from stdin, so just skip to next arg */ | ||
853 | optind++; | ||
854 | } | ||
855 | |||
856 | /* Setup an array of filenames to work with */ | ||
857 | /* TODO: This is the same as in ar, separate function ? */ | ||
858 | while (optind < argc) { | ||
859 | /* kill trailing '/' unless the string is just "/" */ | ||
860 | char *cp = last_char_is(argv[optind], '/'); | ||
861 | if (cp > argv[optind]) | ||
862 | *cp = '\0'; | ||
863 | llist_add_to(&tar_handle->accept, argv[optind]); | ||
864 | optind++; | ||
865 | } | ||
866 | tar_handle->accept = rev_llist(tar_handle->accept); | ||
867 | |||
868 | if (tar_handle->accept || tar_handle->reject) | ||
869 | tar_handle->filter = filter_accept_reject_list; | ||
870 | |||
871 | /* Open the tar file */ | ||
872 | { | ||
873 | FILE *tar_stream; | ||
874 | int flags; | ||
875 | |||
876 | if (opt & OPT_CREATE) { | ||
877 | /* Make sure there is at least one file to tar up. */ | ||
878 | if (tar_handle->accept == NULL) | ||
879 | bb_error_msg_and_die("empty archive"); | ||
880 | |||
881 | tar_stream = stdout; | ||
882 | /* Mimicking GNU tar 1.15.1: */ | ||
883 | flags = O_WRONLY|O_CREAT|O_TRUNC; | ||
884 | /* was doing unlink; open(O_WRONLY|O_CREAT|O_EXCL); why? */ | ||
885 | } else { | ||
886 | tar_stream = stdin; | ||
887 | flags = O_RDONLY; | ||
888 | } | ||
889 | |||
890 | if (LONE_DASH(tar_filename)) { | ||
891 | tar_handle->src_fd = fileno(tar_stream); | ||
892 | tar_handle->seek = seek_by_read; | ||
893 | } else { | ||
894 | tar_handle->src_fd = xopen(tar_filename, flags); | ||
895 | } | ||
896 | } | ||
897 | |||
898 | if (base_dir) | ||
899 | xchdir(base_dir); | ||
900 | |||
901 | #ifdef CHECK_FOR_CHILD_EXITCODE | ||
902 | /* We need to know whether child (gzip/bzip/etc) exits abnormally */ | ||
903 | signal(SIGCHLD, handle_SIGCHLD); | ||
904 | #endif | ||
905 | |||
906 | /* create an archive */ | ||
907 | if (opt & OPT_CREATE) { | ||
908 | int zipMode = 0; | ||
909 | if (ENABLE_FEATURE_TAR_GZIP && get_header_ptr == get_header_tar_gz) | ||
910 | zipMode = 1; | ||
911 | if (ENABLE_FEATURE_TAR_BZIP2 && get_header_ptr == get_header_tar_bz2) | ||
912 | zipMode = 2; | ||
913 | /* NB: writeTarFile() closes tar_handle->src_fd */ | ||
914 | return writeTarFile(tar_handle->src_fd, verboseFlag, opt & OPT_DEREFERENCE, | ||
915 | tar_handle->accept, | ||
916 | tar_handle->reject, zipMode); | ||
917 | } | ||
918 | |||
919 | while (get_header_ptr(tar_handle) == EXIT_SUCCESS) | ||
920 | /* nothing */; | ||
921 | |||
922 | /* Check that every file that should have been extracted was */ | ||
923 | while (tar_handle->accept) { | ||
924 | if (!find_list_entry(tar_handle->reject, tar_handle->accept->data) | ||
925 | && !find_list_entry(tar_handle->passed, tar_handle->accept->data) | ||
926 | ) { | ||
927 | bb_error_msg_and_die("%s: not found in archive", | ||
928 | tar_handle->accept->data); | ||
929 | } | ||
930 | tar_handle->accept = tar_handle->accept->link; | ||
931 | } | ||
932 | if (ENABLE_FEATURE_CLEAN_UP /* && tar_handle->src_fd != STDIN_FILENO */) | ||
933 | close(tar_handle->src_fd); | ||
934 | |||
935 | return EXIT_SUCCESS; | ||
936 | } |