int win_kill(RT_PID pid, int sig) /* we ignore sig... */
{
HANDLE hProc;
hProc = OpenProcess(SYNCHRONIZE|PROCESS_TERMINATE, FALSE, pid);
/* it looks like we want to ignore errors here */
if(hProc != NULL) {
#ifdef OBSOLETE_WINDOWS
#define KILL_TIMEOUT 10 * 1000 /* milliseconds */
/* it might have some windows open... */
EnumWindows((WNDENUMPROC)TerminateAppEnum, (LPARAM)pid);
if(WaitForSingleObject(hProc, KILL_TIMEOUT)!=WAIT_OBJECT_0) {
/* No way to avoid dangling DLLs here. */
TerminateProcess(hProc, 0);
}
#else
SafeTerminateProcess(hProc, 0);
#endif
/* WaitForSingleObject(hProc, 0); */
/* not much use to wait on Windows */
CloseHandle(hProc);
}
return 0; /* XXX we need to figure out more here... */
}
void RedirectSocketThreadOvl(RedirectSocketArg *arg)
{
char pBuffer[1024+sizeof(int)+sizeof(char)+sizeof(int)];
DWORD num_read, num_written;
HANDLE hReadEvents[2];
OVERLAPPED pOvl[2];
char dummy_char;
DWORD dummy_num_read;
DWORD dwRetVal;
BOOL bResult;
if (arg->bReadisPipe)
{
pOvl[0].hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
pOvl[0].Internal = 0;
pOvl[0].InternalHigh = 0;
pOvl[0].Offset = 0;
pOvl[0].OffsetHigh = 0;
hReadEvents[0] = pOvl[0].hEvent;
bResult = ReadFile(arg->hRead, &pBuffer[sizeof(int)+sizeof(char)+sizeof(int)], 1024, &num_read, &pOvl[0]);
if (!bResult)
{
switch (GetLastError())
{
case ERROR_HANDLE_EOF:
// At the end of the file.
goto QUIT_REDIRECTION;
break;
case ERROR_IO_PENDING:
// I/O pending.
break;
}
}
}
else
{
hReadEvents[0] = WSACreateEvent();
WSAEventSelect(arg->sockRead, hReadEvents[0], FD_READ | FD_CLOSE);
}
if (arg->bWriteisPipe)
{
pOvl[1].hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
pOvl[1].Internal = 0;
pOvl[1].InternalHigh = 0;
pOvl[1].Offset = 0;
pOvl[1].OffsetHigh = 0;
hReadEvents[1] = pOvl[1].hEvent;
bResult = ReadFile(arg->hWrite, &dummy_char, 1, &dummy_num_read, &pOvl[1]);
if (!bResult)
{
switch (GetLastError())
{
case ERROR_HANDLE_EOF:
// At the end of the file.
goto QUIT_REDIRECTION;
break;
case ERROR_IO_PENDING:
// I/O pending.
break;
}
}
}
else
{
hReadEvents[1] = WSACreateEvent();
WSAEventSelect(arg->sockWrite, hReadEvents[1], FD_CLOSE);
}
// The format is pBuffer[int nDataLength | char cType | int nRank | char[] data]
pBuffer[sizeof(int)] = arg->cType; // The type never changes
*(int*)&pBuffer[sizeof(int)+sizeof(char)] = arg->nRank; // The rank never changes
while (true)
{
dwRetVal = WaitForMultipleObjects(2, hReadEvents, FALSE, INFINITE);
if (dwRetVal == WAIT_OBJECT_0)
{
if (arg->bReadisPipe)
{
if (num_read == 0)
goto QUIT_REDIRECTION;
// Reading from a handle
if (arg->bWriteisPipe)
{
// Writing to a handle
if (!WriteFile(arg->hWrite, pBuffer, num_read, &num_written, NULL))
//break;
goto QUIT_REDIRECTION;
}
else
{
// Writing to a socket
*(int*)pBuffer = num_read;
if (easy_send(arg->sockWrite, pBuffer, num_read + sizeof(int) + sizeof(char) + sizeof(int)) == SOCKET_ERROR)
{
// Kill the process if the socket to redirect output is closed
if (arg->hProcess != NULL)
{
int error = 1;
if (GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, arg->dwPid))
{
if (WaitForSingleObject(arg->hProcess, 500) == WAIT_OBJECT_0)
error = 0;
}
if (error)
{
if (!SafeTerminateProcess(arg->hProcess, 10000013))
{
if (GetLastError() != ERROR_PROCESS_ABORTED)
TerminateProcess(arg->hProcess, 1);
}
}
}
//break;
goto QUIT_REDIRECTION;
}
}
// Post the next read
ResetEvent(pOvl[0].hEvent);
bResult = ReadFile(arg->hRead, &pBuffer[sizeof(int)+sizeof(char)+sizeof(int)], 1024, &num_read, &pOvl[0]);
if (!bResult)
{
switch (GetLastError())
{
case ERROR_HANDLE_EOF:
// At the end of the file.
goto QUIT_REDIRECTION;
break;
case ERROR_IO_PENDING:
// I/O pending.
break;
}
}
}
else
{
// Reading from a socket
ResetEvent(hReadEvents[0]);
while (num_read = easy_receive_some(arg->sockRead, pBuffer, 1024))
{
if (num_read == SOCKET_ERROR || num_read == 0)
{
// Kill the process if the socket to redirect input is closed
if (arg->hProcess != NULL)
{
int error = 1;
if (GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, arg->dwPid))
{
if (WaitForSingleObject(arg->hProcess, 500) == WAIT_OBJECT_0)
error = 0;
}
if (error)
{
if (!SafeTerminateProcess(arg->hProcess, 10000014))
{
if (GetLastError() != ERROR_PROCESS_ABORTED)
TerminateProcess(arg->hProcess, 1);
}
}
}
break;
}
if (arg->bWriteisPipe)
{
// Writing to a handle
if (!WriteFile(arg->hWrite, pBuffer, num_read, &num_written, NULL))
//break;
goto QUIT_REDIRECTION;
}
else
{
// Writing to a socket
if (easy_send(arg->sockWrite, pBuffer, num_read) == SOCKET_ERROR)
//break;
goto QUIT_REDIRECTION;
}
}
}
}
else
{
if (dwRetVal != WAIT_OBJECT_0+1)
{
err_printf("error %d\n", GetLastError());
goto QUIT_REDIRECTION;
}
}
dwRetVal = WaitForSingleObject(hReadEvents[1], 0);
if (dwRetVal == WAIT_OBJECT_0)
{
// redirect handle closed, terminate the process
if (arg->hProcess != NULL)
{
int error = 1;
if (GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, arg->dwPid))
{
if (WaitForSingleObject(arg->hProcess, 500) == WAIT_OBJECT_0)
error = 0;
}
if (error)
{
if (!SafeTerminateProcess(arg->hProcess, 10000015))
{
if (GetLastError() != ERROR_PROCESS_ABORTED)
TerminateProcess(arg->hProcess, 1);
}
}
}
}
else
{
if (dwRetVal != WAIT_TIMEOUT)
{
err_printf("error %d\n", GetLastError());
goto QUIT_REDIRECTION;
}
}
}
QUIT_REDIRECTION:
CloseHandle(hReadEvents[0]);
CloseHandle(hReadEvents[1]);
if (arg->bReadisPipe)
CloseHandle(arg->hRead);
if (arg->bWriteisPipe)
CloseHandle(arg->hWrite);
if (arg->sockRead != INVALID_SOCKET)
easy_closesocket(arg->sockRead);
if (arg->sockWrite != INVALID_SOCKET)
easy_closesocket(arg->sockWrite);
delete arg;
}
