123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370 |
- #include <cassert>
- #include <iostream>
- #include <list>
- #include <utility>
- #include <string>
- #include <sstream>
- #include "service.h"
- #include "proc-service.h"
- #include "dinit-util.h"
- #include "test_procservice.h"
- // Tests of process-service related functionality.
- //
- // These tests work mostly by completely mocking out the base_process_service class. The mock
- // implementations can be found in test-baseproc.cc.
- extern eventloop_t event_loop;
- constexpr static auto REG = dependency_type::REGULAR;
- constexpr static auto WAITS = dependency_type::WAITS_FOR;
- // Regular service start
- void test_proc_service_start()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Test start with readiness notification
- void test_proc_notify_start()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_notification_fd(3);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- int nfd = base_process_service_test::get_notification_fd(&p);
- assert(nfd > 0);
- char notifystr[] = "ok started\n";
- std::vector<char> rnotifystr;
- rnotifystr.insert(rnotifystr.end(), notifystr, notifystr + sizeof(notifystr));
- bp_sys::supply_read_data(nfd, std::move(rnotifystr));
- event_loop.regd_fd_watchers[nfd]->fd_event(event_loop, nfd, dasynq::IN_EVENTS);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Unexpected termination
- void test_proc_unexpected_term()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
- sset.add_service(&d1);
- d1.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_target_state() == service_state_t::STARTED);
- assert(d1.get_state() == service_state_t::STARTED);
- assert(d1.get_target_state() == service_state_t::STARTED);
- base_process_service_test::handle_exit(&p, 0);
- assert(p.get_target_state() == service_state_t::STOPPED);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
- assert(event_loop.active_timers.size() == 0);
- assert(d1.get_state() == service_state_t::STOPPED);
- assert(d1.get_target_state() == service_state_t::STOPPED);
- sset.remove_service(&d1);
- sset.remove_service(&p);
- }
- // Unexpected termination until restarts exhausted, followed by a normal start.
- void test_proc_term_start()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- // One restart per 1000 interval
- p.set_restart_interval({1000, 0}, 1);
- p.set_auto_restart(true);
- sset.add_service(&p);
- // Start the service
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_target_state() == service_state_t::STARTED);
- // Unexpected termination - should restart
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_target_state() == service_state_t::STARTED);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_target_state() == service_state_t::STARTED);
- // 2nd unexpected termination - should stop
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_target_state() == service_state_t::STOPPED);
- assert(p.get_state() == service_state_t::STOPPED);
- // explicit restart:
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_target_state() == service_state_t::STARTED);
- // Now, again, one automatic restart should go through if the process terminates
- // restart:
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_target_state() == service_state_t::STARTED);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_target_state() == service_state_t::STARTED);
- // and stop:
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_target_state() == service_state_t::STOPPED);
- assert(p.get_state() == service_state_t::STOPPED);
- sset.remove_service(&p);
- }
- // Unexpected termination with restart
- void test_proc_term_restart()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_auto_restart(true);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- pid_t first_pid = bp_sys::last_forked_pid;
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // Starting, restart timer should be armed:
- assert(p.get_state() == service_state_t::STARTING);
- assert(event_loop.active_timers.size() == 1);
- assert(bp_sys::last_forked_pid == first_pid);
- event_loop.advance_time(default_restart_interval);
- // Startup timer now active:
- assert(event_loop.active_timers.size() == 1);
- assert(bp_sys::last_forked_pid == (first_pid + 1));
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Unexpected termination with restart, with dependent
- void test_proc_term_restart2()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- service_record b {&sset, "boot"};
- sset.add_service(&b);
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_auto_restart(true);
- sset.add_service(&p);
- b.add_dep(&p, WAITS);
- pid_t first_pid = bp_sys::last_forked_pid;
- b.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- assert(bp_sys::last_forked_pid == first_pid + 1);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- assert(bp_sys::last_forked_pid == first_pid + 1);
- // simulate process terminating, should then be restarted:
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // Starting, restart timer should be armed:
- assert(p.get_state() == service_state_t::STARTING);
- assert(event_loop.active_timers.size() == 1);
- assert(bp_sys::last_forked_pid == first_pid + 1);
- event_loop.advance_time(time_val(0, 200000000));
- // startup timer will be active:
- assert(event_loop.active_timers.size() == 1);
- sset.process_queues();
- assert(bp_sys::last_forked_pid == first_pid + 2);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- assert(sset.count_active_services() == 2);
- // Request stop, this time it should not restart:
- p.stop(true);
- sset.process_queues();
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- assert(sset.count_active_services() == 1);
- assert(bp_sys::last_forked_pid == first_pid + 2);
- // simulate terminate dinit
- sset.stop_all_services();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- assert(sset.count_active_services() == 0);
- sset.remove_service(&p);
- sset.remove_service(&b);
- }
- // Restart due to dependent, after unexpected termination
- void test_proc_term_restart3()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
- d1.set_auto_restart(true);
- sset.add_service(&d1);
- d1.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_target_state() == service_state_t::STARTED);
- assert(d1.get_state() == service_state_t::STARTED);
- assert(d1.get_target_state() == service_state_t::STARTED);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_target_state() == service_state_t::STARTED);
- assert(p.get_state() == service_state_t::STARTING);
- assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
- assert(d1.get_state() == service_state_t::STARTING);
- assert(d1.get_target_state() == service_state_t::STARTED);
- sset.remove_service(&d1);
- sset.remove_service(&p);
- event_loop.active_timers.clear();
- }
- // Restart after unexpected termination, start times out
- void test_proc_term_restart4()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_auto_restart(true);
- time_val start_timeout {5, 0};
- p.set_start_timeout(start_timeout);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- pid_t first_pid = bp_sys::last_forked_pid;
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // Starting, restart timer should be armed:
- assert(p.get_state() == service_state_t::STARTING);
- assert(event_loop.active_timers.size() == 1);
- assert(bp_sys::last_forked_pid == first_pid);
- event_loop.advance_time(default_restart_interval);
- // restart timer should have stopped, start timeout timer should have started
- assert(event_loop.active_timers.size() == 1);
- assert(bp_sys::last_forked_pid == (first_pid + 1));
- event_loop.advance_time(start_timeout);
- assert(p.get_state() == service_state_t::STOPPING);
- assert(event_loop.active_timers.size() == 1); // stop timeout should be set
- base_process_service_test::exec_succeeded(&p); // the exec must finally succeed...
- base_process_service_test::handle_exit(&p, 0);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Termination via stop request
- void test_term_via_stop()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- assert(event_loop.active_timers.size() == 1);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Termination via stop request, ensure reason is reset:
- void test_term_via_stop2()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- // first set it up with failure reason:
- base_process_service_test::exec_failed(&p, ENOENT);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::EXECFAILED);
- // now restart clean:
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // stop twice
- void test_term_via_stop3()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- assert(event_loop.active_timers.size() == 1);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(event_loop.active_timers.size() == 0);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- bp_sys::last_sig_sent = 0; // make sure signal is re-sent
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- assert(event_loop.active_timers.size() == 1);
- assert(bp_sys::last_sig_sent == SIGTERM);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Time-out during start
- void test_proc_start_timeout()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- scripted_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_start_timeout(time_val(10,0));
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- event_loop.advance_time(time_val(10,0));
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_signal_exit(&p, SIGTERM);
- sset.process_queues();
- // We set no stop script, so state should now be STOPPED with no timer set
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::TIMEDOUT);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Test that a timeout doesn't stop a "waits for" dependent to fail to start
- void test_proc_start_timeout2()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- scripted_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- p.set_start_timeout(time_val {1,0});
- init_service_defaults(p);
- sset.add_service(&p);
- service_record ts {&sset, "test-service-1", service_type_t::INTERNAL,
- {{&p, dependency_type::WAITS_FOR}} };
- ts.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- assert(ts.get_state() == service_state_t::STARTING);
- event_loop.advance_time(time_val {1,0}); // start timer should expire
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::TIMEDOUT);
- assert(ts.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Test exec() failure for process service start.
- void test_proc_start_execfail()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_failed(&p, ENOENT);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::EXECFAILED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Test no ready notification before process terminates
- void test_proc_notify_fail()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_notification_fd(3);
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- int nfd = base_process_service_test::get_notification_fd(&p);
- assert(nfd > 0);
- // Signal EOF on notify fd:
- event_loop.regd_fd_watchers[nfd]->fd_event(event_loop, nfd, dasynq::IN_EVENTS);
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Test stop timeout
- void test_proc_stop_timeout()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_stop_timeout(time_val {10, 0});
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- assert(bp_sys::last_sig_sent == SIGTERM);
- event_loop.advance_time(time_val {10, 0}); // expire stop timer
- sset.process_queues();
- // kill signal (SIGKILL) should have been sent; process not dead until it's dead, however
- assert(p.get_state() == service_state_t::STOPPING);
- assert(bp_sys::last_sig_sent == SIGKILL);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Smooth recovery
- void test_proc_smooth_recovery1()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t first_instance = bp_sys::last_forked_pid;
- assert(p.get_state() == service_state_t::STARTED);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(first_instance == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- event_loop.advance_time(time_val {0, 1000});
- sset.process_queues();
- // Now a new process should've been launched:
- assert(first_instance + 1 == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Smooth recovery without restart delay
- void test_proc_smooth_recovery2()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val(0, 0));
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t first_instance = bp_sys::last_forked_pid;
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // no restart delay, process should restart immediately:
- assert(first_instance + 1 == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // failure during smooth recovery is non-recoverable
- void test_proc_smooth_recovery3()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val(0, 0));
- sset.add_service(&p);
- service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
- d1.set_auto_restart(true);
- sset.add_service(&d1);
- d1.start();
- //p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t first_instance = bp_sys::last_forked_pid;
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // no restart delay, process should attempt restart immediately:
- assert(first_instance + 1 == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 1); // (restart timer)
- base_process_service_test::exec_failed(&p, ENOENT);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_target_state() == service_state_t::STOPPED);
- assert(d1.get_state() == service_state_t::STOPPED);
- assert(d1.get_target_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&d1);
- sset.remove_service(&p);
- }
- void test_proc_smooth_recovery4()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val(2, 0)); // 2 second restart delay
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t first_instance = bp_sys::last_forked_pid;
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // smooth recovery should have begun
- event_loop.advance_time(time_val(1, 0));
- assert(p.get_state() == service_state_t::STARTED);
- // If we stop now, timer should be cancelled
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(first_instance == bp_sys::last_forked_pid); // no more processes launched
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // stop during smooth recovery
- void test_proc_smooth_recovery5()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t first_instance = bp_sys::last_forked_pid;
- assert(p.get_state() == service_state_t::STARTED);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(first_instance == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- event_loop.advance_time(time_val {0, 1000});
- sset.process_queues();
- // Now a new process should've been launched:
- assert(first_instance + 1 == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- // However, at this stage the exec has not succeeded. If we issue a stop, we shouldn't see a signal sent yet,
- // since it's not clear what signal to send (term signal might not be SIGTERM, but if it's something else, the
- // process before exec() may not respond correctly)
- bp_sys::last_sig_sent = -1;
- p.stop(true);
- assert(bp_sys::last_sig_sent == -1);
- assert(p.get_state() == service_state_t::STOPPING);
- // Once the exec succeeds, then we should:
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(bp_sys::last_sig_sent == SIGTERM);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // stop during smooth recovery (while waiting on restart timer)
- void test_proc_smooth_recovery6()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t first_instance = bp_sys::last_forked_pid;
- assert(p.get_state() == service_state_t::STARTED);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(first_instance == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- bp_sys::last_sig_sent = -1;
- // Now issue a stop:
- p.stop(true);
- sset.process_queues();
- // since we were waiting on the restart timer, there should be no process signalled and the
- // state should now be stopped:
- assert(p.get_state() == service_state_t::STOPPED);
- assert(bp_sys::last_sig_sent == -1);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // simulate the launcher process forking a daemon process, and supply the process ID of that
- // daemon process in a pid file.
- static void supply_pid_contents(const char *pid_file, pid_t *daemon_instance_p = nullptr)
- {
- using namespace std;
- pid_t daemon_instance = ++bp_sys::last_forked_pid;
- // Set up the pid file content with the pid of the daemon
- stringstream str;
- str << daemon_instance << std::flush;
- string pid_file_content = str.str();
- vector<char> pid_file_content_v(pid_file_content.begin(), pid_file_content.end());
- bp_sys::supply_file_content(pid_file, std::move(pid_file_content_v));
- if (daemon_instance_p != nullptr) {
- *daemon_instance_p = daemon_instance;
- }
- }
- void test_bgproc_start()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- supply_pid_contents("/run/daemon.pid");
- base_process_service_test::handle_exit(&p, 0);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- void test_bgproc_start_fail()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0x1); // fail status
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- void test_bgproc_start_fail_pid()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_pid_file("/run/no-exist-daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- // launcher returns success, but no pid file exists:
- base_process_service_test::handle_exit(&p, 0x0);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- void test_bgproc_unexpected_term()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
- sset.add_service(&d1);
- d1.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- assert(p.get_state() == service_state_t::STARTING);
- assert(d1.get_state() == service_state_t::STARTING);
- supply_pid_contents("/run/daemon.pid");
- base_process_service_test::handle_exit(&p, 0);
- assert(p.get_target_state() == service_state_t::STARTED);
- assert(p.get_state() == service_state_t::STARTED);
- assert(d1.get_target_state() == service_state_t::STARTED);
- assert(d1.get_state() == service_state_t::STARTED);
- // Now the backgrounded daemon itself terminates:
- base_process_service_test::handle_exit(&p, 0);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_target_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
- assert(event_loop.active_timers.size() == 0);
- assert(d1.get_state() == service_state_t::STOPPED);
- assert(d1.get_target_state() == service_state_t::STOPPED);
- sset.remove_service(&d1);
- sset.remove_service(&p);
- }
- void test_bgproc_smooth_recover()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- // daemon process has been started now, state should be STARTED
- assert(p.get_state() == service_state_t::STARTED);
- assert(daemon_instance == bp_sys::last_forked_pid);
- base_process_service_test::handle_exit(&p, 0); // exit the daemon process
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(p.get_state() == service_state_t::STARTED);
- assert(daemon_instance == bp_sys::last_forked_pid);
- event_loop.advance_time(time_val {0, 1000});
- sset.process_queues();
- // Now a new process should've been launched:
- assert(daemon_instance + 1 == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 1); // start timer
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(event_loop.active_timers.size() == 1);
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- // Now run through it again
- base_process_service_test::handle_exit(&p, 0); // exit the daemon process
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_pid() == -1);
- assert(daemon_instance == bp_sys::last_forked_pid);
- event_loop.advance_time(time_val {0, 1000});
- sset.process_queues();
- // Now a new process should've been launched:
- assert(daemon_instance + 1 == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 1);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(event_loop.active_timers.size() == 1);
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_pid() == daemon_instance);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // stop issued during smooth recovery, with dependent
- void test_bgproc_smooth_recove2()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- bgproc_service p {&sset, "testproc", ha_string(command), command_offsets, {}};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- process_service d1 {&sset, "testproc-d1", std::move(command), command_offsets, {{&p, REG}}};
- init_service_defaults(d1);
- sset.add_service(&d1);
- d1.start();
- sset.process_queues();
- // process for p exec succeds, reads pid file, starts
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- // dependent has been forked already:
- assert(bp_sys::last_forked_pid == daemon_instance + 1);
- // dependent then starts
- base_process_service_test::exec_succeeded(&d1);
- assert(bp_sys::last_forked_pid == daemon_instance + 1);
- // exit daemon process unexpectedly:
- base_process_service_test::handle_exit(&p, 0);
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(p.get_state() == service_state_t::STARTED);
- assert(bp_sys::last_forked_pid == daemon_instance + 1);
- event_loop.advance_time(time_val {0, 1000});
- // Now a new process should've been launched:
- assert(bp_sys::last_forked_pid == daemon_instance + 2);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 1); // start timer
- // We tell the service to stop, before the smooth recovery completes (in fact
- // before the exec even succeeds):
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- assert(d1.get_state() == service_state_t::STOPPING);
- // Now the bgprocess launcher completes but with a bogus pid file contents:
- string pid_file_content_str = "";
- vector<char> pid_file_content_v(pid_file_content_str.begin(), pid_file_content_str.end());
- bp_sys::supply_file_content("/run/daemon.pid", std::move(pid_file_content_v));
- base_process_service_test::exec_succeeded(&p);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- assert(p.get_state() == service_state_t::STOPPING);
- assert(d1.get_state() == service_state_t::STOPPING);
- // Now the dependent stops:
- base_process_service_test::handle_exit(&d1, 0);
- assert(d1.get_state() == service_state_t::STOPPED);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- sset.remove_service(&d1);
- }
- // bgproc smooth recovery failure
- void test_bgproc_smooth_recove3()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- bgproc_service p {&sset, "testproc", ha_string(command), command_offsets, {}};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- // process for p exec succeds, reads pid file, starts
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- // exit daemon process unexpectedly:
- base_process_service_test::handle_exit(&p, 0);
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(p.get_state() == service_state_t::STARTED);
- assert(bp_sys::last_forked_pid == daemon_instance);
- event_loop.advance_time(time_val {0, 1000});
- // Now a new process should've been launched:
- assert(bp_sys::last_forked_pid == daemon_instance + 1);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 1); // start timer
- // Let the new process fail to executable:
- base_process_service_test::exec_failed(&p, ENOMEM);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // stop while in smooth recovery - waiting for restart timer
- void test_bgproc_smooth_recove4()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- bgproc_service p {&sset, "testproc", ha_string(command), command_offsets, {}};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- // process for p exec succeds, reads pid file, starts
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- // exit daemon process unexpectedly:
- base_process_service_test::handle_exit(&p, 0);
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(p.get_state() == service_state_t::STARTED);
- assert(bp_sys::last_forked_pid == daemon_instance);
- // Now issue stop:
- p.stop();
- sset.process_queues();
- // since process was not running, shouldn't need to wait for it to end:
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- assert(bp_sys::last_forked_pid == daemon_instance);
- // now start again:
- p.start();
- sset.process_queues();
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- // and terminate:
- p.stop();
- sset.process_queues();
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- sset.remove_service(&p);
- }
- // Unexpected termination with restart
- void test_bgproc_term_restart()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_auto_restart(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- // Launch process completes:
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_pid() == daemon_instance);
- // Unexpected termination of daemon:
- base_process_service_test::handle_exit(&p, 1);
- // Should re-start:
- assert(p.get_state() == service_state_t::STARTING);
- // Starting, restart timer should be armed:
- assert(p.get_state() == service_state_t::STARTING);
- assert(p.get_pid() == -1);
- assert(event_loop.active_timers.size() == 1);
- event_loop.advance_time(time_val(0, 1000));
- assert(event_loop.active_timers.size() == 1); // start timer
- assert(p.get_pid() != -1);
- // Supply new pid
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- // Launch process completes again
- base_process_service_test::handle_exit(&p, 0);
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_pid() == daemon_instance);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- void test_bgproc_stop()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_auto_restart(true);
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- // Launch process completes:
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- assert(p.get_state() == service_state_t::STARTED);
- assert(p.get_pid() == daemon_instance);
- // Issue stop:
- p.stop();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_exit(&p, 1);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // If launch process completes successfully (exit code 0) after "stop" issued during startup,
- // make sure the pid file is read and the daemon process is signalled.
- void test_bgproc_stop2()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_auto_restart(true);
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- // Now issue stop
- p.stop();
- // doesn't really matter if state is STARTING or STOPPING
- assert(p.get_state() == service_state_t::STARTING || p.get_state() == service_state_t::STOPPING);
- // Launch process completes:
- bp_sys::last_sig_sent = -1;
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- // What should happen now: read the pid file, immediately signal the daemon, and go STOPPING
- assert(p.get_pid() == daemon_instance);
- assert(bp_sys::last_sig_sent == SIGTERM);
- assert(p.get_state() == service_state_t::STOPPING);
- // daemon exits:
- base_process_service_test::handle_exit(&p, 1);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Stop issued during smooth recovery
- void test_bgproc_stop3()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- // daemon process has been started now, state should be STARTED
- assert(p.get_state() == service_state_t::STARTED);
- assert(daemon_instance == bp_sys::last_forked_pid);
- base_process_service_test::handle_exit(&p, 0); // exit the daemon process
- // since time hasn't been changed, we expect that the process has not yet been re-launched:
- assert(p.get_state() == service_state_t::STARTED);
- assert(daemon_instance == bp_sys::last_forked_pid);
- event_loop.advance_time(time_val {0, 1000});
- sset.process_queues();
- // Now a new process should've been launched:
- assert(daemon_instance + 1 == bp_sys::last_forked_pid);
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 1); // start timer
- base_process_service_test::exec_succeeded(&p);
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- // Now, we are in smooth recovery (state = STARTED), launch process is running.
- p.stop();
- assert(p.get_state() == service_state_t::STOPPING);
- assert(event_loop.active_timers.size() == 1);
- base_process_service_test::handle_exit(&p, 0);
- // The launch process terminated, but *successfully*. So the PID file should be read and the
- // daemon process killed also.
- assert(p.get_state() == service_state_t::STOPPING);
- assert(p.get_pid() == daemon_instance);
- base_process_service_test::handle_exit(&p, 0); // exit the daemon process
- assert(p.get_state() == service_state_t::STOPPED);
- assert(daemon_instance == bp_sys::last_forked_pid);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // stop issued via command (service with stop-command set)
- void test_bgproc_stop4()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- ha_string stop_command = "stop-command";
- list<pair<unsigned,unsigned>> stop_command_offsets;
- stop_command_offsets.emplace_back(0, stop_command.length());
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.set_stop_command(stop_command, stop_command_offsets);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- // daemon process has been started now, state should be STARTED
- assert(p.get_state() == service_state_t::STARTED);
- assert(daemon_instance == bp_sys::last_forked_pid);
- // so stop:
- p.stop();
- sset.process_queues();
- base_process_service_test::handle_stop_exit(&p, 0); // exit the daemon process
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // stop issued via command (service with stop-command set); service process dies before stop command
- void test_bgproc_stop5()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- ha_string stop_command = "stop-command";
- list<pair<unsigned,unsigned>> stop_command_offsets;
- stop_command_offsets.emplace_back(0, stop_command.length());
- bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_smooth_recovery(true);
- p.set_restart_delay(time_val {0, 1000});
- p.set_pid_file("/run/daemon.pid");
- sset.add_service(&p);
- p.set_stop_command(stop_command, stop_command_offsets);
- p.start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- pid_t daemon_instance;
- supply_pid_contents("/run/daemon.pid", &daemon_instance);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_exit(&p, 0); // exit the launch process
- sset.process_queues();
- // daemon process has been started now, state should be STARTED
- assert(p.get_state() == service_state_t::STARTED);
- assert(daemon_instance == bp_sys::last_forked_pid);
- // so stop:
- p.stop();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_stop_exit(&p, 0); // exit the daemon process
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- // Test stop timeout
- void test_scripted_stop_timeout()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- ha_string stopcommand = "stop-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_stop_command(stopcommand, command_offsets);
- p.set_stop_timeout(time_val {10, 0});
- sset.add_service(&p);
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- base_process_service_test::handle_exit(&p, 0);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- // should still be stopping:
- assert(p.get_state() == service_state_t::STOPPING);
- event_loop.advance_time(time_val {10, 0}); // expire stop timer
- sset.process_queues();
- // kill signal (SIGKILL) should have been sent; process not dead until it's dead, however
- assert(p.get_state() == service_state_t::STOPPING);
- assert(bp_sys::last_sig_sent == SIGKILL);
- base_process_service_test::handle_exit(&p, SIGKILL);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(event_loop.active_timers.size() == 0);
- sset.remove_service(&p);
- }
- void test_scripted_start_fail()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- ha_string stopcommand = "stop-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_stop_command(stopcommand, command_offsets);
- sset.add_service(&p);
- service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}});
- service_record *s3 = new service_record(&sset, "test-service-3",
- service_type_t::INTERNAL, {{&p, REG}, {s2, REG}});
- sset.add_service(s2);
- sset.add_service(s3);
- s3->start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- base_process_service_test::handle_exit(&p, 0x1); // exit fail
- sset.process_queues();
- // failed to start:
- assert(p.get_state() == service_state_t::STOPPED);
- assert(s2->get_state() == service_state_t::STOPPED);
- assert(s3->get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::FAILED);
- assert(s2->get_stop_reason() == stopped_reason_t::DEPFAILED);
- assert(s3->get_stop_reason() == stopped_reason_t::DEPFAILED);
- event_loop.active_timers.clear();
- sset.remove_service(&p);
- assert(sset.count_active_services() == 0);
- }
- void test_scripted_stop_fail()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- ha_string stopcommand = "stop-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- p.set_stop_command(stopcommand, command_offsets);
- sset.add_service(&p);
- service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {});
- service_record *s3 = new service_record(&sset, "test-service-3", service_type_t::INTERNAL,
- {{s2, REG}, {&p, REG}});
- service_record *s4 = new service_record(&sset, "test-service-4", service_type_t::INTERNAL,
- {{&p, REG}, {s3, REG}});
- sset.add_service(s2);
- sset.add_service(s3);
- sset.add_service(s4);
- s4->start();
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- base_process_service_test::handle_exit(&p, 0x0); // success
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(s2->get_state() == service_state_t::STARTED);
- assert(s3->get_state() == service_state_t::STARTED);
- assert(s4->get_state() == service_state_t::STARTED);
- pid_t last_forked = bp_sys::last_forked_pid;
- s4->stop(true);
- sset.process_queues();
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- base_process_service_test::handle_exit(&p, 0x1); // failure
- sset.process_queues();
- // The stop command should be executed once:
- assert((bp_sys::last_forked_pid - last_forked) == 1);
- assert(p.get_state() == service_state_t::STOPPED);
- assert(s2->get_state() == service_state_t::STOPPED);
- assert(s3->get_state() == service_state_t::STOPPED);
- assert(s4->get_state() == service_state_t::STOPPED);
- event_loop.active_timers.clear();
- sset.remove_service(&p);
- }
- void test_scripted_start_skip()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- service_flags_t sflags;
- sflags.skippable = true;
- p.set_flags(sflags);
- sset.add_service(&p);
- service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}});
- sset.add_service(s2);
- s2->start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::handle_signal_exit(&p, SIGINT); // interrupted
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(s2->get_state() == service_state_t::STARTED);
- assert(p.was_start_skipped());
- assert(! s2->was_start_skipped());
- assert(sset.count_active_services() == 2);
- s2->stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(s2->get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(s2->get_stop_reason() == stopped_reason_t::NORMAL);
- assert(sset.count_active_services() == 0);
- event_loop.active_timers.clear();
- sset.remove_service(&p);
- }
- // Test interrupting start of a service marked skippable
- void test_scripted_start_skip2()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- service_flags_t sflags;
- sflags.skippable = true;
- sflags.start_interruptible = true;
- p.set_flags(sflags);
- sset.add_service(&p);
- service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}});
- sset.add_service(s2);
- s2->start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- s2->stop(true); // abort startup; p should be cancelled
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- base_process_service_test::handle_signal_exit(&p, SIGINT); // interrupted
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPED);
- assert(s2->get_state() == service_state_t::STOPPED);
- assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
- assert(s2->get_stop_reason() == stopped_reason_t::NORMAL);
- assert(sset.count_active_services() == 0);
- event_loop.active_timers.clear();
- sset.remove_service(&p);
- }
- // Test that starting a service with a waits-for dependency on another - currently stopping - service,
- // causes that service to re-start.
- void test_waitsfor_restart()
- {
- using namespace std;
- service_set sset;
- ha_string command = "test-command";
- list<pair<unsigned,unsigned>> command_offsets;
- command_offsets.emplace_back(0, command.length());
- std::list<prelim_dep> depends;
- process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
- init_service_defaults(p);
- sset.add_service(&p);
- service_record tp {&sset, "test-service", service_type_t::INTERNAL, {{&p, WAITS}}};
- sset.add_service(&tp);
- // start p:
- p.start();
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STARTED);
- assert(event_loop.active_timers.size() == 0);
- // begin stopping p:
- p.stop(true);
- sset.process_queues();
- assert(p.get_state() == service_state_t::STOPPING);
- // start tp (which waits-for p):
- tp.start();
- sset.process_queues();
- assert(tp.get_state() == service_state_t::STARTING);
- assert(p.get_state() == service_state_t::STOPPING);
- // p terminates (finishes stopping). Then it should re-start...
- base_process_service_test::handle_signal_exit(&p, SIGTERM);
- sset.process_queues();
- assert(tp.get_state() == service_state_t::STARTING);
- assert(p.get_state() == service_state_t::STARTING);
- base_process_service_test::exec_succeeded(&p);
- sset.process_queues();
- assert(tp.get_state() == service_state_t::STARTED);
- assert(p.get_state() == service_state_t::STARTED);
- sset.remove_service(&tp);
- sset.remove_service(&p);
- }
- #define RUN_TEST(name, spacing) \
- std::cout << #name "..." spacing << std::flush; \
- name(); \
- std::cout << "PASSED" << std::endl;
- int main(int argc, char **argv)
- {
- RUN_TEST(test_proc_service_start, " ");
- RUN_TEST(test_proc_notify_start, " ");
- RUN_TEST(test_proc_unexpected_term, " ");
- RUN_TEST(test_proc_term_start, " ");
- RUN_TEST(test_proc_term_restart, " ");
- RUN_TEST(test_proc_term_restart2, " ");
- RUN_TEST(test_proc_term_restart3, " ");
- RUN_TEST(test_proc_term_restart4, " ");
- RUN_TEST(test_term_via_stop, " ");
- RUN_TEST(test_term_via_stop2, " ");
- RUN_TEST(test_term_via_stop3, " ");
- RUN_TEST(test_proc_start_timeout, " ");
- RUN_TEST(test_proc_start_timeout2, " ");
- RUN_TEST(test_proc_start_execfail, " ");
- RUN_TEST(test_proc_notify_fail, " ");
- RUN_TEST(test_proc_stop_timeout, " ");
- RUN_TEST(test_proc_smooth_recovery1, " ");
- RUN_TEST(test_proc_smooth_recovery2, " ");
- RUN_TEST(test_proc_smooth_recovery3, " ");
- RUN_TEST(test_proc_smooth_recovery4, " ");
- RUN_TEST(test_proc_smooth_recovery5, " ");
- RUN_TEST(test_proc_smooth_recovery6, " ");
- RUN_TEST(test_bgproc_start, " ");
- RUN_TEST(test_bgproc_start_fail, " ");
- RUN_TEST(test_bgproc_start_fail_pid, " ");
- RUN_TEST(test_bgproc_unexpected_term, "");
- RUN_TEST(test_bgproc_smooth_recover, " ");
- RUN_TEST(test_bgproc_smooth_recove2, " ");
- RUN_TEST(test_bgproc_smooth_recove3, " ");
- RUN_TEST(test_bgproc_smooth_recove4, " ");
- RUN_TEST(test_bgproc_term_restart, " ");
- RUN_TEST(test_bgproc_stop, " ");
- RUN_TEST(test_bgproc_stop2, " ");
- RUN_TEST(test_bgproc_stop3, " ");
- RUN_TEST(test_bgproc_stop4, " ");
- RUN_TEST(test_bgproc_stop5, " ");
- RUN_TEST(test_scripted_stop_timeout, " ");
- RUN_TEST(test_scripted_start_fail, " ");
- RUN_TEST(test_scripted_stop_fail, " ");
- RUN_TEST(test_scripted_start_skip, " ");
- RUN_TEST(test_scripted_start_skip2, " ");
- RUN_TEST(test_waitsfor_restart, " ");
- }
|