blob: 6427da970927ba1c8377fbbc55cd0a484b6d039a [file] [log] [blame]
Brian Carlstromb0460ea2011-07-29 10:08:05 -07001/*
2 * Copyright (C) 2008 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#ifndef ART_SRC_ZIP_ARCHIVE_H_
18#define ART_SRC_ZIP_ARCHIVE_H_
19
Brian Carlstromb0460ea2011-07-29 10:08:05 -070020#include <stdint.h>
21#include <sys/mman.h>
22#include <zlib.h>
23
Elliott Hughes90a33692011-08-30 13:27:07 -070024#include <map>
25
26#include "UniquePtr.h"
Brian Carlstromdb4d5402011-08-09 12:18:28 -070027#include "file.h"
Brian Carlstromb0460ea2011-07-29 10:08:05 -070028#include "globals.h"
29#include "logging.h"
Brian Carlstromdb4d5402011-08-09 12:18:28 -070030#include "mem_map.h"
Brian Carlstromb0460ea2011-07-29 10:08:05 -070031#include "stringpiece.h"
Brian Carlstrom7e93b502011-08-04 14:16:22 -070032#include "unordered_map.h"
Brian Carlstromb0460ea2011-07-29 10:08:05 -070033
34namespace art {
35
36class ZipArchive;
37class MemMap;
38
39class ZipEntry {
Brian Carlstromb0460ea2011-07-29 10:08:05 -070040 public:
Brian Carlstrom89521892011-12-07 22:05:07 -080041 bool ExtractToFile(File& file);
42 bool ExtractToMemory(MemMap& mem_map);
Brian Carlstromb0460ea2011-07-29 10:08:05 -070043
Brian Carlstrom89521892011-12-07 22:05:07 -080044 uint32_t GetUncompressedLength();
Brian Carlstromb0460ea2011-07-29 10:08:05 -070045 uint32_t GetCrc32();
46
47 private:
48
Brian Carlstroma6cc8932012-01-04 14:44:07 -080049 ZipEntry(const ZipArchive* zip_archive, const byte* ptr) : zip_archive_(zip_archive), ptr_(ptr) {}
Brian Carlstromb0460ea2011-07-29 10:08:05 -070050
51 // Zip compression methods
52 enum {
53 kCompressStored = 0, // no compression
54 kCompressDeflated = 8, // standard deflate
55 };
56
57 // kCompressStored, kCompressDeflated, ...
58 uint16_t GetCompressionMethod();
59
60 uint32_t GetCompressedLength();
61
Brian Carlstromb0460ea2011-07-29 10:08:05 -070062 // returns -1 on error
63 off_t GetDataOffset();
64
Brian Carlstroma6cc8932012-01-04 14:44:07 -080065 const ZipArchive* zip_archive_;
Brian Carlstromb0460ea2011-07-29 10:08:05 -070066
67 // pointer to zip entry within central directory
Brian Carlstromdb4d5402011-08-09 12:18:28 -070068 const byte* ptr_;
Brian Carlstromb0460ea2011-07-29 10:08:05 -070069
70 friend class ZipArchive;
71};
72
Brian Carlstromb0460ea2011-07-29 10:08:05 -070073class ZipArchive {
74 public:
75
76 // Zip file constants.
77 static const uint32_t kEOCDSignature = 0x06054b50;
78 static const int32_t kEOCDLen = 22;
79 static const int32_t kEOCDNumEntries = 8; // offset to #of entries in file
80 static const int32_t kEOCDSize = 12; // size of the central directory
81 static const int32_t kEOCDFileOffset = 16; // offset to central directory
82
83 static const int32_t kMaxCommentLen = 65535; // longest possible in uint16_t
84 static const int32_t kMaxEOCDSearch = (kMaxCommentLen + kEOCDLen);
85
86 static const uint32_t kLFHSignature = 0x04034b50;
87 static const int32_t kLFHLen = 30; // excluding variable-len fields
88 static const int32_t kLFHNameLen = 26; // offset to filename length
89 static const int32_t kLFHExtraLen = 28; // offset to extra length
90
91 static const uint32_t kCDESignature = 0x02014b50;
92 static const int32_t kCDELen = 46; // excluding variable-len fields
93 static const int32_t kCDEMethod = 10; // offset to compression method
94 static const int32_t kCDEModWhen = 12; // offset to modification timestamp
95 static const int32_t kCDECRC = 16; // offset to entry CRC
96 static const int32_t kCDECompLen = 20; // offset to compressed length
97 static const int32_t kCDEUncompLen = 24; // offset to uncompressed length
98 static const int32_t kCDENameLen = 28; // offset to filename length
99 static const int32_t kCDEExtraLen = 30; // offset to extra length
100 static const int32_t kCDECommentLen = 32; // offset to comment length
101 static const int32_t kCDELocalOffset = 42; // offset to local hdr
102
Brian Carlstromb7bbba42011-10-13 14:58:47 -0700103 // return new ZipArchive instance on success, NULL on error.
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700104 static ZipArchive* Open(const std::string& filename);
Brian Carlstroma6cc8932012-01-04 14:44:07 -0800105 static ZipArchive* OpenFromFd(int fd);
Brian Carlstromb7bbba42011-10-13 14:58:47 -0700106
Brian Carlstroma6cc8932012-01-04 14:44:07 -0800107 ZipEntry* Find(const char* name) const;
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700108
109 ~ZipArchive() {
110 Close();
111 }
112
113 private:
Elliott Hughesa51a3dd2011-10-17 15:19:26 -0700114 explicit ZipArchive(int fd) : fd_(fd), num_entries_(0), dir_offset_(0) {}
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700115
116 bool MapCentralDirectory();
117 bool Parse();
118 void Close();
119
120 int fd_;
121 uint16_t num_entries_;
122 off_t dir_offset_;
Elliott Hughes90a33692011-08-30 13:27:07 -0700123 UniquePtr<MemMap> dir_map_;
Brian Carlstroma663ea52011-08-19 23:33:41 -0700124 typedef std::tr1::unordered_map<StringPiece, const byte*, StringPieceHash> DirEntries;
Brian Carlstrom7e93b502011-08-04 14:16:22 -0700125 DirEntries dir_entries_;
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700126
127 friend class ZipEntry;
128};
129
130} // namespace art
131
132#endif // ART_SRC_ZIP_ARCHIVE_H_