/*
 * NetworkInterfaceInfo.h 
 * Copyright (c) 2011 Antillia.com TOSHIYUKI ARAI. ALL RIGHTS RESERVED. 
 */


// SOL++2000
// 2008/07/15

// 2008/11/10 Modified a bug in the method isConnectedToInternet.

#pragma once

#include <sol\Object.h>
#include <sol\String.h>

//#pragma comment("lib", "ws2_32.lib")
//#pragma comment("lib", "iphlpapi.lib")

#ifndef _SOL_NETWORKINTERFACEINFO
#define _SOL_NETWORKINTERFACEINFO

namespace SOL {

class NetworkInterfaceInfo :public Object {

private:
    INTERFACE_INFO    list[40];
    int        listNum;            //The number of size of list.
    int        upCount;            //The number of UP state.

public:
    /**
     * Constructor
     */
    NetworkInterfaceInfo()
    :listNum(0),
    upCount(0)
    {
    }

public:
    /**
     * Listup all network interface information using WSAIoctl(sd, SIO_GET_INTERFACE_LIST...)
     */
    int listup() 
    {
        clear();

        SOCKET sd = WSASocket(AF_INET, SOCK_DGRAM, 0, 0, 0, 0);
        if (sd == SOCKET_ERROR) {
            throw SOCKET_ERROR;
        }

        unsigned long nBytes =0;

        if (WSAIoctl(sd, SIO_GET_INTERFACE_LIST, 0, 0, &list,
            sizeof(list), &nBytes, 0, 0) == SOCKET_ERROR) {
            throw SOCKET_ERROR;
        }

        closesocket(sd);
        
        this->listNum = nBytes/sizeof(INTERFACE_INFO);

        this->upCount = getUpCount(listNum);
        
        return this->upCount;

    }

private:
    /**
     * Compute the count of Up network interfaces. 
     */
    int getUpCount(int listNum)
    {
        int  n = 0;    //Get the number of up-interfaces,not including a LOOPBACK.
        for (int i = 0; i <listNum; i++) {
            u_long nFlags = list[i].iiFlags;
            
            if (nFlags & IFF_LOOPBACK) {
                continue;    //Skip loopback
            }
            
            if (nFlags & IFF_UP) {// "up";
                n++;
            }
        }
        return n;
    }
 
public:
    /**
     * Destructor
     */
    ~NetworkInterfaceInfo() {
        clear();
    }


private: 
    void clear() {
        memset(list, 0, sizeof(list));

        this->upCount = 0;
        this->listNum = 0;
    }


public:
    /**
     * Get the count of UP network interfaces, not including a loopback(127....).
     */
    int getUpCount() {
        return this->upCount;
    }


public:
    /**
     * Get the IP address of UP network interfaces, not including a loopback(127....).
     *
     */
    bool getUpIPAddress(int index, __out StringT<char>& address) 
    {
        bool rc = false;

        address = "";

        if (index <0 || index>=upCount) {
            // should throw OutOfRangeException.
            return rc;
        }

        int n = 0;
        for (int i = 0; i <listNum; i++) {
            u_long nFlags = list[i].iiFlags;
            
            if (nFlags & IFF_LOOPBACK) {
                continue;    //Skip loopback
            }
            
            if (nFlags & IFF_UP) {
                if (n == index) {
                    sockaddr_in *addr = (sockaddr_in *)&(list[i].iiAddress);
                    address = inet_ntoa(addr->sin_addr);
                    rc = true;
                    break;
                }
                n++;
            }        
        }
        return rc;
    }

public:
    /**
     * Get the IP address of UP network interfaces, not including a loopback(127....).
     *
     */
    //2009/11/01
    bool getUpIPAddress(int index, __out String& address) 
    {
        bool rc = false;

        address = "";

        if (index <0 || index>=upCount) {
            // should throw OutOfRangeException.
            return rc;
        }

        int n = 0;
        for (int i = 0; i <listNum; i++) {
            u_long nFlags = list[i].iiFlags;
            
            if (nFlags & IFF_LOOPBACK) {
                continue;    //Skip loopback
            }
            
            if (nFlags & IFF_UP) {
                if (n == index) {
                    sockaddr_in *addr = (sockaddr_in *)&(list[i].iiAddress);
                    address = inet_ntoa(addr->sin_addr);
                    rc = true;
                    break;
                }
                n++;
            }        
        }
        return rc;
    }

public:
    /**
     * Get UP-network interfaces info, not including a loopback(127....).
     *
     */
    bool getUpInterface(int index, INTERFACE_INFO& info) 
    {
        bool rc = false;


        if (index <0 || index>=upCount) {
            // should throw OutOfRangeException.
            return rc;
        }

        int n = 0;
        for (int i = 0; i <listNum; i++) {
            u_long nFlags = list[i].iiFlags;
            
            if (nFlags & IFF_LOOPBACK) {
                continue;    //Skip loopback
            }
            
            if (nFlags & IFF_UP) {
                if (n == index) {
                    memcpy(&info, &list[i], sizeof(INTERFACE_INFO));
                    rc = true;
                    break;
                }
                n++;
            }        
        }
        return rc;
    }

public:
    /**
     * Is connected to Internet?
     */
    //2008/11/10 Modified
    bool isConnectedToInternet() 
    {
        bool rc = false;
        static const char* CLASS_C = "192.";

        //2008/11/10 Modifed (upCount > 1) 
        if (upCount > 0) {

            for (int i = 0; i< upCount; i++) {
                StringT<char> address = "";
                getUpIPAddress(i, address);
                if (!address.startsWith(CLASS_C)) {
                    rc = true;
                    break;
                }
            }
        }
        return rc;
    }

public:
    /**
     * Is connected to local private address(Class C private) only,
     * not including a loopback?
     * 
     */
    bool isClassCPrivateAddressOnly() {
        static const char* CLASS_C_PRIVATE = "192.168.";
        
        bool rc = false;
        if (upCount == 1) {
            StringT<char> address = "";
            getUpIPAddress(0, address);
            if (address.startsWith(CLASS_C_PRIVATE)) {
                rc = true;
            }
        }
        return rc;
    }

public:
    void dump()
    {
        for (int i = 0; i <listNum; i++) {
            u_long nFlags = list[i].iiFlags;
        
            sockaddr_in *addr = (sockaddr_in *)&(list[i].iiAddress);
            char* address = inet_ntoa(addr->sin_addr);
            char* state = "Unknown";
            if (nFlags & IFF_UP) {
                state = "Up  ";
            } 
            printf("%s %s\n", state, address);
        }

    }

};

}

#endif