Implementace třídy, která spouští podřízený proces s časovým limitem pomocí procesu boost v asynchronním režimu

V následujícím kódu se snažím implementovat program, který spouští zpracuje a vrátí jeho návratový kód stdout a stderr. Má také koncept časového limitu, po kterém je podřízený proces zabit. Tato třída bude použita jako součást vícevláknové aplikace.

#ifndef PROCESS_H_ #define PROCESS_H_ //Header #include <chrono> #include <thread> #include <iostream> #include <boost/process.hpp> #include <boost/asio.hpp namespace bp = boost::process; class Proces { public: Process(const std::string& cmd, const int timeout); void run(); int getReturnStatus(); //Must be only called after run() exits std::string getStdOut(); //Must be only called after run() exits std::string getStdErr(); //Must be only called after run() exits bool wasKilled(); //Must be only called after run() exits private: void initLog(); void timeoutHandler(const boost::system::error_code& ec); void kill(); const std::string command; const int timeout; int returnStatus; std::string stdOut; std::string stdErr; bool killed; bool stopped; boost::process::group group; boost::asio::io_context ioc; boost::asio::deadline_timer deadline_timer; }; #endif //Source #include "Process.h" Process::Process(const std::string& cmd, const int timeout): command(cmd), timeout(timeout), returnStatus(0), stdOut(""), stdErr(""), killed(false), stopped(false), ioc(), deadline_timer(ioc) { } void Process::timeoutHandler(const boost::system::error_code& ec) { if (stopped || ec == boost::asio::error::operation_aborted) { return; } std::cout << "Time Up!"<< std::endl; kill(); deadline_timer.expires_at(boost::posix_time::pos_infin); } void Process::run() { std::future<std::string> dataOut; std::future<std::string> dataErr; std::cout << "Running command: " << command << std::endl; bp::child c(command, bp::std_in.close(), bp::std_out > dataOut, bp::std_err > dataErr, ioc, group, bp::on_exit([=](int e, const std::error_code& ec) { std::cout << "on_exit: " << ec.message() << " -> " << e << std::endl; deadline_timer.cancel(); returnStatus = e; })); deadline_timer.expires_from_now(boost::posix_time::seconds(timeout)); deadline_timer.async_wait(std::bind(&Process::timeoutHandler, this, std::placeholders::_1)); ioc.run(); c.wait(); stdOut = dataOut.get(); stdErr = dataErr.get(); } //Must be only called after run() exits int Process::getReturnStatus() { return returnStatus; } //Must be only called after run() exits std::string Process::getStdOut() { return stdOut; } //Must be only called after run() exits std::string Process::getStdErr() { return stdErr; } void Process::kill() { std::error_code ec; group.terminate(ec); if(ec) { std::cout << "Error occurred while trying to kill the process: " << ec.message(); throw std::runtime_error(ec.message()); } std::cout << "Killed the process and all its descendants" << std::endl; killed = true; stopped = true; } //Must be only called after run() exits bool Process::wasKilled() { return killed; } 

Níže je uveden kód, který k jeho testování používám

#define BOOST_TEST_DYN_LINK #define BOOST_TEST_MODULE ProcessTest #include <boost/test/unit_test.hpp> #include "../src/Process.h" BOOST_AUTO_TEST_CASE( ProcessTest ) { const std::vector<std::string> commands = { "ls" , "pwd" , "uname -a" , "cat /proc/cpuinfo" , "wget https://dl.google.com/dl/earth/client/current/google-earth-pro-stable-current.x86_64.rpm"}; for(const auto& cmd: commands) { Process p(cmd, 3600); p.run(); BOOST_CHECK( p.getReturnStatus() == 0); // #1 continues on error } const std::vector<std::string> commandsThatShouldFail = { "ls doesnotexit" , "cat /proc/doesnotexist" , "wget https://dl.google.com/dl/earth/client/current/doesnotxist.rpm"}; for(const auto& cmd: commandsThatShouldFail) { Process p(cmd, 3600); p.run(); BOOST_CHECK( p.getReturnStatus() != 0); // #1 continues on error } } 

Uveďte své cenné komentáře a návrhy.

Komentáře

  • Vítejte v revizi kódu! Říkáte, že ' se pokoušíte program implementovat. Funguje to tak daleko? Máte obavy ohledně svého kódu? Je to ' v pořádku, pokud kód ' t nemáte, pokud kód funguje, jak má.
  • @Mast Kód funguje pro případy, pro které jsem jej testoval. Ale zajímalo by mě, jestli existují nějaké závodní podmínky a další chyby?
  • Dobré! Mohli byste zahrnout tyto testy?

Odpověď

Stále si zvykám na Boost::Process, takže jsem se snad spletl. Opravdu však potřebujete c.wait() po ios.run()?

Podle jejich dokumentace , ovládání bude blokováno na ios.run() dokud nebude io_context dokončen, nebo samozřejmě pokud deadline_timer platnost vyprší.

Odpověď

Chybějící testy

I nevidím žádné testy na příkazy, které by měly vypršet. Zřetelným příkladem může být sleep 2, pokud jej spustíme s časovým limitem 1. Přísnějším testem by byl příkaz, který ignoruje všechny signály (těch, které lze ignorovat) – měli bychom zajistit, že pokud SIGTERM nebude fungovat, bude mít SIGKILL případně.

Testy založené na datech

Nelíbí se mi vidět smyčky v testovacích případech. Má Boost Test možnost pro opakující se testy s různými daty? Většina testovacích rámců ano; byl bych překvapen, kdyby to bylo m přichází z Boostu.

Napsat komentář

Vaše e-mailová adresa nebude zveřejněna. Vyžadované informace jsou označeny *