傳統(tǒng)CreateProcess傳遞StdIN StdOUT 管道方式存在進(jìn)程輸出不能及時(shí)刷管道問題導(dǎo)致ReadFile阻塞傳統(tǒng)解決方法是子進(jìn)程強(qiáng)制進(jìn)行fflush(stdout), 如果不能改變子進(jìn)程就沒辦法了從win10開始 windows支持偽終端可以完全模擬終端管道的行為傳統(tǒng)管道crt會使用GetFileType檢測STDHANDLE 對于控制臺調(diào)用檢測到的是字符設(shè)備而對于改變后的STDHANDLE 檢測到的是管道設(shè)備對發(fā)現(xiàn)管道設(shè)備時(shí)緩沖方式就變成了塊緩沖方式遇到換行符也不會及時(shí)flush這就導(dǎo)致預(yù)期輸出不能直接被ReadFile讀取到Win10上的偽終端方式接管進(jìn)程輸入輸出代碼//Kernel32.dll #ifndef PROC_THREAD_ATTRIBUTE_PSEUDOCONSOLE #define PROC_THREAD_ATTRIBUTE_PSEUDOCONSOLE 0x20016 #endif typedef HRESULT(WINAPI* PFN_CreatePseudoConsole)( _In_ COORD size, _In_ HANDLE hInput, _In_ HANDLE hOutput, _In_ DWORD dwFlags, _Out_ HPCON* phPC ); HRESULT MyCreatePseudoConsole( _In_ COORD size, _In_ HANDLE hInput, _In_ HANDLE hOutput, _In_ DWORD dwFlags, _Out_ HPCON* phPC ) { static PFN_CreatePseudoConsole fn (PFN_CreatePseudoConsole)::GetProcAddress(GetModuleHandle(LKernel32.dll), CreatePseudoConsole); if (fn) return fn(size, hInput, hOutput, dwFlags, phPC); return E_FAIL; } typedef BOOL(WINAPI* PFN_InitializeProcThreadAttributeList)( LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList, DWORD dwAttributeCount, DWORD dwFlags, PSIZE_T lpSize ); BOOL MyInitializeProcThreadAttributeList( LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList, DWORD dwAttributeCount, DWORD dwFlags, PSIZE_T lpSize ) { static PFN_InitializeProcThreadAttributeList fn (PFN_InitializeProcThreadAttributeList)::GetProcAddress(GetModuleHandle(LKernel32.dll), InitializeProcThreadAttributeList); if (fn) return fn(lpAttributeList, dwAttributeCount, dwFlags, lpSize); return 0; } typedef BOOL(WINAPI* PFN_UpdateProcThreadAttribute)( LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList, DWORD dwFlags, DWORD_PTR Attribute, PVOID lpValue, SIZE_T cbSize, PVOID lpPreviousValue, PSIZE_T lpReturnSize ); BOOL MyUpdateProcThreadAttribute( LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList, DWORD dwFlags, DWORD_PTR Attribute, PVOID lpValue, SIZE_T cbSize, PVOID lpPreviousValue, PSIZE_T lpReturnSize ) { static PFN_UpdateProcThreadAttribute fn (PFN_UpdateProcThreadAttribute)::GetProcAddress(GetModuleHandle(LKernel32.dll), UpdateProcThreadAttribute); if (fn) return fn(lpAttributeList, dwFlags, Attribute, lpValue, cbSize, lpPreviousValue, lpReturnSize); return 0; } typedef void(WINAPI* PFN_ClosePseudoConsole)(_In_ HPCON hPC); void MyClosePseudoConsole(_In_ HPCON hPC) { static PFN_ClosePseudoConsole fn (PFN_ClosePseudoConsole)::GetProcAddress(GetModuleHandle(LKernel32.dll), ClosePseudoConsole); if (fn) fn(hPC); } typedef VOID(WINAPI*PFN_DeleteProcThreadAttributeList)( LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList ); VOID MyDeleteProcThreadAttributeList( LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList ) { static PFN_DeleteProcThreadAttributeList fn (PFN_DeleteProcThreadAttributeList)::GetProcAddress(GetModuleHandle(LKernel32.dll), DeleteProcThreadAttributeList); if (fn) fn(lpAttributeList); } typedef struct _MySTARTUPINFOEXW { STARTUPINFOW StartupInfo; LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList; } MySTARTUPINFOEXW; BidirectionalPipe::~BidirectionalPipe() { Cleanup(); } bool BidirectionalPipe::StartProcess(bool bPsConsole, const std::wstring command) { SECURITY_ATTRIBUTES sa { 0 }; sa.nLength sizeof(SECURITY_ATTRIBUTES); sa.bInheritHandle TRUE; sa.lpSecurityDescriptor NULL; // 創(chuàng)建輸入管道父進(jìn)程寫子進(jìn)程讀 if (!CreatePipe(m_hChildStdinRead, m_hParentWrite, sa, 0)) { Cleanup(); return false; } // 創(chuàng)建輸出管道父進(jìn)程讀子進(jìn)程寫 if (!CreatePipe(m_hParentRead, m_hChildStdoutWrite, sa, 0)) { Cleanup(); return false; } if (bPsConsole) { // 2. 創(chuàng)建偽控制臺 (ConPTY) COORD consoleSize { 80, 25 }; // 初始控制臺大小 HRESULT hr MyCreatePseudoConsole(consoleSize, m_hChildStdinRead, m_hChildStdoutWrite, 0, m_hPC); if (FAILED(hr)) { Cleanup(); return false; } } // 3. 準(zhǔn)備 STARTUPINFOEXW 結(jié)構(gòu)體 MySTARTUPINFOEXW si { 0 }; if (bPsConsole) { si.StartupInfo.cb sizeof(MySTARTUPINFOEXW); } else { si.StartupInfo.cb sizeof(STARTUPINFOW); } si.StartupInfo.dwFlags STARTF_USESTDHANDLES; if (bPsConsole) { // 關(guān)鍵將 ConPTY 句柄附加到 StartupInfo SIZE_T size 0; MyInitializeProcThreadAttributeList(NULL, 1, 0, size); m_lpAttributeList (LPPROC_THREAD_ATTRIBUTE_LIST)HeapAlloc(GetProcessHeap(), 0, size); MyInitializeProcThreadAttributeList(m_lpAttributeList, 1, 0, size); MyUpdateProcThreadAttribute(m_lpAttributeList, 0, PROC_THREAD_ATTRIBUTE_PSEUDOCONSOLE, m_hPC, sizeof(HPCON), NULL, NULL); si.lpAttributeList m_lpAttributeList; si.StartupInfo.wShowWindow SW_HIDE; si.StartupInfo.dwFlags STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES; } else { si.StartupInfo.wShowWindow SW_HIDE; si.StartupInfo.dwFlags STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES; si.StartupInfo.hStdError m_hChildStdoutWrite; si.StartupInfo.hStdOutput m_hChildStdoutWrite; si.StartupInfo.hStdInput m_hChildStdinRead; } // 啟動進(jìn)程 if (!CreateProcessW(NULL, const_castwchar_t*(command.c_str()), NULL, NULL, bPsConsole ? FALSE : TRUE, bPsConsole ? EXTENDED_STARTUPINFO_PRESENT : 0, NULL, NULL, si.StartupInfo, m_pi)) { Cleanup(); return false; } // 【至關(guān)重要】CreateProcess 成功后父進(jìn)程必須關(guān)閉它不需要的管道句柄 // 否則 ReadFile 永遠(yuǎn)不會收到 EOF導(dǎo)致卡死。 if (m_hChildStdinRead) { CloseHandle(m_hChildStdinRead); m_hChildStdinRead 0; } if (m_hChildStdoutWrite) { CloseHandle(m_hChildStdoutWrite); m_hChildStdoutWrite 0; } return true; } void BidirectionalPipe::WriteInput(const std::string input) { if (m_hParentWrite NULL) return; DWORD bytesWritten; WriteFile(m_hParentWrite, input.c_str(), (DWORD)input.size(), bytesWritten, NULL); FlushFileBuffers(m_hParentWrite); } __declspec(noinline) CStringA BidirectionalPipe::ReadOutput() { CStringA strRet; const int bufMaxChars 4095; char buffer[bufMaxChars 10] { 0 }; DWORD bytesRead; //FlushFileBuffers(m_hChildStdoutWrite); // 第一次讀取 if (!ReadFile(m_hParentRead, buffer, bufMaxChars, bytesRead, NULL) || bytesRead 0) { return strRet; // 讀取失敗或無數(shù)據(jù) } buffer[bytesRead] \0; strRet buffer; //// 檢查是否讀取完所有數(shù)據(jù) //if (bytesRead bufMaxChars) { // return strRet; // 緩沖區(qū)未滿說明數(shù)據(jù)已讀完 //} // 緩沖區(qū)已滿檢查管道內(nèi)是否還有剩余數(shù)據(jù) while (true) { DWORD bytesAvailable 0; BOOL peekResult PeekNamedPipe( m_hParentRead, NULL, // 不讀取數(shù)據(jù)只查詢狀態(tài) 0, // 緩沖區(qū)大小為0 NULL, // 讀取的字節(jié)數(shù) bytesAvailable, // 可用字節(jié)數(shù) NULL // 消息剩余字節(jié)數(shù)用于消息模式 ); if (!peekResult || bytesAvailable 0) { break; // 查詢失敗或無剩余數(shù)據(jù) } // 動態(tài)分配緩沖區(qū)讀取剩余數(shù)據(jù)最多讀取bufMaxChars字節(jié) DWORD readSize min(bytesAvailable, (DWORD)bufMaxChars); if (ReadFile(m_hParentRead, buffer, readSize, bytesRead, NULL) bytesRead 0) { buffer[bytesRead] \0; strRet buffer; // 如果讀取的字節(jié)數(shù)等于緩沖區(qū)大小說明可能還有更多數(shù)據(jù) if (bytesRead readSize) { break; // 數(shù)據(jù)已讀完 } } else { break; // 讀取失敗 } } return strRet; } void BidirectionalPipe::Terminate() { if (m_pi.hProcess) { ::TerminateProcess(m_pi.hProcess, 0); WaitForSingleObject(m_pi.hProcess, 10000); } } void BidirectionalPipe::Cleanup() { if (m_hChildStdinRead) { CloseHandle(m_hChildStdinRead); m_hChildStdinRead 0; } if (m_hParentWrite) { CloseHandle(m_hParentWrite); m_hParentWrite 0; } if (m_hParentRead) { CloseHandle(m_hParentRead); m_hParentRead 0; } if (m_hChildStdoutWrite) { CloseHandle(m_hChildStdoutWrite); m_hChildStdoutWrite 0; } if (m_pi.hProcess) { CloseHandle(m_pi.hProcess); m_pi.hProcess 0; } if (m_pi.hThread) { CloseHandle(m_pi.hThread); m_pi.hThread 0; } if (m_lpAttributeList) { MyDeleteProcThreadAttributeList(m_lpAttributeList); HeapFree(GetProcessHeap(), 0, m_lpAttributeList); m_lpAttributeList 0; } if (m_hPC) { MyClosePseudoConsole(m_hPC); m_hPC 0; } } ///////////////////////////////////Header///////////////////////////// #pragma once typedef void* HPCON; class BidirectionalPipe { private: HANDLE m_hChildStdinRead NULL; HANDLE m_hParentWrite NULL; HANDLE m_hParentRead NULL; HANDLE m_hChildStdoutWrite NULL; PROCESS_INFORMATION m_pi { 0 }; LPPROC_THREAD_ATTRIBUTE_LIST m_lpAttributeList 0; HPCON m_hPC 0; public: ~BidirectionalPipe(); bool StartProcess(bool bPsConsole, const std::wstring command); void WriteInput(const std::string input); CStringA ReadOutput(); void Terminate(); private: void Cleanup(); };