windows服务程序的安装和卸载函数

本文连接：https://www.cnblogs.com/jqdy/p/15079152.html

　　安装和卸载服务程序可以在命令行中使用 “sc install 服务名” 和 “sc uninstall 服务名”。

　　也可以将安装和卸载功能加入程序中去。

#include <Windows.h>
#include "WinApiFunc.h" // 引入 ApiPrinft
#include "definition.h"   // 引入 SVCDESCRIPTION，即在服务管理中显示的简介内容

#define BUFFER_SIZE 1020

VOID SvcInstall(PCTSTR pszName)
{
    SC_HANDLE schSCManager = NULL;
    SC_HANDLE schService = NULL;
    
 12     LPTSTR pszBuffer = NULL; // 各处使用的临时缓冲区
    pszBuffer = new TCHAR[BUFFER_SIZE];

    do {

        // Check directory length

        if (0 == GetCurrentDirectory(BUFFER_SIZE - 100, pszBuffer)) // 要预留文件名的位置，100个应该足够
        {
            ApiPrintf(3,
                TEXT("Path is too long, select a short directory to install. "),
                GetWindowsErrorString(pszBuffer, BUFFER_SIZE, GetLastError()), TEXT("
"));
            break;
        }

        // Get a handle to the SCM database.

        schSCManager = OpenSCManager(
            NULL,                   // local computer
            NULL,                   // ServicesActive database
            SC_MANAGER_ALL_ACCESS); // full access rights

        if (NULL == schSCManager)
        {
            ApiPrintf(3,
                TEXT("OpenSCManager failed. "),
                GetWindowsErrorString(pszBuffer, BUFFER_SIZE, GetLastError()), TEXT("
"));
            break;
        }

        // 取得文件当前的路径

        if (GetModuleFileName(NULL, pszBuffer, BUFFER_SIZE) == BUFFER_SIZE)
        {
            ApiPrintf(1, TEXT("Path string is too long, copy install file to a shorter path.
"));
            break;
        }

        // Create the service

        schService = CreateService(
            schSCManager,               // SCM database
            pszName,                    // name of service
            pszName,                    // service name to display
            SERVICE_ALL_ACCESS,         // desired access
            SERVICE_WIN32_OWN_PROCESS,  // service type
            SERVICE_AUTO_START,         // start type
            SERVICE_ERROR_NORMAL,       // error control type
            pszBuffer,                  // path to service's binary
            NULL,                       // no load ordering group
            NULL,                       // no tag identifier
            NULL,                       // no dependencies
            NULL,                       // LocalSystem account
            NULL);                      // no password

        if (NULL == schService)
        {
            ApiPrintf(3,
                TEXT("CreateService failed. "),
                GetWindowsErrorString(pszBuffer, BUFFER_SIZE, GetLastError()),
                TEXT("
"));
            break;
        }
        else
        {
            // 写入服务描述内容
            TCHAR pszText[] = SVCDESCRIPTION;
            SERVICE_DESCRIPTION description{ pszText };
            ChangeServiceConfig2(schService, SERVICE_CONFIG_DESCRIPTION, &description);

            ApiPrintf(1, TEXT("Service installed successfully.
"));
        }
    } while (0);

    if (schSCManager)
    {
        CloseServiceHandle(schSCManager);
        schSCManager = NULL;
    }
    if (schService)
    {
        CloseServiceHandle(schService);
        schService = NULL;
    }
    if (!pszBuffer)
    {
        delete[]pszBuffer;
        pszBuffer = NULL;
    }
}

VOID SvcUnInstall(PCTSTR pszServiceName)
{
    SC_HANDLE schSCManager = NULL;
    SC_HANDLE schService = NULL;
    LPTSTR pszBuffer = NULL;
    SERVICE_STATUS svcStatus;

    pszBuffer = new TCHAR[BUFFER_SIZE];
    do {
        // Open the local default service control manager database
        schSCManager = OpenSCManager(
            NULL,                   // local computer
            NULL,                   // ServiceActive database
            SC_MANAGER_ALL_ACCESS); // full access rights

        if (schSCManager == NULL)
        {
            ApiPrintf(3,
                TEXT("OpenSCManager failed. "),
                GetWindowsErrorString(pszBuffer, BUFFER_SIZE, GetLastError()),
                TEXT("
"));
            break;
        }

        // Open the service with delete, stop, and query status permissions
        schService = OpenService(
            schSCManager,
            pszServiceName,
            SERVICE_STOP | SERVICE_QUERY_STATUS | DELETE);

        if (schService == NULL)
        {
            ApiPrintf(3,
                TEXT("OpenService failed. "),
                GetWindowsErrorString(pszBuffer, BUFFER_SIZE, GetLastError()),
                TEXT("
"));
            break;
        }

        // Try to stop the service
        if (ControlService(schService, SERVICE_CONTROL_STOP, &svcStatus))
        {
            ApiPrintf(2, pszServiceName, TEXT(" is stopping.
"));

            Sleep(1000);

            while (QueryServiceStatus(schService, &svcStatus))
            {
                if (svcStatus.dwCurrentState == SERVICE_STOP_PENDING)
                {
                    Sleep(1000);
                }
                else
                {
                    break;
                }
            }

            ApiPrintf(3,
                pszServiceName,
                (svcStatus.dwCurrentState == SERVICE_STOPPED ? TEXT(" is stopping.") : TEXT(" failed to stop.")),
                TEXT("
"));
        }

        // Now remove the service by calling DeleteService.
        if (!DeleteService(schService))
        {
            ApiPrintf(3,
                TEXT("DeleteService failed. "),
                GetWindowsErrorString(pszBuffer, BUFFER_SIZE, GetLastError()),
                TEXT("
"));
            break;
        }

        ApiPrintf(2, pszServiceName, TEXT(" is uninstalled.
"));

    } while (0);

    // Centralized cleanup for all allocated resources.
    if (schSCManager)
    {
        CloseServiceHandle(schSCManager);
        schSCManager = NULL;
    }
    if (schService)
    {
        CloseServiceHandle(schService);
        schService = NULL;
    }
    if (!pszBuffer)
    {
        delete[] pszBuffer;
        pszBuffer = NULL;
    }
}