mirror of
https://github.com/Fennix-Project/Kernel.git
synced 2025-05-25 22:14:37 +00:00
98 lines
2.3 KiB
C++
98 lines
2.3 KiB
C++
/*
|
|
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__
|