Program Listing for File EndPoint.cpp

↰ Return to documentation for file (src/sockets/EndPoint.cpp)

//
// Created by Aleksey Timin on 12/10/19.
//

#include "EndPoint.h"
#include "BaseSocket.h"
#include "Platform.h"

#if defined(__unix__) || defined(__APPLE__)
#include <arpa/inet.h>
#elif defined(_WIN32) || defined(WIN32) || defined(_WIN64)
#include <winsock2.h>
#include <ws2tcpip.h>
#endif

#include <system_error>
#include <utility>

namespace eipScanner {
namespace sockets {

    EndPoint::EndPoint(struct sockaddr_in &addr)
        : _host(inet_ntoa(addr.sin_addr))
        , _port(htons(addr.sin_port))
        , _addr(addr) {
    }

    EndPoint::EndPoint(std::string host, int port)
        : _host(std::move(host))
        , _port(port)
        , _addr{} {

        _addr.sin_family = AF_INET;
        _addr.sin_port = htons(_port);
#if defined(__unix__) || defined(__APPLE__)
        if (inet_aton(_host.c_str(), &_addr.sin_addr) < 0) {
#elif defined(_WIN32) || defined(WIN32) || defined(_WIN64)
    if (inet_pton(AF_INET, _host.c_str(), &_addr.sin_addr.s_addr) < 0) {
#endif
            throw std::system_error(BaseSocket::getLastError(), BaseSocket::getErrorCategory());
        }
    }

    const std::string &EndPoint::getHost() const {
        return _host;
    }

    int EndPoint::getPort() const {
        return _port;
    }

    const sockaddr_in &EndPoint::getAddr() const {
        return _addr;
    }

    bool EndPoint::operator==(const EndPoint &rhs) const {
        return _host == rhs._host &&
               _port == rhs._port &&
               _addr.sin_addr.s_addr == rhs._addr.sin_addr.s_addr &&
               _addr.sin_port == rhs._addr.sin_port &&
               _addr.sin_family == rhs._addr.sin_family;
    }

    bool EndPoint::operator!=(const EndPoint &rhs) const {
        return !(rhs == *this);
    }

    bool EndPoint::operator< (const EndPoint& rhs) const {
        return _host < rhs._host
               && _port < rhs._port;
    }

    std::string EndPoint::toString() const {
        return _host + ":" + std::to_string(_port);
    }
}
}