Redirect Output of CMD.EXE to a Pipe (5KB)

// ShellView.cpp : implementation file // #include "stdafx.h" #include "gurueditor.h" #include "ShellView.h" #ifdef _DEBUG #define new DEBUG_NEW #undef THIS_FILE static char THIS_FILE[] = __FILE__; #endif ///////////////////////////////////////////////////////////////////////////// // CShellView IMPLEMENT_DYNCREATE(CShellView, CEditView) CShellView::CShellView() { dwProcessId = DWORD(-1); m_pReadThread = NULL; // Initialize LOGFONT structure memset(&m_lf, 0, sizeof(m_lf)); m_lf.lfWeight = FW_NORMAL; m_lf.lfCharSet = GB2312_CHARSET;//ANSI_CHARSET; m_lf.lfOutPrecision = OUT_DEFAULT_PRECIS; m_lf.lfClipPrecision = CLIP_DEFAULT_PRECIS; m_lf.lfQuality = DEFAULT_QUALITY; m_lf.lfPitchAndFamily = DEFAULT_PITCH | FF_DONTCARE; lstrcpy(m_lf.lfFaceName, _T("FixedSys")); } CShellView::~CShellView() { } BEGIN_MESSAGE_MAP(CShellView, CEditView) //{{AFX_MSG_MAP(CShellView) ON_WM_CREATE() ON_WM_DESTROY() ON_WM_CHAR() ON_WM_KEYDOWN() //}}AFX_MSG_MAP END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CShellView drawing void CShellView::OnDraw(CDC* pDC) { CDocument* pDoc = GetDocument(); // TODO: add draw code here } ///////////////////////////////////////////////////////////////////////////// // CShellView diagnostics #ifdef _DEBUG void CShellView::AssertValid() const { CEditView::AssertValid(); } void CShellView::Dump(CDumpContext& dc) const { CEditView::Dump(dc); } #endif //_DEBUG ///////////////////////////////////////////////////////////////////////////// // CShellView message handlers int CShellView::OnCreate(LPCREATESTRUCT lpCreateStruct) { if (CEditView::OnCreate(lpCreateStruct) == -1) return -1; // GetEditCtrl().SetReadOnly( TRUE ); DWORD dwStyle = GetWindowLong( GetEditCtrl().GetSafeHwnd(),GWL_STYLE ); if( dwStyle ) { dwStyle |= DS_LOCALEDIT; SetWindowLong( GetEditCtrl().GetSafeHwnd(),GWL_STYLE,dwStyle ); } if( m_defFont.CreateFontIndirect( &m_lf ) ) GetEditCtrl().SetFont( &m_defFont ); if( !CreateShellRedirect() ) return -1; return 0; } #define BUFSIZE 4096 BOOL CShellView::CreateShellRedirect() { SECURITY_ATTRIBUTES saAttr; BOOL fSuccess; // Set the bInheritHandle flag so pipe handles are inherited. saAttr.nLength = sizeof(SECURITY_ATTRIBUTES); saAttr.bInheritHandle = TRUE; saAttr.lpSecurityDescriptor = NULL; // The steps for redirecting child process's STDOUT: // 1. Save current STDOUT, to be restored later. // 2. Create anonymous pipe to be STDOUT for child process. // 3. Set STDOUT of the parent process to be write handle to // the pipe, so it is inherited by the child process. // 4. Create a noninheritable duplicate of the read handle and // close the inheritable read handle. // Save the handle to the current STDOUT. hSaveStdout = GetStdHandle(STD_OUTPUT_HANDLE); // Create a pipe for the child process's STDOUT. if( !CreatePipe( &hChildStdoutRd, &hChildStdoutWr, &saAttr, 0) ) { TRACE0( _T("Stdout pipe creation failed\n") ); return FALSE; } // Set a write handle to the pipe to be STDOUT. if( !SetStdHandle(STD_OUTPUT_HANDLE, hChildStdoutWr) ) { TRACE0( _T("Redirecting STDOUT failed\n") ); return FALSE; } // Create noninheritable read handle and close the inheritable read handle. fSuccess = DuplicateHandle( GetCurrentProcess(), hChildStdoutRd, GetCurrentProcess(), &hChildStdoutRdDup , 0, FALSE, DUPLICATE_SAME_ACCESS ); if( !fSuccess ) { TRACE0( _T("DuplicateHandle failed\n") ); return FALSE; } CloseHandle( hChildStdoutRd ); // The steps for redirecting child process's STDIN: // 1. Save current STDIN, to be restored later. // 2. Create anonymous pipe to be STDIN for child process. // 3. Set STDIN of the parent to be the read handle to the // pipe, so it is inherited by the child process. // 4. Create a noninheritable duplicate of the write handle, // and close the inheritable write handle. // Save the handle to the current STDIN. hSaveStdin = GetStdHandle(STD_INPUT_HANDLE); // Create a pipe for the child process's STDIN. if( !CreatePipe(&hChildStdinRd, &hChildStdinWr, &saAttr, 0) ) { TRACE0( _T("Stdin pipe creation failed\n") ); return FALSE; } // Set a read handle to the pipe to be STDIN. if( !SetStdHandle(STD_INPUT_HANDLE, hChildStdinRd) ) { TRACE0( _T("Redirecting Stdin failed\n") ); return FALSE; } // Duplicate the write handle to the pipe so it is not inherited. fSuccess = DuplicateHandle(GetCurrentProcess(), hChildStdinWr, GetCurrentProcess(), &hChildStdinWrDup, 0, FALSE, // not inherited DUPLICATE_SAME_ACCESS ); if( !fSuccess ) { TRACE0( _T("DuplicateHandle failed\n") ); return FALSE; } CloseHandle(hChildStdinWr); // Now create the child process. if( !CreateChildProcess(dwProcessId) ) { TRACE0( _T("CreateChildProcess failed\n") ); return FALSE; } // After process creation, restore the saved STDIN and STDOUT. if( !SetStdHandle(STD_INPUT_HANDLE, hSaveStdin) ) { TRACE0( _T("Re-redirecting Stdin failed\n") ); return FALSE; } if( !SetStdHandle(STD_OUTPUT_HANDLE, hSaveStdout) ) { TRACE0( _T("Re-redirecting Stdout failed\n") ); return FALSE; } m_pReadThread = AfxBeginThread( (AFX_THREADPROC)ReadPipeThreadProc,(LPVOID)this ); if( !m_pReadThread ) { TRACE0( _T("Cannot start read-redirect thread!\n") ); return FALSE; } return TRUE; } BOOL CShellView::CreateChildProcess(DWORD& dwProcessId) { PROCESS_INFORMATION piProcInfo; STARTUPINFO siStartInfo; // Set up members of STARTUPINFO structure. ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) ); siStartInfo.cb = sizeof(STARTUPINFO); siStartInfo.dwFlags = STARTF_USESTDHANDLES; siStartInfo.hStdInput = hChildStdinRd; siStartInfo.hStdOutput = hChildStdoutWr; siStartInfo.hStdError = hChildStdoutWr; TCHAR shellCmd[_MAX_PATH]; if( !GetEnvironmentVariable(_T("ComSpec"), shellCmd, _MAX_PATH) ) return FALSE; #ifdef _UNICODE _tcscat( shellCmd, _T(" /U") ); #else _tcscat( shellCmd, _T(" /A") ); #endif // Create the child process. BOOL ret = CreateProcess( NULL, shellCmd, // applicatin name NULL, // process security attributes NULL, // primary thread security attributes TRUE, // handles are inherited DETACHED_PROCESS, // creation flags NULL, // use parent's environment NULL, // use parent's current directory &siStartInfo, // STARTUPINFO pointer &piProcInfo); // receives PROCESS_INFORMATION if( ret ) dwProcessId = piProcInfo.dwProcessId; return ret; } void CShellView::WriteToPipe( LPCTSTR line ) { DWORD dwWritten; WriteFile( hChildStdinWrDup, line, _tcslen(line)*sizeof(TCHAR), &dwWritten, NULL ); } UINT CShellView::ReadPipeThreadProc( LPVOID pParam ) { DWORD dwRead; TCHAR chBuf[BUFSIZE]; CShellView* pView = (CShellView*)pParam; TRACE0( _T("ReadPipe Thread begin run\n") ); for( ;; ) { if( !ReadFile( pView->hChildStdoutRdDup, chBuf, BUFSIZE, &dwRead, NULL) || dwRead == 0) break; chBuf[dwRead/sizeof(TCHAR)] = _T('\0'); pView->AddTexts( chBuf ); pView->m_nLength = pView->GetEditCtrl().SendMessage( WM_GETTEXTLENGTH ); } CloseHandle( pView ->hChildStdinRd); CloseHandle( pView ->hChildStdoutWr); CloseHandle( pView ->hChildStdinWrDup ); CloseHandle( pView ->hChildStdoutRdDup ); pView->m_pReadThread = NULL; pView->dwProcessId = DWORD(-1); pView->PostMessage( WM_CLOSE ); return 1; } void CShellView::OnDestroy() { if( dwProcessId!=DWORD(-1) ) { HANDLE hProcess = OpenProcess( PROCESS_ALL_ACCESS,