1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/zucchini/mapped_file.h"
#include <utility>
#include "base/files/file_util.h"
#include "build/build_config.h"
namespace zucchini {
MappedFileReader::MappedFileReader(base::File file) {
if (!file.IsValid()) {
error_ = "Invalid file.";
return; // |buffer_| will be uninitialized, and therefore invalid.
}
if (!buffer_.Initialize(std::move(file))) {
error_ = "Can't map file to memory.";
}
}
MappedFileWriter::MappedFileWriter(const base::FilePath& file_path,
base::File file,
size_t length)
: file_path_(file_path), delete_behavior_(kManualDeleteOnClose) {
if (!file.IsValid()) {
error_ = "Invalid file.";
return; // |buffer_| will be uninitialized, and therefore invalid.
}
#if BUILDFLAG(IS_WIN)
file_handle_ = file.Duplicate();
// Tell the OS to delete the file when all handles are closed.
if (file_handle_.DeleteOnClose(true)) {
delete_behavior_ = kAutoDeleteOnClose;
} else {
error_ = "Failed to mark file for delete-on-close.";
}
#endif // BUILDFLAG(IS_WIN)
bool is_ok = buffer_.Initialize(std::move(file), {0, length},
base::MemoryMappedFile::READ_WRITE_EXTEND);
if (!is_ok) {
error_ = "Can't map file to memory.";
}
}
MappedFileWriter::~MappedFileWriter() {
if (!HasError() && delete_behavior_ == kManualDeleteOnClose &&
!file_path_.empty() && !base::DeleteFile(file_path_)) {
error_ = "Failed to delete file.";
}
}
bool MappedFileWriter::Keep() {
#if BUILDFLAG(IS_WIN)
if (delete_behavior_ == kAutoDeleteOnClose &&
!file_handle_.DeleteOnClose(false)) {
error_ = "Failed to prevent deletion of file.";
return false;
}
#endif // BUILDFLAG(IS_WIN)
delete_behavior_ = kKeep;
return true;
}
} // namespace zucchini
|
