57d088517c
* feat: add support for validating asar archives on macOS * chore: fix lint * chore: update as per feedback * feat: switch implementation to asar integrity hash checks * feat: make ranged requests work with the asar file validator DataSourceFilter * chore: fix lint * chore: fix missing log include on non-darwin * fix: do not pull block size out of missing optional * fix: match ValidateOrDie symbol on non-darwin * chore: fix up asar specs by repacking archives * fix: maintain integrity chain, do not load file integrity if header integrity was not loaded * debug test * Update node-spec.ts * fix: initialize header_validated_ * chore: update PR per feedback * chore: update per feedback * build: use final asar module * Update fuses.json5
83 lines
2.1 KiB
C++
83 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 "base/logging.h"
|
|
#include "base/threading/thread_restrictions.h"
|
|
#include "shell/common/asar/asar_util.h"
|
|
|
|
namespace asar {
|
|
|
|
ScopedTemporaryFile::ScopedTemporaryFile() = default;
|
|
|
|
ScopedTemporaryFile::~ScopedTemporaryFile() {
|
|
if (!path_.empty()) {
|
|
base::ThreadRestrictions::ScopedAllowIO allow_io;
|
|
// 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 defined(OS_WIN)
|
|
base::DeleteFileAfterReboot(path_);
|
|
#else
|
|
base::DeleteFile(path_);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
bool ScopedTemporaryFile::Init(const base::FilePath::StringType& ext) {
|
|
if (!path_.empty())
|
|
return true;
|
|
|
|
base::ThreadRestrictions::ScopedAllowIO allow_io;
|
|
if (!base::CreateTemporaryFile(&path_))
|
|
return false;
|
|
|
|
#if defined(OS_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 absl::optional<IntegrityPayload>& integrity) {
|
|
if (!src->IsValid())
|
|
return false;
|
|
|
|
if (!Init(ext))
|
|
return false;
|
|
|
|
base::ThreadRestrictions::ScopedAllowIO allow_io;
|
|
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
|