本文的目的是说明如何从任何应用程序(无论其类型(MFC,Win32,控制台))运行CMD命令,等待结果并使用您自己的用户界面查看它们。
背景
作为SecuredGlobe,Inc.日常工作的一部分,我们运行预制程序。他们所做的只是在响应结果的同时以编程方式执行一系列CMD命令。这就提出了为此目的构建通用工具的想法。
“如何”
首先,我们定义3个全局变量来存储命令的异步处理状态。
Command-您将在CMD中键入的命令
CommandResult-执行完成后您将在屏幕上看到的结果
IsRunning-指示命令是否仍在处理中。要测试处理时间更长的命令,请尝试输入“netstat”
CString Command, CommandResult;
BOOL IsRunning = FALSE;12复制代码类型:[cpp]
DoRun()函数
我们编写string发送给ShellExecute()的实际位置是DoRun()。
提提您:您可能更喜欢使用ShellExecuteEx()或CreateProcess()。
该DoRun()函数首先删除result.txt的任何旧版本,然后正确发送命令。
BOOL DoRun(WCHAR *command)
{
BOOL Result = FALSE;
DWORD retSize;
LPTSTR pTemp = NULL;
TCHAR Command[BUFSIZE] = L"";
if (!(DeleteFile(RESULTS_FILE)))
{
//return L"Can't delete previous results";
}
_tcscpy_s(Command, L"/C ");
_tcscat_s(Command, command);
_tcscat_s(Command, L" >");
_tcscat_s(Command, RESULTS_FILE);
wprintf(L"Calling:\n%s\n", Command);
Result = (BOOL) ShellExecute(GetActiveWindow(), L"OPEN", L"cmd", Command, NULL, 0L);
if(!Result)
{
retSize = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_ARGUMENT_ARRAY,
NULL,
GetLastError(),
LANG_NEUTRAL,
(LPTSTR)&pTemp,
0,
NULL);
MessageBox(NULL,pTemp,L"Error",MB_OK);
}
return Result;
}12345678910111213141516171819202122232425262728293031复制代码类型:[cpp]
SetCommand函数
该SetCommand函数用于通过其自己的线程启动新命令。
void SetCommand(CString command)
{
Command = command;
HANDLE hThread = (HANDLE)_beginthread(ThreadFunc, 0, NULL);
}12345复制代码类型:[cpp]
然后线程函数本身如下:
void __cdecl ThreadFunc(void*)
{
CommandResult = L"";
if (DoRun(Command.GetBuffer()))
{
IsRunning = TRUE;
while (IsRunning)
{
if(CheckCommandExecutionStatus())
{
break;
}
}
}
IsRunning = FALSE;
_endthreadex(0);
}123456789101112131415161718复制代码类型:[cpp]
CheckCommandResult()像这样:
bool CheckCommandExecutionStatus()
{
CommandResult = GetResultFromFile();
if (CommandResult != L"")
return true;
else
return false;
}12345678复制代码类型:[cpp]
得到结果
命令执行完成后,我们希望将结果保存在.txt文件中,然后重新使用。文件名为“result.txt”。
该GetResultFromFile()函数读取该文件并返回CString结果。
带注释的部分可用于无法打开文件的情况,但是由于不希望发生这种情况,因此我已注释了该部分。
CString GetResultFromFile()
{
CString strResult = L"";
std::FILE *fp;
fp = NULL;
_wfopen_s(&fp, RESULTS_FILE, L"rb");
if (fp)
{
std::string contents;
std::fseek(fp, 0, SEEK_END);
contents.resize(std::ftell(fp));
std::rewind(fp);
std::fread(&contents[0], 1, contents.size(), fp);
std::fclose(fp);
CString temp1 = (CString)(CStringA)(contents.c_str());
wprintf(L"Result:\n%s\n", temp1.GetBuffer());
if (temp1 == L"") temp1 = L"Unknown command";
strResult = temp1;
}
/*else
{
DWORD lastError = GetLastError();
if (lastError == 32)
{
// File is locked since it is being prepared
return strResult;
}
strResult.Format(L"Can't open file %s. Error %d", RESULTS_FILE, lastError);
return strResult;
}*/
return strResult;
}1234567891011121314151617181920212223242526272829303132复制代码类型:[cpp]
进一步增强
由于应给予笔者EASY-SIZE,马克Richarme。每次调整对话框大小时,都可以轻松使用EASY-SIZE来支持调整对话框中控件的大小。
结果,该对话框最初看起来可能是这样的:
然后调整大小,如下所示:
字体支持
我希望文字清晰且足够大。我选择了140点快递。
首先,我们定义m_font为成员变量:
CFont m_Font;1复制代码类型:[cpp]
然后,作为的一部分OnInitDialog,我们调用:
m_Font.CreatePointFont(140, _T("Courier"));1复制代码类型:[cpp]
然后,要将字体分配给特定控件(在我们的示例中为Command和Command结果,我们使用以下代码:
CommandTyped.SetFont(&m_Font);1复制代码类型:[cpp]
文字颜色
要在黑色文本上设置黄色,我们使用SetTextColor()和SetBkColor()。我们将以下调用添加到OnCtrColor():
case CTLCOLOR_EDIT:
if(ID == IDC_CMD_RESULT || ID == IDC_COMMAND)
{
pDC->SetTextColor(COLOR_YELLOW);
pDC->SetBkColor(COLOR_BLACK);
return (HBRUSH)(m_brush->GetSafeHandle());
}
break;12345678复制代码类型:[cpp]
始终最好分别定义常量,并且在大多数情况下,无需再次输入它们的值即可再次使用它们。在这种情况下,我定义了字体名称以及头文件中的黄色和黑色。
#define FONT_NAME _T("Courier")
#define COLOR_BLACK RGB(0, 0, 0)
#define COLOR_YELLOW RGB(204, 204, 0)