sam_test_agent.c 41 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502
  1. /*
  2. * Copyright (c) 2009 Red Hat, Inc.
  3. *
  4. * All rights reserved.
  5. *
  6. * Author: Jan Friesse (jfriesse@redhat.com)
  7. *
  8. * This software licensed under BSD license, the text of which follows:
  9. *
  10. * Redistribution and use in source and binary forms, with or without
  11. * modification, are permitted provided that the following conditions are met:
  12. *
  13. * - Redistributions of source code must retain the above copyright notice,
  14. * this list of conditions and the following disclaimer.
  15. * - Redistributions in binary form must reproduce the above copyright notice,
  16. * this list of conditions and the following disclaimer in the documentation
  17. * and/or other materials provided with the distribution.
  18. * - Neither the name of the Red Hat, Inc. nor the names of its
  19. * contributors may be used to endorse or promote products derived from this
  20. * software without specific prior written permission.
  21. *
  22. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  23. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  26. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  27. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  28. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  29. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  30. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  31. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  32. * THE POSSIBILITY OF SUCH DAMAGE.
  33. */
  34. /*
  35. * Provides test of SAM API
  36. */
  37. #include <config.h>
  38. #include <stdio.h>
  39. #include <stdint.h>
  40. #include <stdlib.h>
  41. #include <unistd.h>
  42. #include <pthread.h>
  43. #include <limits.h>
  44. #include <sys/types.h>
  45. #include <sys/socket.h>
  46. #include <netinet/in.h>
  47. #include <arpa/inet.h>
  48. #include <netdb.h>
  49. #include <sys/un.h>
  50. #include <corosync/corotypes.h>
  51. #include <corosync/confdb.h>
  52. #include <corosync/sam.h>
  53. #include <signal.h>
  54. #include <string.h>
  55. #include <syslog.h>
  56. #include <sys/wait.h>
  57. #include "common_test_agent.h"
  58. extern const char *__progname;
  59. static int test2_sig_delivered = 0;
  60. static int test5_hc_cb_count = 0;
  61. static int test6_sig_delivered = 0;
  62. /*
  63. * First test will just register SAM, with policy restart. First instance will
  64. * sleep one second, send hc and sleep another 3 seconds. This should force restart.
  65. * Second instance will sleep one second, send hc, stop hc and sleep 3 seconds.
  66. * Then start hc again and sleep 3 seconds. This should force restart again.
  67. * Last instance just calls initialize again. This should end with error.
  68. * Then call start, followed by stop and start again. Finally, we will call finalize
  69. * twice. One should succeed, second should fail. After this, we will call every function
  70. * (none should succeed).
  71. */
  72. static int test1 (void)
  73. {
  74. cs_error_t error;
  75. unsigned int instance_id;
  76. int i;
  77. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  78. error = sam_initialize (2000, SAM_RECOVERY_POLICY_RESTART);
  79. if (error != CS_OK) {
  80. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", error);
  81. return 1;
  82. }
  83. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  84. error = sam_register (&instance_id);
  85. if (error != CS_OK) {
  86. syslog (LOG_ERR, "Can't register. Error %d\n", error);
  87. return 1;
  88. }
  89. if (instance_id == 1 || instance_id == 2) {
  90. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  91. error = sam_start ();
  92. if (error != CS_OK) {
  93. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  94. return 1;
  95. }
  96. for (i = 0; i < 10; i++) {
  97. syslog (LOG_INFO, "%s iid %d: sleep 1\n", __FUNCTION__, instance_id);
  98. sleep (1);
  99. syslog (LOG_INFO, "%s iid %d: hc send\n", __FUNCTION__, instance_id);
  100. error = sam_hc_send ();
  101. if (error != CS_OK) {
  102. syslog (LOG_ERR, "Can't send hc. Error %d\n", error);
  103. return 1;
  104. }
  105. }
  106. if (instance_id == 2) {
  107. syslog (LOG_INFO, "%s iid %d: stop\n", __FUNCTION__, instance_id);
  108. error = sam_stop ();
  109. if (error != CS_OK) {
  110. syslog (LOG_ERR, "Can't send hc. Error %d\n", error);
  111. return 1;
  112. }
  113. }
  114. syslog (LOG_INFO, "%s iid %d: sleep 3\n", __FUNCTION__, instance_id);
  115. sleep (3);
  116. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  117. error = sam_start ();
  118. if (error != CS_OK) {
  119. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  120. return 1;
  121. }
  122. syslog (LOG_INFO, "%s iid %d: sleep 3\n", __FUNCTION__, instance_id);
  123. sleep (3);
  124. return 0;
  125. }
  126. if (instance_id == 3) {
  127. error = sam_initialize (2000, SAM_RECOVERY_POLICY_RESTART);
  128. if (error == CS_OK) {
  129. syslog (LOG_ERR, "Can initialize SAM API after initialization");
  130. return 1;
  131. }
  132. error = sam_start ();
  133. if (error != CS_OK) {
  134. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  135. return 1;
  136. }
  137. error = sam_stop ();
  138. if (error != CS_OK) {
  139. syslog (LOG_ERR, "Can't stop hc. Error %d\n", error);
  140. return 1;
  141. }
  142. error = sam_finalize ();
  143. if (error != CS_OK) {
  144. syslog (LOG_ERR, "Can't finalize sam. Error %d\n", error);
  145. return 1;
  146. }
  147. error = sam_finalize ();
  148. if (error == CS_OK) {
  149. syslog (LOG_ERR, "Can finalize sam after finalization!\n");
  150. return 1;
  151. }
  152. if (sam_initialize (2, SAM_RECOVERY_POLICY_RESTART) == CS_OK ||
  153. sam_start () == CS_OK || sam_stop () == CS_OK ||
  154. sam_register (NULL) == CS_OK || sam_hc_send () == CS_OK ||
  155. sam_hc_callback_register (NULL) == CS_OK) {
  156. syslog (LOG_ERR, "Can call one of function after finalization!\n");
  157. return 1;
  158. }
  159. return 0;
  160. }
  161. return 1;
  162. }
  163. static void test2_signal (int sig) {
  164. syslog (LOG_INFO, "%s\n", __FUNCTION__);
  165. test2_sig_delivered = 1;
  166. }
  167. /*
  168. * This tests recovery policy quit and callback.
  169. */
  170. static int test2 (void) {
  171. cs_error_t error;
  172. unsigned int instance_id;
  173. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  174. error = sam_initialize (2000, SAM_RECOVERY_POLICY_QUIT);
  175. if (error != CS_OK) {
  176. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", error);
  177. return 1;
  178. }
  179. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  180. error = sam_register (&instance_id);
  181. if (error != CS_OK) {
  182. syslog (LOG_ERR, "Can't register. Error %d\n", error);
  183. return 1;
  184. }
  185. if (instance_id == 1) {
  186. signal (SIGTERM, test2_signal);
  187. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  188. error = sam_start ();
  189. if (error != CS_OK) {
  190. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  191. return 1;
  192. }
  193. syslog (LOG_INFO, "%s iid %d: sleep 1\n", __FUNCTION__, instance_id);
  194. sleep (1);
  195. syslog (LOG_INFO, "%s iid %d: hc send\n", __FUNCTION__, instance_id);
  196. error = sam_hc_send ();
  197. if (error != CS_OK) {
  198. syslog (LOG_ERR, "Can't send hc. Error %d\n", error);
  199. return 1;
  200. }
  201. syslog (LOG_INFO, "%s iid %d: wait for delivery of signal\n", __FUNCTION__, instance_id);
  202. while (!test2_sig_delivered) {
  203. sleep (1);
  204. }
  205. syslog (LOG_INFO, "%s iid %d: wait for real kill\n", __FUNCTION__, instance_id);
  206. sleep (3);
  207. }
  208. return 1;
  209. }
  210. /*
  211. * Test sam_data_store, sam_data_restore and sam_data_getsize
  212. */
  213. static int test4 (void)
  214. {
  215. size_t size;
  216. cs_error_t err;
  217. int i;
  218. unsigned int instance_id;
  219. char saved_data[128];
  220. char saved_data2[128];
  221. syslog (LOG_INFO, "%s: sam_data_getsize 1\n", __FUNCTION__);
  222. err = sam_data_getsize (&size);
  223. if (err != CS_ERR_BAD_HANDLE) {
  224. syslog (LOG_ERR, "Test should return CS_ERR_BAD_HANDLE. Error returned %d\n", err);
  225. return 1;
  226. }
  227. syslog (LOG_INFO, "%s: sam_data_getsize 2\n", __FUNCTION__);
  228. err = sam_data_getsize (NULL);
  229. if (err != CS_ERR_INVALID_PARAM) {
  230. syslog (LOG_ERR, "Test should return CS_ERR_INVALID_PARAM. Error returned %d\n", err);
  231. return 1;
  232. }
  233. syslog (LOG_INFO, "%s: sam_data_store 1\n", __FUNCTION__);
  234. err = sam_data_store (NULL, 0);
  235. if (err != CS_ERR_BAD_HANDLE) {
  236. syslog (LOG_ERR, "Test should return CS_ERR_BAD_HANDLE. Error returned %d\n", err);
  237. return 1;
  238. }
  239. syslog (LOG_INFO, "%s: sam_data_restore 1\n", __FUNCTION__);
  240. err = sam_data_restore (saved_data, sizeof (saved_data));
  241. if (err != CS_ERR_BAD_HANDLE) {
  242. syslog (LOG_ERR, "Test should return CS_ERR_BAD_HANDLE. Error returned %d\n", err);
  243. return 1;
  244. }
  245. syslog (LOG_INFO, "%s: sam_initialize\n", __FUNCTION__);
  246. err = sam_initialize (0, SAM_RECOVERY_POLICY_RESTART);
  247. if (err != CS_OK) {
  248. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  249. return 1;
  250. }
  251. syslog (LOG_INFO, "%s: sam_data_getsize 3\n", __FUNCTION__);
  252. err = sam_data_getsize (&size);
  253. if (err != CS_OK) {
  254. syslog (LOG_ERR, "Test should return CS_ERR_BAD_HANDLE. Error returned %d\n", err);
  255. return 1;
  256. }
  257. if (size != 0) {
  258. syslog (LOG_ERR, "Test should return size of 0. Returned %zx\n", size);
  259. return 1;
  260. }
  261. syslog (LOG_INFO, "%s: sam_data_restore 2\n", __FUNCTION__);
  262. err = sam_data_restore (NULL, sizeof (saved_data));
  263. if (err != CS_ERR_INVALID_PARAM) {
  264. syslog (LOG_ERR, "Test should return CS_ERR_INVALID_PARAM. Error returned %d\n", err);
  265. return 1;
  266. }
  267. /*
  268. * Store some real data
  269. */
  270. for (i = 0; i < sizeof (saved_data); i++) {
  271. saved_data[i] = (char)(i + 5);
  272. }
  273. syslog (LOG_INFO, "%s: sam_data_store 2\n", __FUNCTION__);
  274. err = sam_data_store (saved_data, sizeof (saved_data));
  275. if (err != CS_OK) {
  276. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  277. return 1;
  278. }
  279. syslog (LOG_INFO, "%s: sam_data_getsize 4\n", __FUNCTION__);
  280. err = sam_data_getsize (&size);
  281. if (err != CS_OK) {
  282. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  283. return 1;
  284. }
  285. if (size != sizeof (saved_data)) {
  286. syslog (LOG_ERR, "Test should return size of 0. Returned %zx\n", size);
  287. return 1;
  288. }
  289. syslog (LOG_INFO, "%s: sam_data_restore 3\n", __FUNCTION__);
  290. err = sam_data_restore (saved_data2, sizeof (saved_data2) - 1);
  291. if (err != CS_ERR_INVALID_PARAM) {
  292. syslog (LOG_ERR, "Test should return CS_ERR_INVALID_PARAM. Error returned %d\n", err);
  293. return 1;
  294. }
  295. syslog (LOG_INFO, "%s: sam_data_restore 4\n", __FUNCTION__);
  296. err = sam_data_restore (saved_data2, sizeof (saved_data2));
  297. if (err != CS_OK) {
  298. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  299. return 1;
  300. }
  301. if (memcmp (saved_data, saved_data2, sizeof (saved_data2)) != 0) {
  302. syslog (LOG_ERR, "Retored data are not same\n");
  303. return 1;
  304. }
  305. memset (saved_data2, 0, sizeof (saved_data2));
  306. syslog (LOG_INFO, "%s: sam_data_store 3\n", __FUNCTION__);
  307. err = sam_data_store (NULL, 1);
  308. if (err != CS_OK) {
  309. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  310. return 1;
  311. }
  312. syslog (LOG_INFO, "%s: sam_data_getsize 5\n", __FUNCTION__);
  313. err = sam_data_getsize (&size);
  314. if (err != CS_OK) {
  315. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  316. return 1;
  317. }
  318. if (size != 0) {
  319. syslog (LOG_ERR, "Test should return size of 0. Returned %zx\n", size);
  320. return 1;
  321. }
  322. syslog (LOG_INFO, "%s: sam_data_store 4\n", __FUNCTION__);
  323. err = sam_data_store (saved_data, sizeof (saved_data));
  324. if (err != CS_OK) {
  325. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  326. return 1;
  327. }
  328. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  329. err = sam_register (&instance_id);
  330. if (err != CS_OK) {
  331. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  332. return 1;
  333. }
  334. if (instance_id == 1) {
  335. syslog (LOG_INFO, "%s iid %d: sam_start\n", __FUNCTION__, instance_id);
  336. err = sam_start ();
  337. if (err != CS_OK) {
  338. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  339. return 1;
  340. }
  341. syslog (LOG_INFO, "%s iid %d: sam_data_getsize 6\n", __FUNCTION__, instance_id);
  342. err = sam_data_getsize (&size);
  343. if (err != CS_OK) {
  344. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  345. return 1;
  346. }
  347. if (size != sizeof (saved_data2)) {
  348. syslog (LOG_ERR, "Test should return size of 0. Returned %zx\n", size);
  349. return 1;
  350. }
  351. syslog (LOG_INFO, "%s iid %d: sam_data_restore 5\n", __FUNCTION__, instance_id);
  352. err = sam_data_restore (saved_data2, sizeof (saved_data2));
  353. if (err != CS_OK) {
  354. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  355. return 1;
  356. }
  357. if (memcmp (saved_data, saved_data2, sizeof (saved_data2)) != 0) {
  358. syslog (LOG_ERR, "Retored data are not same\n");
  359. return 1;
  360. }
  361. for (i = 0; i < sizeof (saved_data); i++) {
  362. saved_data[i] = (char)(i - 5);
  363. }
  364. syslog (LOG_INFO, "%s iid %d: sam_data_store 5\n", __FUNCTION__, instance_id);
  365. err = sam_data_store (saved_data, sizeof (saved_data) - 7);
  366. if (err != CS_OK) {
  367. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  368. return 1;
  369. }
  370. exit (1);
  371. }
  372. if (instance_id == 2) {
  373. syslog (LOG_INFO, "%s iid %d: sam_start\n", __FUNCTION__, instance_id);
  374. err = sam_start ();
  375. if (err != CS_OK) {
  376. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  377. return 1;
  378. }
  379. syslog (LOG_INFO, "%s iid %d: sam_data_getsize 7\n", __FUNCTION__, instance_id);
  380. err = sam_data_getsize (&size);
  381. if (err != CS_OK) {
  382. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  383. return 1;
  384. }
  385. if (size != sizeof (saved_data2) - 7) {
  386. syslog (LOG_ERR, "Test should return size of 0. Returned %zx\n", size);
  387. return 1;
  388. }
  389. syslog (LOG_INFO, "%s iid %d: sam_data_restore 6\n", __FUNCTION__, instance_id);
  390. err = sam_data_restore (saved_data2, sizeof (saved_data2));
  391. if (err != CS_OK) {
  392. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  393. return 1;
  394. }
  395. for (i = 0; i < sizeof (saved_data); i++) {
  396. saved_data[i] = (char)(i - 5);
  397. }
  398. if (memcmp (saved_data, saved_data2, sizeof (saved_data2) - 7) != 0) {
  399. syslog (LOG_ERR, "Retored data are not same\n");
  400. return 1;
  401. }
  402. syslog (LOG_INFO, "%s iid %d: sam_data_store 6\n", __FUNCTION__, instance_id);
  403. err = sam_data_store (NULL, 0);
  404. if (err != CS_OK) {
  405. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  406. return 1;
  407. }
  408. exit (1);
  409. }
  410. if (instance_id == 3) {
  411. syslog (LOG_INFO, "%s iid %d: sam_data_getsize 8\n", __FUNCTION__, instance_id);
  412. err = sam_data_getsize (&size);
  413. if (err != CS_OK) {
  414. syslog (LOG_ERR, "Test should return CS_OK. Error returned %d\n", err);
  415. return 1;
  416. }
  417. if (size != 0) {
  418. syslog (LOG_ERR, "Test should return size of 0. Returned %zx\n", size);
  419. return 1;
  420. }
  421. }
  422. return (0);
  423. }
  424. static int test5_hc_cb (void)
  425. {
  426. syslog (LOG_INFO, "%s %d\n", __FUNCTION__, ++test5_hc_cb_count);
  427. sam_data_store (&test5_hc_cb_count, sizeof (test5_hc_cb_count));
  428. if (test5_hc_cb_count > 10)
  429. return 1;
  430. return 0;
  431. }
  432. /*
  433. * Test event driven healtchecking.
  434. */
  435. static int test5 (void)
  436. {
  437. cs_error_t error;
  438. unsigned int instance_id;
  439. int hc_cb_count;
  440. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  441. error = sam_initialize (100, SAM_RECOVERY_POLICY_RESTART);
  442. if (error != CS_OK) {
  443. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", error);
  444. return 1;
  445. }
  446. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  447. error = sam_register (&instance_id);
  448. if (error != CS_OK) {
  449. syslog (LOG_ERR, "Can't register. Error %d\n", error);
  450. return 1;
  451. }
  452. if (instance_id == 1) {
  453. syslog (LOG_INFO, "%s iid %d: hc callback register\n", __FUNCTION__, instance_id);
  454. error = sam_hc_callback_register (test5_hc_cb);
  455. if (error != CS_OK) {
  456. syslog (LOG_ERR, "Can't register hc cb. Error %d\n", error);
  457. return 1;
  458. }
  459. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  460. error = sam_start ();
  461. if (error != CS_OK) {
  462. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  463. return 1;
  464. }
  465. sleep (2);
  466. syslog (LOG_INFO, "%s iid %d: Failed. Wasn't killed.\n", __FUNCTION__, instance_id);
  467. return 1;
  468. }
  469. if (instance_id == 2) {
  470. error = sam_data_restore (&hc_cb_count, sizeof (hc_cb_count));
  471. if (error != CS_OK) {
  472. syslog (LOG_ERR, "sam_data_restore should return CS_OK. Error returned %d\n", error);
  473. return 1;
  474. }
  475. if (hc_cb_count != 11) {
  476. syslog (LOG_ERR, "%s iid %d: Premature killed. hc_cb_count should be 11 and it is %d\n",
  477. __FUNCTION__, instance_id - 1, hc_cb_count);
  478. return 1;
  479. }
  480. return 0;
  481. }
  482. return 1;
  483. }
  484. static void test6_signal (int sig) {
  485. cs_error_t error;
  486. syslog (LOG_INFO, "%s\n", __FUNCTION__);
  487. test6_sig_delivered++;
  488. if ((error = sam_data_store (&test6_sig_delivered, sizeof (test6_sig_delivered))) != CS_OK) {
  489. syslog (LOG_ERR, "Can't store data! Error : %d\n", error);
  490. }
  491. }
  492. /*
  493. * Test warn signal set.
  494. */
  495. static int test6 (void) {
  496. cs_error_t error;
  497. unsigned int instance_id;
  498. int test6_sig_del;
  499. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  500. error = sam_initialize (2000, SAM_RECOVERY_POLICY_RESTART);
  501. if (error != CS_OK) {
  502. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", error);
  503. return 1;
  504. }
  505. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  506. error = sam_register (&instance_id);
  507. if (error != CS_OK) {
  508. syslog (LOG_ERR, "Can't register. Error %d\n", error);
  509. return 1;
  510. }
  511. if (instance_id == 1) {
  512. error = sam_warn_signal_set (SIGUSR1);
  513. if (error != CS_OK) {
  514. syslog (LOG_ERR, "Can't set warn signal. Error %d\n", error);
  515. return 1;
  516. }
  517. signal (SIGUSR1, test6_signal);
  518. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  519. error = sam_start ();
  520. if (error != CS_OK) {
  521. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  522. return 1;
  523. }
  524. syslog (LOG_INFO, "%s iid %d: sleep 1\n", __FUNCTION__, instance_id);
  525. sleep (1);
  526. syslog (LOG_INFO, "%s iid %d: hc send\n", __FUNCTION__, instance_id);
  527. error = sam_hc_send ();
  528. if (error != CS_OK) {
  529. syslog (LOG_ERR, "Can't send hc. Error %d\n", error);
  530. return 1;
  531. }
  532. syslog (LOG_INFO, "%s iid %d: wait for delivery of signal\n", __FUNCTION__, instance_id);
  533. while (!test6_sig_delivered) {
  534. sleep (1);
  535. }
  536. syslog (LOG_INFO, "%s iid %d: wait for real kill\n", __FUNCTION__, instance_id);
  537. sleep (3);
  538. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  539. return (1);
  540. }
  541. if (instance_id == 2) {
  542. error = sam_data_restore (&test6_sig_del, sizeof (test6_sig_del));
  543. if (error != CS_OK) {
  544. syslog (LOG_ERR, "Can't restore data. Error %d\n", error);
  545. return 1;
  546. }
  547. if (test6_sig_del != 1) {
  548. syslog (LOG_ERR, "Previous test failed. Signal was not delivered\n");
  549. return 1;
  550. }
  551. error = sam_warn_signal_set (SIGKILL);
  552. if (error != CS_OK) {
  553. syslog (LOG_ERR, "Can't set warn signal. Error %d\n", error);
  554. return 1;
  555. }
  556. signal (SIGUSR1, test6_signal);
  557. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  558. error = sam_start ();
  559. if (error != CS_OK) {
  560. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  561. return 1;
  562. }
  563. syslog (LOG_INFO, "%s iid %d: sleep 1\n", __FUNCTION__, instance_id);
  564. sleep (1);
  565. syslog (LOG_INFO, "%s iid %d: hc send\n", __FUNCTION__, instance_id);
  566. error = sam_hc_send ();
  567. if (error != CS_OK) {
  568. syslog (LOG_ERR, "Can't send hc. Error %d\n", error);
  569. return 1;
  570. }
  571. syslog (LOG_INFO, "%s iid %d: wait for delivery of signal\n", __FUNCTION__, instance_id);
  572. while (!test6_sig_delivered) {
  573. sleep (1);
  574. }
  575. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  576. return (1);
  577. }
  578. if (instance_id == 3) {
  579. error = sam_data_restore (&test6_sig_del, sizeof (test6_sig_del));
  580. if (error != CS_OK) {
  581. syslog (LOG_ERR, "Can't restore data. Error %d\n", error);
  582. return 1;
  583. }
  584. if (test6_sig_del != 1) {
  585. syslog (LOG_ERR, "Previous test failed. Signal WAS delivered\n");
  586. return 1;
  587. }
  588. return (0);
  589. }
  590. return 1;
  591. }
  592. static void *test7_thread (void *arg)
  593. {
  594. /* Wait 5s */
  595. sleep (5);
  596. exit (0);
  597. }
  598. /*
  599. * Test quorum
  600. */
  601. static int test_quorum (void) {
  602. confdb_handle_t cdb_handle;
  603. cs_error_t err;
  604. hdb_handle_t quorum_handle;
  605. size_t value_len;
  606. char key_value[256];
  607. unsigned int instance_id;
  608. pthread_t kill_thread;
  609. err = confdb_initialize (&cdb_handle, NULL);
  610. if (err != CS_OK) {
  611. syslog (LOG_INFO, "Could not initialize Cluster Configuration Database API instance error %d. Test skipped\n", err);
  612. return (1);
  613. }
  614. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  615. if (err != CS_OK) {
  616. syslog (LOG_INFO, "Could not start object_find %d. Test skipped\n", err);
  617. return (1);
  618. }
  619. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "quorum", strlen("quorum"), &quorum_handle);
  620. if (err != CS_OK) {
  621. syslog (LOG_INFO, "Could not object_find \"quorum\": %d. Test skipped\n", err);
  622. return (1);
  623. }
  624. err = confdb_key_get(cdb_handle, quorum_handle, "provider", strlen("provider"), key_value, &value_len);
  625. if (err != CS_OK) {
  626. syslog (LOG_INFO, "Could not get \"provider\" key: %d. Test skipped\n", err);
  627. return (1);
  628. }
  629. if (!(value_len - 1 == strlen ("testquorum") && memcmp (key_value, "testquorum", value_len - 1) == 0)) {
  630. syslog (LOG_INFO, "Provider is not testquorum. Test skipped\n");
  631. return (1);
  632. }
  633. /*
  634. * Set to not quorate
  635. */
  636. err = confdb_key_create(cdb_handle, quorum_handle, "quorate", strlen("quorate"), "0", strlen("0"));
  637. if (err != CS_OK) {
  638. syslog (LOG_INFO, "Can't create confdb key. Error %d\n", err);
  639. return (2);
  640. }
  641. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  642. err = sam_initialize (2000, SAM_RECOVERY_POLICY_QUORUM_RESTART);
  643. if (err != CS_OK) {
  644. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  645. return 2;
  646. }
  647. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  648. err = sam_register (&instance_id);
  649. if (err != CS_OK) {
  650. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  651. return 2;
  652. }
  653. if (instance_id == 1) {
  654. /*
  655. * Sam start should block forever, but 10s for us should be enough
  656. */
  657. pthread_create (&kill_thread, NULL, test7_thread, NULL);
  658. syslog (LOG_INFO, "%s iid %d: start - should block forever (waiting 5s)\n", __FUNCTION__, instance_id);
  659. err = sam_start ();
  660. if (err != CS_OK) {
  661. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  662. return 2;
  663. }
  664. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  665. return (2);
  666. }
  667. if (instance_id == 2) {
  668. /*
  669. * Set to quorate
  670. */
  671. err = confdb_key_create(cdb_handle, quorum_handle, "quorate", strlen("quorate"), "1", strlen("1"));
  672. if (err != CS_OK) {
  673. syslog (LOG_INFO, "Can't create confdb key. Error %d\n", err);
  674. return (2);
  675. }
  676. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  677. err = sam_start ();
  678. if (err != CS_OK) {
  679. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  680. return 2;
  681. }
  682. /*
  683. * Set corosync unquorate
  684. */
  685. err = confdb_key_create(cdb_handle, quorum_handle, "quorate", strlen("quorate"), "0", strlen("0"));
  686. if (err != CS_OK) {
  687. syslog (LOG_INFO, "Can't create confdb key. Error %d\n", err);
  688. return (2);
  689. }
  690. syslog (LOG_INFO, "%s iid %d: sleep 3\n", __FUNCTION__, instance_id);
  691. sleep (3);
  692. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  693. return (2);
  694. }
  695. if (instance_id == 3) {
  696. return (0);
  697. }
  698. return (2);
  699. }
  700. /*
  701. * Test confdb integration + quit policy
  702. */
  703. static int test8 (pid_t pid, pid_t old_pid, int test_n) {
  704. confdb_handle_t cdb_handle;
  705. cs_error_t err;
  706. hdb_handle_t res_handle, proc_handle, pid_handle;
  707. size_t value_len;
  708. uint64_t tstamp1, tstamp2;
  709. int32_t msec_diff;
  710. char key_value[256];
  711. unsigned int instance_id;
  712. char tmp_obj[PATH_MAX];
  713. confdb_value_types_t cdbtype;
  714. err = confdb_initialize (&cdb_handle, NULL);
  715. if (err != CS_OK) {
  716. syslog (LOG_INFO, "Could not initialize Cluster Configuration Database API instance error %d. Test skipped\n", err);
  717. return (1);
  718. }
  719. syslog (LOG_INFO, "%s test %d\n", __FUNCTION__, test_n);
  720. if (test_n == 2) {
  721. /*
  722. * Object should not exist
  723. */
  724. syslog (LOG_INFO, "%s Testing if object exists (it shouldn't)\n", __FUNCTION__);
  725. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  726. if (err != CS_OK) {
  727. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  728. return (2);
  729. }
  730. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  731. if (err != CS_OK) {
  732. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  733. return (2);
  734. }
  735. err = confdb_object_find_start(cdb_handle, res_handle);
  736. if (err != CS_OK) {
  737. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  738. return (2);
  739. }
  740. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  741. if (err != CS_OK) {
  742. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  743. return (2);
  744. }
  745. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  746. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  747. }
  748. err = confdb_object_find_start(cdb_handle, proc_handle);
  749. if (err != CS_OK) {
  750. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  751. return (2);
  752. }
  753. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  754. if (err == CS_OK) {
  755. syslog (LOG_INFO, "Could find object \"%s\": %d.\n", tmp_obj, err);
  756. return (2);
  757. }
  758. }
  759. if (test_n == 1 || test_n == 2) {
  760. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  761. err = sam_initialize (2000, SAM_RECOVERY_POLICY_QUIT | SAM_RECOVERY_POLICY_CONFDB);
  762. if (err != CS_OK) {
  763. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  764. return 2;
  765. }
  766. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  767. err = sam_register (&instance_id);
  768. if (err != CS_OK) {
  769. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  770. return 2;
  771. }
  772. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  773. if (err != CS_OK) {
  774. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  775. return (2);
  776. }
  777. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  778. if (err != CS_OK) {
  779. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  780. return (2);
  781. }
  782. err = confdb_object_find_start(cdb_handle, res_handle);
  783. if (err != CS_OK) {
  784. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  785. return (2);
  786. }
  787. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  788. if (err != CS_OK) {
  789. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  790. return (2);
  791. }
  792. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  793. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  794. }
  795. err = confdb_object_find_start(cdb_handle, proc_handle);
  796. if (err != CS_OK) {
  797. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  798. return (2);
  799. }
  800. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  801. if (err != CS_OK) {
  802. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  803. return (2);
  804. }
  805. err = confdb_key_get(cdb_handle, pid_handle, "recovery", strlen("recovery"), key_value, &value_len);
  806. if (err != CS_OK) {
  807. syslog (LOG_INFO, "Could not get \"recovery\" key: %d.\n", err);
  808. return (2);
  809. }
  810. if (value_len != strlen ("quit") || memcmp (key_value, "quit", value_len) != 0) {
  811. syslog (LOG_INFO, "Recovery key \"%s\" is not \"watchdog\".\n", key_value);
  812. return (2);
  813. }
  814. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  815. if (err != CS_OK) {
  816. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  817. return (2);
  818. }
  819. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  820. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  821. return (2);
  822. }
  823. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  824. err = sam_start ();
  825. if (err != CS_OK) {
  826. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  827. return 2;
  828. }
  829. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  830. if (err != CS_OK) {
  831. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  832. return (2);
  833. }
  834. if (value_len != strlen ("running") || memcmp (key_value, "running", value_len) != 0) {
  835. syslog (LOG_INFO, "State key is not \"running\".\n");
  836. return (2);
  837. }
  838. syslog (LOG_INFO, "%s iid %d: stop\n", __FUNCTION__, instance_id);
  839. err = sam_stop ();
  840. if (err != CS_OK) {
  841. syslog (LOG_ERR, "Can't stop hc. Error %d\n", err);
  842. return 2;
  843. }
  844. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  845. if (err != CS_OK) {
  846. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  847. return (2);
  848. }
  849. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  850. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  851. return (2);
  852. }
  853. syslog (LOG_INFO, "%s iid %d: sleeping 5\n", __FUNCTION__, instance_id);
  854. sleep (5);
  855. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  856. if (err != CS_OK) {
  857. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  858. return (2);
  859. }
  860. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  861. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  862. return (2);
  863. }
  864. syslog (LOG_INFO, "%s iid %d: start 2\n", __FUNCTION__, instance_id);
  865. err = sam_start ();
  866. if (err != CS_OK) {
  867. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  868. return 2;
  869. }
  870. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  871. if (err != CS_OK) {
  872. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  873. return (2);
  874. }
  875. if (value_len != strlen ("running") || memcmp (key_value, "running", value_len) != 0) {
  876. syslog (LOG_INFO, "State key is not \"running\".\n");
  877. return (2);
  878. }
  879. if (test_n == 2) {
  880. syslog (LOG_INFO, "%s iid %d: sleeping 5. Should be killed\n", __FUNCTION__, instance_id);
  881. sleep (5);
  882. return (2);
  883. } else {
  884. syslog (LOG_INFO, "%s iid %d: Test HC\n", __FUNCTION__, instance_id);
  885. err = sam_hc_send ();
  886. if (err != CS_OK) {
  887. syslog (LOG_ERR, "Can't send hc. Error %d\n", err);
  888. return 2;
  889. }
  890. err = confdb_key_get_typed (cdb_handle, pid_handle, "last_updated", &tstamp1, &value_len, &cdbtype);
  891. if (err != CS_OK) {
  892. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  893. return (2);
  894. }
  895. syslog (LOG_INFO, "%s iid %d: Sleep 1\n", __FUNCTION__, instance_id);
  896. sleep (1);
  897. err = sam_hc_send ();
  898. if (err != CS_OK) {
  899. syslog (LOG_ERR, "Can't send hc. Error %d\n", err);
  900. return 2;
  901. }
  902. sleep (1);
  903. err = confdb_key_get_typed (cdb_handle, pid_handle, "last_updated", &tstamp2, &value_len, &cdbtype);
  904. if (err != CS_OK) {
  905. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  906. return (2);
  907. }
  908. msec_diff = (tstamp2 - tstamp1)/CS_TIME_NS_IN_MSEC;
  909. if (msec_diff < 500 || msec_diff > 2000) {
  910. syslog (LOG_INFO, "Difference %d is not within <500, 2000> interval.\n", msec_diff);
  911. return (2);
  912. }
  913. syslog (LOG_INFO, "%s iid %d: stop 2\n", __FUNCTION__, instance_id);
  914. err = sam_stop ();
  915. if (err != CS_OK) {
  916. syslog (LOG_ERR, "Can't stop hc. Error %d\n", err);
  917. return 2;
  918. }
  919. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  920. if (err != CS_OK) {
  921. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  922. return (2);
  923. }
  924. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  925. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  926. return (2);
  927. }
  928. syslog (LOG_INFO, "%s iid %d: exiting\n", __FUNCTION__, instance_id);
  929. return (0);
  930. }
  931. }
  932. if (test_n == 3) {
  933. syslog (LOG_INFO, "%s Testing if status is failed\n", __FUNCTION__);
  934. /*
  935. * Previous should be FAILED
  936. */
  937. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  938. if (err != CS_OK) {
  939. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  940. return (2);
  941. }
  942. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  943. if (err != CS_OK) {
  944. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  945. return (2);
  946. }
  947. err = confdb_object_find_start(cdb_handle, res_handle);
  948. if (err != CS_OK) {
  949. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  950. return (2);
  951. }
  952. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  953. if (err != CS_OK) {
  954. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  955. return (2);
  956. }
  957. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  958. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  959. }
  960. err = confdb_object_find_start(cdb_handle, proc_handle);
  961. if (err != CS_OK) {
  962. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  963. return (2);
  964. }
  965. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  966. if (err != CS_OK) {
  967. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  968. return (2);
  969. }
  970. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  971. if (err != CS_OK) {
  972. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  973. return (2);
  974. }
  975. if (value_len != strlen ("failed") || memcmp (key_value, "failed", value_len) != 0) {
  976. syslog (LOG_INFO, "State key is not \"failed\".\n");
  977. return (2);
  978. }
  979. return (0);
  980. }
  981. return (2);
  982. }
  983. /*
  984. * Test confdb integration + restart policy
  985. */
  986. static int test9 (pid_t pid, pid_t old_pid, int test_n) {
  987. confdb_handle_t cdb_handle;
  988. cs_error_t err;
  989. hdb_handle_t res_handle, proc_handle, pid_handle;
  990. size_t value_len;
  991. char key_value[256];
  992. unsigned int instance_id;
  993. char tmp_obj[PATH_MAX];
  994. err = confdb_initialize (&cdb_handle, NULL);
  995. if (err != CS_OK) {
  996. syslog (LOG_INFO, "Could not initialize Cluster Configuration Database API instance error %d. Test skipped\n", err);
  997. return (1);
  998. }
  999. syslog (LOG_INFO, "%s test %d\n", __FUNCTION__, test_n);
  1000. if (test_n == 1) {
  1001. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  1002. err = sam_initialize (2000, SAM_RECOVERY_POLICY_RESTART | SAM_RECOVERY_POLICY_CONFDB);
  1003. if (err != CS_OK) {
  1004. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  1005. return 2;
  1006. }
  1007. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  1008. err = sam_register (&instance_id);
  1009. if (err != CS_OK) {
  1010. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  1011. return 2;
  1012. }
  1013. syslog (LOG_INFO, "%s: iid %d\n", __FUNCTION__, instance_id);
  1014. if (instance_id < 3) {
  1015. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  1016. if (err != CS_OK) {
  1017. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1018. return (2);
  1019. }
  1020. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"),
  1021. &res_handle);
  1022. if (err != CS_OK) {
  1023. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  1024. return (2);
  1025. }
  1026. err = confdb_object_find_start(cdb_handle, res_handle);
  1027. if (err != CS_OK) {
  1028. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1029. return (2);
  1030. }
  1031. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  1032. if (err != CS_OK) {
  1033. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  1034. return (2);
  1035. }
  1036. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  1037. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  1038. }
  1039. err = confdb_object_find_start(cdb_handle, proc_handle);
  1040. if (err != CS_OK) {
  1041. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1042. return (2);
  1043. }
  1044. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  1045. if (err != CS_OK) {
  1046. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  1047. return (2);
  1048. }
  1049. err = confdb_key_get(cdb_handle, pid_handle, "recovery", strlen("recovery"), key_value, &value_len);
  1050. if (err != CS_OK) {
  1051. syslog (LOG_INFO, "Could not get \"recovery\" key: %d.\n", err);
  1052. return (2);
  1053. }
  1054. if (value_len != strlen ("restart") || memcmp (key_value, "restart", value_len) != 0) {
  1055. syslog (LOG_INFO, "Recovery key \"%s\" is not \"restart\".\n", key_value);
  1056. return (2);
  1057. }
  1058. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  1059. if (err != CS_OK) {
  1060. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  1061. return (2);
  1062. }
  1063. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  1064. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  1065. return (2);
  1066. }
  1067. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  1068. err = sam_start ();
  1069. if (err != CS_OK) {
  1070. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  1071. return 2;
  1072. }
  1073. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  1074. if (err != CS_OK) {
  1075. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  1076. return (2);
  1077. }
  1078. if (value_len != strlen ("running") || memcmp (key_value, "running", value_len) != 0) {
  1079. syslog (LOG_INFO, "State key is not \"running\".\n");
  1080. return (2);
  1081. }
  1082. syslog (LOG_INFO, "%s iid %d: waiting for kill\n", __FUNCTION__, instance_id);
  1083. sleep (10);
  1084. return (2);
  1085. }
  1086. if (instance_id == 3) {
  1087. syslog (LOG_INFO, "%s iid %d: mark failed\n", __FUNCTION__, instance_id);
  1088. err = sam_mark_failed ();
  1089. if (err != CS_OK) {
  1090. syslog (LOG_ERR, "Can't mark failed. Error %d\n", err);
  1091. return 2;
  1092. }
  1093. sleep (10);
  1094. return (2);
  1095. }
  1096. return (2);
  1097. }
  1098. if (test_n == 2) {
  1099. syslog (LOG_INFO, "%s Testing if status is failed\n", __FUNCTION__);
  1100. /*
  1101. * Previous should be FAILED
  1102. */
  1103. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  1104. if (err != CS_OK) {
  1105. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1106. return (2);
  1107. }
  1108. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  1109. if (err != CS_OK) {
  1110. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  1111. return (2);
  1112. }
  1113. err = confdb_object_find_start(cdb_handle, res_handle);
  1114. if (err != CS_OK) {
  1115. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1116. return (2);
  1117. }
  1118. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  1119. if (err != CS_OK) {
  1120. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  1121. return (2);
  1122. }
  1123. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  1124. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  1125. }
  1126. err = confdb_object_find_start(cdb_handle, proc_handle);
  1127. if (err != CS_OK) {
  1128. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1129. return (2);
  1130. }
  1131. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  1132. if (err != CS_OK) {
  1133. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  1134. return (2);
  1135. }
  1136. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  1137. if (err != CS_OK) {
  1138. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  1139. return (2);
  1140. }
  1141. if (value_len != strlen ("failed") || memcmp (key_value, "failed", value_len) != 0) {
  1142. syslog (LOG_INFO, "State key is not \"failed\".\n");
  1143. return (2);
  1144. }
  1145. return (0);
  1146. }
  1147. return (2);
  1148. }
  1149. static int hc_allways_respond_cb(void)
  1150. {
  1151. syslog (LOG_INFO, "%s() -> health check OK.", __FUNCTION__);
  1152. return 0;
  1153. }
  1154. static int setup_hc (void)
  1155. {
  1156. cs_error_t err;
  1157. unsigned int instance_id;
  1158. err = sam_initialize (1000, SAM_RECOVERY_POLICY_QUIT | SAM_RECOVERY_POLICY_CONFDB);
  1159. if (err != CS_OK) {
  1160. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  1161. return 2;
  1162. }
  1163. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  1164. err = sam_register (&instance_id);
  1165. if (err != CS_OK) {
  1166. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  1167. return 2;
  1168. }
  1169. err = sam_hc_callback_register (hc_allways_respond_cb);
  1170. syslog (LOG_INFO, "%s instance id %d: start\n", __FUNCTION__, instance_id);
  1171. err = sam_start ();
  1172. if (err != CS_OK) {
  1173. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  1174. return 2;
  1175. }
  1176. return (0);
  1177. }
  1178. static int do_test_command(int sock, char* func, char*args[], int num_args)
  1179. {
  1180. int err = 0;
  1181. pid_t pid;
  1182. int stat;
  1183. pid = fork ();
  1184. if (pid == -1) {
  1185. syslog (LOG_ERR, "Can't fork\n");
  1186. return -1;
  1187. }
  1188. if (pid == 0) {
  1189. if (strcmp ("test1", func) == 0) {
  1190. err = test1 ();
  1191. } else if (strcmp ("test2", func) == 0) {
  1192. err = test2 ();
  1193. } else if (strcmp ("test4", func) == 0) {
  1194. err = test4 ();
  1195. } else if (strcmp ("test5", func) == 0) {
  1196. err = test5 ();
  1197. } else if (strcmp ("test6", func) == 0) {
  1198. err = test6 ();
  1199. } else if (strcmp ("test_quorum", func) == 0) {
  1200. err = test_quorum ();
  1201. } else if (strcmp ("test8", func) == 0) {
  1202. err = test8 (getpid(), 0, 1);
  1203. } else if (strcmp ("test9", func) == 0) {
  1204. err = test9 (getpid(), 0, 1);
  1205. }
  1206. sam_finalize ();
  1207. exit(err);
  1208. }
  1209. if (pid > 0) {
  1210. waitpid (pid, &stat, 0);
  1211. return WEXITSTATUS (stat);
  1212. }
  1213. return -1;
  1214. }
  1215. static void do_command (int sock, char* func, char*args[], int num_args)
  1216. {
  1217. char response[100];
  1218. int err = 0;
  1219. if (parse_debug) {
  1220. syslog (LOG_INFO, "RPC:%s() called.", func);
  1221. }
  1222. if (strncmp ("test", func, 4) == 0) {
  1223. err = do_test_command(sock, func, args, num_args);
  1224. } else if (strcmp ("setup_hc", func) == 0) {
  1225. err = setup_hc ();
  1226. } else if (strcmp ("sam_stop", func) == 0) {
  1227. sam_stop ();
  1228. sam_finalize();
  1229. } else {
  1230. err = -1;
  1231. syslog (LOG_ERR,"%s RPC:%s not supported!", __func__, func);
  1232. snprintf (response, 100, "%s", NOT_SUPPORTED_STR);
  1233. }
  1234. if (err == 0) {
  1235. snprintf (response, 100, "%s", OK_STR);
  1236. } else if (err == 1) {
  1237. snprintf (response, 100, "%s", FAIL_STR);
  1238. syslog (LOG_ERR, "%s() test skipped?! (%d).", func, err);
  1239. } else {
  1240. snprintf (response, 100, "%s", FAIL_STR);
  1241. syslog (LOG_ERR, "%s() failed (%d).", func, err);
  1242. }
  1243. send (sock, response, strlen (response) + 1, 0);
  1244. }
  1245. int main (int argc, char *argv[])
  1246. {
  1247. int ret;
  1248. openlog (NULL, LOG_CONS|LOG_PID, LOG_DAEMON);
  1249. syslog (LOG_INFO, "%s STARTNG", __FILE__);
  1250. parse_debug = 1;
  1251. ret = test_agent_run (9036, do_command, NULL);
  1252. syslog (LOG_INFO, "%s EXITING", __FILE__);
  1253. return ret;
  1254. }