控制台彩色PING工具ping2发布

REM	UNICODE COMPILATION:
	G++ ping2.c -lWs2_32 -lIphlpapi -lUser32 -D _UNICODE -D UNICODE -municode -O2 -static
	CL  ping2.c /O2 /Oy- /D "WIN32" /D "NDEBUG" /D "_CONSOLE" /D "_UNICODE" /D "UNICODE" /MD
REM	ANSI COMPILATION:
	G++ ping2.c -lWs2_32 -lIphlpapi -lUser32 -O2 -static
	CL  ping2.c /O2 /Oy- /D "WIN32" /D "NDEBUG" /D "_CONSOLE" /MDCOPY

/*
	CONSOLE PING TOOL, COPYRIGHT@2017~2019 BY LEO, VERSION 1.0
	PING2.EXE
	LINK WS2_32 IPHLPAPI USER32
*/
 
#include <stdio.h>
#include <winsock.h>
#include <iphlpapi.h>
#include <icmpapi.h>
#include <tchar.h>
#include <ctype.h>
#include <math.h>
 
#if defined _MSC_VER
#pragma comment(lib,"Ws2_32.lib")
#pragma comment(lib,"Iphlpapi.lib")
#pragma comment(lib,"User32.lib")
#endif
 
#if defined(_UNICODE) || defined(UNICODE)
#define TCHARFORMAT WCHAR
#else
#define TCHARFORMAT CHAR
#endif
 
#define DEFAULT_SEND_SIZE           32
#define DEFAULT_COUNT               4
#define DEFAULT_TTL                 55
#define DEFAULT_TOS                 0
#define DEFAULT_TIMEOUT             1000
#define DEFAULT_TIMEDELAY           1000
 
//定义帮助说明
#define HELP_INFORMATION _T("\
ping2 v1.0 - Console Colour Ping Tool - Copyright (C) 2017-2019 by Leo\n\
Usage: ping2 [host|IP] -t -f -a -n[count] -i[TTL] -v[TOS] -w[time out] -l[size] -d[delay]\n\
\n\
General options:\n\
  -t  Ping the specifed host until interrupted\n\
  -f  Don't fragment\n\
  -a  As IP address\n\
  -n  Send count\n\
  -i  Time to live\n\
  -v  Type of service\n\
  -w  Timeout (in milliseconds)\n\
  -l  Send buffer size\n\
  -d  Delayed time\n\
  -h  Show help information\n\
\n\
Official website:\n\
      http://www.bathome.net/thread-44179-1-1.html\n")
 
#define STR_HEAD _T("\
==============WELCOME YOU TO USE THE COLOR PING TOOL=============")
 
#define STR_LINE _T("\
=================================================================")
 
//开关解析宏名
#define _OPT_TEOF -1
#define _OPT_TILL -2
#define _OPT_TERR -3
 
//开关解析变量
INT OPTIND=1, OPTOPT, UNOPTIND=-1;
TCHAR* OPTARG;
 
//开关解析模块
INT _tgetopt(INT nargc, TCHAR* nargv[], TCHAR* ostr)
{
	static TCHAR* place = (TCHAR*)_T("");
	static TCHAR* lastostr = NULL;
	register TCHAR* oli;
 
	if(ostr!=lastostr)
	{
		lastostr=ostr;
		place=(TCHAR*)_T("");
	}
 
	if(!*place)
	{
		if(
		    (OPTIND>=nargc)                           ||
		    (*(place=nargv[OPTIND]) !=(TCHAR)_T('-')) ||
		    (!*(++place))
		)
		{
			if(*place !=(TCHAR)_T('-') && OPTIND <nargc)
			{
				place =(TCHAR*)_T("");
				if(UNOPTIND == -1)
				{
					UNOPTIND = OPTIND++;
					return _OPT_TILL;
				}
				else
				{
					return _OPT_TERR;
				}
			}
 
			place=(TCHAR*)_T("");
			return _OPT_TEOF;
		}
		if (*place == (TCHAR)_T('-') && *(place+1) == (TCHAR)_T('\0'))
		{
			++OPTIND;
			return _OPT_TEOF;
		}
	}
 
	if (
	    (OPTOPT=*place++) == (TCHAR)_T(':') ||
	    !(oli=(TCHAR*)_tcschr((TCHARFORMAT*)ostr, (TCHAR)OPTOPT))
	)
	{
		if(!*place)
		{
			++OPTIND;
		}
	}
 
	if (oli != NULL && *(++oli) !=(TCHAR)_T(':'))
	{
		OPTARG=NULL;
		if(!*place)
		{
			++OPTIND;
		}
	}
	else
	{
		if(*place)
		{
			OPTARG=place;
		}
		else if(nargc <= ++OPTIND)
		{
			place=(TCHAR*)_T("");
		}
		else
		{
			OPTARG=nargv[OPTIND];
		}
		place=(TCHAR*)_T("");
		++OPTIND;
	}
	return OPTOPT;
}
 
