cpg_test_agent.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775
  1. /*
  2. * Copyright (c) 2010 Red Hat, Inc.
  3. *
  4. * All rights reserved.
  5. *
  6. * Author: Angus Salkeld (asalkeld@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 MontaVista Software, 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. #include <errno.h>
  35. #include <unistd.h>
  36. #include <stdio.h>
  37. #include <stdlib.h>
  38. #include <assert.h>
  39. #include <string.h>
  40. #include <sys/types.h>
  41. #include <sys/socket.h>
  42. #include <netinet/in.h>
  43. #include <arpa/inet.h>
  44. #include <netdb.h>
  45. #include <poll.h>
  46. #include <unistd.h>
  47. #include <fcntl.h>
  48. #include <corosync/list.h>
  49. #include <qb/qbdefs.h>
  50. #include <qb/qbutil.h>
  51. #include <qb/qbloop.h>
  52. #include <qb/qblog.h>
  53. #include <corosync/cpg.h>
  54. #include <corosync/cfg.h>
  55. #include "../../exec/crypto.h"
  56. #include "common_test_agent.h"
  57. typedef enum {
  58. MSG_OK,
  59. MSG_NODEID_ERR,
  60. MSG_PID_ERR,
  61. MSG_SEQ_ERR,
  62. MSG_SIZE_ERR,
  63. MSG_SHA1_ERR,
  64. } msg_status_t;
  65. typedef struct {
  66. uint32_t nodeid;
  67. pid_t pid;
  68. unsigned char sha1[20];
  69. uint32_t seq;
  70. size_t size;
  71. unsigned char payload[0];
  72. } msg_t;
  73. #define LOG_STR_SIZE 80
  74. typedef struct {
  75. char log[LOG_STR_SIZE];
  76. struct list_head list;
  77. } log_entry_t;
  78. static char big_and_buf[HOW_BIG_AND_BUF];
  79. static int32_t record_config_events_g = 0;
  80. static int32_t record_messages_g = 0;
  81. static cpg_handle_t cpg_handle = 0;
  82. static corosync_cfg_handle_t cfg_handle = 0;
  83. static int32_t cpg_fd = -1;
  84. static int32_t cfg_fd = -1;
  85. static struct list_head config_chg_log_head;
  86. static struct list_head msg_log_head;
  87. static pid_t my_pid;
  88. static uint32_t my_nodeid;
  89. static int32_t my_seq;
  90. static int32_t use_zcb = QB_FALSE;
  91. static int32_t my_msgs_to_send;
  92. static int32_t my_msgs_sent;
  93. static int32_t total_stored_msgs = 0;
  94. static int32_t total_msgs_revd = 0;
  95. static int32_t in_cnchg = 0;
  96. static int32_t pcmk_test = 0;
  97. static void send_some_more_messages (void * unused);
  98. static char*
  99. err_status_string (char * buf, size_t buf_len, msg_status_t status)
  100. {
  101. switch (status) {
  102. case MSG_OK:
  103. strncpy (buf, "OK", buf_len);
  104. break;
  105. case MSG_NODEID_ERR:
  106. strncpy (buf, "NODEID_ERR", buf_len);
  107. break;
  108. case MSG_PID_ERR:
  109. strncpy (buf, "PID_ERR", buf_len);
  110. break;
  111. case MSG_SEQ_ERR:
  112. strncpy (buf, "SEQ_ERR", buf_len);
  113. break;
  114. case MSG_SIZE_ERR:
  115. strncpy (buf, "SIZE_ERR", buf_len);
  116. break;
  117. case MSG_SHA1_ERR:
  118. strncpy (buf, "SHA1_ERR", buf_len);
  119. break;
  120. default:
  121. strncpy (buf, "UNKNOWN_ERR", buf_len);
  122. break;
  123. }
  124. if (buf_len > 0) {
  125. buf[buf_len - 1] = '\0';
  126. }
  127. return buf;
  128. }
  129. static void delivery_callback (
  130. cpg_handle_t handle,
  131. const struct cpg_name *groupName,
  132. uint32_t nodeid,
  133. uint32_t pid,
  134. void *msg,
  135. size_t msg_len)
  136. {
  137. log_entry_t *log_pt;
  138. msg_t *msg_pt = (msg_t*)msg;
  139. msg_status_t status = MSG_OK;
  140. char status_buf[20];
  141. unsigned char sha1_compare[20];
  142. hash_state sha1_hash;
  143. if (record_messages_g == 0) {
  144. return;
  145. }
  146. if (nodeid != msg_pt->nodeid) {
  147. status = MSG_NODEID_ERR;
  148. }
  149. if (pid != msg_pt->pid) {
  150. status = MSG_PID_ERR;
  151. }
  152. if (msg_len != msg_pt->size) {
  153. status = MSG_SIZE_ERR;
  154. }
  155. sha1_init (&sha1_hash);
  156. sha1_process (&sha1_hash, msg_pt->payload, (msg_pt->size - sizeof (msg_t)));
  157. sha1_done (&sha1_hash, sha1_compare);
  158. if (memcmp (sha1_compare, msg_pt->sha1, 20) != 0) {
  159. qb_log (LOG_ERR, "msg seq:%d; incorrect hash",
  160. msg_pt->seq);
  161. status = MSG_SHA1_ERR;
  162. }
  163. log_pt = malloc (sizeof(log_entry_t));
  164. list_init (&log_pt->list);
  165. snprintf (log_pt->log, LOG_STR_SIZE, "%d:%d:%d:%s;",
  166. msg_pt->nodeid, msg_pt->seq, my_seq,
  167. err_status_string (status_buf, 20, status));
  168. list_add_tail (&log_pt->list, &msg_log_head);
  169. total_stored_msgs++;
  170. total_msgs_revd++;
  171. my_seq++;
  172. if ((total_msgs_revd % 1000) == 0) {
  173. qb_log (LOG_INFO, "rx %d", total_msgs_revd);
  174. }
  175. }
  176. static void config_change_callback (
  177. cpg_handle_t handle,
  178. const struct cpg_name *groupName,
  179. const struct cpg_address *member_list, size_t member_list_entries,
  180. const struct cpg_address *left_list, size_t left_list_entries,
  181. const struct cpg_address *joined_list, size_t joined_list_entries)
  182. {
  183. int i;
  184. log_entry_t *log_pt;
  185. /* group_name,ip,pid,join|leave */
  186. if (record_config_events_g > 0) {
  187. qb_log (LOG_INFO, "got cpg event[recording] for group %s", groupName->value);
  188. } else {
  189. qb_log (LOG_INFO, "got cpg event[ignoring] for group %s", groupName->value);
  190. }
  191. for (i = 0; i < left_list_entries; i++) {
  192. if (record_config_events_g > 0) {
  193. log_pt = malloc (sizeof(log_entry_t));
  194. list_init (&log_pt->list);
  195. snprintf (log_pt->log, LOG_STR_SIZE, "%s,%d,%d,left",
  196. groupName->value, left_list[i].nodeid,left_list[i].pid);
  197. list_add_tail(&log_pt->list, &config_chg_log_head);
  198. qb_log (LOG_INFO, "cpg event %s", log_pt->log);
  199. }
  200. }
  201. for (i = 0; i < joined_list_entries; i++) {
  202. if (record_config_events_g > 0) {
  203. log_pt = malloc (sizeof(log_entry_t));
  204. list_init (&log_pt->list);
  205. snprintf (log_pt->log, LOG_STR_SIZE, "%s,%d,%d,join",
  206. groupName->value, joined_list[i].nodeid,joined_list[i].pid);
  207. list_add_tail (&log_pt->list, &config_chg_log_head);
  208. qb_log (LOG_INFO, "cpg event %s", log_pt->log);
  209. }
  210. }
  211. if (pcmk_test == 1) {
  212. in_cnchg = 1;
  213. send_some_more_messages (NULL);
  214. in_cnchg = 0;
  215. }
  216. }
  217. static void my_shutdown_callback (corosync_cfg_handle_t handle,
  218. corosync_cfg_shutdown_flags_t flags)
  219. {
  220. qb_log (LOG_CRIT, "flags:%d", flags);
  221. if (flags & COROSYNC_CFG_SHUTDOWN_FLAG_REQUEST) {
  222. corosync_cfg_replyto_shutdown (cfg_handle, COROSYNC_CFG_SHUTDOWN_FLAG_YES);
  223. }
  224. }
  225. static corosync_cfg_callbacks_t cfg_callbacks = {
  226. .corosync_cfg_shutdown_callback = my_shutdown_callback,
  227. .corosync_cfg_state_track_callback = NULL,
  228. };
  229. static cpg_callbacks_t callbacks = {
  230. .cpg_deliver_fn = delivery_callback,
  231. .cpg_confchg_fn = config_change_callback,
  232. };
  233. static void record_messages (void)
  234. {
  235. record_messages_g = 1;
  236. qb_log (LOG_INFO, "record:%d", record_messages_g);
  237. }
  238. static void record_config_events (int sock)
  239. {
  240. char response[100];
  241. record_config_events_g = 1;
  242. qb_log (LOG_INFO, "record:%d", record_config_events_g);
  243. snprintf (response, 100, "%s", OK_STR);
  244. send (sock, response, strlen (response), 0);
  245. }
  246. static void read_config_event (int sock)
  247. {
  248. const char *empty = "None";
  249. struct list_head * list = config_chg_log_head.next;
  250. log_entry_t *entry;
  251. if (list != &config_chg_log_head) {
  252. entry = list_entry (list, log_entry_t, list);
  253. send (sock, entry->log, strlen (entry->log), 0);
  254. list_del (&entry->list);
  255. free (entry);
  256. } else {
  257. qb_log (LOG_DEBUG, "no events in list");
  258. send (sock, empty, strlen (empty), 0);
  259. }
  260. }
  261. static void read_messages (int sock, char* atmost_str)
  262. {
  263. struct list_head * list;
  264. log_entry_t *entry;
  265. int atmost = atoi (atmost_str);
  266. int packed = 0;
  267. ssize_t rc;
  268. if (atmost == 0)
  269. atmost = 1;
  270. if (atmost > (HOW_BIG_AND_BUF / LOG_STR_SIZE))
  271. atmost = (HOW_BIG_AND_BUF / LOG_STR_SIZE);
  272. big_and_buf[0] = '\0';
  273. for (list = msg_log_head.next;
  274. (!list_empty (&msg_log_head) && packed < atmost); ) {
  275. entry = list_entry (list, log_entry_t, list);
  276. strcat (big_and_buf, entry->log);
  277. packed++;
  278. list = list->next;
  279. list_del (&entry->list);
  280. free (entry);
  281. total_stored_msgs--;
  282. }
  283. if (packed == 0) {
  284. strcpy (big_and_buf, "None");
  285. } else {
  286. if ((total_stored_msgs % 1000) == 0) {
  287. qb_log(LOG_INFO, "sending %d; total_stored_msgs:%d; len:%d",
  288. packed, total_stored_msgs, (int)strlen (big_and_buf));
  289. }
  290. }
  291. rc = send (sock, big_and_buf, strlen (big_and_buf), 0);
  292. assert(rc = strlen (big_and_buf));
  293. }
  294. static qb_loop_timer_handle more_messages_timer_handle;
  295. static void send_some_more_messages_later (void)
  296. {
  297. cpg_dispatch (cpg_handle, CS_DISPATCH_ALL);
  298. qb_loop_timer_add (
  299. ta_poll_handle_get(),
  300. QB_LOOP_MED,
  301. 300*QB_TIME_NS_IN_MSEC, NULL,
  302. send_some_more_messages,
  303. &more_messages_timer_handle);
  304. }
  305. static void send_some_more_messages_zcb (void)
  306. {
  307. msg_t *my_msg;
  308. int i;
  309. int send_now;
  310. size_t payload_size;
  311. size_t total_size;
  312. hash_state sha1_hash;
  313. cs_error_t res;
  314. cpg_flow_control_state_t fc_state;
  315. void *zcb_buffer;
  316. if (cpg_fd < 0)
  317. return;
  318. send_now = my_msgs_to_send;
  319. payload_size = (rand() % 100000);
  320. total_size = payload_size + sizeof (msg_t);
  321. cpg_zcb_alloc (cpg_handle, total_size, &zcb_buffer);
  322. my_msg = (msg_t*)zcb_buffer;
  323. qb_log(LOG_DEBUG, "send_now:%d", send_now);
  324. my_msg->pid = my_pid;
  325. my_msg->nodeid = my_nodeid;
  326. my_msg->size = sizeof (msg_t) + payload_size;
  327. my_msg->seq = my_msgs_sent;
  328. for (i = 0; i < payload_size; i++) {
  329. my_msg->payload[i] = i;
  330. }
  331. sha1_init (&sha1_hash);
  332. sha1_process (&sha1_hash, my_msg->payload, payload_size);
  333. sha1_done (&sha1_hash, my_msg->sha1);
  334. for (i = 0; i < send_now; i++) {
  335. res = cpg_flow_control_state_get (cpg_handle, &fc_state);
  336. if (res == CS_OK && fc_state == CPG_FLOW_CONTROL_ENABLED) {
  337. /* lets do this later */
  338. send_some_more_messages_later ();
  339. qb_log (LOG_INFO, "flow control enabled.");
  340. goto free_buffer;
  341. }
  342. res = cpg_zcb_mcast_joined (cpg_handle, CPG_TYPE_AGREED, zcb_buffer, total_size);
  343. if (res == CS_ERR_TRY_AGAIN) {
  344. /* lets do this later */
  345. send_some_more_messages_later ();
  346. goto free_buffer;
  347. } else if (res != CS_OK) {
  348. qb_log (LOG_ERR, "cpg_mcast_joined error:%d, exiting.",
  349. res);
  350. exit (-2);
  351. }
  352. my_msgs_sent++;
  353. my_msgs_to_send--;
  354. }
  355. free_buffer:
  356. cpg_zcb_free (cpg_handle, zcb_buffer);
  357. }
  358. #define cs_repeat(counter, max, code) do { \
  359. code; \
  360. if (res == CS_ERR_TRY_AGAIN) { \
  361. counter++; \
  362. sleep(counter); \
  363. } \
  364. } while (res == CS_ERR_TRY_AGAIN && counter < max)
  365. static unsigned char buffer[200000];
  366. static void send_some_more_messages_normal (void)
  367. {
  368. msg_t my_msg;
  369. struct iovec iov[2];
  370. int i;
  371. int send_now;
  372. size_t payload_size;
  373. hash_state sha1_hash;
  374. cs_error_t res;
  375. cpg_flow_control_state_t fc_state;
  376. int retries = 0;
  377. time_t before;
  378. if (cpg_fd < 0)
  379. return;
  380. send_now = my_msgs_to_send;
  381. qb_log (LOG_TRACE, "send_now:%d", send_now);
  382. my_msg.pid = my_pid;
  383. my_msg.nodeid = my_nodeid;
  384. payload_size = (rand() % 10000);
  385. my_msg.size = sizeof (msg_t) + payload_size;
  386. my_msg.seq = my_msgs_sent;
  387. for (i = 0; i < payload_size; i++) {
  388. buffer[i] = i;
  389. }
  390. sha1_init (&sha1_hash);
  391. sha1_process (&sha1_hash, buffer, payload_size);
  392. sha1_done (&sha1_hash, my_msg.sha1);
  393. iov[0].iov_len = sizeof (msg_t);
  394. iov[0].iov_base = &my_msg;
  395. iov[1].iov_len = payload_size;
  396. iov[1].iov_base = buffer;
  397. for (i = 0; i < send_now; i++) {
  398. if (in_cnchg && pcmk_test) {
  399. retries = 0;
  400. before = time(NULL);
  401. cs_repeat(retries, 30, res = cpg_mcast_joined(cpg_handle, CPG_TYPE_AGREED, iov, 2));
  402. if (retries > 20) {
  403. qb_log (LOG_ERR, "cs_repeat: blocked for :%lu secs.",
  404. (unsigned long)(time(NULL) - before));
  405. }
  406. if (res != CS_OK) {
  407. qb_log (LOG_ERR, "cpg_mcast_joined error:%d.",
  408. res);
  409. return;
  410. }
  411. } else {
  412. res = cpg_flow_control_state_get (cpg_handle, &fc_state);
  413. if (res == CS_OK && fc_state == CPG_FLOW_CONTROL_ENABLED) {
  414. /* lets do this later */
  415. send_some_more_messages_later ();
  416. qb_log (LOG_INFO, "flow control enabled.");
  417. return;
  418. }
  419. res = cpg_mcast_joined (cpg_handle, CPG_TYPE_AGREED, iov, 2);
  420. if (res == CS_ERR_TRY_AGAIN) {
  421. /* lets do this later */
  422. send_some_more_messages_later ();
  423. if (i > 0) {
  424. qb_log (LOG_INFO, "TRY_AGAIN %d to send.",
  425. my_msgs_to_send);
  426. }
  427. return;
  428. } else if (res != CS_OK) {
  429. qb_log (LOG_ERR, "cpg_mcast_joined error:%d, exiting.",
  430. res);
  431. exit (-2);
  432. }
  433. }
  434. my_msgs_sent++;
  435. my_msg.seq = my_msgs_sent;
  436. my_msgs_to_send--;
  437. }
  438. qb_log (LOG_TRACE, "sent %d; to send %d.",
  439. my_msgs_sent, my_msgs_to_send);
  440. }
  441. static void send_some_more_messages (void * unused)
  442. {
  443. if (use_zcb) {
  444. send_some_more_messages_zcb ();
  445. } else {
  446. send_some_more_messages_normal ();
  447. }
  448. }
  449. static void msg_blaster (int sock, char* num_to_send_str)
  450. {
  451. my_msgs_to_send = atoi (num_to_send_str);
  452. my_msgs_sent = 0;
  453. my_seq = 1;
  454. my_pid = getpid();
  455. use_zcb = QB_FALSE;
  456. total_stored_msgs = 0;
  457. cpg_local_get (cpg_handle, &my_nodeid);
  458. /* control the limits */
  459. if (my_msgs_to_send <= 0)
  460. my_msgs_to_send = 1;
  461. if (my_msgs_to_send > 10000)
  462. my_msgs_to_send = 10000;
  463. send_some_more_messages_normal ();
  464. }
  465. static void context_test (int sock)
  466. {
  467. char response[100];
  468. char *cmp;
  469. cpg_context_set (cpg_handle, response);
  470. cpg_context_get (cpg_handle, (void**)&cmp);
  471. if (response != cmp) {
  472. snprintf (response, 100, "%s", FAIL_STR);
  473. }
  474. else {
  475. snprintf (response, 100, "%s", OK_STR);
  476. }
  477. send (sock, response, strlen (response), 0);
  478. }
  479. static void msg_blaster_zcb (int sock, char* num_to_send_str)
  480. {
  481. my_msgs_to_send = atoi (num_to_send_str);
  482. my_seq = 1;
  483. my_pid = getpid();
  484. use_zcb = QB_TRUE;
  485. total_stored_msgs = 0;
  486. cpg_local_get (cpg_handle, &my_nodeid);
  487. /* control the limits */
  488. if (my_msgs_to_send <= 0)
  489. my_msgs_to_send = 1;
  490. if (my_msgs_to_send > 10000)
  491. my_msgs_to_send = 10000;
  492. send_some_more_messages_zcb ();
  493. }
  494. static corosync_cfg_state_notification_t notification_buffer;
  495. static int cfg_dispatch_wrapper_fn (
  496. int fd,
  497. int revents,
  498. void *data)
  499. {
  500. cs_error_t error;
  501. if (revents & POLLHUP || revents & POLLERR) {
  502. qb_log (LOG_ERR, "got POLLHUP disconnecting from CFG");
  503. corosync_cfg_finalize(cfg_handle);
  504. cfg_handle = 0;
  505. return -1;
  506. }
  507. error = corosync_cfg_dispatch (cfg_handle, CS_DISPATCH_ALL);
  508. if (error == CS_ERR_LIBRARY) {
  509. qb_log (LOG_ERR, "got LIB error disconnecting from CFG.");
  510. corosync_cfg_finalize(cfg_handle);
  511. cfg_handle = 0;
  512. return -1;
  513. }
  514. return 0;
  515. }
  516. static int cpg_dispatch_wrapper_fn (
  517. int fd,
  518. int revents,
  519. void *data)
  520. {
  521. cs_error_t error;
  522. if (revents & POLLHUP || revents & POLLERR) {
  523. qb_log (LOG_ERR, "got POLLHUP disconnecting from CPG");
  524. cpg_finalize(cpg_handle);
  525. cpg_handle = 0;
  526. return -1;
  527. }
  528. error = cpg_dispatch (cpg_handle, CS_DISPATCH_ALL);
  529. if (error == CS_ERR_LIBRARY) {
  530. qb_log (LOG_ERR, "got LIB error disconnecting from CPG");
  531. cpg_finalize(cpg_handle);
  532. cpg_handle = 0;
  533. return -1;
  534. }
  535. return 0;
  536. }
  537. static void do_command (int sock, char* func, char*args[], int num_args)
  538. {
  539. int result;
  540. char response[100];
  541. struct cpg_name group_name;
  542. qb_log (LOG_TRACE, "RPC:%s() called.", func);
  543. if (strcmp ("cpg_mcast_joined",func) == 0) {
  544. struct iovec iov[5];
  545. int a;
  546. for (a = 0; a < num_args; a++) {
  547. iov[a].iov_base = args[a];
  548. iov[a].iov_len = strlen(args[a])+1;
  549. }
  550. cpg_mcast_joined (cpg_handle, CPG_TYPE_AGREED, iov, num_args);
  551. } else if (strcmp ("cpg_join",func) == 0) {
  552. strcpy (group_name.value, args[0]);
  553. group_name.length = strlen(args[0]);
  554. result = cpg_join (cpg_handle, &group_name);
  555. if (result != CS_OK) {
  556. qb_log (LOG_ERR,
  557. "Could not join process group, error %d", result);
  558. exit (1);
  559. }
  560. qb_log (LOG_INFO, "called cpg_join(%s)!", group_name.value);
  561. } else if (strcmp ("cpg_leave",func) == 0) {
  562. strcpy (group_name.value, args[0]);
  563. group_name.length = strlen(args[0]);
  564. result = cpg_leave (cpg_handle, &group_name);
  565. if (result != CS_OK) {
  566. qb_log (LOG_ERR,
  567. "Could not leave process group, error %d", result);
  568. exit (1);
  569. }
  570. qb_log (LOG_INFO, "called cpg_leave(%s)!", group_name.value);
  571. } else if (strcmp ("cpg_initialize",func) == 0) {
  572. int retry_count = 0;
  573. result = cpg_initialize (&cpg_handle, &callbacks);
  574. while (result != CS_OK) {
  575. qb_log (LOG_ERR,
  576. "cpg_initialize error %d (attempt %d)",
  577. result, retry_count);
  578. if (retry_count >= 3) {
  579. exit (1);
  580. }
  581. sleep(1);
  582. retry_count++;
  583. result = cpg_initialize (&cpg_handle, &callbacks);
  584. }
  585. cpg_fd_get (cpg_handle, &cpg_fd);
  586. qb_loop_poll_add (ta_poll_handle_get(),
  587. QB_LOOP_MED,
  588. cpg_fd,
  589. POLLIN|POLLNVAL,
  590. NULL,
  591. cpg_dispatch_wrapper_fn);
  592. } else if (strcmp ("cpg_local_get", func) == 0) {
  593. unsigned int local_nodeid;
  594. cpg_local_get (cpg_handle, &local_nodeid);
  595. snprintf (response, 100, "%u",local_nodeid);
  596. send (sock, response, strlen (response), 0);
  597. } else if (strcmp ("cpg_finalize", func) == 0) {
  598. if (cpg_handle > 0) {
  599. cpg_finalize (cpg_handle);
  600. cpg_handle = 0;
  601. }
  602. } else if (strcmp ("record_config_events", func) == 0) {
  603. record_config_events (sock);
  604. } else if (strcmp ("record_messages", func) == 0) {
  605. record_messages ();
  606. } else if (strcmp ("read_config_event", func) == 0) {
  607. read_config_event (sock);
  608. } else if (strcmp ("read_messages", func) == 0) {
  609. read_messages (sock, args[0]);
  610. } else if (strcmp ("msg_blaster_zcb", func) == 0) {
  611. msg_blaster_zcb (sock, args[0]);
  612. } else if (strcmp ("pcmk_test", func) == 0) {
  613. pcmk_test = 1;
  614. } else if (strcmp ("msg_blaster", func) == 0) {
  615. msg_blaster (sock, args[0]);
  616. } else if (strcmp ("context_test", func) == 0) {
  617. context_test (sock);
  618. } else if (strcmp ("are_you_ok_dude", func) == 0) {
  619. snprintf (response, 100, "%s", OK_STR);
  620. send (sock, response, strlen (response), 0);
  621. } else if (strcmp ("cfg_shutdown", func) == 0) {
  622. qb_log (LOG_INFO, "calling %s() called!", func);
  623. result = corosync_cfg_try_shutdown (cfg_handle, COROSYNC_CFG_SHUTDOWN_FLAG_REQUEST);
  624. qb_log (LOG_INFO,"%s() returned %d!", func, result);
  625. } else if (strcmp ("cfg_initialize",func) == 0) {
  626. int retry_count = 0;
  627. qb_log (LOG_INFO,"%s() called!", func);
  628. result = corosync_cfg_initialize (&cfg_handle, &cfg_callbacks);
  629. while (result != CS_OK) {
  630. qb_log (LOG_ERR,
  631. "cfg_initialize error %d (attempt %d)",
  632. result, retry_count);
  633. if (retry_count >= 3) {
  634. exit (1);
  635. }
  636. sleep(1);
  637. retry_count++;
  638. result = corosync_cfg_initialize (&cfg_handle, &cfg_callbacks);
  639. }
  640. qb_log (LOG_INFO,"corosync_cfg_initialize() == %d", result);
  641. result = corosync_cfg_fd_get (cfg_handle, &cfg_fd);
  642. qb_log (LOG_INFO,"corosync_cfg_fd_get() == %d", result);
  643. result = corosync_cfg_state_track (cfg_handle, 0, &notification_buffer);
  644. qb_log (LOG_INFO,"corosync_cfg_state_track() == %d", result);
  645. qb_loop_poll_add (ta_poll_handle_get(),
  646. QB_LOOP_MED,
  647. cfg_fd,
  648. POLLIN|POLLNVAL,
  649. NULL,
  650. cfg_dispatch_wrapper_fn);
  651. } else {
  652. qb_log(LOG_ERR, "RPC:%s not supported!", func);
  653. }
  654. }
  655. static void my_pre_exit(void)
  656. {
  657. qb_log (LOG_INFO, "%s PRE EXIT", __FILE__);
  658. if (cpg_handle > 0) {
  659. cpg_finalize (cpg_handle);
  660. cpg_handle = 0;
  661. }
  662. if (cfg_handle > 0) {
  663. corosync_cfg_finalize (cfg_handle);
  664. cfg_handle = 0;
  665. }
  666. }
  667. int
  668. main(int argc, char *argv[])
  669. {
  670. list_init (&msg_log_head);
  671. list_init (&config_chg_log_head);
  672. return test_agent_run ("cpg_test_agent", 9034, do_command, my_pre_exit);
  673. }