MacAddress.h

Go to the documentation of this file.

#ifndef PCAPPP_MACADDRESS
#define PCAPPP_MACADDRESS

#include <stdint.h>
#include <string.h>
#include <string>

#if __cplusplus > 199711L || _MSC_VER >= 1800
#include <initializer_list>
#include <algorithm>
#include <iterator>
#include <ostream>
#endif


namespace pcpp
{

    class MacAddress
    {
    public:
        MacAddress() : m_IsValid(true) { memset(m_Address, 0, sizeof(m_Address)); }

        MacAddress(const uint8_t* addr) : m_IsValid(true) { memcpy(m_Address, addr, sizeof(m_Address)); }

        MacAddress(const char* addr) { init(addr); }

        MacAddress(const std::string& addr) { init(addr.c_str()); }

        inline MacAddress(uint8_t firstOctest, uint8_t secondOctet, uint8_t thirdOctet, uint8_t fourthOctet, uint8_t fifthOctet, uint8_t sixthOctet);

#if __cplusplus > 199711L || _MSC_VER >= 1800

        MacAddress(std::initializer_list<uint8_t> octets) : m_IsValid { octets.size() == sizeof(m_Address) }
        {
            if(m_IsValid)
            {
                #if _MSC_VER >= 1800
                std::copy(octets.begin(), octets.end(), stdext::checked_array_iterator<uint8_t*>(m_Address, 6));
                #else
                std::copy(octets.begin(), octets.end(), std::begin(m_Address));
                #endif
            }
            else
                memset(m_Address, 0, sizeof(m_Address));
        }
#endif

        bool operator==(const MacAddress& other) const { return memcmp(m_Address, other.m_Address, sizeof(m_Address)) == 0; }

        bool operator!=(const MacAddress& other) const { return !operator==(other); }

#if __cplusplus > 199711L || _MSC_VER >= 1800

        MacAddress& operator=(std::initializer_list<uint8_t> octets)
        {
            m_IsValid = (octets.size() == sizeof m_Address);
            if(m_IsValid)
            {
                #if _MSC_VER >= 1800
                std::copy(octets.begin(), octets.end(), stdext::checked_array_iterator<uint8_t*>(m_Address, sizeof(m_Address)));
                #else
                std::copy(octets.begin(), octets.end(), std::begin(m_Address));
                #endif
            }
            return *this;
        }
#endif

        const uint8_t* getRawData() const { return m_Address; }

        bool isValid() const { return m_IsValid; }

        std::string toString() const;

        void copyTo(uint8_t** arr) const
        {
            *arr = new uint8_t[sizeof(m_Address)];
            memcpy(*arr, m_Address, sizeof(m_Address));
        }

        void copyTo(uint8_t* arr) const { memcpy(arr, m_Address, sizeof(m_Address)); }

        static MacAddress Zero;

    private:
        uint8_t m_Address[6];
        bool m_IsValid;
        void init(const char* addr);
    };

    MacAddress::MacAddress(uint8_t firstOctest, uint8_t secondOctet, uint8_t thirdOctet, uint8_t fourthOctet, uint8_t fifthOctet, uint8_t sixthOctet)
        : m_IsValid(true)
    {
        m_Address[0] = firstOctest;
        m_Address[1] = secondOctet;
        m_Address[2] = thirdOctet;
        m_Address[3] = fourthOctet;
        m_Address[4] = fifthOctet;
        m_Address[5] = sixthOctet;
    }

} // namespace pcpp

inline std::ostream& operator<<(std::ostream& os, const pcpp::MacAddress& macAddress)
{
    os << macAddress.toString();
    return os;
}

#endif /* PCAPPP_MACADDRESS */