C++系统相关操作1
cnblogs 2024-06-22 08:09:09 阅读 72
- 1. 关键词
- 2. sysutil.h
- 3. sysutil.cpp
- 3.1. system_util_unix.cpp
- 3.2. system_util_win.cpp
- 4. 测试代码
- 5. 运行结果
- 6. 源码地址
1. 关键词
关键词:
C++ 系统调用 system popen 跨平台
应用场景:
希望直接调用操作系统的某些命令,并获取命令的返回值。
2. sysutil.h
#pragma once
#include <cstdint>
#include <string>
namespace cutl
{
/**
* @brief Execute a system command.
*
* @param cmd the command to be executed.
* @return true if the command is executed successfully, false otherwise.
*/
bool system(const std::string &cmd);
/**
* @brief Execute a system command and get the output.
*
* @param cmd the command to be executed.
* @param result the output of the command.
* @return true if the command is executed successfully, false otherwise.
*/
bool callcmd(const std::string &cmd, std::string &result);
} // namespace cutl
不需要返回值时,可以直接使用system, 需要获取返回值时,可以调用callcmd。
3. sysutil.cpp
#include <map>
#include <iostream>
#include <strutil.h>
#include <cstdlib>
#include "sysutil.h"
#include "inner/logger.h"
#include "inner/system_util.h"
#include "inner/filesystem.h"
namespace cutl
{
bool system(const std::string &cmd)
{
return call_system(cmd);
}
bool callcmd(const std::string &cmd, std::string &result)
{
// 读取命令执行结果的最大Buffer长度
constexpr int MAX_CMD_BUF_LEN = 1024;
FILE *fp = pipline_open(cmd);
if (fp == NULL)
{
CUTL_ERROR("pipline_open error for cmd:" + cmd);
return false;
}
// 读取命令执行结果
char buffer[MAX_CMD_BUF_LEN] = {0};
char *res = fgets(buffer, sizeof(buffer), fp);
if (res == NULL)
{
CUTL_ERROR("read result error for cmd:" + cmd);
if (pipline_close(fp) != 0)
{
CUTL_ERROR("pipline_close error for cmd:" + cmd);
}
return false;
}
if (pipline_close(fp) != 0)
{
CUTL_ERROR("pipline_close error for cmd:" + cmd);
}
result = strip(std::string(buffer));
return true;
}
} // namespace cutl
3.1. system_util_unix.cpp
#if defined(_WIN32) || defined(__WIN32__)
// do nothing
#else
#include "system_util.h"
#include "inner/logger.h"
namespace cutl
{
bool call_system(const std::string &cmd)
{
if (cmd.empty())
{
CUTL_ERROR("cmd is empty!");
return false;
}
pid_t status;
status = std::system(cmd.c_str());
if (-1 == status)
{
CUTL_ERROR("system error!");
return false;
}
if (!WIFEXITED(status))
{
CUTL_ERROR("exit status:" + std::to_string(WEXITSTATUS(status)));
return false;
}
if (0 != WEXITSTATUS(status))
{
CUTL_ERROR("run shell script fail, script exit code:" + std::to_string(WEXITSTATUS(status)));
return false;
}
return true;
}
FILE *pipline_open(const std::string &cmd)
{
return popen(cmd.c_str(), "r");
}
int pipline_close(FILE *stream)
{
return pclose(stream);
}
} // namespace cutl
#endif // defined(_WIN32) || defined(__WIN32__)
3.2. system_util_win.cpp
#if defined(_WIN32) || defined(__WIN32__)
#include <cstdlib>
#include "system_util.h"
#include "inner/logger.h"
namespace cutl
{
// https://learn.microsoft.com/zh-cn/cpp/c-runtime-library/reference/system-wsystem?view=msvc-170
bool call_system(const std::string &cmd)
{
if (cmd.empty())
{
CUTL_ERROR("cmd is empty!");
return false;
}
int ret = system(cmd.c_str());
if (ret != 0)
{
CUTL_ERROR(std::string("system failure, error") + strerror(errno));
return false;
}
return true;
}
// https://learn.microsoft.com/zh-cn/cpp/c-runtime-library/reference/popen-wpopen?view=msvc-170
FILE *pipline_open(const std::string &cmd)
{
return _popen(cmd.c_str(), "r");
}
// https://learn.microsoft.com/zh-cn/cpp/c-runtime-library/reference/pclose?view=msvc-170
int pipline_close(FILE *stream)
{
return _pclose(stream);
}
} // namespace cutl
#endif // defined(_WIN32) || defined(__WIN32__)
4. 测试代码
#include "common.hpp"
#include "sysutil.h"
void TestSystemCall()
{
PrintSubTitle("TestSystemCall");
bool ret = cutl::system("echo hello");
std::cout << "system call 'echo hello', return: " << ret << std::endl;
auto cmd = "cmake --version";
std::string result_text;
ret = cutl::callcmd(cmd, result_text);
std::cout << "callcmd " << cmd << ", return: " << ret << std::endl;
std::cout << "callcmd " << cmd << ", result text: " << result_text << std::endl;
}
5. 运行结果
-------------------------------------------TestSystemCall-------------------------------------------
hello
system call 'echo hello', return: 1
callcmd cmake --version, return: 1
callcmd cmake --version, result text: cmake version 3.28.3
6. 源码地址
更多详细代码,请查看本人写的C++ 通用工具库: common_util, 本项目已开源,代码简洁,且有详细的文档和Demo。
本文由博客一文多发平台 OpenWrite 发布!
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。