IpAddress.h

Go to the documentation of this file.

/*
  IpAddress.h - Base class that provides IP Address
  Copyright (c) 2011 Adrian McEwen.  All right reserved.

  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License as published by the Free Software Foundation; either
  version 2.1 of the License, or (at your option) any later version.

  This library 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
  Lesser General Public License for more details.

  You should have received a copy of the GNU Lesser General Public
  License along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#pragma once

#include <lwip/init.h>
#include <lwip/ip_addr.h>
#include "Printable.h"
#include "WString.h"

#if LWIP_VERSION_MAJOR == 2
#define LWIP_IP_ADDR_T const ip_addr_t
#else
using ip_addr_t = struct ip_addr;
using ip4_addr_t = ip_addr_t;
#define IP_ADDR4(IP, A, B, C, D) IP4_ADDR(IP, A, B, C, D)
#define ip_addr_set_ip4_u32(IP, U32) ip4_addr_set_u32(IP, U32)
#define ip_addr_get_ip4_u32(IP) ip4_addr_get_u32(IP)
#define ip_2_ip4(IP) (IP)
#define ip4_addr_netcmp(A, B, C) ip_addr_netcmp(A, B, C)
#define LWIP_IP_ADDR_T ip_addr_t
#endif

class IpAddress
{
private:
    ip_addr_t address{0}; 

    void fromString(const String& address);

public:
    // Constructors
    IpAddress()
    {
    }

    IpAddress(uint8_t first_octet, uint8_t second_octet, uint8_t third_octet, uint8_t fourth_octet)
    {
        IP_ADDR4(&address, first_octet, second_octet, third_octet, fourth_octet);
    }

    IpAddress(uint32_t address)
    {
        ip_addr_set_ip4_u32(&this->address, address);
    }

    IpAddress(ip_addr_t& addr)
    {
        address = addr;
    }

    IpAddress(const ip_addr_t& addr)
    {
        address = addr;
    }

#if LWIP_VERSION_MAJOR == 2 && LWIP_IPV6
    IpAddress(ip4_addr_t& addr)
    {
        ip_addr_copy_from_ip4(address, addr);
    }
#endif

    IpAddress(const uint8_t address[4])
    {
        IP_ADDR4(&this->address, address[0], address[1], address[2], address[3]);
    }

    IpAddress(const String& address)
    {
        fromString(address);
    }

    operator uint32_t() const
    {
        return ip_addr_get_ip4_u32(&address);
    }

    operator ip_addr_t() const
    {
        return address;
    }

    operator ip_addr_t*()
    {
        return &address;
    }

#if LWIP_VERSION_MAJOR == 2 && LWIP_IPV6
    operator ip4_addr_t() const
    {
        return *ip_2_ip4(&address);
    }

    operator ip4_addr_t*()
    {
        return ip_2_ip4(&address);
    }
#endif

    bool operator==(const IpAddress& otherAddress) const
    {
        return ip_addr_cmp(&address, &otherAddress.address);
    }

    bool operator==(const uint8_t addr[4]) const
    {
        return *this == IpAddress(addr);
    }

    bool operator!=(const IpAddress& otherAddress) const
    {
        return !ip_addr_cmp(&address, &otherAddress.address);
    }

    bool operator!=(const uint8_t addr[4]) const
    {
        return *this != IpAddress(addr);
    }

    bool isNull() const
    {
        return ip_addr_isany(&address);
    }

    String toString() const;

    bool compare(const IpAddress& addr, const IpAddress& mask) const
    {
        return ip_addr_netcmp(&address, &addr.address, ip_2_ip4(&mask.address));
    }

    // Overloaded index operator to allow getting and setting individual octets of the address
    uint8_t operator[](int index) const
    {
        if(unsigned(index) >= sizeof(ip4_addr_t)) {
            abort();
        }

        return reinterpret_cast<const uint8_t*>(&ip_2_ip4(&address)->addr)[index];
    }

    uint8_t& operator[](int index)
    {
        if(unsigned(index) >= sizeof(ip4_addr_t)) {
            abort();
        }

        return reinterpret_cast<uint8_t*>(&ip_2_ip4(&address)->addr)[index];
    }

    // Overloaded copy operators to allow initialisation of IpAddress objects from other types
    IpAddress& operator=(const uint8_t address[4])
    {
        IP_ADDR4(&this->address, address[0], address[1], address[2], address[3]);
        return *this;
    }

    IpAddress& operator=(uint32_t address)
    {
        ip_addr_set_ip4_u32(&this->address, address);
        return *this;
    }

    IpAddress& operator=(const String& address)
    {
        fromString(address);
        return *this;
    }

    size_t printTo(Print& p) const;
};

inline String toString(IpAddress address)
{
    return address.toString();
}

// Making this extern saves 100's of bytes; each usage otherwise incurs 4 bytes of BSS
#define INADDR_NONE IpAddress()