Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(181)

Side by Side Diff: third_party/ijar/zip.cc

Issue 1323053003: Add ijar to third_party and use it for generating .jar.toc files in GN (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@compiled_action
Patch Set: fix release mode Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « third_party/ijar/zip.h ('k') | third_party/ijar/zip_main.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright 2007 Alan Donovan. All rights reserved.
2 //
3 // Author: Alan Donovan <adonovan@google.com>
4 //
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
8 //
9 // http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16 //
17 // zip.cc -- .zip (.jar) file reading/writing routines.
18 //
19
20 // See README.txt for details.
21 //
22 // See http://www.pkware.com/documents/casestudies/APPNOTE.TXT
23 // for definition of PKZIP file format.
24
25 #define _FILE_OFFSET_BITS 64 // Support zip files larger than 2GB
26
27 #include <errno.h>
28 #include <fcntl.h>
29 #include <stddef.h>
30 #include <stdint.h>
31 #include <stdarg.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include <sys/mman.h>
36 #include <unistd.h>
37 #include <limits.h>
38 #include <limits>
39 #include <vector>
40
41 #include "third_party/ijar/zip.h"
42 #include <zlib.h>
43
44 #define LOCAL_FILE_HEADER_SIGNATURE 0x04034b50
45 #define CENTRAL_FILE_HEADER_SIGNATURE 0x02014b50
46 #define END_OF_CENTRAL_DIR_SIGNATURE 0x06054b50
47 #define DATA_DESCRIPTOR_SIGNATURE 0x08074b50
48
49 // version to extract: 1.0 - default value from APPNOTE.TXT.
50 // Output JAR files contain no extra ZIP features, so this is enough.
51 #define ZIP_VERSION_TO_EXTRACT 10
52 #define COMPRESSION_METHOD_STORED 0 // no compression
53 #define COMPRESSION_METHOD_DEFLATED 8
54
55 #define GENERAL_PURPOSE_BIT_FLAG_COMPRESSED (1 << 3)
56 #define GENERAL_PURPOSE_BIT_FLAG_UTF8_ENCODED (1 << 11)
57 #define GENERAL_PURPOSE_BIT_FLAG_COMPRESSION_SPEED ((1 << 2) | (1 << 1))
58 #define GENERAL_PURPOSE_BIT_FLAG_SUPPORTED \
59 (GENERAL_PURPOSE_BIT_FLAG_COMPRESSED \
60 | GENERAL_PURPOSE_BIT_FLAG_UTF8_ENCODED \
61 | GENERAL_PURPOSE_BIT_FLAG_COMPRESSION_SPEED)
62
63 namespace devtools_ijar {
64 // In the absence of ZIP64 support, zip files are limited to 4GB.
65 // http://www.info-zip.org/FAQ.html#limits
66 static const u8 kMaximumOutputSize = std::numeric_limits<uint32_t>::max();
67
68 static bool ProcessCentralDirEntry(const u1 *&p,
69 size_t *compressed_size,
70 size_t *uncompressed_size,
71 char *filename,
72 size_t filename_size,
73 u4 *attr,
74 u4 *offset);
75
76 //
77 // A class representing a ZipFile for reading. Its public API is exposed
78 // using the ZipExtractor abstract class.
79 //
80 class InputZipFile : public ZipExtractor {
81 public:
82 InputZipFile(ZipExtractorProcessor *processor, int fd, off_t in_length,
83 off_t in_offset, const u1* zipdata_in, const u1* central_dir);
84 virtual ~InputZipFile();
85
86 virtual const char* GetError() {
87 if (errmsg[0] == 0) {
88 return NULL;
89 }
90 return errmsg;
91 }
92
93 virtual bool ProcessNext();
94 virtual void Reset();
95 virtual size_t GetSize() {
96 return in_length_;
97 }
98
99 virtual u8 CalculateOutputLength();
100
101 private:
102 ZipExtractorProcessor *processor;
103
104 int fd_in; // Input file descripor
105
106 // InputZipFile is responsible for maintaining the following
107 // pointers. They are allocated by the Create() method before
108 // the object is actually created using mmap.
109 const u1 * const zipdata_in_; // start of input file mmap
110 const u1 * zipdata_in_mapped_; // start of still mapped region
111 const u1 * const central_dir_; // central directory in input file
112
113 size_t in_length_; // size of the input file
114 size_t in_offset_; // offset the input file
115
116 const u1 *p; // input cursor
117
118 const u1* central_dir_current_; // central dir input cursor
119
120 // Buffer size is initially INITIAL_BUFFER_SIZE. It doubles in size every
121 // time it is found too small, until it reaches MAX_BUFFER_SIZE. If that is
122 // not enough, we bail out. We only decompress class files, so they should
123 // be smaller than 64K anyway, but we give a little leeway.
124 // MAX_BUFFER_SIZE must be bigger than the size of the biggest file in the
125 // ZIP. It is set to 128M here so we can uncompress the Bazel server with
126 // this library.
127 static const size_t INITIAL_BUFFER_SIZE = 256 * 1024; // 256K
128 static const size_t MAX_BUFFER_SIZE = 128 * 1024 * 1024;
129 static const size_t MAX_MAPPED_REGION = 32 * 1024 * 1024;
130
131 // These metadata fields are the fields of the ZIP header of the file being
132 // processed.
133 u2 extract_version_;
134 u2 general_purpose_bit_flag_;
135 u2 compression_method_;
136 u4 uncompressed_size_;
137 u4 compressed_size_;
138 u2 file_name_length_;
139 u2 extra_field_length_;
140 const u1 *file_name_;
141 const u1 *extra_field_;
142
143 // Administration of memory reserved for decompressed data. We use the same
144 // buffer for each file to avoid some malloc()/free() calls and free the
145 // memory only in the dtor. C-style memory management is used so that we
146 // can call realloc.
147 u1 *uncompressed_data_;
148 size_t uncompressed_data_allocated_;
149
150 // Copy of the last filename entry - Null-terminated.
151 char filename[PATH_MAX];
152 // The external file attribute field
153 u4 attr;
154
155 // last error
156 char errmsg[4*PATH_MAX];
157
158 int error(const char *fmt, ...) {
159 va_list ap;
160 va_start(ap, fmt);
161 vsnprintf(errmsg, 4*PATH_MAX, fmt, ap);
162 va_end(ap);
163 return -1;
164 }
165
166 // Check that at least n bytes remain in the input file, otherwise
167 // abort with an error message. "state" is the name of the field
168 // we're about to read, for diagnostics.
169 int EnsureRemaining(size_t n, const char *state) {
170 size_t in_offset = p - zipdata_in_;
171 size_t remaining = in_length_ - in_offset;
172 if (n > remaining) {
173 return error("Premature end of file (at offset %zd, state=%s); "
174 "expected %zd more bytes but found %zd.\n",
175 in_offset, state, n, remaining);
176 }
177 return 0;
178 }
179
180 // Read one entry from input zip file
181 int ProcessLocalFileEntry(size_t compressed_size, size_t uncompressed_size);
182
183 // Uncompress a file from the archive using zlib. The pointer returned
184 // is owned by InputZipFile, so it must not be freed. Advances the input
185 // cursor to the first byte after the compressed data.
186 u1* UncompressFile();
187
188 // Skip a file
189 int SkipFile(const bool compressed);
190
191 // Process a file
192 int ProcessFile(const bool compressed);
193 };
194
195 //
196 // A class implementing ZipBuilder that represent an open zip file for writing.
197 //
198 class OutputZipFile : public ZipBuilder {
199 public:
200 OutputZipFile(int fd, u1 * const zipdata_out) :
201 fd_out(fd),
202 zipdata_out_(zipdata_out),
203 q(zipdata_out) {
204 errmsg[0] = 0;
205 }
206
207 virtual const char* GetError() {
208 if (errmsg[0] == 0) {
209 return NULL;
210 }
211 return errmsg;
212 }
213
214 virtual ~OutputZipFile() { Finish(); }
215 virtual u1* NewFile(const char* filename, const u4 attr);
216 virtual int FinishFile(size_t filelength, bool compress = false);
217 virtual int WriteEmptyFile(const char *filename);
218 virtual size_t GetSize() {
219 return Offset(q);
220 }
221 virtual int GetNumberFiles() {
222 return entries_.size();
223 }
224 virtual int Finish();
225
226 private:
227 struct LocalFileEntry {
228 // Start of the local header (in the output buffer).
229 size_t local_header_offset;
230
231 // Sizes of the file entry
232 size_t uncompressed_length;
233 size_t compressed_length;
234
235 // Compression method
236 u2 compression_method;
237
238 // external attributes field
239 u4 external_attr;
240
241 // Start/length of the file_name in the local header.
242 u1 *file_name;
243 u2 file_name_length;
244
245 // Start/length of the extra_field in the local header.
246 const u1 *extra_field;
247 u2 extra_field_length;
248 };
249
250 int fd_out; // file descriptor for the output file
251
252 // OutputZipFile is responsible for maintaining the following
253 // pointers. They are allocated by the Create() method before
254 // the object is actually created using mmap.
255 u1 * const zipdata_out_; // start of output file mmap
256 u1 *q; // output cursor
257
258 u1 *header_ptr; // Current pointer to "compression method" entry.
259
260 // List of entries to write the central directory
261 std::vector<LocalFileEntry*> entries_;
262
263 // last error
264 char errmsg[4*PATH_MAX];
265
266 int error(const char *fmt, ...) {
267 va_list ap;
268 va_start(ap, fmt);
269 vsnprintf(errmsg, 4*PATH_MAX, fmt, ap);
270 va_end(ap);
271 return -1;
272 }
273
274 // Write the ZIP central directory structure for each local file
275 // entry in "entries".
276 void WriteCentralDirectory();
277
278 // Returns the offset of the pointer relative to the start of the
279 // output zip file.
280 size_t Offset(const u1 *const x) {
281 return x - zipdata_out_;
282 }
283
284 // Write ZIP file header in the output. Since the compressed size is not
285 // known in advance, it must be recorded later. This method returns a pointer
286 // to "compressed size" in the file header that should be passed to
287 // WriteFileSizeInLocalFileHeader() later.
288 u1* WriteLocalFileHeader(const char *filename, const u4 attr);
289
290 // Fill in the "compressed size" and "uncompressed size" fields in a local
291 // file header previously written by WriteLocalFileHeader().
292 size_t WriteFileSizeInLocalFileHeader(u1 *header_ptr, size_t out_length,
293 bool compress = false);
294 };
295
296 //
297 // Implementation of InputZipFile
298 //
299 bool InputZipFile::ProcessNext() {
300 // Process the next entry in the central directory. Also make sure that the
301 // content pointer is in sync.
302 size_t compressed, uncompressed;
303 u4 offset;
304 if (!ProcessCentralDirEntry(central_dir_current_, &compressed, &uncompressed,
305 filename, PATH_MAX, &attr, &offset)) {
306 return false;
307 }
308
309 // There might be an offset specified in the central directory that does
310 // not match the file offset, if so, correct the pointer.
311 if (offset != 0 && (p != (zipdata_in_ + in_offset_ + offset))) {
312 p = zipdata_in_ + offset;
313 }
314
315 if (EnsureRemaining(4, "signature") < 0) {
316 return false;
317 }
318 u4 signature = get_u4le(p);
319 if (signature == LOCAL_FILE_HEADER_SIGNATURE) {
320 if (ProcessLocalFileEntry(compressed, uncompressed) < 0) {
321 return false;
322 }
323 } else {
324 error("local file header signature for file %s not found\n", filename);
325 return false;
326 }
327
328 return true;
329 }
330
331 int InputZipFile::ProcessLocalFileEntry(
332 size_t compressed_size, size_t uncompressed_size) {
333 if (EnsureRemaining(26, "extract_version") < 0) {
334 return -1;
335 }
336 extract_version_ = get_u2le(p);
337 general_purpose_bit_flag_ = get_u2le(p);
338
339 if ((general_purpose_bit_flag_ & ~GENERAL_PURPOSE_BIT_FLAG_SUPPORTED) != 0) {
340 return error("Unsupported value (0x%04x) in general purpose bit flag.\n",
341 general_purpose_bit_flag_);
342 }
343
344 compression_method_ = get_u2le(p);
345
346 if (compression_method_ != COMPRESSION_METHOD_DEFLATED &&
347 compression_method_ != COMPRESSION_METHOD_STORED) {
348 return error("Unsupported compression method (%d).\n",
349 compression_method_);
350 }
351
352 // skip over: last_mod_file_time, last_mod_file_date, crc32
353 p += 2 + 2 + 4;
354 compressed_size_ = get_u4le(p);
355 uncompressed_size_ = get_u4le(p);
356 file_name_length_ = get_u2le(p);
357 extra_field_length_ = get_u2le(p);
358
359 if (EnsureRemaining(file_name_length_, "file_name") < 0) {
360 return -1;
361 }
362 file_name_ = p;
363 p += file_name_length_;
364
365 if (EnsureRemaining(extra_field_length_, "extra_field") < 0) {
366 return -1;
367 }
368 extra_field_ = p;
369 p += extra_field_length_;
370
371 bool is_compressed = compression_method_ == COMPRESSION_METHOD_DEFLATED;
372
373 // If the zip is compressed, compressed and uncompressed size members are
374 // zero in the local file header. If not, check that they are the same as the
375 // lengths from the central directory, otherwise, just believe the central
376 // directory
377 if (compressed_size_ == 0) {
378 compressed_size_ = compressed_size;
379 } else {
380 if (compressed_size_ != compressed_size) {
381 return error("central directory and file header inconsistent\n");
382 }
383 }
384
385 if (uncompressed_size_ == 0) {
386 uncompressed_size_ = uncompressed_size;
387 } else {
388 if (uncompressed_size_ != uncompressed_size) {
389 return error("central directory and file header inconsistent\n");
390 }
391 }
392
393 if (processor->Accept(filename, attr)) {
394 if (ProcessFile(is_compressed) < 0) {
395 return -1;
396 }
397 } else {
398 if (SkipFile(is_compressed) < 0) {
399 return -1;
400 }
401 }
402
403 if (general_purpose_bit_flag_ & GENERAL_PURPOSE_BIT_FLAG_COMPRESSED) {
404 // Skip the data descriptor. Some implementations do not put the signature
405 // here, so check if the next 4 bytes are a signature, and if so, skip the
406 // next 12 bytes (for CRC, compressed/uncompressed size), otherwise skip
407 // the next 8 bytes (because the value just read was the CRC).
408 u4 signature = get_u4le(p);
409 if (signature == DATA_DESCRIPTOR_SIGNATURE) {
410 p += 4 * 3;
411 } else {
412 p += 4 * 2;
413 }
414 }
415
416 if (p > zipdata_in_mapped_ + MAX_MAPPED_REGION) {
417 munmap(const_cast<u1 *>(zipdata_in_mapped_), MAX_MAPPED_REGION);
418 zipdata_in_mapped_ += MAX_MAPPED_REGION;
419 }
420
421 return 0;
422 }
423
424 int InputZipFile::SkipFile(const bool compressed) {
425 if (!compressed) {
426 // In this case, compressed_size_ == uncompressed_size_ (since the file is
427 // uncompressed), so we can use either.
428 if (compressed_size_ != uncompressed_size_) {
429 return error("compressed size != uncompressed size, although the file "
430 "is uncompressed.\n");
431 }
432 }
433
434 if (EnsureRemaining(compressed_size_, "file_data") < 0) {
435 return -1;
436 }
437 p += compressed_size_;
438 return 0;
439 }
440
441 u1* InputZipFile::UncompressFile() {
442 size_t in_offset = p - zipdata_in_;
443 size_t remaining = in_length_ - in_offset;
444 z_stream stream;
445
446 stream.zalloc = Z_NULL;
447 stream.zfree = Z_NULL;
448 stream.opaque = Z_NULL;
449 stream.avail_in = remaining;
450 stream.next_in = (Bytef *) p;
451
452 int ret = inflateInit2(&stream, -MAX_WBITS);
453 if (ret != Z_OK) {
454 error("inflateInit: %d\n", ret);
455 return NULL;
456 }
457
458 int uncompressed_until_now = 0;
459
460 while (true) {
461 stream.avail_out = uncompressed_data_allocated_ - uncompressed_until_now;
462 stream.next_out = uncompressed_data_ + uncompressed_until_now;
463 int old_avail_out = stream.avail_out;
464
465 ret = inflate(&stream, Z_SYNC_FLUSH);
466 int uncompressed_now = old_avail_out - stream.avail_out;
467 uncompressed_until_now += uncompressed_now;
468
469 switch (ret) {
470 case Z_STREAM_END: {
471 // zlib said that there is no more data to decompress.
472
473 u1 *new_p = reinterpret_cast<u1*>(stream.next_in);
474 compressed_size_ = new_p - p;
475 uncompressed_size_ = uncompressed_until_now;
476 p = new_p;
477 inflateEnd(&stream);
478 return uncompressed_data_;
479 }
480
481 case Z_OK: {
482 // zlib said that there is no more room in the buffer allocated for
483 // the decompressed data. Enlarge that buffer and try again.
484
485 if (uncompressed_data_allocated_ == MAX_BUFFER_SIZE) {
486 error("ijar does not support decompressing files "
487 "larger than %dMB.\n",
488 (int) (MAX_BUFFER_SIZE/(1024*1024)));
489 return NULL;
490 }
491
492 uncompressed_data_allocated_ *= 2;
493 if (uncompressed_data_allocated_ > MAX_BUFFER_SIZE) {
494 uncompressed_data_allocated_ = MAX_BUFFER_SIZE;
495 }
496
497 uncompressed_data_ = reinterpret_cast<u1*>(
498 realloc(uncompressed_data_, uncompressed_data_allocated_));
499 break;
500 }
501
502 case Z_DATA_ERROR:
503 case Z_BUF_ERROR:
504 case Z_STREAM_ERROR:
505 case Z_NEED_DICT:
506 default: {
507 error("zlib returned error code %d during inflate.\n", ret);
508 return NULL;
509 }
510 }
511 }
512 }
513
514 int InputZipFile::ProcessFile(const bool compressed) {
515 const u1 *file_data;
516 if (compressed) {
517 file_data = UncompressFile();
518 if (file_data == NULL) {
519 return -1;
520 }
521 } else {
522 // In this case, compressed_size_ == uncompressed_size_ (since the file is
523 // uncompressed), so we can use either.
524 if (compressed_size_ != uncompressed_size_) {
525 return error("compressed size != uncompressed size, although the file "
526 "is uncompressed.\n");
527 }
528
529 if (EnsureRemaining(compressed_size_, "file_data") < 0) {
530 return -1;
531 }
532 file_data = p;
533 p += compressed_size_;
534 }
535 processor->Process(filename, attr, file_data, uncompressed_size_);
536 return 0;
537 }
538
539
540 // Reads and returns some metadata of the next file from the central directory:
541 // - compressed size
542 // - uncompressed size
543 // - whether the entry is a class file (to be included in the output).
544 // Precondition: p points to the beginning of an entry in the central dir
545 // Postcondition: p points to the beginning of the next entry in the central dir
546 // Returns true if the central directory contains another file and false if not.
547 // Of course, in the latter case, the size output variables are not changed.
548 // Note that the central directory is always followed by another data structure
549 // that has a signature, so parsing it this way is safe.
550 static bool ProcessCentralDirEntry(
551 const u1 *&p, size_t *compressed_size, size_t *uncompressed_size,
552 char *filename, size_t filename_size, u4 *attr, u4 *offset) {
553 u4 signature = get_u4le(p);
554 if (signature != CENTRAL_FILE_HEADER_SIGNATURE) {
555 return false;
556 }
557
558 p += 16; // skip to 'compressed size' field
559 *compressed_size = get_u4le(p);
560 *uncompressed_size = get_u4le(p);
561 u2 file_name_length = get_u2le(p);
562 u2 extra_field_length = get_u2le(p);
563 u2 file_comment_length = get_u2le(p);
564 p += 4; // skip to external file attributes field
565 *attr = get_u4le(p);
566 *offset = get_u4le(p);
567 {
568 size_t len = (file_name_length < filename_size)
569 ? file_name_length
570 : (filename_size - 1);
571 memcpy(reinterpret_cast<void*>(filename), p, len);
572 filename[len] = 0;
573 }
574 p += file_name_length;
575 p += extra_field_length;
576 p += file_comment_length;
577 return true;
578 }
579
580 // Gives a maximum bound on the size of the interface JAR. Basically, adds
581 // the difference between the compressed and uncompressed sizes to the size
582 // of the input file.
583 u8 InputZipFile::CalculateOutputLength() {
584 const u1* current = central_dir_;
585
586 u8 compressed_size = 0;
587 u8 uncompressed_size = 0;
588 u8 skipped_compressed_size = 0;
589 u4 attr;
590 u4 offset;
591 char filename[PATH_MAX];
592
593 while (true) {
594 size_t file_compressed, file_uncompressed;
595 if (!ProcessCentralDirEntry(current,
596 &file_compressed, &file_uncompressed,
597 filename, PATH_MAX, &attr, &offset)) {
598 break;
599 }
600
601 if (processor->Accept(filename, attr)) {
602 compressed_size += (u8) file_compressed;
603 uncompressed_size += (u8) file_uncompressed;
604 } else {
605 skipped_compressed_size += file_compressed;
606 }
607 }
608
609 // The worst case is when the output is simply the input uncompressed. The
610 // metadata in the zip file will stay the same, so the file will grow by the
611 // difference between the compressed and uncompressed sizes.
612 return (u8) in_length_ - skipped_compressed_size
613 + (uncompressed_size - compressed_size);
614 }
615
616 // Given the data in the zip file, returns the offset of the central directory
617 // and the number of files contained in it.
618 bool FindZipCentralDirectory(const u1* bytes, size_t in_length,
619 u4* offset, const u1** central_dir) {
620 static const int MAX_COMMENT_LENGTH = 0xffff;
621 static const int CENTRAL_DIR_LOCATOR_SIZE = 22;
622 // Maximum distance of start of central dir locator from end of file
623 static const int MAX_DELTA = MAX_COMMENT_LENGTH + CENTRAL_DIR_LOCATOR_SIZE;
624 const u1* last_pos_to_check = in_length < MAX_DELTA
625 ? bytes
626 : bytes + (in_length - MAX_DELTA);
627 const u1* current;
628 bool found = false;
629
630 for (current = bytes + in_length - CENTRAL_DIR_LOCATOR_SIZE;
631 current >= last_pos_to_check;
632 current-- ) {
633 const u1* p = current;
634 if (get_u4le(p) != END_OF_CENTRAL_DIR_SIGNATURE) {
635 continue;
636 }
637
638 p += 16; // skip to comment length field
639 u2 comment_length = get_u2le(p);
640
641 // Does the comment go exactly till the end of the file?
642 if (current + comment_length + CENTRAL_DIR_LOCATOR_SIZE
643 != bytes + in_length) {
644 continue;
645 }
646
647 // Hooray, we found it!
648 found = true;
649 break;
650 }
651
652 if (!found) {
653 fprintf(stderr, "file is invalid or corrupted (missing end of central "
654 "directory record)\n");
655 return false;
656 }
657
658 const u1* end_of_central_dir = current;
659 get_u4le(current); // central directory locator signature, already checked
660 u2 number_of_this_disk = get_u2le(current);
661 u2 disk_with_central_dir = get_u2le(current);
662 u2 central_dir_entries_on_this_disk = get_u2le(current);
663 u2 central_dir_entries = get_u2le(current);
664 u4 central_dir_size = get_u4le(current);
665 u4 central_dir_offset = get_u4le(current);
666 u2 file_comment_length = get_u2le(current);
667 current += file_comment_length; // set current to the end of the central dir
668
669 if (number_of_this_disk != 0
670 || disk_with_central_dir != 0
671 || central_dir_entries_on_this_disk != central_dir_entries) {
672 fprintf(stderr, "multi-disk JAR files are not supported\n");
673 return false;
674 }
675
676 // Do not change output values before determining that they are OK.
677 *offset = central_dir_offset;
678 // Central directory start can then be used to determine the actual
679 // starts of the zip file (which can be different in case of a non-zip
680 // header like for auto-extractable binaries).
681 *central_dir = end_of_central_dir - central_dir_size;
682 return true;
683 }
684
685 void InputZipFile::Reset() {
686 central_dir_current_ = central_dir_;
687 zipdata_in_mapped_ = zipdata_in_;
688 p = zipdata_in_ + in_offset_;
689 }
690
691 int ZipExtractor::ProcessAll() {
692 while (ProcessNext()) {}
693 if (GetError() != NULL) {
694 return -1;
695 }
696 return 0;
697 }
698
699 ZipExtractor* ZipExtractor::Create(const char* filename,
700 ZipExtractorProcessor *processor) {
701 int fd_in = open(filename, O_RDONLY);
702 if (fd_in < 0) {
703 return NULL;
704 }
705
706 off_t length = lseek(fd_in, 0, SEEK_END);
707 if (length < 0) {
708 return NULL;
709 }
710
711 void *zipdata_in = mmap(NULL, length, PROT_READ, MAP_PRIVATE, fd_in, 0);
712 if (zipdata_in == MAP_FAILED) {
713 return NULL;
714 }
715
716 u4 central_dir_offset;
717 const u1 *central_dir = NULL;
718
719 if (!devtools_ijar::FindZipCentralDirectory(
720 static_cast<const u1*>(zipdata_in), length,
721 &central_dir_offset, &central_dir)) {
722 errno = EIO; // we don't really have a good error number
723 return NULL;
724 }
725 const u1 *zipdata_start = static_cast<const u1*>(zipdata_in);
726 off_t offset = - static_cast<off_t>(zipdata_start
727 + central_dir_offset
728 - central_dir);
729
730 return new InputZipFile(processor, fd_in, length, offset,
731 zipdata_start, central_dir);
732 }
733
734 InputZipFile::InputZipFile(ZipExtractorProcessor *processor, int fd,
735 off_t in_length, off_t in_offset,
736 const u1* zipdata_in, const u1* central_dir)
737 : processor(processor), fd_in(fd),
738 zipdata_in_(zipdata_in), zipdata_in_mapped_(zipdata_in),
739 central_dir_(central_dir), in_length_(in_length), in_offset_(in_offset),
740 p(zipdata_in + in_offset), central_dir_current_(central_dir) {
741 uncompressed_data_allocated_ = INITIAL_BUFFER_SIZE;
742 uncompressed_data_ =
743 reinterpret_cast<u1*>(malloc(uncompressed_data_allocated_));
744 errmsg[0] = 0;
745 }
746
747 InputZipFile::~InputZipFile() {
748 free(uncompressed_data_);
749 close(fd_in);
750 }
751
752
753 //
754 // Implementation of OutputZipFile
755 //
756 int OutputZipFile::WriteEmptyFile(const char *filename) {
757 const u1* file_name = (const u1*) filename;
758 size_t file_name_length = strlen(filename);
759
760 LocalFileEntry *entry = new LocalFileEntry;
761 entry->local_header_offset = Offset(q);
762 entry->external_attr = 0;
763
764 // Output the ZIP local_file_header:
765 put_u4le(q, LOCAL_FILE_HEADER_SIGNATURE);
766 put_u2le(q, 10); // extract_version
767 put_u2le(q, 0); // general_purpose_bit_flag
768 put_u2le(q, 0); // compression_method
769 put_u2le(q, 0); // last_mod_file_time
770 put_u2le(q, 0); // last_mod_file_date
771 put_u4le(q, 0); // crc32
772 put_u4le(q, 0); // compressed_size
773 put_u4le(q, 0); // uncompressed_size
774 put_u2le(q, file_name_length);
775 put_u2le(q, 0); // extra_field_length
776 put_n(q, file_name, file_name_length);
777
778 entry->file_name_length = file_name_length;
779 entry->extra_field_length = 0;
780 entry->compressed_length = 0;
781 entry->uncompressed_length = 0;
782 entry->compression_method = 0;
783 entry->extra_field = (const u1 *)"";
784 entry->file_name = (u1*) strdup((const char *) file_name);
785 entries_.push_back(entry);
786
787 return 0;
788 }
789
790 void OutputZipFile::WriteCentralDirectory() {
791 // central directory:
792 const u1 *central_directory_start = q;
793 for (size_t ii = 0; ii < entries_.size(); ++ii) {
794 LocalFileEntry *entry = entries_[ii];
795 put_u4le(q, CENTRAL_FILE_HEADER_SIGNATURE);
796 put_u2le(q, 0); // version made by
797
798 put_u2le(q, ZIP_VERSION_TO_EXTRACT); // version to extract
799 put_u2le(q, 0); // general purpose bit flag
800 put_u2le(q, entry->compression_method); // compression method:
801 put_u2le(q, 0); // last_mod_file_time
802 put_u2le(q, 0); // last_mod_file_date
803 put_u4le(q, 0); // crc32 (jar/javac tools don't care)
804 put_u4le(q, entry->compressed_length); // compressed_size
805 put_u4le(q, entry->uncompressed_length); // uncompressed_size
806 put_u2le(q, entry->file_name_length);
807 put_u2le(q, entry->extra_field_length);
808
809 put_u2le(q, 0); // file comment length
810 put_u2le(q, 0); // disk number start
811 put_u2le(q, 0); // internal file attributes
812 put_u4le(q, entry->external_attr); // external file attributes
813 // relative offset of local header:
814 put_u4le(q, entry->local_header_offset);
815
816 put_n(q, entry->file_name, entry->file_name_length);
817 put_n(q, entry->extra_field, entry->extra_field_length);
818 }
819 u4 central_directory_size = q - central_directory_start;
820
821 put_u4le(q, END_OF_CENTRAL_DIR_SIGNATURE);
822 put_u2le(q, 0); // number of this disk
823 put_u2le(q, 0); // number of the disk with the start of the central directory
824 put_u2le(q, entries_.size()); // # central dir entries on this disk
825 put_u2le(q, entries_.size()); // total # entries in the central directory
826 put_u4le(q, central_directory_size); // size of the central directory
827 put_u4le(q, Offset(central_directory_start)); // offset of start of central
828 // directory wrt starting disk
829 put_u2le(q, 0); // .ZIP file comment length
830 }
831
832 u1* OutputZipFile::WriteLocalFileHeader(const char* filename, const u4 attr) {
833 off_t file_name_length_ = strlen(filename);
834 LocalFileEntry *entry = new LocalFileEntry;
835 entry->local_header_offset = Offset(q);
836 entry->file_name_length = file_name_length_;
837 entry->file_name = new u1[file_name_length_];
838 entry->external_attr = attr;
839 memcpy(entry->file_name, filename, file_name_length_);
840 entry->extra_field_length = 0;
841 entry->extra_field = (const u1 *)"";
842
843 // Output the ZIP local_file_header:
844 put_u4le(q, LOCAL_FILE_HEADER_SIGNATURE);
845 put_u2le(q, ZIP_VERSION_TO_EXTRACT); // version to extract
846 put_u2le(q, 0); // general purpose bit flag
847 u1 *header_ptr = q;
848 put_u2le(q, COMPRESSION_METHOD_STORED); // compression method = placeholder
849 put_u2le(q, 0); // last_mod_file_time
850 put_u2le(q, 0); // last_mod_file_date
851 put_u4le(q, 0); // crc32 (jar/javac tools don't care)
852 put_u4le(q, 0); // compressed_size = placeholder
853 put_u4le(q, 0); // uncompressed_size = placeholder
854 put_u2le(q, entry->file_name_length);
855 put_u2le(q, entry->extra_field_length);
856
857 put_n(q, entry->file_name, entry->file_name_length);
858 put_n(q, entry->extra_field, entry->extra_field_length);
859 entries_.push_back(entry);
860
861 return header_ptr;
862 }
863
864 // Try to compress a file entry in memory using the deflate algorithm.
865 // It will compress buf (of size length) unless the compressed size is bigger
866 // than the input size. The result will overwrite the content of buf and the
867 // final size is returned.
868 size_t TryDeflate(u1 *buf, size_t length) {
869 u1 *outbuf = reinterpret_cast<u1 *>(malloc(length));
870 z_stream stream;
871
872 // Initialize the z_stream strcut for reading from buf and wrinting in outbuf.
873 stream.zalloc = Z_NULL;
874 stream.zfree = Z_NULL;
875 stream.opaque = Z_NULL;
876 stream.total_in = length;
877 stream.avail_in = length;
878 stream.total_out = length;
879 stream.avail_out = length;
880 stream.next_in = buf;
881 stream.next_out = outbuf;
882
883 if (deflateInit(&stream, Z_DEFAULT_COMPRESSION) != Z_OK) {
884 // Failure to compress => return the buffer uncompressed
885 free(outbuf);
886 return length;
887 }
888
889 if (deflate(&stream, Z_FINISH) == Z_STREAM_END) {
890 // Compression successful and fits in outbuf, let's copy the result in buf.
891 length = stream.total_out;
892 memcpy(buf, outbuf, length);
893 }
894
895 deflateEnd(&stream);
896 free(outbuf);
897
898 // Return the length of the resulting buffer
899 return length;
900 }
901
902 size_t OutputZipFile::WriteFileSizeInLocalFileHeader(u1 *header_ptr,
903 size_t out_length,
904 bool compress) {
905 size_t compressed_size = out_length;
906 if (compress) {
907 compressed_size = TryDeflate(q, out_length);
908 }
909 // compression method
910 if (compressed_size < out_length) {
911 put_u2le(header_ptr, COMPRESSION_METHOD_DEFLATED);
912 } else {
913 put_u2le(header_ptr, COMPRESSION_METHOD_STORED);
914 }
915 header_ptr += 8;
916 put_u4le(header_ptr, compressed_size); // compressed_size
917 put_u4le(header_ptr, out_length); // uncompressed_size
918 return compressed_size;
919 }
920
921 int OutputZipFile::Finish() {
922 if (fd_out > 0) {
923 WriteCentralDirectory();
924 if (ftruncate(fd_out, GetSize()) < 0) {
925 return error("ftruncate(fd_out, GetSize()): %s", strerror(errno));
926 }
927 if (close(fd_out) < 0) {
928 return error("close(fd_out): %s", strerror(errno));
929 }
930 fd_out = -1;
931 }
932 return 0;
933 }
934
935 u1* OutputZipFile::NewFile(const char* filename, const u4 attr) {
936 header_ptr = WriteLocalFileHeader(filename, attr);
937 return q;
938 }
939
940 int OutputZipFile::FinishFile(size_t filelength, bool compress) {
941 size_t compressed_size =
942 WriteFileSizeInLocalFileHeader(header_ptr, filelength, compress);
943 entries_.back()->compressed_length = compressed_size;
944 entries_.back()->uncompressed_length = filelength;
945 if (compressed_size < filelength) {
946 entries_.back()->compression_method = COMPRESSION_METHOD_DEFLATED;
947 } else {
948 entries_.back()->compression_method = COMPRESSION_METHOD_STORED;
949 }
950 q += compressed_size;
951 return 0;
952 }
953
954 ZipBuilder* ZipBuilder::Create(const char* zip_file, u8 estimated_size) {
955 if (estimated_size > kMaximumOutputSize) {
956 fprintf(stderr,
957 "Uncompressed input jar has size %llu, "
958 "which exceeds the maximum supported output size %llu.\n"
959 "Assuming that ijar will be smaller and hoping for the best.\n",
960 estimated_size, kMaximumOutputSize);
961 estimated_size = kMaximumOutputSize;
962 }
963
964 int fd_out = open(zip_file, O_CREAT|O_RDWR|O_TRUNC, 0644);
965 if (fd_out < 0) {
966 return NULL;
967 }
968
969 // Create mmap-able sparse file
970 if (ftruncate(fd_out, estimated_size) < 0) {
971 return NULL;
972 }
973
974 // Ensure that any buffer overflow in JarStripper will result in
975 // SIGSEGV or SIGBUS by over-allocating beyond the end of the file.
976 size_t mmap_length = std::min(estimated_size + sysconf(_SC_PAGESIZE),
977 (u8) std::numeric_limits<size_t>::max());
978
979 void *zipdata_out = mmap(NULL, mmap_length, PROT_WRITE,
980 MAP_SHARED, fd_out, 0);
981 if (zipdata_out == MAP_FAILED) {
982 fprintf(stderr, "output_length=%llu\n", estimated_size);
983 return NULL;
984 }
985
986 return new OutputZipFile(fd_out, (u1*) zipdata_out);
987 }
988
989 u8 ZipBuilder::EstimateSize(char **files) {
990 struct stat statst;
991 // Digital signature field size = 6, End of central directory = 22, Total = 28
992 u8 size = 28;
993 // Count the size of all the files in the input to estimate the size of the
994 // output.
995 for (int i = 0; files[i] != NULL; i++) {
996 if (stat(files[i], &statst) != 0) {
997 fprintf(stderr, "File %s does not seem to exist.", files[i]);
998 return 0;
999 }
1000 size += statst.st_size;
1001 // Add sizes of Zip meta data
1002 // local file header = 30 bytes
1003 // data descriptor = 12 bytes
1004 // central directory descriptor = 46 bytes
1005 // Total: 88bytes
1006 size += 88;
1007 // The filename is stored twice (once in the central directory
1008 // and once in the local file header).
1009 size += strlen(files[i]) * 2;
1010 }
1011 return size;
1012 }
1013
1014 } // namespace devtools_ijar
OLDNEW
« no previous file with comments | « third_party/ijar/zip.h ('k') | third_party/ijar/zip_main.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698