sam_test_agent.c 41 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507
  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. cs_error_t res;
  427. syslog (LOG_INFO, "%s %d\n", __FUNCTION__, ++test5_hc_cb_count);
  428. res = sam_data_store (&test5_hc_cb_count, sizeof (test5_hc_cb_count));
  429. if (res != CS_OK)
  430. return 1;
  431. if (test5_hc_cb_count > 10)
  432. return 1;
  433. return 0;
  434. }
  435. /*
  436. * Test event driven healtchecking.
  437. */
  438. static int test5 (void)
  439. {
  440. cs_error_t error;
  441. unsigned int instance_id;
  442. int hc_cb_count;
  443. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  444. error = sam_initialize (100, SAM_RECOVERY_POLICY_RESTART);
  445. if (error != CS_OK) {
  446. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", error);
  447. return 1;
  448. }
  449. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  450. error = sam_register (&instance_id);
  451. if (error != CS_OK) {
  452. syslog (LOG_ERR, "Can't register. Error %d\n", error);
  453. return 1;
  454. }
  455. if (instance_id == 1) {
  456. syslog (LOG_INFO, "%s iid %d: hc callback register\n", __FUNCTION__, instance_id);
  457. error = sam_hc_callback_register (test5_hc_cb);
  458. if (error != CS_OK) {
  459. syslog (LOG_ERR, "Can't register hc cb. Error %d\n", error);
  460. return 1;
  461. }
  462. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  463. error = sam_start ();
  464. if (error != CS_OK) {
  465. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  466. return 1;
  467. }
  468. sleep (2);
  469. syslog (LOG_INFO, "%s iid %d: Failed. Wasn't killed.\n", __FUNCTION__, instance_id);
  470. return 1;
  471. }
  472. if (instance_id == 2) {
  473. error = sam_data_restore (&hc_cb_count, sizeof (hc_cb_count));
  474. if (error != CS_OK) {
  475. syslog (LOG_ERR, "sam_data_restore should return CS_OK. Error returned %d\n", error);
  476. return 1;
  477. }
  478. if (hc_cb_count != 11) {
  479. syslog (LOG_ERR, "%s iid %d: Premature killed. hc_cb_count should be 11 and it is %d\n",
  480. __FUNCTION__, instance_id - 1, hc_cb_count);
  481. return 1;
  482. }
  483. return 0;
  484. }
  485. return 1;
  486. }
  487. static void test6_signal (int sig) {
  488. cs_error_t error;
  489. syslog (LOG_INFO, "%s\n", __FUNCTION__);
  490. test6_sig_delivered++;
  491. if ((error = sam_data_store (&test6_sig_delivered, sizeof (test6_sig_delivered))) != CS_OK) {
  492. syslog (LOG_ERR, "Can't store data! Error : %d\n", error);
  493. }
  494. }
  495. /*
  496. * Test warn signal set.
  497. */
  498. static int test6 (void) {
  499. cs_error_t error;
  500. unsigned int instance_id;
  501. int test6_sig_del;
  502. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  503. error = sam_initialize (2000, SAM_RECOVERY_POLICY_RESTART);
  504. if (error != CS_OK) {
  505. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", error);
  506. return 1;
  507. }
  508. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  509. error = sam_register (&instance_id);
  510. if (error != CS_OK) {
  511. syslog (LOG_ERR, "Can't register. Error %d\n", error);
  512. return 1;
  513. }
  514. if (instance_id == 1) {
  515. error = sam_warn_signal_set (SIGUSR1);
  516. if (error != CS_OK) {
  517. syslog (LOG_ERR, "Can't set warn signal. Error %d\n", error);
  518. return 1;
  519. }
  520. signal (SIGUSR1, test6_signal);
  521. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  522. error = sam_start ();
  523. if (error != CS_OK) {
  524. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  525. return 1;
  526. }
  527. syslog (LOG_INFO, "%s iid %d: sleep 1\n", __FUNCTION__, instance_id);
  528. sleep (1);
  529. syslog (LOG_INFO, "%s iid %d: hc send\n", __FUNCTION__, instance_id);
  530. error = sam_hc_send ();
  531. if (error != CS_OK) {
  532. syslog (LOG_ERR, "Can't send hc. Error %d\n", error);
  533. return 1;
  534. }
  535. syslog (LOG_INFO, "%s iid %d: wait for delivery of signal\n", __FUNCTION__, instance_id);
  536. while (!test6_sig_delivered) {
  537. sleep (1);
  538. }
  539. syslog (LOG_INFO, "%s iid %d: wait for real kill\n", __FUNCTION__, instance_id);
  540. sleep (3);
  541. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  542. return (1);
  543. }
  544. if (instance_id == 2) {
  545. error = sam_data_restore (&test6_sig_del, sizeof (test6_sig_del));
  546. if (error != CS_OK) {
  547. syslog (LOG_ERR, "Can't restore data. Error %d\n", error);
  548. return 1;
  549. }
  550. if (test6_sig_del != 1) {
  551. syslog (LOG_ERR, "Previous test failed. Signal was not delivered\n");
  552. return 1;
  553. }
  554. error = sam_warn_signal_set (SIGKILL);
  555. if (error != CS_OK) {
  556. syslog (LOG_ERR, "Can't set warn signal. Error %d\n", error);
  557. return 1;
  558. }
  559. signal (SIGUSR1, test6_signal);
  560. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  561. error = sam_start ();
  562. if (error != CS_OK) {
  563. syslog (LOG_ERR, "Can't start hc. Error %d\n", error);
  564. return 1;
  565. }
  566. syslog (LOG_INFO, "%s iid %d: sleep 1\n", __FUNCTION__, instance_id);
  567. sleep (1);
  568. syslog (LOG_INFO, "%s iid %d: hc send\n", __FUNCTION__, instance_id);
  569. error = sam_hc_send ();
  570. if (error != CS_OK) {
  571. syslog (LOG_ERR, "Can't send hc. Error %d\n", error);
  572. return 1;
  573. }
  574. syslog (LOG_INFO, "%s iid %d: wait for delivery of signal\n", __FUNCTION__, instance_id);
  575. while (!test6_sig_delivered) {
  576. sleep (1);
  577. }
  578. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  579. return (1);
  580. }
  581. if (instance_id == 3) {
  582. error = sam_data_restore (&test6_sig_del, sizeof (test6_sig_del));
  583. if (error != CS_OK) {
  584. syslog (LOG_ERR, "Can't restore data. Error %d\n", error);
  585. return 1;
  586. }
  587. if (test6_sig_del != 1) {
  588. syslog (LOG_ERR, "Previous test failed. Signal WAS delivered\n");
  589. return 1;
  590. }
  591. return (0);
  592. }
  593. return 1;
  594. }
  595. static void *test7_thread (void *arg)
  596. {
  597. /* Wait 5s */
  598. sleep (5);
  599. exit (0);
  600. }
  601. /*
  602. * Test quorum
  603. */
  604. static int test_quorum (void) {
  605. confdb_handle_t cdb_handle;
  606. cs_error_t err;
  607. hdb_handle_t quorum_handle;
  608. size_t value_len;
  609. char key_value[256];
  610. unsigned int instance_id;
  611. pthread_t kill_thread;
  612. err = confdb_initialize (&cdb_handle, NULL);
  613. if (err != CS_OK) {
  614. syslog (LOG_INFO, "Could not initialize Cluster Configuration Database API instance error %d. Test skipped\n", err);
  615. return (1);
  616. }
  617. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  618. if (err != CS_OK) {
  619. syslog (LOG_INFO, "Could not start object_find %d. Test skipped\n", err);
  620. return (1);
  621. }
  622. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "quorum", strlen("quorum"), &quorum_handle);
  623. if (err != CS_OK) {
  624. syslog (LOG_INFO, "Could not object_find \"quorum\": %d. Test skipped\n", err);
  625. return (1);
  626. }
  627. err = confdb_key_get(cdb_handle, quorum_handle, "provider", strlen("provider"), key_value, &value_len);
  628. if (err != CS_OK) {
  629. syslog (LOG_INFO, "Could not get \"provider\" key: %d. Test skipped\n", err);
  630. return (1);
  631. }
  632. if (!(value_len - 1 == strlen ("testquorum") && memcmp (key_value, "testquorum", value_len - 1) == 0)) {
  633. syslog (LOG_INFO, "Provider is not testquorum. Test skipped\n");
  634. return (1);
  635. }
  636. /*
  637. * Set to not quorate
  638. */
  639. err = confdb_key_create(cdb_handle, quorum_handle, "quorate", strlen("quorate"), "0", strlen("0"));
  640. if (err != CS_OK) {
  641. syslog (LOG_INFO, "Can't create confdb key. Error %d\n", err);
  642. return (2);
  643. }
  644. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  645. err = sam_initialize (2000, SAM_RECOVERY_POLICY_QUORUM_RESTART);
  646. if (err != CS_OK) {
  647. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  648. return 2;
  649. }
  650. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  651. err = sam_register (&instance_id);
  652. if (err != CS_OK) {
  653. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  654. return 2;
  655. }
  656. if (instance_id == 1) {
  657. /*
  658. * Sam start should block forever, but 10s for us should be enough
  659. */
  660. pthread_create (&kill_thread, NULL, test7_thread, NULL);
  661. syslog (LOG_INFO, "%s iid %d: start - should block forever (waiting 5s)\n", __FUNCTION__, instance_id);
  662. err = sam_start ();
  663. if (err != CS_OK) {
  664. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  665. return 2;
  666. }
  667. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  668. return (2);
  669. }
  670. if (instance_id == 2) {
  671. /*
  672. * Set to quorate
  673. */
  674. err = confdb_key_create(cdb_handle, quorum_handle, "quorate", strlen("quorate"), "1", strlen("1"));
  675. if (err != CS_OK) {
  676. syslog (LOG_INFO, "Can't create confdb key. Error %d\n", err);
  677. return (2);
  678. }
  679. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  680. err = sam_start ();
  681. if (err != CS_OK) {
  682. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  683. return 2;
  684. }
  685. /*
  686. * Set corosync unquorate
  687. */
  688. err = confdb_key_create(cdb_handle, quorum_handle, "quorate", strlen("quorate"), "0", strlen("0"));
  689. if (err != CS_OK) {
  690. syslog (LOG_INFO, "Can't create confdb key. Error %d\n", err);
  691. return (2);
  692. }
  693. syslog (LOG_INFO, "%s iid %d: sleep 3\n", __FUNCTION__, instance_id);
  694. sleep (3);
  695. syslog (LOG_INFO, "%s iid %d: wasn't killed\n", __FUNCTION__, instance_id);
  696. return (2);
  697. }
  698. if (instance_id == 3) {
  699. return (0);
  700. }
  701. return (2);
  702. }
  703. /*
  704. * Test confdb integration + quit policy
  705. */
  706. static int test8 (pid_t pid, pid_t old_pid, int test_n) {
  707. confdb_handle_t cdb_handle;
  708. cs_error_t err;
  709. hdb_handle_t res_handle, proc_handle, pid_handle;
  710. size_t value_len;
  711. uint64_t tstamp1, tstamp2;
  712. int32_t msec_diff;
  713. char key_value[256];
  714. unsigned int instance_id;
  715. char tmp_obj[PATH_MAX];
  716. confdb_value_types_t cdbtype;
  717. err = confdb_initialize (&cdb_handle, NULL);
  718. if (err != CS_OK) {
  719. syslog (LOG_INFO, "Could not initialize Cluster Configuration Database API instance error %d. Test skipped\n", err);
  720. return (1);
  721. }
  722. syslog (LOG_INFO, "%s test %d\n", __FUNCTION__, test_n);
  723. if (test_n == 2) {
  724. /*
  725. * Object should not exist
  726. */
  727. syslog (LOG_INFO, "%s Testing if object exists (it shouldn't)\n", __FUNCTION__);
  728. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  729. if (err != CS_OK) {
  730. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  731. return (2);
  732. }
  733. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  734. if (err != CS_OK) {
  735. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  736. return (2);
  737. }
  738. err = confdb_object_find_start(cdb_handle, res_handle);
  739. if (err != CS_OK) {
  740. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  741. return (2);
  742. }
  743. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  744. if (err != CS_OK) {
  745. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  746. return (2);
  747. }
  748. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  749. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  750. }
  751. err = confdb_object_find_start(cdb_handle, proc_handle);
  752. if (err != CS_OK) {
  753. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  754. return (2);
  755. }
  756. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  757. if (err == CS_OK) {
  758. syslog (LOG_INFO, "Could find object \"%s\": %d.\n", tmp_obj, err);
  759. return (2);
  760. }
  761. }
  762. if (test_n == 1 || test_n == 2) {
  763. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  764. err = sam_initialize (2000, SAM_RECOVERY_POLICY_QUIT | SAM_RECOVERY_POLICY_CONFDB);
  765. if (err != CS_OK) {
  766. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  767. return 2;
  768. }
  769. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  770. err = sam_register (&instance_id);
  771. if (err != CS_OK) {
  772. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  773. return 2;
  774. }
  775. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  776. if (err != CS_OK) {
  777. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  778. return (2);
  779. }
  780. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  781. if (err != CS_OK) {
  782. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  783. return (2);
  784. }
  785. err = confdb_object_find_start(cdb_handle, res_handle);
  786. if (err != CS_OK) {
  787. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  788. return (2);
  789. }
  790. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  791. if (err != CS_OK) {
  792. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  793. return (2);
  794. }
  795. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  796. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  797. }
  798. err = confdb_object_find_start(cdb_handle, proc_handle);
  799. if (err != CS_OK) {
  800. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  801. return (2);
  802. }
  803. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  804. if (err != CS_OK) {
  805. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  806. return (2);
  807. }
  808. err = confdb_key_get(cdb_handle, pid_handle, "recovery", strlen("recovery"), key_value, &value_len);
  809. if (err != CS_OK) {
  810. syslog (LOG_INFO, "Could not get \"recovery\" key: %d.\n", err);
  811. return (2);
  812. }
  813. if (value_len != strlen ("quit") || memcmp (key_value, "quit", value_len) != 0) {
  814. syslog (LOG_INFO, "Recovery key \"%s\" is not \"watchdog\".\n", key_value);
  815. return (2);
  816. }
  817. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  818. if (err != CS_OK) {
  819. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  820. return (2);
  821. }
  822. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  823. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  824. return (2);
  825. }
  826. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  827. err = sam_start ();
  828. if (err != CS_OK) {
  829. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  830. return 2;
  831. }
  832. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  833. if (err != CS_OK) {
  834. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  835. return (2);
  836. }
  837. if (value_len != strlen ("running") || memcmp (key_value, "running", value_len) != 0) {
  838. syslog (LOG_INFO, "State key is not \"running\".\n");
  839. return (2);
  840. }
  841. syslog (LOG_INFO, "%s iid %d: stop\n", __FUNCTION__, instance_id);
  842. err = sam_stop ();
  843. if (err != CS_OK) {
  844. syslog (LOG_ERR, "Can't stop hc. Error %d\n", err);
  845. return 2;
  846. }
  847. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  848. if (err != CS_OK) {
  849. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  850. return (2);
  851. }
  852. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  853. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  854. return (2);
  855. }
  856. syslog (LOG_INFO, "%s iid %d: sleeping 5\n", __FUNCTION__, instance_id);
  857. sleep (5);
  858. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  859. if (err != CS_OK) {
  860. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  861. return (2);
  862. }
  863. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  864. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  865. return (2);
  866. }
  867. syslog (LOG_INFO, "%s iid %d: start 2\n", __FUNCTION__, instance_id);
  868. err = sam_start ();
  869. if (err != CS_OK) {
  870. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  871. return 2;
  872. }
  873. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  874. if (err != CS_OK) {
  875. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  876. return (2);
  877. }
  878. if (value_len != strlen ("running") || memcmp (key_value, "running", value_len) != 0) {
  879. syslog (LOG_INFO, "State key is not \"running\".\n");
  880. return (2);
  881. }
  882. if (test_n == 2) {
  883. syslog (LOG_INFO, "%s iid %d: sleeping 5. Should be killed\n", __FUNCTION__, instance_id);
  884. sleep (5);
  885. return (2);
  886. } else {
  887. syslog (LOG_INFO, "%s iid %d: Test HC\n", __FUNCTION__, instance_id);
  888. err = sam_hc_send ();
  889. if (err != CS_OK) {
  890. syslog (LOG_ERR, "Can't send hc. Error %d\n", err);
  891. return 2;
  892. }
  893. err = confdb_key_get_typed (cdb_handle, pid_handle, "last_updated", &tstamp1, &value_len, &cdbtype);
  894. if (err != CS_OK) {
  895. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  896. return (2);
  897. }
  898. syslog (LOG_INFO, "%s iid %d: Sleep 1\n", __FUNCTION__, instance_id);
  899. sleep (1);
  900. err = sam_hc_send ();
  901. if (err != CS_OK) {
  902. syslog (LOG_ERR, "Can't send hc. Error %d\n", err);
  903. return 2;
  904. }
  905. sleep (1);
  906. err = confdb_key_get_typed (cdb_handle, pid_handle, "last_updated", &tstamp2, &value_len, &cdbtype);
  907. if (err != CS_OK) {
  908. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  909. return (2);
  910. }
  911. msec_diff = (tstamp2 - tstamp1)/CS_TIME_NS_IN_MSEC;
  912. if (msec_diff < 500 || msec_diff > 2000) {
  913. syslog (LOG_INFO, "Difference %d is not within <500, 2000> interval.\n", msec_diff);
  914. return (2);
  915. }
  916. syslog (LOG_INFO, "%s iid %d: stop 2\n", __FUNCTION__, instance_id);
  917. err = sam_stop ();
  918. if (err != CS_OK) {
  919. syslog (LOG_ERR, "Can't stop hc. Error %d\n", err);
  920. return 2;
  921. }
  922. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  923. if (err != CS_OK) {
  924. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  925. return (2);
  926. }
  927. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  928. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  929. return (2);
  930. }
  931. syslog (LOG_INFO, "%s iid %d: exiting\n", __FUNCTION__, instance_id);
  932. return (0);
  933. }
  934. }
  935. if (test_n == 3) {
  936. syslog (LOG_INFO, "%s Testing if status is failed\n", __FUNCTION__);
  937. /*
  938. * Previous should be FAILED
  939. */
  940. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  941. if (err != CS_OK) {
  942. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  943. return (2);
  944. }
  945. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  946. if (err != CS_OK) {
  947. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  948. return (2);
  949. }
  950. err = confdb_object_find_start(cdb_handle, res_handle);
  951. if (err != CS_OK) {
  952. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  953. return (2);
  954. }
  955. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  956. if (err != CS_OK) {
  957. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  958. return (2);
  959. }
  960. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  961. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  962. }
  963. err = confdb_object_find_start(cdb_handle, proc_handle);
  964. if (err != CS_OK) {
  965. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  966. return (2);
  967. }
  968. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  969. if (err != CS_OK) {
  970. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  971. return (2);
  972. }
  973. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  974. if (err != CS_OK) {
  975. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  976. return (2);
  977. }
  978. if (value_len != strlen ("failed") || memcmp (key_value, "failed", value_len) != 0) {
  979. syslog (LOG_INFO, "State key is not \"failed\".\n");
  980. return (2);
  981. }
  982. return (0);
  983. }
  984. return (2);
  985. }
  986. /*
  987. * Test confdb integration + restart policy
  988. */
  989. static int test9 (pid_t pid, pid_t old_pid, int test_n) {
  990. confdb_handle_t cdb_handle;
  991. cs_error_t err;
  992. hdb_handle_t res_handle, proc_handle, pid_handle;
  993. size_t value_len;
  994. char key_value[256];
  995. unsigned int instance_id;
  996. char tmp_obj[PATH_MAX];
  997. err = confdb_initialize (&cdb_handle, NULL);
  998. if (err != CS_OK) {
  999. syslog (LOG_INFO, "Could not initialize Cluster Configuration Database API instance error %d. Test skipped\n", err);
  1000. return (1);
  1001. }
  1002. syslog (LOG_INFO, "%s test %d\n", __FUNCTION__, test_n);
  1003. if (test_n == 1) {
  1004. syslog (LOG_INFO, "%s: initialize\n", __FUNCTION__);
  1005. err = sam_initialize (2000, SAM_RECOVERY_POLICY_RESTART | SAM_RECOVERY_POLICY_CONFDB);
  1006. if (err != CS_OK) {
  1007. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  1008. return 2;
  1009. }
  1010. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  1011. err = sam_register (&instance_id);
  1012. if (err != CS_OK) {
  1013. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  1014. return 2;
  1015. }
  1016. syslog (LOG_INFO, "%s: iid %d\n", __FUNCTION__, instance_id);
  1017. if (instance_id < 3) {
  1018. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  1019. if (err != CS_OK) {
  1020. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1021. return (2);
  1022. }
  1023. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"),
  1024. &res_handle);
  1025. if (err != CS_OK) {
  1026. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  1027. return (2);
  1028. }
  1029. err = confdb_object_find_start(cdb_handle, res_handle);
  1030. if (err != CS_OK) {
  1031. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1032. return (2);
  1033. }
  1034. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  1035. if (err != CS_OK) {
  1036. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  1037. return (2);
  1038. }
  1039. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  1040. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  1041. }
  1042. err = confdb_object_find_start(cdb_handle, proc_handle);
  1043. if (err != CS_OK) {
  1044. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1045. return (2);
  1046. }
  1047. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  1048. if (err != CS_OK) {
  1049. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  1050. return (2);
  1051. }
  1052. err = confdb_key_get(cdb_handle, pid_handle, "recovery", strlen("recovery"), key_value, &value_len);
  1053. if (err != CS_OK) {
  1054. syslog (LOG_INFO, "Could not get \"recovery\" key: %d.\n", err);
  1055. return (2);
  1056. }
  1057. if (value_len != strlen ("restart") || memcmp (key_value, "restart", value_len) != 0) {
  1058. syslog (LOG_INFO, "Recovery key \"%s\" is not \"restart\".\n", key_value);
  1059. return (2);
  1060. }
  1061. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  1062. if (err != CS_OK) {
  1063. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  1064. return (2);
  1065. }
  1066. if (value_len != strlen ("stopped") || memcmp (key_value, "stopped", value_len) != 0) {
  1067. syslog (LOG_INFO, "State key is not \"stopped\".\n");
  1068. return (2);
  1069. }
  1070. syslog (LOG_INFO, "%s iid %d: start\n", __FUNCTION__, instance_id);
  1071. err = sam_start ();
  1072. if (err != CS_OK) {
  1073. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  1074. return 2;
  1075. }
  1076. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  1077. if (err != CS_OK) {
  1078. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  1079. return (2);
  1080. }
  1081. if (value_len != strlen ("running") || memcmp (key_value, "running", value_len) != 0) {
  1082. syslog (LOG_INFO, "State key is not \"running\".\n");
  1083. return (2);
  1084. }
  1085. syslog (LOG_INFO, "%s iid %d: waiting for kill\n", __FUNCTION__, instance_id);
  1086. sleep (10);
  1087. return (2);
  1088. }
  1089. if (instance_id == 3) {
  1090. syslog (LOG_INFO, "%s iid %d: mark failed\n", __FUNCTION__, instance_id);
  1091. err = sam_mark_failed ();
  1092. if (err != CS_OK) {
  1093. syslog (LOG_ERR, "Can't mark failed. Error %d\n", err);
  1094. return 2;
  1095. }
  1096. sleep (10);
  1097. return (2);
  1098. }
  1099. return (2);
  1100. }
  1101. if (test_n == 2) {
  1102. syslog (LOG_INFO, "%s Testing if status is failed\n", __FUNCTION__);
  1103. /*
  1104. * Previous should be FAILED
  1105. */
  1106. err = confdb_object_find_start(cdb_handle, OBJECT_PARENT_HANDLE);
  1107. if (err != CS_OK) {
  1108. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1109. return (2);
  1110. }
  1111. err = confdb_object_find(cdb_handle, OBJECT_PARENT_HANDLE, "resources", strlen("resources"), &res_handle);
  1112. if (err != CS_OK) {
  1113. syslog (LOG_INFO, "Could not object_find \"resources\": %d.\n", err);
  1114. return (2);
  1115. }
  1116. err = confdb_object_find_start(cdb_handle, res_handle);
  1117. if (err != CS_OK) {
  1118. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1119. return (2);
  1120. }
  1121. err = confdb_object_find(cdb_handle, res_handle, "process", strlen("process"), &proc_handle);
  1122. if (err != CS_OK) {
  1123. syslog (LOG_INFO, "Could not object_find \"process\": %d.\n", err);
  1124. return (2);
  1125. }
  1126. if (snprintf (tmp_obj, sizeof (tmp_obj), "%s:%d", __progname, pid) >= sizeof (tmp_obj)) {
  1127. snprintf (tmp_obj, sizeof (tmp_obj), "%d", pid);
  1128. }
  1129. err = confdb_object_find_start(cdb_handle, proc_handle);
  1130. if (err != CS_OK) {
  1131. syslog (LOG_INFO, "Could not start object_find %d.\n", err);
  1132. return (2);
  1133. }
  1134. err = confdb_object_find(cdb_handle, proc_handle, tmp_obj, strlen(tmp_obj), &pid_handle);
  1135. if (err != CS_OK) {
  1136. syslog (LOG_INFO, "Could not object_find \"%s\": %d.\n", tmp_obj, err);
  1137. return (2);
  1138. }
  1139. err = confdb_key_get(cdb_handle, pid_handle, "state", strlen("state"), key_value, &value_len);
  1140. if (err != CS_OK) {
  1141. syslog (LOG_INFO, "Could not get \"state\" key: %d.\n", err);
  1142. return (2);
  1143. }
  1144. if (value_len != strlen ("failed") || memcmp (key_value, "failed", value_len) != 0) {
  1145. syslog (LOG_INFO, "State key is not \"failed\".\n");
  1146. return (2);
  1147. }
  1148. return (0);
  1149. }
  1150. return (2);
  1151. }
  1152. static int hc_allways_respond_cb(void)
  1153. {
  1154. syslog (LOG_INFO, "%s() -> health check OK.", __FUNCTION__);
  1155. return 0;
  1156. }
  1157. static int setup_hc (void)
  1158. {
  1159. cs_error_t err;
  1160. unsigned int instance_id;
  1161. err = sam_initialize (1000, SAM_RECOVERY_POLICY_QUIT | SAM_RECOVERY_POLICY_CONFDB);
  1162. if (err != CS_OK) {
  1163. syslog (LOG_ERR, "Can't initialize SAM API. Error %d\n", err);
  1164. return 2;
  1165. }
  1166. syslog (LOG_INFO, "%s: register\n", __FUNCTION__);
  1167. err = sam_register (&instance_id);
  1168. if (err != CS_OK) {
  1169. syslog (LOG_ERR, "Can't register. Error %d\n", err);
  1170. return 2;
  1171. }
  1172. err = sam_hc_callback_register (hc_allways_respond_cb);
  1173. syslog (LOG_INFO, "%s instance id %d: start\n", __FUNCTION__, instance_id);
  1174. err = sam_start ();
  1175. if (err != CS_OK) {
  1176. syslog (LOG_ERR, "Can't start hc. Error %d\n", err);
  1177. return 2;
  1178. }
  1179. return (0);
  1180. }
  1181. static int do_test_command(int sock, char* func, char*args[], int num_args)
  1182. {
  1183. int err = 0;
  1184. pid_t pid;
  1185. int stat;
  1186. pid = fork ();
  1187. if (pid == -1) {
  1188. syslog (LOG_ERR, "Can't fork\n");
  1189. return -1;
  1190. }
  1191. if (pid == 0) {
  1192. if (strcmp ("test1", func) == 0) {
  1193. err = test1 ();
  1194. } else if (strcmp ("test2", func) == 0) {
  1195. err = test2 ();
  1196. } else if (strcmp ("test4", func) == 0) {
  1197. err = test4 ();
  1198. } else if (strcmp ("test5", func) == 0) {
  1199. err = test5 ();
  1200. } else if (strcmp ("test6", func) == 0) {
  1201. err = test6 ();
  1202. } else if (strcmp ("test_quorum", func) == 0) {
  1203. err = test_quorum ();
  1204. } else if (strcmp ("test8", func) == 0) {
  1205. err = test8 (getpid(), 0, 1);
  1206. } else if (strcmp ("test9", func) == 0) {
  1207. err = test9 (getpid(), 0, 1);
  1208. }
  1209. sam_finalize ();
  1210. exit(err);
  1211. }
  1212. if (pid > 0) {
  1213. waitpid (pid, &stat, 0);
  1214. return WEXITSTATUS (stat);
  1215. }
  1216. return -1;
  1217. }
  1218. static void do_command (int sock, char* func, char*args[], int num_args)
  1219. {
  1220. char response[100];
  1221. int err = 0;
  1222. if (parse_debug) {
  1223. syslog (LOG_INFO, "RPC:%s() called.", func);
  1224. }
  1225. if (strncmp ("test", func, 4) == 0) {
  1226. err = do_test_command(sock, func, args, num_args);
  1227. } else if (strcmp ("setup_hc", func) == 0) {
  1228. err = setup_hc ();
  1229. } else if (strcmp ("sam_stop", func) == 0) {
  1230. sam_stop ();
  1231. sam_finalize();
  1232. } else {
  1233. err = -1;
  1234. syslog (LOG_ERR,"%s RPC:%s not supported!", __func__, func);
  1235. snprintf (response, 100, "%s", NOT_SUPPORTED_STR);
  1236. }
  1237. if (err == 0) {
  1238. snprintf (response, 100, "%s", OK_STR);
  1239. } else if (err == 1) {
  1240. snprintf (response, 100, "%s", FAIL_STR);
  1241. syslog (LOG_ERR, "%s() test skipped?! (%d).", func, err);
  1242. } else {
  1243. snprintf (response, 100, "%s", FAIL_STR);
  1244. syslog (LOG_ERR, "%s() failed (%d).", func, err);
  1245. }
  1246. send (sock, response, strlen (response) + 1, 0);
  1247. }
  1248. int main (int argc, char *argv[])
  1249. {
  1250. int ret;
  1251. openlog (NULL, LOG_CONS|LOG_PID, LOG_DAEMON);
  1252. syslog (LOG_INFO, "%s STARTNG", __FILE__);
  1253. parse_debug = 1;
  1254. ret = test_agent_run (9036, do_command, NULL);
  1255. syslog (LOG_INFO, "%s EXITING", __FILE__);
  1256. return ret;
  1257. }