Bananymous/banan-os
1
1
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Kernel: Move on_close_impl from network socket to udp socket

Browse Source
This commit is contained in:
Bananymous
2024-02-12 04:26:25 +02:00
parent be01ccdb08
commit ba06269b14
7 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
kernel/kernel/Networking/IPv4Layer.cpp
View File

@@ -4,6 +4,7 @@
#include <kernel/Networking/IPv4Layer.h>
#include <kernel/Networking/NetworkManager.h>
#include <kernel/Networking/UDPSocket.h>
#include <kernel/Random.h>
#include <netinet/in.h>
@@ -89,19 +90,18 @@ namespace Kernel
LockGuard _(m_lock);
uint16_t port = NetworkSocket::PORT_NONE;
for (uint32_t temp = 0xC000; temp < 0xFFFF; temp++)
{
if (!m_bound_sockets.contains(temp))
{
for (uint32_t i = 0; i < 100 && port == NetworkSocket::PORT_NONE; i++)
if (uint32_t temp = 0xC000 | (Random::get_u32() & 0x3FFF); !m_bound_sockets.contains(temp))
port = temp;
for (uint32_t temp = 0xC000; temp < 0xFFFF && port == NetworkSocket::PORT_NONE; temp++)
if (!m_bound_sockets.contains(temp))
port = temp;
break;
}
}
if (port == NetworkSocket::PORT_NONE)
{
dwarnln("No ports available");
return BAN::Error::from_errno(EAGAIN);
}
dprintln_if(DEBUG_IPV4, "using port {}", port);
struct sockaddr_in target;
target.sin_family = AF_INET;

10
kernel/kernel/Networking/NetworkSocket.cpp
View File

@@ -15,16 +15,6 @@ namespace Kernel
{
}
void NetworkSocket::on_close_impl()
{
if (is_bound())
{
m_network_layer.unbind_socket(this, m_port);
m_interface = nullptr;
m_port = PORT_NONE;
}
}
void NetworkSocket::bind_interface_and_port(NetworkInterface* interface, uint16_t port)
{
ASSERT(!m_interface);

8
kernel/kernel/Networking/UDPSocket.cpp
View File

@@ -24,6 +24,14 @@ namespace Kernel
: NetworkSocket(network_layer, ino, inode_info)
{ }
void UDPSocket::on_close_impl()
{
if (is_bound())
m_network_layer.unbind_socket(this, m_port);
m_port = PORT_NONE;
m_interface = nullptr;
}
void UDPSocket::add_protocol_header(BAN::ByteSpan packet, uint16_t dst_port, PseudoHeader)
{
auto& header = packet.as<UDPHeader>();