piocontrol.cpp

Windows网络与通信程序设计/王艳平, 张越编著

#include "stdafx.h"

#include "PIOControl.h"


CPIOControl::CPIOControl()
{
	m_fnIoControl = NULL;

	// 加载DLL模块，获取PLSPIoControl函数的指针
	TCHAR szPathName[256];
	TCHAR* p;
	if(::GetFullPathName(PHOENIX_SERVICE_DLL_NAME, 256, szPathName, &p) != 0)
	{
		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::SetPhoenixInstance(HWND hWnd, TCHAR *pszPathName)
{
	// 设置实例句柄
	m_IoControl.hPhoenixWnd = hWnd;
	wcscpy(m_IoControl.szPath, pszPathName);
	m_fnIoControl(&m_IoControl, IO_CONTROL_SET_PHOENIX_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::SetQueryApp(int nIndex, BOOL bPass)
{
	// 返回查询结果
	m_IoControl.nSessionIndex = nIndex;
	m_IoControl.ucWorkMode = bPass;
	m_fnIoControl(&m_IoControl, IO_CONTROL_SET_QUERY_SESSION);
}

LPCTSTR CPIOControl::GetQueryApp(int nIndex)
{
	// 获取查询的应用程序
	m_IoControl.nSessionIndex = nIndex;
	m_fnIoControl(&m_IoControl, IO_CONTROL_GET_QUERY_SESSION);
	return m_IoControl.szPath;
}