MacAddress.h

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <initializer_list>
#include <iterator>
#include <ostream>
#include <cstdint>
#include <string>
#include <array>
 
 
namespace pcpp
{
 
    class MacAddress
    {
    public:
        MacAddress() = default;
 
        explicit MacAddress(const uint8_t addr[6])
        {
            std::copy(addr, addr + 6, m_Address.begin());
        }
 
        explicit MacAddress(const std::array<uint8_t, 6>& addr) : m_Address(addr)
        {}
 
        explicit MacAddress(const std::string& addr);
 
        template <typename T, typename = typename std::enable_if<std::is_convertible<T, std::string>::value>::type>
        MacAddress(const T& addr) : MacAddress(static_cast<std::string>(addr))
        {}
 
        MacAddress(uint8_t firstOctet, uint8_t secondOctet, uint8_t thirdOctet, uint8_t fourthOctet, uint8_t fifthOctet,
                   uint8_t sixthOctet)
            : m_Address{ firstOctet, secondOctet, thirdOctet, fourthOctet, fifthOctet, sixthOctet }
        {}
 
        MacAddress(std::initializer_list<uint8_t> octets)
        {
            if (octets.size() != m_Address.size())
            {
                throw std::invalid_argument("Invalid initializer list size, should be 6");
            }
            std::copy(octets.begin(), octets.end(), std::begin(m_Address));
        }
 
        bool operator==(const MacAddress& other) const
        {
            return m_Address == other.m_Address;
        }
 
        bool operator!=(const MacAddress& other) const
        {
            return !operator==(other);
        }
 
        MacAddress& operator=(std::initializer_list<uint8_t> octets)
        {
            if (octets.size() != sizeof(m_Address))
            {
                throw std::invalid_argument("Invalid initializer list size, should be 6");
            }
 
            std::copy(octets.begin(), octets.end(), m_Address.begin());
            return *this;
        }
 
        const uint8_t* getRawData() const
        {
            return m_Address.data();
        }
 
        std::string toString() const;
 
        std::array<uint8_t, 6> toByteArray() const
        {
            return m_Address;
        }
 
        void copyTo(uint8_t** arr) const
        {
            *arr = new uint8_t[m_Address.size()];
            std::copy(m_Address.begin(), m_Address.end(), *arr);
        }
 
        void copyTo(uint8_t arr[6]) const
        {
            std::copy(m_Address.begin(), m_Address.end(), arr);
        }
 
        static MacAddress Zero;
        static MacAddress Broadcast;
 
    private:
        std::array<uint8_t, 6> m_Address{};
    };
 
    inline std::ostream& operator<<(std::ostream& oss, const pcpp::MacAddress& macAddress)
    {
        oss << macAddress.toString();
        return oss;
    }
}  // namespace pcpp