mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-10-16 07:19:32 +03:00
79fa9765ca
We now use AK::Error and AK::ErrorOr<T> in both kernel and userspace! This was a slightly tedious refactoring that took a long time, so it's not unlikely that some bugs crept in. Nevertheless, it does pass basic functionality testing, and it's just real nice to finally see the same pattern in all contexts. :^)
112 lines
4.5 KiB
C++
112 lines
4.5 KiB
C++
/*
|
|
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/IntrusiveList.h>
|
|
#include <Kernel/DoubleBuffer.h>
|
|
#include <Kernel/Net/Socket.h>
|
|
|
|
namespace Kernel {
|
|
|
|
class OpenFileDescription;
|
|
|
|
struct SocketPair {
|
|
NonnullRefPtr<OpenFileDescription> description0;
|
|
NonnullRefPtr<OpenFileDescription> description1;
|
|
};
|
|
|
|
class LocalSocket final : public Socket {
|
|
|
|
public:
|
|
static ErrorOr<NonnullRefPtr<LocalSocket>> try_create(int type);
|
|
static ErrorOr<SocketPair> try_create_connected_pair(int type);
|
|
virtual ~LocalSocket() override;
|
|
|
|
ErrorOr<void> sendfd(OpenFileDescription const& socket_description, NonnullRefPtr<OpenFileDescription> passing_description);
|
|
ErrorOr<NonnullRefPtr<OpenFileDescription>> recvfd(const OpenFileDescription& socket_description);
|
|
|
|
static void for_each(Function<void(const LocalSocket&)>);
|
|
|
|
StringView socket_path() const;
|
|
ErrorOr<NonnullOwnPtr<KString>> pseudo_path(const OpenFileDescription& description) const override;
|
|
|
|
// ^Socket
|
|
virtual ErrorOr<void> bind(Userspace<const sockaddr*>, socklen_t) override;
|
|
virtual ErrorOr<void> connect(OpenFileDescription&, Userspace<const sockaddr*>, socklen_t, ShouldBlock = ShouldBlock::Yes) override;
|
|
virtual ErrorOr<void> listen(size_t) override;
|
|
virtual void get_local_address(sockaddr*, socklen_t*) override;
|
|
virtual void get_peer_address(sockaddr*, socklen_t*) override;
|
|
virtual ErrorOr<void> attach(OpenFileDescription&) override;
|
|
virtual void detach(OpenFileDescription&) override;
|
|
virtual bool can_read(const OpenFileDescription&, size_t) const override;
|
|
virtual bool can_write(const OpenFileDescription&, size_t) const override;
|
|
virtual ErrorOr<size_t> sendto(OpenFileDescription&, const UserOrKernelBuffer&, size_t, int, Userspace<const sockaddr*>, socklen_t) override;
|
|
virtual ErrorOr<size_t> recvfrom(OpenFileDescription&, UserOrKernelBuffer&, size_t, int flags, Userspace<sockaddr*>, Userspace<socklen_t*>, Time&) override;
|
|
virtual ErrorOr<void> getsockopt(OpenFileDescription&, int level, int option, Userspace<void*>, Userspace<socklen_t*>) override;
|
|
virtual ErrorOr<void> ioctl(OpenFileDescription&, unsigned request, Userspace<void*> arg) override;
|
|
virtual ErrorOr<void> chown(OpenFileDescription&, UserID, GroupID) override;
|
|
virtual ErrorOr<void> chmod(OpenFileDescription&, mode_t) override;
|
|
|
|
private:
|
|
explicit LocalSocket(int type, NonnullOwnPtr<DoubleBuffer> client_buffer, NonnullOwnPtr<DoubleBuffer> server_buffer);
|
|
virtual StringView class_name() const override { return "LocalSocket"sv; }
|
|
virtual bool is_local() const override { return true; }
|
|
bool has_attached_peer(const OpenFileDescription&) const;
|
|
DoubleBuffer* receive_buffer_for(OpenFileDescription&);
|
|
DoubleBuffer* send_buffer_for(OpenFileDescription&);
|
|
NonnullRefPtrVector<OpenFileDescription>& sendfd_queue_for(const OpenFileDescription&);
|
|
NonnullRefPtrVector<OpenFileDescription>& recvfd_queue_for(const OpenFileDescription&);
|
|
|
|
void set_connect_side_role(Role connect_side_role, bool force_evaluate_block_conditions = false)
|
|
{
|
|
auto previous = m_connect_side_role;
|
|
m_connect_side_role = connect_side_role;
|
|
if (previous != m_connect_side_role || force_evaluate_block_conditions)
|
|
evaluate_block_conditions();
|
|
}
|
|
|
|
ErrorOr<void> try_set_path(StringView);
|
|
|
|
// The inode this socket is bound to.
|
|
WeakPtr<Inode> m_inode;
|
|
|
|
UserID m_prebind_uid { 0 };
|
|
GroupID m_prebind_gid { 0 };
|
|
mode_t m_prebind_mode { 0 };
|
|
|
|
// 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 };
|
|
OpenFileDescription* m_connect_side_fd { nullptr };
|
|
|
|
virtual Role role(const OpenFileDescription& 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 };
|
|
OwnPtr<KString> m_path;
|
|
|
|
NonnullOwnPtr<DoubleBuffer> m_for_client;
|
|
NonnullOwnPtr<DoubleBuffer> m_for_server;
|
|
|
|
NonnullRefPtrVector<OpenFileDescription> m_fds_for_client;
|
|
NonnullRefPtrVector<OpenFileDescription> m_fds_for_server;
|
|
|
|
IntrusiveListNode<LocalSocket> m_list_node;
|
|
|
|
public:
|
|
using List = IntrusiveList<&LocalSocket::m_list_node>;
|
|
};
|
|
|
|
}
|