Tiny CMD

• 下载源代码 - 67.4 KB

引言

请访问我们的SourceForge 产品页面。

本文旨在演示如何从任何类型的应用程序（MFC、Win32、控制台）运行 CMD 命令，等待结果并使用您自己的用户界面查看它们。

背景

作为我们在Secured Globe, Inc.日常工作的一部分，我们运行预制程序。它们所做的只是以编程方式执行一系列 CMD 命令，同时响应结果。这产生了构建一个通用工具的想法。

使用方法

首先，我们定义 3 个全局变量来存储命令异步处理的状态。

• Command - 您在 CMD 中键入的命令
• CommandResult - 执行完成后您在屏幕上看到的的结果
• IsRunning - 指示命令是否仍在处理中。要测试处理时间较长的命令，请尝试键入“netstat”

CString Command, CommandResult;
BOOL IsRunning = FALSE;

DoRun() 函数

我们实际编写发送到ShellExecute()的string 的位置是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;
}

SetCommand 函数

SetCommand 函数用于通过其自己的线程启动新命令。

void SetCommand(CString command)
{
    Command = command;
    HANDLE hThread = (HANDLE)_beginthread(ThreadFunc, 0, NULL);
}

然后线程函数本身如下所示

void __cdecl ThreadFunc(void*) 
{
    CommandResult = L"";
    if (DoRun(Command.GetBuffer()))
    {
        IsRunning = TRUE;

        while (IsRunning)
        {
            if(CheckCommandExecutionStatus())
            {
                break;
            }
        }
    }
    IsRunning = FALSE;
    _endthreadex(0);
}

CheckCommandResult() 如下所示

bool CheckCommandExecutionStatus()
{
    CommandResult = GetResultFromFile();
    if (CommandResult != L"")
        return true;
    else
        return false;
}

获取结果

命令执行完成后，我们期望在创建和重复使用的 .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;
}

进一步增强

应该感谢EASY-SIZE 的作者Marc Richarme。无论何时调整对话框大小，都可以轻松使用 EASY-SIZE 来支持调整对话框内控件的大小。

结果，对话框最初可能如下所示

然后如果调整大小，则如下所示

字体支持

我希望文本清晰且足够大。我选择了 Courier，140磅。

首先，我们将m_font定义为成员变量

    CFont m_Font;

然后，作为OnInitDialog的一部分，我们调用

    m_Font.CreatePointFont(140, _T("Courier"));

然后，要将此字体分配给特定控件（在我们的例子中，是我们的命令和命令结果），我们使用以下代码

    CommandTyped.SetFont(&m_Font);

文本颜色

要设置清晰的黄黑文本，我们使用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;

最好始终单独定义常量，并且在大多数情况下，重复使用它们而无需键入它们的值。在本例中，我在头文件中定义了字体名称以及黄色和黑色。

#define FONT_NAME        _T("Courier")
#define COLOR_BLACK        RGB(0, 0, 0)
#define COLOR_YELLOW    RGB(204, 204, 0)

键盘快捷键

为了方便操作，并与CMD的使用方式保持一致，支持以下键盘键

• <ENTER> - 执行命令
• <上 / 下 箭头键> - 在之前给出的命令之间切换

错误修复和校对

感谢evlncrn8 发现了内存泄漏并进行了校对…… :)

历史

• 2019年8月1日：初始版本