//彩显函数
INT ColorTcs(HANDLE handle_out, TCHAR* tcsTEXT1, TCHAR* tcsTEXT2, INT tcsCOLOR1, INT tcsCOLOR2)
{
	SetConsoleTextAttribute(handle_out, tcsCOLOR1);  //项色
	_ftprintf(stdout, _T("%s"), tcsTEXT1);
	SetConsoleTextAttribute(handle_out, tcsCOLOR2);  //文色
	_ftprintf(stdout, _T("%s\n"), tcsTEXT2);
	return 0;
}
 
//转 TCHAR* 函数
TCHAR* toTCHAR(VOID* csDATA, BOOL wcsMODE)
{
#if defined(_UNICODE) || defined(UNICODE)
 
	if(wcsMODE == TRUE)
	{
		return (TCHAR*)csDATA;
	}
	else
	{
		INT wLEN = MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, (CHAR*)csDATA, -1, NULL, 0);
		WCHAR* wsDATA = new WCHAR[wLEN + 1];
		MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, (CHAR*)csDATA, -1, wsDATA, wLEN);
		wsDATA[wLEN]=L'\0';
		return (TCHAR*)wsDATA;
	}
 
#else
 
	if(wcsMODE == TRUE)
	{
		INT aLEN = WideCharToMultiByte(CP_ACP, NULL, (WCHAR*)csDATA, -1, NULL, 0, NULL, NULL);
		CHAR* asDATA = new CHAR[aLEN + 1];
		WideCharToMultiByte(CP_ACP, NULL, (WCHAR*)csDATA, -1, asDATA, aLEN, NULL, NULL);
		asDATA[aLEN] = '\0';
		return (TCHAR*)asDATA;
	}
	else
	{
		return (TCHAR*)csDATA;
	}
 
#endif
}
 
