/*
 * synergy -- 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 "synergy/XSynergy.h"
#include "arch/IArchMultithread.h"

#define WIN32_LEAN_AND_MEAN
#include <Windows.h>
#include <string>
#include <list>
#include <map>

class Thread;
class IpcLogOutputter;
class IpcServer;
class FileLogOutputter;

class MSWindowsWatchdog {
public:
    MSWindowsWatchdog(
        bool autoDetectCommand,
        IpcServer& ipcServer,
        IpcLogOutputter& ipcLogOutputter,
		bool foreground);
    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                startProcessAsUser(String& command, HANDLE userToken, LPSECURITY_ATTRIBUTES sa);
	BOOL                startProcessInForeground(String& command);
    void                sendSas();
    void                getActiveDesktop(LPSECURITY_ATTRIBUTES security);
    void                testOutput(String buffer);
	void				setStartupInfo(STARTUPINFO& si);
	void 				checkChildren();
	/**
	 * @brief This closes the handles held to a child thread
	 * @param pid the ID of the process to kill, will do nothing if PID is not a valid child
	 * @param removeFromMap should the function remove the item from the children map
	 */
	void                closeProcessHandles(unsigned long pid, bool removeFromMap = true);
	/**
	 * @brief This kills off all children's handles created by this process
	 */
	void 				clearAllChildren();

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_foreground;
    std::string         m_activeDesktop;

    /// @brief Save the info of all process made
    ///         We will use this to track all processes we make and
    ///         kill off handels and children that we no longer need
    std::map<unsigned long, PROCESS_INFORMATION>   m_children;
};

//! Relauncher error
/*!
An error occured in the process watchdog.
*/
class XMSWindowsWatchdogError : public XSynergy {
public:
    XMSWindowsWatchdogError(const String& msg) : XSynergy(msg) { }

    // XBase overrides
    virtual String        getWhat() const throw() { return what(); }
};
