WireGuard rework for MacOS and Windows (#314)

WireGuard rework for MacOS and Windows

client/platforms/windows/windowsservicemanager.cpp

@@ -0,0 +1,139 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "windowsservicemanager.h"
+
+#include <QTimer>
+
+#include "Windows.h"
+#include "Winsvc.h"
+#include "logger.h"
+//#include "mozillavpn.h"
+#include "platforms/windows/windowsutils.h"
+
+namespace {
+Logger logger("WindowsServiceManager");
+}
+
+WindowsServiceManager::WindowsServiceManager(LPCWSTR serviceName) {
+  DWORD err = NULL;
+  auto scm_rights = SC_MANAGER_CONNECT | SC_MANAGER_ENUMERATE_SERVICE |
+                    SC_MANAGER_QUERY_LOCK_STATUS | STANDARD_RIGHTS_READ;
+  m_serviceManager = OpenSCManager(NULL,  // local computer
+                                   NULL,  // servicesActive database
+                                   scm_rights);
+  err = GetLastError();
+  if (err != NULL) {
+    logger.error() << " OpenSCManager failed code: " << err;
+    return;
+  }
+  logger.debug() << "OpenSCManager access given - " << err;
+
+  logger.debug() << "Opening Service - "
+                 << QString::fromWCharArray(serviceName);
+  // Try to get an elevated handle
+  m_service = OpenService(m_serviceManager,  // SCM database
+                          serviceName,       // name of service
+                          (GENERIC_READ | SERVICE_START | SERVICE_STOP));
+  err = GetLastError();
+  if (err != NULL) {
+    WindowsUtils::windowsLog("OpenService failed");
+    return;
+  }
+  m_has_access = true;
+  m_timer.setSingleShot(false);
+
+  logger.debug() << "Service manager execute access granted";
+}
+
+WindowsServiceManager::~WindowsServiceManager() {
+  if (m_service != NULL) {
+    CloseServiceHandle(m_service);
+  }
+  if (m_serviceManager != NULL) {
+    CloseServiceHandle(m_serviceManager);
+  }
+}
+
+bool WindowsServiceManager::startPolling(DWORD goal_state, int max_wait_sec) {
+  int tries = 0;
+  while (tries < max_wait_sec) {
+    SERVICE_STATUS status;
+    if (!QueryServiceStatus(m_service, &status)) {
+      WindowsUtils::windowsLog("Failed to retrieve the service status");
+      return false;
+    }
+
+    if (status.dwCurrentState == goal_state) {
+      if (status.dwCurrentState == SERVICE_RUNNING) {
+        emit serviceStarted();
+      }
+      if (status.dwCurrentState == SERVICE_STOPPED) {
+        emit serviceStopped();
+      }
+      return true;
+    }
+
+    logger.debug() << "Polling Status" << m_state_target
+                   << "wanted, has: " << status.dwCurrentState;
+    Sleep(1000);
+    ++tries;
+  }
+  return false;
+}
+
+SERVICE_STATUS_PROCESS WindowsServiceManager::getStatus() {
+  SERVICE_STATUS_PROCESS serviceStatus;
+  if (!m_has_access) {
+    logger.debug() << "Need read access to get service state";
+    return serviceStatus;
+  }
+  DWORD dwBytesNeeded;  // Contains missing bytes if struct is too small?
+  QueryServiceStatusEx(m_service,                       // handle to service
+                       SC_STATUS_PROCESS_INFO,          // information level
+                       (LPBYTE)&serviceStatus,          // address of structure
+                       sizeof(SERVICE_STATUS_PROCESS),  // size of structure
+                       &dwBytesNeeded);
+  return serviceStatus;
+}
+
+bool WindowsServiceManager::startService() {
+  auto state = getStatus().dwCurrentState;
+  if (state != SERVICE_STOPPED && state != SERVICE_STOP_PENDING) {
+    logger.warning() << ("Service start not possible, as its running");
+    emit serviceStarted();
+    return true;
+  }
+
+  bool ok = StartService(m_service,  // handle to service
+                         0,          // number of arguments
+                         NULL);      // no arguments
+  if (ok) {
+    logger.debug() << ("Service start requested");
+    startPolling(SERVICE_RUNNING, 30);
+  } else {
+    WindowsUtils::windowsLog("StartService failed");
+  }
+  return ok;
+}
+
+bool WindowsServiceManager::stopService() {
+  if (!m_has_access) {
+    logger.error() << "Need execute access to stop services";
+    return false;
+  }
+  auto state = getStatus().dwCurrentState;
+  if (state != SERVICE_RUNNING && state != SERVICE_START_PENDING) {
+    logger.warning() << ("Service stop not possible, as its not running");
+  }
+
+  bool ok = ControlService(m_service, SERVICE_CONTROL_STOP, NULL);
+  if (ok) {
+    logger.debug() << ("Service stop requested");
+    startPolling(SERVICE_STOPPED, 10);
+  } else {
+    WindowsUtils::windowsLog("StopService failed");
+  }
+  return ok;
+}