komodiatracertdns.cpp

一个dns解析程序的源代码

/*
 *  Copyright (c) 2000-2003 Barak Weichselbaum <[e-mail]>
 *  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * Contact info:
 * -------------
 *
 * Site:					http://www.komodia.com
 * Main contact:			[e-mail]
 * For custom projects, 
 * consulting, or other
 * paid services:			[e-mail]
 */

#include "stdafx.h"

#include "TraceRouteSocketReverseDNS.h"
#include "ScreenLog.h"

#include <conio.h>

void UsageErrorMessage(const char* pProgramName)
{
	printf("Usage:\n");
	printf("%s <DNSServer> <host>\n",pProgramName);
}

void GeneralErrorMessage(CTraceRouteSocketReverseDNS* pSocket,
						 const char* pError)
{
	//Print the error
	printf("%s\n",pError);

	//Terminate al;
	if (pSocket)
		delete pSocket;

	//Close the sockets
	CSpoofBase::ShutdownSockets();
}

//Socket pointer for map
CTraceRouteSocketReverseDNS* gSocket=NULL;

//Proc for the message map
BOOL ShouldStop()
{
	return gSocket->IsTraceDone() ||
		   kbhit();
}

int main(int argc, char* argv[])
{
	//Welcome message
	printf("KomodiaTracertDNS, Copyright 2000-2003 (c) Barak Weichselbaum! (http://www.komodia.com)\n\n");

	//First check parameters
	if (argc<3)
	{
		//Report it
		UsageErrorMessage(argv[0]);

		//Exit
		return 1;
	}

	//Our IP address
	std::string sIP;

	//Initalize data
	if (!CSpoofBase::InitializeSockets())
	{
		//Report it
		GeneralErrorMessage(NULL,"Sockets couldn't be initialized");

		//Exit
		return 2;
	}

	//Create a log
	CScreenLog aLog;
	CSpoofBase::SetLog(&aLog);

		//Check if it's an address
	if ((*argv[1]>='a' && *argv[1]<='z') || 
		(*argv[1]>='A' && *argv[1]<='Z'))
	{
		//Try to resolve the DNS
		IP aAddr;
		aAddr=CSpoofSocket::ResolveDNS(argv[1]);

		if (aAddr)
			//OK
			sIP=CSpoofSocket::LongToStdString(aAddr);
		else
		{
			//Invalid
			GeneralErrorMessage(NULL,"Invalid DNS!");

			//Exit
			return 3;
		}
	}
	else
	{
		//Check address validity
		if (strlen(argv[1])>15)
		{
			//Report it
			GeneralErrorMessage(NULL,"Invalid DNS!");

			//Exit
			return 3;
		}

		sIP=argv[1];
	
		//Check validity
		if (!CSpoofSocket::ValidAddress(sIP))
		{
			//Report it
			GeneralErrorMessage(NULL,"Invalid DNS!");

			//Exit
			return 4;
		}
	}

	//Our socket
	CTraceRouteSocketReverseDNS* pSocket;
	pSocket=new CTraceRouteSocketReverseDNS(sIP,
											FALSE);

	//Save it to the global
	gSocket=pSocket;

	//Check if it's an address
	if ((*argv[2]>='a' && *argv[2]<='z') || 
		(*argv[2]>='A' && *argv[2]<='Z'))
	{
		//Try to resolve the DNS
		IP aAddr;
		aAddr=pSocket->ResolveDNS(argv[2]);

		if (aAddr)
			//OK
			sIP=pSocket->LongToStdString(aAddr);
		else
		{
			//Invalid
			GeneralErrorMessage(pSocket,"Invalid host!");

			//Exit
			return 5;
		}
	}
	else
	{
		//Check address validity
		if (strlen(argv[2])>15)
		{
			//Report it
			GeneralErrorMessage(pSocket,"Invalid host!");

			//Exit
			return 6;
		}

		sIP=argv[2];
	
		//Check validity
		if (!pSocket->ValidAddress(sIP))
		{
			//Report it
			GeneralErrorMessage(pSocket,"Invalid host!");

			//Exit
			return 7;
		}
	}

	//Create and bind
	if (!pSocket->Create())
	{
		//Report it
		GeneralErrorMessage(pSocket,"Socket couldn't be created");

		//Exit
		return 8;
	}
	
	//Start the trace
	if (!((CTraceRouteSocket*)pSocket)->TraceRoute(sIP,
												   3000,
												   20))
	{
		//Report it
		GeneralErrorMessage(pSocket,"Failed tracing!");

		//Exit
		return 9;
	}

	printf("Tracing:%s\nPress any key to abort\n\n",sIP.c_str());

	//Run a message loop
	CAsyncSocket::SimpleMessageMap(ShouldStop,
								   TRUE,
								   1);

	//Is the trace done
	if (pSocket->IsTraceDone())
	{
		//Take the trace data
		CTraceRouteSocketReverseDNS::DNSTraceRouteVector aData;
		pSocket->TakeDNSData(aData);

		CTraceRouteSocketReverseDNS::DNSTraceRouteVector::const_iterator aIterator=aData.begin();

		//Iterate it
		while (aIterator!=aData.end())
		{
			//Is it a timeout
			if (aIterator->bTimeout)
				printf("Timeout, hop: %i",aIterator->ucTTL);
			else
				printf("Reply from: %s time=%i hop=%i dns: %s\n",
					   CSpoofBase::LongToString(aIterator->aReplyingAddress),
		     		   aIterator->dwElapsedTime,
					   aIterator->ucTTL,
					   aIterator->sReverseDNS.c_str());

			//Increase the iterator
			++aIterator;
		}
	}

	//Shutdown
	delete pSocket;

	//Release all
	CSpoofBase::ShutdownSockets();

	//Done
	return 0;
}