TCPSocketChild.cpp (5146B)
1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ 2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */ 3 /* This Source Code Form is subject to the terms of the Mozilla Public 4 * License, v. 2.0. If a copy of the MPL was not distributed with this file, 5 * You can obtain one at http://mozilla.org/MPL/2.0/. */ 6 7 #include "TCPSocketChild.h" 8 9 #include <algorithm> 10 11 #include "TCPSocket.h" 12 #include "js/ArrayBuffer.h" // JS::NewArrayBufferWithContents 13 #include "js/RootingAPI.h" // JS::MutableHandle 14 #include "js/Utility.h" // js::ArrayBufferContentsArena, JS::FreePolicy, js_pod_arena_malloc 15 #include "js/Value.h" // JS::Value 16 #include "mozilla/HoldDropJSObjects.h" 17 #include "mozilla/UniquePtr.h" 18 #include "mozilla/dom/BrowserChild.h" 19 #include "mozilla/dom/PBrowserChild.h" 20 #include "mozilla/net/NeckoChild.h" 21 #include "nsContentUtils.h" 22 #include "nsITCPSocketCallback.h" 23 24 using mozilla::net::gNeckoChild; 25 26 namespace IPC { 27 28 bool DeserializeArrayBuffer(JSContext* cx, const nsTArray<uint8_t>& aBuffer, 29 JS::MutableHandle<JS::Value> aVal) { 30 mozilla::UniquePtr<uint8_t[], JS::FreePolicy> data( 31 js_pod_arena_malloc<uint8_t>(js::ArrayBufferContentsArena, 32 aBuffer.Length())); 33 if (!data) return false; 34 memcpy(data.get(), aBuffer.Elements(), aBuffer.Length()); 35 36 JSObject* obj = 37 JS::NewArrayBufferWithContents(cx, aBuffer.Length(), std::move(data)); 38 if (!obj) return false; 39 40 aVal.setObject(*obj); 41 return true; 42 } 43 44 } // namespace IPC 45 46 namespace mozilla::dom { 47 48 NS_IMPL_CYCLE_COLLECTION_CLASS(TCPSocketChildBase) 49 50 NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(TCPSocketChildBase) 51 NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mSocket) 52 NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END 53 54 NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(TCPSocketChildBase) 55 NS_IMPL_CYCLE_COLLECTION_UNLINK(mSocket) 56 NS_IMPL_CYCLE_COLLECTION_UNLINK_END 57 58 NS_IMPL_CYCLE_COLLECTION_TRACE_BEGIN(TCPSocketChildBase) 59 NS_IMPL_CYCLE_COLLECTION_TRACE_END 60 61 NS_IMPL_CYCLE_COLLECTING_ADDREF(TCPSocketChildBase) 62 NS_IMPL_CYCLE_COLLECTING_RELEASE(TCPSocketChildBase) 63 64 NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(TCPSocketChildBase) 65 NS_INTERFACE_MAP_ENTRY(nsISupports) 66 NS_INTERFACE_MAP_END 67 68 TCPSocketChildBase::TCPSocketChildBase() : mIPCOpen(false) { 69 mozilla::HoldJSObjects(this); 70 } 71 72 TCPSocketChildBase::~TCPSocketChildBase() { mozilla::DropJSObjects(this); } 73 74 NS_IMETHODIMP_(MozExternalRefCountType) TCPSocketChild::Release(void) { 75 nsrefcnt refcnt = TCPSocketChildBase::Release(); 76 if (refcnt == 1 && mIPCOpen) { 77 PTCPSocketChild::SendRequestDelete(); 78 return 1; 79 } 80 return refcnt; 81 } 82 83 TCPSocketChild::TCPSocketChild(const nsAString& aHost, const uint16_t& aPort, 84 nsISerialEventTarget* aTarget) 85 : mHost(aHost), mPort(aPort), mIPCEventTarget(aTarget) {} 86 87 void TCPSocketChild::SendOpen(nsITCPSocketCallback* aSocket, bool aUseSSL, 88 bool aUseArrayBuffers) { 89 mSocket = aSocket; 90 91 AddIPDLReference(); 92 gNeckoChild->SendPTCPSocketConstructor(this, mHost, mPort); 93 PTCPSocketChild::SendOpen(mHost, mPort, aUseSSL, aUseArrayBuffers); 94 } 95 96 void TCPSocketChildBase::ReleaseIPDLReference() { 97 MOZ_ASSERT(mIPCOpen); 98 mIPCOpen = false; 99 mSocket = nullptr; 100 this->Release(); 101 } 102 103 void TCPSocketChildBase::AddIPDLReference() { 104 MOZ_ASSERT(!mIPCOpen); 105 mIPCOpen = true; 106 this->AddRef(); 107 } 108 109 TCPSocketChild::~TCPSocketChild() = default; 110 111 mozilla::ipc::IPCResult TCPSocketChild::RecvUpdateBufferedAmount( 112 const uint32_t& aBuffered, const uint32_t& aTrackingNumber) { 113 mSocket->UpdateBufferedAmount(aBuffered, aTrackingNumber); 114 return IPC_OK(); 115 } 116 117 mozilla::ipc::IPCResult TCPSocketChild::RecvCallback( 118 const nsString& aType, const CallbackData& aData, 119 const uint32_t& aReadyState) { 120 mSocket->UpdateReadyState(aReadyState); 121 122 if (aData.type() == CallbackData::Tvoid_t) { 123 mSocket->FireEvent(aType); 124 125 } else if (aData.type() == CallbackData::TTCPError) { 126 const TCPError& err(aData.get_TCPError()); 127 mSocket->FireErrorEvent(err.name(), err.message(), err.errorCode()); 128 129 } else if (aData.type() == CallbackData::TSendableData) { 130 const SendableData& data = aData.get_SendableData(); 131 132 if (data.type() == SendableData::TArrayOfuint8_t) { 133 mSocket->FireDataArrayEvent(aType, data.get_ArrayOfuint8_t()); 134 } else if (data.type() == SendableData::TnsCString) { 135 mSocket->FireDataStringEvent(aType, data.get_nsCString()); 136 } else { 137 MOZ_CRASH("Invalid callback data type!"); 138 } 139 } else { 140 MOZ_CRASH("Invalid callback type!"); 141 } 142 return IPC_OK(); 143 } 144 145 void TCPSocketChild::SendSend(const nsACString& aData) { 146 SendData(nsCString(aData)); 147 } 148 149 void TCPSocketChild::SendSend(nsTArray<uint8_t>&& aData) { 150 SendData(SendableData{std::move(aData)}); 151 } 152 153 void TCPSocketChild::SetSocket(TCPSocket* aSocket) { mSocket = aSocket; } 154 155 void TCPSocketChild::GetHost(nsAString& aHost) { aHost = mHost; } 156 157 void TCPSocketChild::GetPort(uint16_t* aPort) const { *aPort = mPort; } 158 159 mozilla::ipc::IPCResult TCPSocketChild::RecvRequestDelete() { 160 (void)Send__delete__(this); 161 return IPC_OK(); 162 } 163 164 } // namespace mozilla::dom