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

Kernel: Make UDPSocket::create() API OOM safe

Browse source
This commit is contained in:
Brian Gianforcaro 2021-05-13 01:34:04 -07:00 • committed by Andreas Kling
parent 858fff979a
commit 2e34714ba1
3 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Kernel/Net/IPv4Socket.cpp
View file

@ -42,8 +42,12 @@ KResultOr<NonnullRefPtr<Socket>> IPv4Socket::create(int type, int protocol)
return tcp_socket.error();
return tcp_socket.release_value();
}
if (type == SOCK_DGRAM)
return UDPSocket::create(protocol);
if (type == SOCK_DGRAM) {
auto udp_socket = UDPSocket::create(protocol);
if (udp_socket.is_error())
return udp_socket.error();
return udp_socket.release_value();
}
if (type == SOCK_RAW) {
auto raw_socket = adopt_ref_if_nonnull(new IPv4Socket(type, protocol));
if (raw_socket)