Index: third_party/zlib/google/zip_reader.cc |
diff --git a/third_party/zlib/google/zip_reader.cc b/third_party/zlib/google/zip_reader.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a962fa10d7cf5db9663d008914ff0b096ca09d17 |
--- /dev/null |
+++ b/third_party/zlib/google/zip_reader.cc |
@@ -0,0 +1,538 @@ |
+// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "third_party/zlib/google/zip_reader.h" |
+ |
+#include <utility> |
+ |
+#include "base/bind.h" |
+#include "base/files/file.h" |
+#include "base/logging.h" |
+#include "base/macros.h" |
+#include "base/message_loop/message_loop.h" |
+#include "base/strings/string_util.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "base/threading/thread_task_runner_handle.h" |
+#include "build/build_config.h" |
+#include "third_party/zlib/google/zip_internal.h" |
+ |
+#if defined(USE_SYSTEM_MINIZIP) |
+#include <minizip/unzip.h> |
+#else |
+#include "third_party/zlib/contrib/minizip/unzip.h" |
+#if defined(OS_WIN) |
+#include "third_party/zlib/contrib/minizip/iowin32.h" |
+#endif // defined(OS_WIN) |
+#endif // defined(USE_SYSTEM_MINIZIP) |
+ |
+namespace zip { |
+ |
+namespace { |
+ |
+// FilePathWriterDelegate ------------------------------------------------------ |
+ |
+// A writer delegate that writes a file at a given path. |
+class FilePathWriterDelegate : public WriterDelegate { |
+ public: |
+ explicit FilePathWriterDelegate(const base::FilePath& output_file_path); |
+ ~FilePathWriterDelegate() override; |
+ |
+ // WriterDelegate methods: |
+ |
+ // Creates the output file and any necessary intermediate directories. |
+ bool PrepareOutput() override; |
+ |
+ // Writes |num_bytes| bytes of |data| to the file, returning false if not all |
+ // bytes could be written. |
+ bool WriteBytes(const char* data, int num_bytes) override; |
+ |
+ private: |
+ base::FilePath output_file_path_; |
+ base::File file_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FilePathWriterDelegate); |
+}; |
+ |
+FilePathWriterDelegate::FilePathWriterDelegate( |
+ const base::FilePath& output_file_path) |
+ : output_file_path_(output_file_path) { |
+} |
+ |
+FilePathWriterDelegate::~FilePathWriterDelegate() { |
+} |
+ |
+bool FilePathWriterDelegate::PrepareOutput() { |
+ // We can't rely on parent directory entries being specified in the |
+ // zip, so we make sure they are created. |
+ if (!base::CreateDirectory(output_file_path_.DirName())) |
+ return false; |
+ |
+ file_.Initialize(output_file_path_, |
+ base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE); |
+ return file_.IsValid(); |
+} |
+ |
+bool FilePathWriterDelegate::WriteBytes(const char* data, int num_bytes) { |
+ return num_bytes == file_.WriteAtCurrentPos(data, num_bytes); |
+} |
+ |
+ |
+// StringWriterDelegate -------------------------------------------------------- |
+ |
+// A writer delegate that writes no more than |max_read_bytes| to a given |
+// std::string. |
+class StringWriterDelegate : public WriterDelegate { |
+ public: |
+ StringWriterDelegate(size_t max_read_bytes, std::string* output); |
+ ~StringWriterDelegate() override; |
+ |
+ // WriterDelegate methods: |
+ |
+ // Returns true. |
+ bool PrepareOutput() override; |
+ |
+ // Appends |num_bytes| bytes from |data| to the output string. Returns false |
+ // if |num_bytes| will cause the string to exceed |max_read_bytes|. |
+ bool WriteBytes(const char* data, int num_bytes) override; |
+ |
+ private: |
+ size_t max_read_bytes_; |
+ std::string* output_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(StringWriterDelegate); |
+}; |
+ |
+StringWriterDelegate::StringWriterDelegate(size_t max_read_bytes, |
+ std::string* output) |
+ : max_read_bytes_(max_read_bytes), |
+ output_(output) { |
+} |
+ |
+StringWriterDelegate::~StringWriterDelegate() { |
+} |
+ |
+bool StringWriterDelegate::PrepareOutput() { |
+ return true; |
+} |
+ |
+bool StringWriterDelegate::WriteBytes(const char* data, int num_bytes) { |
+ if (output_->size() + num_bytes > max_read_bytes_) |
+ return false; |
+ output_->append(data, num_bytes); |
+ return true; |
+} |
+ |
+} // namespace |
+ |
+// TODO(satorux): The implementation assumes that file names in zip files |
+// are encoded in UTF-8. This is true for zip files created by Zip() |
+// function in zip.h, but not true for user-supplied random zip files. |
+ZipReader::EntryInfo::EntryInfo(const std::string& file_name_in_zip, |
+ const unz_file_info& raw_file_info) |
+ : file_path_(base::FilePath::FromUTF8Unsafe(file_name_in_zip)), |
+ is_directory_(false) { |
+ original_size_ = raw_file_info.uncompressed_size; |
+ |
+ // Directory entries in zip files end with "/". |
+ is_directory_ = base::EndsWith(file_name_in_zip, "/", |
+ base::CompareCase::INSENSITIVE_ASCII); |
+ |
+ // Check the file name here for directory traversal issues. |
+ is_unsafe_ = file_path_.ReferencesParent(); |
+ |
+ // We also consider that the file name is unsafe, if it's invalid UTF-8. |
+ base::string16 file_name_utf16; |
+ if (!base::UTF8ToUTF16(file_name_in_zip.data(), file_name_in_zip.size(), |
+ &file_name_utf16)) { |
+ is_unsafe_ = true; |
+ } |
+ |
+ // We also consider that the file name is unsafe, if it's absolute. |
+ // On Windows, IsAbsolute() returns false for paths starting with "/". |
+ if (file_path_.IsAbsolute() || |
+ base::StartsWith(file_name_in_zip, "/", |
+ base::CompareCase::INSENSITIVE_ASCII)) |
+ is_unsafe_ = true; |
+ |
+ // Construct the last modified time. The timezone info is not present in |
+ // zip files, so we construct the time as local time. |
+ base::Time::Exploded exploded_time = {}; // Zero-clear. |
+ exploded_time.year = raw_file_info.tmu_date.tm_year; |
+ // The month in zip file is 0-based, whereas ours is 1-based. |
+ exploded_time.month = raw_file_info.tmu_date.tm_mon + 1; |
+ exploded_time.day_of_month = raw_file_info.tmu_date.tm_mday; |
+ exploded_time.hour = raw_file_info.tmu_date.tm_hour; |
+ exploded_time.minute = raw_file_info.tmu_date.tm_min; |
+ exploded_time.second = raw_file_info.tmu_date.tm_sec; |
+ exploded_time.millisecond = 0; |
+ if (exploded_time.HasValidValues()) { |
+ last_modified_ = base::Time::FromLocalExploded(exploded_time); |
+ } else { |
+ // Use Unix time epoch if the time stamp data is invalid. |
+ last_modified_ = base::Time::UnixEpoch(); |
+ } |
+} |
+ |
+ZipReader::ZipReader() |
+ : weak_ptr_factory_(this) { |
+ Reset(); |
+} |
+ |
+ZipReader::~ZipReader() { |
+ Close(); |
+} |
+ |
+bool ZipReader::Open(const base::FilePath& zip_file_path) { |
+ DCHECK(!zip_file_); |
+ |
+ // Use of "Unsafe" function does not look good, but there is no way to do |
+ // this safely on Linux. See file_util.h for details. |
+ zip_file_ = internal::OpenForUnzipping(zip_file_path.AsUTF8Unsafe()); |
+ if (!zip_file_) { |
+ return false; |
+ } |
+ |
+ return OpenInternal(); |
+} |
+ |
+bool ZipReader::OpenFromPlatformFile(base::PlatformFile zip_fd) { |
+ DCHECK(!zip_file_); |
+ |
+#if defined(OS_POSIX) |
+ zip_file_ = internal::OpenFdForUnzipping(zip_fd); |
+#elif defined(OS_WIN) |
+ zip_file_ = internal::OpenHandleForUnzipping(zip_fd); |
+#endif |
+ if (!zip_file_) { |
+ return false; |
+ } |
+ |
+ return OpenInternal(); |
+} |
+ |
+bool ZipReader::OpenFromString(const std::string& data) { |
+ zip_file_ = internal::PrepareMemoryForUnzipping(data); |
+ if (!zip_file_) |
+ return false; |
+ return OpenInternal(); |
+} |
+ |
+void ZipReader::Close() { |
+ if (zip_file_) { |
+ unzClose(zip_file_); |
+ } |
+ Reset(); |
+} |
+ |
+bool ZipReader::HasMore() { |
+ return !reached_end_; |
+} |
+ |
+bool ZipReader::AdvanceToNextEntry() { |
+ DCHECK(zip_file_); |
+ |
+ // Should not go further if we already reached the end. |
+ if (reached_end_) |
+ return false; |
+ |
+ unz_file_pos position = {}; |
+ if (unzGetFilePos(zip_file_, &position) != UNZ_OK) |
+ return false; |
+ const int current_entry_index = position.num_of_file; |
+ // If we are currently at the last entry, then the next position is the |
+ // end of the zip file, so mark that we reached the end. |
+ if (current_entry_index + 1 == num_entries_) { |
+ reached_end_ = true; |
+ } else { |
+ DCHECK_LT(current_entry_index + 1, num_entries_); |
+ if (unzGoToNextFile(zip_file_) != UNZ_OK) { |
+ return false; |
+ } |
+ } |
+ current_entry_info_.reset(); |
+ return true; |
+} |
+ |
+bool ZipReader::OpenCurrentEntryInZip() { |
+ DCHECK(zip_file_); |
+ |
+ unz_file_info raw_file_info = {}; |
+ char raw_file_name_in_zip[internal::kZipMaxPath] = {}; |
+ const int result = unzGetCurrentFileInfo(zip_file_, |
+ &raw_file_info, |
+ raw_file_name_in_zip, |
+ sizeof(raw_file_name_in_zip) - 1, |
+ NULL, // extraField. |
+ 0, // extraFieldBufferSize. |
+ NULL, // szComment. |
+ 0); // commentBufferSize. |
+ if (result != UNZ_OK) |
+ return false; |
+ if (raw_file_name_in_zip[0] == '\0') |
+ return false; |
+ current_entry_info_.reset( |
+ new EntryInfo(raw_file_name_in_zip, raw_file_info)); |
+ return true; |
+} |
+ |
+bool ZipReader::LocateAndOpenEntry(const base::FilePath& path_in_zip) { |
+ DCHECK(zip_file_); |
+ |
+ current_entry_info_.reset(); |
+ reached_end_ = false; |
+ const int kDefaultCaseSensivityOfOS = 0; |
+ const int result = unzLocateFile(zip_file_, |
+ path_in_zip.AsUTF8Unsafe().c_str(), |
+ kDefaultCaseSensivityOfOS); |
+ if (result != UNZ_OK) |
+ return false; |
+ |
+ // Then Open the entry. |
+ return OpenCurrentEntryInZip(); |
+} |
+ |
+bool ZipReader::ExtractCurrentEntry(WriterDelegate* delegate) const { |
+ DCHECK(zip_file_); |
+ |
+ const int open_result = unzOpenCurrentFile(zip_file_); |
+ if (open_result != UNZ_OK) |
+ return false; |
+ |
+ if (!delegate->PrepareOutput()) |
+ return false; |
+ |
+ bool success = true; // This becomes false when something bad happens. |
+ std::unique_ptr<char[]> buf(new char[internal::kZipBufSize]); |
+ while (true) { |
+ const int num_bytes_read = unzReadCurrentFile(zip_file_, buf.get(), |
+ internal::kZipBufSize); |
+ if (num_bytes_read == 0) { |
+ // Reached the end of the file. |
+ break; |
+ } else if (num_bytes_read < 0) { |
+ // If num_bytes_read < 0, then it's a specific UNZ_* error code. |
+ success = false; |
+ break; |
+ } else if (num_bytes_read > 0) { |
+ // Some data is read. |
+ if (!delegate->WriteBytes(buf.get(), num_bytes_read)) { |
+ success = false; |
+ break; |
+ } |
+ } |
+ } |
+ |
+ unzCloseCurrentFile(zip_file_); |
+ |
+ return success; |
+} |
+ |
+bool ZipReader::ExtractCurrentEntryToFilePath( |
+ const base::FilePath& output_file_path) const { |
+ DCHECK(zip_file_); |
+ |
+ // If this is a directory, just create it and return. |
+ if (current_entry_info()->is_directory()) |
+ return base::CreateDirectory(output_file_path); |
+ |
+ bool success = false; |
+ { |
+ FilePathWriterDelegate writer(output_file_path); |
+ success = ExtractCurrentEntry(&writer); |
+ } |
+ |
+ if (success && |
+ current_entry_info()->last_modified() != base::Time::UnixEpoch()) { |
+ base::TouchFile(output_file_path, |
+ base::Time::Now(), |
+ current_entry_info()->last_modified()); |
+ } |
+ |
+ return success; |
+} |
+ |
+void ZipReader::ExtractCurrentEntryToFilePathAsync( |
+ const base::FilePath& output_file_path, |
+ const SuccessCallback& success_callback, |
+ const FailureCallback& failure_callback, |
+ const ProgressCallback& progress_callback) { |
+ DCHECK(zip_file_); |
+ DCHECK(current_entry_info_.get()); |
+ |
+ // If this is a directory, just create it and return. |
+ if (current_entry_info()->is_directory()) { |
+ if (base::CreateDirectory(output_file_path)) { |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, success_callback); |
+ } else { |
+ DVLOG(1) << "Unzip failed: unable to create directory."; |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, failure_callback); |
+ } |
+ return; |
+ } |
+ |
+ if (unzOpenCurrentFile(zip_file_) != UNZ_OK) { |
+ DVLOG(1) << "Unzip failed: unable to open current zip entry."; |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, failure_callback); |
+ return; |
+ } |
+ |
+ base::FilePath output_dir_path = output_file_path.DirName(); |
+ if (!base::CreateDirectory(output_dir_path)) { |
+ DVLOG(1) << "Unzip failed: unable to create containing directory."; |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, failure_callback); |
+ return; |
+ } |
+ |
+ const int flags = base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE; |
+ base::File output_file(output_file_path, flags); |
+ |
+ if (!output_file.IsValid()) { |
+ DVLOG(1) << "Unzip failed: unable to create platform file at " |
+ << output_file_path.value(); |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, failure_callback); |
+ return; |
+ } |
+ |
+ base::MessageLoop::current()->PostTask( |
+ FROM_HERE, |
+ base::Bind(&ZipReader::ExtractChunk, weak_ptr_factory_.GetWeakPtr(), |
+ Passed(std::move(output_file)), success_callback, |
+ failure_callback, progress_callback, 0 /* initial offset */)); |
+} |
+ |
+bool ZipReader::ExtractCurrentEntryIntoDirectory( |
+ const base::FilePath& output_directory_path) const { |
+ DCHECK(current_entry_info_.get()); |
+ |
+ base::FilePath output_file_path = output_directory_path.Append( |
+ current_entry_info()->file_path()); |
+ return ExtractCurrentEntryToFilePath(output_file_path); |
+} |
+ |
+bool ZipReader::ExtractCurrentEntryToFile(base::File* file) const { |
+ DCHECK(zip_file_); |
+ |
+ // If this is a directory, there's nothing to extract to the file, so return |
+ // false. |
+ if (current_entry_info()->is_directory()) |
+ return false; |
+ |
+ FileWriterDelegate writer(file); |
+ return ExtractCurrentEntry(&writer); |
+} |
+ |
+bool ZipReader::ExtractCurrentEntryToString(size_t max_read_bytes, |
+ std::string* output) const { |
+ DCHECK(output); |
+ DCHECK(zip_file_); |
+ DCHECK_NE(0U, max_read_bytes); |
+ |
+ if (current_entry_info()->is_directory()) { |
+ output->clear(); |
+ return true; |
+ } |
+ |
+ // The original_size() is the best hint for the real size, so it saves |
+ // doing reallocations for the common case when the uncompressed size is |
+ // correct. However, we need to assume that the uncompressed size could be |
+ // incorrect therefore this function needs to read as much data as possible. |
+ std::string contents; |
+ contents.reserve( |
+ static_cast<size_t>(std::min(static_cast<int64_t>(max_read_bytes), |
+ current_entry_info()->original_size()))); |
+ |
+ StringWriterDelegate writer(max_read_bytes, &contents); |
+ if (!ExtractCurrentEntry(&writer)) |
+ return false; |
+ output->swap(contents); |
+ return true; |
+} |
+ |
+bool ZipReader::OpenInternal() { |
+ DCHECK(zip_file_); |
+ |
+ unz_global_info zip_info = {}; // Zero-clear. |
+ if (unzGetGlobalInfo(zip_file_, &zip_info) != UNZ_OK) { |
+ return false; |
+ } |
+ num_entries_ = zip_info.number_entry; |
+ if (num_entries_ < 0) |
+ return false; |
+ |
+ // We are already at the end if the zip file is empty. |
+ reached_end_ = (num_entries_ == 0); |
+ return true; |
+} |
+ |
+void ZipReader::Reset() { |
+ zip_file_ = NULL; |
+ num_entries_ = 0; |
+ reached_end_ = false; |
+ current_entry_info_.reset(); |
+} |
+ |
+void ZipReader::ExtractChunk(base::File output_file, |
+ const SuccessCallback& success_callback, |
+ const FailureCallback& failure_callback, |
+ const ProgressCallback& progress_callback, |
+ const int64_t offset) { |
+ char buffer[internal::kZipBufSize]; |
+ |
+ const int num_bytes_read = unzReadCurrentFile(zip_file_, |
+ buffer, |
+ internal::kZipBufSize); |
+ |
+ if (num_bytes_read == 0) { |
+ unzCloseCurrentFile(zip_file_); |
+ success_callback.Run(); |
+ } else if (num_bytes_read < 0) { |
+ DVLOG(1) << "Unzip failed: error while reading zipfile " |
+ << "(" << num_bytes_read << ")"; |
+ failure_callback.Run(); |
+ } else { |
+ if (num_bytes_read != output_file.Write(offset, buffer, num_bytes_read)) { |
+ DVLOG(1) << "Unzip failed: unable to write all bytes to target."; |
+ failure_callback.Run(); |
+ return; |
+ } |
+ |
+ int64_t current_progress = offset + num_bytes_read; |
+ |
+ progress_callback.Run(current_progress); |
+ |
+ base::MessageLoop::current()->PostTask( |
+ FROM_HERE, |
+ base::Bind(&ZipReader::ExtractChunk, weak_ptr_factory_.GetWeakPtr(), |
+ Passed(std::move(output_file)), success_callback, |
+ failure_callback, progress_callback, current_progress)); |
+ } |
+} |
+ |
+// FileWriterDelegate ---------------------------------------------------------- |
+ |
+FileWriterDelegate::FileWriterDelegate(base::File* file) |
+ : file_(file), |
+ file_length_(0) { |
+} |
+ |
+FileWriterDelegate::~FileWriterDelegate() { |
+#if !defined(NDEBUG) || defined(DCHECK_ALWAYS_ON) |
+ const bool success = |
+#endif |
+ file_->SetLength(file_length_); |
+ DPLOG_IF(ERROR, !success) << "Failed updating length of written file"; |
+} |
+ |
+bool FileWriterDelegate::PrepareOutput() { |
+ return file_->Seek(base::File::FROM_BEGIN, 0) >= 0; |
+} |
+ |
+bool FileWriterDelegate::WriteBytes(const char* data, int num_bytes) { |
+ int bytes_written = file_->WriteAtCurrentPos(data, num_bytes); |
+ if (bytes_written > 0) |
+ file_length_ += bytes_written; |
+ return bytes_written == num_bytes; |
+} |
+ |
+} // namespace zip |