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


// SOL9
// 2010/09/01
// 2010/12/07 Modified to have a member variable of IN_ADDR.

#pragma once

#include <sol/Object.h>
#include <sol/Exception.h>
#include <sol/String.h>
#include <sol/StringT.h>

#include <mstcpip.h>

namespace SOL {

/*
LPTSTR NTAPI RtlIpv4AddressToString(
  __in   const IN_ADDR *Addr,
  __out  LPTSTR S
);


LONG NTAPI RtlIpv4AddressToStringEx(
  __in     const IN_ADDR *Address,
  __in     USHORT Port,
  __out    LPTSTR AddressString,
  __inout  PULONG AddressStringLength
);
*/

typedef char* (NTAPI *_Ipv4AddressToString)(__in const IN_ADDR *address, __out TCHAR* addressString);
typedef LONG  (NTAPI *_Ipv4AddressToStringEx)(__in const IN_ADDR *address, __in USHORT port, 
                                              __out TCHAR* addressString, __inout ULONG* addressStringLength);

class IPv4Address :public Object {
private:
    IN_ADDR inaddr;

private:
    // For Ntdll.dll module
    HMODULE  hNtdll;

private:
    //Utility functions to convert an IPV4 Address (IN_ADDR structure) to a string.
    _Ipv4AddressToString   ipv4ToString;
    _Ipv4AddressToStringEx ipv4ToStringEx;

public:
    /**
     * Constructor
     */
    //2010/12/07: Modified to be able to take a parameter of a pointer to IN_ADDR structure.
    IPv4Address(IN_ADDR* addr=NULL)
        :hNtdll(NULL),
        ipv4ToString(NULL),
        ipv4ToStringEx(NULL)
    {
        memset(&inaddr, 0, sizeof(inaddr));

        if (addr) {
            memcpy(&inaddr, addr, sizeof(inaddr));
        }

        hNtdll = LoadLibrary(_T("Ntdll.dll"));

        if (hNtdll) {
#ifdef UNICODE
            ipv4ToString   = (_Ipv4AddressToString)GetProcAddress(hNtdll, "RtlIpv4AddressToStringW");
            ipv4ToStringEx = (_Ipv4AddressToStringEx)GetProcAddress(hNtdll, "RtlIpv4AddressToStringExW");
#else
            ipv4ToString   = (_Ipv4AddressToString)GetProcAddress(hNtdll, "RtlIpv4AddressToStringA");
            ipv4ToStringEx = (_Ipv4AddressToStringEx)GetProcAddress(hNtdll, "RtlIpv4AddressToStringExA");
#endif
            if (ipv4ToString == NULL) {
                throw Exception("Failed to get a procAddress for Ipv4AddressToStringA");
            }
            if (ipv4ToStringEx == NULL) {
                throw Exception("Failed to get a procAddress for Ipv4AddressToStringExA");
            }

        } else {
            //printf("Failed to get ipV6ToString\n");
            throw Exception("Failed to load Ntdll.dll");
        }
    }

public:
    /**
     * Destructor
     */
    ~IPv4Address()
    {
        if (hNtdll) {
            FreeLibrary(hNtdll);
            hNtdll = NULL;
        }
    }

public:
    void getAddress(IN_ADDR& addr)
    {
        memcpy(&addr, &inaddr, sizeof(inaddr));
    }

public:
    bool toString(__out TCHAR* addressString)
    {
        // Convert this->inaddr of IN_ADDR to a string. 
        return toString(&inaddr, addressString);
    }

public:
    bool toString(__out StringT<TCHAR>& addressString)
    {
        // Convert this->inaddr of IN_ADDR to a string. 
        return toString(&inaddr, addressString);
    }

public:
    bool toStringEx(__in USHORT port, __out TCHAR* addressString, __inout ULONG* addressStringLength)
    {
        // Convert this->inaddr of IN_ADDR to a string. 
        return toStringEx(&inaddr, port, addressString, addressStringLength);
    }


public:
    bool toString(__in const IN_ADDR* address, __out StringT<TCHAR>& addressString)
    {
        bool rc = false;
    
        if (ipv4ToString) {
            TCHAR buffer[128];
            memset(buffer, (TCHAR)0, SizeOf(buffer));
            if ((*ipv4ToString)(address, buffer)) {
                rc = true;
                addressString = buffer;
            }
        }
        return rc;    
    }

public:
    bool toString(__in const IN_ADDR& address, __out StringT<TCHAR>& addressString)
    {
        bool rc = false;
    
        if (ipv4ToString) {
            TCHAR buffer[128];
            memset(buffer, (TCHAR)0, SizeOf(buffer));
            if ((*ipv4ToString)(&address, buffer)) {
                rc = true;
                addressString = buffer;
            }
        }
        return rc;    
    }

public:
    bool toString(__in const DWORD addr, __out StringT<TCHAR>& addressString)
    {
        bool rc = false;
    
        if (ipv4ToString) {
            IN_ADDR address;
            address.s_addr = addr;
            TCHAR buffer[128];
            memset(buffer, (TCHAR)0, SizeOf(buffer));
            if ((*ipv4ToString)(&address, buffer)) {
                rc = true;
                addressString = buffer;
            }
        }
        return rc;    
    }
public:
    bool toString(__in const IN_ADDR* address, __out TCHAR* addressString)
    {
        bool rc = false;
    
        if (ipv4ToString) {
            if ((*ipv4ToString)(address, addressString)) {
                rc = true;
            }
        }
        return rc;    
    }

public:
    bool toStringEx(__in const IN_ADDR *address, __in USHORT port, __out TCHAR* addressString, __inout ULONG* addressStringLength)
    {
        bool rc = false;
    
        if (ipv4ToStringEx) {
            if ((*ipv4ToStringEx)(address, port, addressString, addressStringLength)) {
                rc = true;
            }
        }
        return rc;    
    }

};

}