f25947e68b
Capture all process information as they are created to properly kill off handles when process are stopped, also added in a kill everything when the program cleans up all synergy(s|c) processes
117 lines
4.1 KiB
C++
117 lines
4.1 KiB
C++
/*
|
|
* 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;
|
|
|
|
/// @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(); }
|
|
};
|