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

Unified Diff: third_party/zlib/google/zip_reader.cc

Issue 2023703002: Beginning work on GN build (Closed) Base URL: git@github.com:dart-lang/sdk.git@master
Patch Set: Really add //build. Add dart_bootstrap rule. Created 4 years, 7 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « third_party/zlib/google/zip_reader.h ('k') | third_party/zlib/google/zip_reader_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « third_party/zlib/google/zip_reader.h ('k') | third_party/zlib/google/zip_reader_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698