vx_antivirusdlg.cpp

实现了屏幕截取

// Vx_AntiVirusDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "Vx_AntiVirus.h"
#include "Vx_AntiVirusDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CVx_AntiVirusDlg 对话框

CVx_AntiVirusDlg::CVx_AntiVirusDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CVx_AntiVirusDlg::IDD, pParent)
	, bScanMemory(TRUE)
	, bScanAllFile(TRUE)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CVx_AntiVirusDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_LIST1, m_List);
	DDX_Control(pDX, IDC_PROCESS, m_ProgressCtrl);
	DDX_Check(pDX, IDC_CHECK2, bScanMemory);
	DDX_Check(pDX, IDC_CHECK1, bScanAllFile);
}

BEGIN_MESSAGE_MAP(CVx_AntiVirusDlg, CDialog)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_WM_DESTROY()
	ON_BN_CLICKED(IDOK, &CVx_AntiVirusDlg::OnBnClickedOk)
	ON_BN_CLICKED(ID_STOPKILL, &CVx_AntiVirusDlg::OnBnClickedStopkill)
	ON_BN_CLICKED(IDC_STARTKILL, &CVx_AntiVirusDlg::OnBnClickedStartkill)
	ON_BN_CLICKED(ID_FINVIRUS, &CVx_AntiVirusDlg::OnBnClickedFinvirus)
	ON_BN_CLICKED(IDC_CHECK1, &CVx_AntiVirusDlg::OnBnClickedCheck1)
	ON_BN_CLICKED(IDC_CHECK2, &CVx_AntiVirusDlg::OnBnClickedCheck2)
END_MESSAGE_MAP()


// CVx_AntiVirusDlg 消息处理程序

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

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	ShowWindow(SW_MINIMIZE);

	InitGUI();
	AdjustProcess();

	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CVx_AntiVirusDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		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;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
//
HCURSOR CVx_AntiVirusDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CVx_AntiVirusDlg::OnDestroy()
{
	m_ProgressCtrl.End();
	StopKill();
	CDialog::OnDestroy();
}

void CVx_AntiVirusDlg::InitGUI()
{
	m_List.SetExtendedStyle(LVS_EX_GRIDLINES|LVS_EX_FULLROWSELECT);
	m_List.InsertColumn(0,"路径",LVCFMT_LEFT,260);
	m_List.InsertColumn(1,"区域",LVCFMT_LEFT,40);
	m_List.InsertColumn(2,"病毒",LVCFMT_LEFT,40);
	m_List.InsertColumn(3,"状态",LVCFMT_LEFT,60);
	m_ProgressCtrl.SetSpeed(50);
}
void CVx_AntiVirusDlg::SetStatusText(char* textstr)
{
	if( !textstr ) return;
	SetDlgItemText(IDC_STATUS,textstr);
}
void CVx_AntiVirusDlg::SetCurText(char* textstr)
{
	if( !textstr ) return;
	SetDlgItemText(IDC_CURTEXT,textstr);
}
int CVx_AntiVirusDlg::InsertText(char* strPath,char* area,char* strisname,char* strstatus,DWORD itemdata)
{
	if( !strPath || !area || !strisname || !strstatus )
		return -1;
	int ret = m_List.InsertItem(m_List.GetItemCount(),strPath);
	m_List.SetItemText(ret,1,area);
	m_List.SetItemText(ret,2,strisname);
	m_List.SetItemText(ret,3,strstatus);
	m_List.SetItemData(ret,itemdata);
	return ret;
}
void CVx_AntiVirusDlg::SetText(int idx,char* str)
{
	if( idx == -1 || !str ) return;
	m_List.SetItemText(idx,3,str);
}

void CVx_AntiVirusDlg::OnBnClickedOk()
{
	m_ProgressCtrl.End();
	StopKill();
	OnOK();
}

void CVx_AntiVirusDlg::OnBnClickedFinvirus()
{
	GetDlgItem(ID_FINVIRUS)->EnableWindow(FALSE);
	GetDlgItem(IDC_STARTKILL)->EnableWindow(FALSE);
	GetDlgItem(ID_STOPKILL)->EnableWindow(TRUE);
	StopKill();
	m_ProgressCtrl.Start();
	SetStatusText("正在查毒");
	StartKill(this,TRUE);
}

void CVx_AntiVirusDlg::OnBnClickedStartkill()
{
	GetDlgItem(ID_FINVIRUS)->EnableWindow(FALSE);
	GetDlgItem(IDC_STARTKILL)->EnableWindow(FALSE);
	GetDlgItem(ID_STOPKILL)->EnableWindow(TRUE);
	StopKill();
	m_ProgressCtrl.Start();
	SetStatusText("正在杀毒");
	StartKill(this,FALSE);
}

void CVx_AntiVirusDlg::OnBnClickedStopkill()
{
	GetDlgItem(ID_FINVIRUS)->EnableWindow(TRUE);
	GetDlgItem(IDC_STARTKILL)->EnableWindow(TRUE);
	GetDlgItem(ID_STOPKILL)->EnableWindow(FALSE);

	m_ProgressCtrl.End();
	m_ProgressCtrl.Reset();
	StopKill();
}
void CVx_AntiVirusDlg::OnBnClickedCheck1()
{
	UpdateData(TRUE);
}

void CVx_AntiVirusDlg::OnBnClickedCheck2()
{
	UpdateData(TRUE);
}