Windows-Server-2003/sdktools/remote/srvchild.c

170 lines
3.8 KiB
C

/******************************************************************************\
* This is a part of the Microsoft Source Code Samples.
* Copyright 1992 - 1997 Microsoft Corporation.
* All rights reserved.
* This source code is only intended as a supplement to
* Microsoft Development Tools and/or WinHelp documentation.
* See these sources for detailed information regarding the
* Microsoft samples programs.
\******************************************************************************/
/*++
Copyright 1992 - 1997 Microsoft Corporation
Module Name:
SrvChild.c
Abstract:
The server component of Remote. It spawns a child process
and redirects the stdin/stdout/stderr of child to itself.
Waits for connections from clients - passing the
output of child process to client and the input from clients
to child process.
Author:
Rajivendra Nath 2-Jan-1992
Dave Hart 30 May 1997 split from Server.c
Environment:
Console App. User mode.
Revision History:
--*/
#include <precomp.h>
#include "Remote.h"
#include "Server.h"
VOID
FASTCALL
StartChildOutPipeRead(
VOID
)
{
ReadChildOverlapped.OffsetHigh =
ReadChildOverlapped.Offset = 0;
if ( ! ReadFileEx(
hReadChildOutput,
ReadChildBuffer,
sizeof(ReadChildBuffer) - 1, // allow for null term
&ReadChildOverlapped,
ReadChildOutputCompleted
)) {
if (INVALID_HANDLE_VALUE != hWriteChildStdIn) {
CancelIo( hWriteChildStdIn );
CloseHandle( hWriteChildStdIn );
hWriteChildStdIn = INVALID_HANDLE_VALUE;
}
}
}
VOID
WINAPI
ReadChildOutputCompleted(
DWORD dwError,
DWORD cbRead,
LPOVERLAPPED lpO
)
{
UNREFERENCED_PARAMETER(lpO);
//
// We can get called after hWriteTempFile
// is closed after the child has exited.
//
if (! dwError &&
INVALID_HANDLE_VALUE != hWriteTempFile) {
//
// Start a write to the temp file.
//
ReadChildOverlapped.OffsetHigh = 0;
ReadChildOverlapped.Offset = dwWriteFilePointer;
if ( ! WriteFileEx(
hWriteTempFile,
ReadChildBuffer,
cbRead,
&ReadChildOverlapped,
WriteTempFileCompleted
)) {
dwError = GetLastError();
if (ERROR_DISK_FULL == dwError) {
printf("Remote: disk full writing temp file %s, exiting\n", SaveFileName);
if (INVALID_HANDLE_VALUE != hWriteChildStdIn) {
CancelIo( hWriteChildStdIn );
CloseHandle( hWriteChildStdIn );
hWriteChildStdIn = INVALID_HANDLE_VALUE;
}
} else {
ErrorExit("WriteFileEx for temp file failed.");
}
}
}
}
VOID
WINAPI
WriteTempFileCompleted(
DWORD dwError,
DWORD cbWritten,
LPOVERLAPPED lpO
)
{
UNREFERENCED_PARAMETER(lpO);
if (dwError) {
if (ERROR_DISK_FULL == dwError) {
printf("Remote: disk full writing temp file %s, exiting\n", SaveFileName);
if (INVALID_HANDLE_VALUE != hWriteChildStdIn) {
CancelIo( hWriteChildStdIn );
CloseHandle( hWriteChildStdIn );
hWriteChildStdIn = INVALID_HANDLE_VALUE;
}
return;
} else {
SetLastError(dwError);
ErrorExit("WriteTempFileCompleted may need work");
}
}
dwWriteFilePointer += cbWritten;
TRACE(CHILD, ("Wrote %d bytes to temp file\n", cbWritten));
StartServerToClientFlow();
//
// Start another read against the child input.
//
StartChildOutPipeRead();
}