匿名管道实现控制台重定向

// NPSevDlg.cpp : 实现文件 // #include "stdafx.h" #include "NPSev.h" #include "NPSevDlg.h" #ifdef _DEBUG #define new DEBUG_NEW #endif #define MSG_DATAREC WM_USER+1 // 用于应用程序“关于”菜单项的 CAboutDlg 对话框 class CAboutDlg : public CDialog { public: CAboutDlg(); // 对话框数据 enum { IDD = IDD_ABOUTBOX }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialog) END_MESSAGE_MAP() // CNPSevDlg 对话框 CNPSevDlg::CNPSevDlg(CWnd* pParent /*=NULL*/) : CDialog(CNPSevDlg::IDD, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CNPSevDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); DDX_Control(pDX, IDC_LIST1, m_lstResult); } BEGIN_MESSAGE_MAP(CNPSevDlg, CDialog) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_MESSAGE(MSG_DATAREC,CNPSevDlg::OnDataRec) //}}AFX_MSG_MAP ON_BN_CLICKED(IDC_BUTTON1, &CNPSevDlg::OnBnClickedButton1) END_MESSAGE_MAP() // CNPSevDlg 消息处理程序 BOOL CNPSevDlg::OnInitDialog() { CDialog::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != NULL) { CString strAboutMenu; strAboutMenu.LoadString(IDS_ABOUTBOX); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 // TODO: 在此添加额外的初始化代码 return TRUE; // 除非将焦点设置到控件，否则返回 TRUE } void CNPSevDlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialog::OnSysCommand(nID, lParam); } } // 如果向对话框添加最小化按钮，则需要下面的代码 // 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序， // 这将由框架自动完成。 void CNPSevDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使图标在工作区矩形中居中 int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // 绘制图标 dc.DrawIcon(x, y, m_hIcon); } else { CDialog::OnPaint(); } } //当用户拖动最小化窗口时系统调用此函数取得光标 //显示。 HCURSOR CNPSevDlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } //显示 子进程重定向的输出 LRESULT CNPSevDlg::OnDataRec(WPARAM wParam, LPARAM lParam) { m_lstResult.AddString(m_szData); return 0; } namespace { //读管道线程 UINT ReadPipeProc( LPVOID pParam ) { CNPSevDlg * pAttachWnd = static_cast<CNPSevDlg *>(pParam); HANDLE hRead = pAttachWnd->m_hReadPipe; HWND hWnd = pAttachWnd->GetSafeHwnd(); DWORD bytesRead; while( 1 ) { int len = sizeof(pAttachWnd->m_szData); ZeroMemory(&pAttachWnd->m_szData,len); if( !ReadFile(hRead, pAttachWnd->m_szData, len-1, &bytesRead, NULL) ) break; SendMessage(hWnd,MSG_DATAREC,0,0); } return 0; } } //启用ping命令行 void CNPSevDlg::OnBnClickedButton1() { HANDLE hRead, hWrite; //创建匿名管道 { SECURITY_ATTRIBUTES sa; ZeroMemory(&sa,sizeof(sa)); sa.nLength = sizeof(SECURITY_ATTRIBUTES); sa.lpSecurityDescriptor = NULL; //默认的安全描述符 sa.bInheritHandle = TRUE; //这个必须要设定TRUE,参考资料：《windows核心编程》第三章 if( !CreatePipe(&hRead, &hWrite, &sa, 0) ) { MessageBox(" CreatePipe return FALSE."); return; } m_hReadPipe = hRead; } //创建子进程 { //设置启动参数 STARTUPINFO si; PROCESS_INFORMATION pi; si.cb = sizeof(STARTUPINFO); GetStartupInfo(&si); si.hStdError = hWrite; si.hStdOutput = hWrite; si.wShowWindow = SW_HIDE; si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES; AfxBeginThread(ReadPipeProc,this,NULL); TCHAR szCmdPar[256] = "ping www.baidu.com"; if(!CreateProcess(NULL, szCmdPar,NULL,NULL,TRUE,NULL,NULL,NULL,&si, &pi)) { MessageBox("CreateProcess failed!"); return; } CloseHandle(pi.hProcess); CloseHandle(pi.hThread); } }