diff options
Diffstat (limited to 'base')
-rw-r--r-- | base/file_util.h | 12 | ||||
-rw-r--r-- | base/file_util_win.cc | 26 |
2 files changed, 36 insertions, 2 deletions
diff --git a/base/file_util.h b/base/file_util.h index d89fa71..108c998 100644 --- a/base/file_util.h +++ b/base/file_util.h @@ -506,6 +506,12 @@ class MemoryMappedFile { // ownership of |file| and close it when done. bool Initialize(base::PlatformFile file); +#if defined(OS_WIN) + // Opens an existing file and maps it as an image section. Please refer to + // the Initialize function above for additional information. + bool InitializeAsImageSection(const FilePath& file_name); +#endif // OS_WIN + const uint8* data() const { return data_; } size_t length() const { return length_; } @@ -523,10 +529,14 @@ class MemoryMappedFile { // Closes all open handles. Later we may want to make this public. void CloseHandles(); - base::PlatformFile file_; #if defined(OS_WIN) + // MapFileToMemoryInternal calls this function. It provides the ability to + // pass in flags which control the mapped section. + bool MapFileToMemoryInternalEx(int flags); + HANDLE file_mapping_; #endif + base::PlatformFile file_; uint8* data_; size_t length_; diff --git a/base/file_util_win.cc b/base/file_util_win.cc index ab1dc94..2c26a16 100644 --- a/base/file_util_win.cc +++ b/base/file_util_win.cc @@ -938,7 +938,31 @@ MemoryMappedFile::MemoryMappedFile() length_(INVALID_FILE_SIZE) { } +bool MemoryMappedFile::InitializeAsImageSection(const FilePath& file_name) { + if (IsValid()) + return false; + file_ = base::CreatePlatformFile( + file_name, base::PLATFORM_FILE_OPEN | base::PLATFORM_FILE_READ, + NULL, NULL); + + if (file_ == base::kInvalidPlatformFileValue) { + LOG(ERROR) << "Couldn't open " << file_name.value(); + return false; + } + + if (!MapFileToMemoryInternalEx(SEC_IMAGE)) { + CloseHandles(); + return false; + } + + return true; +} + bool MemoryMappedFile::MapFileToMemoryInternal() { + return MapFileToMemoryInternalEx(0); +} + +bool MemoryMappedFile::MapFileToMemoryInternalEx(int flags) { base::ThreadRestrictions::AssertIOAllowed(); if (file_ == INVALID_HANDLE_VALUE) @@ -950,7 +974,7 @@ bool MemoryMappedFile::MapFileToMemoryInternal() { // length_ value comes from GetFileSize() above. GetFileSize() returns DWORD, // therefore the cast here is safe. - file_mapping_ = ::CreateFileMapping(file_, NULL, PAGE_READONLY, + file_mapping_ = ::CreateFileMapping(file_, NULL, PAGE_READONLY | flags, 0, static_cast<DWORD>(length_), NULL); if (!file_mapping_) { // According to msdn, system error codes are only reserved up to 15999. |