Chromium Code Reviews| 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; | |
|
wuchengli
2014/09/01 09:19:25
You forgot to address this comment?
From MSDN, it
| |
| 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 |