防火墙与安全工具

开发平台：

Visual C++

PIOControl.cpp：源码内容

							// PIOControl.cpp
#include "stdafx.h"
#include "PIOControl.h"
CPIOControl::CPIOControl()
{
	m_fnIoControl = NULL;
	// 加载DLL模块，获取PLSPIoControl函数的指针
	TCHAR szPathName[256];
	wsprintf(szPathName, _T("%s%s"), strCurrentPath, _T("MyLsp.dll"));
	m_hLSPModule = ::LoadLibrary(szPathName);
	if(m_hLSPModule != NULL)
	{
		m_fnIoControl = (PFNLSPIoControl)::GetProcAddress(m_hLSPModule, "PLSPIoControl");
	}
		
	if(m_fnIoControl == NULL)
	{
		::MessageBox(NULL, _T("Can not find LSP module"), _T("error"), 0);
		exit(0);
	}
}
CPIOControl::~CPIOControl()
{
	if(m_hLSPModule != NULL)
		::FreeLibrary(m_hLSPModule);
}
void CPIOControl::SetWorkMode(int nWorkMode)
{
	// 设置工作模式
	m_IoControl.ucWorkMode = nWorkMode;
	m_fnIoControl(&m_IoControl, IO_CONTROL_SET_WORK_MODE);
}
int CPIOControl::GetWorkMode()
{
	// 获取工作模式
	return m_fnIoControl(&m_IoControl, IO_CONTROL_GET_WORK_MODE);
}
void CPIOControl::SetRuleFile(RULE_FILE_HEADER *pHeader, RULE_ITEM *pRules)
{
	// 申请一个临时的RULE_FILE对象
	RULE_FILE RuleFile;
	memcpy(&RuleFile.header, pHeader, sizeof(RULE_FILE_HEADER));
	memcpy(&RuleFile.LspRules, pRules, sizeof(RULE_ITEM) * pHeader->ulLspRuleCount);
	// 设置过滤规则
	m_IoControl.pRuleFile = &RuleFile;
	m_fnIoControl(&m_IoControl, IO_CONTROL_SET_RULE_FILE);
}
void CPIOControl::SetInstance(HWND hWnd, TCHAR *pszPathName)
{
	// 设置实例句柄
	m_IoControl.hWnd = hWnd;
	wcscpy(m_IoControl.szPath, pszPathName);
	m_fnIoControl(&m_IoControl, IO_CONTROL_SET_INSTANCE);
}
void CPIOControl::GetSession(SESSION *pSession, int nIndex)
{
	// 获取有事件发生的会话
	m_IoControl.pSession = pSession;
	m_IoControl.nSessionIndex = nIndex;
	m_fnIoControl(&m_IoControl, IO_CONTROL_GET_SESSION);
}
void CPIOControl::SetQuerySession(int nIndex, UCHAR ucAction)
{
	// 返回查询结果
	m_IoControl.nSessionIndex = nIndex;
	m_IoControl.ucWorkMode = ucAction;
	m_fnIoControl(&m_IoControl, IO_CONTROL_SET_QUERY_SESSION);
}
void CPIOControl::GetQuerySession(SESSION *pSession, int nIndex)
{
	// 获取查询的应用程序
	m_IoControl.pSession = pSession;
	m_IoControl.nSessionIndex = nIndex;
	m_fnIoControl(&m_IoControl, IO_CONTROL_GET_QUERY_SESSION);
}

						