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/logging.h" | |
| 9 #include "base/metrics/histogram.h" | |
| 10 #include "base/strings/string16.h" | 8 #include "base/strings/string16.h" |
| 11 #include "base/threading/thread_restrictions.h" | 9 #include "base/threading/thread_restrictions.h" |
| 12 | 10 |
| 13 namespace base { | 11 namespace base { |
| 14 | 12 |
| 15 MemoryMappedFile::MemoryMappedFile() | 13 MemoryMappedFile::MemoryMappedFile() : data_(NULL), length_(0), image_(false) { |
| 16 : file_(INVALID_HANDLE_VALUE), | |
| 17 file_mapping_(INVALID_HANDLE_VALUE), | |
| 18 data_(NULL), | |
| 19 length_(INVALID_FILE_SIZE) { | |
| 20 } | 14 } |
| 21 | 15 |
| 22 bool MemoryMappedFile::InitializeAsImageSection(const FilePath& file_name) { | 16 bool MemoryMappedFile::InitializeAsImageSection(const FilePath& file_name) { |
| 23 if (IsValid()) | 17 image_ = true; |
| 24 return false; | 18 return Initialize(file_name); |
| 25 file_ = CreatePlatformFile(file_name, PLATFORM_FILE_OPEN | PLATFORM_FILE_READ, | |
| 26 NULL, NULL); | |
| 27 | |
| 28 if (file_ == kInvalidPlatformFileValue) { | |
| 29 DLOG(ERROR) << "Couldn't open " << file_name.AsUTF8Unsafe(); | |
| 30 return false; | |
| 31 } | |
| 32 | |
| 33 if (!MapFileToMemoryInternalEx(SEC_IMAGE)) { | |
| 34 CloseHandles(); | |
| 35 return false; | |
| 36 } | |
| 37 | |
| 38 return true; | |
| 39 } | 19 } |
| 40 | 20 |
| 41 bool MemoryMappedFile::MapFileToMemoryInternal() { | 21 bool MemoryMappedFile::MapFileToMemory() { |
| 42 return MapFileToMemoryInternalEx(0); | |
| 43 } | |
| 44 | |
| 45 bool MemoryMappedFile::MapFileToMemoryInternalEx(int flags) { | |
| 46 ThreadRestrictions::AssertIOAllowed(); | 22 ThreadRestrictions::AssertIOAllowed(); |
| 47 | 23 |
| 48 if (file_ == INVALID_HANDLE_VALUE) | 24 if (!file_.IsValid()) |
| 49 return false; | 25 return false; |
| 50 | 26 |
| 51 length_ = ::GetFileSize(file_, NULL); | 27 int64 len = file_.GetLength(); |
| 52 if (length_ == INVALID_FILE_SIZE) | 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; | |
| 33 | |
| 34 file_mapping_.Set(::CreateFileMapping(file_.GetPlatformFile(), NULL, | |
| 35 flags, 0, 0, NULL)); | |
| 36 if (!file_mapping_.IsValid()) | |
| 53 return false; | 37 return false; |
| 54 | 38 |
| 55 file_mapping_ = ::CreateFileMapping(file_, NULL, PAGE_READONLY | flags, | 39 data_ = static_cast<uint8*>(::MapViewOfFile(file_mapping_.Get(), |
| 56 0, 0, NULL); | 40 FILE_MAP_READ, 0, 0, 0)); |
| 57 if (!file_mapping_) { | |
| 58 // According to msdn, system error codes are only reserved up to 15999. | |
| 59 // http://msdn.microsoft.com/en-us/library/ms681381(v=VS.85).aspx. | |
| 60 UMA_HISTOGRAM_ENUMERATION("MemoryMappedFile.CreateFileMapping", | |
|
cpu_(ooo_6.6-7.5)
2013/12/21 22:59:33
I so hope this two umas are gone gone, not just mo
rvargas (doing something else)
2013/12/27 23:54:39
gone for good
| |
| 61 logging::GetLastSystemErrorCode(), 16000); | |
| 62 return false; | |
| 63 } | |
| 64 | |
| 65 data_ = static_cast<uint8*>( | |
| 66 ::MapViewOfFile(file_mapping_, FILE_MAP_READ, 0, 0, 0)); | |
| 67 if (!data_) { | |
| 68 UMA_HISTOGRAM_ENUMERATION("MemoryMappedFile.MapViewOfFile", | |
| 69 logging::GetLastSystemErrorCode(), 16000); | |
| 70 } | |
| 71 return data_ != NULL; | 41 return data_ != NULL; |
| 72 } | 42 } |
| 73 | 43 |
| 74 void MemoryMappedFile::CloseHandles() { | 44 void MemoryMappedFile::CloseHandles() { |
| 75 if (data_) | 45 if (data_) |
| 76 ::UnmapViewOfFile(data_); | 46 ::UnmapViewOfFile(data_); |
| 77 if (file_mapping_ != INVALID_HANDLE_VALUE) | 47 if (file_mapping_.IsValid()) |
| 78 ::CloseHandle(file_mapping_); | 48 file_mapping_.Close(); |
| 79 if (file_ != INVALID_HANDLE_VALUE) | 49 if (file_.IsValid()) |
| 80 ::CloseHandle(file_); | 50 file_.Close(); |
| 81 | 51 |
| 82 data_ = NULL; | 52 data_ = NULL; |
| 83 file_mapping_ = file_ = INVALID_HANDLE_VALUE; | 53 length_ = 0; |
| 84 length_ = INVALID_FILE_SIZE; | |
| 85 } | 54 } |
| 86 | 55 |
| 87 } // namespace base | 56 } // namespace base |
| OLD | NEW |