IpAddress.h

Go to the documentation of this file.

#ifndef PCAPPP_IPADDRESS
#define PCAPPP_IPADDRESS
 
#include <memory>
#include <stdint.h>
#include <string>
 
#define MAX_ADDR_STRING_LEN 40 //xxxx:xxxx:xxxx:xxxx:xxxx:xxxx:xxxx:xxxx\0
#define MAX_IPV4_STRING_LEN 16 //xxx.xxx.xxx.xxx\0
 
 
struct in_addr;
struct in6_addr;
 
namespace pcpp
{
 
    class IPAddress
    {
    protected:
        bool m_IsValid;
        char m_AddressAsString[MAX_ADDR_STRING_LEN];
 
        // protected c'tor
        IPAddress() : m_IsValid(false) { m_AddressAsString[0] = '\0'; }
    public:
        //Visual studio has always been stupid about returning something useful for __cplusplus
        //Only recently was this fixed - and even then it requires a specific hack to the command line during build
        //Its easier/more consistent to test _MSC_VER in VS 
        //https://docs.microsoft.com/en-us/cpp/build/reference/zc-cplusplus?view=vs-2017
 
#if __cplusplus > 199711L || _MSC_VER >= 1800 //Maybe this can be 1600 for VS2010
        typedef std::unique_ptr<IPAddress> Ptr_t;
#else
        typedef std::auto_ptr<IPAddress> Ptr_t;
#endif
 
        enum AddressType {
            IPv4AddressType,
            IPv6AddressType
        };
 
        virtual ~IPAddress() {}
 
        virtual AddressType getType() const = 0;
 
        std::string toString() const { return std::string(m_AddressAsString); }
 
        bool isValid() const { return m_IsValid; }
 
        virtual IPAddress* clone() const = 0;
 
        bool equals(const IPAddress* other) const;
 
        static Ptr_t fromString(char* addressAsString);
 
        static Ptr_t fromString(std::string addressAsString);
    };
 
    class IPv4Address : public IPAddress
    {
    private:
        in_addr* m_pInAddr;
        void init(const char* addressAsString);
    public:
        IPv4Address(uint32_t addressAsInt);
 
        IPv4Address(const char* addressAsString);
 
        IPv4Address(std::string addressAsString);
 
        IPv4Address(in_addr* inAddr);
 
        ~IPv4Address();
 
        IPv4Address(const IPv4Address& other);
 
        AddressType getType() const { return IPv4AddressType; }
 
        IPAddress* clone() const;
 
        uint32_t toInt() const;
 
        in_addr* toInAddr() const { return m_pInAddr; }
 
        bool operator==(const IPv4Address& other) const { return toInt() == other.toInt(); }
 
        bool operator!=(const IPv4Address& other) const { return toInt() != other.toInt(); }
 
        IPv4Address& operator=(const IPv4Address& other);
 
        bool matchSubnet(const IPv4Address& subnet, const std::string& subnetMask) const;
 
        bool matchSubnet(const IPv4Address& subnet, const IPv4Address& subnetMask) const;
 
        static IPv4Address Zero;
    };
 
 
    class IPv6Address : public IPAddress
    {
    private:
        in6_addr* m_pInAddr;
        void init(char* addressAsString);
    public:
        ~IPv6Address();
 
        IPv6Address(uint8_t* addressAsUintArr);
 
        IPv6Address(char* addressAsString);
 
        IPv6Address(std::string addressAsString);
 
        IPv6Address(const IPv6Address& other);
 
        AddressType getType() const { return IPv6AddressType; }
 
        IPAddress* clone() const;
 
        in6_addr* toIn6Addr() const { return m_pInAddr; }
 
        void copyTo(uint8_t** arr, size_t& length) const;
 
        void copyTo(uint8_t* arr) const;
 
        bool operator==(const IPv6Address& other) const;
 
        bool operator!=(const IPv6Address& other) const;
 
        IPv6Address& operator=(const IPv6Address& other);
 
        static IPv6Address Zero;
    };
 
} // namespace pcpp
 
#endif /* PCAPPP_IPADDRESS */