init
This commit is contained in:
qing committed
commit
70d467d052
97 files changed
+10386
No files matched your search
@@ -0,0 +1,306 @@
|
||||
/* ===============================================
|
||||
* 通用子系统、进程机器、进程托管服务
|
||||
*
|
||||
*
|
||||
* 【主函数文件】
|
||||
*
|
||||
* 直接子系统调用，或者启动一个pmc子系统（可以打开http）。
|
||||
* ===============================================
|
||||
*/
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include <fstream>
|
||||
#include <sstream>
|
||||
#include <thread>
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
#include <list>
|
||||
#include <cstddef>
|
||||
#include <csignal>
|
||||
#include <cctype>
|
||||
#include <iomanip>
|
||||
#include <unistd.h>
|
||||
#include <sys/types.h>
|
||||
#include <boost/json.hpp>
|
||||
#include <boost/program_options.hpp>
|
||||
#include <boost/property_tree/ptree.hpp>
|
||||
|
||||
#include "logs/Logger.hpp"
|
||||
#include <us/Thread.hpp>
|
||||
#include <us/Cv_wait.hpp>
|
||||
#include <us/ITask.hpp>
|
||||
#include "tt/Pipe.h"
|
||||
#include "main/Tttask.hh"
|
||||
#include "main/HttpTask.hh"
|
||||
#include "main/pmc_mtd.hh"
|
||||
#include "main/subsys_call.hh"
|
||||
|
||||
/* 中文模式 Chinese Mode */
|
||||
#include "cn/中文化.hpp"
|
||||
|
||||
/* 静态区 - 存放全局变量 */
|
||||
static const char *PMC_VERSION = "0.0.10"; /* 版本号 */
|
||||
static const size_t MSGLEN = 2048; /* 单条消息的长度 */
|
||||
static const size_t MSGCNT = 100; /* 消息条数 */
|
||||
|
||||
/* ---- 命名空间 ---- */
|
||||
using namespace qing;
|
||||
|
||||
/* ---- 给命令行参数的命名空间取别名 ---- */
|
||||
namespace po = boost::program_options;
|
||||
|
||||
|
||||
void pmc_init(po::variables_map&); /* 初始化pmc并行机器 */
|
||||
void pmc_serve(po::variables_map&); /* pmc的http伺服器模式 */
|
||||
void parse_data_from_mq(std::string&); /* 解析消息队列收到的消息 */
|
||||
|
||||
//-------------------------------------------------------------
|
||||
//-------------------------------------------------------------
|
||||
|
||||
|
||||
/*-------------------------------------------------------------------------*/
|
||||
/*-------------------------------------------------------------------------*/
|
||||
Cv_wait cv = Cv_wait(); /* 条件变量的等待机制（让主程序等待中断信号） */
|
||||
|
||||
auto taskPool = std::vector<std::shared_ptr<ITask>>{}; /* 基层任务池 - !!! 非子进程 !!! */
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
/*-------------------------------------------------------------------------*/
|
||||
|
||||
|
||||
/* 捕获中断信号，结束线程并且退出 */
|
||||
void signalHandler(int signum)
|
||||
{
|
||||
LOG_INFO("收到终止信号，准备停止服务器...");
|
||||
cv.WakeCv(); /* 唤醒条件信号即关闭程序 */
|
||||
//if (server) server->stop(); /* TODO: 封装成ITask */
|
||||
}
|
||||
|
||||
/*---------------- SIGNAL ACTION ----------------*/
|
||||
//
|
||||
/* 对信号的响应行为 TODO: 我应该系统地学习一下这个库 */
|
||||
void set_signal_action() {
|
||||
|
||||
/* 该结构体用于描述信号的处理方式 */
|
||||
struct sigaction sa;
|
||||
|
||||
|
||||
/* 绑定处理函数 */
|
||||
sa.sa_handler = signalHandler;
|
||||
|
||||
/*
|
||||
* 清空并初始化一个信号掩码集 --------------> 用于指定哪些信号在当前或即将执行的信号处理函数期间应该被阻塞（位掩码）
|
||||
* “在执行这个信号处理函数时，除了当前正在处理的信号外，还希望临时阻塞其他某些信号”
|
||||
* 不阻塞其他信号
|
||||
* 确保在执行期间没有其他特定的信号会被阻塞
|
||||
* 防止处理过程中被其他信号中断
|
||||
*/
|
||||
sigemptyset(&sa.sa_mask);
|
||||
|
||||
/*
|
||||
* 用于设置信号处理函数的行为特性。
|
||||
* 0 - 默认选项
|
||||
* SA_NODEFER - 表示在执行信号处理函数时不允许其他信号被阻塞
|
||||
* SA_RESETHAND - 在信号处理函数返回后，将信号的处理方式重置为默认
|
||||
* SA_NOCLDSTOP - 对于子进程发送的停止信号，父进程不会停止。
|
||||
* SA_NOCLDWAIT - 父进程在等待等待子进程终止时不会因为子进程发送信号而收到通知
|
||||
* SA_SIGINFO - 使用扩展的信号信息，信号处理函数接收三个参数而不是一个
|
||||
*/
|
||||
sa.sa_flags = 0;
|
||||
|
||||
/* 注册信号 */
|
||||
sigaction(SIGINT, &sa, nullptr); /* Ctrl + C */
|
||||
sigaction(SIGTERM, &sa, nullptr); /* kill 或 systemctl stop */
|
||||
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------*/
|
||||
/*-------------------------------------------------------------------------*/
|
||||
/* 入口函数 */
|
||||
int main(int argc, char** argv) {
|
||||
|
||||
/*------------------------------*/
|
||||
/* 设置信号行为*/
|
||||
set_signal_action();
|
||||
|
||||
/*------------------------------*/
|
||||
/* 解析命令行参数 */
|
||||
po::options_description desc("pmc subsystem start options");
|
||||
desc.add_options() /* 定义选项 */
|
||||
("help,h", "display help message") /* 帮助菜单 */
|
||||
("version,v", "display version message") /* 显示版本 */
|
||||
/******** 子系统调用模式 *********/
|
||||
("send,s", po::value<std::string>(), "send a message to pmc sys") /* 输入目标pid */
|
||||
("exec", po::value<std::string>(), "message body") /* 启动一个进程 */
|
||||
("list", "list process all run") /* 列出运行中的进程 */
|
||||
("kill", po::value<int>(), "kill a process") /* 杀死一个进程 */
|
||||
/********* 子系统服务模式 ********/
|
||||
("init,i", po::value<std::string>(), "self-init list") /* 输入自启动列表 */
|
||||
("serve", po::value<int>(), "Run as http server"); /* 启用http伺服器 */
|
||||
|
||||
/* 参数变量映射关系 */
|
||||
po::variables_map vm;
|
||||
|
||||
try { /* 开始解析 */
|
||||
po::store(po::parse_command_line(argc, argv, desc), vm);
|
||||
po::notify(vm);
|
||||
if (vm.count("help")) { /* 获取帮助页面 */
|
||||
std::cout << desc << std::endl;
|
||||
std::cout << "--------- API MAP ---------" << std::endl;
|
||||
std::cout << "Available endpoints:" << std::endl;
|
||||
std::cout << " POST /list?target=optional" << std::endl;
|
||||
std::cout << " POST /exec?command=ls&target=optional" << std::endl;
|
||||
std::cout << " POST /kill?pid=1234&target=optional" << std::endl;
|
||||
return 0;
|
||||
}
|
||||
if (vm.count("version")) { /* 获取版本信息 */
|
||||
std::cout << PMC_VERSION << std::endl;
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
catch (const std::exception& e) {
|
||||
std::cerr << "ERROR: " << e.what() << std::endl;
|
||||
std::cerr << "using --help to check options message" << std::endl;
|
||||
throw e;
|
||||
}
|
||||
|
||||
/*----------------------------------------*/
|
||||
|
||||
if ( vm.count("send") && vm.count("exec")) {
|
||||
auto send = vm["send"].as<std::string>();
|
||||
auto exec = vm["exec"].as<std::string> ();
|
||||
auto msg = subsys_call::send_start_and_recv_pipe(send, exec);
|
||||
std::cout << msg << std::endl;
|
||||
return 0;
|
||||
}
|
||||
|
||||
else if ( vm.count("send") && vm.count("list")){
|
||||
auto send = vm["send"].as<std::string> ();
|
||||
auto msg = subsys_call::send_list_and_recv_pipe(send);
|
||||
std::cout << msg << std::endl;
|
||||
return 0;
|
||||
}
|
||||
|
||||
else if (vm.count("send") && vm.count("kill")) {
|
||||
auto send = vm["send"].as<std::string> ();
|
||||
auto kill = vm["kill"].as<int>();
|
||||
auto msg = subsys_call::send_kill_and_recv_pipe(send, kill);
|
||||
std::cout << msg << std::endl;
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* 初始化日志模块 */
|
||||
Logger::getInstance().init(LogLevel::DEBUG, true);
|
||||
|
||||
|
||||
/* 列表自启动 */
|
||||
if (vm.count("init")){
|
||||
std::string path = vm["init"].as<std::string>();
|
||||
pmc_mtd::parse_self_init_list(path);
|
||||
}
|
||||
|
||||
|
||||
/**********************************************/
|
||||
|
||||
/* TODO: 抽象出来 */
|
||||
if (vm.count("serve"))
|
||||
pmc_serve(vm);
|
||||
|
||||
|
||||
/* pmc初始化函数 */
|
||||
pmc_init(vm);
|
||||
|
||||
/*
|
||||
* 主线程等待
|
||||
*
|
||||
* 触发器在信号处理那里
|
||||
*/
|
||||
cv.Wait();
|
||||
|
||||
/* 主线程被唤醒，停止所有任务 */
|
||||
for (auto& task: taskPool) {
|
||||
task->stop();
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
/*-----------------------------------------------------------------------------*/
|
||||
/* 初始化子系统 即是启动通信 */
|
||||
void pmc_init(po::variables_map& vm) {
|
||||
|
||||
/*-------------------*/
|
||||
auto ttt = std::make_shared<Tttask> (MSGLEN, MSGCNT,
|
||||
[](const char *data) { /* 处理收到的数据 */
|
||||
auto json_str = std::string(data);
|
||||
parse_data_from_mq(json_str);
|
||||
});
|
||||
ttt->start();
|
||||
taskPool.push_back(ttt);
|
||||
}
|
||||
|
||||
/* http伺服器模式（只监听本地）TODO: 集成为ITask */
|
||||
void pmc_serve(po::variables_map& vm)
|
||||
{
|
||||
int port = vm["serve"].as<int>();
|
||||
auto server = std::make_shared<HttpTask> (port, [](pmc::net::HttpServer& server_) {
|
||||
process_http_server::registerRoutes(server_);
|
||||
});
|
||||
server->start();
|
||||
taskPool.push_back(server);
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
/* 解析从消息队列中接收到的数据，运行在服务端 */
|
||||
void parse_data_from_mq(std::string& json_str)
|
||||
try {
|
||||
boost::json::value jv = boost::json::parse(json_str);
|
||||
std::string namepipe = jv.at("pipe").as_string().c_str();
|
||||
std::string typ = jv.at("type").as_string().c_str();
|
||||
auto pipe = Pipe(namepipe, Pipe::USER);
|
||||
pipe.openForWrite();
|
||||
|
||||
|
||||
if (typ == "list") {
|
||||
auto msg = pmc_mtd::list();
|
||||
pipe.writePipe(msg);
|
||||
return;
|
||||
}
|
||||
|
||||
else if (typ == "kill") {
|
||||
int trg = jv.at("kill").as_int64();
|
||||
auto flag = pmc_mtd::kill(trg);
|
||||
std::string msg = flag ? "OK" : "Nope";
|
||||
pipe.writePipe(msg);
|
||||
return;
|
||||
}
|
||||
|
||||
else if (typ == "exec"){
|
||||
std::string cmd = jv.at("exec").as_string().c_str();
|
||||
std::string msg = (pmc_mtd::exec(cmd)) ? "OK" : "ERR";
|
||||
pipe.writePipe(msg);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
/* 捕获异常后，仅仅只是打印出来？ */
|
||||
catch (std::exception &exp) {
|
||||
std::cerr << "Exception at: parse_data_from_mq()" << std::endl;
|
||||
std::cerr << "Exception type: " << typeid(exp).name() << std::endl;
|
||||
std::cerr << "Exception info: " << exp.what() << std::endl;
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -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