From 10e1921d30a66db85d8707e554e0eeb5efbb8b00 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sun, 10 May 2015 23:00:08 +0300 Subject: add service mgmt lib & utils --- utils/libservice/src/service.cpp | 235 +++++++++++++++++++++++++++++++ utils/libservice/src/service_handle.cpp | 30 ++++ utils/libservice/src/service_manager.cpp | 48 +++++++ utils/libservice/src/windows_error.cpp | 41 ++++++ 4 files changed, 354 insertions(+) create mode 100644 utils/libservice/src/service.cpp create mode 100644 utils/libservice/src/service_handle.cpp create mode 100644 utils/libservice/src/service_manager.cpp create mode 100644 utils/libservice/src/windows_error.cpp (limited to 'utils/libservice/src') diff --git a/utils/libservice/src/service.cpp b/utils/libservice/src/service.cpp new file mode 100644 index 0000000..1f7cbef --- /dev/null +++ b/utils/libservice/src/service.cpp @@ -0,0 +1,235 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include "libservice/common.hpp" +#include "libservice/service.hpp" +#include "libservice/service_handle.hpp" +#include "libservice/service_manager.hpp" +#include "libservice/windows_error.hpp" + +#include + +#include +#include +#include + +namespace libservice +{ + namespace + { + ServiceHandle open_service(const ServiceManager& mgr, const std::string& name) + { + const auto raw = OpenServiceA( + static_cast(mgr), + name.c_str(), + SERVICE_ALL_ACCESS); + + if (NULL == raw) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + + return ServiceHandle(raw); + } + + ServiceHandle install_service( + const ServiceManager& mgr, + const std::string& name, + const std::string& bin_path) + { + const auto raw = CreateServiceA( + static_cast(mgr), + name.c_str(), + name.c_str(), + SERVICE_ALL_ACCESS, + SERVICE_KERNEL_DRIVER, + SERVICE_DEMAND_START, + SERVICE_ERROR_NORMAL, + bin_path.c_str(), + NULL, + NULL, + NULL, + NULL, + NULL); + + if (NULL == raw) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + + return ServiceHandle(raw); + } + + void start_service(const ServiceHandle& handle) + { + if (!StartService(static_cast(handle), 0, NULL)) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + } + + void stop_service(const ServiceHandle& handle) + { + SERVICE_STATUS service_status; + + if (!ControlService(static_cast(handle), SERVICE_CONTROL_STOP, &service_status)) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + } + + void uninstall_service(const ServiceHandle& handle) + { + if (!DeleteService(static_cast(handle))) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + } + + SERVICE_STATUS_PROCESS query_service_status(const ServiceHandle& handle) + { + SERVICE_STATUS_PROCESS status; + DWORD nbreq; + + if (!QueryServiceStatusEx(static_cast(handle), + SC_STATUS_PROCESS_INFO, + reinterpret_cast(&status), + sizeof(status), + &nbreq)) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + + return status; + } + + DWORD query_service_state(const ServiceHandle& handle) + { + return query_service_status(handle).dwCurrentState; + } + + SERVICE_STATUS_PROCESS wait_for_service_state( + const ServiceHandle& handle, + const DWORD desired_state) + { + auto status = query_service_status(handle); + + DWORD old_timestamp = GetTickCount(); + + DWORD old_check_point = status.dwCheckPoint; + DWORD old_wait_hint = status.dwWaitHint; + + while (desired_state != status.dwCurrentState) + { + DWORD wait_time = old_wait_hint / 10; + + if (wait_time < 1000) + wait_time = 1000; + else if (wait_time > 10000) + wait_time = 10000; + + Sleep(wait_time); + + status = query_service_status(handle); + + if (desired_state == status.dwCurrentState) + break; + + if (status.dwCheckPoint > old_check_point) + { + old_timestamp = GetTickCount(); + + old_check_point = status.dwCheckPoint; + old_wait_hint = status.dwWaitHint; + } + else if (GetTickCount() - old_timestamp > old_wait_hint) + { + return status; + } + } + return status; + } + } + + Service Service::open(const ServiceManager& mgr, const std::string& name) + { + return Service(open_service(mgr, name)); + } + + Service Service::install(const ServiceManager& mgr, + const std::string& name, + const std::string& bin_path) + { + return Service(install_service(mgr, name, bin_path)); + } + + void Service::start() const + { + const auto state = query_service_state(m_handle); + + switch (state) + { + case SERVICE_STOPPED: + break; + + case SERVICE_STOP_PENDING: + wait_for_service_state(m_handle, SERVICE_STOPPED); + break; + + default: + return; + } + + start_service(m_handle); + wait_for_service_state(m_handle, SERVICE_RUNNING); + } + + void Service::stop() const + { + switch (query_service_state(m_handle)) + { + case SERVICE_STOPPED: + return; + + case SERVICE_STOP_PENDING: + wait_for_service_state(m_handle, SERVICE_STOPPED); + return; + } + + stop_service(m_handle); + wait_for_service_state(m_handle, SERVICE_STOPPED); + } + + void Service::uninstall() const + { + stop(); + uninstall_service(m_handle); + } + + void swap(Service& a, Service& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} + +namespace std +{ + template <> + void swap( + libservice::Service& a, + libservice::Service& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} diff --git a/utils/libservice/src/service_handle.cpp b/utils/libservice/src/service_handle.cpp new file mode 100644 index 0000000..5356b50 --- /dev/null +++ b/utils/libservice/src/service_handle.cpp @@ -0,0 +1,30 @@ +/** + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include "libservice/common.hpp" +#include "libservice/service_handle.hpp" + +#include + +namespace libservice +{ + void swap(ServiceHandle& a, ServiceHandle& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} + +namespace std +{ + template <> + void swap( + libservice::ServiceHandle& a, + libservice::ServiceHandle& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} diff --git a/utils/libservice/src/service_manager.cpp b/utils/libservice/src/service_manager.cpp new file mode 100644 index 0000000..4888406 --- /dev/null +++ b/utils/libservice/src/service_manager.cpp @@ -0,0 +1,48 @@ +/** + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include "libservice/common.hpp" +#include "libservice/service_handle.hpp" +#include "libservice/service_manager.hpp" +#include "libservice/windows_error.hpp" + +#include + +#include +#include + +namespace libservice +{ + ServiceManager ServiceManager::open() + { + SC_HANDLE raw = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS); + + if (NULL == raw) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get()); + } + + return ServiceManager(ServiceHandle(raw)); + } + + void swap(ServiceManager& a, ServiceManager& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} + +namespace std +{ + template <> + void swap( + libservice::ServiceManager& a, + libservice::ServiceManager& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} diff --git a/utils/libservice/src/windows_error.cpp b/utils/libservice/src/windows_error.cpp new file mode 100644 index 0000000..6cc7302 --- /dev/null +++ b/utils/libservice/src/windows_error.cpp @@ -0,0 +1,41 @@ +/** + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include "libservice/windows_error.hpp" + +#include + +#include + +namespace libservice +{ + std::string WinErrorCategory::message(int code) const + { + char* buf_ptr; + + DWORD written = FormatMessageA( + FORMAT_MESSAGE_ALLOCATE_BUFFER + | FORMAT_MESSAGE_FROM_SYSTEM + | FORMAT_MESSAGE_IGNORE_INSERTS, + NULL, + code, + MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), + reinterpret_cast(&buf_ptr), + 0, + NULL); + + if (0 == written) + { + LocalFree(buf_ptr); + return "Couldn't format error message"; + } + + std::string str(buf_ptr, written - 2); + LocalFree(buf_ptr); + return str; + } +} -- cgit v1.2.3