/* -*- 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.FirefoxTraceLogger" is:
// {c923f508-96e4-5515-e32c-7539d1b10504}
TRACELOGGING_DEFINE_PROVIDER(kFirefoxTraceLoggingProvider,
"Mozilla.FirefoxTraceLogger",
(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(kFirefoxTraceLoggingProvider, ETWEnableCallback,
nullptr);
}
void Shutdown() { TraceLoggingUnregister(kFirefoxTraceLoggingProvider); }
}
// namespace ETW
