Contents of /trunk/mkinitrd-magellan/busybox/archival/unzip.c
Parent Directory | Revision Log
Revision 532 -
(show annotations)
(download)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 10397 byte(s)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 10397 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 | /* vi: set sw=4 ts=4: */ |
2 | /* |
3 | * Mini unzip implementation for busybox |
4 | * |
5 | * Copyright (C) 2004 by Ed Clark |
6 | * |
7 | * Loosely based on original busybox unzip applet by Laurence Anderson. |
8 | * All options and features should work in this version. |
9 | * |
10 | * Licensed under the GPL v2 or later, see the file LICENSE in this tarball. |
11 | */ |
12 | |
13 | /* For reference see |
14 | * http://www.pkware.com/company/standards/appnote/ |
15 | * http://www.info-zip.org/pub/infozip/doc/appnote-iz-latest.zip |
16 | */ |
17 | |
18 | /* TODO |
19 | * Endian issues |
20 | * Zip64 + other methods |
21 | * Improve handling of zip format, ie. |
22 | * - deferred CRC, comp. & uncomp. lengths (zip header flags bit 3) |
23 | * - unix file permissions, etc. |
24 | * - central directory |
25 | */ |
26 | |
27 | #include "busybox.h" |
28 | #include "unarchive.h" |
29 | |
30 | #define ZIP_FILEHEADER_MAGIC SWAP_LE32(0x04034b50) |
31 | #define ZIP_CDS_MAGIC SWAP_LE32(0x02014b50) |
32 | #define ZIP_CDS_END_MAGIC SWAP_LE32(0x06054b50) |
33 | #define ZIP_DD_MAGIC SWAP_LE32(0x08074b50) |
34 | |
35 | typedef union { |
36 | unsigned char raw[26]; |
37 | struct { |
38 | unsigned short version; /* 0-1 */ |
39 | unsigned short flags; /* 2-3 */ |
40 | unsigned short method; /* 4-5 */ |
41 | unsigned short modtime; /* 6-7 */ |
42 | unsigned short moddate; /* 8-9 */ |
43 | unsigned int crc32 ATTRIBUTE_PACKED; /* 10-13 */ |
44 | unsigned int cmpsize ATTRIBUTE_PACKED; /* 14-17 */ |
45 | unsigned int ucmpsize ATTRIBUTE_PACKED; /* 18-21 */ |
46 | unsigned short filename_len; /* 22-23 */ |
47 | unsigned short extra_len; /* 24-25 */ |
48 | } formatted ATTRIBUTE_PACKED; |
49 | } zip_header_t; |
50 | |
51 | static void unzip_skip(int fd, off_t skip) |
52 | { |
53 | if (lseek(fd, skip, SEEK_CUR) == (off_t)-1) { |
54 | if (errno != ESPIPE) |
55 | bb_error_msg_and_die("seek failure"); |
56 | bb_copyfd_exact_size(fd, -1, skip); |
57 | } |
58 | } |
59 | |
60 | static void unzip_create_leading_dirs(char *fn) |
61 | { |
62 | /* Create all leading directories */ |
63 | char *name = xstrdup(fn); |
64 | if (bb_make_directory(dirname(name), 0777, FILEUTILS_RECUR)) { |
65 | bb_error_msg_and_die("exiting"); /* bb_make_directory is noisy */ |
66 | } |
67 | free(name); |
68 | } |
69 | |
70 | static int unzip_extract(zip_header_t *zip_header, int src_fd, int dst_fd) |
71 | { |
72 | if (zip_header->formatted.method == 0) { |
73 | /* Method 0 - stored (not compressed) */ |
74 | off_t size = zip_header->formatted.ucmpsize; |
75 | if (size) |
76 | bb_copyfd_exact_size(src_fd, dst_fd, size); |
77 | } else { |
78 | /* Method 8 - inflate */ |
79 | inflate_unzip_result res; |
80 | /* err = */ inflate_unzip(&res, zip_header->formatted.cmpsize, src_fd, dst_fd); |
81 | // we should check for -1 error return |
82 | /* Validate decompression - crc */ |
83 | if (zip_header->formatted.crc32 != (res.crc ^ 0xffffffffL)) { |
84 | bb_error_msg("invalid compressed data--%s error", "crc"); |
85 | return 1; |
86 | } |
87 | /* Validate decompression - size */ |
88 | if (zip_header->formatted.ucmpsize != res.bytes_out) { |
89 | bb_error_msg("invalid compressed data--%s error", "length"); |
90 | return 1; |
91 | } |
92 | } |
93 | return 0; |
94 | } |
95 | |
96 | int unzip_main(int argc, char **argv) |
97 | { |
98 | zip_header_t zip_header; |
99 | enum {v_silent, v_normal, v_list} verbosity = v_normal; |
100 | enum {o_prompt, o_never, o_always} overwrite = o_prompt; |
101 | unsigned int total_size = 0; |
102 | unsigned int total_entries = 0; |
103 | int src_fd = -1, dst_fd = -1; |
104 | char *src_fn = NULL, *dst_fn = NULL; |
105 | llist_t *zaccept = NULL; |
106 | llist_t *zreject = NULL; |
107 | char *base_dir = NULL; |
108 | int failed, i, opt, opt_range = 0, list_header_done = 0; |
109 | char key_buf[512]; |
110 | struct stat stat_buf; |
111 | |
112 | while ((opt = getopt(argc, argv, "-d:lnopqx")) != -1) { |
113 | switch (opt_range) { |
114 | case 0: /* Options */ |
115 | switch (opt) { |
116 | case 'l': /* List */ |
117 | verbosity = v_list; |
118 | break; |
119 | |
120 | case 'n': /* Never overwrite existing files */ |
121 | overwrite = o_never; |
122 | break; |
123 | |
124 | case 'o': /* Always overwrite existing files */ |
125 | overwrite = o_always; |
126 | break; |
127 | |
128 | case 'p': /* Extract files to stdout and fall through to set verbosity */ |
129 | dst_fd = STDOUT_FILENO; |
130 | |
131 | case 'q': /* Be quiet */ |
132 | verbosity = (verbosity == v_normal) ? v_silent : verbosity; |
133 | break; |
134 | |
135 | case 1 : /* The zip file */ |
136 | src_fn = xstrndup(optarg, strlen(optarg)+4); |
137 | opt_range++; |
138 | break; |
139 | |
140 | default: |
141 | bb_show_usage(); |
142 | |
143 | } |
144 | break; |
145 | |
146 | case 1: /* Include files */ |
147 | if (opt == 1) { |
148 | llist_add_to(&zaccept, optarg); |
149 | |
150 | } else if (opt == 'd') { |
151 | base_dir = optarg; |
152 | opt_range += 2; |
153 | |
154 | } else if (opt == 'x') { |
155 | opt_range++; |
156 | |
157 | } else { |
158 | bb_show_usage(); |
159 | } |
160 | break; |
161 | |
162 | case 2 : /* Exclude files */ |
163 | if (opt == 1) { |
164 | llist_add_to(&zreject, optarg); |
165 | |
166 | } else if (opt == 'd') { /* Extract to base directory */ |
167 | base_dir = optarg; |
168 | opt_range++; |
169 | |
170 | } else { |
171 | bb_show_usage(); |
172 | } |
173 | break; |
174 | |
175 | default: |
176 | bb_show_usage(); |
177 | } |
178 | } |
179 | |
180 | if (src_fn == NULL) { |
181 | bb_show_usage(); |
182 | } |
183 | |
184 | /* Open input file */ |
185 | if (LONE_DASH(src_fn)) { |
186 | src_fd = STDIN_FILENO; |
187 | /* Cannot use prompt mode since zip data is arriving on STDIN */ |
188 | overwrite = (overwrite == o_prompt) ? o_never : overwrite; |
189 | } else { |
190 | static const char *const extn[] = {"", ".zip", ".ZIP"}; |
191 | int orig_src_fn_len = strlen(src_fn); |
192 | for (i = 0; (i < 3) && (src_fd == -1); i++) { |
193 | strcpy(src_fn + orig_src_fn_len, extn[i]); |
194 | src_fd = open(src_fn, O_RDONLY); |
195 | } |
196 | if (src_fd == -1) { |
197 | src_fn[orig_src_fn_len] = 0; |
198 | bb_error_msg_and_die("cannot open %s, %s.zip, %s.ZIP", src_fn, src_fn, src_fn); |
199 | } |
200 | } |
201 | |
202 | /* Change dir if necessary */ |
203 | if (base_dir) |
204 | xchdir(base_dir); |
205 | |
206 | if (verbosity != v_silent) |
207 | printf("Archive: %s\n", src_fn); |
208 | |
209 | failed = 0; |
210 | |
211 | while (1) { |
212 | unsigned int magic; |
213 | |
214 | /* Check magic number */ |
215 | xread(src_fd, &magic, 4); |
216 | if (magic == ZIP_CDS_MAGIC) { |
217 | break; |
218 | } else if (magic != ZIP_FILEHEADER_MAGIC) { |
219 | bb_error_msg_and_die("invalid zip magic %08X", magic); |
220 | } |
221 | |
222 | /* Read the file header */ |
223 | xread(src_fd, zip_header.raw, 26); |
224 | zip_header.formatted.version = SWAP_LE32(zip_header.formatted.version); |
225 | zip_header.formatted.flags = SWAP_LE32(zip_header.formatted.flags); |
226 | zip_header.formatted.method = SWAP_LE32(zip_header.formatted.method); |
227 | zip_header.formatted.modtime = SWAP_LE32(zip_header.formatted.modtime); |
228 | zip_header.formatted.moddate = SWAP_LE32(zip_header.formatted.moddate); |
229 | zip_header.formatted.crc32 = SWAP_LE32(zip_header.formatted.crc32); |
230 | zip_header.formatted.cmpsize = SWAP_LE32(zip_header.formatted.cmpsize); |
231 | zip_header.formatted.ucmpsize = SWAP_LE32(zip_header.formatted.ucmpsize); |
232 | zip_header.formatted.filename_len = SWAP_LE32(zip_header.formatted.filename_len); |
233 | zip_header.formatted.extra_len = SWAP_LE32(zip_header.formatted.extra_len); |
234 | if ((zip_header.formatted.method != 0) && (zip_header.formatted.method != 8)) { |
235 | bb_error_msg_and_die("unsupported compression method %d", zip_header.formatted.method); |
236 | } |
237 | |
238 | /* Read filename */ |
239 | free(dst_fn); |
240 | dst_fn = xzalloc(zip_header.formatted.filename_len + 1); |
241 | xread(src_fd, dst_fn, zip_header.formatted.filename_len); |
242 | |
243 | /* Skip extra header bytes */ |
244 | unzip_skip(src_fd, zip_header.formatted.extra_len); |
245 | |
246 | if ((verbosity == v_list) && !list_header_done){ |
247 | puts(" Length Date Time Name\n" |
248 | " -------- ---- ---- ----"); |
249 | list_header_done = 1; |
250 | } |
251 | |
252 | /* Filter zip entries */ |
253 | if (find_list_entry(zreject, dst_fn) || |
254 | (zaccept && !find_list_entry(zaccept, dst_fn))) { /* Skip entry */ |
255 | i = 'n'; |
256 | |
257 | } else { /* Extract entry */ |
258 | total_size += zip_header.formatted.ucmpsize; |
259 | |
260 | if (verbosity == v_list) { /* List entry */ |
261 | unsigned int dostime = zip_header.formatted.modtime | (zip_header.formatted.moddate << 16); |
262 | printf("%9u %02u-%02u-%02u %02u:%02u %s\n", |
263 | zip_header.formatted.ucmpsize, |
264 | (dostime & 0x01e00000) >> 21, |
265 | (dostime & 0x001f0000) >> 16, |
266 | (((dostime & 0xfe000000) >> 25) + 1980) % 100, |
267 | (dostime & 0x0000f800) >> 11, |
268 | (dostime & 0x000007e0) >> 5, |
269 | dst_fn); |
270 | total_entries++; |
271 | i = 'n'; |
272 | } else if (dst_fd == STDOUT_FILENO) { /* Extracting to STDOUT */ |
273 | i = -1; |
274 | } else if (last_char_is(dst_fn, '/')) { /* Extract directory */ |
275 | if (stat(dst_fn, &stat_buf) == -1) { |
276 | if (errno != ENOENT) { |
277 | bb_perror_msg_and_die("cannot stat '%s'",dst_fn); |
278 | } |
279 | if (verbosity == v_normal) { |
280 | printf(" creating: %s\n", dst_fn); |
281 | } |
282 | unzip_create_leading_dirs(dst_fn); |
283 | if (bb_make_directory(dst_fn, 0777, 0)) { |
284 | bb_error_msg_and_die("exiting"); |
285 | } |
286 | } else { |
287 | if (!S_ISDIR(stat_buf.st_mode)) { |
288 | bb_error_msg_and_die("'%s' exists but is not directory", dst_fn); |
289 | } |
290 | } |
291 | i = 'n'; |
292 | |
293 | } else { /* Extract file */ |
294 | _check_file: |
295 | if (stat(dst_fn, &stat_buf) == -1) { /* File does not exist */ |
296 | if (errno != ENOENT) { |
297 | bb_perror_msg_and_die("cannot stat '%s'",dst_fn); |
298 | } |
299 | i = 'y'; |
300 | } else { /* File already exists */ |
301 | if (overwrite == o_never) { |
302 | i = 'n'; |
303 | } else if (S_ISREG(stat_buf.st_mode)) { /* File is regular file */ |
304 | if (overwrite == o_always) { |
305 | i = 'y'; |
306 | } else { |
307 | printf("replace %s? [y]es, [n]o, [A]ll, [N]one, [r]ename: ", dst_fn); |
308 | if (!fgets(key_buf, 512, stdin)) { |
309 | bb_perror_msg_and_die("cannot read input"); |
310 | } |
311 | i = key_buf[0]; |
312 | } |
313 | } else { /* File is not regular file */ |
314 | bb_error_msg_and_die("'%s' exists but is not regular file",dst_fn); |
315 | } |
316 | } |
317 | } |
318 | } |
319 | |
320 | switch (i) { |
321 | case 'A': |
322 | overwrite = o_always; |
323 | case 'y': /* Open file and fall into unzip */ |
324 | unzip_create_leading_dirs(dst_fn); |
325 | dst_fd = xopen(dst_fn, O_WRONLY | O_CREAT | O_TRUNC); |
326 | case -1: /* Unzip */ |
327 | if (verbosity == v_normal) { |
328 | printf(" inflating: %s\n", dst_fn); |
329 | } |
330 | if (unzip_extract(&zip_header, src_fd, dst_fd)) { |
331 | failed = 1; |
332 | } |
333 | if (dst_fd != STDOUT_FILENO) { |
334 | /* closing STDOUT is potentially bad for future business */ |
335 | close(dst_fd); |
336 | } |
337 | break; |
338 | |
339 | case 'N': |
340 | overwrite = o_never; |
341 | case 'n': |
342 | /* Skip entry data */ |
343 | unzip_skip(src_fd, zip_header.formatted.cmpsize); |
344 | break; |
345 | |
346 | case 'r': |
347 | /* Prompt for new name */ |
348 | printf("new name: "); |
349 | if (!fgets(key_buf, 512, stdin)) { |
350 | bb_perror_msg_and_die("cannot read input"); |
351 | } |
352 | free(dst_fn); |
353 | dst_fn = xstrdup(key_buf); |
354 | chomp(dst_fn); |
355 | goto _check_file; |
356 | |
357 | default: |
358 | printf("error: invalid response [%c]\n",(char)i); |
359 | goto _check_file; |
360 | } |
361 | |
362 | /* Data descriptor section */ |
363 | if (zip_header.formatted.flags & 4) { |
364 | /* skip over duplicate crc, compressed size and uncompressed size */ |
365 | unzip_skip(src_fd, 12); |
366 | } |
367 | } |
368 | |
369 | if (verbosity == v_list) { |
370 | printf(" -------- -------\n" |
371 | "%9d %d files\n", total_size, total_entries); |
372 | } |
373 | |
374 | return failed; |
375 | } |