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 "UDPSocketChild.h"
8 #include "mozilla/Unused.h"
9 #include "mozilla/ipc/IPCStreamUtils.h"
10 #include "mozilla/net/NeckoChild.h"
11 #include "mozilla/dom/ContentChild.h"
12 #include "mozilla/dom/PermissionMessageUtils.h"
13 #include "mozilla/ipc/BackgroundChild.h"
14 #include "mozilla/ipc/PBackgroundChild.h"
15 #include "mozilla/ipc/BackgroundUtils.h"
16 #include "mozilla/ipc/PBackgroundSharedTypes.h"
17
18 using mozilla::net::gNeckoChild;
19
20 namespace mozilla {
21 namespace dom {
22
NS_IMPL_ISUPPORTS(UDPSocketChildBase,nsISupports)23 NS_IMPL_ISUPPORTS(UDPSocketChildBase, nsISupports)
24
25 UDPSocketChildBase::UDPSocketChildBase() : mIPCOpen(false) {}
26
27 UDPSocketChildBase::~UDPSocketChildBase() = default;
28
ReleaseIPDLReference()29 void UDPSocketChildBase::ReleaseIPDLReference() {
30 MOZ_ASSERT(mIPCOpen);
31 mIPCOpen = false;
32 mSocket = nullptr;
33 this->Release();
34 }
35
AddIPDLReference()36 void UDPSocketChildBase::AddIPDLReference() {
37 MOZ_ASSERT(!mIPCOpen);
38 mIPCOpen = true;
39 this->AddRef();
40 }
41
NS_IMETHODIMP_(MozExternalRefCountType)42 NS_IMETHODIMP_(MozExternalRefCountType) UDPSocketChild::Release(void) {
43 nsrefcnt refcnt = UDPSocketChildBase::Release();
44 if (refcnt == 1 && mIPCOpen) {
45 PUDPSocketChild::SendRequestDelete();
46 return 1;
47 }
48 return refcnt;
49 }
50
UDPSocketChild()51 UDPSocketChild::UDPSocketChild() : mBackgroundManager(nullptr), mLocalPort(0) {}
52
53 UDPSocketChild::~UDPSocketChild() = default;
54
SetBackgroundSpinsEvents()55 nsresult UDPSocketChild::SetBackgroundSpinsEvents() {
56 using mozilla::ipc::BackgroundChild;
57
58 mBackgroundManager = BackgroundChild::GetOrCreateForCurrentThread();
59 if (NS_WARN_IF(!mBackgroundManager)) {
60 return NS_ERROR_FAILURE;
61 }
62
63 return NS_OK;
64 }
65
Bind(nsIUDPSocketInternal * aSocket,nsIPrincipal * aPrincipal,const nsACString & aHost,uint16_t aPort,bool aAddressReuse,bool aLoopback,uint32_t recvBufferSize,uint32_t sendBufferSize,nsIEventTarget * aMainThreadEventTarget)66 nsresult UDPSocketChild::Bind(nsIUDPSocketInternal* aSocket,
67 nsIPrincipal* aPrincipal, const nsACString& aHost,
68 uint16_t aPort, bool aAddressReuse,
69 bool aLoopback, uint32_t recvBufferSize,
70 uint32_t sendBufferSize,
71 nsIEventTarget* aMainThreadEventTarget) {
72 UDPSOCKET_LOG(
73 ("%s: %s:%u", __FUNCTION__, PromiseFlatCString(aHost).get(), aPort));
74
75 NS_ENSURE_ARG(aSocket);
76
77 if (NS_IsMainThread()) {
78 if (aMainThreadEventTarget) {
79 gNeckoChild->SetEventTargetForActor(this, aMainThreadEventTarget);
80 }
81 if (!gNeckoChild->SendPUDPSocketConstructor(this, aPrincipal,
82 mFilterName)) {
83 return NS_ERROR_FAILURE;
84 }
85 } else {
86 if (!mBackgroundManager) {
87 return NS_ERROR_NOT_AVAILABLE;
88 }
89
90 // If we want to support a passed-in principal here we'd need to
91 // convert it to a PrincipalInfo
92 MOZ_ASSERT(!aPrincipal);
93 if (!mBackgroundManager->SendPUDPSocketConstructor(this, Nothing(),
94 mFilterName)) {
95 return NS_ERROR_FAILURE;
96 }
97 }
98
99 mSocket = aSocket;
100 AddIPDLReference();
101
102 SendBind(UDPAddressInfo(nsCString(aHost), aPort), aAddressReuse, aLoopback,
103 recvBufferSize, sendBufferSize);
104 return NS_OK;
105 }
106
Connect(nsIUDPSocketInternal * aSocket,const nsACString & aHost,uint16_t aPort)107 void UDPSocketChild::Connect(nsIUDPSocketInternal* aSocket,
108 const nsACString& aHost, uint16_t aPort) {
109 UDPSOCKET_LOG(
110 ("%s: %s:%u", __FUNCTION__, PromiseFlatCString(aHost).get(), aPort));
111
112 mSocket = aSocket;
113
114 SendConnect(UDPAddressInfo(nsCString(aHost), aPort));
115 }
116
Close()117 void UDPSocketChild::Close() { SendClose(); }
118
SendWithAddress(const NetAddr * aAddr,const uint8_t * aData,uint32_t aByteLength)119 nsresult UDPSocketChild::SendWithAddress(const NetAddr* aAddr,
120 const uint8_t* aData,
121 uint32_t aByteLength) {
122 NS_ENSURE_ARG(aAddr);
123 NS_ENSURE_ARG(aData);
124
125 UDPSOCKET_LOG(("%s: %u bytes", __FUNCTION__, aByteLength));
126 return SendDataInternal(UDPSocketAddr(*aAddr), aData, aByteLength);
127 }
128
SendDataInternal(const UDPSocketAddr & aAddr,const uint8_t * aData,const uint32_t aByteLength)129 nsresult UDPSocketChild::SendDataInternal(const UDPSocketAddr& aAddr,
130 const uint8_t* aData,
131 const uint32_t aByteLength) {
132 NS_ENSURE_ARG(aData);
133
134 FallibleTArray<uint8_t> fallibleArray;
135 if (!fallibleArray.InsertElementsAt(0, aData, aByteLength, fallible)) {
136 return NS_ERROR_OUT_OF_MEMORY;
137 }
138
139 SendOutgoingData(UDPData{std::move(fallibleArray)}, aAddr);
140
141 return NS_OK;
142 }
143
SendBinaryStream(const nsACString & aHost,uint16_t aPort,nsIInputStream * aStream)144 nsresult UDPSocketChild::SendBinaryStream(const nsACString& aHost,
145 uint16_t aPort,
146 nsIInputStream* aStream) {
147 NS_ENSURE_ARG(aStream);
148
149 mozilla::ipc::AutoIPCStream autoStream;
150 autoStream.Serialize(aStream, static_cast<mozilla::dom::ContentChild*>(
151 gNeckoChild->Manager()));
152
153 UDPSOCKET_LOG(
154 ("%s: %s:%u", __FUNCTION__, PromiseFlatCString(aHost).get(), aPort));
155 SendOutgoingData(UDPData(autoStream.TakeValue()),
156 UDPSocketAddr(UDPAddressInfo(nsCString(aHost), aPort)));
157
158 return NS_OK;
159 }
160
JoinMulticast(const nsACString & aMulticastAddress,const nsACString & aInterface)161 void UDPSocketChild::JoinMulticast(const nsACString& aMulticastAddress,
162 const nsACString& aInterface) {
163 SendJoinMulticast(nsCString(aMulticastAddress), nsCString(aInterface));
164 }
165
LeaveMulticast(const nsACString & aMulticastAddress,const nsACString & aInterface)166 void UDPSocketChild::LeaveMulticast(const nsACString& aMulticastAddress,
167 const nsACString& aInterface) {
168 SendLeaveMulticast(nsCString(aMulticastAddress), nsCString(aInterface));
169 }
170
SetFilterName(const nsACString & aFilterName)171 nsresult UDPSocketChild::SetFilterName(const nsACString& aFilterName) {
172 if (!mFilterName.IsEmpty()) {
173 // filter name can only be set once.
174 return NS_ERROR_FAILURE;
175 }
176 mFilterName = aFilterName;
177 return NS_OK;
178 }
179
180 // PUDPSocketChild Methods
RecvCallbackOpened(const UDPAddressInfo & aAddressInfo)181 mozilla::ipc::IPCResult UDPSocketChild::RecvCallbackOpened(
182 const UDPAddressInfo& aAddressInfo) {
183 mLocalAddress = aAddressInfo.addr();
184 mLocalPort = aAddressInfo.port();
185
186 UDPSOCKET_LOG(("%s: %s:%u", __FUNCTION__, mLocalAddress.get(), mLocalPort));
187 nsresult rv = mSocket->CallListenerOpened();
188 mozilla::Unused << NS_WARN_IF(NS_FAILED(rv));
189
190 return IPC_OK();
191 }
192
193 // PUDPSocketChild Methods
RecvCallbackConnected(const UDPAddressInfo & aAddressInfo)194 mozilla::ipc::IPCResult UDPSocketChild::RecvCallbackConnected(
195 const UDPAddressInfo& aAddressInfo) {
196 mLocalAddress = aAddressInfo.addr();
197 mLocalPort = aAddressInfo.port();
198
199 UDPSOCKET_LOG(("%s: %s:%u", __FUNCTION__, mLocalAddress.get(), mLocalPort));
200 nsresult rv = mSocket->CallListenerConnected();
201 mozilla::Unused << NS_WARN_IF(NS_FAILED(rv));
202
203 return IPC_OK();
204 }
205
RecvCallbackClosed()206 mozilla::ipc::IPCResult UDPSocketChild::RecvCallbackClosed() {
207 nsresult rv = mSocket->CallListenerClosed();
208 mozilla::Unused << NS_WARN_IF(NS_FAILED(rv));
209
210 return IPC_OK();
211 }
212
RecvCallbackReceivedData(const UDPAddressInfo & aAddressInfo,nsTArray<uint8_t> && aData)213 mozilla::ipc::IPCResult UDPSocketChild::RecvCallbackReceivedData(
214 const UDPAddressInfo& aAddressInfo, nsTArray<uint8_t>&& aData) {
215 UDPSOCKET_LOG(("%s: %s:%u length %zu", __FUNCTION__,
216 aAddressInfo.addr().get(), aAddressInfo.port(),
217 aData.Length()));
218 nsresult rv = mSocket->CallListenerReceivedData(aAddressInfo.addr(),
219 aAddressInfo.port(), aData);
220 mozilla::Unused << NS_WARN_IF(NS_FAILED(rv));
221
222 return IPC_OK();
223 }
224
RecvCallbackError(const nsCString & aMessage,const nsCString & aFilename,const uint32_t & aLineNumber)225 mozilla::ipc::IPCResult UDPSocketChild::RecvCallbackError(
226 const nsCString& aMessage, const nsCString& aFilename,
227 const uint32_t& aLineNumber) {
228 UDPSOCKET_LOG(("%s: %s:%s:%u", __FUNCTION__, aMessage.get(), aFilename.get(),
229 aLineNumber));
230 nsresult rv = mSocket->CallListenerError(aMessage, aFilename, aLineNumber);
231 mozilla::Unused << NS_WARN_IF(NS_FAILED(rv));
232
233 return IPC_OK();
234 }
235
236 } // namespace dom
237 } // namespace mozilla
238