proctests.cc 73 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430
  1. #include <cassert>
  2. #include <iostream>
  3. #include <list>
  4. #include <utility>
  5. #include <string>
  6. #include <sstream>
  7. #include "service.h"
  8. #include "proc-service.h"
  9. #include "dinit-util.h"
  10. // Tests of process-service related functionality.
  11. //
  12. // These tests work mostly by completely mocking out the base_process_service class. The mock
  13. // implementations can be found in test-baseproc.cc.
  14. extern eventloop_t event_loop;
  15. constexpr static auto REG = dependency_type::REGULAR;
  16. constexpr static auto WAITS = dependency_type::WAITS_FOR;
  17. // Friend interface to access base_process_service private/protected members.
  18. class base_process_service_test
  19. {
  20. public:
  21. static void exec_succeeded(base_process_service *bsp)
  22. {
  23. bsp->waiting_for_execstat = false;
  24. bsp->exec_succeeded();
  25. }
  26. static void exec_failed(base_process_service *bsp, int errcode)
  27. {
  28. run_proc_err err;
  29. err.stage = exec_stage::DO_EXEC;
  30. err.st_errno = errcode;
  31. bsp->waiting_for_execstat = false;
  32. bsp->pid = -1;
  33. bsp->exec_failed(err);
  34. }
  35. static void handle_exit(base_process_service *bsp, int exit_status)
  36. {
  37. bsp->pid = -1;
  38. bsp->handle_exit_status(bp_sys::exit_status(true, false, exit_status));
  39. }
  40. static void handle_signal_exit(base_process_service *bsp, int signo)
  41. {
  42. bsp->pid = -1;
  43. bsp->handle_exit_status(bp_sys::exit_status(false, true, signo));
  44. }
  45. static void handle_stop_exit(process_service *ps, int exit_status)
  46. {
  47. ps->stop_pid = -1;
  48. ps->waiting_for_execstat = false;
  49. ps->stop_status = bp_sys::exit_status(true, false, exit_status);
  50. ps->handle_stop_exit();
  51. }
  52. static int get_notification_fd(base_process_service *bsp)
  53. {
  54. return bsp->notification_fd;
  55. }
  56. };
  57. namespace bp_sys {
  58. // last signal sent:
  59. extern int last_sig_sent;
  60. extern pid_t last_forked_pid;
  61. }
  62. static time_val default_restart_interval = time_val(0, 200000000); // 200 milliseconds
  63. static void init_service_defaults(base_process_service &ps)
  64. {
  65. ps.set_restart_interval(time_val(10,0), 3);
  66. ps.set_restart_delay(default_restart_interval); // 200 milliseconds
  67. ps.set_stop_timeout(time_val(10,0));
  68. }
  69. // Regular service start
  70. void test_proc_service_start()
  71. {
  72. using namespace std;
  73. service_set sset;
  74. ha_string command = "test-command";
  75. list<pair<unsigned,unsigned>> command_offsets;
  76. command_offsets.emplace_back(0, command.length());
  77. std::list<prelim_dep> depends;
  78. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  79. init_service_defaults(p);
  80. sset.add_service(&p);
  81. p.start();
  82. sset.process_queues();
  83. assert(p.get_state() == service_state_t::STARTING);
  84. base_process_service_test::exec_succeeded(&p);
  85. sset.process_queues();
  86. assert(p.get_state() == service_state_t::STARTED);
  87. assert(event_loop.active_timers.size() == 0);
  88. sset.remove_service(&p);
  89. }
  90. // Test start with readiness notification
  91. void test_proc_notify_start()
  92. {
  93. using namespace std;
  94. service_set sset;
  95. ha_string command = "test-command";
  96. list<pair<unsigned,unsigned>> command_offsets;
  97. command_offsets.emplace_back(0, command.length());
  98. std::list<prelim_dep> depends;
  99. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  100. init_service_defaults(p);
  101. p.set_notification_fd(3);
  102. sset.add_service(&p);
  103. p.start();
  104. sset.process_queues();
  105. assert(p.get_state() == service_state_t::STARTING);
  106. base_process_service_test::exec_succeeded(&p);
  107. sset.process_queues();
  108. assert(p.get_state() == service_state_t::STARTING);
  109. int nfd = base_process_service_test::get_notification_fd(&p);
  110. assert(nfd > 0);
  111. char notifystr[] = "ok started\n";
  112. std::vector<char> rnotifystr;
  113. rnotifystr.insert(rnotifystr.end(), notifystr, notifystr + sizeof(notifystr));
  114. bp_sys::supply_read_data(nfd, std::move(rnotifystr));
  115. event_loop.regd_fd_watchers[nfd]->fd_event(event_loop, nfd, dasynq::IN_EVENTS);
  116. assert(p.get_state() == service_state_t::STARTED);
  117. assert(event_loop.active_timers.size() == 0);
  118. sset.remove_service(&p);
  119. }
  120. // Unexpected termination
  121. void test_proc_unexpected_term()
  122. {
  123. using namespace std;
  124. service_set sset;
  125. ha_string command = "test-command";
  126. list<pair<unsigned,unsigned>> command_offsets;
  127. command_offsets.emplace_back(0, command.length());
  128. std::list<prelim_dep> depends;
  129. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  130. init_service_defaults(p);
  131. sset.add_service(&p);
  132. service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
  133. sset.add_service(&d1);
  134. d1.start();
  135. sset.process_queues();
  136. base_process_service_test::exec_succeeded(&p);
  137. sset.process_queues();
  138. assert(p.get_state() == service_state_t::STARTED);
  139. assert(p.get_target_state() == service_state_t::STARTED);
  140. assert(d1.get_state() == service_state_t::STARTED);
  141. assert(d1.get_target_state() == service_state_t::STARTED);
  142. base_process_service_test::handle_exit(&p, 0);
  143. assert(p.get_target_state() == service_state_t::STOPPED);
  144. assert(p.get_state() == service_state_t::STOPPED);
  145. assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
  146. assert(event_loop.active_timers.size() == 0);
  147. assert(d1.get_state() == service_state_t::STOPPED);
  148. assert(d1.get_target_state() == service_state_t::STOPPED);
  149. sset.remove_service(&d1);
  150. sset.remove_service(&p);
  151. }
  152. // Unexpected termination until restarts exhausted, followed by a normal start.
  153. void test_proc_term_start()
  154. {
  155. using namespace std;
  156. service_set sset;
  157. ha_string command = "test-command";
  158. list<pair<unsigned,unsigned>> command_offsets;
  159. command_offsets.emplace_back(0, command.length());
  160. std::list<prelim_dep> depends;
  161. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  162. init_service_defaults(p);
  163. // One restart per 1000 interval
  164. p.set_restart_interval({1000, 0}, 1);
  165. p.set_auto_restart(true);
  166. sset.add_service(&p);
  167. // Start the service
  168. p.start();
  169. sset.process_queues();
  170. base_process_service_test::exec_succeeded(&p);
  171. sset.process_queues();
  172. assert(p.get_state() == service_state_t::STARTED);
  173. assert(p.get_target_state() == service_state_t::STARTED);
  174. // Unexpected termination - should restart
  175. base_process_service_test::handle_exit(&p, 0);
  176. sset.process_queues();
  177. assert(p.get_target_state() == service_state_t::STARTED);
  178. assert(p.get_state() == service_state_t::STARTING);
  179. base_process_service_test::exec_succeeded(&p);
  180. sset.process_queues();
  181. assert(p.get_state() == service_state_t::STARTED);
  182. assert(p.get_target_state() == service_state_t::STARTED);
  183. // 2nd unexpected termination - should stop
  184. base_process_service_test::handle_exit(&p, 0);
  185. sset.process_queues();
  186. assert(p.get_target_state() == service_state_t::STOPPED);
  187. assert(p.get_state() == service_state_t::STOPPED);
  188. // explicit restart:
  189. p.start();
  190. sset.process_queues();
  191. base_process_service_test::exec_succeeded(&p);
  192. sset.process_queues();
  193. assert(p.get_state() == service_state_t::STARTED);
  194. assert(p.get_target_state() == service_state_t::STARTED);
  195. // Now, again, one automatic restart should go through if the process terminates
  196. // restart:
  197. base_process_service_test::handle_exit(&p, 0);
  198. sset.process_queues();
  199. assert(p.get_target_state() == service_state_t::STARTED);
  200. assert(p.get_state() == service_state_t::STARTING);
  201. base_process_service_test::exec_succeeded(&p);
  202. sset.process_queues();
  203. assert(p.get_state() == service_state_t::STARTED);
  204. assert(p.get_target_state() == service_state_t::STARTED);
  205. // and stop:
  206. base_process_service_test::handle_exit(&p, 0);
  207. sset.process_queues();
  208. assert(p.get_target_state() == service_state_t::STOPPED);
  209. assert(p.get_state() == service_state_t::STOPPED);
  210. sset.remove_service(&p);
  211. }
  212. // Unexpected termination with restart
  213. void test_proc_term_restart()
  214. {
  215. using namespace std;
  216. service_set sset;
  217. ha_string command = "test-command";
  218. list<pair<unsigned,unsigned>> command_offsets;
  219. command_offsets.emplace_back(0, command.length());
  220. std::list<prelim_dep> depends;
  221. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  222. init_service_defaults(p);
  223. p.set_auto_restart(true);
  224. sset.add_service(&p);
  225. p.start();
  226. sset.process_queues();
  227. pid_t first_pid = bp_sys::last_forked_pid;
  228. base_process_service_test::exec_succeeded(&p);
  229. sset.process_queues();
  230. assert(p.get_state() == service_state_t::STARTED);
  231. assert(event_loop.active_timers.size() == 0);
  232. base_process_service_test::handle_exit(&p, 0);
  233. sset.process_queues();
  234. // Starting, restart timer should be armed:
  235. assert(p.get_state() == service_state_t::STARTING);
  236. assert(event_loop.active_timers.size() == 1);
  237. assert(bp_sys::last_forked_pid == first_pid);
  238. event_loop.advance_time(default_restart_interval);
  239. // Startup timer now active:
  240. assert(event_loop.active_timers.size() == 1);
  241. assert(bp_sys::last_forked_pid == (first_pid + 1));
  242. base_process_service_test::exec_succeeded(&p);
  243. sset.process_queues();
  244. assert(p.get_state() == service_state_t::STARTED);
  245. assert(event_loop.active_timers.size() == 0);
  246. sset.remove_service(&p);
  247. }
  248. // Unexpected termination with restart, with dependent
  249. void test_proc_term_restart2()
  250. {
  251. using namespace std;
  252. service_set sset;
  253. ha_string command = "test-command";
  254. list<pair<unsigned,unsigned>> command_offsets;
  255. command_offsets.emplace_back(0, command.length());
  256. std::list<prelim_dep> depends;
  257. service_record b {&sset, "boot"};
  258. sset.add_service(&b);
  259. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  260. init_service_defaults(p);
  261. p.set_auto_restart(true);
  262. sset.add_service(&p);
  263. b.add_dep(&p, WAITS);
  264. pid_t first_pid = bp_sys::last_forked_pid;
  265. b.start();
  266. sset.process_queues();
  267. assert(p.get_state() == service_state_t::STARTING);
  268. assert(bp_sys::last_forked_pid == first_pid + 1);
  269. base_process_service_test::exec_succeeded(&p);
  270. sset.process_queues();
  271. assert(p.get_state() == service_state_t::STARTED);
  272. assert(event_loop.active_timers.size() == 0);
  273. assert(bp_sys::last_forked_pid == first_pid + 1);
  274. // simulate process terminating, should then be restarted:
  275. base_process_service_test::handle_exit(&p, 0);
  276. sset.process_queues();
  277. // Starting, restart timer should be armed:
  278. assert(p.get_state() == service_state_t::STARTING);
  279. assert(event_loop.active_timers.size() == 1);
  280. assert(bp_sys::last_forked_pid == first_pid + 1);
  281. event_loop.advance_time(time_val(0, 200000000));
  282. // startup timer will be active:
  283. assert(event_loop.active_timers.size() == 1);
  284. sset.process_queues();
  285. assert(bp_sys::last_forked_pid == first_pid + 2);
  286. base_process_service_test::exec_succeeded(&p);
  287. sset.process_queues();
  288. assert(p.get_state() == service_state_t::STARTED);
  289. assert(event_loop.active_timers.size() == 0);
  290. assert(sset.count_active_services() == 2);
  291. // Request stop, this time it should not restart:
  292. p.stop(true);
  293. sset.process_queues();
  294. base_process_service_test::handle_exit(&p, 0);
  295. sset.process_queues();
  296. assert(p.get_state() == service_state_t::STOPPED);
  297. assert(event_loop.active_timers.size() == 0);
  298. assert(sset.count_active_services() == 1);
  299. assert(bp_sys::last_forked_pid == first_pid + 2);
  300. // simulate terminate dinit
  301. sset.stop_all_services();
  302. sset.process_queues();
  303. assert(p.get_state() == service_state_t::STOPPED);
  304. assert(event_loop.active_timers.size() == 0);
  305. assert(sset.count_active_services() == 0);
  306. sset.remove_service(&p);
  307. sset.remove_service(&b);
  308. }
  309. // Restart due to dependent, after unexpected termination
  310. void test_proc_term_restart3()
  311. {
  312. using namespace std;
  313. service_set sset;
  314. ha_string command = "test-command";
  315. list<pair<unsigned,unsigned>> command_offsets;
  316. command_offsets.emplace_back(0, command.length());
  317. std::list<prelim_dep> depends;
  318. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  319. init_service_defaults(p);
  320. sset.add_service(&p);
  321. service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
  322. d1.set_auto_restart(true);
  323. sset.add_service(&d1);
  324. d1.start();
  325. sset.process_queues();
  326. base_process_service_test::exec_succeeded(&p);
  327. sset.process_queues();
  328. assert(p.get_state() == service_state_t::STARTED);
  329. assert(p.get_target_state() == service_state_t::STARTED);
  330. assert(d1.get_state() == service_state_t::STARTED);
  331. assert(d1.get_target_state() == service_state_t::STARTED);
  332. base_process_service_test::handle_exit(&p, 0);
  333. sset.process_queues();
  334. assert(p.get_target_state() == service_state_t::STARTED);
  335. assert(p.get_state() == service_state_t::STARTING);
  336. assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
  337. assert(d1.get_state() == service_state_t::STARTING);
  338. assert(d1.get_target_state() == service_state_t::STARTED);
  339. sset.remove_service(&d1);
  340. sset.remove_service(&p);
  341. event_loop.active_timers.clear();
  342. }
  343. // Restart after unexpected termination, start times out
  344. void test_proc_term_restart4()
  345. {
  346. using namespace std;
  347. service_set sset;
  348. ha_string command = "test-command";
  349. list<pair<unsigned,unsigned>> command_offsets;
  350. command_offsets.emplace_back(0, command.length());
  351. std::list<prelim_dep> depends;
  352. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  353. init_service_defaults(p);
  354. p.set_auto_restart(true);
  355. time_val start_timeout {5, 0};
  356. p.set_start_timeout(start_timeout);
  357. sset.add_service(&p);
  358. p.start();
  359. sset.process_queues();
  360. pid_t first_pid = bp_sys::last_forked_pid;
  361. base_process_service_test::exec_succeeded(&p);
  362. sset.process_queues();
  363. assert(p.get_state() == service_state_t::STARTED);
  364. assert(event_loop.active_timers.size() == 0);
  365. base_process_service_test::handle_exit(&p, 0);
  366. sset.process_queues();
  367. // Starting, restart timer should be armed:
  368. assert(p.get_state() == service_state_t::STARTING);
  369. assert(event_loop.active_timers.size() == 1);
  370. assert(bp_sys::last_forked_pid == first_pid);
  371. event_loop.advance_time(default_restart_interval);
  372. // restart timer should have stopped, start timeout timer should have started
  373. assert(event_loop.active_timers.size() == 1);
  374. assert(bp_sys::last_forked_pid == (first_pid + 1));
  375. event_loop.advance_time(start_timeout);
  376. assert(p.get_state() == service_state_t::STOPPING);
  377. assert(event_loop.active_timers.size() == 1); // stop timeout should be set
  378. base_process_service_test::exec_succeeded(&p); // the exec must finally succeed...
  379. base_process_service_test::handle_exit(&p, 0);
  380. assert(p.get_state() == service_state_t::STOPPED);
  381. assert(event_loop.active_timers.size() == 0);
  382. sset.remove_service(&p);
  383. }
  384. // Termination via stop request
  385. void test_term_via_stop()
  386. {
  387. using namespace std;
  388. service_set sset;
  389. ha_string command = "test-command";
  390. list<pair<unsigned,unsigned>> command_offsets;
  391. command_offsets.emplace_back(0, command.length());
  392. std::list<prelim_dep> depends;
  393. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  394. init_service_defaults(p);
  395. sset.add_service(&p);
  396. p.start();
  397. sset.process_queues();
  398. base_process_service_test::exec_succeeded(&p);
  399. sset.process_queues();
  400. assert(p.get_state() == service_state_t::STARTED);
  401. assert(event_loop.active_timers.size() == 0);
  402. p.stop(true);
  403. sset.process_queues();
  404. assert(p.get_state() == service_state_t::STOPPING);
  405. assert(event_loop.active_timers.size() == 1);
  406. base_process_service_test::handle_exit(&p, 0);
  407. sset.process_queues();
  408. assert(p.get_state() == service_state_t::STOPPED);
  409. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  410. assert(event_loop.active_timers.size() == 0);
  411. sset.remove_service(&p);
  412. }
  413. // Termination via stop request, ensure reason is reset:
  414. void test_term_via_stop2()
  415. {
  416. using namespace std;
  417. service_set sset;
  418. ha_string command = "test-command";
  419. list<pair<unsigned,unsigned>> command_offsets;
  420. command_offsets.emplace_back(0, command.length());
  421. std::list<prelim_dep> depends;
  422. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  423. init_service_defaults(p);
  424. sset.add_service(&p);
  425. p.start();
  426. sset.process_queues();
  427. // first set it up with failure reason:
  428. base_process_service_test::exec_failed(&p, ENOENT);
  429. sset.process_queues();
  430. assert(p.get_state() == service_state_t::STOPPED);
  431. assert(p.get_stop_reason() == stopped_reason_t::EXECFAILED);
  432. // now restart clean:
  433. p.start();
  434. sset.process_queues();
  435. base_process_service_test::exec_succeeded(&p);
  436. sset.process_queues();
  437. assert(p.get_state() == service_state_t::STARTED);
  438. assert(event_loop.active_timers.size() == 0);
  439. p.stop(true);
  440. sset.process_queues();
  441. assert(p.get_state() == service_state_t::STOPPING);
  442. base_process_service_test::handle_exit(&p, 0);
  443. sset.process_queues();
  444. assert(p.get_state() == service_state_t::STOPPED);
  445. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  446. assert(event_loop.active_timers.size() == 0);
  447. sset.remove_service(&p);
  448. }
  449. // stop twice
  450. void test_term_via_stop3()
  451. {
  452. using namespace std;
  453. service_set sset;
  454. ha_string command = "test-command";
  455. list<pair<unsigned,unsigned>> command_offsets;
  456. command_offsets.emplace_back(0, command.length());
  457. std::list<prelim_dep> depends;
  458. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  459. init_service_defaults(p);
  460. sset.add_service(&p);
  461. p.start();
  462. sset.process_queues();
  463. base_process_service_test::exec_succeeded(&p);
  464. sset.process_queues();
  465. assert(p.get_state() == service_state_t::STARTED);
  466. assert(event_loop.active_timers.size() == 0);
  467. p.stop(true);
  468. sset.process_queues();
  469. assert(p.get_state() == service_state_t::STOPPING);
  470. assert(event_loop.active_timers.size() == 1);
  471. base_process_service_test::handle_exit(&p, 0);
  472. sset.process_queues();
  473. assert(p.get_state() == service_state_t::STOPPED);
  474. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  475. assert(event_loop.active_timers.size() == 0);
  476. p.start();
  477. sset.process_queues();
  478. base_process_service_test::exec_succeeded(&p);
  479. sset.process_queues();
  480. assert(p.get_state() == service_state_t::STARTED);
  481. assert(event_loop.active_timers.size() == 0);
  482. bp_sys::last_sig_sent = 0; // make sure signal is re-sent
  483. p.stop(true);
  484. sset.process_queues();
  485. assert(p.get_state() == service_state_t::STOPPING);
  486. assert(event_loop.active_timers.size() == 1);
  487. assert(bp_sys::last_sig_sent == SIGTERM);
  488. base_process_service_test::handle_exit(&p, 0);
  489. sset.process_queues();
  490. assert(p.get_state() == service_state_t::STOPPED);
  491. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  492. assert(event_loop.active_timers.size() == 0);
  493. sset.remove_service(&p);
  494. }
  495. // Time-out during start
  496. void test_proc_start_timeout()
  497. {
  498. using namespace std;
  499. service_set sset;
  500. ha_string command = "test-command";
  501. list<pair<unsigned,unsigned>> command_offsets;
  502. command_offsets.emplace_back(0, command.length());
  503. std::list<prelim_dep> depends;
  504. scripted_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  505. init_service_defaults(p);
  506. p.set_start_timeout(time_val(10,0));
  507. sset.add_service(&p);
  508. p.start();
  509. sset.process_queues();
  510. assert(p.get_state() == service_state_t::STARTING);
  511. event_loop.advance_time(time_val(10,0));
  512. sset.process_queues();
  513. assert(p.get_state() == service_state_t::STOPPING);
  514. base_process_service_test::handle_signal_exit(&p, SIGTERM);
  515. sset.process_queues();
  516. // We set no stop script, so state should now be STOPPED with no timer set
  517. assert(p.get_state() == service_state_t::STOPPED);
  518. assert(p.get_stop_reason() == stopped_reason_t::TIMEDOUT);
  519. assert(event_loop.active_timers.size() == 0);
  520. sset.remove_service(&p);
  521. }
  522. // Test that a timeout doesn't stop a "waits for" dependent to fail to start
  523. void test_proc_start_timeout2()
  524. {
  525. using namespace std;
  526. service_set sset;
  527. ha_string command = "test-command";
  528. list<pair<unsigned,unsigned>> command_offsets;
  529. command_offsets.emplace_back(0, command.length());
  530. std::list<prelim_dep> depends;
  531. scripted_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  532. p.set_start_timeout(time_val {1,0});
  533. init_service_defaults(p);
  534. sset.add_service(&p);
  535. service_record ts {&sset, "test-service-1", service_type_t::INTERNAL,
  536. {{&p, dependency_type::WAITS_FOR}} };
  537. ts.start();
  538. sset.process_queues();
  539. assert(p.get_state() == service_state_t::STARTING);
  540. assert(ts.get_state() == service_state_t::STARTING);
  541. event_loop.advance_time(time_val {1,0}); // start timer should expire
  542. sset.process_queues();
  543. assert(p.get_state() == service_state_t::STOPPING);
  544. base_process_service_test::handle_exit(&p, 0);
  545. sset.process_queues();
  546. assert(p.get_state() == service_state_t::STOPPED);
  547. assert(p.get_stop_reason() == stopped_reason_t::TIMEDOUT);
  548. assert(ts.get_state() == service_state_t::STARTED);
  549. assert(event_loop.active_timers.size() == 0);
  550. sset.remove_service(&p);
  551. }
  552. // Test exec() failure for process service start.
  553. void test_proc_start_execfail()
  554. {
  555. using namespace std;
  556. service_set sset;
  557. ha_string command = "test-command";
  558. list<pair<unsigned,unsigned>> command_offsets;
  559. command_offsets.emplace_back(0, command.length());
  560. std::list<prelim_dep> depends;
  561. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  562. init_service_defaults(p);
  563. sset.add_service(&p);
  564. p.start();
  565. sset.process_queues();
  566. assert(p.get_state() == service_state_t::STARTING);
  567. base_process_service_test::exec_failed(&p, ENOENT);
  568. sset.process_queues();
  569. assert(p.get_state() == service_state_t::STOPPED);
  570. assert(p.get_stop_reason() == stopped_reason_t::EXECFAILED);
  571. assert(event_loop.active_timers.size() == 0);
  572. sset.remove_service(&p);
  573. }
  574. // Test no ready notification before process terminates
  575. void test_proc_notify_fail()
  576. {
  577. using namespace std;
  578. service_set sset;
  579. ha_string command = "test-command";
  580. list<pair<unsigned,unsigned>> command_offsets;
  581. command_offsets.emplace_back(0, command.length());
  582. std::list<prelim_dep> depends;
  583. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  584. init_service_defaults(p);
  585. p.set_notification_fd(3);
  586. sset.add_service(&p);
  587. p.start();
  588. sset.process_queues();
  589. assert(p.get_state() == service_state_t::STARTING);
  590. base_process_service_test::exec_succeeded(&p);
  591. sset.process_queues();
  592. assert(p.get_state() == service_state_t::STARTING);
  593. int nfd = base_process_service_test::get_notification_fd(&p);
  594. assert(nfd > 0);
  595. // Signal EOF on notify fd:
  596. event_loop.regd_fd_watchers[nfd]->fd_event(event_loop, nfd, dasynq::IN_EVENTS);
  597. assert(p.get_state() == service_state_t::STOPPING);
  598. base_process_service_test::handle_exit(&p, 0);
  599. sset.process_queues();
  600. assert(p.get_state() == service_state_t::STOPPED);
  601. assert(event_loop.active_timers.size() == 0);
  602. sset.remove_service(&p);
  603. }
  604. // Test stop timeout
  605. void test_proc_stop_timeout()
  606. {
  607. using namespace std;
  608. service_set sset;
  609. ha_string command = "test-command";
  610. list<pair<unsigned,unsigned>> command_offsets;
  611. command_offsets.emplace_back(0, command.length());
  612. std::list<prelim_dep> depends;
  613. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  614. init_service_defaults(p);
  615. p.set_stop_timeout(time_val {10, 0});
  616. sset.add_service(&p);
  617. p.start();
  618. sset.process_queues();
  619. assert(p.get_state() == service_state_t::STARTING);
  620. base_process_service_test::exec_succeeded(&p);
  621. sset.process_queues();
  622. assert(p.get_state() == service_state_t::STARTED);
  623. p.stop(true);
  624. sset.process_queues();
  625. assert(p.get_state() == service_state_t::STOPPING);
  626. assert(bp_sys::last_sig_sent == SIGTERM);
  627. event_loop.advance_time(time_val {10, 0}); // expire stop timer
  628. sset.process_queues();
  629. // kill signal (SIGKILL) should have been sent; process not dead until it's dead, however
  630. assert(p.get_state() == service_state_t::STOPPING);
  631. assert(bp_sys::last_sig_sent == SIGKILL);
  632. base_process_service_test::handle_exit(&p, 0);
  633. sset.process_queues();
  634. assert(p.get_state() == service_state_t::STOPPED);
  635. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  636. assert(event_loop.active_timers.size() == 0);
  637. sset.remove_service(&p);
  638. }
  639. // Smooth recovery
  640. void test_proc_smooth_recovery1()
  641. {
  642. using namespace std;
  643. service_set sset;
  644. ha_string command = "test-command";
  645. list<pair<unsigned,unsigned>> command_offsets;
  646. command_offsets.emplace_back(0, command.length());
  647. std::list<prelim_dep> depends;
  648. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  649. init_service_defaults(p);
  650. p.set_smooth_recovery(true);
  651. p.set_restart_delay(time_val {0, 1000});
  652. sset.add_service(&p);
  653. p.start();
  654. sset.process_queues();
  655. base_process_service_test::exec_succeeded(&p);
  656. sset.process_queues();
  657. pid_t first_instance = bp_sys::last_forked_pid;
  658. assert(p.get_state() == service_state_t::STARTED);
  659. base_process_service_test::handle_exit(&p, 0);
  660. sset.process_queues();
  661. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  662. assert(first_instance == bp_sys::last_forked_pid);
  663. assert(p.get_state() == service_state_t::STARTED);
  664. event_loop.advance_time(time_val {0, 1000});
  665. sset.process_queues();
  666. // Now a new process should've been launched:
  667. assert(first_instance + 1 == bp_sys::last_forked_pid);
  668. assert(p.get_state() == service_state_t::STARTED);
  669. base_process_service_test::exec_succeeded(&p);
  670. sset.process_queues();
  671. assert(event_loop.active_timers.size() == 0);
  672. sset.remove_service(&p);
  673. }
  674. // Smooth recovery without restart delay
  675. void test_proc_smooth_recovery2()
  676. {
  677. using namespace std;
  678. service_set sset;
  679. ha_string command = "test-command";
  680. list<pair<unsigned,unsigned>> command_offsets;
  681. command_offsets.emplace_back(0, command.length());
  682. std::list<prelim_dep> depends;
  683. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  684. init_service_defaults(p);
  685. p.set_smooth_recovery(true);
  686. p.set_restart_delay(time_val(0, 0));
  687. sset.add_service(&p);
  688. p.start();
  689. sset.process_queues();
  690. base_process_service_test::exec_succeeded(&p);
  691. sset.process_queues();
  692. pid_t first_instance = bp_sys::last_forked_pid;
  693. assert(p.get_state() == service_state_t::STARTED);
  694. assert(event_loop.active_timers.size() == 0);
  695. base_process_service_test::handle_exit(&p, 0);
  696. sset.process_queues();
  697. // no restart delay, process should restart immediately:
  698. assert(first_instance + 1 == bp_sys::last_forked_pid);
  699. assert(p.get_state() == service_state_t::STARTED);
  700. base_process_service_test::exec_succeeded(&p);
  701. sset.process_queues();
  702. assert(event_loop.active_timers.size() == 0);
  703. sset.remove_service(&p);
  704. }
  705. // failure during smooth recovery is non-recoverable
  706. void test_proc_smooth_recovery3()
  707. {
  708. using namespace std;
  709. service_set sset;
  710. ha_string command = "test-command";
  711. list<pair<unsigned,unsigned>> command_offsets;
  712. command_offsets.emplace_back(0, command.length());
  713. std::list<prelim_dep> depends;
  714. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  715. init_service_defaults(p);
  716. p.set_smooth_recovery(true);
  717. p.set_restart_delay(time_val(0, 0));
  718. sset.add_service(&p);
  719. service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
  720. d1.set_auto_restart(true);
  721. sset.add_service(&d1);
  722. d1.start();
  723. //p.start();
  724. sset.process_queues();
  725. base_process_service_test::exec_succeeded(&p);
  726. sset.process_queues();
  727. pid_t first_instance = bp_sys::last_forked_pid;
  728. assert(p.get_state() == service_state_t::STARTED);
  729. assert(event_loop.active_timers.size() == 0);
  730. base_process_service_test::handle_exit(&p, 0);
  731. sset.process_queues();
  732. // no restart delay, process should attempt restart immediately:
  733. assert(first_instance + 1 == bp_sys::last_forked_pid);
  734. assert(p.get_state() == service_state_t::STARTED);
  735. assert(event_loop.active_timers.size() == 1); // (restart timer)
  736. base_process_service_test::exec_failed(&p, ENOENT);
  737. sset.process_queues();
  738. assert(p.get_state() == service_state_t::STOPPED);
  739. assert(p.get_target_state() == service_state_t::STOPPED);
  740. assert(d1.get_state() == service_state_t::STOPPED);
  741. assert(d1.get_target_state() == service_state_t::STOPPED);
  742. assert(event_loop.active_timers.size() == 0);
  743. sset.remove_service(&d1);
  744. sset.remove_service(&p);
  745. }
  746. void test_proc_smooth_recovery4()
  747. {
  748. using namespace std;
  749. service_set sset;
  750. ha_string command = "test-command";
  751. list<pair<unsigned,unsigned>> command_offsets;
  752. command_offsets.emplace_back(0, command.length());
  753. std::list<prelim_dep> depends;
  754. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  755. init_service_defaults(p);
  756. p.set_smooth_recovery(true);
  757. p.set_restart_delay(time_val(2, 0)); // 2 second restart delay
  758. sset.add_service(&p);
  759. p.start();
  760. sset.process_queues();
  761. base_process_service_test::exec_succeeded(&p);
  762. sset.process_queues();
  763. pid_t first_instance = bp_sys::last_forked_pid;
  764. assert(p.get_state() == service_state_t::STARTED);
  765. assert(event_loop.active_timers.size() == 0);
  766. base_process_service_test::handle_exit(&p, 0);
  767. sset.process_queues();
  768. // smooth recovery should have begun
  769. event_loop.advance_time(time_val(1, 0));
  770. assert(p.get_state() == service_state_t::STARTED);
  771. // If we stop now, timer should be cancelled
  772. p.stop(true);
  773. sset.process_queues();
  774. assert(p.get_state() == service_state_t::STOPPED);
  775. assert(first_instance == bp_sys::last_forked_pid); // no more processes launched
  776. assert(event_loop.active_timers.size() == 0);
  777. sset.remove_service(&p);
  778. }
  779. // stop during smooth recovery
  780. void test_proc_smooth_recovery5()
  781. {
  782. using namespace std;
  783. service_set sset;
  784. ha_string command = "test-command";
  785. list<pair<unsigned,unsigned>> command_offsets;
  786. command_offsets.emplace_back(0, command.length());
  787. std::list<prelim_dep> depends;
  788. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  789. init_service_defaults(p);
  790. p.set_smooth_recovery(true);
  791. p.set_restart_delay(time_val {0, 1000});
  792. sset.add_service(&p);
  793. p.start();
  794. sset.process_queues();
  795. base_process_service_test::exec_succeeded(&p);
  796. sset.process_queues();
  797. pid_t first_instance = bp_sys::last_forked_pid;
  798. assert(p.get_state() == service_state_t::STARTED);
  799. base_process_service_test::handle_exit(&p, 0);
  800. sset.process_queues();
  801. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  802. assert(first_instance == bp_sys::last_forked_pid);
  803. assert(p.get_state() == service_state_t::STARTED);
  804. event_loop.advance_time(time_val {0, 1000});
  805. sset.process_queues();
  806. // Now a new process should've been launched:
  807. assert(first_instance + 1 == bp_sys::last_forked_pid);
  808. assert(p.get_state() == service_state_t::STARTED);
  809. // However, at this stage the exec has not succeeded. If we issue a stop, we shouldn't see a signal sent yet,
  810. // since it's not clear what signal to send (term signal might not be SIGTERM, but if it's something else, the
  811. // process before exec() may not respond correctly)
  812. bp_sys::last_sig_sent = -1;
  813. p.stop(true);
  814. assert(bp_sys::last_sig_sent == -1);
  815. assert(p.get_state() == service_state_t::STOPPING);
  816. // Once the exec succeeds, then we should:
  817. base_process_service_test::exec_succeeded(&p);
  818. sset.process_queues();
  819. assert(bp_sys::last_sig_sent == SIGTERM);
  820. base_process_service_test::handle_exit(&p, 0);
  821. sset.process_queues();
  822. assert(p.get_state() == service_state_t::STOPPED);
  823. assert(event_loop.active_timers.size() == 0);
  824. sset.remove_service(&p);
  825. }
  826. // stop during smooth recovery (while waiting on restart timer)
  827. void test_proc_smooth_recovery6()
  828. {
  829. using namespace std;
  830. service_set sset;
  831. ha_string command = "test-command";
  832. list<pair<unsigned,unsigned>> command_offsets;
  833. command_offsets.emplace_back(0, command.length());
  834. std::list<prelim_dep> depends;
  835. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  836. init_service_defaults(p);
  837. p.set_smooth_recovery(true);
  838. p.set_restart_delay(time_val {0, 1000});
  839. sset.add_service(&p);
  840. p.start();
  841. sset.process_queues();
  842. base_process_service_test::exec_succeeded(&p);
  843. sset.process_queues();
  844. pid_t first_instance = bp_sys::last_forked_pid;
  845. assert(p.get_state() == service_state_t::STARTED);
  846. base_process_service_test::handle_exit(&p, 0);
  847. sset.process_queues();
  848. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  849. assert(first_instance == bp_sys::last_forked_pid);
  850. assert(p.get_state() == service_state_t::STARTED);
  851. bp_sys::last_sig_sent = -1;
  852. // Now issue a stop:
  853. p.stop(true);
  854. sset.process_queues();
  855. // since we were waiting on the restart timer, there should be no process signalled and the
  856. // state should now be stopped:
  857. assert(p.get_state() == service_state_t::STOPPED);
  858. assert(bp_sys::last_sig_sent == -1);
  859. assert(event_loop.active_timers.size() == 0);
  860. sset.remove_service(&p);
  861. }
  862. // simulate the launcher process forking a daemon process, and supply the process ID of that
  863. // daemon process in a pid file.
  864. static void supply_pid_contents(const char *pid_file, pid_t *daemon_instance_p = nullptr)
  865. {
  866. using namespace std;
  867. pid_t daemon_instance = ++bp_sys::last_forked_pid;
  868. // Set up the pid file content with the pid of the daemon
  869. stringstream str;
  870. str << daemon_instance << std::flush;
  871. string pid_file_content = str.str();
  872. vector<char> pid_file_content_v(pid_file_content.begin(), pid_file_content.end());
  873. bp_sys::supply_file_content(pid_file, std::move(pid_file_content_v));
  874. if (daemon_instance_p != nullptr) {
  875. *daemon_instance_p = daemon_instance;
  876. }
  877. }
  878. void test_bgproc_start()
  879. {
  880. using namespace std;
  881. service_set sset;
  882. ha_string command = "test-command";
  883. list<pair<unsigned,unsigned>> command_offsets;
  884. command_offsets.emplace_back(0, command.length());
  885. std::list<prelim_dep> depends;
  886. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  887. init_service_defaults(p);
  888. p.set_pid_file("/run/daemon.pid");
  889. sset.add_service(&p);
  890. p.start();
  891. sset.process_queues();
  892. assert(p.get_state() == service_state_t::STARTING);
  893. base_process_service_test::exec_succeeded(&p);
  894. sset.process_queues();
  895. assert(p.get_state() == service_state_t::STARTING);
  896. supply_pid_contents("/run/daemon.pid");
  897. base_process_service_test::handle_exit(&p, 0);
  898. assert(p.get_state() == service_state_t::STARTED);
  899. assert(event_loop.active_timers.size() == 0);
  900. sset.remove_service(&p);
  901. }
  902. void test_bgproc_start_fail()
  903. {
  904. using namespace std;
  905. service_set sset;
  906. ha_string command = "test-command";
  907. list<pair<unsigned,unsigned>> command_offsets;
  908. command_offsets.emplace_back(0, command.length());
  909. std::list<prelim_dep> depends;
  910. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  911. init_service_defaults(p);
  912. p.set_pid_file("/run/daemon.pid");
  913. sset.add_service(&p);
  914. p.start();
  915. sset.process_queues();
  916. assert(p.get_state() == service_state_t::STARTING);
  917. base_process_service_test::exec_succeeded(&p);
  918. sset.process_queues();
  919. assert(p.get_state() == service_state_t::STARTING);
  920. base_process_service_test::handle_exit(&p, 0x1); // fail status
  921. assert(p.get_state() == service_state_t::STOPPED);
  922. assert(event_loop.active_timers.size() == 0);
  923. sset.remove_service(&p);
  924. }
  925. void test_bgproc_start_fail_pid()
  926. {
  927. using namespace std;
  928. service_set sset;
  929. ha_string command = "test-command";
  930. list<pair<unsigned,unsigned>> command_offsets;
  931. command_offsets.emplace_back(0, command.length());
  932. std::list<prelim_dep> depends;
  933. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  934. init_service_defaults(p);
  935. p.set_pid_file("/run/no-exist-daemon.pid");
  936. sset.add_service(&p);
  937. p.start();
  938. sset.process_queues();
  939. assert(p.get_state() == service_state_t::STARTING);
  940. base_process_service_test::exec_succeeded(&p);
  941. sset.process_queues();
  942. assert(p.get_state() == service_state_t::STARTING);
  943. // launcher returns success, but no pid file exists:
  944. base_process_service_test::handle_exit(&p, 0x0);
  945. assert(p.get_state() == service_state_t::STOPPED);
  946. assert(event_loop.active_timers.size() == 0);
  947. sset.remove_service(&p);
  948. }
  949. void test_bgproc_unexpected_term()
  950. {
  951. using namespace std;
  952. service_set sset;
  953. ha_string command = "test-command";
  954. list<pair<unsigned,unsigned>> command_offsets;
  955. command_offsets.emplace_back(0, command.length());
  956. std::list<prelim_dep> depends;
  957. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  958. init_service_defaults(p);
  959. p.set_pid_file("/run/daemon.pid");
  960. sset.add_service(&p);
  961. service_record d1 {&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}}};
  962. sset.add_service(&d1);
  963. d1.start();
  964. sset.process_queues();
  965. base_process_service_test::exec_succeeded(&p);
  966. assert(p.get_state() == service_state_t::STARTING);
  967. assert(d1.get_state() == service_state_t::STARTING);
  968. supply_pid_contents("/run/daemon.pid");
  969. base_process_service_test::handle_exit(&p, 0);
  970. assert(p.get_target_state() == service_state_t::STARTED);
  971. assert(p.get_state() == service_state_t::STARTED);
  972. assert(d1.get_target_state() == service_state_t::STARTED);
  973. assert(d1.get_state() == service_state_t::STARTED);
  974. // Now the backgrounded daemon itself terminates:
  975. base_process_service_test::handle_exit(&p, 0);
  976. assert(p.get_state() == service_state_t::STOPPED);
  977. assert(p.get_target_state() == service_state_t::STOPPED);
  978. assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
  979. assert(event_loop.active_timers.size() == 0);
  980. assert(d1.get_state() == service_state_t::STOPPED);
  981. assert(d1.get_target_state() == service_state_t::STOPPED);
  982. sset.remove_service(&d1);
  983. sset.remove_service(&p);
  984. }
  985. void test_bgproc_smooth_recover()
  986. {
  987. using namespace std;
  988. service_set sset;
  989. ha_string command = "test-command";
  990. list<pair<unsigned,unsigned>> command_offsets;
  991. command_offsets.emplace_back(0, command.length());
  992. std::list<prelim_dep> depends;
  993. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  994. init_service_defaults(p);
  995. p.set_smooth_recovery(true);
  996. p.set_restart_delay(time_val {0, 1000});
  997. p.set_pid_file("/run/daemon.pid");
  998. sset.add_service(&p);
  999. p.start();
  1000. sset.process_queues();
  1001. base_process_service_test::exec_succeeded(&p);
  1002. sset.process_queues();
  1003. pid_t daemon_instance;
  1004. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1005. assert(p.get_state() == service_state_t::STARTING);
  1006. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1007. sset.process_queues();
  1008. // daemon process has been started now, state should be STARTED
  1009. assert(p.get_state() == service_state_t::STARTED);
  1010. assert(daemon_instance == bp_sys::last_forked_pid);
  1011. base_process_service_test::handle_exit(&p, 0); // exit the daemon process
  1012. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  1013. assert(p.get_state() == service_state_t::STARTED);
  1014. assert(daemon_instance == bp_sys::last_forked_pid);
  1015. event_loop.advance_time(time_val {0, 1000});
  1016. sset.process_queues();
  1017. // Now a new process should've been launched:
  1018. assert(daemon_instance + 1 == bp_sys::last_forked_pid);
  1019. assert(p.get_state() == service_state_t::STARTED);
  1020. assert(event_loop.active_timers.size() == 1); // start timer
  1021. base_process_service_test::exec_succeeded(&p);
  1022. sset.process_queues();
  1023. assert(event_loop.active_timers.size() == 1);
  1024. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1025. base_process_service_test::handle_exit(&p, 0);
  1026. sset.process_queues();
  1027. assert(p.get_state() == service_state_t::STARTED);
  1028. assert(event_loop.active_timers.size() == 0);
  1029. // Now run through it again
  1030. base_process_service_test::handle_exit(&p, 0); // exit the daemon process
  1031. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  1032. assert(p.get_state() == service_state_t::STARTED);
  1033. assert(p.get_pid() == -1);
  1034. assert(daemon_instance == bp_sys::last_forked_pid);
  1035. event_loop.advance_time(time_val {0, 1000});
  1036. sset.process_queues();
  1037. // Now a new process should've been launched:
  1038. assert(daemon_instance + 1 == bp_sys::last_forked_pid);
  1039. assert(p.get_state() == service_state_t::STARTED);
  1040. assert(event_loop.active_timers.size() == 1);
  1041. base_process_service_test::exec_succeeded(&p);
  1042. sset.process_queues();
  1043. assert(event_loop.active_timers.size() == 1);
  1044. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1045. base_process_service_test::handle_exit(&p, 0);
  1046. sset.process_queues();
  1047. assert(p.get_state() == service_state_t::STARTED);
  1048. assert(p.get_pid() == daemon_instance);
  1049. assert(event_loop.active_timers.size() == 0);
  1050. sset.remove_service(&p);
  1051. }
  1052. // stop issued during smooth recovery, with dependent
  1053. void test_bgproc_smooth_recove2()
  1054. {
  1055. using namespace std;
  1056. service_set sset;
  1057. ha_string command = "test-command";
  1058. list<pair<unsigned,unsigned>> command_offsets;
  1059. command_offsets.emplace_back(0, command.length());
  1060. bgproc_service p {&sset, "testproc", ha_string(command), command_offsets, {}};
  1061. init_service_defaults(p);
  1062. p.set_smooth_recovery(true);
  1063. p.set_restart_delay(time_val {0, 1000});
  1064. p.set_pid_file("/run/daemon.pid");
  1065. sset.add_service(&p);
  1066. process_service d1 {&sset, "testproc-d1", std::move(command), command_offsets, {{&p, REG}}};
  1067. init_service_defaults(d1);
  1068. sset.add_service(&d1);
  1069. d1.start();
  1070. sset.process_queues();
  1071. // process for p exec succeds, reads pid file, starts
  1072. base_process_service_test::exec_succeeded(&p);
  1073. sset.process_queues();
  1074. pid_t daemon_instance;
  1075. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1076. assert(p.get_state() == service_state_t::STARTING);
  1077. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1078. sset.process_queues();
  1079. assert(p.get_state() == service_state_t::STARTED);
  1080. // dependent has been forked already:
  1081. assert(bp_sys::last_forked_pid == daemon_instance + 1);
  1082. // dependent then starts
  1083. base_process_service_test::exec_succeeded(&d1);
  1084. assert(bp_sys::last_forked_pid == daemon_instance + 1);
  1085. // exit daemon process unexpectedly:
  1086. base_process_service_test::handle_exit(&p, 0);
  1087. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  1088. assert(p.get_state() == service_state_t::STARTED);
  1089. assert(bp_sys::last_forked_pid == daemon_instance + 1);
  1090. event_loop.advance_time(time_val {0, 1000});
  1091. // Now a new process should've been launched:
  1092. assert(bp_sys::last_forked_pid == daemon_instance + 2);
  1093. assert(p.get_state() == service_state_t::STARTED);
  1094. assert(event_loop.active_timers.size() == 1); // start timer
  1095. // We tell the service to stop, before the smooth recovery completes (in fact
  1096. // before the exec even succeeds):
  1097. p.stop(true);
  1098. sset.process_queues();
  1099. assert(p.get_state() == service_state_t::STOPPING);
  1100. assert(d1.get_state() == service_state_t::STOPPING);
  1101. // Now the bgprocess launcher completes but with a bogus pid file contents:
  1102. string pid_file_content_str = "";
  1103. vector<char> pid_file_content_v(pid_file_content_str.begin(), pid_file_content_str.end());
  1104. bp_sys::supply_file_content("/run/daemon.pid", std::move(pid_file_content_v));
  1105. base_process_service_test::exec_succeeded(&p);
  1106. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1107. assert(p.get_state() == service_state_t::STOPPING);
  1108. assert(d1.get_state() == service_state_t::STOPPING);
  1109. // Now the dependent stops:
  1110. base_process_service_test::handle_exit(&d1, 0);
  1111. assert(d1.get_state() == service_state_t::STOPPED);
  1112. assert(p.get_state() == service_state_t::STOPPED);
  1113. assert(event_loop.active_timers.size() == 0);
  1114. sset.remove_service(&p);
  1115. sset.remove_service(&d1);
  1116. }
  1117. // bgproc smooth recovery failure
  1118. void test_bgproc_smooth_recove3()
  1119. {
  1120. using namespace std;
  1121. service_set sset;
  1122. ha_string command = "test-command";
  1123. list<pair<unsigned,unsigned>> command_offsets;
  1124. command_offsets.emplace_back(0, command.length());
  1125. bgproc_service p {&sset, "testproc", ha_string(command), command_offsets, {}};
  1126. init_service_defaults(p);
  1127. p.set_smooth_recovery(true);
  1128. p.set_restart_delay(time_val {0, 1000});
  1129. p.set_pid_file("/run/daemon.pid");
  1130. sset.add_service(&p);
  1131. p.start();
  1132. sset.process_queues();
  1133. // process for p exec succeds, reads pid file, starts
  1134. base_process_service_test::exec_succeeded(&p);
  1135. sset.process_queues();
  1136. pid_t daemon_instance;
  1137. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1138. assert(p.get_state() == service_state_t::STARTING);
  1139. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1140. sset.process_queues();
  1141. assert(p.get_state() == service_state_t::STARTED);
  1142. // exit daemon process unexpectedly:
  1143. base_process_service_test::handle_exit(&p, 0);
  1144. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  1145. assert(p.get_state() == service_state_t::STARTED);
  1146. assert(bp_sys::last_forked_pid == daemon_instance);
  1147. event_loop.advance_time(time_val {0, 1000});
  1148. // Now a new process should've been launched:
  1149. assert(bp_sys::last_forked_pid == daemon_instance + 1);
  1150. assert(p.get_state() == service_state_t::STARTED);
  1151. assert(event_loop.active_timers.size() == 1); // start timer
  1152. // Let the new process fail to executable:
  1153. base_process_service_test::exec_failed(&p, ENOMEM);
  1154. sset.process_queues();
  1155. assert(p.get_state() == service_state_t::STOPPED);
  1156. assert(p.get_stop_reason() == stopped_reason_t::TERMINATED);
  1157. assert(event_loop.active_timers.size() == 0);
  1158. sset.remove_service(&p);
  1159. }
  1160. // stop while in smooth recovery - waiting for restart timer
  1161. void test_bgproc_smooth_recove4()
  1162. {
  1163. using namespace std;
  1164. service_set sset;
  1165. ha_string command = "test-command";
  1166. list<pair<unsigned,unsigned>> command_offsets;
  1167. command_offsets.emplace_back(0, command.length());
  1168. bgproc_service p {&sset, "testproc", ha_string(command), command_offsets, {}};
  1169. init_service_defaults(p);
  1170. p.set_smooth_recovery(true);
  1171. p.set_restart_delay(time_val {0, 1000});
  1172. p.set_pid_file("/run/daemon.pid");
  1173. sset.add_service(&p);
  1174. p.start();
  1175. sset.process_queues();
  1176. // process for p exec succeds, reads pid file, starts
  1177. base_process_service_test::exec_succeeded(&p);
  1178. sset.process_queues();
  1179. pid_t daemon_instance;
  1180. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1181. assert(p.get_state() == service_state_t::STARTING);
  1182. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1183. sset.process_queues();
  1184. assert(p.get_state() == service_state_t::STARTED);
  1185. // exit daemon process unexpectedly:
  1186. base_process_service_test::handle_exit(&p, 0);
  1187. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  1188. assert(p.get_state() == service_state_t::STARTED);
  1189. assert(bp_sys::last_forked_pid == daemon_instance);
  1190. // Now issue stop:
  1191. p.stop();
  1192. sset.process_queues();
  1193. // since process was not running, shouldn't need to wait for it to end:
  1194. assert(p.get_state() == service_state_t::STOPPED);
  1195. assert(event_loop.active_timers.size() == 0);
  1196. assert(bp_sys::last_forked_pid == daemon_instance);
  1197. // now start again:
  1198. p.start();
  1199. sset.process_queues();
  1200. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1201. assert(p.get_state() == service_state_t::STARTING);
  1202. base_process_service_test::exec_succeeded(&p);
  1203. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1204. sset.process_queues();
  1205. assert(p.get_state() == service_state_t::STARTED);
  1206. // and terminate:
  1207. p.stop();
  1208. sset.process_queues();
  1209. base_process_service_test::handle_exit(&p, 0);
  1210. sset.process_queues();
  1211. assert(p.get_state() == service_state_t::STOPPED);
  1212. sset.remove_service(&p);
  1213. }
  1214. // Unexpected termination with restart
  1215. void test_bgproc_term_restart()
  1216. {
  1217. using namespace std;
  1218. service_set sset;
  1219. ha_string command = "test-command";
  1220. list<pair<unsigned,unsigned>> command_offsets;
  1221. command_offsets.emplace_back(0, command.length());
  1222. std::list<prelim_dep> depends;
  1223. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  1224. init_service_defaults(p);
  1225. p.set_auto_restart(true);
  1226. p.set_restart_delay(time_val {0, 1000});
  1227. p.set_pid_file("/run/daemon.pid");
  1228. sset.add_service(&p);
  1229. p.start();
  1230. sset.process_queues();
  1231. base_process_service_test::exec_succeeded(&p);
  1232. sset.process_queues();
  1233. assert(p.get_state() == service_state_t::STARTING);
  1234. pid_t daemon_instance;
  1235. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1236. // Launch process completes:
  1237. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1238. assert(p.get_state() == service_state_t::STARTED);
  1239. assert(p.get_pid() == daemon_instance);
  1240. // Unexpected termination of daemon:
  1241. base_process_service_test::handle_exit(&p, 1);
  1242. // Should re-start:
  1243. assert(p.get_state() == service_state_t::STARTING);
  1244. // Starting, restart timer should be armed:
  1245. assert(p.get_state() == service_state_t::STARTING);
  1246. assert(p.get_pid() == -1);
  1247. assert(event_loop.active_timers.size() == 1);
  1248. event_loop.advance_time(time_val(0, 1000));
  1249. assert(event_loop.active_timers.size() == 1); // start timer
  1250. assert(p.get_pid() != -1);
  1251. // Supply new pid
  1252. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1253. // Launch process completes again
  1254. base_process_service_test::handle_exit(&p, 0);
  1255. assert(p.get_state() == service_state_t::STARTED);
  1256. assert(p.get_pid() == daemon_instance);
  1257. assert(event_loop.active_timers.size() == 0);
  1258. sset.remove_service(&p);
  1259. }
  1260. void test_bgproc_stop()
  1261. {
  1262. using namespace std;
  1263. service_set sset;
  1264. ha_string command = "test-command";
  1265. list<pair<unsigned,unsigned>> command_offsets;
  1266. command_offsets.emplace_back(0, command.length());
  1267. std::list<prelim_dep> depends;
  1268. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  1269. init_service_defaults(p);
  1270. p.set_auto_restart(true);
  1271. p.set_pid_file("/run/daemon.pid");
  1272. sset.add_service(&p);
  1273. p.start();
  1274. sset.process_queues();
  1275. base_process_service_test::exec_succeeded(&p);
  1276. sset.process_queues();
  1277. assert(p.get_state() == service_state_t::STARTING);
  1278. pid_t daemon_instance;
  1279. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1280. // Launch process completes:
  1281. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1282. assert(p.get_state() == service_state_t::STARTED);
  1283. assert(p.get_pid() == daemon_instance);
  1284. // Issue stop:
  1285. p.stop();
  1286. sset.process_queues();
  1287. assert(p.get_state() == service_state_t::STOPPING);
  1288. base_process_service_test::handle_exit(&p, 1);
  1289. assert(p.get_state() == service_state_t::STOPPED);
  1290. assert(event_loop.active_timers.size() == 0);
  1291. sset.remove_service(&p);
  1292. }
  1293. // If launch process completes successfully (exit code 0) after "stop" issued during startup,
  1294. // make sure the pid file is read and the daemon process is signalled.
  1295. void test_bgproc_stop2()
  1296. {
  1297. using namespace std;
  1298. service_set sset;
  1299. ha_string command = "test-command";
  1300. list<pair<unsigned,unsigned>> command_offsets;
  1301. command_offsets.emplace_back(0, command.length());
  1302. std::list<prelim_dep> depends;
  1303. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  1304. init_service_defaults(p);
  1305. p.set_auto_restart(true);
  1306. p.set_pid_file("/run/daemon.pid");
  1307. sset.add_service(&p);
  1308. p.start();
  1309. sset.process_queues();
  1310. base_process_service_test::exec_succeeded(&p);
  1311. sset.process_queues();
  1312. assert(p.get_state() == service_state_t::STARTING);
  1313. pid_t daemon_instance;
  1314. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1315. // Now issue stop
  1316. p.stop();
  1317. // doesn't really matter if state is STARTING or STOPPING
  1318. assert(p.get_state() == service_state_t::STARTING || p.get_state() == service_state_t::STOPPING);
  1319. // Launch process completes:
  1320. bp_sys::last_sig_sent = -1;
  1321. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1322. // What should happen now: read the pid file, immediately signal the daemon, and go STOPPING
  1323. assert(p.get_pid() == daemon_instance);
  1324. assert(bp_sys::last_sig_sent == SIGTERM);
  1325. assert(p.get_state() == service_state_t::STOPPING);
  1326. // daemon exits:
  1327. base_process_service_test::handle_exit(&p, 1);
  1328. assert(p.get_state() == service_state_t::STOPPED);
  1329. assert(event_loop.active_timers.size() == 0);
  1330. sset.remove_service(&p);
  1331. }
  1332. // Stop issued during smooth recovery
  1333. void test_bgproc_stop3()
  1334. {
  1335. using namespace std;
  1336. service_set sset;
  1337. ha_string command = "test-command";
  1338. list<pair<unsigned,unsigned>> command_offsets;
  1339. command_offsets.emplace_back(0, command.length());
  1340. std::list<prelim_dep> depends;
  1341. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  1342. init_service_defaults(p);
  1343. p.set_smooth_recovery(true);
  1344. p.set_restart_delay(time_val {0, 1000});
  1345. p.set_pid_file("/run/daemon.pid");
  1346. sset.add_service(&p);
  1347. p.start();
  1348. sset.process_queues();
  1349. base_process_service_test::exec_succeeded(&p);
  1350. sset.process_queues();
  1351. pid_t daemon_instance;
  1352. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1353. assert(p.get_state() == service_state_t::STARTING);
  1354. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1355. sset.process_queues();
  1356. // daemon process has been started now, state should be STARTED
  1357. assert(p.get_state() == service_state_t::STARTED);
  1358. assert(daemon_instance == bp_sys::last_forked_pid);
  1359. base_process_service_test::handle_exit(&p, 0); // exit the daemon process
  1360. // since time hasn't been changed, we expect that the process has not yet been re-launched:
  1361. assert(p.get_state() == service_state_t::STARTED);
  1362. assert(daemon_instance == bp_sys::last_forked_pid);
  1363. event_loop.advance_time(time_val {0, 1000});
  1364. sset.process_queues();
  1365. // Now a new process should've been launched:
  1366. assert(daemon_instance + 1 == bp_sys::last_forked_pid);
  1367. assert(p.get_state() == service_state_t::STARTED);
  1368. assert(event_loop.active_timers.size() == 1); // start timer
  1369. base_process_service_test::exec_succeeded(&p);
  1370. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1371. // Now, we are in smooth recovery (state = STARTED), launch process is running.
  1372. p.stop();
  1373. assert(p.get_state() == service_state_t::STOPPING);
  1374. assert(event_loop.active_timers.size() == 1);
  1375. base_process_service_test::handle_exit(&p, 0);
  1376. // The launch process terminated, but *successfully*. So the PID file should be read and the
  1377. // daemon process killed also.
  1378. assert(p.get_state() == service_state_t::STOPPING);
  1379. assert(p.get_pid() == daemon_instance);
  1380. base_process_service_test::handle_exit(&p, 0); // exit the daemon process
  1381. assert(p.get_state() == service_state_t::STOPPED);
  1382. assert(daemon_instance == bp_sys::last_forked_pid);
  1383. assert(event_loop.active_timers.size() == 0);
  1384. sset.remove_service(&p);
  1385. }
  1386. // stop issued via command (service with stop-command set)
  1387. void test_bgproc_stop4()
  1388. {
  1389. using namespace std;
  1390. service_set sset;
  1391. ha_string command = "test-command";
  1392. list<pair<unsigned,unsigned>> command_offsets;
  1393. command_offsets.emplace_back(0, command.length());
  1394. std::list<prelim_dep> depends;
  1395. ha_string stop_command = "stop-command";
  1396. list<pair<unsigned,unsigned>> stop_command_offsets;
  1397. stop_command_offsets.emplace_back(0, stop_command.length());
  1398. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  1399. init_service_defaults(p);
  1400. p.set_smooth_recovery(true);
  1401. p.set_restart_delay(time_val {0, 1000});
  1402. p.set_pid_file("/run/daemon.pid");
  1403. sset.add_service(&p);
  1404. p.set_stop_command(stop_command, stop_command_offsets);
  1405. p.start();
  1406. sset.process_queues();
  1407. base_process_service_test::exec_succeeded(&p);
  1408. sset.process_queues();
  1409. pid_t daemon_instance;
  1410. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1411. assert(p.get_state() == service_state_t::STARTING);
  1412. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1413. sset.process_queues();
  1414. // daemon process has been started now, state should be STARTED
  1415. assert(p.get_state() == service_state_t::STARTED);
  1416. assert(daemon_instance == bp_sys::last_forked_pid);
  1417. // so stop:
  1418. p.stop();
  1419. sset.process_queues();
  1420. base_process_service_test::handle_stop_exit(&p, 0); // exit the daemon process
  1421. sset.process_queues();
  1422. assert(p.get_state() == service_state_t::STOPPING);
  1423. base_process_service_test::handle_exit(&p, 0);
  1424. sset.process_queues();
  1425. assert(p.get_state() == service_state_t::STOPPED);
  1426. assert(event_loop.active_timers.size() == 0);
  1427. sset.remove_service(&p);
  1428. }
  1429. // stop issued via command (service with stop-command set); service process dies before stop command
  1430. void test_bgproc_stop5()
  1431. {
  1432. using namespace std;
  1433. service_set sset;
  1434. ha_string command = "test-command";
  1435. list<pair<unsigned,unsigned>> command_offsets;
  1436. command_offsets.emplace_back(0, command.length());
  1437. std::list<prelim_dep> depends;
  1438. ha_string stop_command = "stop-command";
  1439. list<pair<unsigned,unsigned>> stop_command_offsets;
  1440. stop_command_offsets.emplace_back(0, stop_command.length());
  1441. bgproc_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  1442. init_service_defaults(p);
  1443. p.set_smooth_recovery(true);
  1444. p.set_restart_delay(time_val {0, 1000});
  1445. p.set_pid_file("/run/daemon.pid");
  1446. sset.add_service(&p);
  1447. p.set_stop_command(stop_command, stop_command_offsets);
  1448. p.start();
  1449. sset.process_queues();
  1450. base_process_service_test::exec_succeeded(&p);
  1451. sset.process_queues();
  1452. pid_t daemon_instance;
  1453. supply_pid_contents("/run/daemon.pid", &daemon_instance);
  1454. assert(p.get_state() == service_state_t::STARTING);
  1455. base_process_service_test::handle_exit(&p, 0); // exit the launch process
  1456. sset.process_queues();
  1457. // daemon process has been started now, state should be STARTED
  1458. assert(p.get_state() == service_state_t::STARTED);
  1459. assert(daemon_instance == bp_sys::last_forked_pid);
  1460. // so stop:
  1461. p.stop();
  1462. sset.process_queues();
  1463. assert(p.get_state() == service_state_t::STOPPING);
  1464. base_process_service_test::handle_exit(&p, 0);
  1465. sset.process_queues();
  1466. assert(p.get_state() == service_state_t::STOPPING);
  1467. base_process_service_test::handle_stop_exit(&p, 0); // exit the daemon process
  1468. sset.process_queues();
  1469. assert(p.get_state() == service_state_t::STOPPED);
  1470. assert(event_loop.active_timers.size() == 0);
  1471. sset.remove_service(&p);
  1472. }
  1473. // Test stop timeout
  1474. void test_scripted_stop_timeout()
  1475. {
  1476. using namespace std;
  1477. service_set sset;
  1478. ha_string command = "test-command";
  1479. ha_string stopcommand = "stop-command";
  1480. list<pair<unsigned,unsigned>> command_offsets;
  1481. command_offsets.emplace_back(0, command.length());
  1482. std::list<prelim_dep> depends;
  1483. scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
  1484. init_service_defaults(p);
  1485. p.set_stop_command(stopcommand, command_offsets);
  1486. p.set_stop_timeout(time_val {10, 0});
  1487. sset.add_service(&p);
  1488. p.start();
  1489. sset.process_queues();
  1490. assert(p.get_state() == service_state_t::STARTING);
  1491. base_process_service_test::exec_succeeded(&p);
  1492. sset.process_queues();
  1493. base_process_service_test::handle_exit(&p, 0);
  1494. sset.process_queues();
  1495. assert(p.get_state() == service_state_t::STARTED);
  1496. p.stop(true);
  1497. sset.process_queues();
  1498. assert(p.get_state() == service_state_t::STOPPING);
  1499. base_process_service_test::exec_succeeded(&p);
  1500. sset.process_queues();
  1501. // should still be stopping:
  1502. assert(p.get_state() == service_state_t::STOPPING);
  1503. event_loop.advance_time(time_val {10, 0}); // expire stop timer
  1504. sset.process_queues();
  1505. // kill signal (SIGKILL) should have been sent; process not dead until it's dead, however
  1506. assert(p.get_state() == service_state_t::STOPPING);
  1507. assert(bp_sys::last_sig_sent == SIGKILL);
  1508. base_process_service_test::handle_exit(&p, SIGKILL);
  1509. sset.process_queues();
  1510. assert(p.get_state() == service_state_t::STOPPED);
  1511. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  1512. assert(event_loop.active_timers.size() == 0);
  1513. sset.remove_service(&p);
  1514. }
  1515. void test_scripted_start_fail()
  1516. {
  1517. using namespace std;
  1518. service_set sset;
  1519. ha_string command = "test-command";
  1520. ha_string stopcommand = "stop-command";
  1521. list<pair<unsigned,unsigned>> command_offsets;
  1522. command_offsets.emplace_back(0, command.length());
  1523. std::list<prelim_dep> depends;
  1524. scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
  1525. init_service_defaults(p);
  1526. p.set_stop_command(stopcommand, command_offsets);
  1527. sset.add_service(&p);
  1528. service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}});
  1529. service_record *s3 = new service_record(&sset, "test-service-3",
  1530. service_type_t::INTERNAL, {{&p, REG}, {s2, REG}});
  1531. sset.add_service(s2);
  1532. sset.add_service(s3);
  1533. s3->start();
  1534. sset.process_queues();
  1535. assert(p.get_state() == service_state_t::STARTING);
  1536. base_process_service_test::exec_succeeded(&p);
  1537. sset.process_queues();
  1538. base_process_service_test::handle_exit(&p, 0x1); // exit fail
  1539. sset.process_queues();
  1540. // failed to start:
  1541. assert(p.get_state() == service_state_t::STOPPED);
  1542. assert(s2->get_state() == service_state_t::STOPPED);
  1543. assert(s3->get_state() == service_state_t::STOPPED);
  1544. assert(p.get_stop_reason() == stopped_reason_t::FAILED);
  1545. assert(s2->get_stop_reason() == stopped_reason_t::DEPFAILED);
  1546. assert(s3->get_stop_reason() == stopped_reason_t::DEPFAILED);
  1547. event_loop.active_timers.clear();
  1548. sset.remove_service(&p);
  1549. assert(sset.count_active_services() == 0);
  1550. }
  1551. void test_scripted_stop_fail()
  1552. {
  1553. using namespace std;
  1554. service_set sset;
  1555. ha_string command = "test-command";
  1556. ha_string stopcommand = "stop-command";
  1557. list<pair<unsigned,unsigned>> command_offsets;
  1558. command_offsets.emplace_back(0, command.length());
  1559. std::list<prelim_dep> depends;
  1560. scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
  1561. init_service_defaults(p);
  1562. p.set_stop_command(stopcommand, command_offsets);
  1563. sset.add_service(&p);
  1564. service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {});
  1565. service_record *s3 = new service_record(&sset, "test-service-3", service_type_t::INTERNAL,
  1566. {{s2, REG}, {&p, REG}});
  1567. service_record *s4 = new service_record(&sset, "test-service-4", service_type_t::INTERNAL,
  1568. {{&p, REG}, {s3, REG}});
  1569. sset.add_service(s2);
  1570. sset.add_service(s3);
  1571. sset.add_service(s4);
  1572. s4->start();
  1573. sset.process_queues();
  1574. base_process_service_test::exec_succeeded(&p);
  1575. sset.process_queues();
  1576. base_process_service_test::handle_exit(&p, 0x0); // success
  1577. sset.process_queues();
  1578. assert(p.get_state() == service_state_t::STARTED);
  1579. assert(s2->get_state() == service_state_t::STARTED);
  1580. assert(s3->get_state() == service_state_t::STARTED);
  1581. assert(s4->get_state() == service_state_t::STARTED);
  1582. pid_t last_forked = bp_sys::last_forked_pid;
  1583. s4->stop(true);
  1584. sset.process_queues();
  1585. base_process_service_test::exec_succeeded(&p);
  1586. sset.process_queues();
  1587. base_process_service_test::handle_exit(&p, 0x1); // failure
  1588. sset.process_queues();
  1589. // The stop command should be executed once:
  1590. assert((bp_sys::last_forked_pid - last_forked) == 1);
  1591. assert(p.get_state() == service_state_t::STOPPED);
  1592. assert(s2->get_state() == service_state_t::STOPPED);
  1593. assert(s3->get_state() == service_state_t::STOPPED);
  1594. assert(s4->get_state() == service_state_t::STOPPED);
  1595. event_loop.active_timers.clear();
  1596. sset.remove_service(&p);
  1597. }
  1598. void test_scripted_start_skip()
  1599. {
  1600. using namespace std;
  1601. service_set sset;
  1602. ha_string command = "test-command";
  1603. list<pair<unsigned,unsigned>> command_offsets;
  1604. command_offsets.emplace_back(0, command.length());
  1605. std::list<prelim_dep> depends;
  1606. scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
  1607. init_service_defaults(p);
  1608. service_flags_t sflags;
  1609. sflags.skippable = true;
  1610. p.set_flags(sflags);
  1611. sset.add_service(&p);
  1612. service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}});
  1613. sset.add_service(s2);
  1614. s2->start();
  1615. sset.process_queues();
  1616. assert(p.get_state() == service_state_t::STARTING);
  1617. base_process_service_test::exec_succeeded(&p);
  1618. sset.process_queues();
  1619. assert(p.get_state() == service_state_t::STARTING);
  1620. base_process_service_test::handle_signal_exit(&p, SIGINT); // interrupted
  1621. sset.process_queues();
  1622. assert(p.get_state() == service_state_t::STARTED);
  1623. assert(s2->get_state() == service_state_t::STARTED);
  1624. assert(p.was_start_skipped());
  1625. assert(! s2->was_start_skipped());
  1626. assert(sset.count_active_services() == 2);
  1627. s2->stop(true);
  1628. sset.process_queues();
  1629. assert(p.get_state() == service_state_t::STOPPED);
  1630. assert(s2->get_state() == service_state_t::STOPPED);
  1631. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  1632. assert(s2->get_stop_reason() == stopped_reason_t::NORMAL);
  1633. assert(sset.count_active_services() == 0);
  1634. event_loop.active_timers.clear();
  1635. sset.remove_service(&p);
  1636. }
  1637. // Test interrupting start of a service marked skippable
  1638. void test_scripted_start_skip2()
  1639. {
  1640. using namespace std;
  1641. service_set sset;
  1642. ha_string command = "test-command";
  1643. list<pair<unsigned,unsigned>> command_offsets;
  1644. command_offsets.emplace_back(0, command.length());
  1645. std::list<prelim_dep> depends;
  1646. scripted_service p {&sset, "testscripted", std::move(command), command_offsets, depends};
  1647. init_service_defaults(p);
  1648. service_flags_t sflags;
  1649. sflags.skippable = true;
  1650. sflags.start_interruptible = true;
  1651. p.set_flags(sflags);
  1652. sset.add_service(&p);
  1653. service_record *s2 = new service_record(&sset, "test-service-2", service_type_t::INTERNAL, {{&p, REG}});
  1654. sset.add_service(s2);
  1655. s2->start();
  1656. sset.process_queues();
  1657. assert(p.get_state() == service_state_t::STARTING);
  1658. base_process_service_test::exec_succeeded(&p);
  1659. sset.process_queues();
  1660. assert(p.get_state() == service_state_t::STARTING);
  1661. s2->stop(true); // abort startup; p should be cancelled
  1662. sset.process_queues();
  1663. assert(p.get_state() == service_state_t::STOPPING);
  1664. base_process_service_test::handle_signal_exit(&p, SIGINT); // interrupted
  1665. sset.process_queues();
  1666. assert(p.get_state() == service_state_t::STOPPED);
  1667. assert(s2->get_state() == service_state_t::STOPPED);
  1668. assert(p.get_stop_reason() == stopped_reason_t::NORMAL);
  1669. assert(s2->get_stop_reason() == stopped_reason_t::NORMAL);
  1670. assert(sset.count_active_services() == 0);
  1671. event_loop.active_timers.clear();
  1672. sset.remove_service(&p);
  1673. }
  1674. // Test that starting a service with a waits-for dependency on another - currently stopping - service,
  1675. // causes that service to re-start.
  1676. void test_waitsfor_restart()
  1677. {
  1678. using namespace std;
  1679. service_set sset;
  1680. ha_string command = "test-command";
  1681. list<pair<unsigned,unsigned>> command_offsets;
  1682. command_offsets.emplace_back(0, command.length());
  1683. std::list<prelim_dep> depends;
  1684. process_service p {&sset, "testproc", std::move(command), command_offsets, depends};
  1685. init_service_defaults(p);
  1686. sset.add_service(&p);
  1687. service_record tp {&sset, "test-service", service_type_t::INTERNAL, {{&p, WAITS}}};
  1688. sset.add_service(&tp);
  1689. // start p:
  1690. p.start();
  1691. sset.process_queues();
  1692. assert(p.get_state() == service_state_t::STARTING);
  1693. base_process_service_test::exec_succeeded(&p);
  1694. sset.process_queues();
  1695. assert(p.get_state() == service_state_t::STARTED);
  1696. assert(event_loop.active_timers.size() == 0);
  1697. // begin stopping p:
  1698. p.stop(true);
  1699. sset.process_queues();
  1700. assert(p.get_state() == service_state_t::STOPPING);
  1701. // start tp (which waits-for p):
  1702. tp.start();
  1703. sset.process_queues();
  1704. assert(tp.get_state() == service_state_t::STARTING);
  1705. assert(p.get_state() == service_state_t::STOPPING);
  1706. // p terminates (finishes stopping). Then it should re-start...
  1707. base_process_service_test::handle_signal_exit(&p, SIGTERM);
  1708. sset.process_queues();
  1709. assert(tp.get_state() == service_state_t::STARTING);
  1710. assert(p.get_state() == service_state_t::STARTING);
  1711. base_process_service_test::exec_succeeded(&p);
  1712. sset.process_queues();
  1713. assert(tp.get_state() == service_state_t::STARTED);
  1714. assert(p.get_state() == service_state_t::STARTED);
  1715. sset.remove_service(&tp);
  1716. sset.remove_service(&p);
  1717. }
  1718. #define RUN_TEST(name, spacing) \
  1719. std::cout << #name "..." spacing << std::flush; \
  1720. name(); \
  1721. std::cout << "PASSED" << std::endl;
  1722. int main(int argc, char **argv)
  1723. {
  1724. RUN_TEST(test_proc_service_start, " ");
  1725. RUN_TEST(test_proc_notify_start, " ");
  1726. RUN_TEST(test_proc_unexpected_term, " ");
  1727. RUN_TEST(test_proc_term_start, " ");
  1728. RUN_TEST(test_proc_term_restart, " ");
  1729. RUN_TEST(test_proc_term_restart2, " ");
  1730. RUN_TEST(test_proc_term_restart3, " ");
  1731. RUN_TEST(test_proc_term_restart4, " ");
  1732. RUN_TEST(test_term_via_stop, " ");
  1733. RUN_TEST(test_term_via_stop2, " ");
  1734. RUN_TEST(test_term_via_stop3, " ");
  1735. RUN_TEST(test_proc_start_timeout, " ");
  1736. RUN_TEST(test_proc_start_timeout2, " ");
  1737. RUN_TEST(test_proc_start_execfail, " ");
  1738. RUN_TEST(test_proc_notify_fail, " ");
  1739. RUN_TEST(test_proc_stop_timeout, " ");
  1740. RUN_TEST(test_proc_smooth_recovery1, " ");
  1741. RUN_TEST(test_proc_smooth_recovery2, " ");
  1742. RUN_TEST(test_proc_smooth_recovery3, " ");
  1743. RUN_TEST(test_proc_smooth_recovery4, " ");
  1744. RUN_TEST(test_proc_smooth_recovery5, " ");
  1745. RUN_TEST(test_proc_smooth_recovery6, " ");
  1746. RUN_TEST(test_bgproc_start, " ");
  1747. RUN_TEST(test_bgproc_start_fail, " ");
  1748. RUN_TEST(test_bgproc_start_fail_pid, " ");
  1749. RUN_TEST(test_bgproc_unexpected_term, "");
  1750. RUN_TEST(test_bgproc_smooth_recover, " ");
  1751. RUN_TEST(test_bgproc_smooth_recove2, " ");
  1752. RUN_TEST(test_bgproc_smooth_recove3, " ");
  1753. RUN_TEST(test_bgproc_smooth_recove4, " ");
  1754. RUN_TEST(test_bgproc_term_restart, " ");
  1755. RUN_TEST(test_bgproc_stop, " ");
  1756. RUN_TEST(test_bgproc_stop2, " ");
  1757. RUN_TEST(test_bgproc_stop3, " ");
  1758. RUN_TEST(test_bgproc_stop4, " ");
  1759. RUN_TEST(test_bgproc_stop5, " ");
  1760. RUN_TEST(test_scripted_stop_timeout, " ");
  1761. RUN_TEST(test_scripted_start_fail, " ");
  1762. RUN_TEST(test_scripted_stop_fail, " ");
  1763. RUN_TEST(test_scripted_start_skip, " ");
  1764. RUN_TEST(test_scripted_start_skip2, " ");
  1765. RUN_TEST(test_waitsfor_restart, " ");
  1766. }