1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/*
* Copyright (c) 2021, Dex♪ <dexes.ttp@gmail.com>
* Copyright (c) 2022, Ali Mohammad Pur <mpfard@serenityos.org>
* Copyright (c) 2022, the SerenityOS developers.
* Copyright (c) 2022, Andreas Kling <kling@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#define AK_DONT_REPLACE_STD
#include "WebSocketImplQt.h"
#include "Utilities.h"
#include <LibCore/EventLoop.h>
#include <QSslSocket>
#include <QTcpSocket>
namespace Ladybird {
WebSocketImplQt::~WebSocketImplQt() = default;
WebSocketImplQt::WebSocketImplQt() = default;
bool WebSocketImplQt::can_read_line()
{
return m_socket->canReadLine();
}
bool WebSocketImplQt::send(ReadonlyBytes bytes)
{
auto bytes_written = m_socket->write(reinterpret_cast<char const*>(bytes.data()), bytes.size());
if (bytes_written == -1)
return false;
VERIFY(static_cast<size_t>(bytes_written) == bytes.size());
return true;
}
bool WebSocketImplQt::eof()
{
return m_socket->state() == QTcpSocket::SocketState::UnconnectedState
&& !m_socket->bytesAvailable();
}
void WebSocketImplQt::discard_connection()
{
m_socket = nullptr;
}
void WebSocketImplQt::connect(WebSocket::ConnectionInfo const& connection_info)
{
VERIFY(!m_socket);
VERIFY(on_connected);
VERIFY(on_connection_error);
VERIFY(on_ready_to_read);
if (connection_info.is_secure()) {
auto ssl_socket = make<QSslSocket>();
ssl_socket->connectToHostEncrypted(
qstring_from_akstring(connection_info.url().host()),
connection_info.url().port_or_default());
QObject::connect(ssl_socket.ptr(), &QSslSocket::alertReceived, [this](QSsl::AlertLevel level, QSsl::AlertType, QString const&) {
if (level == QSsl::AlertLevel::Fatal)
on_connection_error();
});
m_socket = move(ssl_socket);
} else {
m_socket = make<QTcpSocket>();
m_socket->connectToHost(
qstring_from_akstring(connection_info.url().host()),
connection_info.url().port_or_default());
}
QObject::connect(m_socket.ptr(), &QTcpSocket::readyRead, [this] {
on_ready_to_read();
});
QObject::connect(m_socket.ptr(), &QTcpSocket::connected, [this] {
on_connected();
});
}
ErrorOr<ByteBuffer> WebSocketImplQt::read(int max_size)
{
auto buffer = TRY(ByteBuffer::create_uninitialized(max_size));
auto bytes_read = m_socket->read(reinterpret_cast<char*>(buffer.data()), buffer.size());
if (bytes_read == -1)
return Error::from_string_literal("WebSocketImplQt::read(): Error reading from socket");
return buffer.slice(0, bytes_read);
}
ErrorOr<String> WebSocketImplQt::read_line(size_t size)
{
auto buffer = TRY(ByteBuffer::create_uninitialized(size));
auto bytes_read = m_socket->readLine(reinterpret_cast<char*>(buffer.data()), buffer.size());
if (bytes_read == -1)
return Error::from_string_literal("WebSocketImplQt::read_line(): Error reading from socket");
return String::copy(buffer.span().slice(0, bytes_read));
}
}
|
