| Index: base/files/memory_mapped_file_win.cc
|
| diff --git a/base/files/memory_mapped_file_win.cc b/base/files/memory_mapped_file_win.cc
|
| index 694212950de18382bfdeb417c10510e40082ea56..f3822873bfdddb877bd73afd9fa2548ca5cd83f2 100644
|
| --- a/base/files/memory_mapped_file_win.cc
|
| +++ b/base/files/memory_mapped_file_win.cc
|
| @@ -5,83 +5,52 @@
|
| #include "base/files/memory_mapped_file.h"
|
|
|
| #include "base/files/file_path.h"
|
| -#include "base/logging.h"
|
| -#include "base/metrics/histogram.h"
|
| #include "base/strings/string16.h"
|
| #include "base/threading/thread_restrictions.h"
|
|
|
| namespace base {
|
|
|
| -MemoryMappedFile::MemoryMappedFile()
|
| - : file_(INVALID_HANDLE_VALUE),
|
| - file_mapping_(INVALID_HANDLE_VALUE),
|
| - data_(NULL),
|
| - length_(INVALID_FILE_SIZE) {
|
| +MemoryMappedFile::MemoryMappedFile() : data_(NULL), length_(0), image_(false) {
|
| }
|
|
|
| bool MemoryMappedFile::InitializeAsImageSection(const FilePath& file_name) {
|
| - if (IsValid())
|
| - return false;
|
| - file_ = CreatePlatformFile(file_name, PLATFORM_FILE_OPEN | PLATFORM_FILE_READ,
|
| - NULL, NULL);
|
| -
|
| - if (file_ == kInvalidPlatformFileValue) {
|
| - DLOG(ERROR) << "Couldn't open " << file_name.AsUTF8Unsafe();
|
| - return false;
|
| - }
|
| -
|
| - if (!MapFileToMemoryInternalEx(SEC_IMAGE)) {
|
| - CloseHandles();
|
| - return false;
|
| - }
|
| -
|
| - return true;
|
| -}
|
| -
|
| -bool MemoryMappedFile::MapFileToMemoryInternal() {
|
| - return MapFileToMemoryInternalEx(0);
|
| + image_ = true;
|
| + return Initialize(file_name);
|
| }
|
|
|
| -bool MemoryMappedFile::MapFileToMemoryInternalEx(int flags) {
|
| +bool MemoryMappedFile::MapFileToMemory() {
|
| ThreadRestrictions::AssertIOAllowed();
|
|
|
| - if (file_ == INVALID_HANDLE_VALUE)
|
| + if (!file_.IsValid())
|
| return false;
|
|
|
| - length_ = ::GetFileSize(file_, NULL);
|
| - if (length_ == INVALID_FILE_SIZE)
|
| + int64 len = file_.GetLength();
|
| + if (len <= 0 || len > kint32max)
|
| return false;
|
| + length_ = static_cast<size_t>(len);
|
| +
|
| + int flags = image_ ? SEC_IMAGE | PAGE_READONLY : PAGE_READONLY;
|
|
|
| - file_mapping_ = ::CreateFileMapping(file_, NULL, PAGE_READONLY | flags,
|
| - 0, 0, NULL);
|
| - if (!file_mapping_) {
|
| - // According to msdn, system error codes are only reserved up to 15999.
|
| - // http://msdn.microsoft.com/en-us/library/ms681381(v=VS.85).aspx.
|
| - UMA_HISTOGRAM_ENUMERATION("MemoryMappedFile.CreateFileMapping",
|
| - logging::GetLastSystemErrorCode(), 16000);
|
| + file_mapping_.Set(::CreateFileMapping(file_.GetPlatformFile(), NULL,
|
| + flags, 0, 0, NULL));
|
| + if (!file_mapping_.IsValid())
|
| return false;
|
| - }
|
|
|
| - data_ = static_cast<uint8*>(
|
| - ::MapViewOfFile(file_mapping_, FILE_MAP_READ, 0, 0, 0));
|
| - if (!data_) {
|
| - UMA_HISTOGRAM_ENUMERATION("MemoryMappedFile.MapViewOfFile",
|
| - logging::GetLastSystemErrorCode(), 16000);
|
| - }
|
| + data_ = static_cast<uint8*>(::MapViewOfFile(file_mapping_.Get(),
|
| + FILE_MAP_READ, 0, 0, 0));
|
| return data_ != NULL;
|
| }
|
|
|
| void MemoryMappedFile::CloseHandles() {
|
| if (data_)
|
| ::UnmapViewOfFile(data_);
|
| - if (file_mapping_ != INVALID_HANDLE_VALUE)
|
| - ::CloseHandle(file_mapping_);
|
| - if (file_ != INVALID_HANDLE_VALUE)
|
| - ::CloseHandle(file_);
|
| + if (file_mapping_.IsValid())
|
| + file_mapping_.Close();
|
| + if (file_.IsValid())
|
| + file_.Close();
|
|
|
| data_ = NULL;
|
| - file_mapping_ = file_ = INVALID_HANDLE_VALUE;
|
| - length_ = INVALID_FILE_SIZE;
|
| + length_ = 0;
|
| }
|
|
|
| } // namespace base
|
|
|