blob: 15cfad216d291be86929ac05a3e99f5ddf8f9231 (
plain)
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
|
/* -*- 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);
};
class MOZ_HEAP_CLASS UntrustedModulesBackupService final {
public:
enum class BackupType : uint32_t {
Staging = 0,
Settled,
Count
};
NS_INLINE_DECL_THREADSAFE_REFCOUNTING(UntrustedModulesBackupService)
static UntrustedModulesBackupService* Get();
void Backup(BackupType aType, UntrustedModulesData&& aData);
void SettleAllStagingData();
const UntrustedModulesBackupData& Ref(BackupType aType) const;
private:
UntrustedModulesBackupData mBackup[static_cast<uint32_t>(BackupType::Count)];
~UntrustedModulesBackupService() = default;
};
} // namespace mozilla
#endif // mozilla_UntrustedModulesBackupService_h
|