diff --git a/Kernel/Net/LocalSocket.cpp b/Kernel/Net/LocalSocket.cpp index 1d0309eb13..135306a272 100644 --- a/Kernel/Net/LocalSocket.cpp +++ b/Kernel/Net/LocalSocket.cpp @@ -229,34 +229,36 @@ ssize_t LocalSocket::sendto(FileDescription& description, const void* data, size ASSERT_NOT_REACHED(); } -ssize_t LocalSocket::recvfrom(FileDescription& description, void* buffer, size_t buffer_size, int, sockaddr*, socklen_t*) +DoubleBuffer& LocalSocket::buffer_for(FileDescription& description) { auto role = this->role(description); - if (role == Role::Accepted) { - if (!description.is_blocking()) { - if (m_for_server.is_empty()) { - if (!has_attached_peer(description)) - return 0; - return -EAGAIN; - } - } - ASSERT(!m_for_server.is_empty()); - return m_for_server.read((u8*)buffer, buffer_size); - } - if (role == Role::Connected) { - if (!description.is_blocking()) { - if (m_for_client.is_empty()) { - if (!has_attached_peer(description)) - return 0; - return -EAGAIN; - } - } - ASSERT(!m_for_client.is_empty()); - return m_for_client.read((u8*)buffer, buffer_size); - } + if (role == Role::Accepted) + return m_for_server; + if (role == Role::Connected) + return m_for_client; ASSERT_NOT_REACHED(); } +ssize_t LocalSocket::recvfrom(FileDescription& description, void* buffer, size_t buffer_size, int, sockaddr*, socklen_t*) +{ + auto& buffer_for_me = buffer_for(description); + if (!description.is_blocking()) { + if (buffer_for_me.is_empty()) { + if (!has_attached_peer(description)) + return 0; + return -EAGAIN; + } + } else if (!can_read(description)) { + auto result = current->block(description); + if (result == Thread::BlockResult::InterruptedBySignal) + return -EINTR; + } + if (!has_attached_peer(description) && buffer_for_me.is_empty()) + return 0; + ASSERT(!buffer_for_me.is_empty()); + return buffer_for_me.read((u8*)buffer, buffer_size); +} + StringView LocalSocket::socket_path() const { int len = strnlen(m_address.sun_path, sizeof(m_address.sun_path)); diff --git a/Kernel/Net/LocalSocket.h b/Kernel/Net/LocalSocket.h index 5ce1753653..2a6ee5bd2b 100644 --- a/Kernel/Net/LocalSocket.h +++ b/Kernel/Net/LocalSocket.h @@ -36,6 +36,7 @@ private: virtual bool is_local() const override { return true; } bool has_attached_peer(const FileDescription&) const; static Lockable>& all_sockets(); + DoubleBuffer& buffer_for(FileDescription&); // An open socket file on the filesystem. RefPtr m_file;