1 files changed, 208 insertions, 0 deletions
diff --git a/toolkit/components/telemetry/core/TelemetryCommon.cpp b/toolkit/components/telemetry/core/TelemetryCommon.cpp
new file mode 100644
index 0000000000..833f6b843b
--- /dev/null
+++ b/toolkit/components/telemetry/core/TelemetryCommon.cpp
@@ -0,0 +1,208 @@
+/* -*- 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 "TelemetryCommon.h"
+
+#include <cstring>
+#include "mozilla/TimeStamp.h"
+#include "mozilla/StaticPrefs_toolkit.h"
+#include "nsComponentManagerUtils.h"
+#include "nsIConsoleService.h"
+#include "nsITelemetry.h"
+#include "nsServiceManagerUtils.h"
+#include "nsThreadUtils.h"
+#include "nsVersionComparator.h"
+#include "TelemetryProcessData.h"
+#include "Telemetry.h"
+#include "mozilla/Uptime.h"
+
+namespace mozilla::Telemetry::Common {
+
+bool IsExpiredVersion(const char* aExpiration) {
+  MOZ_ASSERT(aExpiration);
+  // Note: We intentionally don't construct a static Version object here as we
+  // saw odd crashes around this (see bug 1334105).
+  return strcmp(aExpiration, "never") && strcmp(aExpiration, "default") &&
+         (mozilla::Version(aExpiration) <= MOZ_APP_VERSION);
+}
+
+bool IsInDataset(uint32_t aDataset, uint32_t aContainingDataset) {
+  if (aDataset == aContainingDataset) {
+    return true;
+  }
+
+  // The "optin on release channel" dataset is a superset of the
+  // "optout on release channel one".
+  if (aContainingDataset == nsITelemetry::DATASET_PRERELEASE_CHANNELS &&
+      aDataset == nsITelemetry::DATASET_ALL_CHANNELS) {
+    return true;
+  }
+
+  return false;
+}
+
+bool CanRecordDataset(uint32_t aDataset, bool aCanRecordBase,
+                      bool aCanRecordExtended) {
+  // If we are extended telemetry is enabled, we are allowed to record
+  // regardless of the dataset.
+  if (aCanRecordExtended) {
+    return true;
+  }
+
+  // If base telemetry data is enabled and we're trying to record base
+  // telemetry, allow it.
+  if (aCanRecordBase &&
+      IsInDataset(aDataset, nsITelemetry::DATASET_ALL_CHANNELS)) {
+    return true;
+  }
+
+  // We're not recording extended telemetry or this is not the base
+  // dataset. Bail out.
+  return false;
+}
+
+bool CanRecordInProcess(RecordedProcessType processes,
+                        GeckoProcessType processType) {
+  // We can use (1 << ProcessType) due to the way RecordedProcessType is
+  // defined.
+  bool canRecordProcess =
+      !!(processes & static_cast<RecordedProcessType>(1 << processType));
+
+  return canRecordProcess;
+}
+
+bool CanRecordInProcess(RecordedProcessType processes, ProcessID processId) {
+  return CanRecordInProcess(processes, GetGeckoProcessType(processId));
+}
+
+bool CanRecordProduct(SupportedProduct aProducts) {
+  return mozilla::StaticPrefs::
+             toolkit_telemetry_testing_overrideProductsCheck() ||
+         !!(aProducts & GetCurrentProduct());
+}
+
+nsresult MsSinceProcessStart(double* aResult) {
+  *aResult =
+      (TimeStamp::NowLoRes() - TimeStamp::ProcessCreation()).ToMilliseconds();
+  return NS_OK;
+}
+
+nsresult MsSinceProcessStartIncludingSuspend(double* aResult) {
+  auto rv = mozilla::ProcessUptimeMs();
+  if (rv) {
+    *aResult = rv.value();
+    return NS_OK;
+  }
+  return NS_ERROR_NOT_AVAILABLE;
+}
+
+nsresult MsSinceProcessStartExcludingSuspend(double* aResult) {
+  auto rv = mozilla::ProcessUptimeExcludingSuspendMs();
+  if (rv) {
+    *aResult = rv.value();
+    return NS_OK;
+  }
+  return NS_ERROR_NOT_AVAILABLE;
+}
+
+void LogToBrowserConsole(uint32_t aLogLevel, const nsAString& aMsg) {
+  if (!NS_IsMainThread()) {
+    nsString msg(aMsg);
+    nsCOMPtr<nsIRunnable> task = NS_NewRunnableFunction(
+        "Telemetry::Common::LogToBrowserConsole",
+        [aLogLevel, msg]() { LogToBrowserConsole(aLogLevel, msg); });
+    NS_DispatchToMainThread(task.forget(), NS_DISPATCH_NORMAL);
+    return;
+  }
+
+  nsCOMPtr<nsIConsoleService> console(
+      do_GetService("@mozilla.org/consoleservice;1"));
+  if (!console) {
+    NS_WARNING("Failed to log message to console.");
+    return;
+  }
+
+  nsCOMPtr<nsIScriptError> error(do_CreateInstance(NS_SCRIPTERROR_CONTRACTID));
+  error->Init(aMsg, u""_ns, u""_ns, 0, 0, aLogLevel, "chrome javascript",
+              false /* from private window */, true /* from chrome context */);
+  console->LogMessage(error);
+}
+
+const char* GetNameForProcessID(ProcessID process) {
+  MOZ_ASSERT(process < ProcessID::Count);
+  return ProcessIDToString[static_cast<uint32_t>(process)];
+}
+
+ProcessID GetIDForProcessName(const char* aProcessName) {
+  for (uint32_t id = 0; id < static_cast<uint32_t>(ProcessID::Count); id++) {
+    if (!strcmp(GetNameForProcessID(ProcessID(id)), aProcessName)) {
+      return ProcessID(id);
+    }
+  }
+
+  return ProcessID::Count;
+}
+
+GeckoProcessType GetGeckoProcessType(ProcessID process) {
+  MOZ_ASSERT(process < ProcessID::Count);
+  return ProcessIDToGeckoProcessType[static_cast<uint32_t>(process)];
+}
+
+bool IsStringCharValid(const char aChar, const bool aAllowInfixPeriod,
+                       const bool aAllowInfixUnderscore) {
+  return (aChar >= 'A' && aChar <= 'Z') || (aChar >= 'a' && aChar <= 'z') ||
+         (aChar >= '0' && aChar <= '9') ||
+         (aAllowInfixPeriod && (aChar == '.')) ||
+         (aAllowInfixUnderscore && (aChar == '_'));
+}
+
+bool IsValidIdentifierString(const nsACString& aStr, const size_t aMaxLength,
+                             const bool aAllowInfixPeriod,
+                             const bool aAllowInfixUnderscore) {
+  // Check string length.
+  if (aStr.Length() > aMaxLength) {
+    return false;
+  }
+
+  // Check string characters.
+  const char* first = aStr.BeginReading();
+  const char* end = aStr.EndReading();
+
+  for (const char* cur = first; cur < end; ++cur) {
+    const bool infix = (cur != first) && (cur != (end - 1));
+    if (!IsStringCharValid(*cur, aAllowInfixPeriod && infix,
+                           aAllowInfixUnderscore && infix)) {
+      return false;
+    }
+  }
+
+  return true;
+}
+
+JSString* ToJSString(JSContext* cx, const nsACString& aStr) {
+  const NS_ConvertUTF8toUTF16 wide(aStr);
+  return JS_NewUCStringCopyN(cx, wide.Data(), wide.Length());
+}
+
+JSString* ToJSString(JSContext* cx, const nsAString& aStr) {
+  return JS_NewUCStringCopyN(cx, aStr.Data(), aStr.Length());
+}
+
+SupportedProduct GetCurrentProduct() {
+#if defined(MOZ_WIDGET_ANDROID)
+  if (mozilla::StaticPrefs::toolkit_telemetry_geckoview_streaming()) {
+    return SupportedProduct::GeckoviewStreaming;
+  } else {
+    return SupportedProduct::Fennec;
+  }
+#elif defined(MOZ_THUNDERBIRD)
+  return SupportedProduct::Thunderbird;
+#else
+  return SupportedProduct::Firefox;
+#endif
+}
+
+}  // namespace mozilla::Telemetry::Common