编写简单的 NT 式驱动程序的加载与卸载工具

写驱动的加载需要用到五个函数:

OpenSCManager()

CreateService()

OpenService()

StartService()

CloseServiceHandle()

这五个函数的作用和用法,我在代码中会进行说明。

正常加载驱动的步骤如下:

1、调用 OpenSCManager 这个 Win32 API 函数来打开 SCM 服务管理器,如果返回值为:NULL,则表示打开失败,否则表示打开成功并继续下一步执行。

2、调用 CreateService 这个 Win32 API 函数来创建服务,然后用 GetLastError 函数获取返回值,如果返回值为:ERROR_IO_PENDING,则说明服务已经创建过。

3、然后,此时用 OpenService 这个 Win32 API 函数打开此服务。

4、调用 StartService 这个 Win32 API 函数来开启此服务。

5、最后调用 CloseServiceHandle() 对错误信息进行一个简单的处理。

当我点击加载驱动按钮后,弹出打开文件对话框,然后选择驱动文件,进行加载这个过程我写在 Load Driver 按钮的单击事件中。

下面看代码:

static CStringW DriverName;  //保存驱动名称的全局变量
CEdit * pEditA;
CString CirLf,AddEditSrt;

BOOL LoadNTDriver(WCHAR* pDriverName, WCHAR* pDriverPathName);
void CLoadDriverToolsDlg::On_btnLoadD_Clicked()
{
CEdit * pEdit =(CEdit*)GetDlgItem(IDC_txtGetSysName); //获取 EDIT 控件的句柄,并转换成 CEdit 控件类指针
pEditA = pEdit;
CirLf = ".\r\n"; // \r\n表示换行 WCHAR *Filters = L"All File(*.*)|*.*|Driver File(*.sys)|*.sys|";
WCHAR *DefaultFileter = L"All File(*.*)|*.*|";
WCHAR *DefaultFileName = L"请选择驱动文件"; /*CFileDialog 类封装了Windows常用的文件对话框。常用的文件对话框提供了文件打开和文件存盘对话框功能。
DoModal 代表这个窗口变成当前窗口,而且此函数一直阻塞知道你点OK或者Cancel退出,如果点的时候OK,该函数返回值为IDOK。
IDOK 按钮默认响应CDialog::OnOK();这个虚函数。这个函数调用EndDialog(IDOK);EndDialog设置对话框返回值,IDOK就是返回值。*/
CFileDialog FileDlg(true, DefaultFileter, DefaultFileName, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, Filters, NULL);
if (FileDlg.DoModal() == IDOK){
DriverName = FileDlg.GetFileName();
pEditA->SetWindowTextW(AddEditSrt += L"打开的驱动名称为:" + DriverName + CirLf); } /*能GetBuffer函数,说明这个函数是为一个CString对象,可重新获取其内部字符缓冲区的指针,
返回的LPTSTR为非const的,从而允许直接修改CString中的内容,里面的参数表示缓冲区的大小*/
WCHAR *DriverNameBuffer = DriverName.GetBuffer(DriverName.GetLength());
BOOL Status =LoadNTDriver(DriverNameBuffer, FileDlg.GetPathName().GetBuffer());
if (Status == false){
printf("服务启动失败!!错误代码为:%d \n", GetLastError());
MessageBox(L"服务启动失败!!", L"消息框", MB_OK);
}
else{
printf("服务启动成功! \n");
MessageBox(L"服务启动成功!", L"消息框", MB_OK);
Status = true;
} }
 

这里有个自定义的驱动加载函数,并传递了两个参数,一个驱动名称,一个驱动路径。而整个加载的过程,可以写在头文件里面,我这里就直接写在源文件中,

下面看代码:

