#include "stdafx.h"
#include <winsock2.h>
#include <iphlpapi.h>
#include <icmpapi.h>
#include <iostream>
#include <Windows.h>
#pragma warning(disable: 4996)
using std::cout;
using std::endl;
using std::atoi;

#pragma comment(lib, "iphlpapi.lib")
#pragma comment(lib, "ws2_32.lib")

#define IP_STATUS_BASE 11000
#define IP_SUCCESS 0
#define IP_DEST_NET_UNREACHABLE 11002
#define IP_DEST_HOST_UNREACHABLE 11003
#define IP_DEST_PROT_UNREACHABLE 11004
#define IP_DEST_PORT_UNREACHABLE 11005
#define IP_REQ_TIMED_OUT 11010
#define IP_BAD_REQ 11011
#define IP_BAD_ROUTE 11012
#define IP_TTL_EXPIRED_TRANSIT 11013

void Ping(const char* cHost, unsigned int Timeout, unsigned int RequestCount)
{
	HANDLE hIP = IcmpCreateFile();

	if (hIP == INVALID_HANDLE_VALUE)
	{
		WSACleanup();
		return;
	}

	char SendData[32] = "Data for ping";//буфер для передачи
	int LostPacketsCount = 0; // кол-во потерянных пакетов unsigned
	int MaxMS = 0;// максимальное время ответа (мс)
	int MinMS = -1;    // минимальное время ответа (мс)

	// Выделяем память под пакет – буфер ответа
	PICMP_ECHO_REPLY pIpe = (PICMP_ECHO_REPLY)GlobalAlloc(GHND, sizeof(ICMP_ECHO_REPLY) + sizeof(SendData));
	if (pIpe == 0) {
		IcmpCloseHandle(hIP);
		WSACleanup();
		return;
	}
	pIpe->Data = SendData;
	pIpe->DataSize = sizeof(SendData);
	unsigned long ipaddr = inet_addr(cHost);

	IP_OPTION_INFORMATION option = IP_OPTION_INFORMATION();
	option.Ttl = 100;
	option.Tos = 0;
	option.Flags = IP_FLAG_DF;
	option.OptionsSize = 0;
	option.OptionsData = 0;

	for (unsigned int c = 0; c < RequestCount; c++) {
		int dwStatus = IcmpSendEcho(hIP, ipaddr, SendData, sizeof(SendData), &option, pIpe, sizeof(ICMP_ECHO_REPLY) + sizeof(SendData), Timeout);
		if (dwStatus > 0) {
			for (int i = 0; i< dwStatus; i++) {
				unsigned char* pIpPtr = (unsigned char*)&pIpe->Address;
				cout << "Ответ от  " << (int)*(pIpPtr)
					 << "." << (int)*(pIpPtr + 1)
					 << "." << (int)*(pIpPtr + 2)
					 << "." << (int) *(pIpPtr + 3)
					 << ": число байт = " << pIpe->DataSize
					 << " время = " << pIpe->RoundTripTime
					 << "мс TTL = " << (int)pIpe->Options.Ttl;

				MaxMS = (MaxMS > pIpe->RoundTripTime) ? MaxMS : pIpe->RoundTripTime;

				MinMS = (MinMS < (int)pIpe->RoundTripTime  && MinMS >= 0) ? MinMS : pIpe->RoundTripTime;

				cout << endl;
			}
		}
		else {
			if (pIpe->Status) {
				LostPacketsCount++;
				switch (pIpe->Status) {
				case IP_DEST_NET_UNREACHABLE:
				case IP_DEST_HOST_UNREACHABLE:
				case IP_DEST_PROT_UNREACHABLE:
				case IP_DEST_PORT_UNREACHABLE:
					cout << "Remote host may be down." << endl;
					break;
				case IP_REQ_TIMED_OUT:
					cout << "Request timed out." << endl;
					break;
				case IP_TTL_EXPIRED_TRANSIT:
					cout << "TTL expired in transit." << endl;
					break;
				default:
					cout << "Error code: %ld" << pIpe->Status << endl;
					break;
				}
			}
		}
	}

	IcmpCloseHandle(hIP);
	WSACleanup();

	if (MinMS<0) MinMS = 0;
	unsigned char* pByte = (unsigned char*)&pIpe->Address;
	cout << "Статистика Ping    " << (int)*(pByte)
		 << "." << (int)*(pByte + 1)
		 << "." << (int)*(pByte + 2)
		 << "." << (int)*(pByte + 3) << endl;

	cout << "\tПакетов: отправлено = " << RequestCount
		 << ", получено = " << RequestCount - LostPacketsCount
		 << ", потеряно = " << LostPacketsCount
		 << "<" << (int)(100 / (float)RequestCount)* LostPacketsCount
		 << " % потерь>, " << endl;

	cout << "Приблизительное время приема-передачи:" << endl
		 << "Минимальное = " << MinMS
		 << "мс, Максимальное = " << MaxMS
		 << "мс, Среднее = " << (MaxMS + MinMS) / 2 << "мс" << endl;
}

int main(int argc, char* argv[])
{
	SetConsoleCP(1251);
	SetConsoleOutputCP(1251);

	/*cout << argc << endl;
	for (int i = 0; i < argc; i++) {
		cout << argv[i] << endl;
	}*/

	switch (argc) {
	case 1: {
		Ping("127.0.0.1", 60, 10);
		break;
	}
	case 2: {
		Ping(argv[1], 60, 10);
		break;
	}
	case 3: {
		int timeout = atoi(argv[2]);

		if (timeout == 0) {
			cout << "INCORRECT SECOND PARAMETR" << endl;
			return 1;
		}

		Ping(argv[1], timeout, 10);
		break;
	}
	case 4: {
		int timeout = atoi(argv[2]);
		int request_count = atoi(argv[3]);

		if (timeout == 0) {
			cout << "THE SECOND PARAMETR IS INCORRECT" << endl;
			return 1;
		}
		if (request_count == 0) {
			cout << "THE THIRD PARAMETR IS INCORRECT" << endl;
			return 1;
		}

		Ping(argv[1], timeout, request_count);
		break;
	}
	default: {
		cout << "INCORRECT ARGUMENTS" << endl
			 << "ALLOWED: \nmyPing\nmyPing ip\nmyPing ip timeout\nmyPing ip request_count\n";
	}
	}

    return 0;
}