blob: cd79bb61f3fbd0c17a661060198cfb3101bc06b5 [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#include "zip_archive.h"
18
19#include <fcntl.h>
Ian Rogers8d31bbd2013-10-13 10:44:14 -070020#include <stdio.h>
Brian Carlstromb0460ea2011-07-29 10:08:05 -070021#include <sys/stat.h>
22#include <sys/types.h>
23#include <unistd.h>
Ian Rogers700a4022014-05-19 16:49:03 -070024#include <vector>
Brian Carlstromb0460ea2011-07-29 10:08:05 -070025
Elliott Hughes76160052012-12-12 16:31:20 -080026#include "base/unix_file/fd_file.h"
Elliott Hughes90a33692011-08-30 13:27:07 -070027
Brian Carlstromb0460ea2011-07-29 10:08:05 -070028namespace art {
29
Brian Carlstromb0460ea2011-07-29 10:08:05 -070030uint32_t ZipEntry::GetUncompressedLength() {
Narayan Kamath92572be2013-11-28 14:06:24 +000031 return zip_entry_->uncompressed_length;
Brian Carlstromb0460ea2011-07-29 10:08:05 -070032}
33
34uint32_t ZipEntry::GetCrc32() {
Narayan Kamath92572be2013-11-28 14:06:24 +000035 return zip_entry_->crc32;
Brian Carlstromb0460ea2011-07-29 10:08:05 -070036}
37
Mathieu Chartier661974a2014-01-09 11:23:53 -080038ZipEntry::~ZipEntry() {
39 delete zip_entry_;
40}
Brian Carlstromb0460ea2011-07-29 10:08:05 -070041
Ian Rogers8d31bbd2013-10-13 10:44:14 -070042bool ZipEntry::ExtractToFile(File& file, std::string* error_msg) {
Narayan Kamath92572be2013-11-28 14:06:24 +000043 const int32_t error = ExtractEntryToFile(handle_, zip_entry_, file.Fd());
44 if (error) {
45 *error_msg = std::string(ErrorCodeString(error));
Brian Carlstrom89521892011-12-07 22:05:07 -080046 return false;
47 }
48
Narayan Kamath92572be2013-11-28 14:06:24 +000049 return true;
Brian Carlstromb0460ea2011-07-29 10:08:05 -070050}
51
Brian Carlstrom0aa504b2014-05-23 02:47:28 -070052MemMap* ZipEntry::ExtractToMemMap(const char* zip_filename, const char* entry_filename,
53 std::string* error_msg) {
Brian Carlstrom4922e9d2013-07-09 17:18:47 -070054 std::string name(entry_filename);
55 name += " extracted in memory from ";
Brian Carlstrom0aa504b2014-05-23 02:47:28 -070056 name += zip_filename;
Ian Rogers700a4022014-05-19 16:49:03 -070057 std::unique_ptr<MemMap> map(MemMap::MapAnonymous(name.c_str(),
Mathieu Chartier2cebb242015-04-21 16:50:40 -070058 nullptr, GetUncompressedLength(),
Vladimir Marko5c42c292015-02-25 12:02:49 +000059 PROT_READ | PROT_WRITE, false, false,
60 error_msg));
Ian Rogers8d31bbd2013-10-13 10:44:14 -070061 if (map.get() == nullptr) {
62 DCHECK(!error_msg->empty());
Narayan Kamath92572be2013-11-28 14:06:24 +000063 return nullptr;
Brian Carlstrom4922e9d2013-07-09 17:18:47 -070064 }
65
Narayan Kamath92572be2013-11-28 14:06:24 +000066 const int32_t error = ExtractToMemory(handle_, zip_entry_,
67 map->Begin(), map->Size());
68 if (error) {
69 *error_msg = std::string(ErrorCodeString(error));
70 return nullptr;
Brian Carlstrom4922e9d2013-07-09 17:18:47 -070071 }
72
73 return map.release();
74}
75
Elliott Hughesad6c9c32012-01-19 17:39:12 -080076static void SetCloseOnExec(int fd) {
77 // This dance is more portable than Linux's O_CLOEXEC open(2) flag.
78 int flags = fcntl(fd, F_GETFD);
79 if (flags == -1) {
80 PLOG(WARNING) << "fcntl(" << fd << ", F_GETFD) failed";
81 return;
82 }
83 int rc = fcntl(fd, F_SETFD, flags | FD_CLOEXEC);
84 if (rc == -1) {
85 PLOG(WARNING) << "fcntl(" << fd << ", F_SETFD, " << flags << ") failed";
86 return;
87 }
88}
89
Ian Rogers8d31bbd2013-10-13 10:44:14 -070090ZipArchive* ZipArchive::Open(const char* filename, std::string* error_msg) {
91 DCHECK(filename != nullptr);
Narayan Kamath92572be2013-11-28 14:06:24 +000092
93 ZipArchiveHandle handle;
94 const int32_t error = OpenArchive(filename, &handle);
95 if (error) {
96 *error_msg = std::string(ErrorCodeString(error));
97 CloseArchive(handle);
98 return nullptr;
Brian Carlstromb0460ea2011-07-29 10:08:05 -070099 }
Narayan Kamath92572be2013-11-28 14:06:24 +0000100
101 SetCloseOnExec(GetFileDescriptor(handle));
102 return new ZipArchive(handle);
Brian Carlstromb7bbba42011-10-13 14:58:47 -0700103}
104
Ian Rogers8d31bbd2013-10-13 10:44:14 -0700105ZipArchive* ZipArchive::OpenFromFd(int fd, const char* filename, std::string* error_msg) {
Narayan Kamath92572be2013-11-28 14:06:24 +0000106 DCHECK(filename != nullptr);
107 DCHECK_GT(fd, 0);
108
109 ZipArchiveHandle handle;
110 const int32_t error = OpenArchiveFd(fd, filename, &handle);
111 if (error) {
112 *error_msg = std::string(ErrorCodeString(error));
113 CloseArchive(handle);
114 return nullptr;
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700115 }
Narayan Kamath92572be2013-11-28 14:06:24 +0000116
117 SetCloseOnExec(GetFileDescriptor(handle));
118 return new ZipArchive(handle);
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700119}
120
Narayan Kamath92572be2013-11-28 14:06:24 +0000121ZipEntry* ZipArchive::Find(const char* name, std::string* error_msg) const {
122 DCHECK(name != nullptr);
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700123
Narayan Kamath92572be2013-11-28 14:06:24 +0000124 // Resist the urge to delete the space. <: is a bigraph sequence.
Ian Rogers700a4022014-05-19 16:49:03 -0700125 std::unique_ptr< ::ZipEntry> zip_entry(new ::ZipEntry);
Yusuke Sato64db62d2015-06-25 14:56:49 -0700126 const int32_t error = FindEntry(handle_, ZipString(name), zip_entry.get());
Narayan Kamath92572be2013-11-28 14:06:24 +0000127 if (error) {
128 *error_msg = std::string(ErrorCodeString(error));
129 return nullptr;
Kenny Root72fcca22013-09-19 09:25:34 -0700130 }
131
Narayan Kamath92572be2013-11-28 14:06:24 +0000132 return new ZipEntry(handle_, zip_entry.release());
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700133}
134
Vladimir Marko9bdf1082016-01-21 12:15:52 +0000135ZipArchive::~ZipArchive() {
136 CloseArchive(handle_);
137}
138
Brian Carlstromb0460ea2011-07-29 10:08:05 -0700139} // namespace art