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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
/* -*- 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
|
