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


// SOL9
// 2010/09/01
// 2011/01/18 Fixed bugs on toAddress methods.

#pragma once

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

#include <mstcpip.h>

namespace SOL {

/*
typedef struct in6_addr {
  union {
    u_char  Byte[16];
    u_short Word[8];
  } u;
} IN6_ADDR, *PIN6_ADDR, FAR *LPIN6_ADDR;

LONG NTAPI RtlIpv6StringToAddress(
  __in   PCTSTR S,
  __out  PCTSTR *Terminator,
  __out  IN6_ADDR *Addr
);

LONG NTAPI RtlIpv6StringToAddressEx(
  __in   PCTSTR AddressString,
  __out  IN6_ADDR *Address,
  __out  PULONG ScopeId,
  __out  PUSHORT Port
);

*/

typedef LONG (NTAPI *_Ipv6StringToAddress)(__in const TCHAR* addressString, __out TCHAR** terminator, __out IN6_ADDR *address);
typedef LONG (NTAPI *_Ipv6StringToAddressEx)(__in const TCHAR* addressString, __out IN6_ADDR *address, __out ULONG* scopeId, __out USHORT* port);

class IPv6String :public Object {

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

private:
    //Utility functions to convert a string to IPV6 Address (IN6_ADDR structure)
    _Ipv6StringToAddress   ipv6ToAddress;
    _Ipv6StringToAddressEx ipv6ToAddressEx;

public:
    /**
     * Constructor
     */
    IPv6String()
        :hNtdll(NULL),
        ipv6ToAddress(NULL),
        ipv6ToAddressEx(NULL)
    {
        hNtdll = LoadLibrary(_T("Ntdll.dll"));

        if (hNtdll) {
#ifdef UNICODE
            ipv6ToAddress   = (_Ipv6StringToAddress)GetProcAddress(hNtdll, "RtlIpv6StringToAddressW");
            ipv6ToAddressEx = (_Ipv6StringToAddressEx)GetProcAddress(hNtdll, "RtlIpv6StringToAddressExW");
#else
            ipv6ToAddress   = (_Ipv6StringToAddress)GetProcAddress(hNtdll, "RtlIpv6StringToAddressA");
            ipv6ToAddressEx = (_Ipv6StringToAddressEx)GetProcAddress(hNtdll, "RtlIpv6StringToAddressExA");
#endif
            if (ipv6ToAddress == NULL) {
                throw Exception("Failed to get a procAddress for Ipv6StringToAddressA");
            }
            if (ipv6ToAddressEx == NULL) {
                throw Exception("Failed to get a procAddress for Ipv6StringToAddressExA");
            }

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

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

public:
    bool toAddress(__in const TCHAR* addressString, __out TCHAR** terminator, __out IN6_ADDR* address)
    {
        bool rc = false;
        //2011/1/18
        if (ipv6ToAddress) {
            if ((*ipv6ToAddress)(addressString, terminator, address) ==NO_ERROR) {
                rc = true;
            }
        }
        return rc;    
    }

public:
    bool toAddressEx(__in const TCHAR* addressString, __out IN6_ADDR *address, __out ULONG* scopeId, __out USHORT* port)
    {
        bool rc = false;
        //2011/1/18
        if (ipv6ToAddressEx) {
            if ((*ipv6ToAddressEx)(addressString, address, scopeId, port) == NO_ERROR) {
                rc = true;
            }
        }
        return rc;    
    }

};

}