#include "common.h"
#include "icmp.h"

char data[] = "AAAABBBBCCCCDDDDEEEEFFFFGGGGHHH";
#define EXTRA		(12)
ICMP *ICMP::instance = 0;

ICMP::ICMP():
	timeout(2000),
	functions_loaded(false)
{
	hDLL = LoadLibrary("IPHLPAPI.DLL");
	if(hDLL) {
		pIcmpCreateFile = (pfnHV)GetProcAddress(hDLL, "IcmpCreateFile");
		pIcmpCloseHandle = (pfnBH)GetProcAddress(hDLL, "IcmpCloseHandle");
		pIcmpSendEcho2 = (pfnDHDPWPipPDD)GetProcAddress(hDLL, "IcmpSendEcho2");
	}
	if (hDLL == 0 || pIcmpCreateFile == 0 || pIcmpCloseHandle == 0 || pIcmpSendEcho2 == 0) {
		hDLL = LoadLibrary("ICMP.DLL");
		if(hDLL) {
			pIcmpCreateFile = (pfnHV)GetProcAddress(hDLL, "IcmpCreateFile");
			pIcmpCloseHandle = (pfnBH)GetProcAddress(hDLL, "IcmpCloseHandle");
			pIcmpSendEcho2 = (pfnDHDPWPipPDD)GetProcAddress(hDLL, "IcmpSendEcho2");
		}
		if (hDLL == 0 || pIcmpCreateFile == 0 || pIcmpCloseHandle == 0 || pIcmpSendEcho2 == 0)
			return;
		else
			DBWriteContactSettingString(0, PLUG, "PingLib", "ICMP.DLL"); // for debugging
	} else
		DBWriteContactSettingString(0, PLUG, "PingLib", "IPHLPAPI.DLL"); // for debugging

    WSAData wsaData;
    if (WSAStartup(MAKEWORD(1, 1), &wsaData) != 0) {
		WSACleanup();
		FreeLibrary((HMODULE)hDLL);
		return;
    }

	hIP = pIcmpCreateFile();
	if (hIP == INVALID_HANDLE_VALUE) {
		pIcmpCloseHandle(hIP);
		return;
	}
   
	buff = new char[sizeof(ICMP_ECHO_REPLY) + sizeof(data) + EXTRA];
	functions_loaded = true;
}

void ICMP::stop() {
	if(hIP) {
		pIcmpCloseHandle(hIP);
		hIP = 0;
	}
}

ICMP::~ICMP() {
	if(hIP) stop();
	WSACleanup();
	if(hDLL)
		FreeLibrary(hDLL);	
	delete[] buff;
}

bool ICMP::ping(char *host, ICMP_ECHO_REPLY &reply) {
	if(!functions_loaded) return false;

	unsigned long address;
	HOSTENT *rec;
	IP_OPTION_INFORMATION ipoi;

	address = inet_addr(host);
	if (address == INADDR_NONE) {
		rec = gethostbyname(host);
		if(rec) address = *(unsigned long *)(*rec->h_addr_list);
		else return false;
	}

	ipoi.Ttl = 255;
	ipoi.Tos = 0;
	ipoi.Flags = 0;
	ipoi.OptionsSize = 0;
	ipoi.OptionsData = 0;
	
	reply.Status = 0;	

	//pIcmpSendEcho2(hIP, 0, 0, 0, address, data, sizeof(data), &ipoi, buff, sizeof(ICMP_ECHO_REPLY) + sizeof(data), timeout);
	if(pIcmpSendEcho2(hIP, 0, 0, 0, address, data, sizeof(data), 0, buff, sizeof(ICMP_ECHO_REPLY) + sizeof(data) + EXTRA, timeout) == 0) {
		char winmsg[512], msg[1024];
		FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM, 0, GetLastError(), 0, winmsg, 512, 0);
		mir_snprintf(msg, 1024, "Ping error: %s\nICMP_ECHO_REPLY: %d\ndata: %d\ngiven: %d", winmsg, sizeof(ICMP_ECHO_REPLY), sizeof(data), sizeof(ICMP_ECHO_REPLY) + sizeof(data) + EXTRA);
		PUShowMessage(msg, SM_NOTIFY);
		return false;
	}
	memcpy(&reply, buff, sizeof(ICMP_ECHO_REPLY));

	return (reply.Status == 0);
}

ICMP *ICMP::get_instance() {
	if(!instance)
		instance = new ICMP();
	return instance;
}

void ICMP::cleanup() {
	if(instance) delete instance;
}