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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/* -*- 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 http://mozilla.org/MPL/2.0/. */
#include "MockNetworkLayerController.h"
#include "mozilla/ClearOnShutdown.h"
#include "mozilla/net/SocketProcessParent.h"
#include "nsIOService.h"
#include "nsNetAddr.h"
namespace mozilla::net {
static StaticRefPtr<MockNetworkLayerController> gController;
bool FindNetAddrOverride(const NetAddr& aInput, NetAddr& aOutput) {
RefPtr<MockNetworkLayerController> controller = gController;
if (!controller) {
return false;
}
nsAutoCString addrPort;
aInput.ToAddrPortString(addrPort);
AutoReadLock lock(controller->mLock);
return controller->mNetAddrOverrides.Get(addrPort, &aOutput);
}
bool FindBlockedUDPAddr(const NetAddr& aInput) {
RefPtr<MockNetworkLayerController> controller = gController;
if (!controller) {
return false;
}
nsAutoCString addrPort;
aInput.ToAddrPortString(addrPort);
AutoReadLock lock(controller->mLock);
return controller->mBlockedUDPAddresses.Contains(addrPort);
}
// static
already_AddRefed<nsIMockNetworkLayerController>
MockNetworkLayerController::GetSingleton() {
if (gController) {
return do_AddRef(gController);
}
gController = new MockNetworkLayerController();
ClearOnShutdown(&gController);
return do_AddRef(gController);
}
NS_IMPL_ISUPPORTS(MockNetworkLayerController, nsIMockNetworkLayerController)
NS_IMETHODIMP MockNetworkLayerController::CreateScriptableNetAddr(
const nsACString& aIP, uint16_t aPort, nsINetAddr** aResult) {
NetAddr rawAddr;
if (NS_FAILED(rawAddr.InitFromString(aIP))) {
return NS_ERROR_FAILURE;
}
rawAddr.inet.port = PR_htons(aPort);
RefPtr<nsNetAddr> netaddr = new nsNetAddr(&rawAddr);
netaddr.forget(aResult);
return NS_OK;
}
NS_IMETHODIMP MockNetworkLayerController::AddNetAddrOverride(nsINetAddr* aFrom,
nsINetAddr* aTo) {
MOZ_ASSERT(NS_IsMainThread());
NetAddr fromAddr;
aFrom->GetNetAddr(&fromAddr);
NetAddr toAddr;
aTo->GetNetAddr(&toAddr);
nsAutoCString addrPort;
fromAddr.ToAddrPortString(addrPort);
{
AutoWriteLock lock(mLock);
mNetAddrOverrides.InsertOrUpdate(addrPort, toAddr);
}
if (nsIOService::UseSocketProcess()) {
RefPtr<SocketProcessParent> parent = SocketProcessParent::GetSingleton();
if (parent) {
(void)parent->SendAddNetAddrOverride(fromAddr, toAddr);
}
}
return NS_OK;
}
NS_IMETHODIMP MockNetworkLayerController::ClearNetAddrOverrides() {
MOZ_ASSERT(NS_IsMainThread());
{
AutoWriteLock lock(mLock);
mNetAddrOverrides.Clear();
}
if (nsIOService::UseSocketProcess()) {
RefPtr<SocketProcessParent> parent = SocketProcessParent::GetSingleton();
if (parent) {
(void)parent->SendClearNetAddrOverrides();
}
}
return NS_OK;
}
NS_IMETHODIMP MockNetworkLayerController::BlockUDPAddrIO(nsINetAddr* aAddr) {
MOZ_ASSERT(NS_IsMainThread());
NetAddr addr;
aAddr->GetNetAddr(&addr);
nsAutoCString addrPort;
addr.ToAddrPortString(addrPort);
{
AutoWriteLock lock(mLock);
mBlockedUDPAddresses.Insert(addrPort);
}
return NS_OK;
}
NS_IMETHODIMP MockNetworkLayerController::ClearBlockedUDPAddr() {
MOZ_ASSERT(NS_IsMainThread());
AutoWriteLock lock(mLock);
mBlockedUDPAddresses.Clear();
return NS_OK;
}
} // namespace mozilla::net
|
