ArchDaemonWindows.h

Go to the documentation of this file.

/*
 * Deskflow -- mouse and keyboard sharing utility
 * SPDX-FileCopyrightText: (C) 2012 - 2016 Symless Ltd.
 * SPDX-FileCopyrightText: (C) 2002 Chris Schoeneman
 * SPDX-License-Identifier: GPL-2.0-only WITH LicenseRef-OpenSSL-Exception
 */
 
#pragma once
 
#include "arch/IArchDaemon.h"
#include "arch/IArchMultithread.h"
 
#define WIN32_LEAN_AND_MEAN
#include <Windows.h>
#include <tchar.h>
 
#include <functional>
 
#include <QString>
 
#define ARCH_DAEMON ArchDaemonWindows

class ArchDaemonWindows : public IArchDaemon
{
public:
  using RunFunc = std::function<int()>;
 
  ArchDaemonWindows();
  ~ArchDaemonWindows() override = default;


  static int runDaemon(RunFunc runFunc);


  static void daemonRunning(bool running);


  static void daemonFailed(int result);


  static UINT getDaemonQuitMessage();
 
  // IArchDaemon overrides
  int daemonize(DaemonFunc const &func) override;
  QString commandLine() const override
  {
    return m_commandLine;
  }
 
private:
  static HKEY openNTServicesKey();
 
  int doRunDaemon(RunFunc runFunc);
  void doDaemonRunning(bool running);
  UINT doGetDaemonQuitMessage();
 
  static void setStatus(DWORD state);
  static void setStatus(DWORD state, DWORD step, DWORD waitHint);
  static void setStatusError(DWORD error);
 
  static bool isRunState(DWORD state);
 
  void serviceMain(DWORD, LPTSTR *);
  static void WINAPI serviceMainEntry(DWORD, LPTSTR *);
 
  void serviceHandler(DWORD ctrl);
  static void WINAPI serviceHandlerEntry(DWORD ctrl);
 
  void start(const QString &name);
  void stop(const QString &name);
 
private:
  class ArchDaemonRunException
  {
  public:
    ArchDaemonRunException(int result) : m_result(result)
    {
    }
 
  public:
    int m_result;
  };
 
private:
  static ArchDaemonWindows *s_daemon;
 
  ArchMutex m_serviceMutex;
  ArchCond m_serviceCondVar;
  DWORD m_serviceState;
  bool m_serviceHandlerWaiting;
  bool m_serviceRunning;
 
  DWORD m_daemonThreadID;
  DaemonFunc m_daemonFunc;
  int m_daemonResult;
 
  SERVICE_STATUS_HANDLE m_statusHandle;
 
  UINT m_quitMessage;
 
  QString m_commandLine;
};