Merge remote-tracking branch 'Kernel/master'

Kernel/include/net/udp.hpp

@@ -0,0 +1,97 @@
+/*
+	This file is part of Fennix Kernel.
+
+	Fennix Kernel is free software: you can redistribute it and/or
+	modify it under the terms of the GNU General Public License as
+	published by the Free Software Foundation, either version 3 of
+	the License, or (at your option) any later version.
+
+	Fennix Kernel is distributed in the hope that it will be useful,
+	but WITHOUT ANY WARRANTY; without even the implied warranty of
+	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+	GNU General Public License for more details.
+
+	You should have received a copy of the GNU General Public License
+	along with Fennix Kernel. If not, see <https://www.gnu.org/licenses/>.
+*/
+
+#ifndef __FENNIX_KERNEL_UDP_H__
+#define __FENNIX_KERNEL_UDP_H__
+
+#include <net/ipv4.hpp>
+#include <net/udp.hpp>
+#include <net/nc.hpp>
+#include <types.h>
+
+namespace NetworkUDP
+{
+	struct UDPHeader
+	{
+		uint16_t SourcePort;
+		uint16_t DestinationPort;
+		uint16_t Length;
+		uint16_t Checksum;
+	} __packed;
+
+	struct UDPPacket
+	{
+		UDPHeader Header;
+		uint8_t Data[];
+	};
+
+	class Socket;
+
+	class UDPEvents
+	{
+	protected:
+		UDPEvents();
+		~UDPEvents();
+
+	public:
+		virtual void OnUDPPacketReceived(Socket *Socket, uint8_t *Data, size_t Length)
+		{
+			UNUSED(Socket);
+			UNUSED(Data);
+			UNUSED(Length);
+			warn("Not implemented");
+		}
+	};
+
+	class UDP : public NetworkIPv4::IPv4Events
+	{
+	private:
+		NetworkIPv4::IPv4 *ipv4;
+		NetworkInterfaceManager::DeviceInterface *Interface;
+
+	public:
+		NetworkInterfaceManager::DeviceInterface *GetInterface() { return this->Interface; }
+
+		UDP(NetworkIPv4::IPv4 *ipv4, NetworkInterfaceManager::DeviceInterface *Interface);
+		~UDP();
+
+		virtual Socket *Connect(InternetProtocol IP, uint16_t Port);
+		virtual Socket *Listen(uint16_t Port);
+		virtual void Disconnect(Socket *Socket);
+		virtual void Send(Socket *Socket, uint8_t *Data, size_t Length);
+		virtual void Bind(Socket *Socket, UDPEvents *EventHandler);
+
+		virtual bool OnIPv4PacketReceived(InternetProtocol SourceIP, InternetProtocol DestinationIP, uint8_t *Data, size_t Length);
+	};
+
+	class Socket
+	{
+	public:
+		InternetProtocol LocalIP;
+		uint16_t LocalPort = 0;
+		InternetProtocol RemoteIP;
+		uint16_t RemotePort = 0;
+		bool Listening = false;
+		UDPEvents *EventHandler = nullptr;
+		UDP *SocketUDP = nullptr;
+
+		Socket(UDP *_UDP);
+		~Socket();
+	};
+}
+
+#endif // !__FENNIX_KERNEL_UDP_H__