CWE-ou-minkata/MOULOpenSourceClientPlugin/Plasma20/Sources/Plasma/NucleusLib/pnNetDiag/pnNdDns.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
/*****************************************************************************
*
*   $/Plasma20/Sources/Plasma/NucleusLib/pnNetDiag/pnNdDns.cpp
*   
***/

#include "Pch.h"
#pragma hdrstop


/*****************************************************************************
*
*   Internal types
*
***/

struct DNSParam {
	NetDiag *				diag;
	FNetDiagDumpProc		dump;
	FNetDiagTestCallback	callback;
	void *					param;
	ENetProtocol			protocol;
	unsigned				srv;
};


/*****************************************************************************
*
*   Internal functions
*
***/

//============================================================================
static void LookupCallback (
    void *              param,
    const wchar         name[],
    unsigned            addrCount,
    const NetAddress    addrs[]
) {
	DNSParam * p = (DNSParam *)param;
	if (addrCount) {
		unsigned node = NetAddressGetNode(addrs[0]);
		p->diag->critsect.Enter();
		{
			p->diag->nodes[p->srv] = node;
		}
		p->diag->critsect.Leave();
		wchar nodeStr[64];
		NetAddressNodeToString(p->diag->nodes[p->srv], nodeStr, arrsize(nodeStr));
		p->dump(L"[DNS] Success. %s --> %s", name, nodeStr);
		p->callback(
			p->diag,
			p->protocol,
			kNetSuccess,
			p->param
		);
	}
	else {
		p->diag->critsect.Enter();
		{
			// if the hostname still matches, then clear the node
			if (p->diag->hosts[p->srv] && 0 == StrCmp(p->diag->hosts[p->srv], name))
				p->diag->nodes[p->srv] = 0;
		}
		p->diag->critsect.Leave();
		p->dump(L"[DNS] Failed to resolve hostname %s", name);
		p->callback(
			p->diag,
			p->protocol,
			kNetErrNameLookupFailed,
			p->param
		);
	}
	p->diag->DecRef("DNS");
	DEL(p);
}


/*****************************************************************************
*
*   Module functions
*
***/

//============================================================================
void DnsStartup () {
}

//============================================================================
void DnsShutdown () {
}


/*****************************************************************************
*
*   Exports
*
***/

//============================================================================
void NetDiagDns (
	NetDiag *				diag,
	ENetProtocol			protocol,
	FNetDiagDumpProc		dump,
	FNetDiagTestCallback	callback,
	void *					param
) {
	ASSERT(diag);
	ASSERT(dump);
	ASSERT(callback);
	
	unsigned srv = NetProtocolToSrv(protocol);
	if (srv == kNumDiagSrvs) {
		dump(L"[DNS] Unsupported protocol: %s", NetProtocolToString(protocol));
		callback(diag, protocol, kNetErrNotSupported, param);
		return;
	}

	wchar * host = nil;
	diag->critsect.Enter();
	{
		if (diag->hosts[srv])
			host = StrDup(diag->hosts[srv]);
	}
	diag->critsect.Leave();

	if (!host) {
		dump(L"[DNS] No hostname set for protocol: %s", NetProtocolToString(protocol));
		callback(diag, protocol, kNetSuccess, param);
		return;
	}
			
	diag->IncRef("DNS");
	dump(L"[DNS] Looking up %s...", host);

	DNSParam * dnsParam	= NEWZERO(DNSParam);
	dnsParam->diag		= diag;
	dnsParam->srv		= srv;
	dnsParam->protocol	= protocol;
	dnsParam->dump		= dump;
	dnsParam->callback	= callback;
	dnsParam->param		= param;

	AsyncCancelId cancelId;
	AsyncAddressLookupName(
		&cancelId,
		LookupCallback,
		host,
		0,
		dnsParam
	);
	FREE(host);
}