1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-31 04:18:12 +00:00

Kernel: Move socket role tracking to the Socket class itself

This is more logical and allows us to solve the problem of
non-blocking TCP sockets getting stuck in SocketRole::None.

The only complication is that a single LocalSocket may be shared
between two file descriptions (on the connect and accept sides),
and should have two different roles depending from which side
you look at it. To deal with it, Socket::role() is made a
virtual method that accepts a file description, and LocalSocket
internally tracks which FileDescription is the which one and
returns a correct role.
This commit is contained in:
Sergey Bugaev 2019-08-11 16:38:20 +03:00 committed by Andreas Kling
parent d46c3b0b5b
commit 43ce6c5474
11 changed files with 103 additions and 95 deletions

View file

@ -48,6 +48,7 @@ KResult Socket::listen(int backlog)
if (m_type != SOCK_STREAM)
return KResult(-EOPNOTSUPP);
m_backlog = backlog;
m_role = Role::Listener;
kprintf("Socket{%p} listening with backlog=%d\n", this, m_backlog);
return KSuccess;
}
@ -64,6 +65,7 @@ RefPtr<Socket> Socket::accept()
ASSERT(!client->is_connected());
client->set_setup_state(SetupState::Completed);
client->m_connected = true;
client->m_role = Role::Accepted;
return client;
}
@ -146,14 +148,14 @@ void Socket::load_send_deadline()
m_send_deadline.tv_usec %= 1000000;
}
static const char* to_string(SocketRole role)
static const char* to_string(Socket::Role role)
{
switch (role) {
case SocketRole::Listener:
case Socket::Role::Listener:
return "Listener";
case SocketRole::Accepted:
case Socket::Role::Accepted:
return "Accepted";
case SocketRole::Connected:
case Socket::Role::Connected:
return "Connected";
default:
return "None";
@ -162,7 +164,7 @@ static const char* to_string(SocketRole role)
String Socket::absolute_path(const FileDescription& description) const
{
return String::format("socket:%x (role: %s)", this, ::to_string(description.socket_role()));
return String::format("socket:%x (role: %s)", this, ::to_string(role(description)));
}
ssize_t Socket::read(FileDescription& description, u8* buffer, ssize_t size)