ipv6_address.h

/*
 * Copyright (c) 2017, Matias Fontanini
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 * 
 * * Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * * Redistributions in binary form must reproduce the above
 *   copyright notice, this list of conditions and the following disclaimer
 *   in the documentation and/or other materials provided with the
 *   distribution.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

#ifndef TINS_IPV6_ADDRESS
#define TINS_IPV6_ADDRESS

#include <string>
#include <iosfwd>
#include <functional>
#include <stdint.h>
#include <tins/cxxstd.h>
#include <tins/macros.h>

namespace Tins {

class TINS_API IPv6Address {
public:
    static const size_t address_size = 16;
    
    typedef uint8_t* iterator;
    
    typedef const uint8_t* const_iterator;

    static IPv6Address from_prefix_length(uint32_t prefix_length);

    IPv6Address();
    
    IPv6Address(const char* addr);
    
    IPv6Address(const std::string& addr);
    
    IPv6Address(const_iterator ptr);
    
    std::string to_string() const;
    
    iterator begin() {
        return address_;
    }
    
    const_iterator begin() const {
        return address_;
    }
    
    iterator end() {
        return address_ + address_size;
    }
    
    const_iterator end() const {
        return address_ + address_size;
    }
    
    bool operator==(const IPv6Address& rhs) const {
        return std::equal(begin(), end(), rhs.address_);
    }
    
    bool operator!=(const IPv6Address& rhs) const {
        return !(*this == rhs);
    }
    
    bool operator<(const IPv6Address& rhs) const {
        return std::lexicographical_compare(begin(), end(), rhs.begin(), rhs.end());
    }

    bool operator<=(const IPv6Address& rhs) const {
        return !operator>(rhs);
    }

    bool operator>(const IPv6Address& rhs) const {
        return std::lexicographical_compare(rhs.begin(), rhs.end(), begin(), end());
    }

    bool operator>=(const IPv6Address& rhs) const {
        return !operator<(rhs);
    }

    template<typename OutputIterator>
    OutputIterator copy(OutputIterator iter) const {
        return std::copy(begin(), end(), iter);
    }
    
    bool is_loopback() const;

    bool is_multicast() const;

    size_t size() const {
        return address_size;
    }
    
    TINS_API friend std::ostream& operator<<(std::ostream& os, const IPv6Address& addr);

    IPv6Address operator&(const IPv6Address& rhs) const;

    IPv6Address operator|(const IPv6Address& rhs) const;

    IPv6Address operator~() const ;

private:
    void init(const char* addr);

    uint8_t address_[address_size];
};

} // Tins

#if TINS_IS_CXX11
namespace std {

template<>
struct hash<Tins::IPv6Address> {
    // Implementation taken from boost.functional
    size_t operator()(const Tins::IPv6Address& addr) const {
        std::size_t output = Tins::IPv6Address::address_size;
        Tins::IPv6Address::const_iterator iter = addr.begin();
        for (; iter != addr.end(); ++iter) {
            output ^= *iter + 0x9e3779b9 + (output << 6) + (output >> 2);
        }
        return output;
    }
};

} // std

#endif // TINS_IS_CXX11

#endif // TINS_IPV6_ADDRESS