60c4c9fec6
* chore: iwyu buildflags.h * chore: iwyu dictionary.h * chore: iwyu arguments.h * chore: iwyu values.h * chore: iwyu compiler_specific.h * chore: iwyu binder_map.h * chore: iwyu <vector> * chore: iwyu <set> * chore: iwyu raw_ptr * chore: iwyu gfx/canvas.h * chore: iwyu gfx/color_utils.h * chore: iwyu base/strings/stringprintf.h * chore: iwyu base/task/thread_pool.h * chore: iwyu base/no_destructor.h * chore: iwyu base/path_service.h * chore: iwyu base/files/file_pathh * chore: iwyu base/strings/sys_string_conversions.h * chore: iwyu base/logging.h * chore: iwyu base/command_line.h * chore: iwyu base/files/file_util.h * chore: iwyu base/files/scoped_file.h * chore: iwyu base/strings/utf_string_conversions.h * chore: iwyu base/environment.h * chore: iwyu base/scoped_observation.h * chore: iwyu base/strings/string_split.h * chore: iwyu base/strings/pattern.h * chore: iwyu base/json/string_escape.h * chore: iwyu base/json/json_reader.h * chore: iwyu base/memory/singleton.h * chore: iwyu base/observer_list.h * chore: iwyu base/timer/timer.h * fixup! chore: iwyu values.h * chore: iwyu shell/browser/browser.h * chore: iwyu base/stl_util.h * chore: iwyu base/strings/string_util.h * chore: iwyu shell/browser/javascript_environment.h * chore: iwyu base/memory/ref_counted.h * chore: iwyu base/environment.h * chore: iwyu content/public/browser/browser_thread.h * chore: remove unused typedef gin_helper::EventEmitter::ValueArray * chore: iwyu gin/wrappable.h * chore: iwyu shell/common/gin_helper/function_template_extensions.h * chore: iwyu shell/common/gin_converters/login_item_settings_converter.h * chore: iwyu shell/common/gin_helper/arguments.h * chore: iwyu ui/gfx/skia_util.h * chore: iwyu ui/gfx/geometry/rect.h * chore: iwyu ui/gfx/image/image.h * chore: iwyu base/strings/strcat.h * chore: iwyu ui/native_theme/native_theme.h * fixup! chore: iwyu shell/browser/javascript_environment.h * fixup! chore: iwyu gfx/canvas.h * fixup! chore: iwyu content/public/browser/browser_thread.h * fixup! chore: iwyu ui/native_theme/native_theme.h * fixup! chore: iwyu ui/native_theme/native_theme.h
82 lines
2.1 KiB
C++
82 lines
2.1 KiB
C++
// Copyright (c) 2014 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "shell/common/asar/scoped_temporary_file.h"
|
|
|
|
#include <vector>
|
|
|
|
#include "base/files/file_util.h"
|
|
#include "shell/common/asar/asar_util.h"
|
|
#include "shell/common/thread_restrictions.h"
|
|
|
|
namespace asar {
|
|
|
|
ScopedTemporaryFile::ScopedTemporaryFile() = default;
|
|
|
|
ScopedTemporaryFile::~ScopedTemporaryFile() {
|
|
if (!path_.empty()) {
|
|
electron::ScopedAllowBlockingForElectron allow_blocking;
|
|
// On Windows it is very likely the file is already in use (because it is
|
|
// mostly used for Node native modules), so deleting it now will halt the
|
|
// program.
|
|
#if BUILDFLAG(IS_WIN)
|
|
base::DeleteFileAfterReboot(path_);
|
|
#else
|
|
base::DeleteFile(path_);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
bool ScopedTemporaryFile::Init(const base::FilePath::StringType& ext) {
|
|
if (!path_.empty())
|
|
return true;
|
|
|
|
electron::ScopedAllowBlockingForElectron allow_blocking;
|
|
if (!base::CreateTemporaryFile(&path_))
|
|
return false;
|
|
|
|
#if BUILDFLAG(IS_WIN)
|
|
// Keep the original extension.
|
|
if (!ext.empty()) {
|
|
base::FilePath new_path = path_.AddExtension(ext);
|
|
if (!base::Move(path_, new_path))
|
|
return false;
|
|
path_ = new_path;
|
|
}
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScopedTemporaryFile::InitFromFile(
|
|
base::File* src,
|
|
const base::FilePath::StringType& ext,
|
|
uint64_t offset,
|
|
uint64_t size,
|
|
const std::optional<IntegrityPayload>& integrity) {
|
|
if (!src->IsValid())
|
|
return false;
|
|
|
|
if (!Init(ext))
|
|
return false;
|
|
|
|
electron::ScopedAllowBlockingForElectron allow_blocking;
|
|
std::vector<char> buf(size);
|
|
int len = src->Read(offset, buf.data(), buf.size());
|
|
if (len != static_cast<int>(size))
|
|
return false;
|
|
|
|
if (integrity.has_value()) {
|
|
ValidateIntegrityOrDie(buf.data(), buf.size(), integrity.value());
|
|
}
|
|
|
|
base::File dest(path_, base::File::FLAG_OPEN | base::File::FLAG_WRITE);
|
|
if (!dest.IsValid())
|
|
return false;
|
|
|
|
return dest.WriteAtCurrentPos(buf.data(), buf.size()) ==
|
|
static_cast<int>(size);
|
|
}
|
|
|
|
} // namespace asar
|