/*
 * DnsPtrRecordQuery.cpp 
 * Copyright (c) 2011 Antillia.com TOSHIYUKI ARAI. ALL RIGHTS RESERVED. 
 */


// SOL9
// 2011/1/2
// 2011/1/14 Updated
// 2011/1/18 Modified to use DnsRecordQueryT class

#include <sol/WSAInitializer.h>
#include <sol/dns/DnsRecordQueryT.h>
#include <sol/Locale.h>

namespace SOL {

class DnsPtrRecordQuery :public Object {
private:
    DnsRecordQueryT<DNS_PTR_DATA> dns;

private:
    TCHAR reversedIP[20];

public:
    /**
     * Constructor
     */
    DnsPtrRecordQuery()
        :dns(DNS_TYPE_PTR)
    {
    }

public:
    void start(const TCHAR* ip)
    {
        reverseIP(ip, reversedIP);
        int count = dns.query(reversedIP);

        //Display the query result(all DNS_PTR_DATA).
        _tprintf(_T("All DNS_PTR_DATA(count=%d)\n"), count);                
        dns.display();
        
        //Get the first DNS_PTR_DAT
        DNS_PTR_DATA data;
        dns.getFirst(data);

        //Display a structure data of DNS_PTR_DATA 
        _tprintf(_T("\nFirst DNS_PTR_DATA\n"));
        DnsData dnsData;
        dnsData.display(data);
    }

private:
    void reverseIP(__in const TCHAR* ip, __out TCHAR* reversedIP)
    {
        TCHAR* sep = _T(".");
        TCHAR  array[4][4];
        int i = 0;
        TCHAR* token = strtok((TCHAR*)ip, sep);
        while (token != NULL) {
            _stprintf(array[i], _T("%s"), token);
            token = strtok((TCHAR*)NULL, sep);
            i++;
        }
        _stprintf(reversedIP, _T("%s.%s.%s.%s.%s"), array[3], array[2], array[1], array[0],
            _T("IN-ADDR.ARPA"));
    }

};

}



void _tmain(int argc, TCHAR** argv)
{
    if (argc !=2) {
        _tprintf(_T("Usage:%s ipaddress\n"), argv[0]);
        _tprintf(_T("Ex: %s 216.239.38.10\n"), argv[0]);
        return;
    }
    WSAInitializer initializer;
    Locale locale;

    try {
        DnsPtrRecordQuery recordQuery;

        recordQuery.start(argv[1]);

    } catch (Exception& ex) {
        ex.printf();
    } catch (...) {
        _tprintf(_T("Exception\n"));
    }
}