summaryrefslogtreecommitdiffstats
path: root/tools/profiler/core/ETWTools.cpp
blob: b8fab9b316741bde0878485533fa73f272e51882 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
/* -*- 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
TRACELOGGING_DEFINE_PROVIDER(kFirefoxTraceLoggingProvider,
                             "Mozilla.FirefoxTraceLogger",
                             // This GUID is a hash generated based on the
                             // above string.
                             // {c923f508-96e4-5515-e32c-7539d1b10504}
                             (0xc923f508, 0x96e4, 0x5515, 0xe3, 0x3c, 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