VC++获取控制台命令的输出结果
#include "stdafx.h"
#include <Windows.h>
#include <iostream>
#include <string>
using namespace std;
BOOL ExecDosCmd(string cmd, string &output)
{
SECURITY_ATTRIBUTES sa;
HANDLE hRead, hWrite;
sa.nLength = sizeof(SECURITY_ATTRIBUTES);
sa.lpSecurityDescriptor = NULL;
sa.bInheritHandle = TRUE;
if (!CreatePipe(&hRead, &hWrite, &sa, 0))
{
return FALSE;
}
char command[1024]; // 长达1K的命令行,够用了吧
strcpy(command, "Cmd.exe /C ");
strcat(command, cmd.c_str());
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;
// 关键步骤,CreateProcess函数参数意义请查阅MSDN
if (!CreateProcess(NULL, command, NULL, NULL, TRUE, NULL, NULL, NULL, &si, &pi))
{
CloseHandle(hWrite);
CloseHandle(hRead);
return FALSE;
}
CloseHandle(hWrite);
char buffer[4096] = {0}; // 用4K的空间来存储输出的内容,只要不是显示文件内容,一般情况下是够用了。
DWORD bytesRead;
while (true)
{
if (ReadFile(hRead, buffer, 4095, &bytesRead, NULL) == NULL)
break;
// buffer中就是执行的结果,可以保存到文本,也可以直接输出
// AfxMessageBox(buffer); //这里是弹出对话框显示
// cout << "----:" << endl << buffer << endl;
}
CloseHandle(hRead);
output = buffer;
return TRUE;
}
int main()
{
string output;
ExecDosCmd("wmic diskdrive get Caption,SerialNumber", output);
cout << output << endl;
return 0;
}