IPAddress.h

/*
╔══════════════════════════════════════════════════════════════════════════════╗
║                       SIMPLY POWERFUL TOOLKIT (SPTK)                         ║
╟──────────────────────────────────────────────────────────────────────────────╢
║  copyright            © 1999-2022 Alexey Parshin. All rights reserved.       ║
║  email                alexeyp@gmail.com                                      ║
╚══════════════════════════════════════════════════════════════════════════════╝
┌──────────────────────────────────────────────────────────────────────────────┐
│   This library is free software; you can redistribute it and/or modify it    │
│   under the terms of the GNU Library General Public License as published by  │
│   the Free Software Foundation; either version 2 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 Library   │
│   General Public License for more details.                                   │
│                                                                              │
│   You should have received a copy of the GNU Library General Public License  │
│   along with this library; if not, write to the Free Software Foundation,    │
│   Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.               │
│                                                                              │
│   Please report all bugs and problems to alexeyp@gmail.com.                  │
└──────────────────────────────────────────────────────────────────────────────┘
*/
 
#pragma once
 
#include <sptk5/sptk.h>
 
#ifndef _WIN32
#include <arpa/inet.h>
#include <netdb.h>
#include <netinet/in.h>
#include <sptk5/Strings.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/un.h>
#include <unistd.h>
 
using SOCKET = int;
using SOCKET_ADDRESS_FAMILY = sa_family_t;
 
#ifdef __APPLE__
using socklen_t = int;
#endif
 
#define INVALID_SOCKET -1
 
#else
#include <winsock2.h>
 
#include <windows.h>
using socklen_t = int;
using SOCKET_ADDRESS_FAMILY = unsigned short;
#endif
 
namespace sptk {
 
class SP_EXPORT IPAddress
{
    union
    {
        sockaddr_in ipv4;
        sockaddr_in6 ipv6;
        sockaddr generic;
    } m_address;
 
    String m_addressStr;
 
public:
    IPAddress();
 
    explicit IPAddress(const sockaddr& address);
 
    IPAddress(const IPAddress& other);
 
    IPAddress& operator=(const IPAddress& other);
 
    const sockaddr* address() const
    {
        return &m_address.generic;
    }
 
    size_t length() const
    {
        return addressLength(m_address.generic);
    }
 
    const String& toString() const
    {
        return m_addressStr;
    }
 
    static size_t addressLength(const sockaddr& address)
    {
        if (address.sa_family == AF_INET)
            return sizeof(sockaddr_in);
        return sizeof(sockaddr_in6);
    }
};
 
} // namespace sptk