ipv4.h

Go to the documentation of this file.

 
#ifndef MAXOS_NET_IPV4_H
#define MAXOS_NET_IPV4_H
 
#include <cstdint>
#include <net/ethernetframe.h>
 
 
namespace MaxOS::net {
 
    typedef uint32_t InternetProtocolAddress;   
    typedef uint32_t SubnetMask;                
 
    typedef struct PACKED IPV4Header {
 
        uint8_t header_length : 4;        
        uint8_t version : 4;             
        uint8_t type_of_service;          
        uint16_t total_length;           
 
        uint16_t identifier;            
        uint16_t flags_and_offset;        
 
        uint8_t time_to_live;             
        uint8_t protocol;               
        uint16_t checksum;              
 
        uint32_t source_ip;              
        uint32_t destination_ip;         
 
    } ipv4_header_t;
 
    class InternetProtocolHandler;
 
    class IPV4AddressResolver {
        public:
            explicit IPV4AddressResolver(InternetProtocolHandler* internet_protocol_handler);
            ~IPV4AddressResolver();
            virtual drivers::ethernet::MediaAccessControlAddress resolve(InternetProtocolAddress address);
            virtual void store(InternetProtocolAddress internet_protocol_address, drivers::ethernet::MediaAccessControlAddress media_access_control_address);
    };
 
 
    class IPV4PayloadHandler {
            friend class InternetProtocolHandler;
 
        protected:
            InternetProtocolHandler* internet_protocol_handler;   
            uint8_t ip_protocol;                                 
 
        public:
            IPV4PayloadHandler(InternetProtocolHandler* internet_protocol_handler, uint8_t protocol);
            ~IPV4PayloadHandler();
 
            virtual bool handle_internet_protocol_payload(net::InternetProtocolAddress src_ip_be, net::InternetProtocolAddress dst_ip_be, uint8_t* internetprotocol_payload, uint32_t size);
            void send(InternetProtocolAddress destination_ip, uint8_t* payload_data, uint32_t size);
    };
 
    class InternetProtocolHandler : public EthernetFramePayloadHandler {
 
            friend class IPV4AddressResolver;
 
        protected:
 
            common::Map<uint8_t, IPV4PayloadHandler*> ipv_4_payload_handlers;  
 
            IPV4AddressResolver* resolver = nullptr;                                    
            common::OutputStream* error_messages;                                                    
 
            InternetProtocolAddress own_internet_protocol_address;                                     
            InternetProtocolAddress default_gateway_internet_protocol_address;                          
            SubnetMask subnet_mask;                                                                  
 
            void register_ipv_4_address_resolver(IPV4AddressResolver* ipv4_resolver);
 
        public:
            InternetProtocolHandler(EthernetFrameHandler* backend,
                                    InternetProtocolAddress own_internet_protocol_address,
                                    InternetProtocolAddress default_gateway_internet_protocol_address,
                                    SubnetMask subnet_mask,
                                    common::OutputStream* error_messages);
            ~InternetProtocolHandler();
 
            bool handle_ethernetframe_payload(uint8_t* ethernetframe_payload, uint32_t size) override;
            void send_internet_protocol_packet(uint32_t dst_ip_be, uint8_t protocol, const uint8_t* data, uint32_t size);
 
            static uint16_t checksum(const uint16_t* data, uint32_t length_in_bytes);
 
            static InternetProtocolAddress create_internet_protocol_address(uint8_t digit1, uint8_t digit2, uint8_t digit3, uint8_t digit4);
            static InternetProtocolAddress parse(string address);
            static SubnetMask create_subnet_mask(uint8_t digit1, uint8_t digit2, uint8_t digit3, uint8_t digit4);
            [[nodiscard]] InternetProtocolAddress get_internet_protocol_address() const;
            drivers::ethernet::MediaAccessControlAddress get_media_access_control_address();
 
            void connect_ipv_4_payload_handler(IPV4PayloadHandler* ipv_4_payload_handler);
 
 
    };
}
 
 
#endif //MAXOS_NET_IPV4_H