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 <string>
 
#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
  void installDaemon(
      const char *name, const char *description, const char *pathname, const char *commandLine, const char *dependencies
  ) override;
  void uninstallDaemon(const char *name) override;
  void installDaemon() override;
  void uninstallDaemon() override;
  int daemonize(const char *name, DaemonFunc const &func) override;
  bool canInstallDaemon(const char *name) override;
  bool isDaemonInstalled(const char *name) override;
  std::string 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 char *name);
  void stop(const char *name);
 
private:
  class XArchDaemonRunFailed
  {
  public:
    XArchDaemonRunFailed(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;
 
  std::string m_commandLine;
};
 
#define DEFAULT_DAEMON_INFO _T("Runs the Core process on secure desktops (UAC prompts, login screen, etc).")
 
#define LEGACY_SERVER_DAEMON_NAME _T("Deskflow Server")
#define LEGACY_CLIENT_DAEMON_NAME _T("Deskflow Client")