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::MapFileToMemory() { | 21 bool MemoryMappedFile::MapFileRegionToMemory( |
| 22 const MemoryMappedFile::Region& region) { |
22 ThreadRestrictions::AssertIOAllowed(); | 23 ThreadRestrictions::AssertIOAllowed(); |
23 | 24 |
24 if (!file_.IsValid()) | 25 if (!file_.IsValid()) |
25 return false; | 26 return false; |
26 | 27 |
27 int64 len = file_.GetLength(); | |
28 if (len <= 0 || len > kint32max) | |
29 return false; | |
30 length_ = static_cast<size_t>(len); | |
31 | |
32 int flags = image_ ? SEC_IMAGE | PAGE_READONLY : PAGE_READONLY; | 28 int flags = image_ ? SEC_IMAGE | PAGE_READONLY : PAGE_READONLY; |
33 | 29 |
34 file_mapping_.Set(::CreateFileMapping(file_.GetPlatformFile(), NULL, | 30 file_mapping_.Set(::CreateFileMapping(file_.GetPlatformFile(), NULL, |
35 flags, 0, 0, NULL)); | 31 flags, 0, 0, NULL)); |
36 if (!file_mapping_.IsValid()) | 32 if (!file_mapping_.IsValid()) |
37 return false; | 33 return false; |
38 | 34 |
| 35 LARGE_INTEGER map_start = {0}; |
| 36 SIZE_T map_size = 0; |
| 37 int32 data_offset = 0; |
| 38 |
| 39 if (region == MemoryMappedFile::Region::kWholeFile) { |
| 40 int64 file_len = file_.GetLength(); |
| 41 if (file_len <= 0 || file_len > kint32max) |
| 42 return false; |
| 43 length_ = static_cast<size_t>(file_len); |
| 44 } else { |
| 45 // The region can be arbitrarily aligned. MapViewOfFile, instead, requires |
| 46 // that the start address is aligned to the VM granularity (which is |
| 47 // typically larger than a page size, for instance 32k). |
| 48 // Also, conversely to POSIX's mmap, the |map_size| doesn't have to be |
| 49 // aligned and must be less than or equal the mapped file size. |
| 50 // We map here the outer region [|aligned_start|, |aligned_start+size|] |
| 51 // which contains |region| and then add up the |data_offset| displacement. |
| 52 int64 aligned_start = 0; |
| 53 int64 ignored = 0; |
| 54 CalculateVMAlignedBoundaries( |
| 55 region.offset, region.size, &aligned_start, &ignored, &data_offset); |
| 56 int64 size = region.size + data_offset; |
| 57 |
| 58 // Ensure that the casts below in the MapViewOfFile call are sane. |
| 59 if (aligned_start < 0 || size < 0 || |
| 60 static_cast<uint64>(size) > std::numeric_limits<SIZE_T>::max()) { |
| 61 DLOG(ERROR) << "Region bounds are not valid for MapViewOfFile"; |
| 62 return false; |
| 63 } |
| 64 map_start.QuadPart = aligned_start; |
| 65 map_size = static_cast<SIZE_T>(size); |
| 66 length_ = static_cast<size_t>(region.size); |
| 67 } |
| 68 |
39 data_ = static_cast<uint8*>(::MapViewOfFile(file_mapping_.Get(), | 69 data_ = static_cast<uint8*>(::MapViewOfFile(file_mapping_.Get(), |
40 FILE_MAP_READ, 0, 0, 0)); | 70 FILE_MAP_READ, |
41 return data_ != NULL; | 71 map_start.HighPart, |
| 72 map_start.LowPart, |
| 73 map_size)); |
| 74 if (data_ == NULL) |
| 75 return false; |
| 76 data_ += data_offset; |
| 77 return true; |
42 } | 78 } |
43 | 79 |
44 void MemoryMappedFile::CloseHandles() { | 80 void MemoryMappedFile::CloseHandles() { |
45 if (data_) | 81 if (data_) |
46 ::UnmapViewOfFile(data_); | 82 ::UnmapViewOfFile(data_); |
47 if (file_mapping_.IsValid()) | 83 if (file_mapping_.IsValid()) |
48 file_mapping_.Close(); | 84 file_mapping_.Close(); |
49 if (file_.IsValid()) | 85 if (file_.IsValid()) |
50 file_.Close(); | 86 file_.Close(); |
51 | 87 |
52 data_ = NULL; | 88 data_ = NULL; |
53 length_ = 0; | 89 length_ = 0; |
54 } | 90 } |
55 | 91 |
56 } // namespace base | 92 } // namespace base |
OLD | NEW |