UntrustedModulesBackupService.h

Enable keyboard shortcuts

/* -*- 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 https://mozilla.org/MPL/2.0/. */

#ifndef mozilla_UntrustedModulesBackupService_h

#define mozilla_UntrustedModulesBackupService_h

#include "mozilla/UntrustedModulesData.h"

#include "nsRefPtrHashtable.h"

namespace mozilla {

struct ProcessHashKey {

  GeckoProcessType mType;

  DWORD mPid;

  ProcessHashKey(GeckoProcessType aType, DWORD aPid);

  bool operator==(const ProcessHashKey& aOther) const;

  PLDHashNumber Hash() const;

};

// UntrustedModulesData should not be refcounted as it's exchanged via IPC.

// Instead, we define this container class owning UntrustedModulesData along

// with a refcount.

class MOZ_HEAP_CLASS UntrustedModulesDataContainer final {

  ~UntrustedModulesDataContainer() = default;

 public:

  UntrustedModulesData mData;

  explicit UntrustedModulesDataContainer(UntrustedModulesData&& aData)

      : mData(std::move(aData)) {}

  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(UntrustedModulesDataContainer)

};

class UntrustedModulesBackupData

    : public nsRefPtrHashtable<nsGenericHashKey<ProcessHashKey>,

                               UntrustedModulesDataContainer> {

 public:

  void Add(UntrustedModulesData&& aData);

  void AddWithoutStacks(UntrustedModulesData&& aData);

};

class MOZ_HEAP_CLASS UntrustedModulesBackupService final {

 public:

  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(UntrustedModulesBackupService)

  static UntrustedModulesBackupService* Get();

  // Back up data to mStaging

  void Backup(UntrustedModulesData&& aData);

  void SettleAllStagingData();

  const UntrustedModulesBackupData& Staging() const { return mStaging; }

  const UntrustedModulesBackupData& Settled() const { return mSettled; }

 private:

  // Data not yet submitted as telemetry

  UntrustedModulesBackupData mStaging;

  // Data already submitted as telemetry

  // (This does not have stack information)

  UntrustedModulesBackupData mSettled;

  ~UntrustedModulesBackupService() = default;

};

}  // namespace mozilla

#endif  // mozilla_UntrustedModulesBackupService_h