mirror of
https://github.com/RGBCube/serenity
synced 2025-05-21 22:35:07 +00:00
45cf40653a
Apologies for the enormous commit, but I don't see a way to split this up nicely. In the vast majority of cases it's a simple change. A few extra places can use TRY instead of manual error checking though. :^)
53 lines
1.2 KiB
C++
53 lines
1.2 KiB
C++
/*
|
|
* Copyright (c) 2020, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "Client.h"
|
|
#include "LibCore/EventLoop.h"
|
|
|
|
Client::Client(int id, NonnullOwnPtr<Core::Stream::TCPSocket> socket)
|
|
: m_id(id)
|
|
, m_socket(move(socket))
|
|
{
|
|
m_socket->on_ready_to_read = [this] {
|
|
if (m_socket->is_eof())
|
|
return;
|
|
|
|
auto result = drain_socket();
|
|
if (result.is_error()) {
|
|
dbgln("Failed while trying to drain the socket: {}", result.error());
|
|
Core::deferred_invoke([this, strong_this = NonnullRefPtr(*this)] { quit(); });
|
|
}
|
|
};
|
|
}
|
|
|
|
ErrorOr<void> Client::drain_socket()
|
|
{
|
|
NonnullRefPtr<Client> protect(*this);
|
|
|
|
auto buffer = TRY(ByteBuffer::create_uninitialized(1024));
|
|
|
|
while (TRY(m_socket->can_read_without_blocking())) {
|
|
auto nread = TRY(m_socket->read(buffer));
|
|
|
|
dbgln("Read {} bytes.", nread);
|
|
|
|
if (m_socket->is_eof()) {
|
|
Core::deferred_invoke([this, strong_this = NonnullRefPtr(*this)] { quit(); });
|
|
break;
|
|
}
|
|
|
|
TRY(m_socket->write({ buffer.data(), nread }));
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
void Client::quit()
|
|
{
|
|
m_socket->close();
|
|
if (on_exit)
|
|
on_exit();
|
|
}
|