应用程序与驱动程序通信的源码
- 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
- 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
- 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。
与驱动程序通信的源码
///////////////////////////////////////////////////
// Driver.h文件
#ifndef __DRIVER_H__
#define __DRIVER_H__
#include
class CDriver
{
public:
// 构造函数和析构函数
// 构造函数,pszDriverPath为驱动所在目录,pszLinkName为符号连接名字
// 在类的构造函数中,将试图创建或打开服务,
CDriver(LPCTSTR pszDriverPath, LPCTSTR pszLinkName);
// 析构函数。在这里,将停止服务,
virtual ~CDriver();
// 属性
// 此驱动是否可用
virtual BOOL IsValid() { return (m_hSCM != NULL && m_hService != NULL); }
// 操作
// 开启服务。也就是说驱动的DriverEntry函数将被调用
virtual BOOL StartDriver();
// 结束服务。即驱动程序的DriverUnload例程将被调用
virtual BOOL StopDriver();
// 打开设备,即取得到此驱动的一个句柄
virtual BOOL OpenDevice();
// 向设备发送控制代码
virtual DWORD IoControl(DWORD nCode, PVOID pInBuffer,
DWORD nInCount, PVOID pOutBuffer, DWORD nOutCount);
// 实现
protected:
char m_szLinkName[56]; // 符号连接名称
BOOL m_bStarted; // 指定服务是否启动
BOOL m_bCreateService; // 指定是否创建了服务
HANDLE m_hSCM; // SCM数据库句柄
HANDLE m_hService; // 服务句柄
HANDLE m_hDriver; // 设备句柄
};
CDriver::CDriver(LPCTSTR pszDriverPath, LPCTSTR pszLinkName)
{
strncpy(m_szLinkName, pszLinkName, 55);
m_bStarted = FALSE;
m_bCreateService = FALSE;
m_hSCM = m_hService = NULL;
m_hDriver = INVALID_HANDLE_VALUE;
// 打开SCM管理器
m_hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if(m_hSCM == NULL)
{
MessageBox(0, "打开服务控制管理器失败\n",
"可能是因为您不拥有Administrator权限\n", 0);
return;
}
// 创建或打开服务
m_hService = ::CreateService(m_hSCM, m_szLinkName, m_szLinkName, SERVICE_ALL_ACCESS, SERVICE_KERNEL_DRIVER, SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL,
pszDriverPath, NULL, 0, NULL, NULL, NULL);
if(m_hService == NULL)
{
// 创建服务失败,可能是因为服务已经存在,所以还要试图打开它
int nError = ::GetLastError();
if(nError == ERROR_SERVICE_EXISTS || nError == ERROR_SERVICE_MARKED_FOR_DELETE)
{
m_hService = ::OpenService(m_hSCM, m_szLinkName, SERVICE_ALL_ACCESS);
}
}
else
{
m_bCreateService = TRUE;
}
}
CDriver::~CDriver()
{
// 关闭设备句柄
if(m_hDriver != INVALID_HANDLE_VALUE)
::CloseHandle(m_hDriver);
// 如果创建了服务,就将之删除
if(m_bCreateService)
{
StopDriver();
::DeleteService(m_hService);
}
// 关闭句柄
if(m_hService != NULL)
::CloseServiceHandle(m_hService);
if(m_hSCM != NULL)
::CloseServiceHandle(m_hSCM);
}
BOOL CDriver::StartDriver()
{
if(m_bStarted)
return TRUE;
if(m_hService == NULL)
return FALSE;
// 启动服务
if(!::StartService(m_hService, 0, NULL))
{
int nError = ::GetLastError();
if(nError == ERROR_SERVICE_ALREADY_RUNNING)
m_bStarted = TRUE;
else
::DeleteService(m_hService);
}
else
{
// 启动成功后,等待服务进入运行状态
int nTry = 0;
SERVICE_STATUS ss;
::QueryServiceStatus(m_hService, &ss);
while(ss.dwCurrentState == SERVICE_START_PENDING && nTry++ < 80)
{
::Sleep(50);
::QueryServiceStatus(m_hService, &ss);
}
if(ss.dwCurrentState == SERVICE_RUNNING)
m_bStarted = TRUE;
}