轶哥

📚 Having fun with AI Agent. Always learning.

在Windows平台上使用C++执行外部命令的两种方法
  •   更新:2023-05-27 00:20:32
  •   首发:2023-05-27 00:20:32
  •   教程
  •   1221

在本文中,我将向大家介绍如何在Windows平台上使用C++执行外部命令。我们将探讨两种不同的方法,并对它们进行比较和描述。当我们需要在程序中集成其他应用程序或运行脚本时,这两种方法都非常有用。

在详细讲解这两种方法之前,让我们先了解为什么我们需要在C++程序中执行外部命令。有时,我们需要与其他进程进行交互,例如运行一个脚本、启动一个新进程或收集系统信息。在这些情况下,执行外部命令可以帮助我们轻松地完成这些任务。

现在让我们开始深入了解这两种方法。

方法 1:使用_popen和_pclose函数

// 代码片段1(Method 1)
#include <array>
#include <cstdio>
#include <fstream>
#include <iostream>

std::string exec(const char *cmd) {
  std::array<char, 128> buffer;
  std::string result;
  FILE *pipe = _popen(cmd, "r");

  if (!pipe) {
    throw std::runtime_error("_popen() failed!");
  }

  while (fgets(buffer.data(), buffer.size(), pipe) != nullptr) {
    result += buffer.data();
  }

  int exitStatus = _pclose(pipe);
  if (exitStatus != 0) {
    throw std::runtime_error("Command exited with non-zero status.");
  }

  return result;
}

int main() {
  try {
    // Replace "command" with your command
    std::string aniCommand = "command";
    std::string output = exec(aniCommand.c_str());

    // Save output to file
    std::ofstream outputFile("output.txt");
    if (outputFile.is_open()) {
      outputFile << output;
      outputFile.close();
      std::cout << "Output saved to file." << std::endl;
    } else {
      std::cout << "Error opening file." << std::endl;
    }
  } catch (const std::runtime_error &e) {
    std::cerr << "Error: " << e.what() << std::endl;
  }

  return 0;
}

该方法主要利用了Microsoft提供的_popen_pclose扩展函数。该方法相对简单且易于理解。首先,我们使用_popen创建一个管道并与子进程关联;接着,我们读取子进程通过管道发送的输出。然后,我们关闭管道并检查子进程的退出状态。最后,我们将从子进程获取的输出保存到文件中。

优点

  1. 使用_popen和_pclose函数执行外部命令,较简单易懂。
  2. 将命令输出保存到文件中。

缺点

  1. 可能会遇到兼容性问题,因为_popen和_pclose是Microsoft扩展函数,可能不同编译器的支持程度有所差异。

方法 2:使用Windows API

// 代码片段2(Method 2)
#include <iostream>
#include <stdexcept>
#include <string>
#include <vector>
#include <windows.h>

std::string exec(const char *cmd) {
  SECURITY_ATTRIBUTES sa;
  sa.nLength = sizeof(sa);
  sa.lpSecurityDescriptor = NULL;
  sa.bInheritHandle = TRUE;

  HANDLE hRead, hWrite;

  if (!CreatePipe(&hRead, &hWrite, &sa, 0)) {
    throw std::runtime_error("CreatePipe failed");
  }

  STARTUPINFO si;
  ZeroMemory(&si, sizeof(si));
  si.cb = sizeof(si);
  si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
  si.hStdOutput = hWrite;
  si.hStdError = hWrite;

  PROCESS_INFORMATION pi;
  ZeroMemory(&pi, sizeof(pi));

  if (!CreateProcess(NULL, (LPSTR)cmd, NULL, NULL, TRUE, 0, NULL, NULL, &si,
                     &pi)) {
    CloseHandle(hWrite);
    CloseHandle(hRead);
    throw std::runtime_error("CreateProcess failed");
  }

  CloseHandle(hWrite);

  std::vector<char> buffer(4096);
  DWORD bytesRead;
  std::string output;

  while (true) {
    if (!ReadFile(hRead, buffer.data(), buffer.size(), &bytesRead, NULL)) {
      break;
    }
    output.append(buffer.cbegin(), buffer.cbegin() + bytesRead);
  }

  CloseHandle(hRead);
  CloseHandle(pi.hProcess);
  CloseHandle(pi.hThread);

  return output;
}

int main() {
  // Ensure your command generates ANSI color codes
  std::string output = exec("d:\\Project\\cpp-practice\\build\\cpp-practice."
                            "exe --gtest_color=yes 2>&1");
  std::cout << output;
  return 0;
}

与方法 1 相比,该方法提供了更多底层控制。这里,我们使用Windows API(CreatePipeCreateProcessReadFile)来执行外部命令。首先,我们创建一个匿名管道;接着,我们创建一个新进程并将管道与其关联。然后,我们读取子进程通过管道发送的输出。最后,关闭管道以及相关进程和线程句柄。

优点

  1. 使用Windows API(CreatePipe、CreateProcess和ReadFile)来执行外部命令,更通用且适配多种编译器。
  2. 可以获取子进程stdout和stderr的输出。

缺点

  1. 代码相对复杂,需要处理更多底层Windows API细节。
  2. 没有将命令输出保存到文件中。

结论

在选择使用哪种方法时,应根据具体需求和目标进行权衡。如果你只需执行简单外部命令并保存输出结果到文件,那么方法 1 可能更适合你。然而,如果你需要控制更多底层细节或确保与不同编译器的兼容性,方法 2 可能是更好的选择。

两种方法均有其优点和局限性;最佳策略取决于你的具体需求和项目情况。希望本文能为你在Windows平台上使用C++执行外部命令提供帮助。在未来的编程实践中,你可以灵活地运用这些方法以达到预期目标。

打赏
交流区

暂无内容

尚未登陆
发布
  上一篇 (在 Windows 中迁移 SSH 密钥)
下一篇 (C++ LLVM生成测试覆盖率)  

评论回复提醒