zip_archive.h revision 90a3369d3b6238f1a4c9b19ca68978dab1c39bc4
1b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom/* 2b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * Copyright (C) 2008 The Android Open Source Project 3b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * 4b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * Licensed under the Apache License, Version 2.0 (the "License"); 5b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * you may not use this file except in compliance with the License. 6b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * You may obtain a copy of the License at 7b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * 8b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * http://www.apache.org/licenses/LICENSE-2.0 9b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * 10b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * Unless required by applicable law or agreed to in writing, software 11b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * distributed under the License is distributed on an "AS IS" BASIS, 12b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * See the License for the specific language governing permissions and 14b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom * limitations under the License. 15b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom */ 16b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 17b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#ifndef ART_SRC_ZIP_ARCHIVE_H_ 18b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#define ART_SRC_ZIP_ARCHIVE_H_ 19b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 20b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#include <stdint.h> 21b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#include <sys/mman.h> 22b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#include <zlib.h> 23b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 2490a3369d3b6238f1a4c9b19ca68978dab1c39bc4Elliott Hughes#include <map> 2590a3369d3b6238f1a4c9b19ca68978dab1c39bc4Elliott Hughes 2690a3369d3b6238f1a4c9b19ca68978dab1c39bc4Elliott Hughes#include "UniquePtr.h" 27db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "file.h" 28b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#include "globals.h" 29b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#include "logging.h" 30db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom#include "mem_map.h" 31b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#include "stringpiece.h" 327e93b50433cde2a44d99212e8040299bde498546Brian Carlstrom#include "unordered_map.h" 33b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 34b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstromnamespace art { 35b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 36b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstromclass ZipArchive; 37b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstromclass MemMap; 38b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 39b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstromclass ZipEntry { 40b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 41b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom public: 42b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom // Uncompress an entry, in its entirety, to an open file descriptor. 43db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom bool Extract(File& file); 44b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 45b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom uint32_t GetCrc32(); 46b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 47b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom private: 48b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 49db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom ZipEntry(ZipArchive* zip_archive, const byte* ptr) : zip_archive_(zip_archive), ptr_(ptr) {}; 50b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 51b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom // Zip compression methods 52b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom enum { 53b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom kCompressStored = 0, // no compression 54b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom kCompressDeflated = 8, // standard deflate 55b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom }; 56b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 57b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom // kCompressStored, kCompressDeflated, ... 58b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom uint16_t GetCompressionMethod(); 59b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 60b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom uint32_t GetCompressedLength(); 61b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 62b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom uint32_t GetUncompressedLength(); 63b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 64b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom // returns -1 on error 65b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom off_t GetDataOffset(); 66b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 67b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom ZipArchive* zip_archive_; 68b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 69b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom // pointer to zip entry within central directory 70db4d54081f09abcbe97ffdf615874f2809a9e777Brian Carlstrom const byte* ptr_; 71b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 72b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom friend class ZipArchive; 73b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom}; 74b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 75b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstromclass ZipArchive { 76b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom public: 77b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 78b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom // Zip file constants. 79b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const uint32_t kEOCDSignature = 0x06054b50; 80b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kEOCDLen = 22; 81b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kEOCDNumEntries = 8; // offset to #of entries in file 82b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kEOCDSize = 12; // size of the central directory 83b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kEOCDFileOffset = 16; // offset to central directory 84b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 85b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kMaxCommentLen = 65535; // longest possible in uint16_t 86b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kMaxEOCDSearch = (kMaxCommentLen + kEOCDLen); 87b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 88b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const uint32_t kLFHSignature = 0x04034b50; 89b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kLFHLen = 30; // excluding variable-len fields 90b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kLFHNameLen = 26; // offset to filename length 91b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kLFHExtraLen = 28; // offset to extra length 92b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 93b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const uint32_t kCDESignature = 0x02014b50; 94b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDELen = 46; // excluding variable-len fields 95b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDEMethod = 10; // offset to compression method 96b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDEModWhen = 12; // offset to modification timestamp 97b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDECRC = 16; // offset to entry CRC 98b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDECompLen = 20; // offset to compressed length 99b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDEUncompLen = 24; // offset to uncompressed length 100b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDENameLen = 28; // offset to filename length 101b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDEExtraLen = 30; // offset to extra length 102b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDECommentLen = 32; // offset to comment length 103b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static const int32_t kCDELocalOffset = 42; // offset to local hdr 104b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 105b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom static ZipArchive* Open(const std::string& filename); 106b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom ZipEntry* Find(const char * name); 107b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 108b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom ~ZipArchive() { 109b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom Close(); 110b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom } 111b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 112b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom private: 113b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom ZipArchive(int fd) : fd_(fd), num_entries_(0), dir_offset_(0) {} 114b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 115b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom bool MapCentralDirectory(); 116b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom bool Parse(); 117b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom void Close(); 118b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 119b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom int fd_; 120b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom uint16_t num_entries_; 121b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom off_t dir_offset_; 12290a3369d3b6238f1a4c9b19ca68978dab1c39bc4Elliott Hughes UniquePtr<MemMap> dir_map_; 123a663ea5de4c9ab6b1510fdebd6d8eca77ba699aeBrian Carlstrom typedef std::tr1::unordered_map<StringPiece, const byte*, StringPieceHash> DirEntries; 1247e93b50433cde2a44d99212e8040299bde498546Brian Carlstrom DirEntries dir_entries_; 125b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 126b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom friend class ZipEntry; 127b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom}; 128b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 129b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom} // namespace art 130b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom 131b0460eaa2cb131f1dbdd5a7217bd36b9a9f1b995Brian Carlstrom#endif // ART_SRC_ZIP_ARCHIVE_H_ 132