123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113 |
- /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
- *
- * This is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This software 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 software; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
- * USA.
- */
-
- // -=- LaunchProcess.cxx
-
- #include <rfb_win32/LaunchProcess.h>
- #include <rfb_win32/ModuleFileName.h>
- #include <rfb_win32/Win32Util.h>
- #include <rdr/Exception.h>
- #include <stdio.h>
-
- using namespace rfb;
- using namespace win32;
-
-
- LaunchProcess::LaunchProcess(const TCHAR* exeName_, const TCHAR* params_)
- : exeName(tstrDup(exeName_)), params(tstrDup(params_)) {
- memset(&procInfo, 0, sizeof(procInfo));
- }
-
- LaunchProcess::~LaunchProcess() {
- await();
- }
-
-
- void LaunchProcess::start(HANDLE userToken, bool createConsole) {
- if (procInfo.hProcess && (WaitForSingleObject(procInfo.hProcess, 0) != WAIT_OBJECT_0))
- return;
- await();
- returnCode = STILL_ACTIVE;
-
- DWORD size;
- char desktopName[256];
- char buf[256];
- HDESK desktop = GetThreadDesktop(GetCurrentThreadId());
- if (!GetUserObjectInformation(desktop, UOI_NAME, buf, 256, &size))
- throw rdr::SystemException("unable to launch process", GetLastError());
-
- snprintf(desktopName, 256, "WinSta0\\%s", buf);
-
- // - Create storage for the process startup information
- STARTUPINFO sinfo;
- memset(&sinfo, 0, sizeof(sinfo));
- sinfo.cb = sizeof(sinfo);
- sinfo.lpDesktop = desktopName;
-
- // - Concoct a suitable command-line
- TCharArray exePath;
- if (!tstrContains(exeName.buf, _T('\\'))) {
- ModuleFileName filename;
- TCharArray path; splitPath(filename.buf, &path.buf, 0);
- exePath.buf = new TCHAR[_tcslen(path.buf) + _tcslen(exeName.buf) + 2];
- _stprintf(exePath.buf, _T("%s\\%s"), path.buf, exeName.buf);
- } else {
- exePath.buf = tstrDup(exeName.buf);
- }
-
- // - Start the process
- // Note: We specify the exe's precise path in the ApplicationName parameter,
- // AND include the name as the first part of the CommandLine parameter,
- // because CreateProcess doesn't make ApplicationName argv[0] in C programs.
- TCharArray cmdLine(_tcslen(exeName.buf) + 3 + _tcslen(params.buf) + 1);
- _stprintf(cmdLine.buf, _T("\"%s\" %s"), exeName.buf, params.buf);
- DWORD flags = createConsole ? CREATE_NEW_CONSOLE : CREATE_NO_WINDOW;
- BOOL success;
- if (userToken != INVALID_HANDLE_VALUE)
- success = CreateProcessAsUser(userToken, exePath.buf, cmdLine.buf, 0, 0, FALSE, flags, 0, 0, &sinfo, &procInfo);
- else
- success = CreateProcess(exePath.buf, cmdLine.buf, 0, 0, FALSE, flags, 0, 0, &sinfo, &procInfo);
- if (!success)
- throw rdr::SystemException("unable to launch process", GetLastError());
-
- // Wait for it to finish initialising
- WaitForInputIdle(procInfo.hProcess, 15000);
- }
-
- void LaunchProcess::detach()
- {
- if (!procInfo.hProcess)
- return;
- CloseHandle(procInfo.hProcess);
- CloseHandle(procInfo.hThread);
- memset(&procInfo, 0, sizeof(procInfo));
- }
-
- bool LaunchProcess::await(DWORD timeoutMs) {
- if (!procInfo.hProcess)
- return true;
- DWORD result = WaitForSingleObject(procInfo.hProcess, timeoutMs);
- if (result == WAIT_OBJECT_0) {
- GetExitCodeProcess(procInfo.hProcess, &returnCode);
- detach();
- return true;
- } else if (result == WAIT_FAILED) {
- throw rdr::SystemException("await() failed", GetLastError());
- }
- return false;
- }
|