【正文】 ////////////////////////////////////////////////////////////////////////// // CLiveUpdateDlg dialog CLiveUpdateDlg::CLiveUpdateDlg(CWnd* pParent /*=NULL*/) : CDialog(CLiveUpdateDlg::IDD, pParent) { //{{AFX_DATA_INIT(CLiveUpdateDlg) m_TargetPath = _T()。 } void CLiveUpdateDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX)。 DDX_Control(pDX, ID_REG, m_Reg)。 DDX_Control(pDX, IDOK, m_OK)。 DDX_Control(pDX, IDC_BUTTON_BROWSE, m_Browse)。 //}}AFX_DATA_MAP } BEGIN_MESSAGE_MAP(CLiveUpdateDlg, CDialog) //{{AFX_MSG_MAP(CLiveUpdateDlg) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDC_BUTTON_BROWSE, OnButtonBrowse) 西南石油大學(xué)本科畢業(yè)設(shè)計（論文） 24 ON_BN_CLICKED(ID_REG, OnReg) ON_BN_CLICKED(ID_UNREG, OnUnreg) ON_WM_CTLCOLOR() //}}AFX_MSG_MAP END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CLiveUpdateDlg message handlers // // 名稱 : OnInitDialog // 功能 : 在此進行初始化工作 // 變量 : 無 // 返回 : 無 // BOOL CLiveUpdateDlg::OnInitDialog() { CDialog::OnInitDialog()。 0xFFF0) == IDM_ABOUTBOX)。 CMenu* pSysMenu = GetSystemMenu(FALSE)。 (IDS_ABOUTBOX)。 pSysMenuAppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu)。 // Set big icon 網(wǎng)絡(luò)攻擊與防護之掃面軟件設(shè)計 25 SetIcon(m_hIcon, FALSE)。 GetModuleFileName(NULL, (MAX_PATH+1),MAX_PATH)。 int nPos = 0。\\39。 CString str = (()nPos1)。 // 設(shè)置為當(dāng)前路徑 SetCurrentDirectory ( sPath )。 HANDLE hFind。FindFileData )。 } //列出所有發(fā)現(xiàn)的文件 西南石油大學(xué)本科畢業(yè)設(shè)計（論文） 26 CString strTemp = sPath + \\ + 。 ()。FindFileData ) amp。 hFind != INVALID_HANDLE_VALUE )。 // 同目錄中沒有文件，則退出 if(0 == ()) { CString strInfo。 AfxMessageBox(strInfo)。 // PostMessage(WM_CLOSE)。 (RGB(160,180,220))。 // (100)。 ()。 ()。 ()。 網(wǎng)絡(luò)攻擊與防護之掃面軟件設(shè)計 27 ()。 ()。 // return TRUE unless you set the focus to a control } // // 名稱 : GetSourceTextFile // 功能 : 下載文本文件涵數(shù) // 變量 : theUrl 定下載網(wǎng)址 (如 : // Filename 下載保存的文件名 (如： c:\\) // 返回 : TURE 成功 // FALSE 失敗 // BOOL CLiveUpdateDlg::GetSourceTextFile(CString theUrl,CString Filename) { CInterSession session。 try { // 試著連接到指定 URL file = (CInterFile*) (theUrl)。 m_pExceptionDelete()。 } // 用 dataStore 來保存讀取的文件 CStdioFile dataStore。 //也可采用 LPTSTR 類型，將不會刪除文本中的 \n 回車符 BOOL bIsOk = (Filename, CFile::modeCreate | CFile::modeWrite | CFile::shareDenyWrite | CFile::typeText)。 // 讀寫文件，直到為空 while (fileReadString(somecode) != NULL) //如果采用 LPTSTR 類型，讀取最大個數(shù) nMax 置 0，使它遇空字符時結(jié)束 { (somecode)。 //如果 somecode 采用 LPTSTR 類型 ,可不用此句 } fileClose()。 } else { (_T(到指定服務(wù)器的連接建立失敗 ...))。 } return TRUE。 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout。 } else { CDialog::OnSysCommand(nID, lParam)。 // device context for painting SendMessage(WM_ICONERASEBKGND, (WPARAM) (), 0)。 int cyIcon = GetSystemMetrics(SM_CYICON)。 GetClientRect(amp。 int x = (() cxIcon + 1) / 2。 // Draw the icon (x, y, m_hIcon)。 } 西南石油大學(xué)本科畢業(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。 if(()) { AfxMessageBox(請先選擇安裝目錄 )。 } CString strTmp。 BOOL bSuccess = FALSE。 int nAverage = 100/()。 (Iter_FileName != ())amp。(Iter_Name != ())。 網(wǎng)絡(luò)攻擊與防護之掃面軟件設(shè)計 31 // 判斷路徑下是否有對應(yīng)文件 bSuccess=(strTmp)。 (安裝目錄中沒有原始文件： %s，是否安裝新的升級文件(Y/N), *(Iter_Name))。 if(!bSucced) { AfxMessageBox(被升級的文件可能正在使用或為只讀屬性 !)。 } } } else { // 為防止被升級文件為只讀等屬性，先將置為正常屬性再置回原屬性 DWORD dwAttr = GetFileAttributes(strTmp)。 BOOL bSucced = CopyFile(*(Iter_FileName), strTmp, FALSE)。 return。 // 恢復(fù)文件源屬性 } nProgress += nAverage。 } (100)。 AfxMessageBox(升級成功 !)。 if (IDOK == (this)) { // 設(shè)置選中路徑 m_TargetPath = 。 } } // // 名稱 : OnReg // 功能 : 對控件、組件進行注冊 // 變量 : 無 // 返回 : 無 網(wǎng)絡(luò)攻擊與防護之掃面軟件設(shè)計 33 // void CLiveUpdateDlg::OnReg() { CFileDialog fileDialog( TRUE,*.OCX, *.DLL,NULL,NULL,控件或組件文件 (*.dll,*.ocx)|*.ocx。 if (IDOK == ()) { CString pszDllName = ()。 return。 if (hLib (HINSTANCE)HINSTANCE_ERROR) { MessageBox(不能載入該控件或組件文件 !,錯誤 ,MB_OK)。 } //獲取注冊函數(shù) DllRegisterServer 地址 FARPROC lpDllEntryPoint。 //調(diào)用注冊函數(shù) DllRegisterServer if(lpDllEntryPoint!=NULL) { if(FAILED((*lpDllEntryPoint)())) { MessageBox(控件或組件注冊失??！ ,錯誤 ,MB_OK)。 return。 } else MessageBox(控件或組件注冊失敗！ ,錯誤 ,MB_OK)。 (1,USE_MAXTRD_NUM)。 //傳入函數(shù)應(yīng) (2m_trdPri) m_bAutoKill = FALSE。 UpdateData(FALSE)。 // return TRUE unless you set the focus to a control } 當(dāng)前電腦上所有驅(qū)動器列表 CString CMyKillVirusDlg::GetAllDriverList() { CString tmp=_T(),Dir。i=25。A39。 網(wǎng)絡(luò)攻擊與防護之掃面軟件設(shè)計 35 if(GetDriveType((0))==DRIVE_NO_ROOT_DIR)continue。 } return tmp。 SetMatchFileStr(s)。 } //接收消息掃描過程中發(fā)送到主窗體的各種消息，以得到當(dāng)前文件名等信息 // 參數(shù)： 消息名 傳送的數(shù)據(jù) LRESULT CMyKillVirusDlg::OnScanFileStatus(WPARAM wStatus, LPARAM lValue) { UINT stuStatus = (UINT)wStatus。 (TRUE)。 (FALSE)。 m_bStartScan = FALSE。 break。 CString s[4]。i4。 } (總文件數(shù) :+s[0])。 // (清除數(shù) :+s[2]