Import Upstream version 2.0.0+dfsgupstream/2.0.0+dfsg

1 files changed, 99 insertions, 0 deletions

diff --git a/src/lib/platform/MSWindowsWatchdog.h b/src/lib/platform/MSWindowsWatchdog.h
new file mode 100644
index 0000000..64ffab3
--- /dev/null
+++ b/src/lib/platform/MSWindowsWatchdog.h
@@ -0,0 +1,99 @@
+/*
+ * barrier -- mouse and keyboard sharing utility
+ * Copyright (C) 2012-2016 Symless Ltd.
+ * Copyright (C) 2009 Chris Schoeneman
+ *
+ * This package is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * found in the file LICENSE that should have accompanied this file.
+ * 
+ * This package is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#pragma once
+
+#include "platform/MSWindowsSession.h"
+#include "barrier/XBarrier.h"
+#include "arch/IArchMultithread.h"
+
+#define WIN32_LEAN_AND_MEAN
+#include <Windows.h>
+#include <string>
+#include <list>
+
+class Thread;
+class IpcLogOutputter;
+class IpcServer;
+class FileLogOutputter;
+
+class MSWindowsWatchdog {
+public:
+    MSWindowsWatchdog(
+        bool daemonized,
+        bool autoDetectCommand,
+        IpcServer& ipcServer,
+        IpcLogOutputter& ipcLogOutputter);
+    virtual ~MSWindowsWatchdog();
+
+    void                startAsync();
+    std::string            getCommand() const;
+    void                setCommand(const std::string& command, bool elevate);
+    void                stop();
+    bool                isProcessActive();
+    void                setFileLogOutputter(FileLogOutputter* outputter);
+
+private:
+    void                mainLoop(void*);
+    void                outputLoop(void*);
+    void                shutdownProcess(HANDLE handle, DWORD pid, int timeout);
+    void                shutdownExistingProcesses();
+    HANDLE                duplicateProcessToken(HANDLE process, LPSECURITY_ATTRIBUTES security);
+    HANDLE                getUserToken(LPSECURITY_ATTRIBUTES security);
+    void                startProcess();
+    BOOL                doStartProcessAsUser(String& command, HANDLE userToken, LPSECURITY_ATTRIBUTES sa);
+    BOOL                doStartProcessAsSelf(String& command);
+    void                sendSas();
+    void                getActiveDesktop(LPSECURITY_ATTRIBUTES security);
+    void                testOutput(String buffer);
+
+private:
+    Thread*                m_thread;
+    bool                m_autoDetectCommand;
+    std::string            m_command;
+    bool                m_monitoring;
+    bool                m_commandChanged;
+    HANDLE                m_stdOutWrite;
+    HANDLE                m_stdOutRead;
+    Thread*                m_outputThread;
+    IpcServer&            m_ipcServer;
+    IpcLogOutputter&    m_ipcLogOutputter;
+    bool                m_elevateProcess;
+    MSWindowsSession    m_session;
+    PROCESS_INFORMATION m_processInfo;
+    int                    m_processFailures;
+    bool                m_processRunning;
+    FileLogOutputter*    m_fileLogOutputter;
+    bool                m_autoElevated;
+    ArchMutex            m_mutex;
+    ArchCond            m_condVar;
+    bool                m_ready;
+    bool                m_daemonized;
+};
+
+//! Relauncher error
+/*!
+An error occured in the process watchdog.
+*/
+class XMSWindowsWatchdogError : public XBarrier {
+public:
+    XMSWindowsWatchdogError(const String& msg) : XBarrier(msg) { }
+
+    // XBase overrides
+    virtual String        getWhat() const throw() { return what(); }
+};