third_party/unrar/src/unrar_wrapper.cc

// Copyright 2019 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/unrar/src/unrar_wrapper.h"

#include <memory>

#include "base/files/file_path.h"
#include "base/metrics/histogram_macros.h"
#include "build/build_config.h"
#include "third_party/unrar/src/rar.hpp"

namespace third_party_unrar {

RarReader::RarReader() {}

RarReader::~RarReader() {}

bool RarReader::Open(base::File rar_file, base::File temp_file) {
  rar_file_ = std::move(rar_file);
  temp_file_ = std::move(temp_file);

  archive_ = std::make_unique<Archive>();
  archive_->SetFileHandle(rar_file_.GetPlatformFile());
  archive_->SetTempFileHandle(temp_file_.GetPlatformFile());

  bool open_success = archive_->Open(L"dummy.rar");
  UMA_HISTOGRAM_BOOLEAN("SBClientDownload.RarOpenSuccess", open_success);
  if (!open_success)
    return false;

  bool is_valid_archive = archive_->IsArchive(/*EnableBroken=*/true);
  UMA_HISTOGRAM_BOOLEAN("SBClientDownload.RarValidArchive", is_valid_archive);
  if (!is_valid_archive)
    return false;

  UMA_HISTOGRAM_BOOLEAN("SBClientDownload.RarHeadersEncrypted",
                        archive_->Encrypted);

  command_ = std::make_unique<CommandData>();
  command_->ParseArg(const_cast<wchar_t*>(L"-p"));
  command_->ParseArg(const_cast<wchar_t*>(L"x"));
  command_->ParseDone();

  extractor_ = std::make_unique<CmdExtract>(command_.get());
  extractor_->ExtractArchiveInit(*archive_);

  return true;
}

bool RarReader::ExtractNextEntry() {
  bool success = true, repeat = true;
  while (success || repeat) {
    temp_file_.Seek(base::File::Whence::FROM_BEGIN, 0);
    temp_file_.SetLength(0);
    size_t header_size = archive_->ReadHeader();
    repeat = false;
    success = extractor_->ExtractCurrentFile(
        *archive_, header_size, repeat);  // |repeat| is passed by reference

    if (archive_->GetHeaderType() == HEAD_FILE) {
#if defined(OS_WIN)
      current_entry_.file_path = base::FilePath(archive_->FileHead.FileName);
#else
      std::wstring wide_filename(archive_->FileHead.FileName);
      std::string filename(wide_filename.begin(), wide_filename.end());
      current_entry_.file_path = base::FilePath(filename);
#endif
      current_entry_.is_directory = archive_->FileHead.Dir;
      current_entry_.is_encrypted = archive_->FileHead.Encrypted;
      return true;
    }
  }

  return false;
}

}  // namespace third_party_unrar