57 lines
2.4 KiB
C++
57 lines
2.4 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "ETWTools.h"
|
|
|
|
#include <atomic>
|
|
|
|
namespace ETW {
|
|
std::atomic<ULONGLONG> gETWCollectionMask = 0;
|
|
|
|
// Define a handle to a TraceLogging provider.
|
|
//
|
|
// > All ETW providers are identified by both provider name and provider ID.
|
|
// > [...]
|
|
// > Microsoft recommends generating the provider ID from the provider name
|
|
// > using the ETW name-hashing algorithm described below. This provides
|
|
// > several benefits: it's easier to remember just the name; the ID and the
|
|
// > name are automatically linked; tools such as tracelog, traceview,
|
|
// > EventSource, and WPR have special support for providers that use IDs
|
|
// > generated using this algorithm.
|
|
//
|
|
// https://learn.microsoft.com/en-us/windows/win32/api/traceloggingprovider/nf-traceloggingprovider-tracelogging_define_provider
|
|
//
|
|
// The GUID generated for "Mozilla.IceCatTraceLogger" is:
|
|
// {c923f508-96e4-5515-e32c-7539d1b10504}
|
|
TRACELOGGING_DEFINE_PROVIDER(kIceCatTraceLoggingProvider,
|
|
"Mozilla.IceCatTraceLogger",
|
|
(0xc923f508, 0x96e4, 0x5515, 0xe3, 0x2c, 0x75,
|
|
0x39, 0xd1, 0xb1, 0x05, 0x04));
|
|
|
|
static void NTAPI ETWEnableCallback(LPCGUID aSourceId, ULONG aIsEnabled,
|
|
UCHAR aLevel, ULONGLONG aMatchAnyKeyword,
|
|
ULONGLONG aMatchAllKeyword,
|
|
PEVENT_FILTER_DESCRIPTOR aFilterData,
|
|
PVOID aCallbackContext) {
|
|
// This is called on a CRT worker thread. This means this might race a bit
|
|
// with our main thread, but that is okay.
|
|
if (aIsEnabled) {
|
|
mozilla::profiler::detail::RacyFeatures::SetETWCollectionActive();
|
|
} else {
|
|
mozilla::profiler::detail::RacyFeatures::SetETWCollectionInactive();
|
|
}
|
|
// The lower 48 bits of the provider flags are used to mask markers.
|
|
gETWCollectionMask = aMatchAnyKeyword;
|
|
}
|
|
|
|
void Init() {
|
|
TraceLoggingRegisterEx(kIceCatTraceLoggingProvider, ETWEnableCallback,
|
|
nullptr);
|
|
}
|
|
|
|
void Shutdown() { TraceLoggingUnregister(kIceCatTraceLoggingProvider); }
|
|
|
|
} // namespace ETW
|