BOOL LoadNTDriver(WCHAR* pDriverName, WCHAR* pDriverPathName){
BOOL Status = false;
SC_HANDLE hServiceManager = NULL;
SC_HANDLE hServiceDriver = NULL; //打开服务管理器
hServiceManager = OpenSCManager(
NULL, //指定目标计算机的名称。如果该指针为NULL ,就连接到本地主机的服务控制器。
SERVICES_ACTIVE_DATABASE, //指定将要打开的服务控制管理数据库的名称。
SC_MANAGER_CREATE_SERVICE //如果该指针为NULL ,则打开默认的 SERVICES_ACTIVE_DATABASE
); //返回值为服务管理器的句柄
if (hServiceManager != NULL){ /*TRACE 宏有点象C语言中用的Printf函数,使程序在运行中输出一些调试信息。
但有一点不同的是:TRACE 宏只有在调试状态下才有所输出,而以前用的Printf 函数在任何情况下都有输出。*/
printf("打开服务管理器成功! \n");
pEditA->SetWindowTextW(AddEditSrt += L"打开服务管理器成功!" + CirLf); //创建服务
hServiceDriver = CreateService(
hServiceManager, //服务管理器的句柄,由系统函数OpenSCManager 返回
pDriverName, //驱动程序在注册表中的名称,最大字符串长度为 256 个字符
pDriverName, //被用户界面程序用来识别服务的显示名称。此字符串具有最大长度为 256 个字符。
SERVICE_ALL_ACCESS, //加载驱动程序时所具备的访问权限
SERVICE_KERNEL_DRIVER, //服务类型,此枚举表示加载的服务是驱动程序
SERVICE_DEMAND_START, //服务启动选项,此枚举表示由服务控制管理器(SCM)启动的服务,也就是需手动启动。
SERVICE_ERROR_IGNORE, //当该启动服务失败时产生错误及采取的保护措施, 此枚举表示服务启动程序将忽略该错误并返回继续执行。
pDriverPathName, //注册表中驱动程序所在的路径
NULL,
NULL,
NULL,
NULL,
NULL
); //返回值为服务的句柄
if (hServiceDriver != NULL){
printf("创建服务成功! \n");
pEditA->SetWindowTextW(AddEditSrt += L"创建服务成功!" + CirLf); //打开创建的服务
hServiceDriver = OpenService(
hServiceManager, //服务管理器的句柄
pDriverName, //服务的名称
SERVICE_ALL_ACCESS //打开服务所需的访问权限,此枚举为完全权限
); //返回值为服务的句柄
if (hServiceDriver != NULL){
printf("打开服务成功! \n");
pEditA->SetWindowTextW(AddEditSrt += L"打开服务成功!" + CirLf); //启动创建的服务
INT reValue = StartService(
hServiceDriver, //要打开的服务的服务句柄
NULL, //服务向量数组中的字符串数量,如果字符串数量为空,该参数可以是零。
NULL //字符串被传递给服务的ServiceMain函数作为参数。如果没有参数,可以为空。
); //如果启动成功返回非零的数,否则返回零
if (reValue != ){
TRACE("启动服务成功! \n");
pEditA->SetWindowTextW(AddEditSrt += L"启动服务成功!" + CirLf);
Status = true;
return Status;
}
else{
printf("StartService() Fild %d ! \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"启动服务失败!!错误代码为:" + GetLastError() + CirLf);
goto a; //出现错误后,调到a处,进行处理
}
}
else{
printf("OpenService() Fild %d ! \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"打开服务失败!!错误代码为:" + GetLastError() + CirLf);
goto a;
}
}
else{
printf("CreateService() Fild %d ! \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"创建服务失败!!错误代码为:" + GetLastError() + CirLf);
goto a;
}
}
else{
printf("OpenSCManager() Fild %d ! \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"打开服务管理器失败!!错误代码为:" + GetLastError() + CirLf);
goto a;
} //用 CloseServiceHandle 函数来关闭服务管理器或服务
a: if (hServiceManager){ //如果打开了服务管理器,就关闭服务管理器
CloseServiceHandle(hServiceManager);
}
if (hServiceDriver){ //如果打开了服务,就关闭服务
CloseServiceHandle(hServiceDriver);
}
return Status;
}

好了,加载驱动的过程以编写完毕,下面看下效果:

编写简单的 NT 式驱动程序的加载与卸载工具

编写简单的 NT 式驱动程序的加载与卸载工具

编写简单的 NT 式驱动程序的加载与卸载工具

驱动加载并成功启动!!

下面我们写一个卸载驱动的过来,让这个工具能加载也能卸载!

卸载驱动的过程需要用到四个函数:

OpenSCManager()     //打开服务管理器

OpenService()             //打开要卸载的服务

ControlService()        //发送控制命令来停止要卸载的服务

DeleteService()        //删除要卸载的服务

同样,当我点击卸载驱动按钮后,就进行卸载,这个过程我写在 Unload Driver 按钮的单击事件中。

下面看代码:

BOOL UnloadNTDriver(WCHAR* lpDriverName);
void CLoadDriverToolsDlg::On_btnUnloadD_Clicked()
{ BOOL Status = UnloadNTDriver(DriverName.GetBuffer(DriverName.GetLength()));
if (Status = true){
printf("服务卸载成功!\n");
MessageBox(L"服务卸载成功!", L"消息框", MB_OK);
}
else{
printf("服务卸载失败!!错误代码为:%d \n",GetLastError());
MessageBox(L"服务卸载失败!!", L"消息框", MB_OK);
}
}

这里有个自定义的驱动卸载函数,并传递了一个参数,一个驱动名称的缓冲区指针。而整个卸载过程,可以写在头文件里面,我这里就直接写在源文件中,

下面看代码:

BOOL UnloadNTDriver(WCHAR* lpDriverName){
BOOL Status = false;
SC_HANDLE hServiceManager = NULL;
SC_HANDLE hServiceDriver = NULL;
SERVICE_STATUS SaveValue; // SERVICE_STATUS 显示了服务器上运行的服务状态信息。 //打开服务管理器
hServiceManager = OpenSCManager(NULL, SERVICES_ACTIVE_DATABASE, SC_MANAGER_CREATE_SERVICE);
if (hServiceManager != NULL){
printf("服务管理器打开成功!\n");
pEditA->SetWindowTextW(AddEditSrt += L"服务管理器打开成功!" + CirLf); //打开服务
hServiceDriver = OpenService(hServiceManager, lpDriverName, SERVICE_ALL_ACCESS);
if (hServiceDriver != NULL){
printf("服务打开成功!\n");
pEditA->SetWindowTextW(AddEditSrt += L"服务打开成功!" + CirLf);
int StatusValue; //想要顺利卸载驱动,必先停止驱动,所以用ControlService函数发送控制命令.
StatusValue = ControlService(hServiceDriver, SERVICE_CONTROL_STOP, &SaveValue);
if (StatusValue > ){
printf("服务停止成功!\n");
pEditA->SetWindowTextW(AddEditSrt += L"服务停止成功!" + CirLf);
/*我们把 ControlService 函数读出的 SERVICE_STATUS 服务状态信息打印出来看看,
上面,我们以保存在 SaveValue 这个变量中了,所以现在只需读取这个变量就行。*/
printf("停止服务的过程已经执行了,那现在的状态是:=%d \n", &SaveValue); //卸载驱动,此函数从服务控制管理器数据库删除指定的服务
StatusValue = DeleteService(hServiceDriver);
if (StatusValue > ){
printf("服务卸载成功!\n");
pEditA->SetWindowTextW(AddEditSrt += L"服务卸载成功!" + CirLf);
Status = true;
return Status;
}
else{
printf("服务卸载失败!!错误代码为:=%d \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"服务卸载失败!!错误代码为:" + GetLastError() + CirLf);
goto b;
}
}
else{
printf("服务停止失败!!错误代码为:=%d \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"服务停止失败!!错误代码为:" + GetLastError() + CirLf);
goto b;
}
}
else{
printf("服务打开失败!!错误代码为:=%d \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"服务打开失败!!错误代码为:" + GetLastError() + CirLf);
goto b;
}
}
else{
printf("服务管理器打开失败!!错误代码为:=%d \n", GetLastError());
pEditA->SetWindowTextW(AddEditSrt += L"服务管理器打开失败!!错误代码为:" + GetLastError() + CirLf);
goto b;
} //如果出现错误,就对句柄进行处理
b: if (hServiceManager){
CloseServiceHandle(hServiceManager);
}
if (hServiceDriver){
CloseServiceHandle(hServiceDriver);
}
return Status;
}

大功告成!!验证结果见下图:

编写简单的 NT 式驱动程序的加载与卸载工具

上一篇:【P1813】8的倍数


下一篇:demo