OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 CONTENT_BROWSER_TRACING_ETW_SYSTEM_EVENT_CONSUMER_WIN_H_ | 5 #ifndef CONTENT_BROWSER_TRACING_ETW_SYSTEM_EVENT_CONSUMER_WIN_H_ |
6 #define CONTENT_BROWSER_TRACING_ETW_SYSTEM_EVENT_CONSUMER_WIN_H_ | 6 #define CONTENT_BROWSER_TRACING_ETW_SYSTEM_EVENT_CONSUMER_WIN_H_ |
7 | 7 |
8 #include "base/bind.h" | 8 #include "base/bind.h" |
9 #include "base/memory/ref_counted_memory.h" | 9 #include "base/memory/ref_counted_memory.h" |
10 #include "base/threading/thread.h" | 10 #include "base/threading/thread.h" |
| 11 #include "base/trace_event/tracing_agent.h" |
11 #include "base/values.h" | 12 #include "base/values.h" |
12 #include "base/win/event_trace_consumer.h" | 13 #include "base/win/event_trace_consumer.h" |
13 #include "base/win/event_trace_controller.h" | 14 #include "base/win/event_trace_controller.h" |
14 | 15 |
15 namespace base { | 16 namespace base { |
16 template <typename Type> | 17 template <typename Type> |
17 struct DefaultSingletonTraits; | 18 struct DefaultSingletonTraits; |
18 } | 19 } |
19 | 20 |
20 namespace content { | 21 namespace content { |
21 | 22 |
22 class EtwSystemEventConsumer : | 23 class EtwSystemEventConsumer |
23 public base::win::EtwTraceConsumerBase<EtwSystemEventConsumer> { | 24 : public base::win::EtwTraceConsumerBase<EtwSystemEventConsumer>, |
| 25 public base::trace_event::TracingAgent { |
24 public: | 26 public: |
25 typedef base::Callback<void(const scoped_refptr<base::RefCountedString>&)> | 27 // base::trace_event::TracingAgent implementation. |
26 OutputCallback; | 28 std::string GetTracingAgentName() override; |
27 | 29 std::string GetTraceEventLabel() override; |
28 bool StartSystemTracing(); | 30 bool StartAgentTracing( |
29 void StopSystemTracing(const OutputCallback& callback); | 31 const base::trace_event::TraceConfig& trace_config) override; |
| 32 void StopAgentTracing(const StopAgentTracingCallback& callback) override; |
30 | 33 |
31 // Retrieve the ETW consumer instance. | 34 // Retrieve the ETW consumer instance. |
32 static EtwSystemEventConsumer* GetInstance(); | 35 static EtwSystemEventConsumer* GetInstance(); |
33 | 36 |
34 private: | 37 private: |
35 // This allows constructor and destructor to be private and usable only | 38 // This allows constructor and destructor to be private and usable only |
36 // by the Singleton class. | 39 // by the Singleton class. |
37 friend struct base::DefaultSingletonTraits<EtwSystemEventConsumer>; | 40 friend struct base::DefaultSingletonTraits<EtwSystemEventConsumer>; |
38 | 41 |
39 // Constructor. | 42 // Constructor. |
40 EtwSystemEventConsumer(); | 43 EtwSystemEventConsumer(); |
41 virtual ~EtwSystemEventConsumer(); | 44 ~EtwSystemEventConsumer() override; |
42 | 45 |
43 void AddSyncEventToBuffer(); | 46 void AddSyncEventToBuffer(); |
44 void AppendEventToBuffer(EVENT_TRACE* event); | 47 void AppendEventToBuffer(EVENT_TRACE* event); |
45 | 48 |
46 // Static override of EtwTraceConsumerBase::ProcessEvent. | 49 // Static override of EtwTraceConsumerBase::ProcessEvent. |
47 // @param event the raw ETW event to process. | 50 // @param event the raw ETW event to process. |
48 friend class base::win::EtwTraceConsumerBase<EtwSystemEventConsumer>; | 51 friend class base::win::EtwTraceConsumerBase<EtwSystemEventConsumer>; |
49 static void ProcessEvent(EVENT_TRACE* event); | 52 static void ProcessEvent(EVENT_TRACE* event); |
50 | 53 |
51 // Request the ETW trace controller to activate the kernel tracing. | 54 // Request the ETW trace controller to activate the kernel tracing. |
52 // returns true on success, false if the kernel tracing isn't activated. | 55 // returns true on success, false if the kernel tracing isn't activated. |
53 bool StartKernelSessionTracing(); | 56 bool StartKernelSessionTracing(); |
54 | 57 |
55 // Request the ETW trace controller to deactivate the kernel tracing. | 58 // Request the ETW trace controller to deactivate the kernel tracing. |
56 // @param callback the callback to call with the consumed events. | 59 // @param callback the callback to call with the consumed events. |
57 // @returns true on success, false if an error occurred. | 60 // @returns true on success, false if an error occurred. |
58 bool StopKernelSessionTracing(); | 61 bool StopKernelSessionTracing(); |
59 | 62 |
60 void OnStopSystemTracingDone( | 63 void OnStopSystemTracingDone( |
61 const OutputCallback& callback, | 64 const StopAgentTracingCallback& callback, |
62 const scoped_refptr<base::RefCountedString>& result); | 65 const scoped_refptr<base::RefCountedString>& result); |
63 | 66 |
64 void TraceAndConsumeOnThread(); | 67 void TraceAndConsumeOnThread(); |
65 void FlushOnThread(const OutputCallback& callback); | 68 void FlushOnThread(const StopAgentTracingCallback& callback); |
66 | 69 |
67 scoped_ptr<base::ListValue> events_; | 70 scoped_ptr<base::ListValue> events_; |
68 base::Thread thread_; | 71 base::Thread thread_; |
69 TRACEHANDLE session_handle_; | 72 TRACEHANDLE session_handle_; |
70 base::win::EtwTraceProperties properties_; | 73 base::win::EtwTraceProperties properties_; |
71 | 74 |
72 DISALLOW_COPY_AND_ASSIGN(EtwSystemEventConsumer); | 75 DISALLOW_COPY_AND_ASSIGN(EtwSystemEventConsumer); |
73 }; | 76 }; |
74 | 77 |
75 } // namespace content | 78 } // namespace content |
76 | 79 |
77 #endif // CONTENT_BROWSER_TRACING_ETW_SYSTEM_EVENT_CONSUMER_WIN_H_ | 80 #endif // CONTENT_BROWSER_TRACING_ETW_SYSTEM_EVENT_CONSUMER_WIN_H_ |
OLD | NEW |