init
This commit is contained in:
commit
70d467d052
97 files changed
+10386
No files matched your search
@@ -0,0 +1,347 @@
|
||||
#include <iostream>
|
||||
#include <boost/property_tree/ptree.hpp>
|
||||
#include <boost/property_tree/json_parser.hpp>
|
||||
#include "main/HttpTask.hh"
|
||||
#include "main/pmc_mtd.hh"
|
||||
#include "main/subsys_call.hh"
|
||||
|
||||
using namespace qing;
|
||||
namespace qing {
|
||||
|
||||
|
||||
/* -------- HttpTask -------- */
|
||||
|
||||
HttpTask::HttpTask(int port, std::function<void(pmc::net::HttpServer& server)> handler)
|
||||
: port(port), handler(handler) {
|
||||
this->th = std::make_unique<Thread> ();
|
||||
init_thread();
|
||||
}
|
||||
|
||||
bool HttpTask::isRunning() {
|
||||
auto stat = th->check();
|
||||
return !(stat == Fsm::Stat::STOP || stat == Fsm::Stat::SHUT);
|
||||
}
|
||||
|
||||
|
||||
void HttpTask::start() {
|
||||
if (!this->isRunning()) {
|
||||
th->Activate(); /* FIXME: 这里获取了资源,不知道会不会造成内存泄漏 */
|
||||
th->WaitStart(); /* TODO: add timeout 增加超时 */
|
||||
}
|
||||
}
|
||||
|
||||
void HttpTask::stop() {
|
||||
this->server->stop();
|
||||
if (this->isRunning())
|
||||
th->WaitClose();
|
||||
}
|
||||
|
||||
void HttpTask::init_thread() {
|
||||
|
||||
/* 构造通信线程 */
|
||||
this->th = std::make_unique<Thread> ();
|
||||
|
||||
/* stop callback 静止事件 */
|
||||
this->th->set_stop( [](Thread& th) -> void {
|
||||
/* 由线程提供的等待状态机发生改变信号的方法 */
|
||||
th.suspend();
|
||||
});
|
||||
|
||||
|
||||
/* start callback 唤醒事件 */
|
||||
this->th->set_wake( [this](Thread& th) -> void {
|
||||
this->server = std::make_unique<pmc::net::HttpServer> ("127.0.0.1", port, "server.crt", "server.key", 4);
|
||||
handler(*this->server);
|
||||
th.run();
|
||||
|
||||
this->server->start(); /* 这里包含了端口启动提示符输出 */
|
||||
this->server->run(); /* 堵死线程,等待服务器运行结束 */
|
||||
});
|
||||
|
||||
|
||||
/* loop callback 循环事件 */
|
||||
this->th->set_loop( [this](Thread& th) -> void {
|
||||
/* 如果服务器是主动退出的 */
|
||||
th.stop(); /* 停止线程,防止跌入死循环 */
|
||||
});
|
||||
|
||||
|
||||
/* clean callback 清理事件 */
|
||||
this->th->set_clear( [this] (Thread& th) -> void {
|
||||
this->server.reset();;
|
||||
});
|
||||
|
||||
}
|
||||
|
||||
|
||||
/*==============================================================*/
|
||||
/* 这些是尚未整理的工具 */
|
||||
|
||||
/* --------------------------- */
|
||||
/* json字符串转boost.ptree
|
||||
* FIXME: 这个方法好像在ProcessManagerService进程托管服务中集成了*/
|
||||
boost::property_tree::ptree parseJsonToPtree(const std::string& jsonStr) {
|
||||
boost::property_tree::ptree pt;
|
||||
try {
|
||||
std::stringstream ss(jsonStr);
|
||||
boost::property_tree::read_json(ss, pt);
|
||||
} catch (const boost::property_tree::json_parser_error& e) {
|
||||
// 解析失败处理
|
||||
std::cerr << "JSON parse error: " << e.what() << std::endl;
|
||||
}
|
||||
return pt;
|
||||
}
|
||||
|
||||
/* --------------------------- */
|
||||
/* url 解码 */
|
||||
static std::string url_decode(const std::string& str) {
|
||||
std::string result;
|
||||
std::istringstream iss(str);
|
||||
char ch;
|
||||
|
||||
while (iss.get(ch)) {
|
||||
if (ch == '%') {
|
||||
int hex;
|
||||
if (iss >> std::hex >> hex) {
|
||||
result += static_cast<char>(hex);
|
||||
}
|
||||
} else if (ch == '+') {
|
||||
result += ' ';
|
||||
} else {
|
||||
result += ch;
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
/*==============================================================*/
|
||||
|
||||
|
||||
/**
|
||||
* @brief 进程托管服务 - 相关模块
|
||||
*
|
||||
* 提供进程的列表、执行和终止功能
|
||||
* TODO(20260629-qing): 将这个模块合入HttpTask类模块
|
||||
*/
|
||||
|
||||
namespace process_http_server {
|
||||
|
||||
/* ---- 注册POST路由 ---- */
|
||||
void registerRoutes(pmc::net::HttpServer& server) {
|
||||
server.post("/list", [](const auto& req, const auto& params) {
|
||||
return handleList(req, params);
|
||||
});
|
||||
|
||||
server.post("/exec", [](const auto& req, const auto& params) {
|
||||
return handleExec(req, params);
|
||||
});
|
||||
|
||||
server.post("/kill", [](const auto& req, const auto& params) {
|
||||
return handleKill(req, params);
|
||||
});
|
||||
}
|
||||
|
||||
/* ---- 解析json ---- */
|
||||
boost::property_tree::ptree parseJsonBody(const std::string& body) {
|
||||
boost::property_tree::ptree pt;
|
||||
try {
|
||||
std::stringstream ss(body);
|
||||
boost::property_tree::read_json(ss, pt);
|
||||
}
|
||||
catch (...) {
|
||||
/* ---- JSON解析失败,返回空树 ---- */
|
||||
}
|
||||
return pt;
|
||||
}
|
||||
|
||||
std::string buildJsonResponse(bool success, const std::string& message,
|
||||
const boost::property_tree::ptree& data) {
|
||||
boost::property_tree::ptree response;
|
||||
response.put("success", success);
|
||||
response.put("message", message);
|
||||
if (!data.empty()) {
|
||||
response.add_child("data", data);
|
||||
}
|
||||
|
||||
std::stringstream ss;
|
||||
boost::property_tree::write_json(ss, response);
|
||||
return ss.str();
|
||||
}
|
||||
|
||||
std::string getTarget(const std::unordered_map<std::string, std::string>& params) {
|
||||
auto it = params.find("target");
|
||||
if (it != params.end())
|
||||
return url_decode(it->second); // <--- 这里
|
||||
return "";
|
||||
}
|
||||
|
||||
std::string getExecCommand(const std::unordered_map<std::string, std::string>& params) {
|
||||
auto it = params.find("command");
|
||||
if (it != params.end())
|
||||
return url_decode(it->second); // <--- 这里
|
||||
return "";
|
||||
}
|
||||
|
||||
int getPid(const std::unordered_map<std::string, std::string>& params) {
|
||||
auto it = params.find("pid");
|
||||
if (it != params.end()) {
|
||||
std::string decoded = url_decode(it->second); // <--- 这里
|
||||
return std::stoi(decoded);
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
http::response<http::string_body> handleList(
|
||||
const http::request<http::string_body>& req,
|
||||
const std::unordered_map<std::string, std::string>& params) {
|
||||
|
||||
http::response<http::string_body> res;
|
||||
res.version(req.version());
|
||||
res.set(http::field::content_type, "application/json");
|
||||
|
||||
try {
|
||||
std::string target = getTarget(params);
|
||||
|
||||
std::string result = listProcesses(target);
|
||||
boost::property_tree::ptree data = parseJsonToPtree(result);
|
||||
|
||||
res.result(http::status::ok);
|
||||
res.body() = buildJsonResponse(true, "Process list retrieved successfully", data);
|
||||
}
|
||||
|
||||
catch (const std::exception& e) {
|
||||
res.result(http::status::internal_server_error);
|
||||
res.body() = buildJsonResponse(false, std::string("Error: ") + e.what());
|
||||
}
|
||||
|
||||
res.prepare_payload();
|
||||
return res;
|
||||
}
|
||||
|
||||
http::response<http::string_body> handleExec(
|
||||
const http::request<http::string_body>& req,
|
||||
const std::unordered_map<std::string, std::string>& params) {
|
||||
|
||||
http::response<http::string_body> res;
|
||||
res.version(req.version());
|
||||
res.set(http::field::content_type, "application/json");
|
||||
|
||||
try {
|
||||
std::string target = getTarget(params);
|
||||
std::string command = getExecCommand(params);
|
||||
|
||||
if (command.empty()) {
|
||||
// 尝试从POST body中获取command
|
||||
auto body_json = parseJsonBody(req.body());
|
||||
command = body_json.get<std::string>("command", "");
|
||||
}
|
||||
|
||||
if (command.empty()) {
|
||||
res.result(http::status::bad_request);
|
||||
res.body() = buildJsonResponse(false, "Missing required parameter: command");
|
||||
res.prepare_payload();
|
||||
return res;
|
||||
}
|
||||
|
||||
// 实现命令执行逻辑
|
||||
std::string result = execCommand(command, target);
|
||||
|
||||
boost::property_tree::ptree data;
|
||||
data.put("command", command);
|
||||
data.put("output", result);
|
||||
|
||||
res.result(http::status::ok);
|
||||
res.body() = buildJsonResponse(true, "Command executed successfully", data);
|
||||
}
|
||||
|
||||
catch (const std::exception& e) {
|
||||
res.result(http::status::internal_server_error);
|
||||
res.body() = buildJsonResponse(false, std::string("Error: ") + e.what());
|
||||
}
|
||||
|
||||
res.prepare_payload();
|
||||
return res;
|
||||
}
|
||||
|
||||
http::response<http::string_body> handleKill(
|
||||
const http::request<http::string_body>& req,
|
||||
const std::unordered_map<std::string, std::string>& params) {
|
||||
|
||||
http::response<http::string_body> res;
|
||||
res.version(req.version());
|
||||
res.set(http::field::content_type, "application/json");
|
||||
|
||||
try {
|
||||
std::string target = getTarget(params);
|
||||
int pid = getPid(params);
|
||||
|
||||
if (pid <= 0 && !req.body().empty()) {
|
||||
// 尝试从POST body中获取pid
|
||||
auto body_json = parseJsonBody(req.body());
|
||||
pid = body_json.get<int>("pid", -1);
|
||||
}
|
||||
|
||||
if (pid <= 0) {
|
||||
res.result(http::status::bad_request);
|
||||
res.body() = buildJsonResponse(false, "Missing or invalid parameter: pid");
|
||||
res.prepare_payload();
|
||||
return res;
|
||||
}
|
||||
|
||||
bool success = killProcess(pid, target);
|
||||
|
||||
if (success) {
|
||||
boost::property_tree::ptree data;
|
||||
data.put("pid", pid);
|
||||
res.result(http::status::ok);
|
||||
res.body() = buildJsonResponse(true, "Process terminated successfully", data);
|
||||
} else {
|
||||
res.result(http::status::internal_server_error);
|
||||
res.body() = buildJsonResponse(false, "Failed to terminate process");
|
||||
}
|
||||
}
|
||||
|
||||
catch (const std::exception& e) {
|
||||
res.result(http::status::internal_server_error);
|
||||
res.body() = buildJsonResponse(false, std::string("Error: ") + e.what());
|
||||
}
|
||||
|
||||
res.prepare_payload();
|
||||
return res;
|
||||
}
|
||||
|
||||
// 如果target为空,在本机执行
|
||||
// 否则在目标服务器执行
|
||||
std::string listProcesses(const std::string& target)
|
||||
{
|
||||
if (target == "")
|
||||
return pmc_mtd::list();
|
||||
else
|
||||
return subsys_call::send_list_and_recv_pipe(target);
|
||||
}
|
||||
|
||||
std::string execCommand(const std::string& command, const std::string& target)
|
||||
{
|
||||
if (target == "")
|
||||
return pmc_mtd::exec(command) ? "OK" :"ERR";
|
||||
else
|
||||
return subsys_call::send_start_and_recv_pipe(target, command); /* 发送启动命令并且接收回复 */
|
||||
}
|
||||
|
||||
bool killProcess(int pid, const std::string& target)
|
||||
{
|
||||
if (target == "")
|
||||
return pmc_mtd::kill(pid);
|
||||
else
|
||||
return subsys_call::send_kill_and_recv_pipe(target, pid) == "OK";
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,61 @@
|
||||
#include <iostream>
|
||||
#include <stdexcept>
|
||||
#include "main/PPool.hh"
|
||||
|
||||
namespace qing {
|
||||
|
||||
void PPool::crtp(const std::string& cmd) try
|
||||
{
|
||||
auto pt = std::make_shared<ProcessTask>(cmd);
|
||||
pt->start(); /* 直接创建了进程丢进容器 */
|
||||
pool.push_back(pt);
|
||||
}
|
||||
|
||||
/* 如果boost的子进程模块报错,表示子进程创建失败? */
|
||||
catch (boost::system::system_error &exp) {
|
||||
std::cerr << "Failed to execute this command: " << cmd << std::endl;
|
||||
}
|
||||
|
||||
|
||||
void PPool::clr() {
|
||||
auto it = --pool.end(); /* 由最后一个元素开始 */
|
||||
for (; it != pool.begin(); --it) /* 回溯遍历 */
|
||||
if (!(*it)-> isRunning()) { /* 如果任务已经终止 */
|
||||
auto tmp= it; /* 储存当前位置 */
|
||||
it++; /* 返回上一个结点 */
|
||||
pool.erase(tmp); /* 自动释放终止的节点 */
|
||||
}
|
||||
|
||||
/* 循环结束时it == begin(),如果容器为空则end() == begin() */
|
||||
if (pool.size() > 0 && !(*it)->isRunning()) /* */
|
||||
pool.erase(it); /* 如果end() == begin(),此步将会试图删除哨兵 */
|
||||
}
|
||||
|
||||
void PPool::kill(int idx) {
|
||||
|
||||
|
||||
/* 从头部向后计数idx */
|
||||
auto it = pool.begin();
|
||||
int i = 0;
|
||||
for (; i < idx && it != pool.end(); ++i, ++it);
|
||||
|
||||
if (it != pool.end()) { /* 如果找到了目标(因到达idx而停止)(计数器没有指向尾节点) */
|
||||
(*it)->stop(); /* 关闭当前任务 */
|
||||
pool.erase(it); /* 从容器中移除当前任务 */
|
||||
}
|
||||
|
||||
/* 因抵达终点而停下(下标越界) */
|
||||
else {
|
||||
throw std::out_of_range("PPool::kill()");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
ProcessTask& PPool::operator[](const size_t idx) {
|
||||
auto it = pool.begin();
|
||||
for (int i = 0; i < idx; ++it, ++i) /* 遍历idx次 */
|
||||
if (it == pool.end()) /* 如果该节点是尾部哨兵(下标越界) */
|
||||
throw std::out_of_range("PPool::operator[]()");
|
||||
return **it; /* 返回当前节点 */
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,30 @@
|
||||
#include <iostream>
|
||||
#include "main/ProcessTask.hh"
|
||||
|
||||
#ifdef __NEW_VER
|
||||
namespace qing {
|
||||
bool ProcessTask::isRunning() {
|
||||
return p && p->running(exit_code);
|
||||
}
|
||||
int ProcessTask::getExitCode() {
|
||||
return exit_code.value();
|
||||
}
|
||||
void ProcessTask::start() {
|
||||
if (!isRunning()) {
|
||||
auto ccmd = boost::process::v2::shell(cmd);
|
||||
auto exe = ccmd.exe();
|
||||
auto args = ccmd.args();
|
||||
this->p = std::make_unique<boost::process::v2::process>(ctx, exe, args);
|
||||
}
|
||||
}
|
||||
void ProcessTask::stop() {
|
||||
p->terminate(exit_code);
|
||||
if (exit_code) {
|
||||
std::cerr << "杀死进程失败: " << exit_code.message() << std::endl;
|
||||
}
|
||||
else {
|
||||
p.reset();
|
||||
}
|
||||
}
|
||||
}
|
||||
#endif
|
||||
@@ -0,0 +1,69 @@
|
||||
#include <iostream>
|
||||
#include <memory>
|
||||
#include <us/Thread.hpp>
|
||||
#include <us/ITask.hpp>
|
||||
#include "main/Tttask.hh"
|
||||
|
||||
namespace qing {
|
||||
Tttask::Tttask(const size_t msglen, size_t msgcnt, const sf_t callback)
|
||||
: msglen(msglen), msgcnt(msgcnt) {
|
||||
this->callback = std::make_unique<sf_t>(callback); /* FIXME: 使用指针似乎有所不妥 */
|
||||
init_thread();
|
||||
}
|
||||
|
||||
void Tttask::start() {
|
||||
if (!this->isRunning()) {
|
||||
|
||||
th->Activate(); /* FIXME: 这里获取了资源,不知道会不会造成内存泄漏 */
|
||||
th->WaitStart(); /* TODO: add timeout 增加超时 */
|
||||
}
|
||||
}
|
||||
|
||||
bool Tttask::isRunning() {
|
||||
auto stat = th->check();
|
||||
return !(stat == Fsm::Stat::STOP
|
||||
|| stat == Fsm::Stat::SHUT);
|
||||
}
|
||||
|
||||
void Tttask::init_thread() {
|
||||
|
||||
/* 构造通信线程 */
|
||||
this->th = std::make_unique<Thread> ();
|
||||
|
||||
/* stop callback 静止事件 */
|
||||
this->th->set_stop( [](Thread& th) -> void {
|
||||
|
||||
/* 由线程提供的等待状态机发生改变信号的方法 */
|
||||
th.suspend();
|
||||
});
|
||||
|
||||
/* start callback 唤醒事件 */
|
||||
this->th->set_wake( [this](Thread& th) -> void {
|
||||
name = std::to_string(getpid()); /* 以进程名构建消息队列 */
|
||||
mq = std::make_unique<Mq> (name, msglen, msgcnt, Mq::CREATOR);
|
||||
th.run();
|
||||
});
|
||||
|
||||
|
||||
/* loop callback 循环事件 */
|
||||
this->th->set_loop( [this](Thread& th) -> void {
|
||||
|
||||
/* 第一个是接收成功的处理事件,第二个是接收失败的处理事件
|
||||
*
|
||||
* TODO: 作为参数传入延时
|
||||
* TODO: 使用标准库的线程延时*/
|
||||
mq->recv(*callback, []() { usleep(10000); });
|
||||
|
||||
});
|
||||
|
||||
|
||||
/* clean callback 清理事件 */
|
||||
this->th->set_clear( [this] (Thread& th) -> void {
|
||||
|
||||
/* 直接重置指针 */
|
||||
mq.reset();
|
||||
});
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,132 @@
|
||||
|
||||
|
||||
#include <boost/json.hpp>
|
||||
#include <boost/property_tree/ptree.hpp>
|
||||
#include <boost/property_tree/json_parser.hpp>
|
||||
#include "main/PPool.hh"
|
||||
#include "main/pmc_mtd.hh"
|
||||
|
||||
namespace qing {
|
||||
|
||||
/* -------- 用于存放子进程的进程池 -------- */
|
||||
static PPool ppool;
|
||||
|
||||
/*-------------------------------------------------------------------------*/
|
||||
|
||||
/* --------【pmc method子进程操作】集合 -------- */
|
||||
namespace pmc_mtd {
|
||||
|
||||
/*
|
||||
* 列举出所有子进程。
|
||||
*
|
||||
* {
|
||||
* "pipe": "<int>",
|
||||
* "type": "list",
|
||||
* } */
|
||||
auto list() -> std::string {
|
||||
boost::json::array processes;
|
||||
|
||||
/* 遍历子进程池
|
||||
*
|
||||
* FIXME: 子进程池的迭代器完成后,改为使用for-in迭代 */
|
||||
//ppool.clr() /* 保留已经跑完的进程,因为可能会涉及到输出的提取 *?
|
||||
for (int i = 0; i < ppool.size(); ++i) {
|
||||
processes.push_back({ /* 向json中推入对象 */
|
||||
{"index", i},
|
||||
{"pid", ppool[i].pid()},
|
||||
{"status", ppool[i].check()}
|
||||
});
|
||||
}
|
||||
|
||||
/* 返回序列化的json对象 */
|
||||
return boost::json::serialize(processes);
|
||||
}
|
||||
|
||||
/*
|
||||
* 删除一个子进程,需要提供pid
|
||||
*
|
||||
* {
|
||||
* "pipe": "<integer>",
|
||||
* "type": "kill",
|
||||
* "kill": <integer>
|
||||
* } */
|
||||
auto kill(const int pid) -> bool {
|
||||
|
||||
/* 遍历子进程池对比pid */
|
||||
for (int i=0; i<ppool.size(); ++i)
|
||||
if ( ppool[i].pid() == pid) { /* 寻找目标进程 */
|
||||
ppool.kill(i); /* 杀死该进程 */
|
||||
return true; } /* 成功返回逻辑真 */
|
||||
|
||||
/* 失败(没找到)返回逻辑假 */
|
||||
return false;
|
||||
}
|
||||
|
||||
/*
|
||||
* 启动一个子进程
|
||||
*
|
||||
* {
|
||||
* "pipe": "<integer>",
|
||||
* "type": "exec",
|
||||
* "exec": "<commandline>"
|
||||
* } */
|
||||
auto exec(const std::string& cmd) -> bool try
|
||||
{
|
||||
//ppool.clr(); /* 保留跑完的子进程 */
|
||||
ppool.crtp(cmd);
|
||||
return true; /* 成功返回逻辑真 */
|
||||
}
|
||||
|
||||
/* 如果出现任何异常,返回假 */
|
||||
catch(std::exception& exp) {
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/* 解析自启动列表、添加自启动项
|
||||
*
|
||||
* 如果在解析过程中出错,只是打印异常? */
|
||||
void parse_self_init_list(std::string &path)
|
||||
try {
|
||||
|
||||
std::ifstream file(path);
|
||||
if (!file.is_open()) { /* ---- 没能打开自启动文件 ---- */
|
||||
std::cerr << "Failed to open self-init list" << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
std::string row = "";
|
||||
char ch;
|
||||
while (file.get(ch))
|
||||
|
||||
if (ch == '\n') {
|
||||
|
||||
auto pos = row.find("#");
|
||||
if (pos != -1)
|
||||
row = row.substr(0, pos);
|
||||
if (row != "") { /* trim() */
|
||||
ppool.clr();
|
||||
ppool.crtp(row); /* 创建进程 */
|
||||
row = "";
|
||||
}
|
||||
}
|
||||
|
||||
else row += ch;
|
||||
|
||||
file.close();
|
||||
|
||||
}
|
||||
catch (std::exception &exp) {
|
||||
std::cerr << "Exception at: parse_self_init_list()" << std::endl;
|
||||
std::cerr << "Exception type: " << typeid(exp).name() << std::endl;
|
||||
std::cerr << "Exception info: " << exp.what() << std::endl;
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
@@ -0,0 +1,163 @@
|
||||
|
||||
#include "iostream"
|
||||
#include "fstream"
|
||||
#include <boost/json.hpp>
|
||||
#include "tt/Mq.h"
|
||||
#include "tt/Pipe.h"
|
||||
#include "main/subsys_call.hh"
|
||||
|
||||
namespace qing {
|
||||
namespace subsys_call {
|
||||
|
||||
|
||||
/*-------------------------------------------------------------------------*/
|
||||
|
||||
|
||||
/* -------- 向pmc的消息队列发送启动命令并且从管道接收回复 -------- */
|
||||
std::string send_start_and_recv_pipe(const std::string& name_mq, const std::string& exec)
|
||||
try {
|
||||
|
||||
auto name_pipe = std::to_string(getpid());
|
||||
auto mq = Mq(name_mq, 2048, 100, Mq::USER); /* 放在这里避免管道文件的残留 */
|
||||
|
||||
boost::json::object obj;
|
||||
obj["pipe"] = name_pipe;
|
||||
obj["type"] = "exec";
|
||||
obj["exec"] = exec;
|
||||
auto jsonstr = boost::json::serialize(obj);
|
||||
|
||||
|
||||
auto pipe = Pipe(name_pipe, Pipe::CREATOR); /* FIXME: 创建管道文件到/tmp下面 */
|
||||
if (!pipe.openForRead(true)) {
|
||||
std::cerr << "Failed to open pipe" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
mq.send(jsonstr);
|
||||
if (!pipe.waitForRead(1, 0)) {
|
||||
std::cerr << "Pipe Read Timeout" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
std::string res = "";
|
||||
if (pipe.readPipe(&res) < 0) {
|
||||
std::cerr << "Failed to read pipe" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
return res;
|
||||
}
|
||||
|
||||
//catch (boost::interprocess::interprocess_exception &exp) {
|
||||
// std::cerr << "Target Pmc Is Not Found" << std::endl;
|
||||
// return 1;
|
||||
//}
|
||||
|
||||
catch (std::exception &exp) {
|
||||
std::cerr << "Exception at: send_start_and_recv_pipe()" << std::endl;
|
||||
std::cerr << "Exception type: " << typeid(exp).name() << std::endl;
|
||||
std::cerr << "Exception info: " << exp.what() << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
|
||||
|
||||
/* -------- 向目标子系统发送列举命令并收集结果 -------- */
|
||||
std::string send_list_and_recv_pipe(const std::string& name_mq)
|
||||
try {
|
||||
|
||||
auto name_pipe = std::to_string(getpid());
|
||||
auto mq = Mq(name_mq, 2048, 100, Mq::USER);
|
||||
|
||||
boost::json::object obj;
|
||||
obj["pipe"] = name_pipe;
|
||||
obj["type"] = "list";
|
||||
auto jsonstr = boost::json::serialize(obj);
|
||||
auto pipe = Pipe(name_pipe, Pipe::CREATOR);
|
||||
|
||||
if (!pipe.openForRead(true)) {
|
||||
std::cerr << "Failed to open pipe" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
|
||||
mq.send(jsonstr);
|
||||
if (!pipe.waitForRead(1, 0)) {
|
||||
std::cerr << "Pipe Read Timeout" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
std::string res = "";
|
||||
if (pipe.readPipe(&res) < 0) {
|
||||
std::cerr << "Failed to read pipe" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
|
||||
return res;
|
||||
|
||||
}
|
||||
|
||||
//catch (boost::interprocess::interprocess_exception &exp) {
|
||||
// std::cerr << "Target Pmc Is Not Found" << std::endl;
|
||||
// return 1;
|
||||
//}
|
||||
|
||||
catch (std::exception &exp) {
|
||||
std::cerr << "Exception at: send_list_and_recv_pipe()" << std::endl;
|
||||
std::cerr << "Exception type: " << typeid(exp).name() << std::endl;
|
||||
std::cerr << "Exception info: " << exp.what() << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
/* -------- 删除操作调用 -------- */
|
||||
std::string send_kill_and_recv_pipe(const std::string& name_mq, const int kill)
|
||||
try{
|
||||
|
||||
auto name_pipe = std::to_string(getpid());
|
||||
auto mq = Mq(name_mq, 2048, 100, Mq::USER);
|
||||
|
||||
boost::json::object obj;
|
||||
obj["pipe"] = name_pipe;
|
||||
obj["type"] = "kill";
|
||||
obj["kill"] = kill;
|
||||
auto jsonstr = boost::json::serialize(obj);
|
||||
|
||||
auto pipe = Pipe(name_pipe, Pipe::CREATOR);
|
||||
if (!pipe.openForRead(true)) {
|
||||
std::cerr << "Failed to open pipe" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
mq.send(jsonstr);
|
||||
if (!pipe.waitForRead(1, 0)) {
|
||||
std::cerr << "Pipe Read Timeout" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
std::string res = "";
|
||||
if (pipe.readPipe(&res) < 0) {
|
||||
std::cerr << "Failed to read pipe" << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
else return res;
|
||||
|
||||
}
|
||||
|
||||
/* 没有打开目标子系统的消息队列时,会触发这个异常
|
||||
* 但是实际上这里捕获的是进程间通信通用异常(旧版) */
|
||||
//catch (boost::interprocess::interprocess_exception &exp) {
|
||||
// std::cerr << "Target Pmc Is Not Found" << std::endl;
|
||||
// return 1;
|
||||
//}
|
||||
|
||||
/* 这些函数抛出异常后可能导致管道文件残留 */
|
||||
catch (std::exception &exp) {
|
||||
std::cerr << "Exception at: send_kill_and_recv_pipe()" << std::endl;
|
||||
std::cerr << "Exception type: " << typeid(exp).name() << std::endl;
|
||||
std::cerr << "Exception info: " << exp.what() << std::endl;
|
||||
return "ERR";
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user