OLD | NEW |
---|---|
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 #ifndef BASE_PROFILER_STACK_SAMPLING_PROFILER_H_ | 5 #ifndef BASE_PROFILER_STACK_SAMPLING_PROFILER_H_ |
6 #define BASE_PROFILER_STACK_SAMPLING_PROFILER_H_ | 6 #define BASE_PROFILER_STACK_SAMPLING_PROFILER_H_ |
7 | 7 |
8 #include <stddef.h> | 8 #include <stddef.h> |
9 | 9 |
10 #include <memory> | 10 #include <memory> |
11 #include <string> | 11 #include <string> |
12 #include <vector> | 12 #include <vector> |
13 | 13 |
14 #include "base/atomicops.h" | 14 #include "base/atomicops.h" |
15 #include "base/base_export.h" | 15 #include "base/base_export.h" |
16 #include "base/callback.h" | 16 #include "base/callback.h" |
17 #include "base/files/file_path.h" | 17 #include "base/files/file_path.h" |
18 #include "base/macros.h" | 18 #include "base/macros.h" |
19 #include "base/strings/string16.h" | 19 #include "base/strings/string16.h" |
20 #include "base/synchronization/waitable_event.h" | 20 #include "base/synchronization/waitable_event.h" |
21 #include "base/threading/platform_thread.h" | 21 #include "base/threading/platform_thread.h" |
22 #include "base/time/time.h" | 22 #include "base/time/time.h" |
23 | 23 |
24 namespace base { | 24 namespace base { |
25 | 25 |
26 class NativeStackSampler; | 26 class NativeStackSampler; |
27 class NativeStackSamplerTestDelegate; | 27 class NativeStackSamplerTestDelegate; |
28 class WaitableEvent; | |
28 | 29 |
29 // StackSamplingProfiler periodically stops a thread to sample its stack, for | 30 // StackSamplingProfiler periodically stops a thread to sample its stack, for |
30 // the purpose of collecting information about which code paths are | 31 // the purpose of collecting information about which code paths are |
31 // executing. This information is used in aggregate by UMA to identify hot | 32 // executing. This information is used in aggregate by UMA to identify hot |
32 // and/or janky code paths. | 33 // and/or janky code paths. |
33 // | 34 // |
35 // IMPORTANT: Only threads guaranteed to live beyond the lifetime of the | |
36 // profiler object can be safely sampled. Sampling the activity of the current | |
37 // thread within the scope of this object is safe, but sampling any other | |
38 // thread could cause Bad Things(tm) to occur should that thread exit before | |
39 // sampling is complete; ensure the profiler object gets destructed before | |
40 // exit of the thread under test. USE WITH CAUTION! | |
41 // | |
Mike Wittman
2017/02/13 22:35:57
If we refactor the constructors to:
StackSampli
bcwhite
2017/02/14 14:33:02
I like it. Done.
| |
34 // Sample StackSamplingProfiler usage: | 42 // Sample StackSamplingProfiler usage: |
35 // | 43 // |
36 // // Create and customize params as desired. | 44 // // Create and customize params as desired. |
37 // base::StackStackSamplingProfiler::SamplingParams params; | 45 // base::StackStackSamplingProfiler::SamplingParams params; |
38 // // Any thread's ID may be passed as the target. | 46 // // Any thread's ID may be passed as the target. |
39 // base::StackSamplingProfiler profiler(base::PlatformThread::CurrentId()), | 47 // base::StackSamplingProfiler profiler(base::PlatformThread::CurrentId()), |
40 // params); | 48 // params); |
41 // | 49 // |
42 // // Or, to process the profiles within Chrome rather than via UMA, use a | 50 // // Or, to process the profiles within Chrome rather than via UMA, use a |
43 // // custom completed callback: | 51 // // custom completed callback: |
(...skipping 154 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
198 StackSamplingProfiler(PlatformThreadId thread_id, | 206 StackSamplingProfiler(PlatformThreadId thread_id, |
199 const SamplingParams& params, | 207 const SamplingParams& params, |
200 const CompletedCallback& callback); | 208 const CompletedCallback& callback); |
201 StackSamplingProfiler(PlatformThreadId thread_id, | 209 StackSamplingProfiler(PlatformThreadId thread_id, |
202 const SamplingParams& params, | 210 const SamplingParams& params, |
203 const CompletedCallback& callback, | 211 const CompletedCallback& callback, |
204 NativeStackSamplerTestDelegate* test_delegate); | 212 NativeStackSamplerTestDelegate* test_delegate); |
205 // Stops any profiling currently taking place before destroying the profiler. | 213 // Stops any profiling currently taking place before destroying the profiler. |
206 ~StackSamplingProfiler(); | 214 ~StackSamplingProfiler(); |
207 | 215 |
208 // The fire-and-forget interface: starts a profiler and allows it to complete | |
209 // without the caller needing to manage the profiler lifetime. May be invoked | |
210 // from any thread, but requires that the calling thread has a message loop. | |
211 static void StartAndRunAsync(PlatformThreadId thread_id, | |
212 const SamplingParams& params, | |
213 const CompletedCallback& callback); | |
214 | |
215 // Initializes the profiler and starts sampling. | 216 // Initializes the profiler and starts sampling. |
216 void Start(); | 217 void Start(); |
217 | 218 |
218 // Stops the profiler and any ongoing sampling. Calling this function is | 219 // Stops the profiler and any ongoing sampling. Calling this function is |
219 // optional; if not invoked profiling terminates when all the profiling bursts | 220 // optional; if not invoked profiling terminates when all the profiling bursts |
220 // specified in the SamplingParams are completed or the profiler is destroyed, | 221 // specified in the SamplingParams are completed or the profiler is destroyed, |
221 // whichever occurs first. | 222 // whichever occurs first. |
222 void Stop(); | 223 void Stop(); |
223 | 224 |
225 // Stops all active profiles and cleans up any resources in anticipation of | |
226 // a shutdown of the current process. | |
227 static void Shutdown(); | |
228 | |
229 // Resets the "shutdown" state so that the sampling thread can be restarted. | |
230 static void UndoShutdownForTesting(); | |
231 | |
232 // Returns whether the sampling thread is currently running or not. | |
233 static bool IsSamplingThreadRunningForTesting(); | |
234 | |
235 // Sets the auto-shutdown delay time for the sampling thread, in ms. Set this | |
236 // to zero to disable it. | |
237 static void SetSamplingThreadIdleShutdownTimeForTesting(int shutdown_ms); | |
238 | |
239 // Initiates an idle shutdown task, as though the idle timer had expired, | |
240 // causing the thread to exit if there are no more sampling tasks pending. | |
241 // This returns immediately. Watch IsSamplingThreadRunningForTesting() to | |
242 // know when the thread has exited, though it will never do so if it is | |
243 // not idle. | |
244 static void InitiateSamplingThreadIdleShutdownForTesting(); | |
245 | |
224 // Set the current system state that is recorded with each captured stack | 246 // Set the current system state that is recorded with each captured stack |
225 // frame. This is thread-safe so can be called from anywhere. The parameter | 247 // frame. This is thread-safe so can be called from anywhere. The parameter |
226 // value should be from an enumeration of the appropriate type with values | 248 // value should be from an enumeration of the appropriate type with values |
227 // ranging from 0 to 31, inclusive. This sets bits within Sample field of | 249 // ranging from 0 to 31, inclusive. This sets bits within Sample field of |
228 // |process_milestones|. The actual meanings of these bits are defined | 250 // |process_milestones|. The actual meanings of these bits are defined |
229 // (globally) by the caller(s). | 251 // (globally) by the caller(s). |
230 static void SetProcessMilestone(int milestone); | 252 static void SetProcessMilestone(int milestone); |
231 static void ResetAnnotationsForTesting(); | 253 static void ResetAnnotationsForTesting(); |
232 | 254 |
233 private: | 255 private: |
234 // SamplingThread is a separate thread used to suspend and sample stacks from | 256 // SamplingThread is a separate thread used to suspend and sample stacks from |
235 // the target thread. | 257 // the target thread. |
236 class SamplingThread : public PlatformThread::Delegate { | 258 class SamplingThread; |
237 public: | |
238 // Samples stacks using |native_sampler|. When complete, invokes | |
239 // |completed_callback| with the collected call stack profiles. | |
240 // |completed_callback| must be callable on any thread. | |
241 SamplingThread(std::unique_ptr<NativeStackSampler> native_sampler, | |
242 const SamplingParams& params, | |
243 const CompletedCallback& completed_callback); | |
244 ~SamplingThread() override; | |
245 | |
246 // PlatformThread::Delegate: | |
247 void ThreadMain() override; | |
248 | |
249 void Stop(); | |
250 | |
251 private: | |
252 // Collects |profile| from a single burst. If the profiler was stopped | |
253 // during collection, sets |was_stopped| and provides the set of samples | |
254 // collected up to that point. | |
255 void CollectProfile(CallStackProfile* profile, TimeDelta* elapsed_time, | |
256 bool* was_stopped); | |
257 | |
258 // Collects call stack profiles from all bursts, or until the sampling is | |
259 // stopped. If stopped before complete, the last profile in | |
260 // |call_stack_profiles| may contain a partial burst. | |
261 void CollectProfiles(CallStackProfiles* profiles); | |
262 | |
263 std::unique_ptr<NativeStackSampler> native_sampler_; | |
264 const SamplingParams params_; | |
265 | |
266 // If Stop() is called, it signals this event to force the sampling to | |
267 // terminate before all the samples specified in |params_| are collected. | |
268 WaitableEvent stop_event_; | |
269 | |
270 const CompletedCallback completed_callback_; | |
271 | |
272 DISALLOW_COPY_AND_ASSIGN(SamplingThread); | |
273 }; | |
274 | 259 |
275 // Adds annotations to a Sample. | 260 // Adds annotations to a Sample. |
276 static void RecordAnnotations(Sample* sample); | 261 static void RecordAnnotations(Sample* sample); |
277 | 262 |
278 // This global variables holds the current system state and is recorded with | 263 // This global variables holds the current system state and is recorded with |
279 // every captured sample, done on a separate thread which is why updates to | 264 // every captured sample, done on a separate thread which is why updates to |
280 // this must be atomic. A PostTask to move the the updates to that thread | 265 // this must be atomic. A PostTask to move the the updates to that thread |
281 // would skew the timing and a lock could result in deadlock if the thread | 266 // would skew the timing and a lock could result in deadlock if the thread |
282 // making a change was also being profiled and got stopped. | 267 // making a change was also being profiled and got stopped. |
283 static subtle::Atomic32 process_milestones_; | 268 static subtle::Atomic32 process_milestones_; |
284 | 269 |
285 // The thread whose stack will be sampled. | 270 // The thread whose stack will be sampled. |
286 PlatformThreadId thread_id_; | 271 PlatformThreadId thread_id_; |
287 | 272 |
288 const SamplingParams params_; | 273 const SamplingParams params_; |
289 | 274 |
290 std::unique_ptr<SamplingThread> sampling_thread_; | 275 const CompletedCallback completed_callback_; |
291 PlatformThreadHandle sampling_thread_handle_; | |
292 | 276 |
293 const CompletedCallback completed_callback_; | 277 // An event signaled when all sampling is complete and the callback done. |
278 WaitableEvent finished_event_; | |
279 | |
280 // Object that does the native sampling. This is created during construction | |
281 // and later passed to the sampling thread when profiling is started. | |
282 std::unique_ptr<NativeStackSampler> native_sampler_; | |
283 | |
284 // An ID uniquely identifying this collection to the sampling thread. | |
285 int collection_id_ = -1; | |
294 | 286 |
295 // Stored until it can be passed to the NativeStackSampler created in Start(). | 287 // Stored until it can be passed to the NativeStackSampler created in Start(). |
296 NativeStackSamplerTestDelegate* const test_delegate_; | 288 NativeStackSamplerTestDelegate* const test_delegate_; |
297 | 289 |
298 DISALLOW_COPY_AND_ASSIGN(StackSamplingProfiler); | 290 DISALLOW_COPY_AND_ASSIGN(StackSamplingProfiler); |
299 }; | 291 }; |
300 | 292 |
301 // These operators permit types to be compared and used in a map of Samples, as | 293 // These operators permit types to be compared and used in a map of Samples, as |
302 // done in tests and by the metrics provider code. | 294 // done in tests and by the metrics provider code. |
303 BASE_EXPORT bool operator==(const StackSamplingProfiler::Module& a, | 295 BASE_EXPORT bool operator==(const StackSamplingProfiler::Module& a, |
304 const StackSamplingProfiler::Module& b); | 296 const StackSamplingProfiler::Module& b); |
305 BASE_EXPORT bool operator==(const StackSamplingProfiler::Sample& a, | 297 BASE_EXPORT bool operator==(const StackSamplingProfiler::Sample& a, |
306 const StackSamplingProfiler::Sample& b); | 298 const StackSamplingProfiler::Sample& b); |
307 BASE_EXPORT bool operator!=(const StackSamplingProfiler::Sample& a, | 299 BASE_EXPORT bool operator!=(const StackSamplingProfiler::Sample& a, |
308 const StackSamplingProfiler::Sample& b); | 300 const StackSamplingProfiler::Sample& b); |
309 BASE_EXPORT bool operator<(const StackSamplingProfiler::Sample& a, | 301 BASE_EXPORT bool operator<(const StackSamplingProfiler::Sample& a, |
310 const StackSamplingProfiler::Sample& b); | 302 const StackSamplingProfiler::Sample& b); |
311 BASE_EXPORT bool operator==(const StackSamplingProfiler::Frame& a, | 303 BASE_EXPORT bool operator==(const StackSamplingProfiler::Frame& a, |
312 const StackSamplingProfiler::Frame& b); | 304 const StackSamplingProfiler::Frame& b); |
313 BASE_EXPORT bool operator<(const StackSamplingProfiler::Frame& a, | 305 BASE_EXPORT bool operator<(const StackSamplingProfiler::Frame& a, |
314 const StackSamplingProfiler::Frame& b); | 306 const StackSamplingProfiler::Frame& b); |
315 | 307 |
316 } // namespace base | 308 } // namespace base |
317 | 309 |
318 #endif // BASE_PROFILER_STACK_SAMPLING_PROFILER_H_ | 310 #endif // BASE_PROFILER_STACK_SAMPLING_PROFILER_H_ |
OLD | NEW |