NetworkHelper.cpp

Go to the documentation of this file.

#include <NetworkHelper.h>

#ifdef HAVE_WINDOWS
#include <winsock2.h>
#include <ws2tcpip.h>
#include <windows.h>
#include <locale>
#include <codecvt>
#endif

#ifdef HAVE_LINUX
#include <ifaddrs.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#endif

#include <iostream>
#include <cstring>
using namespace std;

NetworkHelper::NetworkHelper() {
#ifdef HAVE_WINDOWS
    WORD wVersionRequested;
    WSADATA wsaData;
    
    wVersionRequested = MAKEWORD(2,2);
    if(WSAStartup(wVersionRequested, &wsaData) != 0) {
        cerr << "COULD NOT LOAD WINSOCK LIBRARY!" << endl;
    }
#endif
}

NetworkHelper::~NetworkHelper() {
#ifdef HAVE_WINDOWS
    WSACleanup();
#endif
}

std::list<std::string> NetworkHelper::getIpAddresses(bool allowIPv6) {
    list<string> out;
#ifdef HAVE_WINDOWS
    char ac[80];
    if (gethostname(ac, sizeof(ac)) == SOCKET_ERROR) {
        cerr << "Error " << WSAGetLastError() <<
                " when getting local host name." << endl;
        return out;
    }
    cout << "Host name is " << ac << "." << endl;

    PADDRINFOA addrInfo;

    if (0 != getaddrinfo(ac, nullptr, nullptr, &addrInfo)) {
        cerr << "Yow! Bad host lookup." << endl;
        return out;
    }
    
    while(addrInfo != nullptr) {
        switch(addrInfo->ai_family) {
            case AF_INET6:
                if(!allowIPv6) {
                    break;
                }
                /* fallthrough */
            case AF_INET:
            {
                WCHAR str[64];
                DWORD strlen = (DWORD)(sizeof(str) / sizeof(WCHAR));
                WSAAddressToStringW(addrInfo->ai_addr,
                                    (DWORD)addrInfo->ai_addrlen,
                                    NULL,
                                    str,
                                    &strlen);
                std::wstring_convert<std::codecvt_utf8<wchar_t>, wchar_t> converter;
                out.emplace_back(converter.to_bytes(str));
            }
        }
        addrInfo = addrInfo->ai_next;
    }
#endif
#ifdef HAVE_LINUX
    struct ifaddrs * ifAddrStruct=nullptr;
    struct ifaddrs * ifa=nullptr;
    void * tmpAddrPtr=nullptr;

    getifaddrs(&ifAddrStruct);

    for (ifa = ifAddrStruct; ifa != nullptr; ifa = ifa->ifa_next) {
        if (ifa ->ifa_addr->sa_family==AF_INET) { // check it is IP4
            // is a valid IP4 Address
            tmpAddrPtr=&((struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
            char addressBuffer[INET_ADDRSTRLEN];
            inet_ntop(AF_INET, tmpAddrPtr, addressBuffer, INET_ADDRSTRLEN);
            if(strcmp(addressBuffer, "127.0.0.1") == 0)
                continue;
            out.emplace_back(addressBuffer);
        } else if (ifa->ifa_addr->sa_family==AF_INET6 && allowIPv6) { // check it is IP6
            // is a valid IP6 Address
            tmpAddrPtr=&((struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
            char addressBuffer[INET6_ADDRSTRLEN];
            inet_ntop(AF_INET6, tmpAddrPtr, addressBuffer, INET6_ADDRSTRLEN);
            if(strcmp(addressBuffer, "::1") == 0)
                continue;
            out.emplace_back(addressBuffer);
        }
    }
    if (ifAddrStruct!=nullptr) freeifaddrs(ifAddrStruct);
#endif
    return out;
}