| // Copyright 2013 The Flutter 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 "flutter/assets/zip_asset_store.h" |
| #include "flutter/fml/build_config.h" |
| |
| #include <fcntl.h> |
| |
| #if !defined(OS_WIN) |
| #include <unistd.h> |
| #endif |
| |
| #include <string> |
| #include <utility> |
| |
| #include "flutter/fml/trace_event.h" |
| |
| namespace flutter { |
| |
| void UniqueUnzipperTraits::Free(void* file) { |
| unzClose(file); |
| } |
| |
| ZipAssetStore::ZipAssetStore(std::string file_path, std::string directory) |
| : file_path_(std::move(file_path)), directory_(std::move(directory)) { |
| BuildStatCache(); |
| } |
| |
| ZipAssetStore::~ZipAssetStore() = default; |
| |
| UniqueUnzipper ZipAssetStore::CreateUnzipper() const { |
| return UniqueUnzipper{::unzOpen2(file_path_.c_str(), nullptr)}; |
| } |
| |
| // |AssetResolver| |
| bool ZipAssetStore::IsValid() const { |
| return stat_cache_.size() > 0; |
| } |
| |
| // |AssetResolver| |
| std::unique_ptr<fml::Mapping> ZipAssetStore::GetAsMapping( |
| const std::string& asset_name) const { |
| TRACE_EVENT1("flutter", "ZipAssetStore::GetAsMapping", "name", |
| asset_name.c_str()); |
| |
| auto found = stat_cache_.find(directory_ + "/" + asset_name); |
| if (found == stat_cache_.end()) { |
| return nullptr; |
| } |
| |
| auto unzipper = CreateUnzipper(); |
| |
| if (!unzipper.is_valid()) { |
| return nullptr; |
| } |
| |
| int result = UNZ_OK; |
| |
| result = unzGoToFilePos(unzipper.get(), &(found->second.file_pos)); |
| if (result != UNZ_OK) { |
| FML_LOG(WARNING) << "unzGetCurrentFileInfo failed, error=" << result; |
| return nullptr; |
| } |
| |
| result = unzOpenCurrentFile(unzipper.get()); |
| if (result != UNZ_OK) { |
| FML_LOG(WARNING) << "unzOpenCurrentFile failed, error=" << result; |
| return nullptr; |
| } |
| |
| std::vector<uint8_t> data(found->second.uncompressed_size); |
| int total_read = 0; |
| while (total_read < static_cast<int>(data.size())) { |
| int bytes_read = unzReadCurrentFile( |
| unzipper.get(), data.data() + total_read, data.size() - total_read); |
| if (bytes_read <= 0) { |
| return nullptr; |
| } |
| total_read += bytes_read; |
| } |
| |
| return std::make_unique<fml::DataMapping>(std::move(data)); |
| } |
| |
| void ZipAssetStore::BuildStatCache() { |
| TRACE_EVENT0("flutter", "ZipAssetStore::BuildStatCache"); |
| |
| auto unzipper = CreateUnzipper(); |
| |
| if (!unzipper.is_valid()) { |
| return; |
| } |
| |
| if (unzGoToFirstFile(unzipper.get()) != UNZ_OK) { |
| return; |
| } |
| |
| do { |
| int result = UNZ_OK; |
| |
| // Get the current file name. |
| unz_file_info file_info = {}; |
| char file_name[255]; |
| result = unzGetCurrentFileInfo(unzipper.get(), &file_info, file_name, |
| sizeof(file_name), nullptr, 0, nullptr, 0); |
| if (result != UNZ_OK) { |
| continue; |
| } |
| |
| if (file_info.uncompressed_size == 0) { |
| continue; |
| } |
| |
| // Get the current file position. |
| unz_file_pos file_pos = {}; |
| result = unzGetFilePos(unzipper.get(), &file_pos); |
| if (result != UNZ_OK) { |
| continue; |
| } |
| |
| std::string file_name_key(file_name, file_info.size_filename); |
| CacheEntry entry(file_pos, file_info.uncompressed_size); |
| stat_cache_.emplace(std::move(file_name_key), std::move(entry)); |
| |
| } while (unzGoToNextFile(unzipper.get()) == UNZ_OK); |
| } |
| |
| } // namespace flutter |