LibWebRTCSocketFactory.cpp [plain text]
#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 <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;
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 {
callOnMainThread([] {
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)
{
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), 0);
else {
callOnMainThread([] {
WebProcess::singleton().ensureNetworkProcessConnection();
});
m_pendingMessageTasks.append([identifier = socket->identifier(), address = RTCNetwork::SocketAddress(address), minPort, maxPort](auto& connection) {
connection.send(Messages::NetworkRTCProvider::CreateUDPSocket(identifier, address, minPort, maxPort), 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)
{
ASSERT(!WTF::isMainRunLoop());
auto socket = makeUnique<LibWebRTCSocket>(*this, socketGroup, LibWebRTCSocket::Type::ClientTCP, localAddress, remoteAddress);
socket->setState(LibWebRTCSocket::STATE_CONNECTING);
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), 0);
else {
callOnMainThread([] {
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](auto& connection) {
connection.send(Messages::NetworkRTCProvider::CreateClientTCPSocket(identifier, localAddress, remoteAddress, userAgent, opts), 0);
});
}
return socket.release();
}
rtc::AsyncPacketSocket* LibWebRTCSocketFactory::createNewConnectionSocket(LibWebRTCSocket& serverSocket, LibWebRTCSocketIdentifier newConnectionSocketIdentifier, const rtc::SocketAddress& remoteAddress)
{
ASSERT(!WTF::isMainRunLoop());
if (!m_connection) {
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;
}
}
#endif // USE(LIBWEBRTC)