mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-06 21:05:37 +00:00
c5fb688d6a
Differential Revision: https://phabricator.services.mozilla.com/D36947 --HG-- extra : moz-landing-system : lando
345 lines
10 KiB
C++
345 lines
10 KiB
C++
/* -*- 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 "TCPSocketParent.h"
|
|
#include "jsapi.h"
|
|
#include "jsfriendapi.h"
|
|
#include "nsJSUtils.h"
|
|
#include "mozilla/Unused.h"
|
|
#include "mozilla/net/NeckoCommon.h"
|
|
#include "mozilla/net/PNeckoParent.h"
|
|
#include "mozilla/dom/ScriptSettings.h"
|
|
#include "mozilla/dom/BrowserParent.h"
|
|
#include "mozilla/HoldDropJSObjects.h"
|
|
#include "nsISocketTransportService.h"
|
|
#include "nsISocketTransport.h"
|
|
#include "nsIScriptSecurityManager.h"
|
|
#include "nsNetUtil.h"
|
|
|
|
namespace IPC {
|
|
|
|
// Defined in TCPSocketChild.cpp
|
|
extern bool DeserializeArrayBuffer(JSContext* aCx,
|
|
const nsTArray<uint8_t>& aBuffer,
|
|
JS::MutableHandle<JS::Value> aVal);
|
|
|
|
} // namespace IPC
|
|
|
|
namespace mozilla {
|
|
|
|
namespace net {
|
|
//
|
|
// set MOZ_LOG=TCPSocket:5
|
|
//
|
|
extern LazyLogModule gTCPSocketLog;
|
|
#define TCPSOCKET_LOG(args) MOZ_LOG(gTCPSocketLog, LogLevel::Debug, args)
|
|
#define TCPSOCKET_LOG_ENABLED() MOZ_LOG_TEST(gTCPSocketLog, LogLevel::Debug)
|
|
} // namespace net
|
|
|
|
using namespace net;
|
|
|
|
namespace dom {
|
|
|
|
static void FireInteralError(TCPSocketParent* aActor, uint32_t aLineNo) {
|
|
MOZ_ASSERT(aActor->IPCOpen());
|
|
|
|
mozilla::Unused << aActor->SendCallback(
|
|
NS_LITERAL_STRING("onerror"),
|
|
TCPError(NS_LITERAL_STRING("InvalidStateError"),
|
|
NS_LITERAL_STRING("Internal error")),
|
|
static_cast<uint32_t>(TCPReadyState::Connecting));
|
|
}
|
|
|
|
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(TCPSocketParentBase)
|
|
NS_INTERFACE_MAP_ENTRY(nsISupports)
|
|
NS_INTERFACE_MAP_END
|
|
|
|
NS_IMPL_CYCLE_COLLECTION(TCPSocketParentBase, mSocket)
|
|
NS_IMPL_CYCLE_COLLECTING_ADDREF(TCPSocketParentBase)
|
|
NS_IMPL_CYCLE_COLLECTING_RELEASE(TCPSocketParentBase)
|
|
|
|
TCPSocketParentBase::TCPSocketParentBase() : mIPCOpen(false) {}
|
|
|
|
TCPSocketParentBase::~TCPSocketParentBase() {}
|
|
|
|
void TCPSocketParentBase::ReleaseIPDLReference() {
|
|
MOZ_ASSERT(mIPCOpen);
|
|
mIPCOpen = false;
|
|
this->Release();
|
|
}
|
|
|
|
void TCPSocketParentBase::AddIPDLReference() {
|
|
MOZ_ASSERT(!mIPCOpen);
|
|
mIPCOpen = true;
|
|
this->AddRef();
|
|
}
|
|
|
|
NS_IMETHODIMP_(MozExternalRefCountType) TCPSocketParent::Release(void) {
|
|
nsrefcnt refcnt = TCPSocketParentBase::Release();
|
|
if (refcnt == 1 && mIPCOpen) {
|
|
mozilla::Unused << PTCPSocketParent::SendRequestDelete();
|
|
return 1;
|
|
}
|
|
return refcnt;
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvOpen(
|
|
const nsString& aHost, const uint16_t& aPort, const bool& aUseSSL,
|
|
const bool& aUseArrayBuffers) {
|
|
mSocket = new TCPSocket(nullptr, aHost, aPort, aUseSSL, aUseArrayBuffers);
|
|
mSocket->SetSocketBridgeParent(this);
|
|
NS_ENSURE_SUCCESS(mSocket->Init(), IPC_OK());
|
|
return IPC_OK();
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvOpenBind(
|
|
const nsCString& aRemoteHost, const uint16_t& aRemotePort,
|
|
const nsCString& aLocalAddr, const uint16_t& aLocalPort,
|
|
const bool& aUseSSL, const bool& aReuseAddrPort,
|
|
const bool& aUseArrayBuffers, const nsCString& aFilter) {
|
|
nsresult rv;
|
|
nsCOMPtr<nsISocketTransportService> sts =
|
|
do_GetService("@mozilla.org/network/socket-transport-service;1", &rv);
|
|
if (NS_FAILED(rv)) {
|
|
FireInteralError(this, __LINE__);
|
|
return IPC_OK();
|
|
}
|
|
|
|
nsCOMPtr<nsISocketTransport> socketTransport;
|
|
if (aUseSSL) {
|
|
AutoTArray<nsCString, 1> socketTypes = {NS_LITERAL_CSTRING("ssl")};
|
|
rv = sts->CreateTransport(socketTypes, aRemoteHost, aRemotePort, nullptr,
|
|
getter_AddRefs(socketTransport));
|
|
} else {
|
|
rv = sts->CreateTransport(nsTArray<nsCString>(), aRemoteHost, aRemotePort,
|
|
nullptr, getter_AddRefs(socketTransport));
|
|
}
|
|
|
|
if (NS_FAILED(rv)) {
|
|
FireInteralError(this, __LINE__);
|
|
return IPC_OK();
|
|
}
|
|
|
|
// in most cases aReuseAddrPort is false, but ICE TCP needs
|
|
// sockets options set that allow addr/port reuse
|
|
socketTransport->SetReuseAddrPort(aReuseAddrPort);
|
|
|
|
PRNetAddr prAddr;
|
|
if (PR_SUCCESS != PR_InitializeNetAddr(PR_IpAddrAny, aLocalPort, &prAddr)) {
|
|
FireInteralError(this, __LINE__);
|
|
return IPC_OK();
|
|
}
|
|
if (PR_SUCCESS != PR_StringToNetAddr(aLocalAddr.BeginReading(), &prAddr)) {
|
|
FireInteralError(this, __LINE__);
|
|
return IPC_OK();
|
|
}
|
|
|
|
mozilla::net::NetAddr addr;
|
|
PRNetAddrToNetAddr(&prAddr, &addr);
|
|
rv = socketTransport->Bind(&addr);
|
|
if (NS_FAILED(rv)) {
|
|
FireInteralError(this, __LINE__);
|
|
return IPC_OK();
|
|
}
|
|
|
|
if (!aFilter.IsEmpty()) {
|
|
nsAutoCString contractId(NS_NETWORK_TCP_SOCKET_FILTER_HANDLER_PREFIX);
|
|
contractId.Append(aFilter);
|
|
nsCOMPtr<nsISocketFilterHandler> filterHandler =
|
|
do_GetService(contractId.get());
|
|
if (!filterHandler) {
|
|
NS_ERROR("Content doesn't have a valid filter");
|
|
FireInteralError(this, __LINE__);
|
|
return IPC_OK();
|
|
}
|
|
rv = filterHandler->NewFilter(getter_AddRefs(mFilter));
|
|
if (NS_FAILED(rv)) {
|
|
NS_ERROR("Cannot create filter that content specified");
|
|
FireInteralError(this, __LINE__);
|
|
return IPC_OK();
|
|
}
|
|
}
|
|
|
|
mSocket = new TCPSocket(nullptr, NS_ConvertUTF8toUTF16(aRemoteHost),
|
|
aRemotePort, aUseSSL, aUseArrayBuffers);
|
|
mSocket->SetSocketBridgeParent(this);
|
|
rv = mSocket->InitWithUnconnectedTransport(socketTransport);
|
|
NS_ENSURE_SUCCESS(rv, IPC_OK());
|
|
return IPC_OK();
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvStartTLS() {
|
|
NS_ENSURE_TRUE(mSocket, IPC_OK());
|
|
ErrorResult rv;
|
|
mSocket->UpgradeToSecure(rv);
|
|
if (NS_WARN_IF(rv.Failed())) {
|
|
rv.SuppressException();
|
|
}
|
|
|
|
return IPC_OK();
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvSuspend() {
|
|
NS_ENSURE_TRUE(mSocket, IPC_OK());
|
|
mSocket->Suspend();
|
|
return IPC_OK();
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvResume() {
|
|
NS_ENSURE_TRUE(mSocket, IPC_OK());
|
|
ErrorResult rv;
|
|
mSocket->Resume(rv);
|
|
if (NS_WARN_IF(rv.Failed())) {
|
|
rv.SuppressException();
|
|
}
|
|
|
|
return IPC_OK();
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvData(
|
|
const SendableData& aData, const uint32_t& aTrackingNumber) {
|
|
ErrorResult rv;
|
|
|
|
if (mFilter) {
|
|
mozilla::net::NetAddr addr; // dummy value
|
|
bool allowed;
|
|
MOZ_ASSERT(aData.type() == SendableData::TArrayOfuint8_t,
|
|
"Unsupported data type for filtering");
|
|
const nsTArray<uint8_t>& data(aData.get_ArrayOfuint8_t());
|
|
nsresult nsrv =
|
|
mFilter->FilterPacket(&addr, data.Elements(), data.Length(),
|
|
nsISocketFilter::SF_OUTGOING, &allowed);
|
|
|
|
// Reject sending of unallowed data
|
|
if (NS_WARN_IF(NS_FAILED(nsrv)) || !allowed) {
|
|
TCPSOCKET_LOG(("%s: Dropping outgoing TCP packet", __FUNCTION__));
|
|
return IPC_FAIL_NO_REASON(this);
|
|
}
|
|
}
|
|
|
|
switch (aData.type()) {
|
|
case SendableData::TArrayOfuint8_t: {
|
|
AutoSafeJSContext autoCx;
|
|
JS::Rooted<JS::Value> val(autoCx);
|
|
const nsTArray<uint8_t>& buffer = aData.get_ArrayOfuint8_t();
|
|
bool ok = IPC::DeserializeArrayBuffer(autoCx, buffer, &val);
|
|
NS_ENSURE_TRUE(ok, IPC_OK());
|
|
RootedSpiderMonkeyInterface<ArrayBuffer> data(autoCx);
|
|
if (!data.Init(&val.toObject())) {
|
|
TCPSOCKET_LOG(("%s: Failed to allocate memory", __FUNCTION__));
|
|
return IPC_FAIL_NO_REASON(this);
|
|
}
|
|
Optional<uint32_t> byteLength(buffer.Length());
|
|
mSocket->SendWithTrackingNumber(autoCx, data, 0, byteLength,
|
|
aTrackingNumber, rv);
|
|
break;
|
|
}
|
|
|
|
case SendableData::TnsCString: {
|
|
const nsCString& strData = aData.get_nsCString();
|
|
mSocket->SendWithTrackingNumber(strData, aTrackingNumber, rv);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
MOZ_CRASH("unexpected SendableData type");
|
|
}
|
|
NS_ENSURE_SUCCESS(rv.StealNSResult(), IPC_OK());
|
|
return IPC_OK();
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvClose() {
|
|
NS_ENSURE_TRUE(mSocket, IPC_OK());
|
|
mSocket->Close();
|
|
return IPC_OK();
|
|
}
|
|
|
|
void TCPSocketParent::FireErrorEvent(const nsAString& aName,
|
|
const nsAString& aType,
|
|
TCPReadyState aReadyState) {
|
|
SendEvent(NS_LITERAL_STRING("error"),
|
|
TCPError(nsString(aName), nsString(aType)), aReadyState);
|
|
}
|
|
|
|
void TCPSocketParent::FireEvent(const nsAString& aType,
|
|
TCPReadyState aReadyState) {
|
|
return SendEvent(aType, mozilla::void_t(), aReadyState);
|
|
}
|
|
|
|
void TCPSocketParent::FireArrayBufferDataEvent(nsTArray<uint8_t>& aBuffer,
|
|
TCPReadyState aReadyState) {
|
|
nsTArray<uint8_t> arr;
|
|
arr.SwapElements(aBuffer);
|
|
|
|
if (mFilter) {
|
|
bool allowed;
|
|
mozilla::net::NetAddr addr;
|
|
nsresult nsrv =
|
|
mFilter->FilterPacket(&addr, arr.Elements(), arr.Length(),
|
|
nsISocketFilter::SF_INCOMING, &allowed);
|
|
// receiving unallowed data, drop it.
|
|
if (NS_WARN_IF(NS_FAILED(nsrv)) || !allowed) {
|
|
TCPSOCKET_LOG(("%s: Dropping incoming TCP packet", __FUNCTION__));
|
|
return;
|
|
}
|
|
}
|
|
|
|
SendableData data(arr);
|
|
SendEvent(NS_LITERAL_STRING("data"), data, aReadyState);
|
|
}
|
|
|
|
void TCPSocketParent::FireStringDataEvent(const nsACString& aData,
|
|
TCPReadyState aReadyState) {
|
|
SendableData data((nsCString(aData)));
|
|
|
|
MOZ_ASSERT(!mFilter, "Socket filtering doesn't support nsCString");
|
|
|
|
SendEvent(NS_LITERAL_STRING("data"), data, aReadyState);
|
|
}
|
|
|
|
void TCPSocketParent::SendEvent(const nsAString& aType, CallbackData aData,
|
|
TCPReadyState aReadyState) {
|
|
if (mIPCOpen) {
|
|
mozilla::Unused << PTCPSocketParent::SendCallback(
|
|
nsString(aType), aData, static_cast<uint32_t>(aReadyState));
|
|
}
|
|
}
|
|
|
|
void TCPSocketParent::SetSocket(TCPSocket* socket) { mSocket = socket; }
|
|
|
|
nsresult TCPSocketParent::GetHost(nsAString& aHost) {
|
|
if (!mSocket) {
|
|
NS_ERROR("No internal socket instance mSocket!");
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
mSocket->GetHost(aHost);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult TCPSocketParent::GetPort(uint16_t* aPort) {
|
|
if (!mSocket) {
|
|
NS_ERROR("No internal socket instance mSocket!");
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
*aPort = mSocket->Port();
|
|
return NS_OK;
|
|
}
|
|
|
|
void TCPSocketParent::ActorDestroy(ActorDestroyReason why) {
|
|
if (mSocket) {
|
|
mSocket->Close();
|
|
}
|
|
mSocket = nullptr;
|
|
}
|
|
|
|
mozilla::ipc::IPCResult TCPSocketParent::RecvRequestDelete() {
|
|
mozilla::Unused << Send__delete__(this);
|
|
return IPC_OK();
|
|
}
|
|
|
|
} // namespace dom
|
|
} // namespace mozilla
|