Source code
Revision control
Copy as Markdown
Other Tools
/* -*- 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
#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);
}
// 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) {
Unused << 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) {
Unused << parent->SendClearNetAddrOverrides();
}
}
return NS_OK;
}
} // namespace mozilla::net