Отправить команду на сервис из C++

Как я могу отправить команду в службу Windows из C++? Эквивалентный.NET-код:

ServiceController sc = new ServiceController("MyService");
sc.ExecuteCommand(255);

3 ответа

Решение

Из нативного C++ вам необходимо:

  1. Откройте дескриптор диспетчера управления службами,
  2. Используйте диспетчер управления сервисами, чтобы получить дескриптор сервиса для сервиса, которым вы хотите управлять,
  3. Отправьте контрольный код или коды в службу, и
  4. Закройте ручки, открытые в шагах 1 и 2.

Например, этот код перезапускает службу синхронизации времени. Сначала я создаю класс-оболочку для служебных дескрипторов, чтобы они автоматически закрывались при выходе из блока.

class CSC_HANDLE
{
public:
 CSC_HANDLE(SC_HANDLE h) : m_h(h) { }
 ~CSC_HANDLE() { ::CloseServiceHandle(m_h); }
 operator SC_HANDLE () { return m_h; }
private:
 SC_HANDLE m_h;
};

Затем я открываю диспетчер управления службами (используя OpenSCManager ()) и службу, которой хочу управлять. Обратите внимание, что параметр dwDesiredAccess для OpenService() должен включать разрешения для каждого элемента управления, который я хочу отправить, иначе соответствующие функции элемента управления не будут работать.

BOOL RestartTimeService()
{
    CSC_HANDLE hSCM(::OpenSCManager(NULL, SERVICES_ACTIVE_DATABASE, GENERIC_READ));
    if (NULL == hSCM) return FALSE;

    CSC_HANDLE hW32Time(::OpenService(hSCM, L"W32Time", SERVICE_START | SERVICE_STOP | SERVICE_QUERY_STATUS));
    if (NULL == hW32Time) return FALSE;

Чтобы остановить службу, я использую ControlService() для отправки кода SERVICE_CONTROL_STOP, а затем проверяю возвращаемое значение, чтобы убедиться, что команда выполнена успешно. Если сообщается о какой-либо ошибке, кроме ERROR_SERVICE_NOT_ACTIVE, я предполагаю, что запуск службы не будет успешным.

    SERVICE_STATUS ss = { 0 };
    ::SetLastError(0);
    BOOL success = ::ControlService(hW32Time, SERVICE_CONTROL_STOP, &ss);
    if (!success)
    {
        DWORD le = ::GetLastError();
        switch (le)
        {
        case ERROR_ACCESS_DENIED:
        case ERROR_DEPENDENT_SERVICES_RUNNING:
        case ERROR_INVALID_HANDLE:
        case ERROR_INVALID_PARAMETER:
        case ERROR_INVALID_SERVICE_CONTROL:
        case ERROR_SERVICE_CANNOT_ACCEPT_CTRL:
        case ERROR_SERVICE_REQUEST_TIMEOUT:
        case ERROR_SHUTDOWN_IN_PROGRESS:
            return FALSE;

        case ERROR_SERVICE_NOT_ACTIVE:
        default:
            break;
        }
    }

После указания остановки службы я жду, когда менеджер службы сообщит, что служба фактически остановлена. В этом коде есть две потенциальные ошибки, которые вы можете исправить в рабочем коде:

  1. Sleep (1000) приостановит цикл обработки сообщений в этом потоке, поэтому вам следует использовать другой метод для задержки выполнения, если эта функция будет выполняться в потоке пользовательского интерфейса. Вы можете создать подходящий цикл сна с сообщением, используя MsgWaitForMultipleObjectsEx ().
  2. DWORD, возвращаемый из GetTickCount(), в конце концов обернется до нуля; если это произойдет, пока эта функция ожидает, ожидание может сдаться раньше, чем я планировал.

    DWORD waitstart(::GetTickCount());
    while (true)
    {
        ZeroMemory(&ss, sizeof(ss));
        ::QueryServiceStatus(hW32Time, &ss);
        if (SERVICE_STOPPED == ss.dwCurrentState) break;
        ::Sleep(1000);
        DWORD tick(::GetTickCount());
        if ((tick < waitstart) || (tick > (waitstart + 30000))) return FALSE;
    }
    

Наконец, зная, что служба находится в остановленном состоянии, я вызываю StartService() и запускаю ее снова.

    success = ::StartService(hW32Time, 0, NULL);
    if (!success) return FALSE;

    return TRUE;
}

Вот небольшая программа, которая подключится к сервису под названием "MYSERVICE", а затем отправит команду 141 (которая определяется сервисом)

// ServiceCommunicator.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <Windows.h>
#include <iostream>

using namespace std;

int main()
{
    SC_HANDLE managerHandle;
    SC_HANDLE serviceHandle;

    SERVICE_STATUS   controlParms;
    DWORD retStatus;

    managerHandle = OpenSCManager(NULL, NULL, GENERIC_READ);
    if (NULL != managerHandle)
    {
        serviceHandle = OpenService(managerHandle, L"MYSERVICE", SERVICE_USER_DEFINED_CONTROL | SERVICE_QUERY_STATUS);

        if (NULL != serviceHandle)
        {
            cout << "connected to Service" << endl;
            retStatus = ControlService(serviceHandle, 141, &controlParms);

            if (retStatus)
            {
                //Get the return code from the service
                cout << "For command 141, return code from service was " << controlParms.dwWin32ExitCode << endl;
            }
            else
                cout << "Sending command 141 failed" << endl;

            CloseServiceHandle(serviceHandle);
        }
        else
        {
            cout << "could not connect to Service" << endl;
        }

        CloseServiceHandle(managerHandle);
    }
    else
    {
        cout << "could not open service manager" << endl;
    }
    return 0;
}
Другие вопросы по тегам