1/*
2 * Copyright (C) 2019 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23 * THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "config.h"
27#include "NetworkSocketChannel.h"
28
29#include "DataReference.h"
30#include "NetworkConnectionToWebProcess.h"
31#include "NetworkProcess.h"
32#include "NetworkSession.h"
33#include "WebSocketChannelMessages.h"
34#include "WebSocketTask.h"
35
36namespace WebKit {
37using namespace WebCore;
38
39std::unique_ptr<NetworkSocketChannel> NetworkSocketChannel::create(NetworkConnectionToWebProcess& connection, PAL::SessionID sessionID, const ResourceRequest& request, const String& protocol, uint64_t identifier)
40{
41 auto result = std::make_unique<NetworkSocketChannel>(connection, connection.networkProcess().networkSession(sessionID), request, protocol, identifier);
42 if (!result->m_socket) {
43 result->didClose(0, "Cannot create a web socket task"_s);
44 return nullptr;
45 }
46 return result;
47}
48
49NetworkSocketChannel::NetworkSocketChannel(NetworkConnectionToWebProcess& connection, RefPtr<NetworkSession>&& session, const ResourceRequest& request, const String& protocol, uint64_t identifier)
50 : m_connectionToWebProcess(connection)
51 , m_identifier(identifier)
52 , m_session(WTFMove(session))
53{
54 if (!m_session)
55 return;
56
57 m_socket = m_session->createWebSocketTask(*this, request, protocol);
58
59 m_session->addWebSocketTask(*m_socket);
60 m_socket->resume();
61}
62
63NetworkSocketChannel::~NetworkSocketChannel()
64{
65 if (!m_socket)
66 return;
67
68 m_socket->cancel();
69 m_session->removeWebSocketTask(*m_socket);
70}
71
72void NetworkSocketChannel::sendString(const String& message, CompletionHandler<void()>&& callback)
73{
74 m_socket->sendString(message, WTFMove(callback));
75}
76
77void NetworkSocketChannel::sendData(const IPC::DataReference& data, CompletionHandler<void()>&& callback)
78{
79 m_socket->sendData(data, WTFMove(callback));
80}
81
82void NetworkSocketChannel::finishClosingIfPossible()
83{
84 if (m_state == State::Open) {
85 m_state = State::Closing;
86 return;
87 }
88 ASSERT(m_state == State::Closing);
89 m_state = State::Closed;
90 m_connectionToWebProcess.removeSocketChannel(m_identifier);
91}
92
93void NetworkSocketChannel::close(int32_t code, const String& reason)
94{
95 m_socket->close(code, reason);
96 finishClosingIfPossible();
97}
98
99void NetworkSocketChannel::didConnect()
100{
101 send(Messages::WebSocketChannel::DidConnect { });
102}
103
104void NetworkSocketChannel::didReceiveText(const String& text)
105{
106 send(Messages::WebSocketChannel::DidReceiveText { text });
107}
108
109void NetworkSocketChannel::didReceiveBinaryData(const uint8_t* data, size_t length)
110{
111 send(Messages::WebSocketChannel::DidReceiveBinaryData { { data, length } });
112}
113
114void NetworkSocketChannel::didClose(unsigned short code, const String& reason)
115{
116 send(Messages::WebSocketChannel::DidClose { code, reason });
117 finishClosingIfPossible();
118}
119
120IPC::Connection* NetworkSocketChannel::messageSenderConnection() const
121{
122 return &m_connectionToWebProcess.connection();
123}
124
125} // namespace WebKit
126