//PING核心函数
INT SendPingByUIP(HANDLE handle_out, ULONG uIPADRR, BYTE opeFLAG, INT opeCOUNT, INT opeTTL, INT opeTOS, INT opeTIMEOUT, INT opeSENDSIZE, INT opeTIMEDELAY)
{
	HANDLE hicmpFILE = IcmpCreateFile();
	if (hicmpFILE == INVALID_HANDLE_VALUE)
	{
		//创建包文失败
		_ftprintf(stderr, _T("The ICMP header creats failed\n"));
		exit(1);
	}
 
	//填充包文头
	LPVOID sendDATA  = (LPVOID)(new BYTE[opeSENDSIZE]);
	memset(sendDATA, 'E', opeSENDSIZE);
 
	DWORD  replySIZE = sizeof(ICMP_ECHO_REPLY) + opeSENDSIZE;
	LPVOID replyBUFF = (LPVOID)(new BYTE[replySIZE]);
 
	//设置 IP_OPTION_INFORMATION 参数
	PIP_OPTION_INFORMATION sendOPTS = new IP_OPTION_INFORMATION;
	sendOPTS->OptionsData = 0;
	sendOPTS->OptionsSize = 0;
	sendOPTS->Ttl         = opeTTL;
	sendOPTS->Tos         = opeTOS;
	sendOPTS->Flags       = (opeFLAG|0x02) ?IP_FLAG_DF :0;
 
	//创建一个地址容器
	PIN_ADDR replyADDR = new IN_ADDR;
 
#if defined(_UNICODE) || defined(UNICODE)
 
	WCHAR* replyADDRW=new WCHAR[_MAX_PATH];
 
#endif
 
	INT succeedICMP=0, failedICMP=0, minRTT=-1, maxRTT=0;
	DOUBLE totalRTT=0.0f, totalSquareRTT=0.0f;
 
	if(opeFLAG & 0x01){opeCOUNT = 65535;}
 
	for(INT i=0; i<opeCOUNT; i++)
	{
		//发送包文
		if(IcmpSendEcho(hicmpFILE, uIPADRR, sendDATA, opeSENDSIZE, sendOPTS, replyBUFF, replySIZE, opeTIMEOUT) !=0)
		{
			PICMP_ECHO_REPLY pechoREPLY =(PICMP_ECHO_REPLY)replyBUFF;
			if(pechoREPLY->Status == IP_SUCCESS)
			{
				replyADDR->S_un.S_addr = pechoREPLY->Address;
				CHAR* replyADDRA=inet_ntoa(*replyADDR);
 
#if defined(_UNICODE) || defined(UNICODE)
 
				INT rLEN=MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, replyADDRA, -1, NULL, 0);
				MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, replyADDRA, -1, replyADDRW, rLEN);
				replyADDRW[rLEN]='\0';
				_ftprintf(stdout, _T("      REPLY FROM %s: BYTES=%u, TIME=%2ldms, TTL=%u\n"), (TCHARFORMAT*)replyADDRW, pechoREPLY->DataSize, pechoREPLY->RoundTripTime, pechoREPLY->Options.Ttl);
 
#else
 
				_ftprintf(stdout, _T("      REPLY FROM %s: BYTES=%u, TIME=%2ldms, TTL=%u\n"), (TCHARFORMAT*)replyADDRA, pechoREPLY->DataSize, pechoREPLY->RoundTripTime, pechoREPLY->Options.Ttl);
 
#endif
 
				//每PING间隔
				if(i+1 < opeCOUNT)
				{
					Sleep(opeTIMEDELAY);
				}
 
				//发包计数器
				succeedICMP ++;
 
				//包统计模块
				totalRTT += (DOUBLE)pechoREPLY->RoundTripTime;
				totalSquareRTT += pow((DOUBLE)pechoREPLY->RoundTripTime, 2);
				if(minRTT==-1)
				{
					minRTT =(INT)pechoREPLY->RoundTripTime;
				}
				if(minRTT > (INT)pechoREPLY->RoundTripTime)
				{
					minRTT =(INT)pechoREPLY->RoundTripTime;
				}
				if(maxRTT < (INT)pechoREPLY->RoundTripTime)
				{
					maxRTT =(INT)pechoREPLY->RoundTripTime;
				}
 
			}
			else
			{
				//丢包计数器
				failedICMP ++;
 
				SetConsoleTextAttribute(handle_out, 4|8);
				_ftprintf(stdout, _T("ERROR NUM: %u\n"), pechoREPLY->Status);
				SetConsoleTextAttribute(handle_out, 2|8);
			}
		}
		else
		{
			failedICMP ++;
			SetConsoleTextAttribute(handle_out, 4|8);
			_ftprintf(stderr, _T("THE ICMP MESSAGE SENDS FAILED\n"));
			break;
		}
	}
 
	//计算回环方差
	DOUBLE avageTIME=totalRTT/succeedICMP, sdevTIME=sqrt(totalSquareRTT/succeedICMP-pow((DOUBLE)(totalRTT/succeedICMP), 2)), pingQUA=(avageTIME==0.0f)?1:(1.0f-sdevTIME/avageTIME);
 
	//设置显示颜色
	SetConsoleTextAttribute(handle_out, 1|2|8);
 
	//统计PING质率
	_ftprintf(stdout, _T("PINGQUA : %-1.1lf\n"), pingQUA);
 
	//统计包达成率
	_ftprintf(stdout, _T("PACKETS : SEND=%d, RECEIVED=%d, LOST=%d (%2.0lf%% LOSS)\n"), succeedICMP+failedICMP, succeedICMP, failedICMP, (DOUBLE)failedICMP*100.0f/(succeedICMP+failedICMP));
 
	//显示统计信息
	_ftprintf(stdout, _T("TRIPTIME: MIN(TIME)=%dms, MAX(TIME)=%dms, E(TIME)=%dms, SDT=%.2lfms\n"), minRTT, maxRTT, (INT)avageTIME, sdevTIME);
 
	//释放句柄
	CloseHandle(hicmpFILE);
 
	//释放资源
	operator delete (sendDATA);
	operator delete (replyBUFF);
	delete sendOPTS;
	delete replyADDR;
 
