2003-06-10 23:46:31 +04:00
|
|
|
/*=========================================================================
|
|
|
|
|
2003-12-16 00:30:46 +03:00
|
|
|
Program: KWSys - Kitware System Library
|
|
|
|
Module: $RCSfile$
|
|
|
|
|
|
|
|
Copyright (c) Kitware, Inc., Insight Consortium. All rights reserved.
|
|
|
|
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
|
|
|
|
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even
|
|
|
|
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
|
|
PURPOSE. See the above copyright notices for more information.
|
2003-06-10 23:46:31 +04:00
|
|
|
|
|
|
|
=========================================================================*/
|
2003-07-10 22:32:30 +04:00
|
|
|
#include "kwsysPrivate.h"
|
|
|
|
#include KWSYS_HEADER(Process.h)
|
2003-06-10 23:46:31 +04:00
|
|
|
|
2005-04-14 00:46:09 +04:00
|
|
|
/* Work-around CMake dependency scanning limitation. This must
|
|
|
|
duplicate the above list of headers. */
|
|
|
|
#if 0
|
|
|
|
# include "Process.h.in"
|
|
|
|
#endif
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/*
|
|
|
|
|
|
|
|
Implementation for Windows
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
On windows, a thread is created to wait for data on each pipe. The
|
|
|
|
threads are synchronized with the main thread to simulate the use of
|
|
|
|
a UNIX-style select system call.
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
On Windows9x platforms, a small WIN32 console application is spawned
|
|
|
|
in-between the calling process and the actual child to be executed.
|
|
|
|
This is to work-around a problem with connecting pipes from WIN16
|
|
|
|
console applications to WIN32 applications.
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
For more information, please check Microsoft Knowledge Base Articles
|
|
|
|
Q190351 and Q150956.
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
2003-06-30 18:44:35 +04:00
|
|
|
#ifdef _MSC_VER
|
|
|
|
#pragma warning (push, 1)
|
|
|
|
#endif
|
2003-06-10 23:46:31 +04:00
|
|
|
#include <windows.h> /* Windows API */
|
|
|
|
#include <string.h> /* strlen, strdup */
|
|
|
|
#include <stdio.h> /* sprintf */
|
|
|
|
#include <io.h> /* _unlink */
|
2004-01-26 21:32:20 +03:00
|
|
|
|
|
|
|
#ifndef _MAX_FNAME
|
|
|
|
#define _MAX_FNAME 4096
|
|
|
|
#endif
|
|
|
|
#ifndef _MAX_PATH
|
|
|
|
#define _MAX_PATH 4096
|
|
|
|
#endif
|
|
|
|
|
2003-06-30 18:44:35 +04:00
|
|
|
#ifdef _MSC_VER
|
|
|
|
#pragma warning (pop)
|
|
|
|
#pragma warning (disable: 4514)
|
|
|
|
#pragma warning (disable: 4706)
|
|
|
|
#endif
|
2003-06-10 23:46:31 +04:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* There are pipes for the process pipeline's stdout and stderr. */
|
|
|
|
#define KWSYSPE_PIPE_COUNT 2
|
|
|
|
#define KWSYSPE_PIPE_STDOUT 0
|
|
|
|
#define KWSYSPE_PIPE_STDERR 1
|
2003-06-10 23:46:31 +04:00
|
|
|
|
|
|
|
/* The maximum amount to read from a pipe at a time. */
|
2003-12-03 17:20:05 +03:00
|
|
|
#define KWSYSPE_PIPE_BUFFER_SIZE 1024
|
2003-06-10 23:46:31 +04:00
|
|
|
|
2004-05-10 21:38:33 +04:00
|
|
|
#define kwsysEncodedWriteArrayProcessFwd9x kwsys_ns(EncodedWriteArrayProcessFwd9x)
|
2003-06-10 23:46:31 +04:00
|
|
|
|
|
|
|
typedef LARGE_INTEGER kwsysProcessTime;
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
typedef struct kwsysProcessCreateInformation_s
|
|
|
|
{
|
|
|
|
/* Windows child startup control data. */
|
|
|
|
STARTUPINFO StartupInfo;
|
|
|
|
|
|
|
|
/* Special error reporting pipe for Win9x forwarding executable. */
|
|
|
|
HANDLE ErrorPipeRead;
|
|
|
|
HANDLE ErrorPipeWrite;
|
|
|
|
} kwsysProcessCreateInformation;
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
typedef struct kwsysProcessPipeData_s kwsysProcessPipeData;
|
2004-07-14 00:50:55 +04:00
|
|
|
static DWORD WINAPI kwsysProcessPipeThreadRead(LPVOID ptd);
|
|
|
|
static void kwsysProcessPipeThreadReadPipe(kwsysProcess* cp,
|
|
|
|
kwsysProcessPipeData* td);
|
2004-07-14 00:23:49 +04:00
|
|
|
static DWORD WINAPI kwsysProcessPipeThreadWake(LPVOID ptd);
|
2004-07-14 00:50:55 +04:00
|
|
|
static void kwsysProcessPipeThreadWakePipe(kwsysProcess* cp,
|
|
|
|
kwsysProcessPipeData* td);
|
2003-12-03 17:20:05 +03:00
|
|
|
static int kwsysProcessInitialize(kwsysProcess* cp);
|
|
|
|
static int kwsysProcessCreate(kwsysProcess* cp, int index,
|
|
|
|
kwsysProcessCreateInformation* si,
|
|
|
|
PHANDLE readEnd);
|
|
|
|
static void kwsysProcessDestroy(kwsysProcess* cp, int event);
|
2003-12-13 22:13:54 +03:00
|
|
|
static int kwsysProcessSetupOutputPipeFile(PHANDLE handle, const char* name);
|
2003-06-10 23:46:31 +04:00
|
|
|
static void kwsysProcessCleanupHandle(PHANDLE h);
|
|
|
|
static void kwsysProcessCleanup(kwsysProcess* cp, int error);
|
2003-08-14 02:08:07 +04:00
|
|
|
static void kwsysProcessCleanErrorMessage(kwsysProcess* cp);
|
2003-06-10 23:46:31 +04:00
|
|
|
static int kwsysProcessGetTimeoutTime(kwsysProcess* cp, double* userTimeout,
|
|
|
|
kwsysProcessTime* timeoutTime);
|
2003-06-30 18:44:35 +04:00
|
|
|
static int kwsysProcessGetTimeoutLeft(kwsysProcessTime* timeoutTime,
|
2005-06-17 22:05:35 +04:00
|
|
|
double* userTimeout,
|
2003-06-10 23:46:31 +04:00
|
|
|
kwsysProcessTime* timeoutLength);
|
|
|
|
static kwsysProcessTime kwsysProcessTimeGetCurrent();
|
|
|
|
static DWORD kwsysProcessTimeToDWORD(kwsysProcessTime t);
|
|
|
|
static double kwsysProcessTimeToDouble(kwsysProcessTime t);
|
|
|
|
static kwsysProcessTime kwsysProcessTimeFromDouble(double d);
|
|
|
|
static int kwsysProcessTimeLess(kwsysProcessTime in1, kwsysProcessTime in2);
|
|
|
|
static kwsysProcessTime kwsysProcessTimeAdd(kwsysProcessTime in1, kwsysProcessTime in2);
|
|
|
|
static kwsysProcessTime kwsysProcessTimeSubtract(kwsysProcessTime in1, kwsysProcessTime in2);
|
2003-12-31 00:23:16 +03:00
|
|
|
static void kwsysProcessSetExitException(kwsysProcess* cp, int code);
|
2004-05-10 22:54:22 +04:00
|
|
|
static void kwsysProcessKillTree(int pid);
|
2004-07-08 02:15:20 +04:00
|
|
|
static void kwsysProcessDisablePipeThreads(kwsysProcess* cp);
|
2003-06-30 18:30:38 +04:00
|
|
|
extern kwsysEXPORT int kwsysEncodedWriteArrayProcessFwd9x(const char* fname);
|
2003-06-10 23:46:31 +04:00
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
2004-07-14 00:50:55 +04:00
|
|
|
/* A structure containing synchronization data for each thread. */
|
|
|
|
typedef struct kwsysProcessPipeSync_s kwsysProcessPipeSync;
|
|
|
|
struct kwsysProcessPipeSync_s
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Handle to the thread. */
|
2003-06-10 23:46:31 +04:00
|
|
|
HANDLE Thread;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Semaphore indicating to the thread that a process has started. */
|
2003-06-10 23:46:31 +04:00
|
|
|
HANDLE Ready;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Semaphore indicating to the thread that it should begin work. */
|
|
|
|
HANDLE Go;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Semaphore indicating thread has reset for another process. */
|
2003-06-10 23:46:31 +04:00
|
|
|
HANDLE Reset;
|
2004-07-14 00:50:55 +04:00
|
|
|
};
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* A structure containing data for each pipe's threads. */
|
|
|
|
struct kwsysProcessPipeData_s
|
|
|
|
{
|
|
|
|
/* ------------- Data managed per instance of kwsysProcess ------------- */
|
|
|
|
|
|
|
|
/* Synchronization data for reading thread. */
|
|
|
|
kwsysProcessPipeSync Reader;
|
2004-07-14 00:23:49 +04:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Synchronization data for waking thread. */
|
|
|
|
kwsysProcessPipeSync Waker;
|
2004-07-14 00:23:49 +04:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Index of this pipe. */
|
|
|
|
int Index;
|
|
|
|
|
|
|
|
/* The kwsysProcess instance owning this pipe. */
|
|
|
|
kwsysProcess* Process;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* ------------- Data managed per call to Execute ------------- */
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Buffer for data read in this pipe's thread. */
|
2003-12-03 17:20:05 +03:00
|
|
|
char DataBuffer[KWSYSPE_PIPE_BUFFER_SIZE];
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* The length of the data stored in the buffer. */
|
|
|
|
DWORD DataLength;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Whether the pipe has been closed. */
|
|
|
|
int Closed;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Handle for the read end of this pipe. */
|
|
|
|
HANDLE Read;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Handle for the write end of this pipe. */
|
|
|
|
HANDLE Write;
|
2003-12-05 19:37:28 +03:00
|
|
|
};
|
2003-06-10 23:46:31 +04:00
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* Structure containing data used to implement the child's execution. */
|
|
|
|
struct kwsysProcess_s
|
|
|
|
{
|
|
|
|
/* ------------- Data managed per instance of kwsysProcess ------------- */
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* The status of the process structure. */
|
2003-06-10 23:46:31 +04:00
|
|
|
int State;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* The command lines to execute. */
|
|
|
|
char** Commands;
|
|
|
|
int NumberOfCommands;
|
|
|
|
|
|
|
|
/* The exit code of each command. */
|
|
|
|
DWORD* CommandExitCodes;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-07-07 17:10:08 +04:00
|
|
|
/* The working directory for the child process. */
|
|
|
|
char* WorkingDirectory;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-08 02:15:20 +04:00
|
|
|
/* Whether to create the child as a detached process. */
|
|
|
|
int OptionDetach;
|
|
|
|
|
|
|
|
/* Whether the child was created as a detached process. */
|
|
|
|
int Detached;
|
|
|
|
|
2003-08-05 23:10:33 +04:00
|
|
|
/* Whether to hide the child process's window. */
|
|
|
|
int HideWindow;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* On Win9x platforms, the path to the forwarding executable. */
|
|
|
|
char* Win9x;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* On Win9x platforms, the resume event for the forwarding executable. */
|
|
|
|
HANDLE Win9xResumeEvent;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* On Win9x platforms, the kill event for the forwarding executable. */
|
|
|
|
HANDLE Win9xKillEvent;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Mutex to protect the shared index used by threads to report data. */
|
|
|
|
HANDLE SharedIndexMutex;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Semaphore used by threads to signal data ready. */
|
|
|
|
HANDLE Full;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Whether we are currently deleting this kwsysProcess instance. */
|
|
|
|
int Deleting;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Data specific to each pipe and its thread. */
|
2003-12-05 19:37:28 +03:00
|
|
|
kwsysProcessPipeData Pipe[KWSYSPE_PIPE_COUNT];
|
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
/* Name of files to which stdin and stdout pipes are attached. */
|
|
|
|
char* PipeFileSTDIN;
|
|
|
|
char* PipeFileSTDOUT;
|
|
|
|
char* PipeFileSTDERR;
|
|
|
|
|
2003-12-17 01:20:01 +03:00
|
|
|
/* Whether each pipe is shared with the parent process. */
|
|
|
|
int PipeSharedSTDIN;
|
|
|
|
int PipeSharedSTDOUT;
|
|
|
|
int PipeSharedSTDERR;
|
|
|
|
|
2003-12-14 21:47:39 +03:00
|
|
|
/* Handle to automatically delete the Win9x forwarding executable. */
|
|
|
|
HANDLE Win9xHandle;
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* ------------- Data managed per call to Execute ------------- */
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-19 01:06:21 +04:00
|
|
|
/* The exceptional behavior that terminated the process, if any. */
|
|
|
|
int ExitException;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-19 01:06:21 +04:00
|
|
|
/* The process exit code. */
|
2003-06-30 18:44:35 +04:00
|
|
|
DWORD ExitCode;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-19 01:06:21 +04:00
|
|
|
/* The process return code, if any. */
|
|
|
|
int ExitValue;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Index of last pipe to report data, if any. */
|
|
|
|
int CurrentIndex;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
|
|
|
/* Index shared by threads to report data. */
|
2003-06-10 23:46:31 +04:00
|
|
|
int SharedIndex;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* The timeout length. */
|
|
|
|
double Timeout;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Time at which the child started. */
|
|
|
|
kwsysProcessTime StartTime;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Time at which the child will timeout. Negative for no timeout. */
|
|
|
|
kwsysProcessTime TimeoutTime;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Flag for whether the process was killed. */
|
|
|
|
int Killed;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Flag for whether the timeout expired. */
|
|
|
|
int TimeoutExpired;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Flag for whether the process has terminated. */
|
|
|
|
int Terminated;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* The number of pipes still open during execution and while waiting
|
|
|
|
for pipes to close after process termination. */
|
|
|
|
int PipesLeft;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Buffer for error messages (possibly from Win9x child). */
|
2003-12-03 17:20:05 +03:00
|
|
|
char ErrorMessage[KWSYSPE_PIPE_BUFFER_SIZE+1];
|
2003-06-10 23:46:31 +04:00
|
|
|
|
2003-12-31 00:23:16 +03:00
|
|
|
/* Description for the ExitException. */
|
|
|
|
char ExitExceptionString[KWSYSPE_PIPE_BUFFER_SIZE+1];
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Windows process information data. */
|
2003-12-03 17:20:05 +03:00
|
|
|
PROCESS_INFORMATION* ProcessInformation;
|
|
|
|
|
|
|
|
/* Data and process termination events for which to wait. */
|
|
|
|
PHANDLE ProcessEvents;
|
|
|
|
int ProcessEventsLength;
|
2003-12-13 22:13:54 +03:00
|
|
|
|
|
|
|
/* Real working directory of our own process. */
|
|
|
|
DWORD RealWorkingDirectoryLength;
|
|
|
|
char* RealWorkingDirectory;
|
2003-06-10 23:46:31 +04:00
|
|
|
};
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
kwsysProcess* kwsysProcess_New()
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
|
|
|
/* Process control structure. */
|
|
|
|
kwsysProcess* cp;
|
|
|
|
|
|
|
|
/* Path to Win9x forwarding executable. */
|
|
|
|
char* win9x = 0;
|
|
|
|
|
|
|
|
/* Windows version number data. */
|
|
|
|
OSVERSIONINFO osv;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Allocate a process control structure. */
|
|
|
|
cp = (kwsysProcess*)malloc(sizeof(kwsysProcess));
|
2003-06-13 00:18:07 +04:00
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
/* Could not allocate memory for the control structure. */
|
|
|
|
return 0;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
ZeroMemory(cp, sizeof(*cp));
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-17 01:20:01 +03:00
|
|
|
/* Share stdin with the parent process by default. */
|
|
|
|
cp->PipeSharedSTDIN = 1;
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Set initial status. */
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->State = kwsysProcess_State_Starting;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Choose a method of running the child based on version of
|
|
|
|
windows. */
|
|
|
|
ZeroMemory(&osv, sizeof(osv));
|
|
|
|
osv.dwOSVersionInfoSize = sizeof(osv);
|
|
|
|
GetVersionEx(&osv);
|
|
|
|
if(osv.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS)
|
|
|
|
{
|
|
|
|
/* This is Win9x. We need the console forwarding executable to
|
|
|
|
work-around a Windows 9x bug. */
|
|
|
|
char fwdName[_MAX_FNAME+1] = "";
|
|
|
|
char tempDir[_MAX_PATH+1] = "";
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* We will try putting the executable in the system temp
|
2003-07-07 04:40:51 +04:00
|
|
|
directory. Note that the returned path already has a trailing
|
|
|
|
slash. */
|
|
|
|
DWORD length = GetTempPath(_MAX_PATH+1, tempDir);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Construct the executable name from the process id and kwsysProcess
|
|
|
|
instance. This should be unique. */
|
2003-12-14 21:47:39 +03:00
|
|
|
sprintf(fwdName, KWSYS_NAMESPACE_STRING "pew9xfwd_%u_%p.exe",
|
|
|
|
GetCurrentProcessId(), cp);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-07-07 04:40:51 +04:00
|
|
|
/* If we have a temp directory, use it. */
|
2003-06-10 23:46:31 +04:00
|
|
|
if(length > 0 && length <= _MAX_PATH)
|
|
|
|
{
|
|
|
|
/* Allocate a buffer to hold the forwarding executable path. */
|
2003-07-07 04:40:51 +04:00
|
|
|
size_t tdlen = strlen(tempDir);
|
2003-06-10 23:46:31 +04:00
|
|
|
win9x = (char*)malloc(tdlen + strlen(fwdName) + 2);
|
|
|
|
if(!win9x)
|
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Construct the full path to the forwarding executable. */
|
2003-07-07 04:40:51 +04:00
|
|
|
sprintf(win9x, "%s%s", tempDir, fwdName);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* If we found a place to put the forwarding executable, try to
|
|
|
|
write it. */
|
|
|
|
if(win9x)
|
|
|
|
{
|
|
|
|
if(!kwsysEncodedWriteArrayProcessFwd9x(win9x))
|
|
|
|
{
|
|
|
|
/* Failed to create forwarding executable. Give up. */
|
|
|
|
free(win9x);
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-14 21:47:39 +03:00
|
|
|
|
|
|
|
/* Get a handle to the file that will delete it when closed. */
|
|
|
|
cp->Win9xHandle = CreateFile(win9x, GENERIC_READ, FILE_SHARE_READ, 0,
|
|
|
|
OPEN_EXISTING, FILE_FLAG_DELETE_ON_CLOSE, 0);
|
|
|
|
if(cp->Win9xHandle == INVALID_HANDLE_VALUE)
|
|
|
|
{
|
|
|
|
/* We were not able to get a read handle for the forwarding
|
|
|
|
executable. It will not be deleted properly. Give up. */
|
|
|
|
_unlink(win9x);
|
|
|
|
free(win9x);
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* Failed to find a place to put forwarding executable. */
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Save the path to the forwarding executable. */
|
2003-06-10 23:46:31 +04:00
|
|
|
cp->Win9x = win9x;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Initially no thread owns the mutex. Initialize semaphore to 1. */
|
|
|
|
if(!(cp->SharedIndexMutex = CreateSemaphore(0, 1, 1, 0)))
|
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Initially no data are available. Initialize semaphore to 0. */
|
|
|
|
if(!(cp->Full = CreateSemaphore(0, 0, 1, 0)))
|
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
|
|
|
SECURITY_ATTRIBUTES sa;
|
|
|
|
ZeroMemory(&sa, sizeof(sa));
|
|
|
|
sa.nLength = sizeof(sa);
|
|
|
|
sa.bInheritHandle = TRUE;
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Create an event to tell the forwarding executable to resume the
|
|
|
|
child. */
|
|
|
|
if(!(cp->Win9xResumeEvent = CreateEvent(&sa, TRUE, 0, 0)))
|
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Create an event to tell the forwarding executable to kill the
|
|
|
|
child. */
|
2003-06-10 23:46:31 +04:00
|
|
|
if(!(cp->Win9xKillEvent = CreateEvent(&sa, TRUE, 0, 0)))
|
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Create the thread to read each pipe. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
DWORD dummy=0;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Assign the thread its index. */
|
|
|
|
cp->Pipe[i].Index = i;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Give the thread a pointer back to the kwsysProcess instance. */
|
|
|
|
cp->Pipe[i].Process = cp;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* No process is yet running. Initialize semaphore to 0. */
|
|
|
|
if(!(cp->Pipe[i].Reader.Ready = CreateSemaphore(0, 0, 1, 0)))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* The pipe is not yet reset. Initialize semaphore to 0. */
|
2004-07-14 00:50:55 +04:00
|
|
|
if(!(cp->Pipe[i].Reader.Reset = CreateSemaphore(0, 0, 1, 0)))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* The thread's buffer is initially empty. Initialize semaphore to 1. */
|
2004-07-14 00:50:55 +04:00
|
|
|
if(!(cp->Pipe[i].Reader.Go = CreateSemaphore(0, 1, 1, 0)))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Create the reading thread. It will block immediately. The
|
|
|
|
thread will not make deeply nested calls, so we need only a
|
|
|
|
small stack. */
|
|
|
|
if(!(cp->Pipe[i].Reader.Thread = CreateThread(0, 1024,
|
|
|
|
kwsysProcessPipeThreadRead,
|
|
|
|
&cp->Pipe[i], 0, &dummy)))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2004-07-14 00:23:49 +04:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* No process is yet running. Initialize semaphore to 0. */
|
|
|
|
if(!(cp->Pipe[i].Waker.Ready = CreateSemaphore(0, 0, 1, 0)))
|
2004-07-14 00:23:49 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* The pipe is not yet reset. Initialize semaphore to 0. */
|
|
|
|
if(!(cp->Pipe[i].Waker.Reset = CreateSemaphore(0, 0, 1, 0)))
|
2004-07-14 00:23:49 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* The waker should not wake immediately. Initialize semaphore to 0. */
|
|
|
|
if(!(cp->Pipe[i].Waker.Go = CreateSemaphore(0, 0, 1, 0)))
|
2004-07-14 00:23:49 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Create the waking thread. It will block immediately. The
|
|
|
|
thread will not make deeply nested calls, so we need only a
|
|
|
|
small stack. */
|
2004-07-14 00:50:55 +04:00
|
|
|
if(!(cp->Pipe[i].Waker.Thread = CreateThread(0, 1024,
|
|
|
|
kwsysProcessPipeThreadWake,
|
|
|
|
&cp->Pipe[i], 0, &dummy)))
|
2004-07-14 00:23:49 +04:00
|
|
|
{
|
|
|
|
kwsysProcess_Delete(cp);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
return cp;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcess_Delete(kwsysProcess* cp)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
2003-12-13 18:36:22 +03:00
|
|
|
/* Make sure we have an instance. */
|
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* If the process is executing, wait for it to finish. */
|
2003-06-19 01:06:21 +04:00
|
|
|
if(cp->State == kwsysProcess_State_Executing)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2004-07-08 02:15:20 +04:00
|
|
|
if(cp->Detached)
|
|
|
|
{
|
|
|
|
kwsysProcess_Disown(cp);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
kwsysProcess_WaitForExit(cp, 0);
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* We are deleting the kwsysProcess instance. */
|
|
|
|
cp->Deleting = 1;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Terminate each of the threads. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Terminate this reading thread. */
|
|
|
|
if(cp->Pipe[i].Reader.Thread)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
/* Signal the thread we are ready for it. It will terminate
|
|
|
|
immediately since Deleting is set. */
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(cp->Pipe[i].Reader.Ready, 1, 0);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Wait for the thread to exit. */
|
2004-07-14 00:50:55 +04:00
|
|
|
WaitForSingleObject(cp->Pipe[i].Reader.Thread, INFINITE);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Close the handle to the thread. */
|
2004-07-14 00:50:55 +04:00
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Reader.Thread);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Terminate this waking thread. */
|
|
|
|
if(cp->Pipe[i].Waker.Thread)
|
2004-07-14 00:23:49 +04:00
|
|
|
{
|
|
|
|
/* Signal the thread we are ready for it. It will terminate
|
|
|
|
immediately since Deleting is set. */
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(cp->Pipe[i].Waker.Ready, 1, 0);
|
2004-07-14 00:23:49 +04:00
|
|
|
|
|
|
|
/* Wait for the thread to exit. */
|
2004-07-14 00:50:55 +04:00
|
|
|
WaitForSingleObject(cp->Pipe[i].Waker.Thread, INFINITE);
|
2004-07-14 00:23:49 +04:00
|
|
|
|
|
|
|
/* Close the handle to the thread. */
|
2004-07-14 00:50:55 +04:00
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Waker.Thread);
|
2004-07-14 00:23:49 +04:00
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Cleanup the pipe's semaphores. */
|
2004-07-14 00:50:55 +04:00
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Reader.Ready);
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Reader.Go);
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Reader.Reset);
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Waker.Ready);
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Waker.Go);
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Waker.Reset);
|
2003-12-05 19:37:28 +03:00
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Close the shared semaphores. */
|
|
|
|
kwsysProcessCleanupHandle(&cp->SharedIndexMutex);
|
|
|
|
kwsysProcessCleanupHandle(&cp->Full);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Close the Win9x resume and kill event handles. */
|
2003-06-10 23:46:31 +04:00
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
kwsysProcessCleanupHandle(&cp->Win9xResumeEvent);
|
2003-06-10 23:46:31 +04:00
|
|
|
kwsysProcessCleanupHandle(&cp->Win9xKillEvent);
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Free memory. */
|
|
|
|
kwsysProcess_SetCommand(cp, 0);
|
2003-07-07 17:10:08 +04:00
|
|
|
kwsysProcess_SetWorkingDirectory(cp, 0);
|
2003-12-13 22:13:54 +03:00
|
|
|
kwsysProcess_SetPipeFile(cp, kwsysProcess_Pipe_STDIN, 0);
|
|
|
|
kwsysProcess_SetPipeFile(cp, kwsysProcess_Pipe_STDOUT, 0);
|
|
|
|
kwsysProcess_SetPipeFile(cp, kwsysProcess_Pipe_STDERR, 0);
|
2003-12-03 17:20:05 +03:00
|
|
|
if(cp->CommandExitCodes)
|
|
|
|
{
|
|
|
|
free(cp->CommandExitCodes);
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
2003-12-14 21:48:18 +03:00
|
|
|
/* Close our handle to the forwarding executable file. This will
|
2003-12-14 21:47:39 +03:00
|
|
|
cause it to be deleted. */
|
|
|
|
kwsysProcessCleanupHandle(&cp->Win9xHandle);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
free(cp);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
2003-12-03 17:20:05 +03:00
|
|
|
int kwsysProcess_SetCommand(kwsysProcess* cp, char const* const* command)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
int i;
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
|
|
|
free(cp->Commands[i]);
|
|
|
|
}
|
|
|
|
cp->NumberOfCommands = 0;
|
|
|
|
if(cp->Commands)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
free(cp->Commands);
|
|
|
|
cp->Commands = 0;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
if(command)
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
return kwsysProcess_AddCommand(cp, command);
|
|
|
|
}
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_AddCommand(kwsysProcess* cp, char const* const* command)
|
|
|
|
{
|
|
|
|
int newNumberOfCommands;
|
|
|
|
char** newCommands;
|
|
|
|
|
|
|
|
/* Make sure we have a command to add. */
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp || !command)
|
2003-12-03 17:20:05 +03:00
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Allocate a new array for command pointers. */
|
|
|
|
newNumberOfCommands = cp->NumberOfCommands + 1;
|
|
|
|
if(!(newCommands = (char**)malloc(sizeof(char*) * newNumberOfCommands)))
|
|
|
|
{
|
|
|
|
/* Out of memory. */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Copy any existing commands into the new array. */
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
|
|
|
newCommands[i] = cp->Commands[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* We need to construct a single string representing the command
|
2003-07-04 02:33:53 +04:00
|
|
|
and its arguments. We will surround each argument containing
|
|
|
|
spaces with double-quotes. Inside a double-quoted argument, we
|
|
|
|
need to escape double-quotes and all backslashes before them.
|
|
|
|
We also need to escape backslashes at the end of an argument
|
|
|
|
because they come before the closing double-quote for the
|
|
|
|
argument. */
|
2003-12-03 17:20:05 +03:00
|
|
|
{
|
|
|
|
char* cmd;
|
|
|
|
char const* const* arg;
|
|
|
|
int length = 0;
|
|
|
|
/* First determine the length of the final string. */
|
|
|
|
for(arg = command; *arg; ++arg)
|
|
|
|
{
|
|
|
|
/* Keep track of how many backslashes have been encountered in a
|
|
|
|
row in this argument. */
|
|
|
|
int backslashes = 0;
|
|
|
|
int spaces = 0;
|
|
|
|
const char* c;
|
2003-06-10 23:46:31 +04:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Scan the string for spaces. If there are no spaces, we can
|
2003-07-04 02:33:53 +04:00
|
|
|
pass the argument verbatim. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(c=*arg; *c; ++c)
|
|
|
|
{
|
|
|
|
if(*c == ' ' || *c == '\t')
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
spaces = 1;
|
|
|
|
break;
|
2003-07-04 02:33:53 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Add the length of the argument, plus 1 for the space
|
2003-07-04 02:33:53 +04:00
|
|
|
separating the arguments. */
|
2003-12-03 17:20:05 +03:00
|
|
|
length += (int)strlen(*arg) + 1;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
if(spaces)
|
|
|
|
{
|
|
|
|
/* Add 2 for double quotes since spaces are present. */
|
|
|
|
length += 2;
|
2003-07-04 02:33:53 +04:00
|
|
|
|
|
|
|
/* Scan the string to find characters that need escaping. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(c=*arg; *c; ++c)
|
|
|
|
{
|
|
|
|
if(*c == '\\')
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Found a backslash. It may need to be escaped later. */
|
|
|
|
++backslashes;
|
|
|
|
}
|
|
|
|
else if(*c == '"')
|
|
|
|
{
|
|
|
|
/* Found a double-quote. We need to escape it and all
|
|
|
|
immediately preceding backslashes. */
|
|
|
|
length += backslashes + 1;
|
|
|
|
backslashes = 0;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* Found another character. This eliminates the possibility
|
|
|
|
that any immediately preceding backslashes will be
|
|
|
|
escaped. */
|
|
|
|
backslashes = 0;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-07-04 02:33:53 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* We need to escape all ending backslashes. */
|
|
|
|
length += backslashes;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Allocate enough space for the command. We do not need an extra
|
2003-06-10 23:46:31 +04:00
|
|
|
byte for the terminating null because we allocated a space for
|
|
|
|
the first argument that we will not use. */
|
2003-12-03 17:20:05 +03:00
|
|
|
newCommands[cp->NumberOfCommands] = (char*)malloc(length);
|
|
|
|
if(!newCommands[cp->NumberOfCommands])
|
|
|
|
{
|
|
|
|
/* Out of memory. */
|
|
|
|
free(newCommands);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Construct the command line in the allocated buffer. */
|
|
|
|
cmd = newCommands[cp->NumberOfCommands];
|
|
|
|
for(arg = command; *arg; ++arg)
|
|
|
|
{
|
|
|
|
/* Keep track of how many backslashes have been encountered in a
|
|
|
|
row in an argument. */
|
|
|
|
int backslashes = 0;
|
|
|
|
int spaces = 0;
|
|
|
|
const char* c;
|
2003-07-04 02:33:53 +04:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Scan the string for spaces. If there are no spaces, we can
|
2003-07-04 02:33:53 +04:00
|
|
|
pass the argument verbatim. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(c=*arg; *c; ++c)
|
|
|
|
{
|
|
|
|
if(*c == ' ' || *c == '\t')
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
spaces = 1;
|
|
|
|
break;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
}
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Add the separating space if this is not the first argument. */
|
|
|
|
if(arg != command)
|
|
|
|
{
|
|
|
|
*cmd++ = ' ';
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
if(spaces)
|
|
|
|
{
|
|
|
|
/* Add the opening double-quote for this argument. */
|
|
|
|
*cmd++ = '"';
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-07-04 02:33:53 +04:00
|
|
|
/* Add the characters of the argument, possibly escaping them. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(c=*arg; *c; ++c)
|
|
|
|
{
|
|
|
|
if(*c == '\\')
|
|
|
|
{
|
|
|
|
/* Found a backslash. It may need to be escaped later. */
|
|
|
|
++backslashes;
|
|
|
|
*cmd++ = '\\';
|
|
|
|
}
|
|
|
|
else if(*c == '"')
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Add enough backslashes to escape any that preceded the
|
|
|
|
double-quote. */
|
|
|
|
while(backslashes > 0)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
--backslashes;
|
2003-06-10 23:46:31 +04:00
|
|
|
*cmd++ = '\\';
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Add the backslash to escape the double-quote. */
|
|
|
|
*cmd++ = '\\';
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Add the double-quote itself. */
|
|
|
|
*cmd++ = '"';
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
else
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* We encountered a normal character. This eliminates any
|
|
|
|
escaping needed for preceding backslashes. Add the
|
|
|
|
character. */
|
|
|
|
backslashes = 0;
|
|
|
|
*cmd++ = *c;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Add enough backslashes to escape any trailing ones. */
|
|
|
|
while(backslashes > 0)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
--backslashes;
|
|
|
|
*cmd++ = '\\';
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Add the closing double-quote for this argument. */
|
|
|
|
*cmd++ = '"';
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* No spaces. Add the argument verbatim. */
|
|
|
|
for(c=*arg; *c; ++c)
|
|
|
|
{
|
|
|
|
*cmd++ = *c;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Add the terminating null character to the command line. */
|
|
|
|
*cmd = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Save the new array of commands. */
|
|
|
|
free(cp->Commands);
|
|
|
|
cp->Commands = newCommands;
|
|
|
|
cp->NumberOfCommands = newNumberOfCommands;
|
|
|
|
return 1;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcess_SetTimeout(kwsysProcess* cp, double timeout)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
cp->Timeout = timeout;
|
2003-12-13 18:36:22 +03:00
|
|
|
if(cp->Timeout < 0)
|
|
|
|
{
|
|
|
|
cp->Timeout = 0;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
2003-07-07 17:10:08 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
2003-12-13 22:13:54 +03:00
|
|
|
int kwsysProcess_SetWorkingDirectory(kwsysProcess* cp, const char* dir)
|
2003-07-07 17:10:08 +04:00
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp)
|
|
|
|
{
|
2003-12-13 22:13:54 +03:00
|
|
|
return 0;
|
2003-12-13 18:36:22 +03:00
|
|
|
}
|
2003-07-07 17:10:08 +04:00
|
|
|
if(cp->WorkingDirectory)
|
|
|
|
{
|
|
|
|
free(cp->WorkingDirectory);
|
|
|
|
cp->WorkingDirectory = 0;
|
|
|
|
}
|
|
|
|
if(dir && dir[0])
|
|
|
|
{
|
|
|
|
/* We must convert the working directory to a full path. */
|
|
|
|
DWORD length = GetFullPathName(dir, 0, 0, 0);
|
|
|
|
if(length > 0)
|
|
|
|
{
|
|
|
|
cp->WorkingDirectory = (char*)malloc(length);
|
2003-12-13 22:13:54 +03:00
|
|
|
if(!cp->WorkingDirectory)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
2003-07-07 17:10:08 +04:00
|
|
|
if(!GetFullPathName(dir, length, cp->WorkingDirectory, 0))
|
|
|
|
{
|
|
|
|
free(cp->WorkingDirectory);
|
|
|
|
cp->WorkingDirectory = 0;
|
2003-12-13 22:13:54 +03:00
|
|
|
return 0;
|
2003-07-07 17:10:08 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2003-12-13 22:13:54 +03:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_SetPipeFile(kwsysProcess* cp, int pipe, const char* file)
|
|
|
|
{
|
|
|
|
char** pfile;
|
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
switch(pipe)
|
|
|
|
{
|
|
|
|
case kwsysProcess_Pipe_STDIN: pfile = &cp->PipeFileSTDIN; break;
|
|
|
|
case kwsysProcess_Pipe_STDOUT: pfile = &cp->PipeFileSTDOUT; break;
|
|
|
|
case kwsysProcess_Pipe_STDERR: pfile = &cp->PipeFileSTDERR; break;
|
|
|
|
default: return 0;
|
|
|
|
}
|
|
|
|
if(*pfile)
|
|
|
|
{
|
|
|
|
free(*pfile);
|
|
|
|
*pfile = 0;
|
|
|
|
}
|
|
|
|
if(file)
|
|
|
|
{
|
|
|
|
*pfile = malloc(strlen(file)+1);
|
|
|
|
if(!*pfile)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
strcpy(*pfile, file);
|
|
|
|
}
|
2003-12-17 01:20:01 +03:00
|
|
|
|
|
|
|
/* If we are redirecting the pipe, do not share it. */
|
|
|
|
if(*pfile)
|
|
|
|
{
|
|
|
|
kwsysProcess_SetPipeShared(cp, pipe, 0);
|
|
|
|
}
|
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
return 1;
|
2003-07-07 17:10:08 +04:00
|
|
|
}
|
|
|
|
|
2003-12-17 01:20:01 +03:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcess_SetPipeShared(kwsysProcess* cp, int pipe, int shared)
|
|
|
|
{
|
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
switch(pipe)
|
|
|
|
{
|
|
|
|
case kwsysProcess_Pipe_STDIN: cp->PipeSharedSTDIN = shared?1:0; break;
|
|
|
|
case kwsysProcess_Pipe_STDOUT: cp->PipeSharedSTDOUT = shared?1:0; break;
|
|
|
|
case kwsysProcess_Pipe_STDERR: cp->PipeSharedSTDERR = shared?1:0; break;
|
|
|
|
default: return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If we are sharing the pipe, do not redirect it to a file. */
|
|
|
|
if(shared)
|
|
|
|
{
|
|
|
|
kwsysProcess_SetPipeFile(cp, pipe, 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-08-05 22:27:47 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_GetOption(kwsysProcess* cp, int optionId)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2003-08-05 23:10:33 +04:00
|
|
|
switch(optionId)
|
|
|
|
{
|
2004-07-08 02:15:20 +04:00
|
|
|
case kwsysProcess_Option_Detach: return cp->OptionDetach;
|
2003-08-05 23:10:33 +04:00
|
|
|
case kwsysProcess_Option_HideWindow: return cp->HideWindow;
|
|
|
|
default: return 0;
|
|
|
|
}
|
2003-08-05 22:27:47 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcess_SetOption(kwsysProcess* cp, int optionId, int value)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2003-08-05 23:10:33 +04:00
|
|
|
switch(optionId)
|
|
|
|
{
|
2004-07-08 02:15:20 +04:00
|
|
|
case kwsysProcess_Option_Detach: cp->OptionDetach = value; break;
|
2003-08-05 23:10:33 +04:00
|
|
|
case kwsysProcess_Option_HideWindow: cp->HideWindow = value; break;
|
|
|
|
default: break;
|
|
|
|
}
|
2003-08-05 22:27:47 +04:00
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_GetState(kwsysProcess* cp)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
return cp? cp->State : kwsysProcess_State_Error;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
2003-06-19 01:06:21 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_GetExitException(kwsysProcess* cp)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
return cp? cp->ExitException : kwsysProcess_Exception_Other;
|
2003-06-19 01:06:21 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_GetExitValue(kwsysProcess* cp)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
return cp? cp->ExitValue : -1;
|
2003-06-19 01:06:21 +04:00
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_GetExitCode(kwsysProcess* cp)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
return cp? cp->ExitCode : 0;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
const char* kwsysProcess_GetErrorString(kwsysProcess* cp)
|
|
|
|
{
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp)
|
|
|
|
{
|
2003-12-31 00:23:16 +03:00
|
|
|
return "Process management structure could not be allocated";
|
2003-12-13 18:36:22 +03:00
|
|
|
}
|
|
|
|
else if(cp->State == kwsysProcess_State_Error)
|
2003-06-19 01:06:21 +04:00
|
|
|
{
|
|
|
|
return cp->ErrorMessage;
|
|
|
|
}
|
2003-12-31 00:23:16 +03:00
|
|
|
return "Success";
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
const char* kwsysProcess_GetExceptionString(kwsysProcess* cp)
|
|
|
|
{
|
|
|
|
if(!cp)
|
|
|
|
{
|
|
|
|
return "GetExceptionString called with NULL process management structure";
|
|
|
|
}
|
|
|
|
else if(cp->State == kwsysProcess_State_Exception)
|
|
|
|
{
|
|
|
|
return cp->ExitExceptionString;
|
|
|
|
}
|
|
|
|
return "No exception";
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcess_Execute(kwsysProcess* cp)
|
|
|
|
{
|
2003-11-04 17:00:18 +03:00
|
|
|
int i;
|
2003-06-10 23:46:31 +04:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Child startup control data. */
|
|
|
|
kwsysProcessCreateInformation si;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Do not execute a second time. */
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp || cp->State == kwsysProcess_State_Executing)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Initialize the control structure for a new process. */
|
|
|
|
if(!kwsysProcessInitialize(cp))
|
|
|
|
{
|
|
|
|
strcpy(cp->ErrorMessage, "Out of memory");
|
|
|
|
cp->State = kwsysProcess_State_Error;
|
|
|
|
return;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
/* Save the real working directory of this process and change to
|
|
|
|
the working directory for the child processes. This is needed
|
|
|
|
to make pipe file paths evaluate correctly. */
|
|
|
|
if(cp->WorkingDirectory)
|
|
|
|
{
|
|
|
|
if(!GetCurrentDirectory(cp->RealWorkingDirectoryLength,
|
|
|
|
cp->RealWorkingDirectory))
|
|
|
|
{
|
|
|
|
kwsysProcessCleanup(cp, 1);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
SetCurrentDirectory(cp->WorkingDirectory);
|
|
|
|
}
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Reset the Win9x resume and kill events. */
|
2003-06-10 23:46:31 +04:00
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
if(!ResetEvent(cp->Win9xResumeEvent))
|
|
|
|
{
|
|
|
|
kwsysProcessCleanup(cp, 1);
|
|
|
|
return;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
if(!ResetEvent(cp->Win9xKillEvent))
|
|
|
|
{
|
|
|
|
kwsysProcessCleanup(cp, 1);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Initialize startup info data. */
|
|
|
|
ZeroMemory(&si, sizeof(si));
|
|
|
|
si.StartupInfo.cb = sizeof(si.StartupInfo);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Decide whether a child window should be shown. */
|
|
|
|
si.StartupInfo.dwFlags |= STARTF_USESHOWWINDOW;
|
|
|
|
si.StartupInfo.wShowWindow =
|
|
|
|
(unsigned short)(cp->HideWindow?SW_HIDE:SW_SHOWDEFAULT);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Connect the child's output pipes to the threads. */
|
2003-12-05 19:19:43 +03:00
|
|
|
si.StartupInfo.dwFlags |= STARTF_USESTDHANDLES;
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Create stderr pipe to be shared by all processes in the pipeline.
|
|
|
|
Neither end is directly inherited. */
|
|
|
|
if(!CreatePipe(&cp->Pipe[KWSYSPE_PIPE_STDERR].Read,
|
|
|
|
&cp->Pipe[KWSYSPE_PIPE_STDERR].Write, 0, 0))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
kwsysProcessCleanup(cp, 1);
|
|
|
|
return;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Create an inherited duplicate of the write end, but do not
|
|
|
|
close the non-inherited version. We need to keep it open
|
|
|
|
to use in waking up the pipe threads. */
|
|
|
|
if(!DuplicateHandle(GetCurrentProcess(), cp->Pipe[KWSYSPE_PIPE_STDERR].Write,
|
|
|
|
GetCurrentProcess(), &si.StartupInfo.hStdError,
|
|
|
|
0, TRUE, DUPLICATE_SAME_ACCESS))
|
|
|
|
{
|
|
|
|
kwsysProcessCleanup(cp, 1);
|
|
|
|
kwsysProcessCleanupHandle(&si.StartupInfo.hStdError);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
/* Replace the stderr pipe with a file if requested. In this case
|
|
|
|
the pipe thread will still run but never report data. */
|
|
|
|
if(cp->PipeFileSTDERR)
|
|
|
|
{
|
|
|
|
if(!kwsysProcessSetupOutputPipeFile(&si.StartupInfo.hStdError,
|
|
|
|
cp->PipeFileSTDERR))
|
|
|
|
{
|
|
|
|
kwsysProcessCleanup(cp, 1);
|
|
|
|
kwsysProcessCleanupHandle(&si.StartupInfo.hStdError);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-12-17 01:20:01 +03:00
|
|
|
/* Replace the stderr pipe with the parent process's if requested.
|
|
|
|
In this case the pipe thread will still run but never report
|
|
|
|
data. */
|
|
|
|
if(cp->PipeSharedSTDERR)
|
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&si.StartupInfo.hStdError);
|
|
|
|
si.StartupInfo.hStdError = GetStdHandle(STD_ERROR_HANDLE);
|
|
|
|
}
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Create the pipeline of processes. */
|
|
|
|
{
|
|
|
|
HANDLE readEnd = 0;
|
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
|
|
|
if(kwsysProcessCreate(cp, i, &si, &readEnd))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
cp->ProcessEvents[i+1] = cp->ProcessInformation[i].hProcess;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
else
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
kwsysProcessCleanup(cp, 1);
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Release resources that may have been allocated for this
|
|
|
|
process before an error occurred. */
|
|
|
|
kwsysProcessCleanupHandle(&readEnd);
|
2003-12-17 01:20:01 +03:00
|
|
|
if(si.StartupInfo.hStdInput != GetStdHandle(STD_INPUT_HANDLE))
|
2003-12-03 17:20:05 +03:00
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&si.StartupInfo.hStdInput);
|
|
|
|
}
|
2003-12-17 01:20:01 +03:00
|
|
|
if(si.StartupInfo.hStdOutput != GetStdHandle(STD_OUTPUT_HANDLE))
|
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&si.StartupInfo.hStdOutput);
|
|
|
|
}
|
2005-04-22 00:05:09 +04:00
|
|
|
if(si.StartupInfo.hStdError != GetStdHandle(STD_ERROR_HANDLE))
|
2003-12-17 01:20:01 +03:00
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&si.StartupInfo.hStdError);
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
kwsysProcessCleanupHandle(&si.ErrorPipeRead);
|
|
|
|
kwsysProcessCleanupHandle(&si.ErrorPipeWrite);
|
2003-06-10 23:46:31 +04:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Save a handle to the output pipe for the last process. */
|
|
|
|
cp->Pipe[KWSYSPE_PIPE_STDOUT].Read = readEnd;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Close the inherited handles to the stderr pipe shared by all
|
|
|
|
processes in the pipeline. */
|
2005-04-22 00:05:09 +04:00
|
|
|
if(si.StartupInfo.hStdError != GetStdHandle(STD_ERROR_HANDLE))
|
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&si.StartupInfo.hStdError);
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
/* Restore the working directory. */
|
|
|
|
if(cp->RealWorkingDirectory)
|
|
|
|
{
|
|
|
|
SetCurrentDirectory(cp->RealWorkingDirectory);
|
|
|
|
free(cp->RealWorkingDirectory);
|
|
|
|
cp->RealWorkingDirectory = 0;
|
|
|
|
}
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* The timeout period starts now. */
|
|
|
|
cp->StartTime = kwsysProcessTimeGetCurrent();
|
|
|
|
cp->TimeoutTime = kwsysProcessTimeFromDouble(-1);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* All processes in the pipeline have been started in suspended
|
|
|
|
mode. Resume them all now. */
|
2003-06-10 23:46:31 +04:00
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
SetEvent(cp->Win9xResumeEvent);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
|
|
|
ResumeThread(cp->ProcessInformation[i].hThread);
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
2003-12-05 19:37:28 +03:00
|
|
|
/* ---- It is no longer safe to call kwsysProcessCleanup. ----- */
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Tell the pipe threads that a process has started. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(cp->Pipe[i].Reader.Ready, 1, 0);
|
|
|
|
ReleaseSemaphore(cp->Pipe[i].Waker.Ready, 1, 0);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* We don't care about the children's main threads. */
|
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&cp->ProcessInformation[i].hThread);
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* No pipe has reported data. */
|
2003-12-03 17:20:05 +03:00
|
|
|
cp->CurrentIndex = KWSYSPE_PIPE_COUNT;
|
|
|
|
cp->PipesLeft = KWSYSPE_PIPE_COUNT;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* The process has now started. */
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->State = kwsysProcess_State_Executing;
|
2004-07-08 02:15:20 +04:00
|
|
|
cp->Detached = cp->OptionDetach;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
2004-07-08 01:27:50 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcess_Disown(kwsysProcess* cp)
|
|
|
|
{
|
2004-07-08 02:15:20 +04:00
|
|
|
int i;
|
|
|
|
|
|
|
|
/* Make sure we are executing a detached process. */
|
|
|
|
if(!cp || !cp->Detached || cp->State != kwsysProcess_State_Executing ||
|
|
|
|
cp->TimeoutExpired || cp->Killed || cp->Terminated)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Disable the reading threads. */
|
|
|
|
kwsysProcessDisablePipeThreads(cp);
|
|
|
|
|
|
|
|
/* Wait for all pipe threads to reset. */
|
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
WaitForSingleObject(cp->Pipe[i].Reader.Reset, INFINITE);
|
|
|
|
WaitForSingleObject(cp->Pipe[i].Waker.Reset, INFINITE);
|
2004-07-08 02:15:20 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/* We will not wait for exit, so cleanup now. */
|
|
|
|
kwsysProcessCleanup(cp, 0);
|
|
|
|
|
|
|
|
/* The process has been disowned. */
|
|
|
|
cp->State = kwsysProcess_State_Disowned;
|
2004-07-08 01:27:50 +04:00
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
|
2003-12-05 19:53:17 +03:00
|
|
|
int kwsysProcess_WaitForData(kwsysProcess* cp, char** data, int* length,
|
2003-06-10 23:46:31 +04:00
|
|
|
double* userTimeout)
|
|
|
|
{
|
|
|
|
kwsysProcessTime userStartTime;
|
|
|
|
kwsysProcessTime timeoutLength;
|
|
|
|
kwsysProcessTime timeoutTime;
|
|
|
|
DWORD timeout;
|
|
|
|
int user;
|
|
|
|
int done = 0;
|
|
|
|
int expired = 0;
|
2003-12-07 22:09:58 +03:00
|
|
|
int pipeId = kwsysProcess_Pipe_None;
|
2003-06-10 23:46:31 +04:00
|
|
|
DWORD w;
|
2003-06-30 18:44:35 +04:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Make sure we are executing a process. */
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp || cp->State != kwsysProcess_State_Executing || cp->Killed ||
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->TimeoutExpired)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-07 22:09:58 +03:00
|
|
|
return kwsysProcess_Pipe_None;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Record the time at which user timeout period starts. */
|
2003-06-30 18:44:35 +04:00
|
|
|
userStartTime = kwsysProcessTimeGetCurrent();
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Calculate the time at which a timeout will expire, and whether it
|
|
|
|
is the user or process timeout. */
|
|
|
|
user = kwsysProcessGetTimeoutTime(cp, userTimeout, &timeoutTime);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Loop until we have a reason to return. */
|
|
|
|
while(!done && cp->PipesLeft > 0)
|
|
|
|
{
|
|
|
|
/* If we previously got data from a thread, let it know we are
|
|
|
|
done with the data. */
|
2003-12-03 17:20:05 +03:00
|
|
|
if(cp->CurrentIndex < KWSYSPE_PIPE_COUNT)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(cp->Pipe[cp->CurrentIndex].Reader.Go, 1, 0);
|
2003-12-03 17:20:05 +03:00
|
|
|
cp->CurrentIndex = KWSYSPE_PIPE_COUNT;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Setup a timeout if required. */
|
2005-06-17 22:05:35 +04:00
|
|
|
if(kwsysProcessGetTimeoutLeft(&timeoutTime, user?userTimeout:0,
|
|
|
|
&timeoutLength))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
/* Timeout has already expired. */
|
|
|
|
expired = 1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(timeoutTime.QuadPart < 0)
|
|
|
|
{
|
|
|
|
timeout = INFINITE;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
timeout = kwsysProcessTimeToDWORD(timeoutLength);
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Wait for a pipe's thread to signal or a process to terminate. */
|
|
|
|
w = WaitForMultipleObjects(cp->ProcessEventsLength, cp->ProcessEvents,
|
|
|
|
0, timeout);
|
2003-06-10 23:46:31 +04:00
|
|
|
if(w == WAIT_TIMEOUT)
|
|
|
|
{
|
|
|
|
/* Timeout has expired. */
|
|
|
|
expired = 1;
|
|
|
|
done = 1;
|
|
|
|
}
|
|
|
|
else if(w == WAIT_OBJECT_0)
|
|
|
|
{
|
|
|
|
/* Save the index of the reporting thread and release the mutex.
|
|
|
|
The thread will block until we signal its Empty mutex. */
|
|
|
|
cp->CurrentIndex = cp->SharedIndex;
|
|
|
|
ReleaseSemaphore(cp->SharedIndexMutex, 1, 0);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Data are available or a pipe closed. */
|
|
|
|
if(cp->Pipe[cp->CurrentIndex].Closed)
|
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
/* The pipe closed at the write end. Close the read end and
|
|
|
|
inform the wakeup thread it is done with this process. */
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[cp->CurrentIndex].Read);
|
|
|
|
ReleaseSemaphore(cp->Pipe[cp->CurrentIndex].Waker.Go, 1, 0);
|
2003-06-10 23:46:31 +04:00
|
|
|
--cp->PipesLeft;
|
|
|
|
}
|
2003-12-05 19:53:17 +03:00
|
|
|
else if(data && length)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-05 19:53:17 +03:00
|
|
|
/* Report this data. */
|
2003-06-10 23:46:31 +04:00
|
|
|
*data = cp->Pipe[cp->CurrentIndex].DataBuffer;
|
|
|
|
*length = cp->Pipe[cp->CurrentIndex].DataLength;
|
2003-12-07 22:09:58 +03:00
|
|
|
switch(cp->CurrentIndex)
|
|
|
|
{
|
|
|
|
case KWSYSPE_PIPE_STDOUT:
|
|
|
|
pipeId = kwsysProcess_Pipe_STDOUT; break;
|
|
|
|
case KWSYSPE_PIPE_STDERR:
|
|
|
|
pipeId = kwsysProcess_Pipe_STDERR; break;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
done = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* A process has terminated. */
|
|
|
|
kwsysProcessDestroy(cp, w-WAIT_OBJECT_0);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Update the user timeout. */
|
|
|
|
if(userTimeout)
|
|
|
|
{
|
|
|
|
kwsysProcessTime userEndTime = kwsysProcessTimeGetCurrent();
|
|
|
|
kwsysProcessTime difference = kwsysProcessTimeSubtract(userEndTime,
|
|
|
|
userStartTime);
|
|
|
|
double d = kwsysProcessTimeToDouble(difference);
|
|
|
|
*userTimeout -= d;
|
|
|
|
if(*userTimeout < 0)
|
|
|
|
{
|
|
|
|
*userTimeout = 0;
|
|
|
|
}
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Check what happened. */
|
|
|
|
if(pipeId)
|
|
|
|
{
|
|
|
|
/* Data are ready on a pipe. */
|
|
|
|
return pipeId;
|
|
|
|
}
|
|
|
|
else if(expired)
|
|
|
|
{
|
|
|
|
/* A timeout has expired. */
|
|
|
|
if(user)
|
|
|
|
{
|
|
|
|
/* The user timeout has expired. It has no time left. */
|
2003-06-19 01:06:21 +04:00
|
|
|
return kwsysProcess_Pipe_Timeout;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* The process timeout has expired. Kill the child now. */
|
|
|
|
kwsysProcess_Kill(cp);
|
|
|
|
cp->TimeoutExpired = 1;
|
|
|
|
cp->Killed = 0;
|
2003-12-07 22:09:58 +03:00
|
|
|
return kwsysProcess_Pipe_None;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* The children have terminated and no more data are available. */
|
2003-12-07 22:09:58 +03:00
|
|
|
return kwsysProcess_Pipe_None;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcess_WaitForExit(kwsysProcess* cp, double* userTimeout)
|
|
|
|
{
|
|
|
|
int i;
|
2003-11-04 17:00:18 +03:00
|
|
|
int pipe;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Make sure we are executing a process. */
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp || cp->State != kwsysProcess_State_Executing)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
return 1;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Wait for the process to terminate. Ignore all data. */
|
2003-12-05 19:53:17 +03:00
|
|
|
while((pipe = kwsysProcess_WaitForData(cp, 0, 0, userTimeout)) > 0)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-06-19 01:06:21 +04:00
|
|
|
if(pipe == kwsysProcess_Pipe_Timeout)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
/* The user timeout has expired. */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* When the last pipe closes in WaitForData, the loop terminates
|
2005-07-07 17:05:59 +04:00
|
|
|
without releasing the pipe's thread. Release it now. */
|
2003-12-03 17:20:05 +03:00
|
|
|
if(cp->CurrentIndex < KWSYSPE_PIPE_COUNT)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(cp->Pipe[cp->CurrentIndex].Reader.Go, 1, 0);
|
2003-12-03 17:20:05 +03:00
|
|
|
cp->CurrentIndex = KWSYSPE_PIPE_COUNT;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Wait for all pipe threads to reset. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
WaitForSingleObject(cp->Pipe[i].Reader.Reset, INFINITE);
|
|
|
|
WaitForSingleObject(cp->Pipe[i].Waker.Reset, INFINITE);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* ---- It is now safe again to call kwsysProcessCleanup. ----- */
|
|
|
|
/* Close all the pipes. */
|
|
|
|
kwsysProcessCleanup(cp, 0);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Determine the outcome. */
|
|
|
|
if(cp->Killed)
|
|
|
|
{
|
|
|
|
/* We killed the child. */
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->State = kwsysProcess_State_Killed;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
else if(cp->TimeoutExpired)
|
|
|
|
{
|
|
|
|
/* The timeout expired. */
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->State = kwsysProcess_State_Expired;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
else
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* The children exited. Report the outcome of the last process. */
|
|
|
|
cp->ExitCode = cp->CommandExitCodes[cp->NumberOfCommands-1];
|
2004-01-28 18:59:46 +03:00
|
|
|
if((cp->ExitCode & 0xF0000000) == 0xC0000000)
|
2003-06-19 01:06:21 +04:00
|
|
|
{
|
|
|
|
/* Child terminated due to exceptional behavior. */
|
2003-06-19 01:27:10 +04:00
|
|
|
cp->State = kwsysProcess_State_Exception;
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->ExitValue = 1;
|
2003-12-31 00:23:16 +03:00
|
|
|
kwsysProcessSetExitException(cp, cp->ExitCode);
|
2003-06-19 01:06:21 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2004-01-28 18:59:46 +03:00
|
|
|
/* Child exited without exception. */
|
2003-06-19 01:27:10 +04:00
|
|
|
cp->State = kwsysProcess_State_Exited;
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->ExitException = kwsysProcess_Exception_None;
|
2004-01-28 18:59:46 +03:00
|
|
|
cp->ExitValue = cp->ExitCode;
|
2003-06-19 01:06:21 +04:00
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcess_Kill(kwsysProcess* cp)
|
|
|
|
{
|
|
|
|
int i;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Make sure we are executing a process. */
|
2003-12-13 18:36:22 +03:00
|
|
|
if(!cp || cp->State != kwsysProcess_State_Executing || cp->TimeoutExpired ||
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->Killed || cp->Terminated)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2004-07-08 02:15:20 +04:00
|
|
|
/* Disable the reading threads. */
|
|
|
|
kwsysProcessDisablePipeThreads(cp);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Kill the children. */
|
2003-06-10 23:46:31 +04:00
|
|
|
cp->Killed = 1;
|
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
|
|
|
/* Windows 9x. Tell the forwarding executable to kill the child. */
|
|
|
|
SetEvent(cp->Win9xKillEvent);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Not Windows 9x. Just terminate the children. */
|
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
2004-05-10 22:54:22 +04:00
|
|
|
kwsysProcessKillTree(cp->ProcessInformation[i].dwProcessId);
|
2003-12-03 17:20:05 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-05-13 18:08:09 +04:00
|
|
|
/* We are killing the children and ignoring all data. Do not wait
|
|
|
|
for them to exit. */
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
Function executed for each pipe's thread. Argument is a pointer to
|
|
|
|
the kwsysProcessPipeData instance for this thread.
|
|
|
|
*/
|
2004-07-14 00:50:55 +04:00
|
|
|
DWORD WINAPI kwsysProcessPipeThreadRead(LPVOID ptd)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
kwsysProcessPipeData* td = (kwsysProcessPipeData*)ptd;
|
|
|
|
kwsysProcess* cp = td->Process;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Wait for a process to be ready. */
|
2004-07-14 00:50:55 +04:00
|
|
|
while((WaitForSingleObject(td->Reader.Ready, INFINITE), !cp->Deleting))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
/* Read output from the process for this thread's pipe. */
|
|
|
|
kwsysProcessPipeThreadReadPipe(cp, td);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Signal the main thread we have reset for a new process. */
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(td->Reader.Reset, 1, 0);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
Function called in each pipe's thread to handle data for one
|
|
|
|
execution of a subprocess.
|
|
|
|
*/
|
|
|
|
void kwsysProcessPipeThreadReadPipe(kwsysProcess* cp, kwsysProcessPipeData* td)
|
|
|
|
{
|
|
|
|
/* Wait for space in the thread's buffer. */
|
2004-07-14 00:50:55 +04:00
|
|
|
while((WaitForSingleObject(td->Reader.Go, INFINITE), !td->Closed))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
/* Read data from the pipe. This may block until data are available. */
|
2003-12-03 17:20:05 +03:00
|
|
|
if(!ReadFile(td->Read, td->DataBuffer, KWSYSPE_PIPE_BUFFER_SIZE,
|
2003-06-10 23:46:31 +04:00
|
|
|
&td->DataLength, 0))
|
|
|
|
{
|
|
|
|
if(GetLastError() != ERROR_BROKEN_PIPE)
|
|
|
|
{
|
|
|
|
/* UNEXPECTED failure to read the pipe. */
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* The pipe closed. There are no more data to read. */
|
|
|
|
td->Closed = 1;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Wait for our turn to be handled by the main thread. */
|
|
|
|
WaitForSingleObject(cp->SharedIndexMutex, INFINITE);
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Tell the main thread we have something to report. */
|
|
|
|
cp->SharedIndex = td->Index;
|
|
|
|
ReleaseSemaphore(cp->Full, 1, 0);
|
|
|
|
}
|
2004-07-14 00:50:55 +04:00
|
|
|
|
|
|
|
/* We were signalled to exit with our buffer empty. Reset the
|
|
|
|
mutex for a new process. */
|
|
|
|
ReleaseSemaphore(td->Reader.Go, 1, 0);
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
2004-07-14 00:23:49 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
Function executed for each pipe's thread. Argument is a pointer to
|
|
|
|
the kwsysProcessPipeData instance for this thread.
|
|
|
|
*/
|
|
|
|
DWORD WINAPI kwsysProcessPipeThreadWake(LPVOID ptd)
|
|
|
|
{
|
|
|
|
kwsysProcessPipeData* td = (kwsysProcessPipeData*)ptd;
|
|
|
|
kwsysProcess* cp = td->Process;
|
|
|
|
|
|
|
|
/* Wait for a process to be ready. */
|
2004-07-14 00:50:55 +04:00
|
|
|
while((WaitForSingleObject(td->Waker.Ready, INFINITE), !cp->Deleting))
|
2004-07-14 00:23:49 +04:00
|
|
|
{
|
|
|
|
/* Wait for a possible wakeup. */
|
|
|
|
kwsysProcessPipeThreadWakePipe(cp, td);
|
|
|
|
|
|
|
|
/* Signal the main thread we have reset for a new process. */
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(td->Waker.Reset, 1, 0);
|
2004-07-14 00:23:49 +04:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
Function called in each pipe's thread to handle reading thread
|
|
|
|
wakeup for one execution of a subprocess.
|
|
|
|
*/
|
2004-07-14 00:23:49 +04:00
|
|
|
void kwsysProcessPipeThreadWakePipe(kwsysProcess* cp, kwsysProcessPipeData* td)
|
|
|
|
{
|
2005-04-22 17:21:19 +04:00
|
|
|
(void)cp;
|
|
|
|
|
2004-07-14 00:23:49 +04:00
|
|
|
/* Wait for a possible wake command. */
|
2004-07-14 00:50:55 +04:00
|
|
|
WaitForSingleObject(td->Waker.Go, INFINITE);
|
|
|
|
|
|
|
|
/* If the pipe is not closed, we need to wake up the reading thread. */
|
|
|
|
if(!td->Closed)
|
2004-07-14 00:23:49 +04:00
|
|
|
{
|
|
|
|
DWORD dummy;
|
|
|
|
WriteFile(td->Write, "", 1, &dummy, 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* Initialize a process control structure for kwsysProcess_Execute. */
|
|
|
|
int kwsysProcessInitialize(kwsysProcess* cp)
|
|
|
|
{
|
|
|
|
/* Reset internal status flags. */
|
|
|
|
cp->TimeoutExpired = 0;
|
|
|
|
cp->Terminated = 0;
|
|
|
|
cp->Killed = 0;
|
|
|
|
cp->ExitException = kwsysProcess_Exception_None;
|
|
|
|
cp->ExitCode = 1;
|
|
|
|
cp->ExitValue = 1;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Reset error data. */
|
|
|
|
cp->ErrorMessage[0] = 0;
|
2003-12-31 00:23:16 +03:00
|
|
|
strcpy(cp->ExitExceptionString, "No exception");
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Allocate process information for each process. */
|
|
|
|
cp->ProcessInformation =
|
|
|
|
(PROCESS_INFORMATION*)malloc(sizeof(PROCESS_INFORMATION) *
|
|
|
|
cp->NumberOfCommands);
|
|
|
|
if(!cp->ProcessInformation)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
ZeroMemory(cp->ProcessInformation,
|
|
|
|
sizeof(PROCESS_INFORMATION) * cp->NumberOfCommands);
|
|
|
|
if(cp->CommandExitCodes)
|
|
|
|
{
|
|
|
|
free(cp->CommandExitCodes);
|
|
|
|
}
|
|
|
|
cp->CommandExitCodes = (DWORD*)malloc(sizeof(DWORD)*cp->NumberOfCommands);
|
|
|
|
if(!cp->CommandExitCodes)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
ZeroMemory(cp->CommandExitCodes, sizeof(DWORD)*cp->NumberOfCommands);
|
|
|
|
|
|
|
|
/* Allocate event wait array. The first event is cp->Full, the rest
|
|
|
|
are the process termination events. */
|
|
|
|
cp->ProcessEvents = (PHANDLE)malloc(sizeof(HANDLE)*(cp->NumberOfCommands+1));
|
|
|
|
if(!cp->ProcessEvents)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
ZeroMemory(cp->ProcessEvents, sizeof(HANDLE) * (cp->NumberOfCommands+1));
|
|
|
|
cp->ProcessEvents[0] = cp->Full;
|
|
|
|
cp->ProcessEventsLength = cp->NumberOfCommands+1;
|
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
/* Allocate space to save the real working directory of this process. */
|
|
|
|
if(cp->WorkingDirectory)
|
|
|
|
{
|
|
|
|
cp->RealWorkingDirectoryLength = GetCurrentDirectory(0, 0);
|
|
|
|
if(cp->RealWorkingDirectoryLength > 0)
|
|
|
|
{
|
|
|
|
cp->RealWorkingDirectory = malloc(cp->RealWorkingDirectoryLength);
|
|
|
|
if(!cp->RealWorkingDirectory)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcessCreate(kwsysProcess* cp, int index,
|
|
|
|
kwsysProcessCreateInformation* si,
|
|
|
|
PHANDLE readEnd)
|
|
|
|
{
|
|
|
|
/* Setup the process's stdin. */
|
|
|
|
if(*readEnd)
|
|
|
|
{
|
|
|
|
/* Create an inherited duplicate of the read end from the output
|
|
|
|
pipe of the previous process. This also closes the
|
|
|
|
non-inherited version. */
|
|
|
|
if(!DuplicateHandle(GetCurrentProcess(), *readEnd,
|
|
|
|
GetCurrentProcess(), readEnd,
|
|
|
|
0, TRUE, (DUPLICATE_CLOSE_SOURCE |
|
|
|
|
DUPLICATE_SAME_ACCESS)))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
si->StartupInfo.hStdInput = *readEnd;
|
|
|
|
|
|
|
|
/* This function is done with this handle. */
|
|
|
|
*readEnd = 0;
|
|
|
|
}
|
2003-12-13 22:13:54 +03:00
|
|
|
else if(cp->PipeFileSTDIN)
|
|
|
|
{
|
|
|
|
/* Create a handle to read a file for stdin. */
|
|
|
|
HANDLE fin = CreateFile(cp->PipeFileSTDIN, GENERIC_READ,
|
|
|
|
FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);
|
|
|
|
if(fin == INVALID_HANDLE_VALUE)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
/* Create an inherited duplicate of the handle. This also closes
|
|
|
|
the non-inherited version. */
|
|
|
|
if(!DuplicateHandle(GetCurrentProcess(), fin,
|
|
|
|
GetCurrentProcess(), &fin,
|
|
|
|
0, TRUE, (DUPLICATE_CLOSE_SOURCE |
|
|
|
|
DUPLICATE_SAME_ACCESS)))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
si->StartupInfo.hStdInput = fin;
|
|
|
|
}
|
2003-12-17 01:20:01 +03:00
|
|
|
else if(cp->PipeSharedSTDIN)
|
2003-12-03 17:20:05 +03:00
|
|
|
{
|
|
|
|
si->StartupInfo.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
|
|
|
|
}
|
2003-12-17 01:20:01 +03:00
|
|
|
else
|
|
|
|
{
|
|
|
|
si->StartupInfo.hStdInput = INVALID_HANDLE_VALUE;
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
/* Setup the process's stdout. */
|
|
|
|
{
|
|
|
|
DWORD maybeClose = DUPLICATE_CLOSE_SOURCE;
|
|
|
|
HANDLE writeEnd;
|
|
|
|
|
|
|
|
/* Create the output pipe for this process. Neither end is directly
|
|
|
|
inherited. */
|
|
|
|
if(!CreatePipe(readEnd, &writeEnd, 0, 0))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Create an inherited duplicate of the write end. Close the
|
|
|
|
non-inherited version unless this is the last process. Save the
|
|
|
|
non-inherited write end of the last process. */
|
|
|
|
if(index == cp->NumberOfCommands-1)
|
|
|
|
{
|
|
|
|
cp->Pipe[KWSYSPE_PIPE_STDOUT].Write = writeEnd;
|
|
|
|
maybeClose = 0;
|
|
|
|
}
|
|
|
|
if(!DuplicateHandle(GetCurrentProcess(), writeEnd,
|
2003-12-13 22:13:54 +03:00
|
|
|
GetCurrentProcess(), &writeEnd,
|
2003-12-03 17:20:05 +03:00
|
|
|
0, TRUE, (maybeClose | DUPLICATE_SAME_ACCESS)))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-13 22:13:54 +03:00
|
|
|
si->StartupInfo.hStdOutput = writeEnd;
|
2003-12-03 17:20:05 +03:00
|
|
|
}
|
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
/* Replace the stdout pipe with a file if requested. In this case
|
|
|
|
the pipe thread will still run but never report data. */
|
|
|
|
if(index == cp->NumberOfCommands-1 && cp->PipeFileSTDOUT)
|
|
|
|
{
|
|
|
|
if(!kwsysProcessSetupOutputPipeFile(&si->StartupInfo.hStdOutput,
|
|
|
|
cp->PipeFileSTDOUT))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-12-17 01:20:01 +03:00
|
|
|
/* Replace the stdout pipe with the parent process's if requested.
|
|
|
|
In this case the pipe thread will still run but never report
|
|
|
|
data. */
|
|
|
|
if(index == cp->NumberOfCommands-1 && cp->PipeSharedSTDOUT)
|
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&si->StartupInfo.hStdOutput);
|
|
|
|
si->StartupInfo.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
|
|
|
|
}
|
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Create the child process. */
|
|
|
|
{
|
|
|
|
BOOL r;
|
|
|
|
char* realCommand;
|
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
|
|
|
/* Create an error reporting pipe for the forwarding executable.
|
|
|
|
Neither end is directly inherited. */
|
|
|
|
if(!CreatePipe(&si->ErrorPipeRead, &si->ErrorPipeWrite, 0, 0))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Create an inherited duplicate of the write end. This also closes
|
|
|
|
the non-inherited version. */
|
|
|
|
if(!DuplicateHandle(GetCurrentProcess(), si->ErrorPipeWrite,
|
|
|
|
GetCurrentProcess(), &si->ErrorPipeWrite,
|
|
|
|
0, TRUE, (DUPLICATE_CLOSE_SOURCE |
|
|
|
|
DUPLICATE_SAME_ACCESS)))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* The forwarding executable is given a handle to the error pipe
|
|
|
|
and resume and kill events. */
|
|
|
|
realCommand = malloc(strlen(cp->Win9x)+strlen(cp->Commands[index])+100);
|
|
|
|
if(!realCommand)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
sprintf(realCommand, "%s %p %p %p %d %s", cp->Win9x,
|
|
|
|
si->ErrorPipeWrite, cp->Win9xResumeEvent, cp->Win9xKillEvent,
|
|
|
|
cp->HideWindow, cp->Commands[index]);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
realCommand = cp->Commands[index];
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Create the child in a suspended state so we can wait until all
|
|
|
|
children have been created before running any one. */
|
|
|
|
r = CreateProcess(0, realCommand, 0, 0, TRUE,
|
2003-12-13 22:13:54 +03:00
|
|
|
cp->Win9x? 0 : CREATE_SUSPENDED, 0, 0,
|
|
|
|
&si->StartupInfo, &cp->ProcessInformation[index]);
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
|
|
|
/* Free memory. */
|
|
|
|
free(realCommand);
|
|
|
|
|
|
|
|
/* Close the error pipe write end so we can detect when the
|
|
|
|
forwarding executable closes it. */
|
|
|
|
kwsysProcessCleanupHandle(&si->ErrorPipeWrite);
|
|
|
|
if(r)
|
|
|
|
{
|
|
|
|
/* Wait for the forwarding executable to report an error or
|
|
|
|
close the error pipe to report success. */
|
|
|
|
DWORD total = 0;
|
|
|
|
DWORD n = 1;
|
|
|
|
while(total < KWSYSPE_PIPE_BUFFER_SIZE && n > 0)
|
|
|
|
{
|
|
|
|
if(ReadFile(si->ErrorPipeRead, cp->ErrorMessage+total,
|
|
|
|
KWSYSPE_PIPE_BUFFER_SIZE-total, &n, 0))
|
|
|
|
{
|
|
|
|
total += n;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
n = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(total > 0 || GetLastError() != ERROR_BROKEN_PIPE)
|
|
|
|
{
|
|
|
|
/* The forwarding executable could not run the process, or
|
|
|
|
there was an error reading from its error pipe. Preserve
|
|
|
|
the last error while cleaning up the forwarding executable
|
|
|
|
so the cleanup our caller does reports the proper error. */
|
|
|
|
DWORD error = GetLastError();
|
|
|
|
kwsysProcessCleanupHandle(&cp->ProcessInformation[index].hThread);
|
|
|
|
kwsysProcessCleanupHandle(&cp->ProcessInformation[index].hProcess);
|
|
|
|
SetLastError(error);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
kwsysProcessCleanupHandle(&si->ErrorPipeRead);
|
|
|
|
}
|
|
|
|
|
|
|
|
if(!r)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Successfully created this child process. */
|
|
|
|
if(index > 0)
|
|
|
|
{
|
|
|
|
/* Close our handle to the input pipe for the current process. */
|
|
|
|
kwsysProcessCleanupHandle(&si->StartupInfo.hStdInput);
|
|
|
|
}
|
|
|
|
|
2003-12-17 01:20:01 +03:00
|
|
|
if(si->StartupInfo.hStdOutput != GetStdHandle(STD_OUTPUT_HANDLE))
|
|
|
|
{
|
|
|
|
/* The parent process does not need the inhertied pipe write end. */
|
|
|
|
kwsysProcessCleanupHandle(&si->StartupInfo.hStdOutput);
|
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcessDestroy(kwsysProcess* cp, int event)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
int index;
|
|
|
|
|
|
|
|
/* Find the process index for the termination event. */
|
|
|
|
for(index=0; index < cp->NumberOfCommands; ++index)
|
|
|
|
{
|
|
|
|
if(cp->ProcessInformation[index].hProcess == cp->ProcessEvents[event])
|
|
|
|
{
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Check the exit code of the process. */
|
|
|
|
GetExitCodeProcess(cp->ProcessInformation[index].hProcess,
|
|
|
|
&cp->CommandExitCodes[index]);
|
|
|
|
|
|
|
|
/* Close the process handle for the terminated process. */
|
|
|
|
kwsysProcessCleanupHandle(&cp->ProcessInformation[index].hProcess);
|
|
|
|
|
|
|
|
/* Remove the process from the available events. */
|
|
|
|
cp->ProcessEventsLength -= 1;
|
|
|
|
for(i=event; i < cp->ProcessEventsLength; ++i)
|
|
|
|
{
|
|
|
|
cp->ProcessEvents[i] = cp->ProcessEvents[i+1];
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Check if all processes have terminated. */
|
|
|
|
if(cp->ProcessEventsLength == 1)
|
|
|
|
{
|
|
|
|
cp->Terminated = 1;
|
|
|
|
|
|
|
|
/* Close our copies of the pipe write handles so the pipe threads
|
|
|
|
can detect end-of-data. */
|
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Write);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-12-13 22:13:54 +03:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcessSetupOutputPipeFile(PHANDLE phandle, const char* name)
|
|
|
|
{
|
|
|
|
HANDLE fout;
|
|
|
|
if(!name)
|
|
|
|
{
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Close the existing inherited handle. */
|
|
|
|
kwsysProcessCleanupHandle(phandle);
|
|
|
|
|
|
|
|
/* Create a handle to write a file for the pipe. */
|
|
|
|
fout = CreateFile(name, GENERIC_WRITE, FILE_SHARE_READ, 0,
|
|
|
|
CREATE_ALWAYS, 0, 0);
|
|
|
|
if(fout == INVALID_HANDLE_VALUE)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Create an inherited duplicate of the handle. This also closes
|
|
|
|
the non-inherited version. */
|
|
|
|
if(!DuplicateHandle(GetCurrentProcess(), fout,
|
|
|
|
GetCurrentProcess(), &fout,
|
|
|
|
0, TRUE, (DUPLICATE_CLOSE_SOURCE |
|
|
|
|
DUPLICATE_SAME_ACCESS)))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Assign the replacement handle. */
|
|
|
|
*phandle = fout;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
|
|
|
|
/* Close the given handle if it is open. Reset its value to 0. */
|
|
|
|
void kwsysProcessCleanupHandle(PHANDLE h)
|
|
|
|
{
|
|
|
|
if(h && *h)
|
|
|
|
{
|
|
|
|
CloseHandle(*h);
|
|
|
|
*h = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
|
|
|
|
/* Close all handles created by kwsysProcess_Execute. */
|
|
|
|
void kwsysProcessCleanup(kwsysProcess* cp, int error)
|
|
|
|
{
|
|
|
|
int i;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* If this is an error case, report the error. */
|
|
|
|
if(error)
|
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Construct an error message if one has not been provided already. */
|
|
|
|
if(cp->ErrorMessage[0] == 0)
|
2003-07-07 17:38:14 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Format the error message. */
|
|
|
|
DWORD original = GetLastError();
|
|
|
|
DWORD length = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM |
|
|
|
|
FORMAT_MESSAGE_IGNORE_INSERTS, 0, original,
|
|
|
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
|
|
|
cp->ErrorMessage, KWSYSPE_PIPE_BUFFER_SIZE, 0);
|
|
|
|
if(length < 1)
|
|
|
|
{
|
|
|
|
/* FormatMessage failed. Use a default message. */
|
|
|
|
_snprintf(cp->ErrorMessage, KWSYSPE_PIPE_BUFFER_SIZE,
|
|
|
|
"Process execution failed with error 0x%X. "
|
|
|
|
"FormatMessage failed with error 0x%X",
|
|
|
|
original, GetLastError());
|
|
|
|
}
|
2003-07-07 17:38:14 +04:00
|
|
|
}
|
2003-08-14 02:08:07 +04:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Remove trailing period and newline, if any. */
|
2003-12-05 19:37:28 +03:00
|
|
|
kwsysProcessCleanErrorMessage(cp);
|
2003-12-03 17:20:05 +03:00
|
|
|
|
2003-07-07 17:38:14 +04:00
|
|
|
/* Set the error state. */
|
2003-06-19 01:06:21 +04:00
|
|
|
cp->State = kwsysProcess_State_Error;
|
2003-08-14 02:08:07 +04:00
|
|
|
|
2003-12-03 17:20:05 +03:00
|
|
|
/* Cleanup any processes already started in a suspended state. */
|
|
|
|
if(cp->ProcessInformation)
|
|
|
|
{
|
|
|
|
if(cp->Win9x)
|
|
|
|
{
|
|
|
|
SetEvent(cp->Win9xKillEvent);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
|
|
|
if(cp->ProcessInformation[i].hProcess)
|
|
|
|
{
|
|
|
|
TerminateProcess(cp->ProcessInformation[i].hProcess, 255);
|
|
|
|
WaitForSingleObject(cp->ProcessInformation[i].hProcess, INFINITE);
|
|
|
|
}
|
|
|
|
}
|
2003-12-05 19:37:28 +03:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < cp->NumberOfCommands; ++i)
|
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&cp->ProcessInformation[i].hThread);
|
|
|
|
kwsysProcessCleanupHandle(&cp->ProcessInformation[i].hProcess);
|
|
|
|
}
|
|
|
|
}
|
2003-12-13 22:13:54 +03:00
|
|
|
|
|
|
|
/* Restore the working directory. */
|
|
|
|
if(cp->RealWorkingDirectory)
|
|
|
|
{
|
|
|
|
SetCurrentDirectory(cp->RealWorkingDirectory);
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-03 17:20:05 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Free memory. */
|
2003-12-03 17:20:05 +03:00
|
|
|
if(cp->ProcessInformation)
|
|
|
|
{
|
|
|
|
free(cp->ProcessInformation);
|
|
|
|
cp->ProcessInformation = 0;
|
|
|
|
}
|
|
|
|
if(cp->ProcessEvents)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
2003-12-03 17:20:05 +03:00
|
|
|
free(cp->ProcessEvents);
|
|
|
|
cp->ProcessEvents = 0;
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
2003-12-13 22:13:54 +03:00
|
|
|
if(cp->RealWorkingDirectory)
|
|
|
|
{
|
|
|
|
free(cp->RealWorkingDirectory);
|
|
|
|
cp->RealWorkingDirectory = 0;
|
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
|
|
|
|
/* Close each pipe. */
|
2003-12-03 17:20:05 +03:00
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Write);
|
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[i].Read);
|
2005-07-07 17:05:59 +04:00
|
|
|
cp->Pipe[i].Closed = 0;
|
2003-12-05 19:37:28 +03:00
|
|
|
}
|
2003-06-10 23:46:31 +04:00
|
|
|
}
|
|
|
|
|
2003-08-14 02:08:07 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
void kwsysProcessCleanErrorMessage(kwsysProcess* cp)
|
|
|
|
{
|
|
|
|
/* Remove trailing period and newline, if any. */
|
|
|
|
int length = strlen(cp->ErrorMessage);
|
|
|
|
if(cp->ErrorMessage[length-1] == '\n')
|
|
|
|
{
|
|
|
|
cp->ErrorMessage[length-1] = 0;
|
|
|
|
--length;
|
|
|
|
if(length > 0 && cp->ErrorMessage[length-1] == '\r')
|
|
|
|
{
|
|
|
|
cp->ErrorMessage[length-1] = 0;
|
|
|
|
--length;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(length > 0 && cp->ErrorMessage[length-1] == '.')
|
|
|
|
{
|
|
|
|
cp->ErrorMessage[length-1] = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* Get the time at which either the process or user timeout will
|
|
|
|
expire. Returns 1 if the user timeout is first, and 0 otherwise. */
|
|
|
|
int kwsysProcessGetTimeoutTime(kwsysProcess* cp, double* userTimeout,
|
|
|
|
kwsysProcessTime* timeoutTime)
|
|
|
|
{
|
|
|
|
/* The first time this is called, we need to calculate the time at
|
|
|
|
which the child will timeout. */
|
|
|
|
if(cp->Timeout && cp->TimeoutTime.QuadPart < 0)
|
|
|
|
{
|
|
|
|
kwsysProcessTime length = kwsysProcessTimeFromDouble(cp->Timeout);
|
|
|
|
cp->TimeoutTime = kwsysProcessTimeAdd(cp->StartTime, length);
|
2003-12-05 19:37:28 +03:00
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Start with process timeout. */
|
|
|
|
*timeoutTime = cp->TimeoutTime;
|
2003-12-05 19:37:28 +03:00
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
/* Check if the user timeout is earlier. */
|
|
|
|
if(userTimeout)
|
|
|
|
{
|
|
|
|
kwsysProcessTime currentTime = kwsysProcessTimeGetCurrent();
|
|
|
|
kwsysProcessTime userTimeoutLength = kwsysProcessTimeFromDouble(*userTimeout);
|
|
|
|
kwsysProcessTime userTimeoutTime = kwsysProcessTimeAdd(currentTime,
|
|
|
|
userTimeoutLength);
|
2005-06-17 22:05:35 +04:00
|
|
|
if(timeoutTime->QuadPart < 0 ||
|
|
|
|
kwsysProcessTimeLess(userTimeoutTime, *timeoutTime))
|
2003-06-10 23:46:31 +04:00
|
|
|
{
|
|
|
|
*timeoutTime = userTimeoutTime;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* Get the length of time before the given timeout time arrives.
|
|
|
|
Returns 1 if the time has already arrived, and 0 otherwise. */
|
2003-06-30 18:44:35 +04:00
|
|
|
int kwsysProcessGetTimeoutLeft(kwsysProcessTime* timeoutTime,
|
2005-06-17 22:05:35 +04:00
|
|
|
double* userTimeout,
|
2003-06-10 23:46:31 +04:00
|
|
|
kwsysProcessTime* timeoutLength)
|
|
|
|
{
|
|
|
|
if(timeoutTime->QuadPart < 0)
|
|
|
|
{
|
|
|
|
/* No timeout time has been requested. */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* Calculate the remaining time. */
|
|
|
|
kwsysProcessTime currentTime = kwsysProcessTimeGetCurrent();
|
|
|
|
*timeoutLength = kwsysProcessTimeSubtract(*timeoutTime, currentTime);
|
2005-06-17 22:05:35 +04:00
|
|
|
|
|
|
|
if(timeoutLength->QuadPart < 0 && userTimeout && *userTimeout <= 0)
|
|
|
|
{
|
|
|
|
/* Caller has explicitly requested a zero timeout. */
|
|
|
|
timeoutLength->QuadPart = 0;
|
|
|
|
}
|
|
|
|
|
2003-06-10 23:46:31 +04:00
|
|
|
if(timeoutLength->QuadPart < 0)
|
|
|
|
{
|
|
|
|
/* Timeout has already expired. */
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* There is some time left. */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
kwsysProcessTime kwsysProcessTimeGetCurrent()
|
|
|
|
{
|
|
|
|
kwsysProcessTime current;
|
|
|
|
FILETIME ft;
|
|
|
|
GetSystemTimeAsFileTime(&ft);
|
|
|
|
current.LowPart = ft.dwLowDateTime;
|
|
|
|
current.HighPart = ft.dwHighDateTime;
|
|
|
|
return current;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
DWORD kwsysProcessTimeToDWORD(kwsysProcessTime t)
|
|
|
|
{
|
|
|
|
return (DWORD)(t.QuadPart * 0.0001);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
double kwsysProcessTimeToDouble(kwsysProcessTime t)
|
|
|
|
{
|
|
|
|
return t.QuadPart * 0.0000001;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
kwsysProcessTime kwsysProcessTimeFromDouble(double d)
|
|
|
|
{
|
|
|
|
kwsysProcessTime t;
|
|
|
|
t.QuadPart = (LONGLONG)(d*10000000);
|
|
|
|
return t;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
int kwsysProcessTimeLess(kwsysProcessTime in1, kwsysProcessTime in2)
|
|
|
|
{
|
|
|
|
return in1.QuadPart < in2.QuadPart;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
kwsysProcessTime kwsysProcessTimeAdd(kwsysProcessTime in1, kwsysProcessTime in2)
|
|
|
|
{
|
|
|
|
kwsysProcessTime out;
|
|
|
|
out.QuadPart = in1.QuadPart + in2.QuadPart;
|
|
|
|
return out;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
kwsysProcessTime kwsysProcessTimeSubtract(kwsysProcessTime in1, kwsysProcessTime in2)
|
|
|
|
{
|
|
|
|
kwsysProcessTime out;
|
|
|
|
out.QuadPart = in1.QuadPart - in2.QuadPart;
|
|
|
|
return out;
|
|
|
|
}
|
2003-12-31 00:23:16 +03:00
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
#define KWSYSPE_CASE(type, str) \
|
|
|
|
cp->ExitException = kwsysProcess_Exception_##type; \
|
|
|
|
strcpy(cp->ExitExceptionString, str)
|
|
|
|
static void kwsysProcessSetExitException(kwsysProcess* cp, int code)
|
|
|
|
{
|
|
|
|
switch (code)
|
|
|
|
{
|
|
|
|
case STATUS_CONTROL_C_EXIT:
|
|
|
|
KWSYSPE_CASE(Interrupt, "User interrupt"); break;
|
|
|
|
|
|
|
|
case STATUS_FLOAT_DENORMAL_OPERAND:
|
|
|
|
KWSYSPE_CASE(Numerical, "Floating-point exception (denormal operand)"); break;
|
|
|
|
case STATUS_FLOAT_DIVIDE_BY_ZERO:
|
|
|
|
KWSYSPE_CASE(Numerical, "Divide-by-zero"); break;
|
|
|
|
case STATUS_FLOAT_INEXACT_RESULT:
|
|
|
|
KWSYSPE_CASE(Numerical, "Floating-point exception (inexact result)"); break;
|
|
|
|
case STATUS_FLOAT_INVALID_OPERATION:
|
|
|
|
KWSYSPE_CASE(Numerical, "Invalid floating-point operation"); break;
|
|
|
|
case STATUS_FLOAT_OVERFLOW:
|
|
|
|
KWSYSPE_CASE(Numerical, "Floating-point overflow"); break;
|
|
|
|
case STATUS_FLOAT_STACK_CHECK:
|
|
|
|
KWSYSPE_CASE(Numerical, "Floating-point stack check failed"); break;
|
|
|
|
case STATUS_FLOAT_UNDERFLOW:
|
|
|
|
KWSYSPE_CASE(Numerical, "Floating-point underflow"); break;
|
|
|
|
#ifdef STATUS_FLOAT_MULTIPLE_FAULTS
|
|
|
|
case STATUS_FLOAT_MULTIPLE_FAULTS:
|
|
|
|
KWSYSPE_CASE(Numerical, "Floating-point exception (multiple faults)"); break;
|
|
|
|
#endif
|
|
|
|
#ifdef STATUS_FLOAT_MULTIPLE_TRAPS
|
|
|
|
case STATUS_FLOAT_MULTIPLE_TRAPS:
|
|
|
|
KWSYSPE_CASE(Numerical, "Floating-point exception (multiple traps)"); break;
|
|
|
|
#endif
|
|
|
|
case STATUS_INTEGER_DIVIDE_BY_ZERO:
|
|
|
|
KWSYSPE_CASE(Numerical, "Integer divide-by-zero"); break;
|
|
|
|
case STATUS_INTEGER_OVERFLOW:
|
|
|
|
KWSYSPE_CASE(Numerical, "Integer overflow"); break;
|
|
|
|
|
|
|
|
case STATUS_DATATYPE_MISALIGNMENT:
|
|
|
|
KWSYSPE_CASE(Fault, "Datatype misalignment"); break;
|
|
|
|
case STATUS_ACCESS_VIOLATION:
|
|
|
|
KWSYSPE_CASE(Fault, "Access violation"); break;
|
|
|
|
case STATUS_IN_PAGE_ERROR:
|
|
|
|
KWSYSPE_CASE(Fault, "In-page error"); break;
|
|
|
|
case STATUS_INVALID_HANDLE:
|
|
|
|
KWSYSPE_CASE(Fault, "Invalid hanlde"); break;
|
|
|
|
case STATUS_NONCONTINUABLE_EXCEPTION:
|
|
|
|
KWSYSPE_CASE(Fault, "Noncontinuable exception"); break;
|
|
|
|
case STATUS_INVALID_DISPOSITION:
|
|
|
|
KWSYSPE_CASE(Fault, "Invalid disposition"); break;
|
|
|
|
case STATUS_ARRAY_BOUNDS_EXCEEDED:
|
|
|
|
KWSYSPE_CASE(Fault, "Array bounds exceeded"); break;
|
|
|
|
case STATUS_STACK_OVERFLOW:
|
|
|
|
KWSYSPE_CASE(Fault, "Stack overflow"); break;
|
|
|
|
|
|
|
|
case STATUS_ILLEGAL_INSTRUCTION:
|
|
|
|
KWSYSPE_CASE(Illegal, "Illegal instruction"); break;
|
|
|
|
case STATUS_PRIVILEGED_INSTRUCTION:
|
|
|
|
KWSYSPE_CASE(Illegal, "Privileged instruction"); break;
|
|
|
|
|
|
|
|
case STATUS_NO_MEMORY:
|
|
|
|
default:
|
|
|
|
cp->ExitException = kwsysProcess_Exception_Other;
|
|
|
|
sprintf(cp->ExitExceptionString, "Exit code 0x%x\n", code);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#undef KWSYSPE_CASE
|
|
|
|
|
2004-05-10 22:54:22 +04:00
|
|
|
typedef struct kwsysProcess_List_s kwsysProcess_List;
|
|
|
|
static kwsysProcess_List* kwsysProcess_List_New();
|
|
|
|
static void kwsysProcess_List_Delete(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List_Update(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List_NextProcess(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List_GetCurrentProcessId(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List_GetCurrentParentId(kwsysProcess_List* self);
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* Windows NT 4 API definitions. */
|
|
|
|
#define STATUS_INFO_LENGTH_MISMATCH ((NTSTATUS)0xC0000004L)
|
|
|
|
typedef LONG NTSTATUS;
|
|
|
|
typedef LONG KPRIORITY;
|
|
|
|
typedef struct _UNICODE_STRING UNICODE_STRING;
|
|
|
|
struct _UNICODE_STRING
|
|
|
|
{
|
|
|
|
USHORT Length;
|
|
|
|
USHORT MaximumLength;
|
|
|
|
PWSTR Buffer;
|
|
|
|
};
|
|
|
|
|
|
|
|
/* The process information structure. Declare only enough to get
|
|
|
|
process identifiers. The rest may be ignored because we use the
|
|
|
|
NextEntryDelta to move through an array of instances. */
|
|
|
|
typedef struct _SYSTEM_PROCESS_INFORMATION SYSTEM_PROCESS_INFORMATION;
|
|
|
|
typedef SYSTEM_PROCESS_INFORMATION* PSYSTEM_PROCESS_INFORMATION;
|
|
|
|
struct _SYSTEM_PROCESS_INFORMATION
|
|
|
|
{
|
|
|
|
ULONG NextEntryDelta;
|
|
|
|
ULONG ThreadCount;
|
|
|
|
ULONG Reserved1[6];
|
|
|
|
LARGE_INTEGER CreateTime;
|
|
|
|
LARGE_INTEGER UserTime;
|
|
|
|
LARGE_INTEGER KernelTime;
|
|
|
|
UNICODE_STRING ProcessName;
|
|
|
|
KPRIORITY BasePriority;
|
|
|
|
ULONG ProcessId;
|
|
|
|
ULONG InheritedFromProcessId;
|
|
|
|
};
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* Toolhelp32 API definitions. */
|
|
|
|
#define TH32CS_SNAPPROCESS 0x00000002
|
|
|
|
typedef struct tagPROCESSENTRY32 PROCESSENTRY32;
|
|
|
|
typedef PROCESSENTRY32* LPPROCESSENTRY32;
|
|
|
|
struct tagPROCESSENTRY32
|
|
|
|
{
|
|
|
|
DWORD dwSize;
|
|
|
|
DWORD cntUsage;
|
|
|
|
DWORD th32ProcessID;
|
|
|
|
DWORD th32DefaultHeapID;
|
|
|
|
DWORD th32ModuleID;
|
|
|
|
DWORD cntThreads;
|
|
|
|
DWORD th32ParentProcessID;
|
|
|
|
LONG pcPriClassBase;
|
|
|
|
DWORD dwFlags;
|
|
|
|
char szExeFile[MAX_PATH];
|
|
|
|
};
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
/* Windows API function types. */
|
|
|
|
typedef HANDLE (WINAPI* CreateToolhelp32SnapshotType)(DWORD, DWORD);
|
|
|
|
typedef BOOL (WINAPI* Process32FirstType)(HANDLE, LPPROCESSENTRY32);
|
|
|
|
typedef BOOL (WINAPI* Process32NextType)(HANDLE, LPPROCESSENTRY32);
|
|
|
|
typedef NTSTATUS (WINAPI* ZwQuerySystemInformationType)(ULONG, PVOID,
|
|
|
|
ULONG, PULONG);
|
|
|
|
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__New_NT4(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__New_Snapshot(kwsysProcess_List* self);
|
|
|
|
static void kwsysProcess_List__Delete_NT4(kwsysProcess_List* self);
|
|
|
|
static void kwsysProcess_List__Delete_Snapshot(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__Update_NT4(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__Update_Snapshot(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__Next_NT4(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__Next_Snapshot(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__GetProcessId_NT4(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__GetProcessId_Snapshot(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__GetParentId_NT4(kwsysProcess_List* self);
|
|
|
|
static int kwsysProcess_List__GetParentId_Snapshot(kwsysProcess_List* self);
|
|
|
|
|
|
|
|
struct kwsysProcess_List_s
|
|
|
|
{
|
|
|
|
/* Implementation switches at runtime based on version of Windows. */
|
|
|
|
int NT4;
|
|
|
|
|
|
|
|
/* Implementation functions and data for NT 4. */
|
|
|
|
ZwQuerySystemInformationType P_ZwQuerySystemInformation;
|
|
|
|
char* Buffer;
|
|
|
|
int BufferSize;
|
|
|
|
PSYSTEM_PROCESS_INFORMATION CurrentInfo;
|
|
|
|
|
|
|
|
/* Implementation functions and data for other Windows versions. */
|
|
|
|
CreateToolhelp32SnapshotType P_CreateToolhelp32Snapshot;
|
|
|
|
Process32FirstType P_Process32First;
|
|
|
|
Process32NextType P_Process32Next;
|
|
|
|
HANDLE Snapshot;
|
|
|
|
PROCESSENTRY32 CurrentEntry;
|
|
|
|
};
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static kwsysProcess_List* kwsysProcess_List_New()
|
|
|
|
{
|
|
|
|
OSVERSIONINFO osv;
|
|
|
|
kwsysProcess_List* self;
|
|
|
|
|
|
|
|
/* Allocate and initialize the list object. */
|
|
|
|
if(!(self = (kwsysProcess_List*)malloc(sizeof(kwsysProcess_List))))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
memset(self, 0, sizeof(*self));
|
|
|
|
|
|
|
|
/* Select an implementation. */
|
|
|
|
ZeroMemory(&osv, sizeof(osv));
|
|
|
|
osv.dwOSVersionInfoSize = sizeof(osv);
|
|
|
|
GetVersionEx(&osv);
|
|
|
|
self->NT4 = (osv.dwPlatformId == VER_PLATFORM_WIN32_NT &&
|
|
|
|
osv.dwMajorVersion < 5)? 1:0;
|
|
|
|
|
|
|
|
/* Initialize the selected implementation. */
|
|
|
|
if(!(self->NT4?
|
|
|
|
kwsysProcess_List__New_NT4(self) :
|
|
|
|
kwsysProcess_List__New_Snapshot(self)))
|
|
|
|
{
|
|
|
|
kwsysProcess_List_Delete(self);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Update to the current set of processes. */
|
|
|
|
if(!kwsysProcess_List_Update(self))
|
|
|
|
{
|
|
|
|
kwsysProcess_List_Delete(self);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return self;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static void kwsysProcess_List_Delete(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
if(self)
|
|
|
|
{
|
|
|
|
if(self->NT4)
|
|
|
|
{
|
|
|
|
kwsysProcess_List__Delete_NT4(self);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
kwsysProcess_List__Delete_Snapshot(self);
|
|
|
|
}
|
|
|
|
free(self);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List_Update(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return self? (self->NT4?
|
|
|
|
kwsysProcess_List__Update_NT4(self) :
|
|
|
|
kwsysProcess_List__Update_Snapshot(self)) : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List_GetCurrentProcessId(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return self? (self->NT4?
|
|
|
|
kwsysProcess_List__GetProcessId_NT4(self) :
|
|
|
|
kwsysProcess_List__GetProcessId_Snapshot(self)) : -1;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List_GetCurrentParentId(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return self? (self->NT4?
|
|
|
|
kwsysProcess_List__GetParentId_NT4(self) :
|
|
|
|
kwsysProcess_List__GetParentId_Snapshot(self)) : -1;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List_NextProcess(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return (self? (self->NT4?
|
|
|
|
kwsysProcess_List__Next_NT4(self) :
|
|
|
|
kwsysProcess_List__Next_Snapshot(self)) : 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__New_NT4(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
HANDLE hNT = GetModuleHandle("ntdll.dll");
|
|
|
|
if(hNT)
|
|
|
|
{
|
|
|
|
/* Get pointers to the needed API functions. */
|
|
|
|
self->P_ZwQuerySystemInformation =
|
|
|
|
((ZwQuerySystemInformationType)
|
|
|
|
GetProcAddress(hNT, "ZwQuerySystemInformation"));
|
|
|
|
CloseHandle(hNT);
|
|
|
|
}
|
|
|
|
if(!self->P_ZwQuerySystemInformation)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Allocate an initial process information buffer. */
|
|
|
|
self->BufferSize = 32768;
|
|
|
|
self->Buffer = (char*)malloc(self->BufferSize);
|
|
|
|
return self->Buffer? 1:0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static void kwsysProcess_List__Delete_NT4(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
/* Free the process information buffer. */
|
|
|
|
if(self->Buffer)
|
|
|
|
{
|
|
|
|
free(self->Buffer);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__Update_NT4(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
self->CurrentInfo = 0;
|
2005-04-22 17:21:19 +04:00
|
|
|
for(;;)
|
2004-05-10 22:54:22 +04:00
|
|
|
{
|
|
|
|
/* Query number 5 is for system process list. */
|
|
|
|
NTSTATUS status =
|
|
|
|
self->P_ZwQuerySystemInformation(5, self->Buffer, self->BufferSize, 0);
|
|
|
|
if(status == STATUS_INFO_LENGTH_MISMATCH)
|
|
|
|
{
|
|
|
|
/* The query requires a bigger buffer. */
|
|
|
|
int newBufferSize = self->BufferSize * 2;
|
|
|
|
char* newBuffer = (char*)malloc(newBufferSize);
|
|
|
|
if(newBuffer)
|
|
|
|
{
|
|
|
|
free(self->Buffer);
|
|
|
|
self->Buffer = newBuffer;
|
|
|
|
self->BufferSize = newBufferSize;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if(status >= 0)
|
|
|
|
{
|
|
|
|
/* The query succeeded. Initialize traversal of the process list. */
|
|
|
|
self->CurrentInfo = (PSYSTEM_PROCESS_INFORMATION)self->Buffer;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* The query failed. */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__Next_NT4(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
if(self->CurrentInfo)
|
|
|
|
{
|
|
|
|
if(self->CurrentInfo->NextEntryDelta > 0)
|
|
|
|
{
|
|
|
|
self->CurrentInfo = ((PSYSTEM_PROCESS_INFORMATION)
|
|
|
|
((char*)self->CurrentInfo +
|
|
|
|
self->CurrentInfo->NextEntryDelta));
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
self->CurrentInfo = 0;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__GetProcessId_NT4(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return self->CurrentInfo? self->CurrentInfo->ProcessId : -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__GetParentId_NT4(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return self->CurrentInfo? self->CurrentInfo->InheritedFromProcessId : -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__New_Snapshot(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
HANDLE hKernel = GetModuleHandle("kernel32.dll");
|
|
|
|
if(hKernel)
|
|
|
|
{
|
|
|
|
self->P_CreateToolhelp32Snapshot =
|
|
|
|
((CreateToolhelp32SnapshotType)
|
|
|
|
GetProcAddress(hKernel, "CreateToolhelp32Snapshot"));
|
|
|
|
self->P_Process32First =
|
|
|
|
((Process32FirstType)
|
|
|
|
GetProcAddress(hKernel, "Process32First"));
|
|
|
|
self->P_Process32Next =
|
|
|
|
((Process32NextType)
|
|
|
|
GetProcAddress(hKernel, "Process32Next"));
|
|
|
|
CloseHandle(hKernel);
|
|
|
|
}
|
|
|
|
return (self->P_CreateToolhelp32Snapshot &&
|
|
|
|
self->P_Process32First &&
|
|
|
|
self->P_Process32Next)? 1:0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static void kwsysProcess_List__Delete_Snapshot(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
if(self->Snapshot)
|
|
|
|
{
|
|
|
|
CloseHandle(self->Snapshot);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__Update_Snapshot(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
if(self->Snapshot)
|
|
|
|
{
|
|
|
|
CloseHandle(self->Snapshot);
|
|
|
|
}
|
|
|
|
if(!(self->Snapshot =
|
|
|
|
self->P_CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0)))
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
ZeroMemory(&self->CurrentEntry, sizeof(self->CurrentEntry));
|
|
|
|
self->CurrentEntry.dwSize = sizeof(self->CurrentEntry);
|
|
|
|
if(!self->P_Process32First(self->Snapshot, &self->CurrentEntry))
|
|
|
|
{
|
|
|
|
CloseHandle(self->Snapshot);
|
|
|
|
self->Snapshot = 0;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__Next_Snapshot(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
if(self->Snapshot)
|
|
|
|
{
|
|
|
|
if(self->P_Process32Next(self->Snapshot, &self->CurrentEntry))
|
|
|
|
{
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
CloseHandle(self->Snapshot);
|
|
|
|
self->Snapshot = 0;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__GetProcessId_Snapshot(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return self->Snapshot? self->CurrentEntry.th32ProcessID : -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static int kwsysProcess_List__GetParentId_Snapshot(kwsysProcess_List* self)
|
|
|
|
{
|
|
|
|
return self->Snapshot? self->CurrentEntry.th32ParentProcessID : -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static void kwsysProcessKill(DWORD pid)
|
|
|
|
{
|
|
|
|
HANDLE h = OpenProcess(PROCESS_TERMINATE, 0, pid);
|
|
|
|
if(h)
|
|
|
|
{
|
|
|
|
TerminateProcess(h, 255);
|
|
|
|
WaitForSingleObject(h, INFINITE);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static void kwsysProcessKillTree(int pid)
|
|
|
|
{
|
|
|
|
kwsysProcess_List* plist = kwsysProcess_List_New();
|
|
|
|
kwsysProcessKill(pid);
|
|
|
|
if(plist)
|
|
|
|
{
|
|
|
|
do
|
|
|
|
{
|
|
|
|
if(kwsysProcess_List_GetCurrentParentId(plist) == pid)
|
|
|
|
{
|
|
|
|
int ppid = kwsysProcess_List_GetCurrentProcessId(plist);
|
|
|
|
kwsysProcessKillTree(ppid);
|
|
|
|
}
|
|
|
|
} while(kwsysProcess_List_NextProcess(plist));
|
|
|
|
kwsysProcess_List_Delete(plist);
|
|
|
|
}
|
|
|
|
}
|
2004-07-08 02:15:20 +04:00
|
|
|
|
|
|
|
/*--------------------------------------------------------------------------*/
|
|
|
|
static void kwsysProcessDisablePipeThreads(kwsysProcess* cp)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
|
|
|
/* If data were just reported data, release the pipe's thread. */
|
|
|
|
if(cp->CurrentIndex < KWSYSPE_PIPE_COUNT)
|
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
ReleaseSemaphore(cp->Pipe[cp->CurrentIndex].Reader.Go, 1, 0);
|
2004-07-08 02:15:20 +04:00
|
|
|
cp->CurrentIndex = KWSYSPE_PIPE_COUNT;
|
|
|
|
}
|
|
|
|
|
2004-07-14 00:50:55 +04:00
|
|
|
/* Wakeup all reading threads that are not on closed pipes. */
|
2004-07-08 02:15:20 +04:00
|
|
|
for(i=0; i < KWSYSPE_PIPE_COUNT; ++i)
|
|
|
|
{
|
2004-07-14 00:23:49 +04:00
|
|
|
/* The wakeup threads will write one byte to the pipe write ends.
|
|
|
|
If there are no data in the pipe then this is enough to wakeup
|
|
|
|
the reading threads. If there are already data in the pipe
|
|
|
|
this may block. We cannot use PeekNamedPipe to check whether
|
|
|
|
there are data because an outside process might still be
|
|
|
|
writing data if we are disowning it. Also, PeekNamedPipe will
|
|
|
|
block if checking a pipe on which the reading thread is
|
|
|
|
currently calling ReadPipe. Therefore we need a separate
|
|
|
|
thread to call WriteFile. If it blocks, that is okay because
|
|
|
|
it will unblock when we close the read end and break the pipe
|
|
|
|
below. */
|
2004-07-14 00:50:55 +04:00
|
|
|
if(cp->Pipe[i].Read)
|
|
|
|
{
|
|
|
|
ReleaseSemaphore(cp->Pipe[i].Waker.Go, 1, 0);
|
|
|
|
}
|
2004-07-08 02:15:20 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Tell pipe threads to reset until we run another process. */
|
|
|
|
while(cp->PipesLeft > 0)
|
|
|
|
{
|
2004-07-14 00:50:55 +04:00
|
|
|
/* The waking threads will cause all reading threads to report.
|
|
|
|
Wait for the next one and save its index. */
|
2004-07-08 02:15:20 +04:00
|
|
|
WaitForSingleObject(cp->Full, INFINITE);
|
|
|
|
cp->CurrentIndex = cp->SharedIndex;
|
|
|
|
ReleaseSemaphore(cp->SharedIndexMutex, 1, 0);
|
2004-07-14 00:50:55 +04:00
|
|
|
|
|
|
|
/* We are done reading this pipe. Close its read handle. */
|
2004-07-08 02:15:20 +04:00
|
|
|
cp->Pipe[cp->CurrentIndex].Closed = 1;
|
2004-07-14 00:50:55 +04:00
|
|
|
kwsysProcessCleanupHandle(&cp->Pipe[cp->CurrentIndex].Read);
|
2004-07-08 02:15:20 +04:00
|
|
|
--cp->PipesLeft;
|
2004-07-14 00:50:55 +04:00
|
|
|
|
|
|
|
/* Tell the reading thread we are done with the data. It will
|
|
|
|
reset immediately because the pipe is closed. */
|
|
|
|
ReleaseSemaphore(cp->Pipe[cp->CurrentIndex].Reader.Go, 1, 0);
|
2004-07-08 02:15:20 +04:00
|
|
|
}
|
|
|
|
}
|