RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-31 07:08:10 +00:00
Code Issues Activity Actions

Kernel: Make TCPSocket::create API OOM safe

Browse source 
Note that the changes to IPv4Socket::create are unfortunately needed as
the return type of TCPSocket::create and IPv4Socket::create don't match.

 - KResultOr<NonnullRefPtr<TcpSocket>>>
   vs
 - KResultOr<NonnullRefPtr<Socket>>>

To handle this we are forced to manually decompose the KResultOr<T> and
return the value() and error() separately.
This commit is contained in:
Brian Gianforcaro 2021-05-13 01:01:38 -07:00 • committed by Andreas Kling
parent 9375f3dc09
commit 46ce7adf7b
3 changed files with 17 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
Kernel/Net/TCPSocket.cpp
View file

@ -98,8 +98,11 @@ RefPtr<TCPSocket> TCPSocket::create_client(const IPv4Address& new_local_address,
if (sockets_by_tuple().resource().contains(tuple))
return {};
auto client = TCPSocket::create(protocol());
auto result = TCPSocket::create(protocol());
if (result.is_error())
return {};
auto client = result.release_value();
client->set_setup_state(SetupState::InProgress);
client->set_local_address(new_local_address);
client->set_local_port(new_local_port);
@ -142,9 +145,12 @@ TCPSocket::~TCPSocket()
dbgln_if(TCP_SOCKET_DEBUG, "~TCPSocket in state {}", to_string(state()));
}
NonnullRefPtr<TCPSocket> TCPSocket::create(int protocol)
KResultOr<NonnullRefPtr<TCPSocket>> TCPSocket::create(int protocol)
{
return adopt_ref(*new TCPSocket(protocol));
auto socket = adopt_ref_if_nonnull(new TCPSocket(protocol));
if (socket)
return socket.release_nonnull();
return ENOMEM;
}
KResultOr<size_t> TCPSocket::protocol_receive(ReadonlyBytes raw_ipv4_packet, UserOrKernelBuffer& buffer, size_t buffer_size, [[maybe_unused]] int flags)