【正文】 (fileReadString(somecode) != NULL) //如果采用 LPTSTR 類型，讀取最大個數(shù) nMax 置 0，使它遇空字符時結(jié)束 { (somecode)。 (\n)。 //如果 somecode 采用 LPTSTR 類型 ,可不用此句 } fileClose()。 delete file。 } else { (_T(到指定服務器的連接建立失敗 ...))。 return FALSE。 } return TRUE。 } void CLiveUpdateDlg::OnSysCommand(UINT nID, LPARAM lParam) { 網(wǎng)絡攻擊與防護之掃面軟件設(shè)計 29 if ((nID amp。 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout。 ()。 } else { CDialog::OnSysCommand(nID, lParam)。 } } void CLiveUpdateDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this)。 // device context for painting SendMessage(WM_ICONERASEBKGND, (WPARAM) (), 0)。 // Center icon in client rectangle int cxIcon = GetSystemMetrics(SM_CXICON)。 int cyIcon = GetSystemMetrics(SM_CYICON)。 CRect rect。 GetClientRect(amp。rect)。 int x = (() cxIcon + 1) / 2。 int y = (() cyIcon + 1) / 2。 // Draw the icon (x, y, m_hIcon)。 } else { CDialog::OnPaint()。 } 西南石油大學本科畢業(yè)設(shè)計（論文） 30 } // The system calls this to obtain the cursor to display while the user drags // the minimized window. HCURSOR CLiveUpdateDlg::OnQueryDragIcon() { return (HCURSOR) m_hIcon。 } // // 名稱 : OnOK // 功能 : 復制升級程序到目標目錄中 // 變量 : 無 // 返回 : 無 // void CLiveUpdateDlg::OnOK() { UpdateData(TRUE)。 if(()) { AfxMessageBox(請先選擇安裝目錄 )。 return。 } CString strTmp。 CFileFind fFind。 BOOL bSuccess = FALSE。 int nProgress = 0。 int nAverage = 100/()。 for(Iter_FileName = (),Iter_Name = ()。 (Iter_FileName != ())amp。amp。(Iter_Name != ())。 Iter_FileName++,Iter_Name++) { strTmp = m_TargetPath + \\ + *(Iter_Name)。 網(wǎng)絡攻擊與防護之掃面軟件設(shè)計 31 // 判斷路徑下是否有對應文件 bSuccess=(strTmp)。 if(!bSuccess) { CString str。 (安裝目錄中沒有原始文件： %s，是否安裝新的升級文件(Y/N), *(Iter_Name))。 if(IDYES == AfxMessageBox(str,MB_YESNO)) // 復制新文件 { BOOL bSucced = CopyFile(*(Iter_FileName), strTmp, FALSE)。 if(!bSucced) { AfxMessageBox(被升級的文件可能正在使用或為只讀屬性 !)。 return。 } } } else { // 為防止被升級文件為只讀等屬性，先將置為正常屬性再置回原屬性 DWORD dwAttr = GetFileAttributes(strTmp)。 SetFileAttributes(strTmp, FILE_ATTRIBUTE_ARCHIVE|FILE_ATTRIBUTE_NORMAL)。 BOOL bSucced = CopyFile(*(Iter_FileName), strTmp, FALSE)。 if(!bSucced) { AfxMessageBox(正被升級的文件可能正在使用 !)。 return。 } 西南石油大學本科畢業(yè)設(shè)計（論文） 32 SetFileAttributes(strTmp, dwAttr)。 // 恢復文件源屬性 } nProgress += nAverage。 (nProgress)。 } (100)。 ()。 AfxMessageBox(升級成功 !)。 } // // 名稱 : OnButtonBrowse // 功能 : 選擇目標目錄并進行判斷操作 // 變量 : 無 // 返回 : 無 // void CLiveUpdateDlg::OnButtonBrowse() { CDirDialog dlg。 if (IDOK == (this)) { // 設(shè)置選中路徑 m_TargetPath = 。 UpdateData(FALSE)。 } } // // 名稱 : OnReg // 功能 : 對控件、組件進行注冊 // 變量 : 無 // 返回 : 無 網(wǎng)絡攻擊與防護之掃面軟件設(shè)計 33 // void CLiveUpdateDlg::OnReg() { CFileDialog fileDialog( TRUE,*.OCX, *.DLL,NULL,NULL,控件或組件文件 (*.dll,*.ocx)|*.ocx。*.dll|所有文件 (*.*)|*.*||)。 if (IDOK == ()) { CString pszDllName = ()。 //已選控件、組件名是否為空 if(pszDllName==) { MessageBox(請先選擇您需要注冊的控件或組件名稱！ ,提示,MB_OK)。 return。 } //裝載控件或組件 HINSTANCE hLib = LoadLibrary(pszDllName)。 if (hLib (HINSTANCE)HINSTANCE_ERROR) { MessageBox(不能載入該控件或組件文件 !,錯誤 ,MB_OK)。 return。 } //獲取注冊函數(shù) DllRegisterServer 地址 FARPROC lpDllEntryPoint。 lpDllEntryPoint = GetProcAddress(hLib,_T(DllRegisterServer))。 //調(diào)用注冊函數(shù) DllRegisterServer if(lpDllEntryPoint!=NULL) { if(FAILED((*lpDllEntryPoint)())) { MessageBox(控件或組件注冊失敗！ ,錯誤 ,MB_OK)。 西南石油大學本科畢業(yè)設(shè)計（論文） 34 FreeLibrary(hLib)。 return。 } MessageBox(控件或組件注冊成功 ,注冊提示 ,MB_OK)。 } else MessageBox(控件或組件注冊失??！ ,錯誤 ,MB_OK)。 } } 掃描文件算法 編譯環(huán)境： VC++ 開發(fā)語言： c++ 實現(xiàn)主要代碼： 最大線程數(shù) 代碼 m_maxTrdCount = 20。 (1,USE_MAXTRD_NUM)。 //線程優(yōu)先級 m_trdPri = 2。 //傳入函數(shù)應 (2m_trdPri) m_bAutoKill = FALSE。 m_strDir = GetAllDriverList()。 UpdateData(FALSE)。 return TRUE。 // return TRUE unless you set the focus to a control } 當前電腦上所有驅(qū)動器列表 CString CMyKillVirusDlg::GetAllDriverList() { CString tmp=_T(),Dir。 for(int i=1。i=25。i++) { Dir=CString(39。A39。+i)+_T(:\\)。 網(wǎng)絡攻擊與防護之掃面軟件設(shè)計 35 if(GetDriveType((0))==DRIVE_NO_ROOT_DIR)continue。 tmp += Dir + 。 } return tmp。 } 自動生成文件擴展名窗體發(fā)回來的消息 LRESULT CMyKillVirusDlg::OnGetMatchStr(WPARAM pMatchStr,LPARAM pval) { CString s = CString(LPTSTR(pMatchStr))。 SetMatchFileStr(s)。 return 0。 } //接收消息掃描過程中發(fā)送到主窗體的各種消息，以得到當前文件名等信息 // 參數(shù)： 消息名 傳送的數(shù)據(jù) LRESULT CMyKillVirusDlg::OnScanFileStatus(WPARAM wStatus, LPARAM lValue) { UINT stuStatus = (UINT)wStatus。 switch(stuStatus) { case WM_SCEAN_END: //掃描結(jié)束 { INT value = (INT)lValue。 (TRUE)。 (TRUE)。 (FALSE)。 (FALSE)。 m_bStartScan = FALSE。 (掃描結(jié)束 !)。 break。 西南石油大學本科畢業(yè)設(shè)計（論文） 36 } case WM_SCEAN_NUMBER: //正在掃描過程中的 4 個數(shù)字信息 { INT *pInt = (INT *)(lValue)。 CString s[4]。 for(INT i=0。i4。i++) { s[i].Format(%d,*(pInt + i))。 } (總文件數(shù) :+s[0])。 // (病毒數(shù) :+s[1])。 // (清除數(shù) :+s[2])。 // (錯誤數(shù) :+s