diff options
author | siggi@chromium.org <siggi@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2010-11-06 15:39:40 +0000 |
---|---|---|
committer | siggi@chromium.org <siggi@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2010-11-06 15:39:40 +0000 |
commit | ba50d192694e3bbdd99d153205e7fcef236abf8a (patch) | |
tree | d4621c1cbad3e63cd87c40689c0ae5daaafab1fe /base/win/event_trace_provider.cc | |
parent | f2505c82c17948056214d3437d53620610fba837 (diff) | |
download | chromium_src-ba50d192694e3bbdd99d153205e7fcef236abf8a.zip chromium_src-ba50d192694e3bbdd99d153205e7fcef236abf8a.tar.gz chromium_src-ba50d192694e3bbdd99d153205e7fcef236abf8a.tar.bz2 |
Move Windows-specific Event Tracing for Windows implementation to base/win.
fix users of same.
BUG=none
TEST=unittests
Review URL: http://codereview.chromium.org/4517004
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@65310 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'base/win/event_trace_provider.cc')
-rw-r--r-- | base/win/event_trace_provider.cc | 134 |
1 files changed, 134 insertions, 0 deletions
diff --git a/base/win/event_trace_provider.cc b/base/win/event_trace_provider.cc new file mode 100644 index 0000000..332fac3 --- /dev/null +++ b/base/win/event_trace_provider.cc @@ -0,0 +1,134 @@ +// Copyright (c) 2009 The Chromium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. +// +#include "base/win/event_trace_provider.h" +#include <windows.h> +#include <cguid.h> + +namespace base { +namespace win { + +TRACE_GUID_REGISTRATION EtwTraceProvider::obligatory_guid_registration_ = { + &GUID_NULL, + NULL +}; + +EtwTraceProvider::EtwTraceProvider(const GUID& provider_name) + : provider_name_(provider_name), registration_handle_(NULL), + session_handle_(NULL), enable_flags_(0), enable_level_(0) { +} + +EtwTraceProvider::EtwTraceProvider() + : provider_name_(GUID_NULL), registration_handle_(NULL), + session_handle_(NULL), enable_flags_(0), enable_level_(0) { +} + +EtwTraceProvider::~EtwTraceProvider() { + Unregister(); +} + +ULONG EtwTraceProvider::EnableEvents(void* buffer) { + session_handle_ = ::GetTraceLoggerHandle(buffer); + if (NULL == session_handle_) { + return ::GetLastError(); + } + + enable_flags_ = ::GetTraceEnableFlags(session_handle_); + enable_level_ = ::GetTraceEnableLevel(session_handle_); + + // Give subclasses a chance to digest the state change. + OnEventsEnabled(); + + return ERROR_SUCCESS; +} + +ULONG EtwTraceProvider::DisableEvents() { + // Give subclasses a chance to digest the state change. + OnEventsDisabled(); + + enable_level_ = 0; + enable_flags_ = 0; + session_handle_ = NULL; + + PostEventsDisabled(); + + return ERROR_SUCCESS; +} + +ULONG EtwTraceProvider::Callback(WMIDPREQUESTCODE request, void* buffer) { + switch (request) { + case WMI_ENABLE_EVENTS: + return EnableEvents(buffer); + case WMI_DISABLE_EVENTS: + return DisableEvents(); + default: + return ERROR_INVALID_PARAMETER; + } + // Not reached. +} + +ULONG WINAPI EtwTraceProvider::ControlCallback(WMIDPREQUESTCODE request, + void* context, ULONG *reserved, void* buffer) { + EtwTraceProvider *provider = reinterpret_cast<EtwTraceProvider*>(context); + + return provider->Callback(request, buffer); +} + +ULONG EtwTraceProvider::Register() { + if (provider_name_ == GUID_NULL) + return ERROR_INVALID_NAME; + + return ::RegisterTraceGuids(ControlCallback, this, &provider_name_, + 1, &obligatory_guid_registration_, NULL, NULL, ®istration_handle_); +} + +ULONG EtwTraceProvider::Unregister() { + ULONG ret = ::UnregisterTraceGuids(registration_handle_); + + // Make sure we don't log anything from here on. + enable_level_ = 0; + enable_flags_ = 0; + session_handle_ = NULL; + registration_handle_ = NULL; + + return ret; +} + +ULONG EtwTraceProvider::Log(const EtwEventClass& event_class, + EtwEventType type, EtwEventLevel level, const char *message) { + if (NULL == session_handle_ || enable_level_ < level) + return ERROR_SUCCESS; // No one listening. + + EtwMofEvent<1> event(event_class, type, level); + + event.fields[0].DataPtr = reinterpret_cast<ULONG64>(message); + event.fields[0].Length = message ? + static_cast<ULONG>(sizeof(message[0]) * (1 + strlen(message))) : 0; + + return ::TraceEvent(session_handle_, &event.header); +} + +ULONG EtwTraceProvider::Log(const EtwEventClass& event_class, + EtwEventType type, EtwEventLevel level, const wchar_t *message) { + if (NULL == session_handle_ || enable_level_ < level) + return ERROR_SUCCESS; // No one listening. + + EtwMofEvent<1> event(event_class, type, level); + + event.fields[0].DataPtr = reinterpret_cast<ULONG64>(message); + event.fields[0].Length = message ? + static_cast<ULONG>(sizeof(message[0]) * (1 + wcslen(message))) : 0; + + return ::TraceEvent(session_handle_, &event.header); +} + +ULONG EtwTraceProvider::Log(EVENT_TRACE_HEADER* event) { + if (enable_level_ < event->Class.Level) + return ERROR_SUCCESS; + + return ::TraceEvent(session_handle_, event); +} + +} // namespace win +} // namespace base |