DNS Query MFC based Application

// DnsDialog.cpp : implementation file
//

#include "stdafx.h"
#include "Dns.h"
#include "DnsDialog.h"
#include ".\dnsdialog.h"


// CDnsDialog dialog













IMPLEMENT_DYNAMIC(CDnsDialog, CDialog)
CDnsDialog::CDnsDialog(INITPRM *params, CWnd* pParent /*=NULL*/)
                : CDialog(CDnsDialog::IDD, pParent)
                , m_options(0)
                , m_ip4array(0)
{
        this->params = params;
}

CDnsDialog::~CDnsDialog()
{
        if (m_ip4array)
                delete[] m_ip4array;
}

void CDnsDialog::DoDataExchange(CDataExchange* pDX)
{
        CDialog::DoDataExchange(pDX);
        DDX_Text(pDX, IDC_EDIT_NAME, m_name);
        DDX_Control(pDX, IDC_LIST_OPTIONS, m_listoptions);
        DDX_Text(pDX, IDC_EDIT_SERVER, m_server);
        DDX_Control(pDX, IDC_EDIT_NAME, m_queryEdit);
        DDX_Control(pDX, IDC_COMBO_TYPE, m_typeCombo);
}


BEGIN_MESSAGE_MAP(CDnsDialog, CDialog)
END_MESSAGE_MAP()


// CDnsDialog message handlers

BOOL CDnsDialog::OnInitDialog()
{
        CDialog::OnInitDialog();

        // TODO:  Add extra initialization here

        for (int i = 0; i < 13; i++)
                m_listoptions.AddString(Options[i].title);
        m_listoptions.SetSel(0);

        RECT cr;
        m_typeCombo.GetClientRect(&cr);
        m_typeCombo.SetWindowPos(0, 0, 0, cr.right, cr.bottom + 12*10, SWP_NOMOVE | SWP_NOZORDER);

        for (int i = 0; i < 55; i++)
                m_typeCombo.AddString(Type[i].title);
        m_typeCombo.SetCurSel(params->type);

        m_server = params->servers;
        m_name = params->name;

        UpdateData(false);

        m_queryEdit.SetFocus();

        return false;  // return TRUE unless you set the focus to a control
        // EXCEPTION: OCX Property Pages should return FALSE
}

void CDnsDialog::OnOK()
{
        // TODO: Add your specialized code here and/or call the base class
        UpdateData();

        m_type = Type[m_typeCombo.GetCurSel()].type;
        m_options = 0;
        for (int i = 0; i < 13; i++) {
                if (m_listoptions.GetSel(i))
                        m_options |= Options[i].options;
        }


        params->name = m_name;
        params->type = m_typeCombo.GetCurSel();
        params->servers = m_server;


        if (m_server != "") {
                if (m_ip4array)
                        delete[] m_ip4array;

                in_addr saddr;

                if ((saddr.S_un.S_addr = inet_addr(m_server)) == INADDR_NONE) {
                        hostent *host = gethostbyname(m_server);
                        if (host) {
                                saddr = *((in_addr *)host->h_addr_list[0]);
                        } else {
                                CDialog::OnOK();
                                return;
                        }

                }

                m_ip4array = new IP4_ARRAY;
                m_ip4array->AddrCount = 1;
                m_ip4array->AddrArray[0] = saddr.S_un.S_addr;
                m_options |= DNS_QUERY_BYPASS_CACHE;
        }


        CDialog::OnOK();
}