OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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/installer/test/resource_loader.h" | 5 #include "chrome/installer/test/resource_loader.h" |
6 | 6 |
7 #include <windows.h> | 7 #include <windows.h> |
8 | 8 |
9 #include "base/files/file_path.h" | 9 #include "base/files/file_path.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 | 11 |
12 namespace { | 12 namespace { |
13 | 13 |
14 // Populates |resource_data| with the address and size of the resource in | 14 // Populates |resource_data| with the address and size of the resource in |
15 // |module| identified by |name_or_id| of type |type_name_or_id|, returning | 15 // |module| identified by |name_or_id| of type |type_name_or_id|, returning |
16 // true on success. | 16 // true on success. |
17 bool DoLoad(HMODULE module, const wchar_t* name_or_id, | 17 bool DoLoad(HMODULE module, |
| 18 const wchar_t* name_or_id, |
18 const wchar_t* type_name_or_id, | 19 const wchar_t* type_name_or_id, |
19 std::pair<const uint8*, DWORD>* resource_data) { | 20 std::pair<const uint8_t*, DWORD>* resource_data) { |
20 bool loaded = false; | 21 bool loaded = false; |
21 HRSRC resource_info; | 22 HRSRC resource_info; |
22 | 23 |
23 resource_info = FindResource(module, name_or_id, type_name_or_id); | 24 resource_info = FindResource(module, name_or_id, type_name_or_id); |
24 if (resource_info != NULL) { | 25 if (resource_info != NULL) { |
25 HGLOBAL loaded_resource; | 26 HGLOBAL loaded_resource; |
26 | 27 |
27 loaded_resource = LoadResource(module, resource_info); | 28 loaded_resource = LoadResource(module, resource_info); |
28 if (loaded_resource != NULL) { | 29 if (loaded_resource != NULL) { |
29 resource_data->first = | 30 resource_data->first = |
30 static_cast<const uint8*>(LockResource(loaded_resource)); | 31 static_cast<const uint8_t*>(LockResource(loaded_resource)); |
31 if (resource_data->first != NULL) { | 32 if (resource_data->first != NULL) { |
32 resource_data->second = SizeofResource(module, resource_info); | 33 resource_data->second = SizeofResource(module, resource_info); |
33 DPCHECK(resource_data->second != 0); | 34 DPCHECK(resource_data->second != 0); |
34 loaded = true; | 35 loaded = true; |
35 } else { | 36 } else { |
36 DPCHECK(false) << "LockResource failed"; | 37 DPCHECK(false) << "LockResource failed"; |
37 } | 38 } |
38 } else { | 39 } else { |
39 DPCHECK(false) << "LoadResource failed"; | 40 DPCHECK(false) << "LoadResource failed"; |
40 } | 41 } |
(...skipping 23 matching lines...) Expand all Loading... |
64 module_ = LoadLibraryEx(pe_image_path.value().c_str(), NULL, | 65 module_ = LoadLibraryEx(pe_image_path.value().c_str(), NULL, |
65 (LOAD_LIBRARY_AS_DATAFILE_EXCLUSIVE | | 66 (LOAD_LIBRARY_AS_DATAFILE_EXCLUSIVE | |
66 LOAD_LIBRARY_AS_IMAGE_RESOURCE)); | 67 LOAD_LIBRARY_AS_IMAGE_RESOURCE)); |
67 DPLOG_IF(INFO, module_ == NULL) | 68 DPLOG_IF(INFO, module_ == NULL) |
68 << "Failed loading \"" << pe_image_path.value() << "\""; | 69 << "Failed loading \"" << pe_image_path.value() << "\""; |
69 return module_ != NULL; | 70 return module_ != NULL; |
70 } | 71 } |
71 | 72 |
72 bool ResourceLoader::Load(const std::wstring& name, | 73 bool ResourceLoader::Load(const std::wstring& name, |
73 const std::wstring& type, | 74 const std::wstring& type, |
74 std::pair<const uint8*, DWORD>* resource_data) { | 75 std::pair<const uint8_t*, DWORD>* resource_data) { |
75 DCHECK(resource_data != NULL); | 76 DCHECK(resource_data != NULL); |
76 DCHECK(module_ != NULL); | 77 DCHECK(module_ != NULL); |
77 | 78 |
78 return DoLoad(module_, name.c_str(), type.c_str(), resource_data); | 79 return DoLoad(module_, name.c_str(), type.c_str(), resource_data); |
79 } | 80 } |
80 | 81 |
81 bool ResourceLoader::Load(WORD id, | 82 bool ResourceLoader::Load(WORD id, |
82 WORD type, | 83 WORD type, |
83 std::pair<const uint8*, DWORD>* resource_data) { | 84 std::pair<const uint8_t*, DWORD>* resource_data) { |
84 DCHECK(resource_data != NULL); | 85 DCHECK(resource_data != NULL); |
85 DCHECK(module_ != NULL); | 86 DCHECK(module_ != NULL); |
86 | 87 |
87 return DoLoad(module_, MAKEINTRESOURCE(id), MAKEINTRESOURCE(type), | 88 return DoLoad(module_, MAKEINTRESOURCE(id), MAKEINTRESOURCE(type), |
88 resource_data); | 89 resource_data); |
89 } | 90 } |
90 | 91 |
91 } // namespace upgrade_test | 92 } // namespace upgrade_test |
OLD | NEW |