[C++] 纯文本查看 复制代码 // 安装驱动
BOOL installDvr(CONST WCHAR drvPath[50], CONST WCHAR serviceName[20]) {
// 打开服务控制管理器数据库
SC_HANDLE schSCManager = OpenSCManager(
NULL, // 目标计算机的名称,NULL:连接本地计算机上的服务控制管理器
NULL, // 服务控制管理器数据库的名称,NULL:打开 SERVICES_ACTIVE_DATABASE 数据库
SC_MANAGER_ALL_ACCESS // 所有权限
);
if (schSCManager == NULL) {
CloseServiceHandle(schSCManager);
return FALSE;
}
// 创建服务对象,添加至服务控制管理器数据库
SC_HANDLE schService = CreateService(
schSCManager, // 服务控件管理器数据库的句柄
serviceName, // 要安装的服务的名称
serviceName, // 用户界面程序用来标识服务的显示名称
SERVICE_ALL_ACCESS, // 对服务的访问权限:所有全权限
SERVICE_KERNEL_DRIVER, // 服务类型:驱动服务
SERVICE_DEMAND_START, // 服务启动选项:进程调用 StartService 时启动
SERVICE_ERROR_IGNORE, // 如果无法启动:忽略错误继续运行
drvPath, // 驱动文件绝对路径,如果包含空格需要多加双引号
NULL, // 服务所属的负载订购组:服务不属于某个组
NULL, // 接收订购组唯一标记值:不接收
NULL, // 服务加载顺序数组:服务没有依赖项
NULL, // 运行服务的账户名:使用 LocalSystem 账户
NULL // LocalSystem 账户密码
);
if (schService == NULL) {
CloseServiceHandle(schService);
CloseServiceHandle(schSCManager);
return FALSE;
}
CloseServiceHandle(schService);
CloseServiceHandle(schSCManager);
return TRUE;
}
// 启动服务
BOOL startDvr(CONST WCHAR serviceName[20]) {
// 打开服务控制管理器数据库
SC_HANDLE schSCManager = OpenSCManager(
NULL, // 目标计算机的名称,NULL:连接本地计算机上的服务控制管理器
NULL, // 服务控制管理器数据库的名称,NULL:打开 SERVICES_ACTIVE_DATABASE 数据库
SC_MANAGER_ALL_ACCESS // 所有权限
);
if (schSCManager == NULL) {
CloseServiceHandle(schSCManager);
return FALSE;
}
// 打开服务
SC_HANDLE hs = OpenService(
schSCManager, // 服务控件管理器数据库的句柄
serviceName, // 要打开的服务名
SERVICE_ALL_ACCESS // 服务访问权限:所有权限
);
if (hs == NULL) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
if (StartService(hs, 0, 0) == 0) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return TRUE;
}
// 停止服务
BOOL stopDvr(CONST WCHAR serviceName[20]) {
// 打开服务控制管理器数据库
SC_HANDLE schSCManager = OpenSCManager(
NULL, // 目标计算机的名称,NULL:连接本地计算机上的服务控制管理器
NULL, // 服务控制管理器数据库的名称,NULL:打开 SERVICES_ACTIVE_DATABASE 数据库
SC_MANAGER_ALL_ACCESS // 所有权限
);
if (schSCManager == NULL) {
CloseServiceHandle(schSCManager);
return FALSE;
}
// 打开服务
SC_HANDLE hs = OpenService(
schSCManager, // 服务控件管理器数据库的句柄
serviceName, // 要打开的服务名
SERVICE_ALL_ACCESS // 服务访问权限:所有权限
);
if (hs == NULL) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
// 如果服务正在运行
SERVICE_STATUS status;
if (QueryServiceStatus(hs, &status) == 0) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
if (status.dwCurrentState != SERVICE_STOPPED &&
status.dwCurrentState != SERVICE_STOP_PENDING
) {
// 发送关闭服务请求
if (ControlService(
hs, // 服务句柄
SERVICE_CONTROL_STOP, // 控制码:通知服务应该停止
&status // 接收最新的服务状态信息
) == 0) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
// 判断超时
INT timeOut = 0;
while (status.dwCurrentState != SERVICE_STOPPED) {
timeOut++;
QueryServiceStatus(hs, &status);
Sleep(50);
}
if (timeOut > 80) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
}
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return TRUE;
}
// 卸载驱动
BOOL unloadDvr(CONST WCHAR serviceName[20]) {
// 打开服务控制管理器数据库
SC_HANDLE schSCManager = OpenSCManager(
NULL, // 目标计算机的名称,NULL:连接本地计算机上的服务控制管理器
NULL, // 服务控制管理器数据库的名称,NULL:打开 SERVICES_ACTIVE_DATABASE 数据库
SC_MANAGER_ALL_ACCESS // 所有权限
);
if (schSCManager == NULL) {
CloseServiceHandle(schSCManager);
return FALSE;
}
// 打开服务
SC_HANDLE hs = OpenService(
schSCManager, // 服务控件管理器数据库的句柄
serviceName, // 要打开的服务名
SERVICE_ALL_ACCESS // 服务访问权限:所有权限
);
if (hs == NULL) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
// 删除服务
if (DeleteService(hs) == 0) {
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return FALSE;
}
CloseServiceHandle(hs);
CloseServiceHandle(schSCManager);
return TRUE;
}
int main()
{
if (installDvr(L"procexp64.sys", L"我的驱动名称") == TRUE) {
printf("安装驱动成功!\n");
}
if (startDvr(L"我的驱动名称") == TRUE) {
printf("启动驱动成功!\n");
}
if (stopDvr(L"我的驱动名称") == TRUE) {
printf("停止驱动成功!\n");
}
if (unloadDvr(L"我的驱动名称") == TRUE) {
printf("卸载驱动成功!\n");
}
getchar();
return 0;
}
|