mirror of
https://github.com/RGBCube/serenity
synced 2025-05-31 11:28:12 +00:00
43ce6c5474
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.
54 lines
1.9 KiB
C++
54 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include <Kernel/DoubleBuffer.h>
|
|
#include <Kernel/Net/Socket.h>
|
|
|
|
class FileDescription;
|
|
|
|
class LocalSocket final : public Socket {
|
|
public:
|
|
static NonnullRefPtr<LocalSocket> create(int type);
|
|
virtual ~LocalSocket() override;
|
|
|
|
// ^Socket
|
|
virtual KResult bind(const sockaddr*, socklen_t) override;
|
|
virtual KResult connect(FileDescription&, const sockaddr*, socklen_t, ShouldBlock = ShouldBlock::Yes) override;
|
|
virtual KResult listen(int) override;
|
|
virtual bool get_local_address(sockaddr*, socklen_t*) override;
|
|
virtual bool get_peer_address(sockaddr*, socklen_t*) override;
|
|
virtual void attach(FileDescription&) override;
|
|
virtual void detach(FileDescription&) override;
|
|
virtual bool can_read(FileDescription&) const override;
|
|
virtual bool can_write(FileDescription&) const override;
|
|
virtual ssize_t sendto(FileDescription&, const void*, size_t, int, const sockaddr*, socklen_t) override;
|
|
virtual ssize_t recvfrom(FileDescription&, void*, size_t, int flags, sockaddr*, socklen_t*) override;
|
|
|
|
private:
|
|
explicit LocalSocket(int type);
|
|
virtual bool is_local() const override { return true; }
|
|
bool has_attached_peer(const FileDescription&) const;
|
|
|
|
// An open socket file on the filesystem.
|
|
RefPtr<FileDescription> m_file;
|
|
|
|
// A single LocalSocket is shared between two file descriptions
|
|
// on the connect side and the accept side; so we need to store
|
|
// an additional role for the connect side and differentiate
|
|
// between them.
|
|
Role m_connect_side_role { Role::None };
|
|
FileDescription* m_connect_side_fd { nullptr };
|
|
|
|
virtual Role role(const FileDescription& description) const override
|
|
{
|
|
if (m_connect_side_fd == &description)
|
|
return m_connect_side_role;
|
|
return m_role;
|
|
}
|
|
|
|
bool m_bound { false };
|
|
bool m_accept_side_fd_open { false };
|
|
sockaddr_un m_address;
|
|
|
|
DoubleBuffer m_for_client;
|
|
DoubleBuffer m_for_server;
|
|
};
|