fune/toolkit/components/telemetry/other/UntrustedModulesBackupService.h
Toshihito Kikuchi 59fddfbcff Bug 1750791 - Part 4: Reduce backup data of module loading events. r=mhowell
We use third-party loading events data for two purposes.  One is to
send it as the third-party-modules ping.  The other is to display it
on about:third-party.  Currently we keep all loading events in memory,
but it may consume too much memory if modules injection happens too
often intentionally or unintentionally.

This patch is to discard old loading events after telemetry submission,
and also drops stack information after telemetry submission because
about:third-party does not use stack information.

Differential Revision: https://phabricator.services.mozilla.com/D137247
2022-02-24 15:00:28 +00:00

73 lines
2.2 KiB
C++

/* -*- 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