创建系统服务

绝地灬酷狼 2022-08-28 05:48 165阅读 0赞

// 系统进程自启动是通过创建系统服务并设置服务启动类型为自动启动来实现的。
    
    // PS：OpenSCManager函数：
    // 功能：建立一个到服务控制管理器的连接，并打开指定的数据库。
    // 原型：SC_HANDLE OpenSCManager (
    // LPCTSTR lpMachineName,
    // LPCTSTR lpDatabaseName,
    // DWORD dwDesiredAccess
    // )
    // 参数：lpMachineName：指定目标计算机名称，如果为NULL或空字符串，则表示指定本地计算机，那么该函数连接到本地计算机的SCM。
    // lpDatabaseName：指定将要打开的SCM数据库名称，这个值应该为SERVICE_ACTIVE_DATABASE,如果为NULL，就表示SERVICE_ACTIVE_DATABASE。
    // dwDesiredAccess：访问权限。
    // 结果：成功，返回SCM数据库句柄，失败，返回NULL。
    
    // PS：CreateService函数：
    // 功能：创建一个服务对象，并将其添加到指定的SCM数据库中。
    // 原型：SC_HANDLE CreateService (
    // SC_HANDLE hSCManager,
    // LPCTSTR lpServiceName,
    // LPCTSTR lpDisplayName,
    // DWORD dwDesiredAccess,
    // DWORD dwServiceType,
    // DWORD dwStartType,
    // DWORD dwErrorControl,
    // LPCTSTR lpBinaryPathName,
    // lPCTSTR lpLoadOrderGroup,
    // LPDWORD lpdwTagId,
    // LPCTSTR lpDependencies,
    // LPCTSTR lpServiceStartName,
    // LPCTSTR lpPassword
    // )
    // 参数：hSCManager：由OpenSCManager函数返回的SCM数据库句柄。
    // lpServiceName：服务名称，一般和文件名相同。
    // lpDsiplayName：UI用来识别服务的显示名称，一般和文件名相同。
    // dwDesiredAccess：访问权限。
    // dwServiceType：服务类型，可以是以下枚举之一：
    // SERVICE_ADAPTER：保留。
    // SERVICE_FILE_SYSTEM_DRIVER：文件系统驱动服务程序。
    // SERVICE_KERNEL_DRIVER：内核驱动服务程序。
    // SERVICE_RECOGNIZER_DRIVER：保留。
    // SERVICE_WIN32_OWN_PROCESS：运行在独立进程中的服务程序。
    // SERVICE_WIN32_SHARE_PROCESS：由多个进程共享的服务程序。
    // dwStartType：服务启动选项，为以下枚举之一：
    // SERVICE_AUTO_START：系统启动时由SCM自动启动该服务程序。
    // SERVICE_BOOT_START：用在由系统加载器创建的设备驱动程序中，它只能用于驱动服务程序。
    // SERVICE_DEMAND_START：由SCM启动的服务。
    // SERVICE_DISABLED：表示该服务不可启动。
    // SERVICE_SYSTEM_START：用于由IoInitSystem函数创建的设备驱动程序。
    // dwErrorControl：当该服务启动失败时，指定产生错误的严重程度以及应采取的保护措施。
    // lpBinaryPathName：服务程序的二进制文件路径。
    // lpLoadOrderGroup：指向加载排序组的名称。
    // lpdwTagId：接收由lpLoadOrderGroup参数指定的组中唯一的标记值变量。
    // lpDependencies：空分隔名称的服务或加载顺序组系统在这个服务开始之前的双空终止数组的指针。
    // lpServiceStartName：该服务应运行的账户名称。
    // lpPassword：账户名对应的密码。
    // 结果，返回，返回该服务的句柄，失败，返回NULL。
    
    // PS：OpenService函数：
    // 功能：打开一个已存在的服务。
    // 原型：SC_HANDLE WINAPI OpenService (
    // _In_ SC_HANDLE hSCManager,
    // _In_ LPCTSTR lpServiceName,
    // _In_ DWORD dwDesiredAccess
    // )
    // 参数：hSCManager：由OpenSCManager函数获取的SCM数据库句柄。
    // lpServiceName：服务名称。
    // dwDesiredAccess：访问权限。
    // 结果：成功，返回TRUE，失败，返回FALSE。
    
    // PS：StartService函数：
    // 功能：启动服务。
    // 原型：BOOL WINAPI StartService (
    // _In_ SC_HANDLE hService,
    // _In_DWORD dwNumServiceArgs,
    // _In_opt_ lPCTSTR *lpServiceArgVectors
    // )
    // 参数：hService：由CreateService或OpenService获取的服务句柄。
    // dwNumServiceArgs：指向lpServiceArgVectors中的个数。
    // lpServiceArgVectors：传递给ServiceMain的参数，如果没有，可以设置为NULL；否则，第一个形参lpServiceArgVectors[0]为服务的名字，其他则为需要传入的参数。
    // 结果：成功，返回TRUE，失败，返回FALSE。
    // 备注：驱动不接受lpServiceArgVector为空，dwNumServiceArgs为0。
    
    // PS：StartServiceCtrlDispatcher函数：
    // 功能：将服务进程的主线程连接到SCM，该线程将作为调用过程的服务控制分派器线程。
    // 原型：BOOL WINAPI StartServiceCtrlDispatcher(_In_ const SERVICE_TABLE_ENTRY *lpServiceTable)
    // 参数：lpServiceTable：指向一系列SERVICE_TABLE_ENTRY结构的指针，其中包含可在调用进程中执行的每个服务的条目。表中最后一个条目必须具有NULL值以指定表的结束。
    // 结果：陈宫，返回TRUE，失败，返回FALSE。
    
    // PS：实现原理：一、创建启动系统服务。二、系统服务程序的编写。
    
    // PS：创建启动系统服务：
    
    // 示例代码：
    
    // 0 创建服务 1 启动服务 2 停止服务 3 删除服务
    BOOL SystemServiceOperate(char *lpszDriverPath, uint nOperateType)
    { 
        BOOL bRet = TRUE;
        char szName[MAX_SIZE] = { 0};
        ::lstrcpy(szName, lpszDriverPath);
    
        // 过滤掉文件目录，获取文件名
        ::PathStripPath(szName);
        
        SC_HANDLE shOSCM = NULL, shCS = NULL;
        SERVICE_STATUS ss;
        DWORD dwErrorCode = 0;
        BOOL bSuccess = FALSE;
    
        // 打开SCM数据库
        shOSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
        if (!shOSCM)
        { 
            ShowError("OpenSCManager");
            return FALSE;
        }
    
        // 打开已创建的服务
        if (0 != nOperateType)
        { 
            shCS = OpenService(shOSCM, szName, szName, SERVICE_ALL_ACCESS);
            if (!shCS)
            { 
                ShowError("OpenService");
                ::CloseServiceHandle(shOSCM);
                shOSCM = NULL;
                return FALSE;
            }
        }
    
        switch (nOperateType)
        { 
            case 0:
            { 
                // 创建服务
                shCS = ::CreateService(shOSCM, szName, szName, SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS,
                             SERVICE_AUTO_START, SERVICE_ERROR_NORMAL, lpszDriverPath, NULL, NULL, NULL, NULL, NULL);
                if (!shCS)
                { 
                    ShowError("CreateService");
                    bRet = FALSE;
                }
                break;
            }
            case 1:
            { 
                // 启动服务
                if(!::StartService(shCS, 0, NULL))
                { 
                    ShowError("StartService");
                    bRet = FALSE;
                }
                break;
            }
            case 2:
            { 
                // 终止服务
                if (!::ControlService(shCS, SERVICE_CONTROL_STOP, &ss))
                { 
                    ShowError("ControlService");
                    bRet = FALSE;
                }
                break;
            }
            case 3:
            { 
                // 删除服务
                if(!::DeleteService(shCS))
                { 
                    ShowError("DeleteService");
                    bRet = FALSE;
                }
                break;
            }
            default:
                break;
        }
    
        // 关闭句柄
        if (shCS)
        { 
            ::CloseServiceHandle(shCS);
            shCS = NULL;
        }
    
        if (shOSCM)
        { 
            ::CloseServiceHandle(shOSCM);
            shOSCM = NULL;
        }
    
        return bRet;
    }
    
    // PS：系统服务程序的编写
    
    // 服务程序main入口函数的代码：
    
    int _tmain(int argc, _TCHAR* argv[])
    { 
        // 注册服务入口函数
        SERVICE_TABLE_ENTRY stDispatchTable[] = { { g_szServiceName, (LPSERVICE_MAIN_FUNCTION)ServiceMain}, { NULL, NULL}};
        ::StartServiceCtrlDispatcher(stDispatchTable);
        return 0;
    }
    
    // 服务程序ServiceMain入口函数示例代码：
    
    void __stdcall ServiceMain(DWORD dwArgc, char *lpszArgv)
    { 
        g_ServiceStatusHandle = ::RegisterServiceCtrlHandler(g_szServiceName, ServiceCtrlHandle);
        TellSCM(SERVICE_START_PENDING, 0, 1);
        TellSCM(SERVICE_RUNNING, 0, 0);
    
        // 自定义功能代码放在这里
        while (TRUE)
        { 
            Sleep(5000);
            DoTask();
        }
    }