toolkit/components/telemetry/other/UntrustedModulesBackupService.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef mozilla_UntrustedModulesBackupService_h
   8 #define mozilla_UntrustedModulesBackupService_h
   9
  10 #include "mozilla/UntrustedModulesData.h"
  11 #include "nsRefPtrHashtable.h"
  12
  13 namespace mozilla {
  14
  15 struct ProcessHashKey {
  16   GeckoProcessType mType;
  17   DWORD mPid;
  18   ProcessHashKey(GeckoProcessType aType, DWORD aPid);
  19   bool operator==(const ProcessHashKey& aOther) const;
  20   PLDHashNumber Hash() const;
  21 };
  22
  23 // UntrustedModulesData should not be refcounted as it's exchanged via IPC.
  24 // Instead, we define this container class owning UntrustedModulesData along
  25 // with a refcount.
  26 class MOZ_HEAP_CLASS UntrustedModulesDataContainer final {
  27   ~UntrustedModulesDataContainer() = default;
  28
  29  public:
  30   UntrustedModulesData mData;
  31
  32   explicit UntrustedModulesDataContainer(UntrustedModulesData&& aData)
  33       : mData(std::move(aData)) {}
  34
  35   NS_INLINE_DECL_THREADSAFE_REFCOUNTING(UntrustedModulesDataContainer)
  36 };
  37
  38 class UntrustedModulesBackupData
  39     : public nsRefPtrHashtable<nsGenericHashKey<ProcessHashKey>,
  40                                UntrustedModulesDataContainer> {
  41  public:
  42   void Add(UntrustedModulesData&& aData);
  43   void AddWithoutStacks(UntrustedModulesData&& aData);
  44 };
  45
  46 class MOZ_HEAP_CLASS UntrustedModulesBackupService final {
  47  public:
  48   NS_INLINE_DECL_THREADSAFE_REFCOUNTING(UntrustedModulesBackupService)
  49
  50   static UntrustedModulesBackupService* Get();
  51
  52   // Back up data to mStaging
  53   void Backup(UntrustedModulesData&& aData);
  54
  55   void SettleAllStagingData();
  56
  57   const UntrustedModulesBackupData& Staging() const { return mStaging; }
  58   const UntrustedModulesBackupData& Settled() const { return mSettled; }
  59
  60  private:
  61   // Data not yet submitted as telemetry
  62   UntrustedModulesBackupData mStaging;
  63
  64   // Data already submitted as telemetry
  65   // (This does not have stack information)
  66   UntrustedModulesBackupData mSettled;
  67
  68   ~UntrustedModulesBackupService() = default;
  69 };
  70
  71 }  // namespace mozilla
  72
  73 #endif  // mozilla_UntrustedModulesBackupService_h