mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
3.1 KiB
C++
113 lines
3.1 KiB
C++
#include <windows.h>
|
|
#include <chrono>
|
|
#include <llarp.hpp>
|
|
#include <llarp/util/logging.hpp>
|
|
#include "service_manager.hpp"
|
|
#include <dbghelp.h>
|
|
#include <cassert>
|
|
#include <csignal>
|
|
#include <optional>
|
|
|
|
namespace llarp::sys
|
|
{
|
|
|
|
static auto logcat = log::Cat("svc");
|
|
|
|
namespace
|
|
{
|
|
|
|
std::optional<DWORD>
|
|
to_win32_state(ServiceState st)
|
|
{
|
|
switch (st)
|
|
{
|
|
case ServiceState::Starting:
|
|
return SERVICE_START_PENDING;
|
|
case ServiceState::Running:
|
|
return SERVICE_RUNNING;
|
|
case ServiceState::Stopping:
|
|
return SERVICE_STOP_PENDING;
|
|
case ServiceState::Stopped:
|
|
return SERVICE_STOPPED;
|
|
default:
|
|
return std::nullopt;
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
SVC_Manager::SVC_Manager()
|
|
{
|
|
_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
|
|
}
|
|
|
|
void
|
|
SVC_Manager::system_changed_our_state(ServiceState st)
|
|
{
|
|
if (m_disable)
|
|
return;
|
|
if (st == ServiceState::Stopping)
|
|
{
|
|
we_changed_our_state(st);
|
|
}
|
|
}
|
|
|
|
void
|
|
SVC_Manager::report_changed_state()
|
|
{
|
|
if (m_disable)
|
|
return;
|
|
|
|
log::debug(
|
|
logcat,
|
|
"Reporting Windows service status '{}', exit code {}, wait hint {}, dwCP {}, dwCA {}",
|
|
_status.dwCurrentState == SERVICE_START_PENDING ? "start pending"
|
|
: _status.dwCurrentState == SERVICE_RUNNING ? "running"
|
|
: _status.dwCurrentState == SERVICE_STOPPED ? "stopped"
|
|
: _status.dwCurrentState == SERVICE_STOP_PENDING
|
|
? "stop pending"
|
|
: fmt::format("unknown: {}", _status.dwCurrentState),
|
|
_status.dwWin32ExitCode,
|
|
_status.dwWaitHint,
|
|
_status.dwCheckPoint,
|
|
_status.dwControlsAccepted);
|
|
|
|
SetServiceStatus(handle, &_status);
|
|
}
|
|
|
|
void
|
|
SVC_Manager::we_changed_our_state(ServiceState st)
|
|
{
|
|
if (st == ServiceState::Failed)
|
|
{
|
|
_status.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
|
|
_status.dwServiceSpecificExitCode = 2; // TODO: propagate more info ?
|
|
report_changed_state();
|
|
}
|
|
else if (auto maybe_state = to_win32_state(st))
|
|
{
|
|
auto new_state = *maybe_state;
|
|
_status.dwWin32ExitCode = NO_ERROR;
|
|
_status.dwCurrentState = new_state;
|
|
_status.dwControlsAccepted = st == ServiceState::Running ? SERVICE_ACCEPT_STOP : 0;
|
|
_status.dwWaitHint =
|
|
std::chrono::milliseconds{
|
|
st == ServiceState::Starting ? StartupTimeout
|
|
: st == ServiceState::Stopping ? StopTimeout
|
|
: 0s}
|
|
.count();
|
|
// dwCheckPoint gets incremented during a start/stop to tell windows "we're still
|
|
// starting/stopping" and to reset its must-be-hung timer. We increment it here so that this
|
|
// can be called multiple times to tells Windows something is happening.
|
|
if (st == ServiceState::Starting or st == ServiceState::Stopping)
|
|
_status.dwCheckPoint++;
|
|
else
|
|
_status.dwCheckPoint = 0;
|
|
|
|
report_changed_state();
|
|
}
|
|
}
|
|
|
|
SVC_Manager _manager{};
|
|
I_SystemLayerManager* const service_manager = &_manager;
|
|
} // namespace llarp::sys
|