IPV4.h

Go to the documentation of this file.

/*
 * Copyright (C) 2015 Niek Linnenbank
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef __LIB_LIBNET_IPV4_H
#define __LIB_LIBNET_IPV4_H
 
#include <Types.h>
#include <String.h>
#include "NetworkProtocol.h"
 
class ICMP;
class ARP;
class UDP;
 
class IPV4 : public NetworkProtocol
{
  public:
 
    typedef u32 Address;
 
    enum Protocol
    {
        ICMP = 1,
        IGMP = 2,
        UDP  = 17,
        TCP  = 6
    };
 
    typedef struct Header
    {
        u8  versionIHL; 
        u8  typeOfService;
        u16 length;
        u16 identification;
        u16 fragmentOffset;
        u8  timeToLive;
        u8  protocol;
        u16 checksum;
        Address source;
        Address destination;
    }
    PACKED Header;
 
    typedef struct PseudoHeader
    {
        Address source;
        Address destination;
        u8      reserved;
        u8      protocol;
        u16     length;
    }
    PseudoHeader;
 
  public:
 
    IPV4(NetworkServer &server,
         NetworkDevice &device,
         NetworkProtocol &parent);
 
    virtual ~IPV4();
 
    virtual FileSystem::Result initialize();
 
    void setICMP(::ICMP *icmp);
 
    void setARP(::ARP *arp);
 
    void setUDP(::UDP *udp);
 
    virtual FileSystem::Result getAddress(Address *address);
 
    virtual FileSystem::Result setAddress(const Address *address);
 
    static const String toString(const Address address);
 
    static const Address toAddress(const char *address);
 
    virtual FileSystem::Result getTransmitPacket(NetworkQueue::Packet **pkt,
                                                 const void *address,
                                                 const Size addressSize,
                                                 const Identifier protocol,
                                                 const Size payloadSize);
 
    virtual FileSystem::Result process(const NetworkQueue::Packet *pkt,
                                       const Size offset);
 
    static const u16 checksum(const void *buffer,
                              const Size length);
 
  private:
 
    Protocol getProtocolByIdentifier(const NetworkProtocol::Identifier id) const;
 
  private:
 
    Address m_address;
 
    ::ICMP *m_icmp;
 
    ::ARP *m_arp;
 
    ::UDP *m_udp;
 
    u16 m_id;
};
 
#endif /* __LIB_LIBNET_IPV4_H */