| /* |
| * Copyright (C) 2017 Apple Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' |
| * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, |
| * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS |
| * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF |
| * THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "config.h" |
| #include "LibWebRTCSocketFactory.h" |
| |
| #if USE(LIBWEBRTC) |
| |
| #include "LibWebRTCNetwork.h" |
| #include "Logging.h" |
| #include "NetworkProcessConnection.h" |
| #include "NetworkRTCProviderMessages.h" |
| #include "WebProcess.h" |
| #include <WebCore/RuntimeEnabledFeatures.h> |
| #include <wtf/MainThread.h> |
| |
| namespace WebKit { |
| using namespace WebCore; |
| |
| static inline rtc::SocketAddress prepareSocketAddress(const rtc::SocketAddress& address, bool disableNonLocalhostConnections) |
| { |
| auto result = RTCNetwork::isolatedCopy(address); |
| if (disableNonLocalhostConnections) |
| result.SetIP("127.0.0.1"); |
| return result; |
| } |
| |
| void LibWebRTCSocketFactory::setConnection(RefPtr<IPC::Connection>&& connection) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| m_connection = WTFMove(connection); |
| if (!m_connection) |
| return; |
| |
| m_connection->send(Messages::NetworkRTCProvider::SetPlatformTCPSocketsEnabled(RuntimeEnabledFeatures::sharedFeatures().webRTCPlatformTCPSocketsEnabled()), 0); |
| m_connection->send(Messages::NetworkRTCProvider::SetPlatformUDPSocketsEnabled(RuntimeEnabledFeatures::sharedFeatures().webRTCPlatformUDPSocketsEnabled()), 0); |
| |
| while (!m_pendingMessageTasks.isEmpty()) |
| m_pendingMessageTasks.takeFirst()(*m_connection); |
| } |
| |
| IPC::Connection* LibWebRTCSocketFactory::connection() |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| return m_connection.get(); |
| } |
| |
| rtc::AsyncPacketSocket* LibWebRTCSocketFactory::createServerTcpSocket(const void* socketGroup, const rtc::SocketAddress& address, uint16_t minPort, uint16_t maxPort, int options) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| auto socket = makeUnique<LibWebRTCSocket>(*this, socketGroup, LibWebRTCSocket::Type::ServerTCP, address, rtc::SocketAddress()); |
| |
| if (m_connection) |
| m_connection->send(Messages::NetworkRTCProvider::CreateServerTCPSocket(socket->identifier(), RTCNetwork::SocketAddress(address), minPort, maxPort, options), 0); |
| else { |
| callOnMainRunLoop([] { |
| WebProcess::singleton().ensureNetworkProcessConnection(); |
| }); |
| m_pendingMessageTasks.append([identifier = socket->identifier(), address = RTCNetwork::SocketAddress(address), minPort, maxPort, options](auto& connection) { |
| connection.send(Messages::NetworkRTCProvider::CreateServerTCPSocket(identifier, address, minPort, maxPort, options), 0); |
| }); |
| } |
| return socket.release(); |
| } |
| |
| rtc::AsyncPacketSocket* LibWebRTCSocketFactory::createUdpSocket(const void* socketGroup, const rtc::SocketAddress& address, uint16_t minPort, uint16_t maxPort, WebPageProxyIdentifier pageIdentifier, bool isFirstParty, bool isRelayDisabled, const WebCore::RegistrableDomain& domain) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| auto socket = makeUnique<LibWebRTCSocket>(*this, socketGroup, LibWebRTCSocket::Type::UDP, address, rtc::SocketAddress()); |
| |
| if (m_connection) |
| m_connection->send(Messages::NetworkRTCProvider::CreateUDPSocket(socket->identifier(), RTCNetwork::SocketAddress(address), minPort, maxPort, pageIdentifier, isFirstParty, isRelayDisabled, domain), 0); |
| else { |
| callOnMainRunLoop([] { |
| WebProcess::singleton().ensureNetworkProcessConnection(); |
| }); |
| m_pendingMessageTasks.append([identifier = socket->identifier(), address = RTCNetwork::SocketAddress(address), minPort, maxPort, pageIdentifier, isFirstParty, isRelayDisabled, domain](auto& connection) { |
| connection.send(Messages::NetworkRTCProvider::CreateUDPSocket(identifier, address, minPort, maxPort, pageIdentifier, isFirstParty, isRelayDisabled, domain), 0); |
| }); |
| } |
| |
| return socket.release(); |
| } |
| |
| rtc::AsyncPacketSocket* LibWebRTCSocketFactory::createClientTcpSocket(const void* socketGroup, const rtc::SocketAddress& localAddress, const rtc::SocketAddress& remoteAddress, String&& userAgent, const rtc::PacketSocketTcpOptions& options, WebPageProxyIdentifier pageIdentifier, bool isFirstParty, bool isRelayDisabled, const WebCore::RegistrableDomain& domain) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| |
| auto socket = makeUnique<LibWebRTCSocket>(*this, socketGroup, LibWebRTCSocket::Type::ClientTCP, localAddress, remoteAddress); |
| socket->setState(LibWebRTCSocket::STATE_CONNECTING); |
| |
| // FIXME: We only transfer options.opts but should also handle other members. |
| if (m_connection) |
| m_connection->send(Messages::NetworkRTCProvider::CreateClientTCPSocket(socket->identifier(), RTCNetwork::SocketAddress(prepareSocketAddress(localAddress, m_disableNonLocalhostConnections)), RTCNetwork::SocketAddress(prepareSocketAddress(remoteAddress, m_disableNonLocalhostConnections)), userAgent, options.opts, pageIdentifier, isFirstParty, isRelayDisabled, domain), 0); |
| else { |
| callOnMainRunLoop([] { |
| WebProcess::singleton().ensureNetworkProcessConnection(); |
| }); |
| m_pendingMessageTasks.append([identifier = socket->identifier(), localAddress = RTCNetwork::SocketAddress(prepareSocketAddress(localAddress, m_disableNonLocalhostConnections)), remoteAddress = RTCNetwork::SocketAddress(prepareSocketAddress(remoteAddress, m_disableNonLocalhostConnections)), userAgent, opts = options.opts, pageIdentifier, isFirstParty, isRelayDisabled, domain](auto& connection) { |
| connection.send(Messages::NetworkRTCProvider::CreateClientTCPSocket(identifier, localAddress, remoteAddress, userAgent, opts, pageIdentifier, isFirstParty, isRelayDisabled, domain), 0); |
| }); |
| } |
| |
| return socket.release(); |
| } |
| |
| rtc::AsyncPacketSocket* LibWebRTCSocketFactory::createNewConnectionSocket(LibWebRTCSocket& serverSocket, LibWebRTCSocketIdentifier newConnectionSocketIdentifier, const rtc::SocketAddress& remoteAddress) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| if (!m_connection) { |
| // No need to enqueue a message in this case since it means the network process handling the incoming socket is gone. |
| RELEASE_LOG(WebRTC, "No connection to create incoming TCP socket"); |
| return nullptr; |
| } |
| |
| auto socket = makeUnique<LibWebRTCSocket>(*this, serverSocket.socketGroup(), LibWebRTCSocket::Type::ServerConnectionTCP, serverSocket.localAddress(), remoteAddress); |
| socket->setState(LibWebRTCSocket::STATE_CONNECTED); |
| |
| m_connection->send(Messages::NetworkRTCProvider::WrapNewTCPConnection(socket->identifier(), newConnectionSocketIdentifier), 0); |
| |
| return socket.release(); |
| } |
| |
| void LibWebRTCSocketFactory::addSocket(LibWebRTCSocket& socket) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| ASSERT(!m_sockets.contains(socket.identifier())); |
| m_sockets.add(socket.identifier(), &socket); |
| } |
| |
| void LibWebRTCSocketFactory::removeSocket(LibWebRTCSocket& socket) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| ASSERT(m_sockets.contains(socket.identifier())); |
| m_sockets.remove(socket.identifier()); |
| } |
| |
| void LibWebRTCSocketFactory::forSocketInGroup(const void* socketGroup, const Function<void(LibWebRTCSocket&)>& callback) |
| { |
| ASSERT(!WTF::isMainRunLoop()); |
| for (auto* socket : m_sockets.values()) { |
| if (socket->socketGroup() == socketGroup) |
| callback(*socket); |
| } |
| } |
| |
| rtc::AsyncResolverInterface* LibWebRTCSocketFactory::createAsyncResolver() |
| { |
| auto resolver = makeUnique<LibWebRTCResolver>(); |
| auto* resolverPointer = resolver.get(); |
| m_resolvers.set(resolverPointer->identifier(), WTFMove(resolver)); |
| return resolverPointer; |
| } |
| |
| } // namespace WebKit |
| |
| #endif // USE(LIBWEBRTC) |