jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#pragma once
8
9#include <AK/Error.h>
10#include <Kernel/Locking/MutexProtected.h>
11#include <Kernel/Net/IPv4Socket.h>
12
13namespace Kernel {
14
15class UDPSocket final : public IPv4Socket {
16public:
17 static ErrorOr<NonnullRefPtr<UDPSocket>> try_create(int protocol, NonnullOwnPtr<DoubleBuffer> receive_buffer);
18 virtual ~UDPSocket() override;
19
20 static RefPtr<UDPSocket> from_port(u16);
21 static void for_each(Function<void(UDPSocket const&)>);
22 static ErrorOr<void> try_for_each(Function<ErrorOr<void>(UDPSocket const&)>);
23
24private:
25 explicit UDPSocket(int protocol, NonnullOwnPtr<DoubleBuffer> receive_buffer);
26 virtual StringView class_name() const override { return "UDPSocket"sv; }
27 static MutexProtected<HashMap<u16, UDPSocket*>>& sockets_by_port();
28
29 virtual ErrorOr<size_t> protocol_receive(ReadonlyBytes raw_ipv4_packet, UserOrKernelBuffer& buffer, size_t buffer_size, int flags) override;
30 virtual ErrorOr<size_t> protocol_send(UserOrKernelBuffer const&, size_t) override;
31 virtual ErrorOr<size_t> protocol_size(ReadonlyBytes raw_ipv4_packet) override;
32 virtual ErrorOr<void> protocol_connect(OpenFileDescription&) override;
33 virtual ErrorOr<u16> protocol_allocate_local_port() override;
34 virtual ErrorOr<void> protocol_bind() override;
35};
36
37}