2014-10-31 18:17:05 +00:00
|
|
|
// Copyright (c) 2014 GitHub, Inc.
|
2014-09-25 08:56:50 +00:00
|
|
|
// Use of this source code is governed by the MIT license that can be
|
|
|
|
// found in the LICENSE file.
|
|
|
|
|
|
|
|
#include "atom/common/asar/scoped_temporary_file.h"
|
|
|
|
|
2014-09-25 15:25:17 +00:00
|
|
|
#include <vector>
|
|
|
|
|
2015-01-10 01:45:50 +00:00
|
|
|
#include "base/files/file_util.h"
|
2014-09-25 08:56:50 +00:00
|
|
|
#include "base/threading/thread_restrictions.h"
|
|
|
|
|
|
|
|
namespace asar {
|
|
|
|
|
|
|
|
ScopedTemporaryFile::ScopedTemporaryFile() {
|
|
|
|
}
|
|
|
|
|
|
|
|
ScopedTemporaryFile::~ScopedTemporaryFile() {
|
|
|
|
if (!path_.empty()) {
|
2015-02-03 01:03:52 +00:00
|
|
|
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.
|
2015-02-03 00:09:35 +00:00
|
|
|
#if defined(OS_WIN)
|
2015-02-03 01:03:52 +00:00
|
|
|
base::DeleteFileAfterReboot(path_);
|
2015-02-03 00:09:35 +00:00
|
|
|
#else
|
2014-09-25 08:56:50 +00:00
|
|
|
base::DeleteFile(path_, false);
|
2015-02-03 00:09:35 +00:00
|
|
|
#endif
|
2014-09-25 08:56:50 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-02 03:04:47 +00:00
|
|
|
bool ScopedTemporaryFile::Init(const base::FilePath::StringType& ext) {
|
2014-09-25 08:56:50 +00:00
|
|
|
if (!path_.empty())
|
|
|
|
return true;
|
|
|
|
|
|
|
|
base::ThreadRestrictions::ScopedAllowIO allow_io;
|
2015-12-02 03:36:29 +00:00
|
|
|
if (!base::CreateTemporaryFile(&path_))
|
2015-12-01 15:57:32 +00:00
|
|
|
return false;
|
|
|
|
|
2015-12-02 03:36:29 +00:00
|
|
|
#if defined(OS_WIN)
|
2015-12-02 03:04:47 +00:00
|
|
|
// Keep the original extension.
|
2015-12-02 03:36:29 +00:00
|
|
|
if (!ext.empty()) {
|
|
|
|
base::FilePath new_path = path_.AddExtension(ext);
|
|
|
|
if (!base::Move(path_, new_path))
|
|
|
|
return false;
|
|
|
|
path_ = new_path;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return true;
|
2014-09-25 08:56:50 +00:00
|
|
|
}
|
|
|
|
|
2015-05-11 03:02:17 +00:00
|
|
|
bool ScopedTemporaryFile::InitFromFile(base::File* src,
|
2015-12-02 03:04:47 +00:00
|
|
|
const base::FilePath::StringType& ext,
|
2014-09-25 08:56:50 +00:00
|
|
|
uint64 offset, uint64 size) {
|
2015-05-11 03:02:17 +00:00
|
|
|
if (!src->IsValid())
|
2014-09-25 08:56:50 +00:00
|
|
|
return false;
|
|
|
|
|
2015-12-01 15:57:32 +00:00
|
|
|
if (!Init(ext))
|
2014-09-25 08:56:50 +00:00
|
|
|
return false;
|
|
|
|
|
|
|
|
std::vector<char> buf(size);
|
2015-05-11 03:02:17 +00:00
|
|
|
int len = src->Read(offset, buf.data(), buf.size());
|
2014-09-25 08:56:50 +00:00
|
|
|
if (len != static_cast<int>(size))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
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
|