// MyPingExampleDlg.cpp : implementation file
//

#include "stdafx.h"
#include "MyPingExample.h"
#include "MyPingExampleDlg.h"
#include <icmpapi.h>
#include <winsock.h>
#include <ipExport.h>

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CMyPingExampleDlg dialog

CMyPingExampleDlg::CMyPingExampleDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CMyPingExampleDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CMyPingExampleDlg)
	m_ipAddr = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CMyPingExampleDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CMyPingExampleDlg)
	DDX_Control(pDX, IDC_LIST_RECV, m_lstRecv);
	DDX_Text(pDX, IDC_EDIT_IP, m_ipAddr);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CMyPingExampleDlg, CDialog)
	//{{AFX_MSG_MAP(CMyPingExampleDlg)
	ON_WM_HELPINFO()
	ON_BN_CLICKED(IDC_BTNPING, OnBtnping)
	ON_BN_CLICKED(IDC_BTNCLEAR, OnBtnclear)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMyPingExampleDlg message handlers

BOOL CMyPingExampleDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	CenterWindow(GetDesktopWindow());	// center to the hpc screen

	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CMyPingExampleDlg::OnHelpInfo()
{
	// TODO: implement help here
	MessageBox(_T("Help"));
}

/*
*函數介紹：CE下Ping方法
*入口參數：cIPAddr  : IP地址
		   pWnd     : CWnd對象，用于指定窗體對象    
*出口參數：pListBox : CListBox對象，用于顯示Ping信息
*返回值：(無)
*/
void CEPing(CString cIPAddr,CListBox * pListBox,CWnd * pWnd)
{
   	WSAData wsaData;
	//初始化Socket動態鏈接庫
    if (WSAStartup(MAKEWORD(1, 1), &wsaData) != 0) {
        return ;
    }
	
	/*將IP地址轉換成單字節*/
    char szDestIPAddr[255];
	ZeroMemory(szDestIPAddr,255);
	WideCharToMultiByte(CP_ACP,WC_COMPOSITECHECK
		,cIPAddr.GetBuffer(cIPAddr.GetLength()),cIPAddr.GetLength()
		,szDestIPAddr,cIPAddr.GetLength(),NULL,NULL);
	
	IPAddr ipAddr;
	//將目標字符串IP地址轉換成IPAddr結構
    ipAddr = inet_addr(szDestIPAddr);      
	
	if (ipAddr == INADDR_NONE) 
	{
		AfxMessageBox(_T("地址無效"));
		return ;
	}
	
	// 打開ping服務
    HANDLE hIP = IcmpCreateFile();
    if (hIP == INVALID_HANDLE_VALUE) {
        AfxMessageBox(_T("不能打開Ping服務"));
        return ;
    }
	
	
	// 構造ping數據包
    char acPingBuffer[64];
    memset(acPingBuffer, '*', sizeof(acPingBuffer));
    PICMP_ECHO_REPLY pIpe = (PICMP_ECHO_REPLY)GlobalAlloc( GMEM_FIXED | GMEM_ZEROINIT,
		sizeof(ICMP_ECHO_REPLY) + sizeof(acPingBuffer)); 
    if (pIpe == 0) {
		AfxMessageBox(_T("分配ping包緩沖區失敗"));
        return ;
    }
    
	CString m_recv;
    // 發送ping數據包
	m_recv = "ping "+cIPAddr+" with 64 Bytes";
	pListBox->AddString(m_recv);
    for (int i=0;i<4;i++)
	{
		//發送ping服務包，等待接收時間為1秒
		DWORD dwStatus = IcmpSendEcho(hIP,ipAddr, 
            acPingBuffer, sizeof(acPingBuffer), NULL, pIpe, 
            sizeof(ICMP_ECHO_REPLY) + sizeof(acPingBuffer), 1000);
		//當dwStatus不等于0，代表接收到回應
		if (dwStatus != 0) {
			m_recv.Format(_T("Reply From %d.%d.%d.%d :bytes=%d time=%d TTL")
				,int(LOBYTE(LOWORD(pIpe->Address)))
				,int(HIBYTE(LOWORD(pIpe->Address)))
				,int(LOBYTE(HIWORD(pIpe->Address)))
				,int(HIBYTE(HIWORD(pIpe->Address)))
				,int(pIpe->DataSize)
				,int(pIpe->RoundTripTime)
				,int(pIpe->Options.Ttl));
			pListBox->AddString(m_recv);
			Sleep(500);
		}
		else 
		{
			pListBox->AddString(_T("Error obtaining info from ping packet."));
		}
		pWnd->UpdateWindow();
	}
	
    //釋放已分配的內存
    GlobalFree(pIpe);
	//關閉Ping服務
    IcmpCloseHandle(hIP);
    //釋放Socket資源
	WSACleanup();
}

void CMyPingExampleDlg::OnBtnping() 
{
	// TODO: Add your control notification handler code here
	UpdateData(TRUE);
  CEPing(m_ipAddr,&m_lstRecv,this);
  UpdateWindow();	
}

void CMyPingExampleDlg::OnBtnclear() 
{
	// TODO: Add your control notification handler code here
		int nCount = m_lstRecv.GetCount();
	
	for (int i=0;i < nCount;i++)
	{
		m_lstRecv.DeleteString(0);
	}
}