corosync-notifyd.c 26 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082
  1. /*
  2. * Copyright (c) 2011 Red Hat
  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 <config.h>
  35. #include <sys/select.h>
  36. #include <sys/socket.h>
  37. #include <sys/un.h>
  38. #include <sys/types.h>
  39. #include <netdb.h>
  40. #include <arpa/inet.h>
  41. #include <stdio.h>
  42. #include <stdlib.h>
  43. #include <errno.h>
  44. #include <unistd.h>
  45. #include <string.h>
  46. #include <ctype.h>
  47. #include <poll.h>
  48. #include <signal.h>
  49. #include <syslog.h>
  50. #include <corosync/corotypes.h>
  51. #include <corosync/totem/coropoll.h>
  52. #include <corosync/confdb.h>
  53. #include <corosync/cfg.h>
  54. #include <corosync/quorum.h>
  55. /*
  56. * generic declarations
  57. */
  58. enum {
  59. CS_NTF_LOG,
  60. CS_NTF_STDOUT,
  61. CS_NTF_SNMP,
  62. CS_NTF_DBUS,
  63. CS_NTF_FG,
  64. CS_NTF_MAX,
  65. };
  66. static int conf[CS_NTF_MAX];
  67. typedef void (*node_membership_fn_t)(char *nodename, uint32_t nodeid, char *state, char* ip);
  68. typedef void (*node_quorum_fn_t)(char *nodename, uint32_t nodeid, const char *state);
  69. typedef void (*application_connection_fn_t)(char *nodename, uint32_t nodeid, char *app_name, const char *state);
  70. struct notify_callbacks {
  71. node_membership_fn_t node_membership_fn;
  72. node_quorum_fn_t node_quorum_fn;
  73. application_connection_fn_t application_connection_fn;
  74. };
  75. #define MAX_NOTIFIERS 5
  76. static int num_notifiers = 0;
  77. static struct notify_callbacks notifiers[MAX_NOTIFIERS];
  78. static uint32_t local_nodeid = 0;
  79. static char local_nodename[CS_MAX_NAME_LENGTH];
  80. static hdb_handle_t poll_handle;
  81. static quorum_handle_t quorum_handle;
  82. static void _cs_node_membership_event(char *nodename, uint32_t nodeid, char *state, char* ip);
  83. static void _cs_node_quorum_event(const char *state);
  84. static void _cs_application_connection_event(char *app_name, const char *state);
  85. #ifdef HAVE_DBUS
  86. #include <dbus/dbus.h>
  87. /*
  88. * dbus
  89. */
  90. #define DBUS_CS_NAME "org.corosync"
  91. #define DBUS_CS_IFACE "org.corosync"
  92. #define DBUS_CS_PATH "/org/corosync"
  93. static DBusConnection *db = NULL;
  94. static char _err[512];
  95. static int err_set = 0;
  96. static void _cs_dbus_init(void);
  97. #endif /* HAVE_DBUS */
  98. #ifdef ENABLE_SNMP
  99. #include <net-snmp/net-snmp-config.h>
  100. #include <net-snmp/snmpv3_api.h>
  101. #include <net-snmp/agent/agent_trap.h>
  102. #include <net-snmp/library/mib.h>
  103. #include <net-snmp/library/snmp_api.h>
  104. #include <net-snmp/library/snmp_client.h>
  105. #include <net-snmp/library/snmp_debug.h>
  106. enum snmp_node_status {
  107. SNMP_NODE_STATUS_UNKNOWN = 0,
  108. SNMP_NODE_STATUS_JOINED = 1,
  109. SNMP_NODE_STATUS_LEFT = 2
  110. };
  111. #define SNMP_OID_COROSYNC "1.3.6.1.4.1.35488"
  112. #define SNMP_OID_OBJECT_ROOT SNMP_OID_COROSYNC ".1"
  113. #define SNMP_OID_OBJECT_NODE_NAME SNMP_OID_OBJECT_ROOT ".1"
  114. #define SNMP_OID_OBJECT_NODE_ID SNMP_OID_OBJECT_ROOT ".2"
  115. #define SNMP_OID_OBJECT_NODE_STATUS SNMP_OID_OBJECT_ROOT ".3"
  116. #define SNMP_OID_OBJECT_NODE_ADDR SNMP_OID_OBJECT_ROOT ".4"
  117. #define SNMP_OID_OBJECT_RINGSEQ SNMP_OID_OBJECT_ROOT ".20"
  118. #define SNMP_OID_OBJECT_QUORUM SNMP_OID_OBJECT_ROOT ".21"
  119. #define SNMP_OID_OBJECT_APP_NAME SNMP_OID_OBJECT_ROOT ".40"
  120. #define SNMP_OID_OBJECT_APP_STATUS SNMP_OID_OBJECT_ROOT ".41"
  121. #define SNMP_OID_TRAPS_ROOT SNMP_OID_COROSYNC ".0"
  122. #define SNMP_OID_TRAPS_NODE SNMP_OID_TRAPS_ROOT ".1"
  123. #define SNMP_OID_TRAPS_QUORUM SNMP_OID_TRAPS_ROOT ".2"
  124. #define SNMP_OID_TRAPS_APP SNMP_OID_TRAPS_ROOT ".3"
  125. #define CS_TIMESTAMP_STR_LEN 20
  126. static const char *local_host = "localhost";
  127. #endif /* ENABLE_SNMP */
  128. static char snmp_manager_buf[CS_MAX_NAME_LENGTH];
  129. static char *snmp_manager = NULL;
  130. /*
  131. * confdb
  132. */
  133. #define SEPERATOR_STR "."
  134. static confdb_handle_t confdb_handle;
  135. static void _cs_confdb_key_changed(confdb_handle_t handle,
  136. confdb_change_type_t change_type,
  137. hdb_handle_t parent_object_handle,
  138. hdb_handle_t object_handle,
  139. const void *object_name, size_t object_name_len,
  140. const void *key_name, size_t key_name_len,
  141. const void *key_value, size_t key_value_len);
  142. static void _cs_confdb_object_created(confdb_handle_t handle,
  143. hdb_handle_t parent_object_handle,
  144. hdb_handle_t object_handle,
  145. const void *name_pt, size_t name_len);
  146. static void _cs_confdb_object_deleted(confdb_handle_t handle,
  147. hdb_handle_t parent_object_handle,
  148. const void *name_pt, size_t name_len);
  149. static confdb_callbacks_t callbacks = {
  150. .confdb_key_change_notify_fn = _cs_confdb_key_changed,
  151. .confdb_object_create_change_notify_fn = _cs_confdb_object_created,
  152. .confdb_object_delete_change_notify_fn = _cs_confdb_object_deleted,
  153. };
  154. static int32_t _cs_ip_to_hostname(char* ip, char* name_out)
  155. {
  156. struct sockaddr_in sa;
  157. int rc;
  158. if (strchr(ip, ':') == NULL) {
  159. sa.sin_family = AF_INET;
  160. } else {
  161. sa.sin_family = AF_INET6;
  162. }
  163. rc = inet_pton(sa.sin_family, ip, &sa.sin_addr);
  164. if (rc == 0) {
  165. return -EINVAL;
  166. }
  167. rc = getnameinfo((struct sockaddr*)&sa, sizeof(sa),
  168. name_out, CS_MAX_NAME_LENGTH, NULL, 0, 0);
  169. if (rc != 0) {
  170. syslog (LOG_ERR, "error looking up %s : %s\n", ip, gai_strerror(rc));
  171. return -EINVAL;
  172. }
  173. return 0;
  174. }
  175. static void
  176. _cs_confdb_key_changed(confdb_handle_t handle,
  177. confdb_change_type_t change_type,
  178. hdb_handle_t parent_object_handle,
  179. hdb_handle_t object_handle,
  180. const void *object_name_pt, size_t object_name_len,
  181. const void *key_name_pt, size_t key_name_len,
  182. const void *key_value_pt, size_t key_value_len)
  183. {
  184. char parent_name[CS_MAX_NAME_LENGTH];
  185. size_t len = 0;
  186. hdb_handle_t real_parent_object_handle;
  187. cs_error_t rc = CS_OK;
  188. char nodename[CS_MAX_NAME_LENGTH];
  189. char nodeid_str[CS_MAX_NAME_LENGTH];
  190. uint32_t nodeid;
  191. char status[CS_MAX_NAME_LENGTH];
  192. char ip[CS_MAX_NAME_LENGTH];
  193. size_t ip_len;
  194. confdb_value_types_t type;
  195. char* open_bracket = NULL;
  196. char* close_bracket = NULL;
  197. rc = confdb_object_parent_get (handle,
  198. parent_object_handle, &real_parent_object_handle);
  199. assert(rc == CS_OK);
  200. rc = confdb_object_name_get (handle,
  201. real_parent_object_handle,
  202. parent_name,
  203. &len);
  204. parent_name[len] = '\0';
  205. assert(rc == CS_OK);
  206. if (strcmp(parent_name, "members") == 0) {
  207. if (strncmp(key_name_pt, "status", strlen("status")) == 0) {
  208. memcpy(nodeid_str, object_name_pt, object_name_len);
  209. nodeid_str[object_name_len] = '\0';
  210. nodeid = atoi(nodeid_str);
  211. memcpy(status, key_value_pt, key_value_len);
  212. status[key_value_len] = '\0';
  213. rc = confdb_key_get_typed(handle, parent_object_handle,
  214. "ip", ip, &ip_len, &type);
  215. assert(rc == CS_OK);
  216. ip[ip_len-1] = '\0';
  217. /*
  218. * We want the ip out of: "r(0) ip(192.168.100.92)"
  219. */
  220. open_bracket = strrchr(ip, '(');
  221. open_bracket++;
  222. close_bracket = strrchr(open_bracket, ')');
  223. *close_bracket = '\0';
  224. _cs_ip_to_hostname(open_bracket, nodename);
  225. _cs_node_membership_event(nodename, nodeid, status, open_bracket);
  226. }
  227. }
  228. }
  229. static void
  230. _cs_confdb_object_created(confdb_handle_t handle,
  231. hdb_handle_t parent_object_handle,
  232. hdb_handle_t object_handle,
  233. const void *name_pt,
  234. size_t name_len)
  235. {
  236. char parent_name[CS_MAX_NAME_LENGTH];
  237. size_t len = 0;
  238. char obj_name[CS_MAX_NAME_LENGTH];
  239. hdb_handle_t real_parent_object_handle;
  240. cs_error_t rc = CS_OK;
  241. memcpy(obj_name, name_pt, name_len);
  242. obj_name[name_len] = '\0';
  243. rc = confdb_object_parent_get (handle,
  244. parent_object_handle, &real_parent_object_handle);
  245. if (rc != CS_OK) {
  246. /* this error is normally from our own cfg connection
  247. * which is short lived.
  248. */
  249. return;
  250. }
  251. rc = confdb_object_name_get (handle,
  252. real_parent_object_handle, parent_name, &len);
  253. parent_name[len] = '\0';
  254. assert(rc == CS_OK);
  255. if (strcmp(parent_name, "connections") == 0) {
  256. _cs_application_connection_event(obj_name, "connected");
  257. }
  258. }
  259. static void
  260. _cs_confdb_object_deleted(confdb_handle_t handle,
  261. hdb_handle_t parent_object_handle,
  262. const void *name_pt,
  263. size_t name_len)
  264. {
  265. char obj_name[CS_MAX_NAME_LENGTH];
  266. char parent_name[CS_MAX_NAME_LENGTH];
  267. size_t len = 0;
  268. cs_error_t rc;
  269. memcpy(obj_name, name_pt, name_len);
  270. obj_name[name_len] = '\0';
  271. rc = confdb_object_name_get (handle,
  272. parent_object_handle, parent_name, &len);
  273. parent_name[len] = '\0';
  274. assert(rc == CS_OK);
  275. if (strcmp(parent_name, "connections") == 0) {
  276. _cs_application_connection_event(obj_name, "disconnected");
  277. }
  278. }
  279. static cs_error_t
  280. _cs_confdb_find_object (confdb_handle_t handle,
  281. const char * name_pt,
  282. hdb_handle_t * out_handle)
  283. {
  284. char * obj_name_pt;
  285. char * save_pt;
  286. hdb_handle_t obj_handle;
  287. confdb_handle_t parent_object_handle = OBJECT_PARENT_HANDLE;
  288. char tmp_name[CS_MAX_NAME_LENGTH];
  289. cs_error_t res = CS_OK;
  290. strncpy (tmp_name, name_pt, CS_MAX_NAME_LENGTH);
  291. obj_name_pt = strtok_r(tmp_name, SEPERATOR_STR, &save_pt);
  292. while (obj_name_pt != NULL) {
  293. res = confdb_object_find_start(handle, parent_object_handle);
  294. if (res != CS_OK) {
  295. syslog (LOG_ERR, "Could not start object_find %d\n", res);
  296. exit (EXIT_FAILURE);
  297. }
  298. res = confdb_object_find(handle, parent_object_handle,
  299. obj_name_pt, strlen (obj_name_pt), &obj_handle);
  300. if (res != CS_OK) {
  301. return res;
  302. }
  303. parent_object_handle = obj_handle;
  304. obj_name_pt = strtok_r (NULL, SEPERATOR_STR, &save_pt);
  305. }
  306. *out_handle = parent_object_handle;
  307. return res;
  308. }
  309. static int
  310. _cs_confdb_dispatch(hdb_handle_t handle,
  311. int fd, int revents, void *data)
  312. {
  313. confdb_dispatch(confdb_handle, CONFDB_DISPATCH_ALL);
  314. return 0;
  315. }
  316. static void _cs_quorum_notification(quorum_handle_t handle,
  317. uint32_t quorate, uint64_t ring_seq,
  318. uint32_t view_list_entries, uint32_t *view_list)
  319. {
  320. if (quorate) {
  321. _cs_node_quorum_event("quorate");
  322. } else {
  323. _cs_node_quorum_event("not quorate");
  324. }
  325. }
  326. static int
  327. _cs_quorum_dispatch(hdb_handle_t handle,
  328. int fd, int revents, void *data)
  329. {
  330. quorum_dispatch(quorum_handle, CS_DISPATCH_ALL);
  331. return 0;
  332. }
  333. static void
  334. _cs_quorum_init(void)
  335. {
  336. cs_error_t rc;
  337. int fd;
  338. quorum_callbacks_t quorum_callbacks = {
  339. .quorum_notify_fn = _cs_quorum_notification,
  340. };
  341. rc = quorum_initialize (&quorum_handle, &quorum_callbacks);
  342. if (rc != CS_OK) {
  343. syslog(LOG_ERR, "Could not connect to corosync(quorum)");
  344. return;
  345. }
  346. quorum_fd_get(quorum_handle, &fd);
  347. poll_dispatch_add (poll_handle, fd, POLLIN|POLLNVAL, NULL,
  348. _cs_quorum_dispatch);
  349. quorum_trackstart(quorum_handle, CS_TRACK_CHANGES);
  350. }
  351. static void
  352. _cs_quorum_finalize(void)
  353. {
  354. quorum_finalize (quorum_handle);
  355. }
  356. #ifdef HAVE_DBUS
  357. /*
  358. * dbus notifications
  359. */
  360. static void
  361. _cs_dbus_auto_flush(void)
  362. {
  363. dbus_connection_ref(db);
  364. dbus_connection_read_write(db, 500);
  365. dbus_connection_unref(db);
  366. }
  367. static void
  368. _cs_dbus_release(void)
  369. {
  370. DBusError err;
  371. if (!db)
  372. return;
  373. dbus_error_init(&err);
  374. dbus_bus_release_name(db, DBUS_CS_NAME, &err);
  375. dbus_error_free(&err);
  376. dbus_connection_unref(db);
  377. db = NULL;
  378. }
  379. static void
  380. _cs_dbus_node_quorum_event(char *nodename, uint32_t nodeid, const char *state)
  381. {
  382. DBusMessage *msg = NULL;
  383. int ret = -1;
  384. if (err_set) {
  385. syslog (LOG_ERR, "%s\n", _err);
  386. err_set = 0;
  387. }
  388. if (!db) {
  389. goto out_free;
  390. }
  391. if (dbus_connection_get_is_connected(db) != TRUE) {
  392. err_set = 1;
  393. snprintf(_err, sizeof(_err), "DBus connection lost");
  394. _cs_dbus_release();
  395. goto out_unlock;
  396. }
  397. _cs_dbus_auto_flush();
  398. if (!(msg = dbus_message_new_signal(DBUS_CS_PATH,
  399. DBUS_CS_IFACE,
  400. "QuorumStateChange"))) {
  401. syslog (LOG_ERR, "%s(%d) error\n", __func__, __LINE__);
  402. goto out_unlock;
  403. }
  404. if (!dbus_message_append_args(msg,
  405. DBUS_TYPE_STRING, &nodename,
  406. DBUS_TYPE_UINT32, &nodeid,
  407. DBUS_TYPE_STRING, &state,
  408. DBUS_TYPE_INVALID)) {
  409. syslog (LOG_ERR, "%s(%d) error\n", __func__, __LINE__);
  410. goto out_unlock;
  411. }
  412. dbus_connection_send(db, msg, NULL);
  413. ret = 0;
  414. out_unlock:
  415. if (ret == -1) {
  416. syslog (LOG_ERR, "%s() error\n", __func__);
  417. }
  418. if (msg)
  419. dbus_message_unref(msg);
  420. out_free:
  421. return;
  422. }
  423. static void
  424. _cs_dbus_node_membership_event(char *nodename, uint32_t nodeid, char *state, char* ip)
  425. {
  426. DBusMessage *msg = NULL;
  427. int ret = -1;
  428. if (err_set) {
  429. syslog (LOG_ERR, "%s\n", _err);
  430. err_set = 0;
  431. }
  432. if (!db) {
  433. goto out_free;
  434. }
  435. if (dbus_connection_get_is_connected(db) != TRUE) {
  436. err_set = 1;
  437. snprintf(_err, sizeof(_err), "DBus connection lost");
  438. _cs_dbus_release();
  439. goto out_unlock;
  440. }
  441. _cs_dbus_auto_flush();
  442. if (!(msg = dbus_message_new_signal(DBUS_CS_PATH,
  443. DBUS_CS_IFACE,
  444. "NodeStateChange"))) {
  445. syslog (LOG_ERR, "%s(%d) error\n", __func__, __LINE__);
  446. goto out_unlock;
  447. }
  448. if (!dbus_message_append_args(msg,
  449. DBUS_TYPE_STRING, &nodename,
  450. DBUS_TYPE_UINT32, &nodeid,
  451. DBUS_TYPE_STRING, &ip,
  452. DBUS_TYPE_STRING, &state,
  453. DBUS_TYPE_INVALID)) {
  454. syslog (LOG_ERR, "%s(%d) error\n", __func__, __LINE__);
  455. goto out_unlock;
  456. }
  457. dbus_connection_send(db, msg, NULL);
  458. ret = 0;
  459. out_unlock:
  460. if (ret == -1) {
  461. syslog (LOG_ERR, "%s() error\n", __func__);
  462. }
  463. if (msg)
  464. dbus_message_unref(msg);
  465. out_free:
  466. return;
  467. }
  468. static void
  469. _cs_dbus_application_connection_event(char *nodename, uint32_t nodeid, char *app_name, const char *state)
  470. {
  471. DBusMessage *msg = NULL;
  472. int ret = -1;
  473. if (err_set) {
  474. syslog (LOG_ERR, "%s\n", _err);
  475. err_set = 0;
  476. }
  477. if (!db) {
  478. goto out_free;
  479. }
  480. if (dbus_connection_get_is_connected(db) != TRUE) {
  481. err_set = 1;
  482. snprintf(_err, sizeof(_err), "DBus connection lost");
  483. _cs_dbus_release();
  484. goto out_unlock;
  485. }
  486. _cs_dbus_auto_flush();
  487. if (!(msg = dbus_message_new_signal(DBUS_CS_PATH,
  488. DBUS_CS_IFACE,
  489. "ConnectionStateChange"))) {
  490. syslog (LOG_ERR, "%s(%d) error\n", __func__, __LINE__);
  491. goto out_unlock;
  492. }
  493. if (!dbus_message_append_args(msg,
  494. DBUS_TYPE_STRING, &nodename,
  495. DBUS_TYPE_UINT32, &nodeid,
  496. DBUS_TYPE_STRING, &app_name,
  497. DBUS_TYPE_STRING, &state,
  498. DBUS_TYPE_INVALID)) {
  499. syslog (LOG_ERR, "%s(%d) error\n", __func__, __LINE__);
  500. goto out_unlock;
  501. }
  502. dbus_connection_send(db, msg, NULL);
  503. ret = 0;
  504. out_unlock:
  505. if (msg)
  506. dbus_message_unref(msg);
  507. out_free:
  508. return;
  509. }
  510. static void
  511. _cs_dbus_init(void)
  512. {
  513. DBusConnection *dbc = NULL;
  514. DBusError err;
  515. dbus_error_init(&err);
  516. dbc = dbus_bus_get(DBUS_BUS_SYSTEM, &err);
  517. if (!dbc) {
  518. snprintf(_err, sizeof(_err),
  519. "dbus_bus_get: %s", err.message);
  520. err_set = 1;
  521. dbus_error_free(&err);
  522. return;
  523. }
  524. dbus_connection_set_exit_on_disconnect(dbc, FALSE);
  525. db = dbc;
  526. notifiers[num_notifiers].node_membership_fn =
  527. _cs_dbus_node_membership_event;
  528. notifiers[num_notifiers].node_quorum_fn =
  529. _cs_dbus_node_quorum_event;
  530. notifiers[num_notifiers].application_connection_fn =
  531. _cs_dbus_application_connection_event;
  532. num_notifiers++;
  533. }
  534. #endif /* HAVE_DBUS */
  535. #ifdef ENABLE_SNMP
  536. static netsnmp_session *snmp_init (const char *target)
  537. {
  538. static netsnmp_session *session = NULL;
  539. #ifndef NETSNMPV54
  540. char default_port[128];
  541. snprintf (default_port, sizeof (default_port), "%s:162", target);
  542. #endif
  543. if (session) {
  544. return (session);
  545. }
  546. if (target == NULL) {
  547. return NULL;
  548. }
  549. session = malloc (sizeof (netsnmp_session));
  550. snmp_sess_init (session);
  551. session->version = SNMP_VERSION_2c;
  552. session->callback = NULL;
  553. session->callback_magic = NULL;
  554. session = snmp_add(session,
  555. #ifdef NETSNMPV54
  556. netsnmp_transport_open_client ("snmptrap", target),
  557. #else
  558. netsnmp_tdomain_transport (default_port, 0, "udp"),
  559. #endif
  560. NULL, NULL);
  561. if (session == NULL) {
  562. syslog(LOG_ERR, "Could not create snmp transport");
  563. }
  564. return (session);
  565. }
  566. static inline void add_field (
  567. netsnmp_pdu *trap_pdu,
  568. u_char asn_type,
  569. const char *prefix,
  570. void *value,
  571. size_t value_size)
  572. {
  573. oid _oid[MAX_OID_LEN];
  574. size_t _oid_len = MAX_OID_LEN;
  575. if (snmp_parse_oid(prefix, _oid, &_oid_len)) {
  576. snmp_pdu_add_variable (trap_pdu, _oid, _oid_len, asn_type, (u_char *) value, value_size);
  577. }
  578. }
  579. static void
  580. _cs_snmp_node_membership_event(char *nodename, uint32_t nodeid, char *state, char* ip)
  581. {
  582. int ret;
  583. char csysuptime[CS_TIMESTAMP_STR_LEN];
  584. static oid snmptrap_oid[] = { 1,3,6,1,6,3,1,1,4,1,0 };
  585. static oid sysuptime_oid[] = { 1,3,6,1,2,1,1,3,0 };
  586. time_t now = time (NULL);
  587. netsnmp_pdu *trap_pdu;
  588. netsnmp_session *session = snmp_init (snmp_manager);
  589. if (session == NULL) {
  590. syslog (LOG_NOTICE, "Failed to init SNMP session.\n");
  591. return ;
  592. }
  593. trap_pdu = snmp_pdu_create (SNMP_MSG_TRAP2);
  594. if (!trap_pdu) {
  595. syslog (LOG_NOTICE, "Failed to create SNMP notification.\n");
  596. return ;
  597. }
  598. /* send uptime */
  599. snprintf (csysuptime, CS_TIMESTAMP_STR_LEN, "%ld", now);
  600. snmp_add_var (trap_pdu, sysuptime_oid, sizeof (sysuptime_oid) / sizeof (oid), 't', csysuptime);
  601. snmp_add_var (trap_pdu, snmptrap_oid, sizeof (snmptrap_oid) / sizeof (oid), 'o', SNMP_OID_TRAPS_NODE);
  602. /* Add extries to the trap */
  603. add_field (trap_pdu, ASN_OCTET_STR, SNMP_OID_OBJECT_NODE_NAME, (void*)nodename, strlen (nodename));
  604. add_field (trap_pdu, ASN_INTEGER, SNMP_OID_OBJECT_NODE_ID, (void*)&nodeid, sizeof (nodeid));
  605. add_field (trap_pdu, ASN_OCTET_STR, SNMP_OID_OBJECT_NODE_ADDR, (void*)ip, strlen (ip));
  606. add_field (trap_pdu, ASN_OCTET_STR, SNMP_OID_OBJECT_NODE_STATUS, (void*)state, strlen (state));
  607. /* Send and cleanup */
  608. ret = snmp_send (session, trap_pdu);
  609. if (ret == 0) {
  610. /* error */
  611. syslog (LOG_ERR, "Could not send SNMP trap");
  612. snmp_free_pdu (trap_pdu);
  613. }
  614. }
  615. static void
  616. _cs_snmp_node_quorum_event(char *nodename, uint32_t nodeid,
  617. const char *state)
  618. {
  619. int ret;
  620. char csysuptime[20];
  621. static oid snmptrap_oid[] = { 1,3,6,1,6,3,1,1,4,1,0 };
  622. static oid sysuptime_oid[] = { 1,3,6,1,2,1,1,3,0 };
  623. time_t now = time (NULL);
  624. netsnmp_pdu *trap_pdu;
  625. netsnmp_session *session = snmp_init (snmp_manager);
  626. if (session == NULL) {
  627. syslog (LOG_NOTICE, "Failed to init SNMP session.\n");
  628. return ;
  629. }
  630. trap_pdu = snmp_pdu_create (SNMP_MSG_TRAP2);
  631. if (!trap_pdu) {
  632. syslog (LOG_NOTICE, "Failed to create SNMP notification.\n");
  633. return ;
  634. }
  635. /* send uptime */
  636. sprintf (csysuptime, "%ld", now);
  637. snmp_add_var (trap_pdu, sysuptime_oid, sizeof (sysuptime_oid) / sizeof (oid), 't', csysuptime);
  638. snmp_add_var (trap_pdu, snmptrap_oid, sizeof (snmptrap_oid) / sizeof (oid), 'o', SNMP_OID_TRAPS_NODE);
  639. /* Add extries to the trap */
  640. add_field (trap_pdu, ASN_OCTET_STR, SNMP_OID_OBJECT_NODE_NAME, (void*)nodename, strlen (nodename));
  641. add_field (trap_pdu, ASN_INTEGER, SNMP_OID_OBJECT_NODE_ID, (void*)&nodeid, sizeof (nodeid));
  642. add_field (trap_pdu, ASN_OCTET_STR, SNMP_OID_OBJECT_QUORUM, (void*)state, strlen (state));
  643. /* Send and cleanup */
  644. ret = snmp_send (session, trap_pdu);
  645. if (ret == 0) {
  646. /* error */
  647. syslog (LOG_ERR, "Could not send SNMP trap");
  648. snmp_free_pdu (trap_pdu);
  649. }
  650. }
  651. static void
  652. _cs_snmp_init(void)
  653. {
  654. if (snmp_manager == NULL) {
  655. snmp_manager = (char*)local_host;
  656. }
  657. notifiers[num_notifiers].node_membership_fn =
  658. _cs_snmp_node_membership_event;
  659. notifiers[num_notifiers].node_quorum_fn =
  660. _cs_snmp_node_quorum_event;
  661. notifiers[num_notifiers].application_connection_fn = NULL;
  662. num_notifiers++;
  663. }
  664. #endif /* ENABLE_SNMP */
  665. static void
  666. _cs_syslog_node_membership_event(char *nodename, uint32_t nodeid, char *state, char* ip)
  667. {
  668. syslog (LOG_NOTICE, "%s[%d] ip:%s %s\n", nodename, nodeid, ip, state);
  669. }
  670. static void
  671. _cs_syslog_node_quorum_event(char *nodename, uint32_t nodeid, const char *state)
  672. {
  673. if (strcmp(state, "quorate") == 0) {
  674. syslog (LOG_NOTICE, "%s[%d] is now %s\n", nodename, nodeid, state);
  675. } else {
  676. syslog (LOG_NOTICE, "%s[%d] has lost quorum\n", nodename, nodeid);
  677. }
  678. }
  679. static void
  680. _cs_syslog_application_connection_event(char *nodename, uint32_t nodeid, char* app_name, const char *state)
  681. {
  682. if (strcmp(state, "connected") == 0) {
  683. syslog (LOG_ERR, "%s[%d] %s is now %s to corosync\n", nodename, nodeid, app_name, state);
  684. } else {
  685. syslog (LOG_ERR, "%s[%d] %s is now %s from corosync\n", nodename, nodeid, app_name, state);
  686. }
  687. }
  688. static void
  689. _cs_node_membership_event(char *nodename, uint32_t nodeid, char *state, char* ip)
  690. {
  691. int i;
  692. for (i = 0; i < num_notifiers; i++) {
  693. if (notifiers[i].node_membership_fn) {
  694. notifiers[i].node_membership_fn(nodename, nodeid, state, ip);
  695. }
  696. }
  697. }
  698. static void
  699. _cs_local_node_info_get(char **nodename, uint32_t *nodeid)
  700. {
  701. cs_error_t rc;
  702. corosync_cfg_handle_t cfg_handle;
  703. if (local_nodeid == 0) {
  704. corosync_cfg_initialize(&cfg_handle, NULL);
  705. rc = corosync_cfg_local_get (cfg_handle, &local_nodeid);
  706. corosync_cfg_finalize(cfg_handle);
  707. if (rc != CS_OK) {
  708. local_nodeid = 0;
  709. strncpy(local_nodename, "localhost", CS_MAX_NAME_LENGTH);
  710. } else {
  711. gethostname(local_nodename, CS_MAX_NAME_LENGTH);
  712. }
  713. }
  714. *nodeid = local_nodeid;
  715. *nodename = local_nodename;
  716. }
  717. static void
  718. _cs_node_quorum_event(const char *state)
  719. {
  720. int i;
  721. char *nodename;
  722. uint32_t nodeid;
  723. _cs_local_node_info_get(&nodename, &nodeid);
  724. for (i = 0; i < num_notifiers; i++) {
  725. if (notifiers[i].node_quorum_fn) {
  726. notifiers[i].node_quorum_fn(nodename, nodeid, state);
  727. }
  728. }
  729. }
  730. static void
  731. _cs_application_connection_event(char *app_name, const char *state)
  732. {
  733. int i;
  734. char *nodename;
  735. uint32_t nodeid;
  736. _cs_local_node_info_get(&nodename, &nodeid);
  737. for (i = 0; i < num_notifiers; i++) {
  738. if (notifiers[i].application_connection_fn) {
  739. notifiers[i].application_connection_fn(nodename, nodeid, app_name, state);
  740. }
  741. }
  742. }
  743. static void
  744. sig_exit_handler (int num)
  745. {
  746. poll_stop(poll_handle);
  747. }
  748. static void
  749. _cs_confdb_init(void)
  750. {
  751. hdb_handle_t obj_handle;
  752. cs_error_t rc;
  753. int conf_fd = 0;
  754. rc = confdb_initialize (&confdb_handle, &callbacks);
  755. if (rc != CS_OK) {
  756. syslog (LOG_ERR, "Failed to initialize the objdb API. Error %d\n", rc);
  757. exit (EXIT_FAILURE);
  758. }
  759. confdb_fd_get(confdb_handle, &conf_fd);
  760. poll_dispatch_add (poll_handle, conf_fd, POLLIN|POLLNVAL, NULL,
  761. _cs_confdb_dispatch);
  762. rc = _cs_confdb_find_object (confdb_handle, "runtime.connections.",
  763. &obj_handle);
  764. if (rc != CS_OK) {
  765. syslog (LOG_ERR,
  766. "Failed to find the connections object. Error %d\n", rc);
  767. exit (EXIT_FAILURE);
  768. }
  769. rc = confdb_track_changes (confdb_handle, obj_handle,
  770. CONFDB_TRACK_DEPTH_ONE);
  771. if (rc != CS_OK) {
  772. syslog (LOG_ERR,
  773. "Failed to track the connections object. Error %d\n", rc);
  774. exit (EXIT_FAILURE);
  775. }
  776. rc = _cs_confdb_find_object(confdb_handle,
  777. "runtime.totem.pg.mrp.srp.members.", &obj_handle);
  778. if (rc != CS_OK) {
  779. syslog (LOG_ERR, "Failed to find the object. Error %d\n", rc);
  780. exit (EXIT_FAILURE);
  781. }
  782. rc = confdb_track_changes(confdb_handle,
  783. obj_handle, CONFDB_TRACK_DEPTH_RECURSIVE);
  784. if (rc != CS_OK) {
  785. syslog (LOG_ERR,
  786. "Failed to track the object. Error %d\n", rc);
  787. exit (EXIT_FAILURE);
  788. }
  789. }
  790. static void
  791. _cs_confdb_finalize(void)
  792. {
  793. confdb_stop_track_changes (confdb_handle);
  794. confdb_finalize (confdb_handle);
  795. }
  796. static void
  797. _cs_check_config(void)
  798. {
  799. if (conf[CS_NTF_LOG] == 0 &&
  800. conf[CS_NTF_STDOUT] == 0 &&
  801. conf[CS_NTF_SNMP] == 0 &&
  802. conf[CS_NTF_DBUS] == 0) {
  803. syslog(LOG_ERR, "no event type enabled, see corosync-notifyd -h, exiting.");
  804. exit(EXIT_FAILURE);
  805. }
  806. #ifndef ENABLE_SNMP
  807. if (conf[CS_NTF_SNMP]) {
  808. syslog(LOG_ERR, "Not compiled with SNMP support enabled, exiting.");
  809. exit(EXIT_FAILURE);
  810. }
  811. #endif
  812. #ifndef HAVE_DBUS
  813. if (conf[CS_NTF_DBUS]) {
  814. syslog(LOG_ERR, "Not compiled with DBus support enabled, exiting.");
  815. exit(EXIT_FAILURE);
  816. }
  817. #endif
  818. if (conf[CS_NTF_STDOUT] && !conf[CS_NTF_FG]) {
  819. syslog(LOG_ERR, "configured to print to stdout and run in the background, exiting");
  820. exit(EXIT_FAILURE);
  821. }
  822. if (conf[CS_NTF_SNMP] && conf[CS_NTF_DBUS]) {
  823. syslog(LOG_ERR, "configured to send snmp traps and dbus signals - are you sure?.");
  824. }
  825. }
  826. static void
  827. _cs_usage(void)
  828. {
  829. fprintf(stderr, "usage:\n"\
  830. " -f : Start application in foreground.\n"\
  831. " -l : Log all events.\n"\
  832. " -o : Print events to stdout (turns on -l).\n"\
  833. " -s : Send SNMP traps on all events.\n"\
  834. " -m : SNMP Manager IP address (defaults to localhost).\n"\
  835. " -d : Send DBUS signals on all events.\n"\
  836. " -h : Print this help\n\n");
  837. }
  838. int
  839. main(int argc, char *argv[])
  840. {
  841. int ch;
  842. conf[CS_NTF_FG] = 0;
  843. conf[CS_NTF_LOG] = 0;
  844. conf[CS_NTF_STDOUT] = 0;
  845. conf[CS_NTF_SNMP] = 0;
  846. conf[CS_NTF_DBUS] = 0;
  847. while ((ch = getopt (argc, argv, "floshdm:")) != EOF) {
  848. switch (ch) {
  849. case 'f':
  850. conf[CS_NTF_FG] = 1;
  851. break;
  852. case 'l':
  853. conf[CS_NTF_LOG] = 1;
  854. break;
  855. case 'm':
  856. conf[CS_NTF_SNMP] = 1;
  857. strncpy(snmp_manager_buf, optarg, CS_MAX_NAME_LENGTH);
  858. snmp_manager = snmp_manager_buf;
  859. break;
  860. case 'o':
  861. conf[CS_NTF_LOG] = 1;
  862. conf[CS_NTF_STDOUT] = 1;
  863. break;
  864. case 's':
  865. conf[CS_NTF_SNMP] = 1;
  866. break;
  867. case 'd':
  868. conf[CS_NTF_DBUS] = 1;
  869. break;
  870. case 'h':
  871. default:
  872. _cs_usage();
  873. return EXIT_FAILURE;
  874. }
  875. }
  876. if (conf[CS_NTF_STDOUT]) {
  877. openlog(NULL, LOG_PID|LOG_PERROR, LOG_DAEMON);
  878. } else {
  879. openlog(NULL, LOG_PID, LOG_DAEMON);
  880. }
  881. _cs_check_config();
  882. if (!conf[CS_NTF_FG]) {
  883. daemon(0, 0);
  884. }
  885. num_notifiers = 0;
  886. if (conf[CS_NTF_LOG]) {
  887. notifiers[num_notifiers].node_membership_fn =
  888. _cs_syslog_node_membership_event;
  889. notifiers[num_notifiers].node_quorum_fn =
  890. _cs_syslog_node_quorum_event;
  891. notifiers[num_notifiers].application_connection_fn =
  892. _cs_syslog_application_connection_event;
  893. num_notifiers++;
  894. }
  895. poll_handle = poll_create();
  896. _cs_confdb_init();
  897. _cs_quorum_init();
  898. #ifdef HAVE_DBUS
  899. if (conf[CS_NTF_DBUS]) {
  900. _cs_dbus_init();
  901. }
  902. #endif /* HAVE_DBUS */
  903. #ifdef ENABLE_SNMP
  904. if (conf[CS_NTF_SNMP]) {
  905. _cs_snmp_init();
  906. }
  907. #endif /* ENABLE_SNMP */
  908. (void)signal (SIGINT, sig_exit_handler);
  909. (void)signal (SIGQUIT, sig_exit_handler);
  910. (void)signal (SIGTERM, sig_exit_handler);
  911. poll_run(poll_handle);
  912. #ifdef HAVE_DBUS
  913. if (conf[CS_NTF_DBUS]) {
  914. _cs_dbus_release();
  915. }
  916. #endif /* HAVE_DBUS */
  917. _cs_quorum_finalize();
  918. _cs_confdb_finalize();
  919. return 0;
  920. }