防火墙与安全工具

开发平台：
Visual C++

AccessCtrlPage.cpp：源码内容
							// AccessCtrlPage.cpp : implementation file
//
#include "stdafx.h"
#include "MyFireWall.h"
#include "AccessCtrlPage.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
extern CMyFireWallApp theApp;
/////////////////////////////////////////////////////////////////////////////
// CAccessCtrlPage property page
IMPLEMENT_DYNCREATE(CAccessCtrlPage, CPropertyPage)
CAccessCtrlPage::CAccessCtrlPage() : CPropertyPage(CAccessCtrlPage::IDD)
{
	//{{AFX_DATA_INIT(CAccessCtrlPage)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
}
CAccessCtrlPage::~CAccessCtrlPage()
{
}
void CAccessCtrlPage::DoDataExchange(CDataExchange* pDX)
{
	CPropertyPage::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAccessCtrlPage)
	DDX_Control(pDX, IDC_ACCESS_CTRL_LIST, m_AccessCtrlList);
	//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CAccessCtrlPage, CPropertyPage)
	//{{AFX_MSG_MAP(CAccessCtrlPage)
	ON_NOTIFY(NM_DBLCLK, IDC_ACCESS_CTRL_LIST, OnDblclkAccessCtrlList)
	ON_NOTIFY(NM_RCLICK, IDC_ACCESS_CTRL_LIST, OnRclickAccessCtrlList)
	ON_COMMAND(IDM_ACL_DELETE, OnAclDeleteRule)
	ON_COMMAND(IDM_ACL_MODIFY, OnAclModifyRule)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CAccessCtrlPage message handlers
BOOL CAccessCtrlPage::OnInitDialog() 
{
	CPropertyPage::OnInitDialog();
	
	// TODO: Add extra initialization here
	// 初始化 AccessCtrlList 的设置和风格
	m_AccessCtrlList.InsertColumn(0,L"程序名称",LVCFMT_LEFT,sizeof(L"应用程序")*12,-1);
	m_AccessCtrlList.InsertColumn(1,L"状态",LVCFMT_LEFT,sizeof(L"动作")*8,-1);
	m_AccessCtrlList.InsertColumn(2,L"类型",LVCFMT_LEFT,sizeof(L"类型")*6,-1);
	m_AccessCtrlList.InsertColumn(3,L"应用程序路径",LVCFMT_LEFT,sizeof(L"应用程序路径")*18,-1);
	m_AccessCtrlList.InsertColumn(4,L"描述",LVCFMT_LEFT,sizeof(L"描述")*18,-1);
	m_AccessCtrlList.ModifyStyle(LVS_TYPEMASK,LVS_REPORT);
	m_AccessCtrlList.SetExtendedStyle(LVS_EX_GRIDLINES | LVS_EX_FULLROWSELECT);
	
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}
void CAccessCtrlPage::OnDblclkAccessCtrlList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	// TODO: Add your control notification handler code here
	LPNMLISTVIEW pNMListView = (NM_LISTVIEW *)pNMHDR;
	
    //是否点击了有效的项目。
    if (pNMListView->iItem != -1)	//点击了有效的项目
    {
        if (pNMListView->iSubItem == 1)	// 双击列1时，修改应用程序的状态，允许 --> 禁止  或者  禁止 --> 允许
        {
			OnAclModifyRule();
        }
    }
	*pResult = 0;
}
void CAccessCtrlPage::OnRclickAccessCtrlList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	// TODO: Add your control notification handler code here
	CMenu menu,*pSubMenu;
	menu.LoadMenu(IDR_ACL_MENU);
	pSubMenu=menu.GetSubMenu(0);
	
	// 根据应用程序的状态来修改右键菜单的内容
	CString strAction;
	strAction = m_AccessCtrlList.GetItemText(m_AccessCtrlList.GetNextItem(-1,LVNI_ALL|LVNI_SELECTED),1);
	if (strAction == " 允许 ")
	{
		pSubMenu->ModifyMenu(IDM_ACL_MODIFY, MF_BYCOMMAND, IDM_ACL_MODIFY, L" 禁止 ");
	} 
	else
	{
		pSubMenu->ModifyMenu(IDM_ACL_MODIFY, MF_BYCOMMAND, IDM_ACL_MODIFY, L" 允许 ");
	}
	
	CPoint point;
	GetCursorPos(&point);
	
	pSubMenu->TrackPopupMenu(TPM_LEFTALIGN,point.x,point.y,this);
	*pResult = 0;
}
void CAccessCtrlPage::OnAclModifyRule() 
{
	// TODO: Add your command handler code here
	TCHAR szAppPath[MAX_PATH];
	// 得到当前选中的进程路径
	_tcscpy(szAppPath, m_AccessCtrlList.GetItemText(m_AccessCtrlList.GetNextItem(-1,LVNI_ALL|LVNI_SELECTED),3));
	
	// 修改LSP过滤规则
	if(g_RuleFile.ModifyLspRule(szAppPath) == FALSE)
	{
		return ;
	}
	
	// 应用到Dll层
	theApp.ApplyLspRules();
	
	char szContent[256];
	USES_CONVERSION;
	// 修改 AccessCtrlList 上的显示
	CString strAction;
	strAction = m_AccessCtrlList.GetItemText(m_AccessCtrlList.GetNextItem(-1,LVNI_ALL|LVNI_SELECTED),1);
	if (strAction == " 允许 ")
	{
		strAction = " 禁止 ";
		sprintf(szContent, "将程序 %s 对应的应用层规则状态修改为：禁止。", W2A(GetFileName(szAppPath)));
	} 
	else
	{
		strAction = " 允许 ";
		sprintf(szContent, "将程序 %s 对应的应用层规则状态修改为：允许。", W2A(GetFileName(szAppPath)));
	}
	m_AccessCtrlList.SetItemText(m_AccessCtrlList.GetNextItem(-1,LVNI_ALL|LVNI_SELECTED), 1, strAction);
	// 保存日志
	SaveLogFile(szContent);
	
}
void CAccessCtrlPage::OnAclDeleteRule()
{
	// TODO: Add your command handler code here
	// 获得选中的行号
	int nItem = m_AccessCtrlList.GetNextItem(-1,LVNI_ALL|LVNI_SELECTED);
	
	// 删除规则
	g_RuleFile.DelLspRule(nItem);
	
	// 将规则保存到文件
	g_RuleFile.SaveRules();
	
	// 将规则在 AccessCtrlList 控件上显示出来
	::PostMessage(g_hMainWnd, UPDATE_ACCESS_CTRL_LIST, 0, 0);
	
	// 将规则应用到Dll模块
	theApp.ApplyLspRules();
	// 保存日志文件
	TCHAR szAppPath[MAX_PATH];
	// 得到当前选中的进程路径
	_tcscpy(szAppPath, m_AccessCtrlList.GetItemText(nItem,3));
	char szContent[256];
	USES_CONVERSION;
	sprintf(szContent, "删除程序 %s 对应的应用层过滤规则。", W2A(GetFileName(szAppPath)));
	SaveLogFile(szContent);
}

						