全球作弊论坛  
  
返回列表 发新帖
查看: 387|回复: 0

驱动控制类

[复制链接]
river

2

主题

0

回帖

15

积分

凡人

Rank: 1

积分
15
发表于 2024-5-10 14:15:19 | 显示全部楼层 |阅读模式
本帖最后由 river 于 2024-5-10 14:17 编辑

Driver.h
  1. #pragma once
  2. #include "Alastor.h"

  4. namespace Alastor
  5. {
  6.         class ALASTOR_API Driver {
  7.         public:

  9.                 Driver();
  10.                 ~Driver();


  13.                 bool load(
  14.                         std::string path,
  15.                         std::string deviceName = "",
  16.                         std::string symbolName = "",
  17.                         unsigned long startOptions = SERVICE_DEMAND_START
  18.                 );

  20.                 bool unload();

  22.                 int getControlCode(int function, int deviceType = FILE_DEVICE_UNKNOWN, int method = METHOD_BUFFERED, int access = FILE_ANY_ACCESS);

  24.                 bool control(int code, void* input, int inputLength, void* output, int outputLength);

  26.                 std::string lasterror();


  29.         private:
  30.                 std::string m_path;
  31.                 std::string m_deviceName;
  32.                 std::string m_symbolName;
  33.                 std::string m_serviceName;
  34.                 HANDLE m_deviceHandle;
  35.                 int m_error;
  36.         };
  37. }
复制代码


