// 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;
}
VC中调用CMD命令并获取调用结果
5星 · 超过95%的资源 需积分: 37 9 浏览量
2014-06-26
14:48:53
上传
评论 10
收藏 2.28MB 7Z 举报
wwkaven
- 粉丝: 228
- 资源: 53
- 1
- 2
- 3
前往页