| OLD | NEW | 
|    1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |    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 |    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 "chrome/app/image_pre_reader_win.h" |    5 #include "chrome/app/image_pre_reader_win.h" | 
|    6  |    6  | 
|    7 #include <windows.h> |    7 #include <windows.h> | 
|    8 #include <algorithm> |    8 #include <algorithm> | 
|    9 #include <limits> |    9 #include <limits> | 
|   10 #include <vector> |   10 #include <vector> | 
|   11  |   11  | 
|   12 #include "base/logging.h" |   12 #include "base/logging.h" | 
|   13 #include "base/memory/scoped_ptr.h" |   13 #include "base/memory/scoped_ptr.h" | 
|   14 #include "base/threading/thread_restrictions.h" |   14 #include "base/threading/thread_restrictions.h" | 
|   15 #include "base/win/pe_image.h" |   15 #include "base/win/pe_image.h" | 
|   16 #include "base/win/scoped_handle.h" |   16 #include "base/win/scoped_handle.h" | 
|   17 #include "base/win/windows_version.h" |   17 #include "base/win/windows_version.h" | 
|   18  |   18  | 
|   19 namespace { |   19 namespace { | 
|   20  |   20  | 
|   21 // The minimum buffer size to allocate when reading the PE file headers. |   21 // The minimum buffer size to allocate when reading the PE file headers. | 
|   22 // |   22 // | 
|   23 // The PE file headers usually fit into a single 1KB page, and a PE file must |   23 // The PE file headers usually fit into a single 1KB page, and a PE file must | 
|   24 // at least contain the initial page with the headers. That said, as long as |   24 // at least contain the initial page with the headers. That said, as long as | 
|   25 // we expect at least sizeof(IMAGE_DOS_HEADER) bytes, we're ok. |   25 // we expect at least sizeof(IMAGE_DOS_HEADER) bytes, we're ok. | 
|   26 const size_t kMinHeaderBufferSize = 0x400; |   26 const size_t kMinHeaderBufferSize = 0x400; | 
|   27  |   27  | 
|   28 // A handy symbolic constant. |   28 // A handy symbolic constant. | 
|   29 const uint8 kOneHundredPercent = 100; |   29 const size_t kOneHundredPercent = 100; | 
|   30  |   30  | 
|   31 void StaticAssertions() { |   31 void StaticAssertions() { | 
|   32   COMPILE_ASSERT(kMinHeaderBufferSize >= sizeof(IMAGE_DOS_HEADER), |   32   COMPILE_ASSERT(kMinHeaderBufferSize >= sizeof(IMAGE_DOS_HEADER), | 
|   33                  min_header_buffer_size_at_least_as_big_as_the_dos_header); |   33                  min_header_buffer_size_at_least_as_big_as_the_dos_header); | 
|   34 } |   34 } | 
|   35  |   35  | 
|   36 // This struct provides a deallocation functor for use with scoped_ptr<T> |   36 // This struct provides a deallocation functor for use with scoped_ptr<T> | 
|   37 // allocated with ::VirtualAlloc(). |   37 // allocated with ::VirtualAlloc(). | 
|   38 struct ScopedPtrVirtualFree { |   38 struct ScopedPtrVirtualFree { | 
|   39   void operator() (void* ptr) { |   39   void operator() (void* ptr) { | 
| (...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
|   98 // |   98 // | 
|   99 // Note that sections can have their tails implicitly initialized to zero |   99 // Note that sections can have their tails implicitly initialized to zero | 
|  100 // (i.e., their virtual size is larger than the raw size) and that raw data |  100 // (i.e., their virtual size is larger than the raw size) and that raw data | 
|  101 // is padded to the PE page size if the entire section is initialized (i.e., |  101 // is padded to the PE page size if the entire section is initialized (i.e., | 
|  102 // their raw data size will be larger than the virtual size). |  102 // their raw data size will be larger than the virtual size). | 
|  103 // |  103 // | 
|  104 // Any data after the initialized portion of the section will be soft-faulted |  104 // Any data after the initialized portion of the section will be soft-faulted | 
|  105 // in (very quickly) as needed, so we don't need to include it in the returned |  105 // in (very quickly) as needed, so we don't need to include it in the returned | 
|  106 // length. |  106 // length. | 
|  107 size_t GetPercentageOfSectionLength(const IMAGE_SECTION_HEADER* section, |  107 size_t GetPercentageOfSectionLength(const IMAGE_SECTION_HEADER* section, | 
|  108                                     uint8 percentage) { |  108                                     size_t percentage) { | 
|  109   DCHECK(section != NULL); |  109   DCHECK(section != NULL); | 
|  110   DCHECK_GT(percentage, 0); |  110   DCHECK_GT(percentage, 0u); | 
|  111   DCHECK_LE(percentage, kOneHundredPercent); |  111   DCHECK_LE(percentage, kOneHundredPercent); | 
|  112  |  112  | 
|  113   size_t initialized_length = std::min(section->SizeOfRawData, |  113   size_t initialized_length = std::min(section->SizeOfRawData, | 
|  114                                        section->Misc.VirtualSize); |  114                                        section->Misc.VirtualSize); | 
|  115  |  115  | 
|  116   if (initialized_length == 0) |  116   if (initialized_length == 0) | 
|  117     return 0; |  117     return 0; | 
|  118  |  118  | 
|  119   size_t length = (initialized_length * percentage) / kOneHundredPercent; |  119   size_t length = (initialized_length * percentage) / kOneHundredPercent; | 
|  120  |  120  | 
|  121   return std::max<size_t>(length, 1); |  121   return std::max<size_t>(length, 1); | 
|  122 } |  122 } | 
|  123  |  123  | 
|  124 // Helper function to read through a |percentage| of the given |section| |  124 // Helper function to read through a |percentage| of the given |section| | 
|  125 // of the file denoted by |file_handle|. The |temp_buffer| is (re)used as |  125 // of the file denoted by |file_handle|. The |temp_buffer| is (re)used as | 
|  126 // a transient storage area as the section is read in chunks of |  126 // a transient storage area as the section is read in chunks of | 
|  127 // |temp_buffer_size| bytes. |  127 // |temp_buffer_size| bytes. | 
|  128 bool ReadThroughSection(HANDLE file_handle, |  128 bool ReadThroughSection(HANDLE file_handle, | 
|  129                         const IMAGE_SECTION_HEADER* section, |  129                         const IMAGE_SECTION_HEADER* section, | 
|  130                         uint8 percentage, |  130                         size_t percentage, | 
|  131                         void* temp_buffer, |  131                         void* temp_buffer, | 
|  132                         size_t temp_buffer_size) { |  132                         size_t temp_buffer_size) { | 
|  133   DCHECK(file_handle != INVALID_HANDLE_VALUE); |  133   DCHECK(file_handle != INVALID_HANDLE_VALUE); | 
|  134   DCHECK(section != NULL); |  134   DCHECK(section != NULL); | 
|  135   DCHECK_LE(percentage, kOneHundredPercent); |  135   DCHECK_LE(percentage, kOneHundredPercent); | 
|  136   DCHECK(temp_buffer != NULL); |  136   DCHECK(temp_buffer != NULL); | 
|  137   DCHECK(temp_buffer_size > 0); |  137   DCHECK(temp_buffer_size > 0); | 
|  138  |  138  | 
|  139   size_t bytes_to_read = GetPercentageOfSectionLength(section, percentage); |  139   size_t bytes_to_read = GetPercentageOfSectionLength(section, percentage); | 
|  140   if (bytes_to_read == 0) |  140   if (bytes_to_read == 0) | 
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
|  182   while (touch_ptr < final_touch_ptr) { |  182   while (touch_ptr < final_touch_ptr) { | 
|  183     dummy = *touch_ptr; |  183     dummy = *touch_ptr; | 
|  184     touch_ptr += system_info.dwPageSize; |  184     touch_ptr += system_info.dwPageSize; | 
|  185   } |  185   } | 
|  186   dummy = *final_touch_ptr; |  186   dummy = *final_touch_ptr; | 
|  187 } |  187 } | 
|  188  |  188  | 
|  189 }  // namespace |  189 }  // namespace | 
|  190  |  190  | 
|  191 bool ImagePreReader::PartialPreReadImageOnDisk(const wchar_t* file_path, |  191 bool ImagePreReader::PartialPreReadImageOnDisk(const wchar_t* file_path, | 
|  192                                                uint8 percentage, |  192                                                size_t percentage, | 
|  193                                                size_t max_chunk_size) { |  193                                                size_t max_chunk_size) { | 
|  194   // TODO(rogerm): change this to have the number of bytes pre-read per |  194   // TODO(rogerm): change this to have the number of bytes pre-read per | 
|  195   //     section be driven by a static table within the PE file (defaulting to |  195   //     section be driven by a static table within the PE file (defaulting to | 
|  196   //     full read if it's not there?) that's initialized by the optimization |  196   //     full read if it's not there?) that's initialized by the optimization | 
|  197   //     toolchain. |  197   //     toolchain. | 
|  198   DCHECK(file_path != NULL); |  198   DCHECK(file_path != NULL); | 
|  199  |  199  | 
|  200   if (percentage == 0) |  200   if (percentage == 0) | 
|  201     return true; |  201     return true; | 
|  202  |  202  | 
| (...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
|  266     if (!ReadThroughSection( |  266     if (!ReadThroughSection( | 
|  267             file, section, percentage, buffer.get(), max_chunk_size)) |  267             file, section, percentage, buffer.get(), max_chunk_size)) | 
|  268       return false; |  268       return false; | 
|  269   } |  269   } | 
|  270  |  270  | 
|  271   // We're done. |  271   // We're done. | 
|  272   return true; |  272   return true; | 
|  273 } |  273 } | 
|  274  |  274  | 
|  275 bool ImagePreReader::PartialPreReadImageInMemory(const wchar_t* file_path, |  275 bool ImagePreReader::PartialPreReadImageInMemory(const wchar_t* file_path, | 
|  276                                                  uint8 percentage) { |  276                                                  size_t percentage) { | 
|  277   // TODO(rogerm): change this to have the number of bytes pre-read per |  277   // TODO(rogerm): change this to have the number of bytes pre-read per | 
|  278   //     section be driven by a static table within the PE file (defaulting to |  278   //     section be driven by a static table within the PE file (defaulting to | 
|  279   //     full read if it's not there?) that's initialized by the optimization |  279   //     full read if it's not there?) that's initialized by the optimization | 
|  280   //     toolchain. |  280   //     toolchain. | 
|  281   DCHECK(file_path != NULL); |  281   DCHECK(file_path != NULL); | 
|  282  |  282  | 
|  283   if (percentage == 0) |  283   if (percentage == 0) | 
|  284     return true; |  284     return true; | 
|  285  |  285  | 
|  286   if (percentage > kOneHundredPercent) |  286   if (percentage > kOneHundredPercent) | 
| (...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
|  389  |  389  | 
|  390     // Page in then release the module. |  390     // Page in then release the module. | 
|  391     TouchPagesInRange(dll_module, dll_module_length); |  391     TouchPagesInRange(dll_module, dll_module_length); | 
|  392     FreeLibrary(dll_module); |  392     FreeLibrary(dll_module); | 
|  393   } |  393   } | 
|  394  |  394  | 
|  395   return true; |  395   return true; | 
|  396 } |  396 } | 
|  397  |  397  | 
|  398 bool ImagePreReader::PartialPreReadImage(const wchar_t* file_path, |  398 bool ImagePreReader::PartialPreReadImage(const wchar_t* file_path, | 
|  399                                          uint8 percentage, |  399                                          size_t percentage, | 
|  400                                          size_t max_chunk_size) { |  400                                          size_t max_chunk_size) { | 
|  401   base::ThreadRestrictions::AssertIOAllowed(); |  401   base::ThreadRestrictions::AssertIOAllowed(); | 
|  402  |  402  | 
|  403   if (percentage >= kOneHundredPercent) { |  403   if (percentage >= kOneHundredPercent) { | 
|  404     // If we're reading the whole image, we don't need to parse headers and |  404     // If we're reading the whole image, we don't need to parse headers and | 
|  405     // navigate sections, the basic PreReadImage() can be used to just step |  405     // navigate sections, the basic PreReadImage() can be used to just step | 
|  406     // blindly through the entire file / address-space. |  406     // blindly through the entire file / address-space. | 
|  407     return PreReadImage(file_path, 0, max_chunk_size); |  407     return PreReadImage(file_path, 0, max_chunk_size); | 
|  408   } |  408   } | 
|  409  |  409  | 
|  410   if (base::win::GetVersion() > base::win::VERSION_XP) { |  410   if (base::win::GetVersion() > base::win::VERSION_XP) { | 
|  411     // Vista+ branch. On these OSes, we warm up the Image by reading its |  411     // Vista+ branch. On these OSes, we warm up the Image by reading its | 
|  412     // file off the disk. |  412     // file off the disk. | 
|  413     return PartialPreReadImageOnDisk(file_path, percentage, max_chunk_size); |  413     return PartialPreReadImageOnDisk(file_path, percentage, max_chunk_size); | 
|  414   } |  414   } | 
|  415  |  415  | 
|  416   // WinXP branch. For XP, reading the image from disk doesn't do what we want |  416   // WinXP branch. For XP, reading the image from disk doesn't do what we want | 
|  417   // so instead we pull the pages into memory by loading the DLL and touching |  417   // so instead we pull the pages into memory by loading the DLL and touching | 
|  418   // initialized pages at a stride. |  418   // initialized pages at a stride. | 
|  419   return PartialPreReadImageInMemory(file_path, percentage); |  419   return PartialPreReadImageInMemory(file_path, percentage); | 
|  420 } |  420 } | 
| OLD | NEW |