VC中调用CMD命令并获取调用结果

// SysCmdTestDlg.cpp : implementation file // #include "stdafx.h" #include "SysCmdTest.h" #include "SysCmdTestDlg.h" #ifdef _DEBUG #define new DEBUG_NEW #undef THIS_FILE static char THIS_FILE[] = __FILE__; #endif // 全局变量 HANDLE h1 = NULL; // 线程句柄 DWORD pid = 0; // 保存cmd窗口的PID CString inputCmd, outputCmd; // 保存cmd命令和返回的结果 CEdit * m_edit; // 输出框 ///////////////////////////////////////////////////////////////////////////// // CSysCmdTestDlg dialog CSysCmdTestDlg::CSysCmdTestDlg(CWnd* pParent /*=NULL*/) : CDialog(CSysCmdTestDlg::IDD, pParent) { //{{AFX_DATA_INIT(CSysCmdTestDlg) // NOTE: the ClassWizard will add member initialization here //}}AFX_DATA_INIT m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CSysCmdTestDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); //{{AFX_DATA_MAP(CSysCmdTestDlg) // NOTE: the ClassWizard will add DDX and DDV calls here //}}AFX_DATA_MAP } BEGIN_MESSAGE_MAP(CSysCmdTestDlg, CDialog) //{{AFX_MSG_MAP(CSysCmdTestDlg) ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDC_BUTTON1, OnButton1) ON_WM_CLOSE() //}}AFX_MSG_MAP END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CSysCmdTestDlg message handlers BOOL CSysCmdTestDlg::OnInitDialog() { CDialog::OnInitDialog(); SetIcon(m_hIcon, TRUE); // Set big icon SetIcon(m_hIcon, FALSE); // Set small icon // TODO: Add extra initialization here // 保存输出框对象 m_edit = (CEdit*)GetDlgItem(IDC_EDIT1); return TRUE; // return TRUE unless you set the focus to a control } // If you add a minimize button to your dialog, you will need the code below // to draw the icon. For MFC applications using the document/view model, // this is automatically done for you by the framework. void CSysCmdTestDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // device context for painting SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0); // Center icon in client rectangle 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; // Draw the icon dc.DrawIcon(x, y, m_hIcon); } else { CDialog::OnPaint(); } } HCURSOR CSysCmdTestDlg::OnQueryDragIcon() { return (HCURSOR) m_hIcon; } void CSysCmdTestDlg::OnButton1() { // TODO: Add your control notification handler code here GetDlgItemText(IDC_EDIT2,inputCmd); //获取编辑框中输入的命令行 // 关闭CMD窗口 ::SendMessage(GetWindowHandleByPID(pid), WM_CLOSE, NULL, NULL); // 销毁线程 CloseHandle(h1); // 创建线程 h1 = ::CreateThread(NULL, 0, ThreadFun, NULL, 0, 0); } /************************************************************************/ /* 线程函数 */ /************************************************************************/ DWORD WINAPI CSysCmdTestDlg::ThreadFun(LPVOID lpParam) { SECURITY_ATTRIBUTES sa; HANDLE hRead,hWrite; sa.nLength = sizeof(SECURITY_ATTRIBUTES); sa.lpSecurityDescriptor = NULL; //使用系统默认的安全描述符 sa.bInheritHandle = TRUE; //创建的进程继承句柄 if (!CreatePipe(&hRead,&hWrite,&sa,0)) //创建匿名管道 { ::MessageBox(NULL,"CreatePipe Failed!","提示",MB_OK | MB_ICONWARNING); return 0; } STARTUPINFO si; PROCESS_INFORMATION pi; ZeroMemory(&si,sizeof(STARTUPINFO)); si.cb = sizeof(STARTUPINFO); GetStartupInfo(&si); si.hStdError = hWrite; si.hStdOutput = hWrite; //新创建进程的标准输出连在写管道一端 si.wShowWindow = SW_HIDE; //隐藏窗口 si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES; char cmdline[200]; CString tmp,stredit2; tmp.Format("cmd /C %s",inputCmd); // inputCmd为输入的cmd命令 sprintf(cmdline,"%s",tmp); if (!CreateProcess(NULL,cmdline,NULL,NULL,TRUE,NULL,NULL,NULL,&si,&pi)) //创建子进程 { ::MessageBox(NULL,"CreateProcess Failed!","提示",MB_OK | MB_ICONWARNING); return 0; } pid = pi.dwProcessId; CloseHandle(hWrite); //关闭管道句柄 char buffer[4096] = {0}; DWORD bytesRead; while (true) { if (ReadFile(hRead,buffer,4095,&bytesRead,NULL) == NULL) //读取管道 break; CString temp; temp = buffer; // outpuuCmd为输出的结果 temp += _T("\r\n\r\n"); outputCmd += temp; //显示输出信息到编辑框,并刷新窗口 int len = outputCmd.GetLength(); m_edit->SetWindowText(outputCmd); m_edit->SetSel(len, len); } CloseHandle(hRead); return 1; } void CSysCmdTestDlg::OnClose() { // TODO: Add your message handler code here and/or call default // 关闭CMD窗口 ::SendMessage(GetWindowHandleByPID(pid), WM_CLOSE, NULL, NULL); // 销毁线程 CloseHandle(h1); CDialog::OnClose(); } /************************************************************************/ /* 根据进程ID（PID）获取窗体句柄 */ /************************************************************************/ HWND CSysCmdTestDlg::GetWindowHandleByPID(DWORD dwProcessID) { HWND h = ::GetTopWindow(0); while ( h ) { DWORD pid = 0; DWORD dwTheardId = GetWindowThreadProcessId( h,&pid); if (dwTheardId != 0) { if ( pid == dwProcessID/*your process id*/ ) { // here h is the handle to the window return h; } } h = ::GetNextWindow( h , GW_HWNDNEXT); } return NULL; }