开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 系统/网络安全 > 杀毒 > 查看源码
nimdaDlg.cpp
资源名称：antinimda.zip　[点击查看]
上传用户：leon2013
上传日期：2007-01-10
资源大小：186k
文件大小：8k
源码类别：

杀毒

开发平台：

Visual C++

nimdaDlg.cpp：源码内容
  1. // nimdaDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "nimda.h"
  6. #include "nimdaDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. CString GetText(CWnd& wnd)
  15. {
  16. CString s;
  17. wnd.GetWindowText(s);
  18. return s;
  19. }
  23. /////////////////////////////////////////////////////////////////////////////
  24. // CAboutDlg dialog used for App About
  26. class CAboutDlg : public CDialog
  27. {
  28. public:
  29. CAboutDlg();
  31. // Dialog Data
  32. //{{AFX_DATA(CAboutDlg)
  33. enum { IDD = IDD_ABOUTBOX };
  34. //}}AFX_DATA
  36. // ClassWizard generated virtual function overrides
  37. //{{AFX_VIRTUAL(CAboutDlg)
  38. protected:
  39. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  40. //}}AFX_VIRTUAL
  42. // Implementation
  43. protected:
  44. //{{AFX_MSG(CAboutDlg)
  45. //}}AFX_MSG
  46. DECLARE_MESSAGE_MAP()
  47. };
  49. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  50. {
  51. //{{AFX_DATA_INIT(CAboutDlg)
  52. //}}AFX_DATA_INIT
  53. }
  55. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  56. {
  57. CDialog::DoDataExchange(pDX);
  58. //{{AFX_DATA_MAP(CAboutDlg)
  59. //}}AFX_DATA_MAP
  60. }
  62. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  63. //{{AFX_MSG_MAP(CAboutDlg)
  64. // No message handlers
  65. //}}AFX_MSG_MAP
  66. END_MESSAGE_MAP()
  68. /////////////////////////////////////////////////////////////////////////////
  69. // CNimdaDlg dialog
  71. CNimdaDlg::CNimdaDlg(CWnd* pParent /*=NULL*/, bool auto_scan /*=false*/)
  72. : CDialog(CNimdaDlg::IDD, pParent), m_autoscan(auto_scan)
  73. {
  74. //{{AFX_DATA_INIT(CNimdaDlg)
  75. //}}AFX_DATA_INIT
  76. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  77. }
  79. void CNimdaDlg::DoDataExchange(CDataExchange* pDX)
  80. {
  81. CDialog::DoDataExchange(pDX);
  82. //{{AFX_DATA_MAP(CNimdaDlg)
  83. DDX_Control(pDX, IDC_MBSCANNED, bytesscanned);
  84. DDX_Control(pDX, IDVIEWLOG, btnViewLog);
  85. DDX_Control(pDX, IDPAUSE, btnPause);
  86. DDX_Control(pDX, IDC_ROOT, root);
  87. DDX_Control(pDX, IDOK, btnOk);
  88. DDX_Control(pDX, IDC_OUTPUT, output);
  89. DDX_Control(pDX, IDC_SKIPPED, skipped);
  90. DDX_Control(pDX, IDC_SCANRATE, scanrate);
  91. DDX_Control(pDX, IDC_ERRORS, errors);
  92. DDX_Control(pDX, IDC_DELETED, deleted);
  93. DDX_Control(pDX, IDC_VELOCITY, velocity);
  94. DDX_Control(pDX, IDC_TIME, time);
  95. DDX_Control(pDX, IDC_SCANNED, scanned);
  96. DDX_Control(pDX, IDC_INFECTED, infected);
  97. DDX_Control(pDX, IDC_FILE, file);
  98. DDX_Control(pDX, IDC_DIRECTORIES, directories);
  99. DDX_Control(pDX, IDC_CLEANED, cleaned);
  100. //}}AFX_DATA_MAP
  101. }
  103. BEGIN_MESSAGE_MAP(CNimdaDlg, CDialog)
  104. //{{AFX_MSG_MAP(CNimdaDlg)
  105. ON_WM_SYSCOMMAND()
  106. ON_WM_PAINT()
  107. ON_WM_QUERYDRAGICON()
  108. ON_WM_TIMER()
  109. ON_BN_CLICKED(IDPAUSE, OnPause)
  110. ON_BN_CLICKED(IDVIEWLOG, OnViewlog)
  111. //}}AFX_MSG_MAP
  112. END_MESSAGE_MAP()
  114. /////////////////////////////////////////////////////////////////////////////
  115. // CNimdaDlg message handlers
  117. BOOL CNimdaDlg::OnInitDialog()
  118. {
  119. CDialog::OnInitDialog();
  121. // Add "About..." menu item to system menu.
  123. // IDM_ABOUTBOX must be in the system command range.
  124. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  125. ASSERT(IDM_ABOUTBOX < 0xF000);
  127. CMenu* pSysMenu = GetSystemMenu(FALSE);
  128. if (pSysMenu != NULL)
  129. {
  130. CString strAboutMenu;
  131. strAboutMenu.LoadString(IDS_ABOUTBOX);
  132. if (!strAboutMenu.IsEmpty())
  133. {
  134. pSysMenu->AppendMenu(MF_SEPARATOR);
  135. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  136. }
  137. }
  139. SetIcon(m_hIcon, TRUE); // Set big icon
  140. SetIcon(m_hIcon, FALSE); // Set small icon
  142. /* setup columns in output list */
  143. output.InsertColumn(0,"File",LVCFMT_LEFT,150);
  144. output.InsertColumn(1,"Status",LVCFMT_LEFT,80);
  145. output.InsertColumn(2,"Path",LVCFMT_LEFT,240);
  147. /* initial defaults */
  148. root.SetWindowText("c:");
  150. /* set the statistics update timer */
  151. SetTimer(1,200,NULL);
  153. /* if set to auto-scan start the scan now */
  154. if (m_autoscan) OnOK();
  156. return TRUE;  // return TRUE  unless you set the focus to a control
  157. }
  159. void CNimdaDlg::OnInfected(LPCTSTR filename, VirusKiller::SCANRESULT result)
  160. {
  161. if ((result&~VIRUS_SCANNED)>0) {
  162. CString file(filename), path;
  164. /* find the path break */
  165. int brk;
  166. if ((brk=file.ReverseFind('\'))==-1) brk=0;
  168. /* extract path */
  169. if (brk) path=file.Left(brk);
  171. /* extract file */
  172. file=file.Mid(brk+1);
  174. /* build the action string */
  175. CString action;
  176. switch (result) {
  177. case VIRUS_SKIPPED: action="skipped"; break;
  178. case VIRUS_SCANNED: action="scanned"; break;
  179. case VIRUS_INFECTED: action="infected"; break;
  180. case VIRUS_CLEANED: action="cleaned"; break;
  181. case VIRUS_DELETED: action="deleted"; break;
  182. default:
  183. action="error"; break;
  184. }
  186. /* add item to output list control */
  187. int itemid=output.InsertItem(0,file);
  189. /* set fields */
  190. output.SetItemText(itemid,1,action);
  191. output.SetItemText(itemid,2,path);
  192. }
  193. }
  195. void CNimdaDlg::OnSysCommand(UINT nID, LPARAM lParam)
  196. {
  197. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  198. {
  199. CAboutDlg dlgAbout;
  200. dlgAbout.DoModal();
  201. }
  202. else
  203. {
  204. CDialog::OnSysCommand(nID, lParam);
  205. }
  206. }
  208. // If you add a minimize button to your dialog, you will need the code below
  209. //  to draw the icon.  For MFC applications using the document/view model,
  210. //  this is automatically done for you by the framework.
  212. void CNimdaDlg::OnPaint() 
  213. {
  214. if (IsIconic())
  215. {
  216. CPaintDC dc(this); // device context for painting
  218. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  220. // Center icon in client rectangle
  221. int cxIcon = GetSystemMetrics(SM_CXICON);
  222. int cyIcon = GetSystemMetrics(SM_CYICON);
  223. CRect rect;
  224. GetClientRect(&rect);
  225. int x = (rect.Width() - cxIcon + 1) / 2;
  226. int y = (rect.Height() - cyIcon + 1) / 2;
  228. // Draw the icon
  229. dc.DrawIcon(x, y, m_hIcon);
  230. }
  231. else
  232. {
  233. CDialog::OnPaint();
  234. }
  235. }
  237. HCURSOR CNimdaDlg::OnQueryDragIcon()
  238. {
  239. return (HCURSOR) m_hIcon;
  240. }
  242. void CNimdaDlg::OnTimer(UINT nIDEvent) 
  243. {
  244. if (nIDEvent==1)
  245. UpdateStatistics();
  246. else
  247. CDialog::OnTimer(nIDEvent);
  248. }
  250. void CNimdaDlg::UpdateStatistics() 
  251. {
  252. if (GetText(btnOk)!="Start") {
  253. CString s;
  255. /* scanned files */
  256. s.Format("%d ", Scanned());
  257. scanned.SetWindowText(s);
  259. /* directories */
  260. s.Format("%d ", Directories());
  261. directories.SetWindowText(s);
  263. /* errors */
  264. s.Format("%d ", Errors());
  265. errors.SetWindowText(s);
  267. /* skipped */
  268. s.Format("%d ", Skipped());
  269. skipped.SetWindowText(s);
  271. /* deleted */
  272. s.Format("%d ", Deleted());
  273. deleted.SetWindowText(s);
  275. /* scan rate */
  276. int seconds=Duration().GetTotalSeconds();
  277. if (seconds>3) {
  278. s.Format("%6.2f ", ((double)BytesScanned() / seconds) / 1048576 * 60 );
  279. scanrate.SetWindowText(s);
  280. }
  282. /* megabytes scanned rate */
  283. s.Format("%d MB ", BytesScanned() /1048576 );
  284. bytesscanned.SetWindowText(s);
  286. /* velocity */
  287. if (seconds>3) {
  288. s.Format("%8.1f ", Velocity());
  289. velocity.SetWindowText(s);
  290. }
  292. /* infected files */
  293. s.Format("%d ", Infected());
  294. infected.SetWindowText(s);
  296. /* cleaned files */
  297. s.Format("%d ", Cleaned());
  298. cleaned.SetWindowText(s);
  300. /* elapsed time */
  301. time.SetWindowText(Duration().Format("%H : %M : %S "));
  303. /* current file */
  304. file.SetWindowText(LastFileScanned());
  306. if (!Running())
  307. if (GetText(btnOk)=="Stop") OnOK();
  308. }
  309. }
  311. void CNimdaDlg::OnOK() 
  312. {
  313. if (GetText(btnOk)=="Close") {
  314. /* exit program */
  315. CDialog::OnOK();
  316. }
  317. else if (Scanned()>0) {
  318. /* stop the thread if it is started */
  319. if (Running()) Stop();
  321. /* hide pause button */
  322. btnPause.ShowWindow(SW_HIDE);
  324. /* show view log button */
  325. btnViewLog.ShowWindow(SW_SHOW);
  327. /* set main button to close */
  328. btnOk.SetWindowText("Close");
  330. /* if this was an autoscan we can close now */
  331. if (m_autoscan) CDialog::OnOK();
  332. }
  333. else {
  334. /* get the root directory */
  335. CString scan_root=GetText(root);
  337. /* set root directory of scan */
  338. Root(scan_root);
  340. /* start the scan */
  341. Start();
  343. /* change button text */
  344. btnOk.SetWindowText("Stop");
  345. btnPause.ShowWindow(SW_SHOW);
  346. }
  347. }
  349. void CNimdaDlg::OnPause() 
  350. {
  351. if (GetText(btnPause)=="Pause") {
  352. btnPause.SetWindowText("Resume");
  353. Pause();
  354. }
  355. else {
  356. btnPause.SetWindowText("Pause");
  357. Continue();
  358. }
  359. }
  361. void CNimdaDlg::OnViewlog() 
  362. {
  363. ShellExecute(NULL,"open","notepad","c:\anti-nimba.log",NULL,SW_SHOWNORMAL);
  364. }