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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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 http://mozilla.org/MPL/2.0/. */
#ifndef mozilla_windowsdllblocklist_h
#define mozilla_windowsdllblocklist_h
#if (defined(_MSC_VER) || defined(__MINGW32__)) && \
(defined(_M_IX86) || defined(_M_X64) || defined(_M_ARM64))
# include <windows.h>
# include "CrashAnnotations.h"
# include "mozilla/Attributes.h"
# include "mozilla/ProcessType.h"
# include "mozilla/Types.h"
# define HAS_DLL_BLOCKLIST
enum DllBlocklistInitFlags {
eDllBlocklistInitFlagDefault = 0,
eDllBlocklistInitFlagIsChildProcess = 1 << 0,
eDllBlocklistInitFlagWasBootstrapped = 1 << 1,
eDllBlocklistInitFlagIsUtilityProcess = 1 << 2,
eDllBlocklistInitFlagIsSocketProcess = 1 << 3,
eDllBlocklistInitFlagIsGPUProcess = 1 << 4,
eDllBlocklistInitFlagIsGMPluginProcess = 1 << 5,
};
inline void SetDllBlocklistProcessTypeFlags(uint32_t& aFlags,
GeckoProcessType aProcessType) {
if (aProcessType == GeckoProcessType_Utility) {
aFlags |= eDllBlocklistInitFlagIsUtilityProcess;
} else if (aProcessType == GeckoProcessType_Socket) {
aFlags |= eDllBlocklistInitFlagIsSocketProcess;
} else if (aProcessType == GeckoProcessType_GPU) {
aFlags |= eDllBlocklistInitFlagIsGPUProcess;
} else if (aProcessType == GeckoProcessType_GMPlugin) {
aFlags |= eDllBlocklistInitFlagIsGMPluginProcess;
}
}
// Only available from within firefox.exe
# if !defined(IMPL_MFBT) && !defined(MOZILLA_INTERNAL_API)
extern uint32_t gBlocklistInitFlags;
# endif // !defined(IMPL_MFBT) && !defined(MOZILLA_INTERNAL_API)
MFBT_API void DllBlocklist_Initialize(
uint32_t aInitFlags = eDllBlocklistInitFlagDefault);
MFBT_API void DllBlocklist_WriteNotes(CrashReporter::AnnotationWriter& aWriter);
MFBT_API bool DllBlocklist_CheckStatus();
// This export intends to clean up after DllBlocklist_Initialize().
// It's disabled in release builds for performance and to limit callers' ability
// to interfere with dll blocking.
# ifdef DEBUG
MFBT_API void DllBlocklist_Shutdown();
# endif // DEBUG
namespace mozilla {
namespace glue {
namespace detail {
// Forward declaration
class DllServicesBase;
template <size_t N>
class WritableBuffer {
char mBuffer[N];
size_t mLen;
size_t Available() const { return sizeof(mBuffer) - mLen; }
public:
WritableBuffer() : mBuffer{0}, mLen(0) {}
void Write(const char* aData, size_t aLen) {
size_t writable_len = std::min(aLen, Available());
memcpy(mBuffer + mLen, aData, writable_len);
mLen += writable_len;
}
size_t Length() const { return mLen; }
const char* Data() const { return mBuffer; }
};
} // namespace detail
} // namespace glue
} // namespace mozilla
MFBT_API void DllBlocklist_SetFullDllServices(
mozilla::glue::detail::DllServicesBase* aSvc);
MFBT_API void DllBlocklist_SetBasicDllServices(
mozilla::glue::detail::DllServicesBase* aSvc);
#endif // defined(_MSC_VER) && (defined(_M_IX86) || defined(_M_X64))
#endif // mozilla_windowsdllblocklist_h
|