#if defined(_UNICODE) || defined(UNICODE)
 
	delete replyADDRW;
 
#endif
 
	return 0;
}
 
#if defined _MSC_VER
#else
extern "C"
#endif
 
//MAIN主函数
INT _tmain(INT argc, TCHAR** argv)
{
	if(argc<2)
	{
		//无参数则退出
		_ftprintf(stdout, HELP_INFORMATION);
		exit(0);
	}
 
	//启动安全套接字
	WSADATA  Ws;
	if(WSAStartup(MAKEWORD(2, 2), &Ws) !=0)
	{
		_ftprintf(stderr, _T("Starting the winsocket failed\n"));
		return 1;
	}
 
	//设置传入参数
	TCHAR* opeTHOST=NULL;
	BYTE opeFLAG=0x00;
	INT K=_OPT_TEOF, opeCOUNT=DEFAULT_COUNT, opeTTL=DEFAULT_TTL, opeTOS=DEFAULT_TOS, opeTIMEOUT=DEFAULT_TIMEOUT, opeSENDSIZE=DEFAULT_SEND_SIZE, opeTIMEDELAY=DEFAULT_TIMEDELAY;
 
	//开关解析
	while((K=_tgetopt(argc, argv, (TCHAR*)_T("htfan:i:v:w:l:d:HTFAN:I:V:W:L:D:"))) != _OPT_TEOF)
	{
		switch(K)
		{
		case 't':
		case 'T':
			opeFLAG |= 0x01;
			break;
 
		case 'f':
		case 'F':
			opeFLAG |= 0x02;
			break;
 
		case 'a':
		case 'A':
			;
			opeFLAG |= 0x04;
			break;
 
		case 'n':
		case 'N':
			if(OPTARG !=NULL && _istdigit(*OPTARG) !=0)
			{
				opeCOUNT =_ttoi((TCHARFORMAT*)OPTARG);
				if(opeCOUNT > 0)
				{
					opeFLAG |= 0x08;
					break;
				}
			}
			_ftprintf(stderr, _T("The switch '-n' needs a positive number\n"));
			exit(1);
 
		case 'i':
		case 'I':
			if(OPTARG !=NULL && _istdigit(*OPTARG) !=0)
			{
				opeTTL =_ttoi((TCHARFORMAT*)OPTARG);
				if(opeTTL >= 0)
				{
 
					opeFLAG |= 0x0F;
					break;
				}
			}
			_ftprintf(stderr, _T("The switch '-i' needs a positive number\n"));
			exit(1);
 
		case 'v':
		case 'V':
			if(OPTARG !=NULL && _istdigit(*OPTARG) !=0)
			{
				opeTOS =_ttoi((TCHARFORMAT*)OPTARG);
				if(opeTOS >= 0)
				{
					opeFLAG |= 0x10;
					break;
				}
			}
			_ftprintf(stderr, _T("The switch '-v' needs a positive number\n"));
			exit(1);
 
		case 'w':
		case 'W':
			if(OPTARG !=NULL && _istdigit(*OPTARG) !=0)
			{
				opeTIMEOUT =_ttoi((TCHARFORMAT*)OPTARG);
				if(opeTIMEOUT >= 0)
				{
					opeFLAG |= 0x20;
					break;
				}
			}
			_ftprintf(stderr, _T("The switch '-w' needs a positive number\n"));
			exit(1);
 
		case 'l':
		case 'L':
			if(OPTARG !=NULL && _istdigit(*OPTARG) !=0)
			{
				opeSENDSIZE =_ttoi((TCHARFORMAT*)OPTARG);
				if(0<= opeSENDSIZE && opeSENDSIZE <=65500)
				{
					opeFLAG |= 0x40;
					break;
				}
			}
			_ftprintf(stderr, _T("The switch '-l' needs a number less 65KB\n"));
			exit(1);
 
		case 'd':
		case 'D':
			if(OPTARG !=NULL && _istdigit(*OPTARG) !=0)
			{
				opeTIMEDELAY =_ttoi((TCHARFORMAT*)OPTARG);
				if(opeTIMEDELAY >= 0)
				{
					opeFLAG |= 0x80;
					break;
				}
			}
			_ftprintf(stderr, _T("The switch '-d' needs a positive number\n"));
			exit(1);
 
		case 'h':
		case 'H':
			_ftprintf(stdout, HELP_INFORMATION);
			exit(0);
 
		case _OPT_TILL:
			//第一个无选项的参数识别为主机名
			opeTHOST = argv[UNOPTIND];
			break;
 
		case _OPT_TERR:
			_ftprintf(stderr, _T("Extra parameters \"%s\"\n"), argv[OPTIND]);
			exit(1);
 
		default:
			_ftprintf(stderr, _T("Unknown switch '-%c'\n"), K);
			exit(1);
 
		}
	}
 
	//无主机名，强制退出
	if(opeTHOST == NULL)
	{
		_ftprintf(stderr, _T("Needs host name or IP address\n"));
		exit(1);
	}
 
#if defined(_UNICODE) || defined(UNICODE)
 
	INT thostLEN = WideCharToMultiByte(CP_ACP, NULL, opeTHOST, -1, NULL, 0, NULL, NULL);
	CHAR* opeHOSTNAME = new CHAR[thostLEN + 1];
	WideCharToMultiByte(CP_ACP, NULL, opeTHOST, -1, opeHOSTNAME, thostLEN, NULL, NULL);
	opeHOSTNAME[thostLEN] = '\0';
 
#else
 
	CHAR* opeHOSTNAME = opeTHOST;
 
#endif
 
	//获取主机HOST
	PHOSTENT pHOSTENT = gethostbyname(opeHOSTNAME);
	ULONG uIPADRR = INADDR_NONE;
 
	//尝试作为IP地址解析
	if (! pHOSTENT)
	{
		uIPADRR = inet_addr(opeHOSTNAME);
		//识别失败，非正常主机名
		if (uIPADRR == INADDR_NONE)
		{
			_ftprintf(stderr, _T("Not a correct hostname\n"), K);
			exit(1);
		}
	}
 
	//获取UIP值
	PIN_ADDR pINADRR = (PIN_ADDR)pHOSTENT->h_addr;
	if((uIPADRR=pINADRR->S_un.S_addr) == INADDR_NONE)
	{
		//获取UIP失败，执行退出
		_ftprintf(stderr, _T("Get UIP failed\n"), K);
		exit(1);
	}
 
	//获取控制台输出句柄
	HANDLE handle_out=GetStdHandle(STD_OUTPUT_HANDLE);
 
	//获取控制台初始配色
	WORD orgCOLOR;
	CONSOLE_SCREEN_BUFFER_INFO buffINFO;
	GetConsoleScreenBufferInfo(handle_out, &buffINFO);
	orgCOLOR=buffINFO.wAttributes;
 
	//显示题头
	ColorTcs(handle_out, (TCHARFORMAT*)STR_HEAD, (TCHARFORMAT*)_T(""), 4|8, 2|4|8);
	//显示主机名
	ColorTcs(handle_out, (TCHARFORMAT*)_T("<HOST-NAME> "), _tcsupr(toTCHAR(pHOSTENT->h_name, FALSE)), 4|8, 1|2|8);
 
	//显示主机别名
	INT iINDEX = 0;
	while(pHOSTENT->h_aliases[iINDEX])
	{
		ColorTcs(handle_out, (TCHARFORMAT*)_T("<ALIA-NAME> "), _tcsupr(toTCHAR(pHOSTENT->h_aliases[iINDEX], FALSE)), 4|8, 1|2|8);
		iINDEX ++;
	}
	//显示题尾
	ColorTcs(handle_out, (TCHARFORMAT*)STR_LINE, (TCHARFORMAT*)_T(""), 4|8, 2|8);
 
	//调用 PING 核心函数
	INT retVALUE = SendPingByUIP(handle_out, uIPADRR, opeFLAG, opeCOUNT, opeTTL, opeTOS, opeTIMEOUT, opeSENDSIZE, opeTIMEDELAY);
 
#if defined(_UNICODE) || defined(UNICODE)
 
	//释放内存
	delete opeHOSTNAME;
 
#endif
 
	//还原字体颜色
	SetConsoleTextAttribute(handle_out, orgCOLOR);
 
	//关闭安全套接字
	WSACleanup();
 
	return retVALUE;
}COPY