OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/files/memory_mapped_file.h" | 5 #include "base/files/memory_mapped_file.h" |
6 | 6 |
7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
8 #include "base/strings/string16.h" | 8 #include "base/strings/string16.h" |
9 #include "base/threading/thread_restrictions.h" | 9 #include "base/threading/thread_restrictions.h" |
10 | 10 |
11 namespace base { | 11 namespace base { |
12 | 12 |
13 MemoryMappedFile::MemoryMappedFile() : data_(NULL), length_(0), image_(false) { | 13 MemoryMappedFile::MemoryMappedFile() : data_(NULL), length_(0), image_(false) { |
14 } | 14 } |
15 | 15 |
16 bool MemoryMappedFile::InitializeAsImageSection(const FilePath& file_name) { | 16 bool MemoryMappedFile::InitializeAsImageSection(const FilePath& file_name) { |
17 image_ = true; | 17 image_ = true; |
18 return Initialize(file_name); | 18 return Initialize(file_name); |
19 } | 19 } |
20 | 20 |
21 bool MemoryMappedFile::MapFileRegionToMemory( | 21 bool MemoryMappedFile::MapFileRegionToMemory( |
22 const MemoryMappedFile::Region& region) { | 22 const MemoryMappedFile::Region& region, bool write) { |
23 ThreadRestrictions::AssertIOAllowed(); | 23 ThreadRestrictions::AssertIOAllowed(); |
24 | 24 |
25 if (!file_.IsValid()) | 25 if (!file_.IsValid()) |
26 return false; | 26 return false; |
27 | 27 |
28 int flags = image_ ? SEC_IMAGE | PAGE_READONLY : PAGE_READONLY; | 28 int flags = 0; |
| 29 if (image_) |
| 30 flags = SEC_IMAGE | PAGE_READONLY; |
| 31 else |
| 32 flags = write ? PAGE_READWRITE : PAGE_READONLY; |
29 | 33 |
30 file_mapping_.Set(::CreateFileMapping(file_.GetPlatformFile(), NULL, | 34 file_mapping_.Set(::CreateFileMapping(file_.GetPlatformFile(), NULL, |
31 flags, 0, 0, NULL)); | 35 flags, 0, 0, NULL)); |
32 if (!file_mapping_.IsValid()) | 36 if (!file_mapping_.IsValid()) |
33 return false; | 37 return false; |
34 | 38 |
35 LARGE_INTEGER map_start = {0}; | 39 LARGE_INTEGER map_start = {0}; |
36 SIZE_T map_size = 0; | 40 SIZE_T map_size = 0; |
37 int32 data_offset = 0; | 41 int32 data_offset = 0; |
38 | 42 |
(...skipping 20 matching lines...) Expand all Loading... |
59 if (aligned_start < 0 || size < 0 || | 63 if (aligned_start < 0 || size < 0 || |
60 static_cast<uint64>(size) > std::numeric_limits<SIZE_T>::max()) { | 64 static_cast<uint64>(size) > std::numeric_limits<SIZE_T>::max()) { |
61 DLOG(ERROR) << "Region bounds are not valid for MapViewOfFile"; | 65 DLOG(ERROR) << "Region bounds are not valid for MapViewOfFile"; |
62 return false; | 66 return false; |
63 } | 67 } |
64 map_start.QuadPart = aligned_start; | 68 map_start.QuadPart = aligned_start; |
65 map_size = static_cast<SIZE_T>(size); | 69 map_size = static_cast<SIZE_T>(size); |
66 length_ = static_cast<size_t>(region.size); | 70 length_ = static_cast<size_t>(region.size); |
67 } | 71 } |
68 | 72 |
| 73 int map_access = write ? FILE_MAP_WRITE : FILE_MAP_READ; |
69 data_ = static_cast<uint8*>(::MapViewOfFile(file_mapping_.Get(), | 74 data_ = static_cast<uint8*>(::MapViewOfFile(file_mapping_.Get(), |
70 FILE_MAP_READ, | 75 map_access, |
71 map_start.HighPart, | 76 map_start.HighPart, |
72 map_start.LowPart, | 77 map_start.LowPart, |
73 map_size)); | 78 map_size)); |
74 if (data_ == NULL) | 79 if (data_ == NULL) |
75 return false; | 80 return false; |
76 data_ += data_offset; | 81 data_ += data_offset; |
77 return true; | 82 return true; |
78 } | 83 } |
79 | 84 |
80 void MemoryMappedFile::CloseHandles() { | 85 void MemoryMappedFile::CloseHandles() { |
81 if (data_) | 86 if (data_) |
82 ::UnmapViewOfFile(data_); | 87 ::UnmapViewOfFile(data_); |
83 if (file_mapping_.IsValid()) | 88 if (file_mapping_.IsValid()) |
84 file_mapping_.Close(); | 89 file_mapping_.Close(); |
85 if (file_.IsValid()) | 90 if (file_.IsValid()) |
86 file_.Close(); | 91 file_.Close(); |
87 | 92 |
88 data_ = NULL; | 93 data_ = NULL; |
89 length_ = 0; | 94 length_ = 0; |
90 } | 95 } |
91 | 96 |
92 } // namespace base | 97 } // namespace base |
OLD | NEW |