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 "base/debug/profiler.h" | 5 #include "base/debug/profiler.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/process/process_handle.h" | 9 #include "base/process/process_handle.h" |
10 #include "base/strings/string_util.h" | 10 #include "base/strings/string_util.h" |
11 #include "base/strings/stringprintf.h" | 11 #include "base/strings/stringprintf.h" |
12 | 12 |
13 #if defined(OS_WIN) | |
14 #include "base/win/pe_image.h" | |
15 #endif // defined(OS_WIN) | |
16 | |
17 // TODO(peria): Enable profiling on Windows. | 13 // TODO(peria): Enable profiling on Windows. |
18 #if defined(ENABLE_PROFILING) && !defined(NO_TCMALLOC) && !defined(OS_WIN) | 14 #if defined(ENABLE_PROFILING) && !defined(NO_TCMALLOC) |
19 #include "third_party/tcmalloc/chromium/src/gperftools/profiler.h" | 15 #include "third_party/tcmalloc/chromium/src/gperftools/profiler.h" |
20 #endif | 16 #endif |
21 | 17 |
22 namespace base { | 18 namespace base { |
23 namespace debug { | 19 namespace debug { |
24 | 20 |
25 // TODO(peria): Enable profiling on Windows. | 21 // TODO(peria): Enable profiling on Windows. |
26 #if defined(ENABLE_PROFILING) && !defined(NO_TCMALLOC) && !defined(OS_WIN) | 22 #if defined(ENABLE_PROFILING) && !defined(NO_TCMALLOC) |
27 | 23 |
28 static int profile_count = 0; | 24 static int profile_count = 0; |
29 | 25 |
30 void StartProfiling(const std::string& name) { | 26 void StartProfiling(const std::string& name) { |
31 ++profile_count; | 27 ++profile_count; |
32 std::string full_name(name); | 28 std::string full_name(name); |
33 std::string pid = StringPrintf("%d", GetCurrentProcId()); | 29 std::string pid = StringPrintf("%d", GetCurrentProcId()); |
34 std::string count = StringPrintf("%d", profile_count); | 30 std::string count = StringPrintf("%d", profile_count); |
35 ReplaceSubstringsAfterOffset(&full_name, 0, "{pid}", pid); | 31 ReplaceSubstringsAfterOffset(&full_name, 0, "{pid}", pid); |
36 ReplaceSubstringsAfterOffset(&full_name, 0, "{count}", count); | 32 ReplaceSubstringsAfterOffset(&full_name, 0, "{count}", count); |
(...skipping 30 matching lines...) Expand all Loading... |
67 | 63 |
68 bool BeingProfiled() { | 64 bool BeingProfiled() { |
69 return false; | 65 return false; |
70 } | 66 } |
71 | 67 |
72 void RestartProfilingAfterFork() { | 68 void RestartProfilingAfterFork() { |
73 } | 69 } |
74 | 70 |
75 #endif | 71 #endif |
76 | 72 |
77 #if !defined(OS_WIN) | |
78 | |
79 bool IsBinaryInstrumented() { | 73 bool IsBinaryInstrumented() { |
80 return false; | 74 return false; |
81 } | 75 } |
82 | 76 |
83 ReturnAddressLocationResolver GetProfilerReturnAddrResolutionFunc() { | 77 ReturnAddressLocationResolver GetProfilerReturnAddrResolutionFunc() { |
84 return NULL; | 78 return NULL; |
85 } | 79 } |
86 | 80 |
87 DynamicFunctionEntryHook GetProfilerDynamicFunctionEntryHookFunc() { | 81 DynamicFunctionEntryHook GetProfilerDynamicFunctionEntryHookFunc() { |
88 return NULL; | 82 return NULL; |
89 } | 83 } |
90 | 84 |
91 AddDynamicSymbol GetProfilerAddDynamicSymbolFunc() { | 85 AddDynamicSymbol GetProfilerAddDynamicSymbolFunc() { |
92 return NULL; | 86 return NULL; |
93 } | 87 } |
94 | 88 |
95 MoveDynamicSymbol GetProfilerMoveDynamicSymbolFunc() { | 89 MoveDynamicSymbol GetProfilerMoveDynamicSymbolFunc() { |
96 return NULL; | 90 return NULL; |
97 } | 91 } |
98 | 92 |
99 #else // defined(OS_WIN) | |
100 | |
101 // http://blogs.msdn.com/oldnewthing/archive/2004/10/25/247180.aspx | |
102 extern "C" IMAGE_DOS_HEADER __ImageBase; | |
103 | |
104 bool IsBinaryInstrumented() { | |
105 enum InstrumentationCheckState { | |
106 UNINITIALIZED, | |
107 INSTRUMENTED_IMAGE, | |
108 NON_INSTRUMENTED_IMAGE, | |
109 }; | |
110 | |
111 static InstrumentationCheckState state = UNINITIALIZED; | |
112 | |
113 if (state == UNINITIALIZED) { | |
114 HMODULE this_module = reinterpret_cast<HMODULE>(&__ImageBase); | |
115 base::win::PEImage image(this_module); | |
116 | |
117 // Check to be sure our image is structured as we'd expect. | |
118 DCHECK(image.VerifyMagic()); | |
119 | |
120 // Syzygy-instrumented binaries contain a PE image section named ".thunks", | |
121 // and all Syzygy-modified binaries contain the ".syzygy" image section. | |
122 // This is a very fast check, as it only looks at the image header. | |
123 if ((image.GetImageSectionHeaderByName(".thunks") != NULL) && | |
124 (image.GetImageSectionHeaderByName(".syzygy") != NULL)) { | |
125 state = INSTRUMENTED_IMAGE; | |
126 } else { | |
127 state = NON_INSTRUMENTED_IMAGE; | |
128 } | |
129 } | |
130 DCHECK(state != UNINITIALIZED); | |
131 | |
132 return state == INSTRUMENTED_IMAGE; | |
133 } | |
134 | |
135 namespace { | |
136 | |
137 struct FunctionSearchContext { | |
138 const char* name; | |
139 FARPROC function; | |
140 }; | |
141 | |
142 // Callback function to PEImage::EnumImportChunks. | |
143 bool FindResolutionFunctionInImports( | |
144 const base::win::PEImage &image, const char* module_name, | |
145 PIMAGE_THUNK_DATA unused_name_table, PIMAGE_THUNK_DATA import_address_table, | |
146 PVOID cookie) { | |
147 FunctionSearchContext* context = | |
148 reinterpret_cast<FunctionSearchContext*>(cookie); | |
149 | |
150 DCHECK_NE(static_cast<FunctionSearchContext*>(NULL), context); | |
151 DCHECK_EQ(static_cast<FARPROC>(NULL), context->function); | |
152 | |
153 // Our import address table contains pointers to the functions we import | |
154 // at this point. Let's retrieve the first such function and use it to | |
155 // find the module this import was resolved to by the loader. | |
156 const wchar_t* function_in_module = | |
157 reinterpret_cast<const wchar_t*>(import_address_table->u1.Function); | |
158 | |
159 // Retrieve the module by a function in the module. | |
160 const DWORD kFlags = GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | | |
161 GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT; | |
162 HMODULE module = NULL; | |
163 if (!::GetModuleHandleEx(kFlags, function_in_module, &module)) { | |
164 // This can happen if someone IAT patches us to a thunk. | |
165 return true; | |
166 } | |
167 | |
168 // See whether this module exports the function we're looking for. | |
169 FARPROC exported_func = ::GetProcAddress(module, context->name); | |
170 if (exported_func != NULL) { | |
171 // We found it, return the function and terminate the enumeration. | |
172 context->function = exported_func; | |
173 return false; | |
174 } | |
175 | |
176 // Keep going. | |
177 return true; | |
178 } | |
179 | |
180 template <typename FunctionType> | |
181 FunctionType FindFunctionInImports(const char* function_name) { | |
182 if (!IsBinaryInstrumented()) | |
183 return NULL; | |
184 | |
185 HMODULE this_module = reinterpret_cast<HMODULE>(&__ImageBase); | |
186 base::win::PEImage image(this_module); | |
187 | |
188 FunctionSearchContext ctx = { function_name, NULL }; | |
189 image.EnumImportChunks(FindResolutionFunctionInImports, &ctx); | |
190 | |
191 return reinterpret_cast<FunctionType>(ctx.function); | |
192 } | |
193 | |
194 } // namespace | |
195 | |
196 ReturnAddressLocationResolver GetProfilerReturnAddrResolutionFunc() { | |
197 return FindFunctionInImports<ReturnAddressLocationResolver>( | |
198 "ResolveReturnAddressLocation"); | |
199 } | |
200 | |
201 DynamicFunctionEntryHook GetProfilerDynamicFunctionEntryHookFunc() { | |
202 return FindFunctionInImports<DynamicFunctionEntryHook>( | |
203 "OnDynamicFunctionEntry"); | |
204 } | |
205 | |
206 AddDynamicSymbol GetProfilerAddDynamicSymbolFunc() { | |
207 return FindFunctionInImports<AddDynamicSymbol>( | |
208 "AddDynamicSymbol"); | |
209 } | |
210 | |
211 MoveDynamicSymbol GetProfilerMoveDynamicSymbolFunc() { | |
212 return FindFunctionInImports<MoveDynamicSymbol>( | |
213 "MoveDynamicSymbol"); | |
214 } | |
215 | |
216 #endif // defined(OS_WIN) | |
217 | 93 |
218 } // namespace debug | 94 } // namespace debug |
219 } // namespace base | 95 } // namespace base |
OLD | NEW |