Windows平台下的图形化的Ping工具(2)（基于MFC的Dialog工程）

// WinPingDlg.cpp : implementation file // #include "stdafx.h" #include "WinPing.h" #include "WinPingDlg.h" #include "ProtoInfo.h" #ifdef _DEBUG #define new DEBUG_NEW #endif // CAboutDlg dialog used for App About class CAboutDlg : public CDialog { public: CAboutDlg(); // Dialog Data enum { IDD = IDD_ABOUTBOX }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support // Implementation protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialog) END_MESSAGE_MAP() // CWinPingDlg dialog CString CWinPingDlg::m_strResult = _T(""); #define UM_PINGMSG WM_USER+10011 #define UM_PINGFIN WM_USER+10012 #define DEFAULT_DATA_SIZE 32 #define DEFAULT_SEND_COUNT 4 #define DEFAULT_RECV_TIMEOUT 6000 #define MAX_BUFFER_LEN 1024 CWinPingDlg::CWinPingDlg(CWnd* pParent /*=NULL*/) : CDialog(CWinPingDlg::IDD, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CWinPingDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); DDX_Control(pDX, IDC_COMBO_ADDRESS, m_combo); DDX_Control(pDX, IDC_EDIT_RESULT, m_edit); } BEGIN_MESSAGE_MAP(CWinPingDlg, CDialog) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() //}}AFX_MSG_MAP ON_BN_CLICKED(IDOK, &CWinPingDlg::OnBnClickedOk) ON_MESSAGE(UM_PINGMSG, &CWinPingDlg::OnPingMsg) ON_MESSAGE(UM_PINGFIN, &CWinPingDlg::OnPingFin) END_MESSAGE_MAP() // CWinPingDlg message handlers BOOL CWinPingDlg::OnInitDialog() { CDialog::OnInitDialog(); // Add "About..." menu item to system menu. // IDM_ABOUTBOX must be in the system command range. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != NULL) { BOOL bNameValid; CString strAboutMenu; bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX); ASSERT(bNameValid); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 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 // TODO: Add extra initialization here m_combo.AddString(_T("127.0.0.1")); m_combo.SetCurSel(0); return TRUE; // return TRUE unless you set the focus to a control } void CWinPingDlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialog::OnSysCommand(nID, lParam); } } // If you add a minimize button to your dialog, you will need the code below // to draw the icon. For MFC applications using the document/view model, // this is automatically done for you by the framework. void CWinPingDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // device context for painting SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // Center icon in client rectangle int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // Draw the icon dc.DrawIcon(x, y, m_hIcon); } else { CDialog::OnPaint(); } } // The system calls this function to obtain the cursor to display while the user drags // the minimized window. HCURSOR CWinPingDlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } void CWinPingDlg::OnBnClickedOk() { // TODO: Add your control notification handler code here //OnOK(); m_edit.SetWindowText(_T("")); GetDlgItem(IDOK)->EnableWindow(FALSE); AddStringToComboBox(); CWinThread* pThread = AfxBeginThread(PingThreadProc, (LPVOID)GetSafeHwnd()); CloseHandle(pThread->m_hThread); } void CWinPingDlg::AddStringToComboBox() { CString strText(_T("")); m_combo.GetWindowText(strText); if(CB_ERR == m_combo.FindString(-1, strText)) { m_combo.AddString(strText); } } UINT CWinPingDlg::PingThreadProc(LPVOID lParam) { HWND hWnd = (HWND)lParam; ASSERT(hWnd); WSAOVERLAPPED ol; SOCKET sRaw = INVALID_SOCKET; char* icmpbuf = NULL; struct addrinfo* dest = NULL; struct addrinfo* local = NULL; SOCKADDR_STORAGE from; DWORD bytes = 0; DWORD flags = 0; int packlen = 0; int fromlen = sizeof(from); int time = 0; int rc = 0; int status = 0; ol.hEvent = WSA_INVALID_EVENT; char szDestination[MAX_BUFFER_LEN] = {0}; char recvBuf[MAX_BUFFER_LEN] = {0}; GetDlgItemTextA(hWnd, IDC_COMBO_ADDRESS, szDestination, MAX_BUFFER_LEN); dest = CProtoInfo::ResolveAddress(szDestination, "0", AF_INET, 0, 0); if(NULL == dest) { AfxMessageBox(_T("ResolveAddress 1 failed.")); return 0; } local = CProtoInfo::ResolveAddress(NULL, "0", AF_INET, 0, 0); if(NULL == local) { AfxMessageBox(_T("ResolveAddress 2 failed.")); return 0; } sRaw = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP); if(INVALID_SOCKET == sRaw) { AfxMessageBox(_T("socket failed.")); return 0; } CProtoInfo::SetTTL(sRaw); packlen += sizeof(ICMP_HDR); int nDataSize = DEFAULT_DATA_SIZE; packlen += nDataSize; icmpbuf = (char*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, packlen); if(NULL == icmpbuf) { AfxMessageBox(_T("HeapAlloc failed.")); closesocket(sRaw); return 0; } CProtoInfo::InitIcmpHeader(icmpbuf, nDataSize); if(SOCKET_ERROR == bind(sRaw, local->ai_addr, (int)local->ai_addrlen)) { AfxMessageBox(_T("bind failed.")); closesocket(sRaw); return 0; } memset(&ol, 0, sizeof(ol)); ol.hEvent = WSACreateEvent(); if(WSA_INVALID_EVENT == ol.hEvent) { AfxMessageBox(_T("WSACreateEvent failed.")); closesocket(sRaw); return 0; } WSABUF wbuf; wbuf.buf = recvBuf; wbuf.len = 1024; if(SOCKET_ERROR == WSARecvFrom(sRaw, &wbuf, 1, &bytes, &flags, (LPSOCKADDR)&from, &fromlen, &ol, NULL)) { if(WSA_IO_PENDING != WSAGetLastError()) { AfxMessageBox(_T("WSARecvFrom failed.")); closesocket(sRaw); return 0; } } BOOL bAddr = FALSE; if(0 == atoi(szDestination)) { bAddr = TRUE; } CString strTmp (_T("")); strTmp = CProtoInfo::PrintAddress(dest->ai_addr, (int)dest->ai_addrlen); if(!bAddr) { m_strResult.Format(_T("Ping %s with %d bytes of data:\r\n"), strTmp, nDataSize); } else { m_strResult.Format(_T("Ping %s [%s] with %d bytes of data:\r\n"), CString(szDestination), strTmp, nDataSize); } ::SendMessage(hWnd, UM_PINGMSG, (WPARAM)0, (LPARAM)m_strResult.GetBuffer(m_strResult.GetLength())); m_strResult.ReleaseBuffer(); int nRecive = DEFAULT_SEND_COUNT; int nSend = DEFAULT_SEND_COUNT; int nMin = DEFAULT_RECV_TIMEOUT; int nMax = 0; int nTime[DEFAULT_SEND_COUNT] = {0}; for(int i=0; i<DEFAULT_SEND_COUNT; i++) { CProtoInfo::SetIcmpSequence(icmpbuf); CProtoInfo::ComputeIcmpChecksum(icmpbuf, packlen); time = GetTickCount(); if(SOCKET_ERROR == sendto(sRaw, icmpbuf, packlen, 0, dest->ai_addr, (int)dest->ai_addrlen)) { --nSend; continue; } rc = WaitForSingleObject((HANDLE)ol.hEvent, DEFAULT_RECV_TIMEOUT); if(WAIT_FAILED == rc) { continue; } else if(WAIT_TIMEOUT == rc) { m_strResult.Format(_T("Request timeout.")); ::SendMessage(hWnd, UM_PINGMSG, (WPARAM)0, (LPARAM)m_strResult.GetBuffer(m_strResult.GetLength())); m_strResult.ReleaseBuffer(); --nRecive; } else { WSAGetOverlappedResult(sRaw, &ol, &bytes, FALSE, &flags); time = GetTickCount() - time; nTime[i] = time; nMin = min(nMin, time); nMax = max(nMax, tim