Driver.cpp
  1. #include "Driver.h"
  2. #include <Windows.h>

  4. /*
  5.         Author:River
  6. */
  7. namespace Alastor
  8. {
  9.         Driver::Driver()
  10.         {
  11.                 m_deviceHandle = NULL;
  12.                 m_error = 0;
  13.         }

  15.         Driver::~Driver()
  16.         {
  17.         }



  21.         bool Driver::load(
  22.                 std::string path,
  23.                 std::string deviceName,
  24.                 std::string symbolName,
  25.                 unsigned long startOptions
  26.         )
  27.         {
  28.                 SC_HANDLE hSCManager = OpenSCManagerA(NULL, NULL, startOptions);
  29.                 if (!hSCManager)
  30.                 {
  31.                         m_error = GetLastError();
  32.                         return false;
  33.                 }
  34.                 SC_HANDLE hService = CreateServiceA(hSCManager, m_serviceName.c_str(), m_serviceName.c_str(), SERVICE_ALL_ACCESS, SERVICE_KERNEL_DRIVER, startOptions, SERVICE_ERROR_IGNORE, m_path.c_str(), 0, 0, 0, 0, 0);
  35.                 if (hService == NULL)
  36.                 {
  37.                         if (ERROR_SERVICE_EXISTS == GetLastError())
  38.                         {
  39.                                 hService = OpenServiceA(hSCManager, m_serviceName.c_str(), SERVICE_ALL_ACCESS);
  40.                                 if (hService == NULL)
  41.                                 {
  42.                                         CloseServiceHandle(hSCManager);
  43.                                         m_error = GetLastError();
  44.                                         return false;
  45.                                 }
  46.                         }
  47.                         else
  48.                         {
  49.                                 if (ERROR_SERVICE_MARKED_FOR_DELETE == GetLastError() || ERROR_DUPLICATE_SERVICE_NAME == GetLastError())
  50.                                 {
  51.                                         hService = OpenServiceA(hSCManager, m_serviceName.c_str(), SERVICE_ALL_ACCESS);
  52.                                         if (hService == NULL)
  53.                                         {
  54.                                                 CloseServiceHandle(hSCManager);
  55.                                                 m_error = GetLastError();
  56.                                                 return false;
  57.                                         }
  58.                                         SERVICE_STATUS status;
  59.                                         ControlService(hService, SERVICE_CONTROL_STOP,&status);
  60.                                         DeleteService(hService);
  61.                                         CloseServiceHandle(hService);
  62.                                         CloseServiceHandle(hSCManager);
  63.                                         hSCManager = OpenSCManagerA(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  64.                                         if (hSCManager == NULL)
  65.                                         {
  66.                                                 m_error = GetLastError();
  67.                                                 return false;
  68.                                         }

  70.                                         hService = CreateServiceA(hSCManager, m_serviceName.c_str(), m_serviceName.c_str(), SERVICE_ALL_ACCESS, SERVICE_KERNEL_DRIVER, startOptions, SERVICE_ERROR_IGNORE, m_path.c_str(), 0, 0, 0, 0, 0);
  71.                                         if (hService == NULL)
  72.                                         {
  73.                                                 CloseServiceHandle(hSCManager);
  74.                                                 m_error = GetLastError();
  75.                                                 return false;
  76.                                         }
  77.                                 }
  78.                                 else
  79.                                 {
  80.                                         m_path = "";
  81.                                         m_deviceHandle = NULL;
  82.                                         CloseServiceHandle(hSCManager);
  83.                                         m_error = GetLastError();
  84.                                         return false;
  85.                                 }
  86.                         }
  87.                 }
  88.                
  89.                 if (StartServiceA(hService, NULL, NULL) == 0)
  90.                 {
  91.                         if (ERROR_SERVICE_ALREADY_RUNNING != GetLastError())
  92.                         {
  93.                                 CloseServiceHandle(hService);
  94.                                 CloseServiceHandle(hSCManager);
  95.                                 m_path = "";
  96.                                 m_error = GetLastError();
  97.                                 return false;
  98.                         }
  99.                 }
  100.                
  101.                 m_deviceHandle = CreateFileA(("\\\\.\" + m_symbolName).c_str(),GENERIC_READ|GENERIC_WRITE, NULL, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL,NULL);
  102.                 if (m_deviceHandle == NULL)
  103.                 {
  104.                         SERVICE_STATUS status;
  105.                         ControlService(hService, SERVICE_CONTROL_STOP, &status);
  106.                         DeleteService(hService);
  107.                         CloseServiceHandle(hService);
  108.                         CloseServiceHandle(hSCManager);

  110.                         m_path = "";
  111.                         m_error = GetLastError();
  112.                         return false;
  113.                 }

  115.                 CloseServiceHandle(hService);
  116.                 CloseServiceHandle(hSCManager);
  117.                 return true;
  118.         }

  120.         bool Driver::unload()
  121.         {
  122.                 if (m_deviceHandle)
  123.                 {
  124.                         CloseHandle(m_deviceHandle);
  125.                 }

  127.                 SC_HANDLE hSCManager = OpenSCManagerA(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  128.                 if (hSCManager == NULL)
  129.                 {
  130.                         m_path = "";
  131.                         m_deviceHandle = NULL;
  132.                         m_error = GetLastError();
  133.                         return false;
  134.                 }
  135.                 SC_HANDLE hService = OpenServiceA(hSCManager, m_serviceName.c_str(), SERVICE_ALL_ACCESS);
  136.                 if (hService == NULL)
  137.                 {
  138.                         CloseServiceHandle(hSCManager);
  139.                         m_path = "";
  140.                         m_deviceHandle = NULL;
  141.                         m_error = GetLastError();
  142.                         return false;
  143.                 }
  144.                 SERVICE_STATUS status;
  145.                 if (ControlService(hService, SERVICE_CONTROL_STOP, &status) == FALSE)
  146.                 {
  147.                         CloseServiceHandle(hService);
  148.                         CloseServiceHandle(hSCManager);
  149.                         m_path = "";
  150.                         m_deviceHandle = NULL;
  151.                         m_error = GetLastError();
  152.                         return false;
  153.                 }
  154.                 if (DeleteService(hService) == FALSE)
  155.                 {
  156.                         CloseServiceHandle(hService);
  157.                         CloseServiceHandle(hSCManager);
  158.                         m_path = "";
  159.                         m_deviceHandle = NULL;
  160.                         m_error = GetLastError();
  161.                         return false;
  162.                 }
  163.                 CloseServiceHandle(hService);
  164.                 CloseServiceHandle(hSCManager);
  165.                 m_path = "";
  166.                 m_deviceHandle = NULL;
  167.                 return false;
  168.         }

  170.         int Driver::getControlCode(int function, int deviceType, int method, int access)
  171.         {
  172.                 return (deviceType << 2) + (access << 14) + (function << 2) + method;
  173.         }

  175.         bool Driver::control(int code, void* input, int inputLength, void* output, int outputLength)
  176.         {
  177.                 OVERLAPPED overlapped;
  178.                 return DeviceIoControl(m_deviceHandle, code, input, inputLength, output, outputLength, NULL, &overlapped)==1;
  179.         }

  181.         std::string Driver::lasterror()
  182.         {
  183.                 return std::string();
  184.         }
  185. }
复制代码
回复

使用道具 举报

返回列表 发新帖
高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表