| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 // | |
| 5 // Implementation of a Windows event trace controller class. | |
| 6 #include "base/win/event_trace_controller.h" | |
| 7 #include "base/logging.h" | |
| 8 | |
| 9 namespace base { | |
| 10 namespace win { | |
| 11 | |
| 12 EtwTraceProperties::EtwTraceProperties() { | |
| 13 memset(buffer_, 0, sizeof(buffer_)); | |
| 14 EVENT_TRACE_PROPERTIES* prop = get(); | |
| 15 | |
| 16 prop->Wnode.BufferSize = sizeof(buffer_); | |
| 17 prop->Wnode.Flags = WNODE_FLAG_TRACED_GUID; | |
| 18 prop->LoggerNameOffset = sizeof(EVENT_TRACE_PROPERTIES); | |
| 19 prop->LogFileNameOffset = sizeof(EVENT_TRACE_PROPERTIES) + | |
| 20 sizeof(wchar_t) * kMaxStringLen; | |
| 21 } | |
| 22 | |
| 23 HRESULT EtwTraceProperties::SetLoggerName(const wchar_t* logger_name) { | |
| 24 size_t len = wcslen(logger_name) + 1; | |
| 25 if (kMaxStringLen < len) | |
| 26 return E_INVALIDARG; | |
| 27 | |
| 28 memcpy(buffer_ + get()->LoggerNameOffset, | |
| 29 logger_name, | |
| 30 sizeof(wchar_t) * len); | |
| 31 return S_OK; | |
| 32 } | |
| 33 | |
| 34 HRESULT EtwTraceProperties::SetLoggerFileName(const wchar_t* logger_file_name) { | |
| 35 size_t len = wcslen(logger_file_name) + 1; | |
| 36 if (kMaxStringLen < len) | |
| 37 return E_INVALIDARG; | |
| 38 | |
| 39 memcpy(buffer_ + get()->LogFileNameOffset, | |
| 40 logger_file_name, | |
| 41 sizeof(wchar_t) * len); | |
| 42 return S_OK; | |
| 43 } | |
| 44 | |
| 45 EtwTraceController::EtwTraceController() : session_(NULL) { | |
| 46 } | |
| 47 | |
| 48 EtwTraceController::~EtwTraceController() { | |
| 49 Stop(NULL); | |
| 50 } | |
| 51 | |
| 52 HRESULT EtwTraceController::Start(const wchar_t* session_name, | |
| 53 EtwTraceProperties* prop) { | |
| 54 DCHECK(NULL == session_ && session_name_.empty()); | |
| 55 EtwTraceProperties ignore; | |
| 56 if (prop == NULL) | |
| 57 prop = &ignore; | |
| 58 | |
| 59 HRESULT hr = Start(session_name, prop, &session_); | |
| 60 if (SUCCEEDED(hr)) | |
| 61 session_name_ = session_name; | |
| 62 | |
| 63 return hr; | |
| 64 } | |
| 65 | |
| 66 HRESULT EtwTraceController::StartFileSession(const wchar_t* session_name, | |
| 67 const wchar_t* logfile_path, bool realtime) { | |
| 68 DCHECK(NULL == session_ && session_name_.empty()); | |
| 69 | |
| 70 EtwTraceProperties prop; | |
| 71 prop.SetLoggerFileName(logfile_path); | |
| 72 EVENT_TRACE_PROPERTIES& p = *prop.get(); | |
| 73 p.Wnode.ClientContext = 1; // QPC timer accuracy. | |
| 74 p.LogFileMode = EVENT_TRACE_FILE_MODE_SEQUENTIAL; // Sequential log. | |
| 75 if (realtime) | |
| 76 p.LogFileMode |= EVENT_TRACE_REAL_TIME_MODE; | |
| 77 | |
| 78 p.MaximumFileSize = 100; // 100M file size. | |
| 79 p.FlushTimer = 30; // 30 seconds flush lag. | |
| 80 return Start(session_name, &prop); | |
| 81 } | |
| 82 | |
| 83 HRESULT EtwTraceController::StartRealtimeSession(const wchar_t* session_name, | |
| 84 size_t buffer_size) { | |
| 85 DCHECK(NULL == session_ && session_name_.empty()); | |
| 86 EtwTraceProperties prop; | |
| 87 EVENT_TRACE_PROPERTIES& p = *prop.get(); | |
| 88 p.LogFileMode = EVENT_TRACE_REAL_TIME_MODE | EVENT_TRACE_USE_PAGED_MEMORY; | |
| 89 p.FlushTimer = 1; // flush every second. | |
| 90 p.BufferSize = 16; // 16 K buffers. | |
| 91 p.LogFileNameOffset = 0; | |
| 92 return Start(session_name, &prop); | |
| 93 } | |
| 94 | |
| 95 HRESULT EtwTraceController::EnableProvider(REFGUID provider, UCHAR level, | |
| 96 ULONG flags) { | |
| 97 ULONG error = ::EnableTrace(TRUE, flags, level, &provider, session_); | |
| 98 return HRESULT_FROM_WIN32(error); | |
| 99 } | |
| 100 | |
| 101 HRESULT EtwTraceController::DisableProvider(REFGUID provider) { | |
| 102 ULONG error = ::EnableTrace(FALSE, 0, 0, &provider, session_); | |
| 103 return HRESULT_FROM_WIN32(error); | |
| 104 } | |
| 105 | |
| 106 HRESULT EtwTraceController::Stop(EtwTraceProperties* properties) { | |
| 107 EtwTraceProperties ignore; | |
| 108 if (properties == NULL) | |
| 109 properties = &ignore; | |
| 110 | |
| 111 ULONG error = ::ControlTrace(session_, NULL, properties->get(), | |
| 112 EVENT_TRACE_CONTROL_STOP); | |
| 113 if (ERROR_SUCCESS != error) | |
| 114 return HRESULT_FROM_WIN32(error); | |
| 115 | |
| 116 session_ = NULL; | |
| 117 session_name_.clear(); | |
| 118 return S_OK; | |
| 119 } | |
| 120 | |
| 121 HRESULT EtwTraceController::Flush(EtwTraceProperties* properties) { | |
| 122 EtwTraceProperties ignore; | |
| 123 if (properties == NULL) | |
| 124 properties = &ignore; | |
| 125 | |
| 126 ULONG error = ::ControlTrace(session_, NULL, properties->get(), | |
| 127 EVENT_TRACE_CONTROL_FLUSH); | |
| 128 if (ERROR_SUCCESS != error) | |
| 129 return HRESULT_FROM_WIN32(error); | |
| 130 | |
| 131 return S_OK; | |
| 132 } | |
| 133 | |
| 134 HRESULT EtwTraceController::Start(const wchar_t* session_name, | |
| 135 EtwTraceProperties* properties, TRACEHANDLE* session_handle) { | |
| 136 DCHECK(properties != NULL); | |
| 137 ULONG err = ::StartTrace(session_handle, session_name, properties->get()); | |
| 138 return HRESULT_FROM_WIN32(err); | |
| 139 } | |
| 140 | |
| 141 HRESULT EtwTraceController::Query(const wchar_t* session_name, | |
| 142 EtwTraceProperties* properties) { | |
| 143 ULONG err = ::ControlTrace(NULL, session_name, properties->get(), | |
| 144 EVENT_TRACE_CONTROL_QUERY); | |
| 145 return HRESULT_FROM_WIN32(err); | |
| 146 }; | |
| 147 | |
| 148 HRESULT EtwTraceController::Update(const wchar_t* session_name, | |
| 149 EtwTraceProperties* properties) { | |
| 150 DCHECK(properties != NULL); | |
| 151 ULONG err = ::ControlTrace(NULL, session_name, properties->get(), | |
| 152 EVENT_TRACE_CONTROL_UPDATE); | |
| 153 return HRESULT_FROM_WIN32(err); | |
| 154 } | |
| 155 | |
| 156 HRESULT EtwTraceController::Stop(const wchar_t* session_name, | |
| 157 EtwTraceProperties* properties) { | |
| 158 DCHECK(properties != NULL); | |
| 159 ULONG err = ::ControlTrace(NULL, session_name, properties->get(), | |
| 160 EVENT_TRACE_CONTROL_STOP); | |
| 161 return HRESULT_FROM_WIN32(err); | |
| 162 } | |
| 163 | |
| 164 HRESULT EtwTraceController::Flush(const wchar_t* session_name, | |
| 165 EtwTraceProperties* properties) { | |
| 166 DCHECK(properties != NULL); | |
| 167 ULONG err = ::ControlTrace(NULL, session_name, properties->get(), | |
| 168 EVENT_TRACE_CONTROL_FLUSH); | |
| 169 return HRESULT_FROM_WIN32(err); | |
| 170 } | |
| 171 | |
| 172 } // namespace win | |
| 173 } // namespace base | |
| OLD | NEW |