generic-library/poco
1
0
Fork 0
mirror of https://github.com/pocoproject/poco.git synced 2025-10-28 11:31:53 +01:00
Code Issues Projects Releases Wiki Activity

Merge branch 'poco-1.10.1' of github.com:pocoproject/poco into devel

Browse Source
This commit is contained in:
Günter Obiltschnig
2020-02-12 20:37:04 +01:00
parent 7e8c9bc0fd bb63b28121
commit e795d45023
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
Util/src/WinService.cpp
View File

@@ -125,6 +125,7 @@ void WinService::unregisterService()
open(); open();
if (!DeleteService(_svcHandle)) if (!DeleteService(_svcHandle))
throw SystemException("cannot unregister service", _name); throw SystemException("cannot unregister service", _name);
close();
} }
@@ -236,10 +237,17 @@ WinService::Startup WinService::getStartup() const
void WinService::setFailureActions(FailureActionVector failureActions, const std::string& command, const std::string& rebootMessage) void WinService::setFailureActions(FailureActionVector failureActions, const std::string& command, const std::string& rebootMessage)
{ {
if (failureActions.size() > 3) {
throw InvalidArgumentException{ "Only 0-3 Failure Actions are supported" };
}
open(); open();
auto actions = new SC_ACTION[3]; auto actions = new SC_ACTION[failureActions.size()];
SERVICE_FAILURE_ACTIONSW ac; SERVICE_FAILURE_ACTIONSW ac;
ac.lpCommand = NULL;
ac.lpRebootMsg = NULL;
std::wstring urebootMessage; std::wstring urebootMessage;
Poco::UnicodeConverter::toUTF16(rebootMessage, urebootMessage); Poco::UnicodeConverter::toUTF16(rebootMessage, urebootMessage);
std::vector<wchar_t> rebootMessageVector{ urebootMessage.begin(), urebootMessage.end() }; std::vector<wchar_t> rebootMessageVector{ urebootMessage.begin(), urebootMessage.end() };
@@ -250,7 +258,7 @@ void WinService::setFailureActions(FailureActionVector failureActions, const std
std::vector<wchar_t> commandVector{ uComamnd.begin(), uComamnd.end() }; std::vector<wchar_t> commandVector{ uComamnd.begin(), uComamnd.end() };
commandVector.push_back('\0'); commandVector.push_back('\0');
for (auto i = 0; i < 3; i++) for (auto i = 0; i < failureActions.size(); i++)
{ {
switch (failureActions[i].type) switch (failureActions[i].type)
{ {
@@ -276,7 +284,7 @@ void WinService::setFailureActions(FailureActionVector failureActions, const std
} }
ac.dwResetPeriod = 0; ac.dwResetPeriod = 0;
ac.cActions = 3; ac.cActions = failureActions.size();
ac.lpsaActions = actions; ac.lpsaActions = actions;
if (!ChangeServiceConfig2W(_svcHandle, SERVICE_CONFIG_FAILURE_ACTIONS, &ac)) if (!ChangeServiceConfig2W(_svcHandle, SERVICE_CONFIG_FAILURE_ACTIONS, &ac))