Host.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 <cstring>
#include <mutex>
#include <sptk5/Strings.h>
#include <sptk5/threads/Locks.h>
#include <sstream>
 
#ifndef _WIN32
 
#include <netinet/in.h>
 
#else
#include <WS2tcpip.h>
#include <winsock2.h>
#endif
 
namespace sptk {
 
class SP_EXPORT Host
{
    mutable SharedMutex m_mutex;                            
    String m_hostname;                                      
    uint16_t m_port {0};                                    
    std::array<uint8_t, sizeof(sockaddr_in6)> m_address {}; 
 
    sockaddr& any()
    {
        return *(sockaddr*) m_address.data();
    }
 
    const sockaddr& any() const
    {
        return *(const sockaddr*) m_address.data();
    }
 
    sockaddr_in& ip_v4()
    {
        return *(sockaddr_in*) m_address.data();
    }
 
    const sockaddr_in& ip_v4() const
    {
        return *(const sockaddr_in*) m_address.data();
    }
 
    sockaddr_in6& ip_v6()
    {
        return *(sockaddr_in6*) m_address.data();
    }
 
    const sockaddr_in6& ip_v6() const
    {
        return *(const sockaddr_in6*) m_address.data();
    }
 
    void getHostAddress();
 
    void setPort(uint16_t p);
 
public:
    Host() noexcept;
 
    Host(const String& hostname, uint16_t port);
 
    explicit Host(const String& hostAndPort);
 
    explicit Host(const sockaddr_in* addressAndPort);
 
    explicit Host(const sockaddr_in6* addressAndPort);
 
    Host(const Host& other);
 
    Host(Host&& other) noexcept;
 
    virtual ~Host() = default;
 
    Host& operator=(const Host& other);
 
    Host& operator=(Host&& other) noexcept;
 
    bool operator==(const Host& other) const;
 
    bool operator!=(const Host& other) const;
 
    const String& hostname() const
    {
        SharedLock(m_mutex);
        return m_hostname;
    }
 
    void port(uint16_t p)
    {
        setPort(p);
    }
 
    uint16_t port() const
    {
        SharedLock(m_mutex);
        return m_port;
    }
 
    String toString(bool forceAddress = false) const;
 
    void getAddress(sockaddr_in& address) const
    {
        SharedLock(m_mutex);
        memcpy(&address, &m_address, sizeof(address));
    }
 
    void getAddress(sockaddr_in6& address) const
    {
        SharedLock(m_mutex);
        memcpy(&address, &m_address, sizeof(address));
    }
 
    void setHostNameFromAddress(socklen_t addressLen);
};
 
using SHost = std::shared_ptr<Host>;
 
class SP_EXPORT HostCompare
{
public:
    bool operator()(const Host& s1, const Host& s2) const
    {
#ifdef WIN32
        return stricmp(s1.toString(true).c_str(), s2.toString(true).c_str()) > 0;
#else
        return strcasecmp(s1.toString(true).c_str(), s2.toString(true).c_str()) > 0;
#endif
    }
};
 
 
} // namespace sptk