Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/app/file_pre_reader_win.h" | |
| 6 | |
| 7 #include <windows.h> | |
| 8 #include <algorithm> | |
| 9 | |
| 10 #include "base/files/memory_mapped_file.h" | |
| 11 #include "base/logging.h" | |
| 12 #include "base/threading/thread_restrictions.h" | |
| 13 #include "base/win/scoped_handle.h" | |
| 14 #include "base/win/windows_version.h" | |
| 15 | |
| 16 namespace { | |
| 17 | |
| 18 // A helper function to touch all pages in the range | |
| 19 // [base_addr, base_addr + length). | |
| 20 void TouchPagesInRange(const void* base_addr, size_t length) { | |
| 21 DCHECK(base_addr != NULL); | |
| 22 DCHECK(length > 0); | |
| 23 | |
| 24 // Get the system info so we know the page size. Also, make sure we use a | |
| 25 // non-zero value for the page size; GetSystemInfo() is hookable/patchable, | |
| 26 // and you never know what shenanigans someone could get up to. | |
| 27 SYSTEM_INFO system_info = {}; | |
| 28 GetSystemInfo(&system_info); | |
| 29 if (system_info.dwPageSize == 0) | |
| 30 system_info.dwPageSize = 4096; | |
| 31 | |
| 32 // We don't want to read outside the byte range (which could trigger an | |
| 33 // access violation), so let's figure out the exact locations of the first | |
| 34 // and final bytes we want to read. | |
| 35 volatile uint8 const* touch_ptr = reinterpret_cast<uint8 const*>(base_addr); | |
|
grt (UTC plus 2)
2015/10/29 19:15:06
uint8_t here and elsewhere
fdoray
2015/10/30 21:08:54
Done.
| |
| 36 volatile uint8 const* final_touch_ptr = touch_ptr + length - 1; | |
| 37 | |
| 38 // Read the memory in the range [touch_ptr, final_touch_ptr] with a stride | |
| 39 // of the system page size, to ensure that it's been paged in. | |
| 40 uint8 dummy; | |
| 41 while (touch_ptr < final_touch_ptr) { | |
| 42 dummy = *touch_ptr; | |
| 43 touch_ptr += system_info.dwPageSize; | |
| 44 } | |
| 45 dummy = *final_touch_ptr; | |
| 46 } | |
| 47 | |
| 48 } // namespace | |
| 49 | |
| 50 bool FilePreReader::PreReadImage(const wchar_t* file_path, size_t step_size) { | |
| 51 base::ThreadRestrictions::AssertIOAllowed(); | |
| 52 if (base::win::GetVersion() > base::win::VERSION_XP) { | |
| 53 // Vista+ branch. On these OSes, the forced reads through the DLL actually | |
| 54 // slows warm starts. The solution is to sequentially read file contents | |
| 55 // with an optional cap on total amount to read. | |
| 56 base::win::ScopedHandle file( | |
|
grt (UTC plus 2)
2015/10/29 19:15:07
please add a comment that base::File isn't used be
fdoray
2015/10/30 21:08:54
Done. base::File::FLAG_SEQUENTIAL_SCAN was added i
| |
| 57 CreateFile(file_path, GENERIC_READ, | |
| 58 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, | |
| 59 OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL)); | |
| 60 | |
| 61 if (!file.IsValid()) | |
| 62 return false; | |
| 63 | |
| 64 // Default to 1MB sequential reads. | |
| 65 const DWORD actual_step_size = std::max(static_cast<DWORD>(step_size), | |
| 66 static_cast<DWORD>(1024 * 1024)); | |
| 67 LPVOID buffer = | |
| 68 ::VirtualAlloc(NULL, actual_step_size, MEM_COMMIT, PAGE_READWRITE); | |
| 69 | |
| 70 if (buffer == NULL) | |
| 71 return false; | |
| 72 | |
| 73 DWORD len; | |
| 74 size_t total_read = 0; | |
|
grt (UTC plus 2)
2015/10/29 19:15:06
file sizes are 64bits on windows, so uint64_t.
bet
fdoray
2015/10/30 21:08:54
Done.
| |
| 75 while (::ReadFile(file.Get(), buffer, actual_step_size, &len, NULL) && | |
| 76 len > 0) { | |
| 77 total_read += static_cast<size_t>(len); | |
| 78 } | |
| 79 ::VirtualFree(buffer, 0, MEM_RELEASE); | |
| 80 } else { | |
| 81 // WinXP branch. Here, reading the DLL from disk doesn't do | |
| 82 // what we want so instead we pull the pages into memory and touch pages at | |
| 83 // a stride. We use the system's page size as the stride, ignoring the | |
| 84 // passed in step_size, to make sure each page in the range is touched. | |
| 85 base::MemoryMappedFile file_memory_map; | |
| 86 CHECK(file_memory_map.Initialize(base::FilePath(file_path))); | |
|
grt (UTC plus 2)
2015/10/29 19:15:06
why crash here but not on line 62?
fdoray
2015/10/30 21:08:54
Done. We shouldn't crash.
| |
| 87 TouchPagesInRange(file_memory_map.data(), file_memory_map.length()); | |
| 88 } | |
| 89 | |
| 90 return true; | |
| 91 } | |
| OLD | NEW |