4
0

totemknet.c 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407
  1. /*
  2. * Copyright (c) 2016-2017 Red Hat, Inc.
  3. *
  4. * All rights reserved.
  5. *
  6. * Author: Christine Caulfield (ccaulfie@redhat.com)
  7. * This software licensed under BSD license, the text of which follows:
  8. *
  9. * Redistribution and use in source and binary forms, with or without
  10. * modification, are permitted provided that the following conditions are met:
  11. *
  12. * - Redistributions of source code must retain the above copyright notice,
  13. * this list of conditions and the following disclaimer.
  14. * - Redistributions in binary form must reproduce the above copyright notice,
  15. * this list of conditions and the following disclaimer in the documentation
  16. * and/or other materials provided with the distribution.
  17. * - Neither the name of the MontaVista Software, Inc. nor the names of its
  18. * contributors may be used to endorse or promote products derived from this
  19. * software without specific prior written permission.
  20. *
  21. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  22. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  25. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  26. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  27. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  28. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  29. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  30. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  31. * THE POSSIBILITY OF SUCH DAMAGE.
  32. */
  33. #include <config.h>
  34. #include <assert.h>
  35. #include <sys/mman.h>
  36. #include <sys/types.h>
  37. #include <sys/stat.h>
  38. #include <sys/socket.h>
  39. #include <netdb.h>
  40. #include <sys/un.h>
  41. #include <sys/ioctl.h>
  42. #include <sys/param.h>
  43. #include <netinet/in.h>
  44. #include <arpa/inet.h>
  45. #include <unistd.h>
  46. #include <fcntl.h>
  47. #include <stdlib.h>
  48. #include <stdio.h>
  49. #include <errno.h>
  50. #include <sched.h>
  51. #include <time.h>
  52. #include <sys/time.h>
  53. #include <sys/poll.h>
  54. #include <sys/uio.h>
  55. #include <limits.h>
  56. #include <qb/qbdefs.h>
  57. #include <qb/qbloop.h>
  58. #include <corosync/sq.h>
  59. #include <corosync/swab.h>
  60. #include <corosync/logsys.h>
  61. #include <corosync/icmap.h>
  62. #include <corosync/totem/totemip.h>
  63. #include "totemknet.h"
  64. #include "util.h"
  65. #include <libknet.h>
  66. #include <corosync/totem/totemstats.h>
  67. #ifndef MSG_NOSIGNAL
  68. #define MSG_NOSIGNAL 0
  69. #endif
  70. /* Should match that used by cfg */
  71. #define CFG_INTERFACE_STATUS_MAX_LEN 512
  72. struct totemknet_instance {
  73. struct crypto_instance *crypto_inst;
  74. qb_loop_t *poll_handle;
  75. knet_handle_t knet_handle;
  76. int link_mode;
  77. void *context;
  78. void (*totemknet_deliver_fn) (
  79. void *context,
  80. const void *msg,
  81. unsigned int msg_len);
  82. void (*totemknet_iface_change_fn) (
  83. void *context,
  84. const struct totem_ip_address *iface_address,
  85. unsigned int link_no);
  86. void (*totemknet_mtu_changed) (
  87. void *context,
  88. int net_mtu);
  89. void (*totemknet_target_set_completed) (void *context);
  90. /*
  91. * Function and data used to log messages
  92. */
  93. int totemknet_log_level_security;
  94. int totemknet_log_level_error;
  95. int totemknet_log_level_warning;
  96. int totemknet_log_level_notice;
  97. int totemknet_log_level_debug;
  98. int totemknet_subsys_id;
  99. int knet_subsys_id;
  100. void (*totemknet_log_printf) (
  101. int level,
  102. int subsys,
  103. const char *function,
  104. const char *file,
  105. int line,
  106. const char *format,
  107. ...)__attribute__((format(printf, 6, 7)));
  108. void *knet_context;
  109. char iov_buffer[KNET_MAX_PACKET_SIZE];
  110. char *link_status[INTERFACE_MAX];
  111. struct totem_ip_address my_ids[INTERFACE_MAX];
  112. uint16_t ip_port[INTERFACE_MAX];
  113. int our_nodeid;
  114. struct totem_config *totem_config;
  115. struct totem_ip_address token_target;
  116. qb_loop_timer_handle timer_netif_check_timeout;
  117. qb_loop_timer_handle timer_merge_detect_timeout;
  118. int send_merge_detect_message;
  119. unsigned int merge_detect_messages_sent_before_timeout;
  120. int logpipes[2];
  121. int knet_fd;
  122. };
  123. /* Awkward. But needed to get stats from knet */
  124. struct totemknet_instance *global_instance;
  125. struct work_item {
  126. const void *msg;
  127. unsigned int msg_len;
  128. struct totemknet_instance *instance;
  129. };
  130. int totemknet_member_list_rebind_ip (
  131. void *knet_context);
  132. static void totemknet_start_merge_detect_timeout(
  133. void *knet_context);
  134. static void totemknet_stop_merge_detect_timeout(
  135. void *knet_context);
  136. static void log_flush_messages (
  137. void *knet_context);
  138. static void totemknet_instance_initialize (struct totemknet_instance *instance)
  139. {
  140. memset (instance, 0, sizeof (struct totemknet_instance));
  141. }
  142. #define knet_log_printf(level, format, args...) \
  143. do { \
  144. instance->totemknet_log_printf ( \
  145. level, instance->totemknet_subsys_id, \
  146. __FUNCTION__, __FILE__, __LINE__, \
  147. (const char *)format, ##args); \
  148. } while (0);
  149. #define libknet_log_printf(level, format, args...) \
  150. do { \
  151. instance->totemknet_log_printf ( \
  152. level, instance->knet_subsys_id, \
  153. __FUNCTION__, "libknet.h", __LINE__, \
  154. (const char *)format, ##args); \
  155. } while (0);
  156. #define KNET_LOGSYS_PERROR(err_num, level, fmt, args...) \
  157. do { \
  158. char _error_str[LOGSYS_MAX_PERROR_MSG_LEN]; \
  159. const char *_error_ptr = qb_strerror_r(err_num, _error_str, sizeof(_error_str)); \
  160. instance->totemknet_log_printf ( \
  161. level, instance->totemknet_subsys_id, \
  162. __FUNCTION__, __FILE__, __LINE__, \
  163. fmt ": %s (%d)", ##args, _error_ptr, err_num); \
  164. } while(0)
  165. static int dst_host_filter_callback_fn(void *private_data,
  166. const unsigned char *outdata,
  167. ssize_t outdata_len,
  168. uint8_t tx_rx,
  169. knet_node_id_t this_host_id,
  170. knet_node_id_t src_host_id,
  171. int8_t *channel,
  172. knet_node_id_t *dst_host_ids,
  173. size_t *dst_host_ids_entries)
  174. {
  175. struct totem_message_header *header = (struct totem_message_header *)outdata;
  176. int res;
  177. *channel = 0;
  178. if (header->target_nodeid) {
  179. dst_host_ids[0] = header->target_nodeid;
  180. *dst_host_ids_entries = 1;
  181. res = 0; /* unicast message */
  182. }
  183. else {
  184. *dst_host_ids_entries = 0;
  185. res = 1; /* multicast message */
  186. }
  187. return res;
  188. }
  189. static void socket_error_callback_fn(void *private_data, int datafd, int8_t channel, uint8_t tx_rx, int error, int errorno)
  190. {
  191. struct totemknet_instance *instance = (struct totemknet_instance *)private_data;
  192. knet_log_printf (LOGSYS_LEVEL_DEBUG, "Knet socket ERROR notification called: txrx=%d, error=%d, errorno=%d", tx_rx, error, errorno);
  193. if ((error == -1 && errorno != EAGAIN) || (error == 0)) {
  194. knet_handle_remove_datafd(instance->knet_handle, datafd);
  195. }
  196. }
  197. static void host_change_callback_fn(void *private_data, knet_node_id_t host_id, uint8_t reachable, uint8_t remote, uint8_t external)
  198. {
  199. struct totemknet_instance *instance = (struct totemknet_instance *)private_data;
  200. // TODO: what? if anything.
  201. knet_log_printf (LOGSYS_LEVEL_DEBUG, "Knet host change callback. nodeid: %d reachable: %d", host_id, reachable);
  202. }
  203. static void pmtu_change_callback_fn(void *private_data, unsigned int data_mtu)
  204. {
  205. struct totemknet_instance *instance = (struct totemknet_instance *)private_data;
  206. knet_log_printf (LOGSYS_LEVEL_DEBUG, "Knet pMTU change: %d", data_mtu);
  207. /* We don't need to tell corosync the actual knet MTU */
  208. // instance->totemknet_mtu_changed(instance->context, data_mtu);
  209. }
  210. int totemknet_crypto_set (
  211. void *knet_context,
  212. const char *cipher_type,
  213. const char *hash_type)
  214. {
  215. return (0);
  216. }
  217. static inline void ucast_sendmsg (
  218. struct totemknet_instance *instance,
  219. struct totem_ip_address *system_to,
  220. const void *msg,
  221. unsigned int msg_len)
  222. {
  223. int res = 0;
  224. struct totem_message_header *header = (struct totem_message_header *)msg;
  225. struct msghdr msg_ucast;
  226. struct iovec iovec;
  227. header->target_nodeid = system_to->nodeid;
  228. iovec.iov_base = (void *)msg;
  229. iovec.iov_len = msg_len;
  230. /*
  231. * Build unicast message
  232. */
  233. memset(&msg_ucast, 0, sizeof(msg_ucast));
  234. msg_ucast.msg_iov = (void *)&iovec;
  235. msg_ucast.msg_iovlen = 1;
  236. #ifdef HAVE_MSGHDR_CONTROL
  237. msg_ucast.msg_control = 0;
  238. #endif
  239. #ifdef HAVE_MSGHDR_CONTROLLEN
  240. msg_ucast.msg_controllen = 0;
  241. #endif
  242. #ifdef HAVE_MSGHDR_FLAGS
  243. msg_ucast.msg_flags = 0;
  244. #endif
  245. #ifdef HAVE_MSGHDR_ACCRIGHTS
  246. msg_ucast.msg_accrights = NULL;
  247. #endif
  248. #ifdef HAVE_MSGHDR_ACCRIGHTSLEN
  249. msg_ucast.msg_accrightslen = 0;
  250. #endif
  251. /*
  252. * Transmit unicast message
  253. * An error here is recovered by totemsrp
  254. */
  255. res = sendmsg (instance->knet_fd, &msg_ucast, MSG_NOSIGNAL);
  256. if (res < 0) {
  257. KNET_LOGSYS_PERROR (errno, instance->totemknet_log_level_debug,
  258. "sendmsg(ucast) failed (non-critical)");
  259. }
  260. }
  261. static inline void mcast_sendmsg (
  262. struct totemknet_instance *instance,
  263. const void *msg,
  264. unsigned int msg_len,
  265. int only_active)
  266. {
  267. int res;
  268. struct totem_message_header *header = (struct totem_message_header *)msg;
  269. struct msghdr msg_mcast;
  270. struct iovec iovec;
  271. iovec.iov_base = (void *)msg;
  272. iovec.iov_len = msg_len;
  273. header->target_nodeid = 0;
  274. /*
  275. * Build multicast message
  276. */
  277. memset(&msg_mcast, 0, sizeof(msg_mcast));
  278. msg_mcast.msg_iov = (void *)&iovec;
  279. msg_mcast.msg_iovlen = 1;
  280. #ifdef HAVE_MSGHDR_CONTROL
  281. msg_mcast.msg_control = 0;
  282. #endif
  283. #ifdef HAVE_MSGHDR_CONTROLLEN
  284. msg_mcast.msg_controllen = 0;
  285. #endif
  286. #ifdef HAVE_MSGHDR_FLAGS
  287. msg_mcast.msg_flags = 0;
  288. #endif
  289. #ifdef HAVE_MSGHDR_ACCRIGHTS
  290. msg_mcast.msg_accrights = NULL;
  291. #endif
  292. #ifdef HAVE_MSGHDR_ACCRIGHTSLEN
  293. msg_mcast.msg_accrightslen = 0;
  294. #endif
  295. // log_printf (LOGSYS_LEVEL_DEBUG, "totemknet: mcast_sendmsg. only_active=%d, len=%d", only_active, msg_len);
  296. res = sendmsg (instance->knet_fd, &msg_mcast, MSG_NOSIGNAL);
  297. if (res < msg_len) {
  298. knet_log_printf (LOGSYS_LEVEL_DEBUG, "totemknet: mcast_send sendmsg returned %d", res);
  299. }
  300. if (!only_active || instance->send_merge_detect_message) {
  301. /*
  302. * Current message was sent to all nodes
  303. */
  304. instance->merge_detect_messages_sent_before_timeout++;
  305. instance->send_merge_detect_message = 0;
  306. }
  307. }
  308. static int node_compare(const void *aptr, const void *bptr)
  309. {
  310. uint16_t a,b;
  311. a = *(uint16_t *)aptr;
  312. b = *(uint16_t *)bptr;
  313. return a > b;
  314. }
  315. int totemknet_ifaces_get (void *knet_context,
  316. char ***status,
  317. unsigned int *iface_count)
  318. {
  319. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  320. struct knet_link_status link_status;
  321. knet_node_id_t host_list[KNET_MAX_HOST];
  322. uint8_t link_list[KNET_MAX_LINK];
  323. size_t num_hosts;
  324. size_t num_links;
  325. int i,j;
  326. char *ptr;
  327. int res = 0;
  328. /*
  329. * Don't do the whole 'link_info' bit if the caller just wants
  330. * a count of interfaces.
  331. */
  332. if (status) {
  333. res = knet_host_get_host_list(instance->knet_handle,
  334. host_list, &num_hosts);
  335. if (res) {
  336. return (-1);
  337. }
  338. qsort(host_list, num_hosts, sizeof(uint16_t), node_compare);
  339. for (i=0; i<INTERFACE_MAX; i++) {
  340. memset(instance->link_status[i], 'n', CFG_INTERFACE_STATUS_MAX_LEN-1);
  341. instance->link_status[i][num_hosts] = '\0';
  342. }
  343. /* This is all a bit "inside-out" because "status" is a set of strings per link
  344. * and knet orders things by host
  345. */
  346. for (j=0; j<num_hosts; j++) {
  347. res = knet_link_get_link_list(instance->knet_handle,
  348. host_list[j], link_list, &num_links);
  349. if (res) {
  350. return (-1);
  351. }
  352. for (i=0; i < num_links; i++) {
  353. ptr = instance->link_status[link_list[i]];
  354. res = knet_link_get_status(instance->knet_handle,
  355. host_list[j],
  356. link_list[i],
  357. &link_status,
  358. sizeof(link_status));
  359. if (res == 0) {
  360. ptr[j] = '0' + (link_status.enabled |
  361. link_status.connected<<1 |
  362. link_status.dynconnected<<2);
  363. }
  364. else {
  365. ptr[j] = '?';
  366. }
  367. }
  368. }
  369. *status = instance->link_status;
  370. }
  371. *iface_count = INTERFACE_MAX;
  372. return (res);
  373. }
  374. int totemknet_finalize (
  375. void *knet_context)
  376. {
  377. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  378. int res = 0;
  379. int i,j;
  380. static knet_node_id_t nodes[KNET_MAX_HOST]; /* static to save stack */
  381. uint8_t links[KNET_MAX_LINK];
  382. size_t num_nodes;
  383. size_t num_links;
  384. knet_log_printf(LOG_DEBUG, "totemknet: finalize");
  385. qb_loop_poll_del (instance->poll_handle, instance->logpipes[0]);
  386. qb_loop_poll_del (instance->poll_handle, instance->knet_fd);
  387. res = knet_host_get_host_list(instance->knet_handle, nodes, &num_nodes);
  388. if (res) {
  389. knet_log_printf (LOGSYS_LEVEL_ERROR, "Cannot get knet node list for shutdown: %s", strerror(errno));
  390. /* Crash out anyway */
  391. goto finalise_error;
  392. }
  393. /* Tidily shut down all nodes & links. This ensures that the LEAVE message will be sent */
  394. for (i=0; i<num_nodes; i++) {
  395. res = knet_link_get_link_list(instance->knet_handle, nodes[i], links, &num_links);
  396. if (res) {
  397. knet_log_printf (LOGSYS_LEVEL_ERROR, "Cannot get knet link list for node %d: %s", nodes[i], strerror(errno));
  398. goto finalise_error;
  399. }
  400. for (j=0; j<num_links; j++) {
  401. res = knet_link_set_enable(instance->knet_handle, nodes[i], links[j], 0);
  402. if (res) {
  403. knet_log_printf (LOGSYS_LEVEL_ERROR, "totemknet: knet_link_set_enable(node %d, link %d) failed: %s", nodes[i], links[j], strerror(errno));
  404. }
  405. res = knet_link_clear_config(instance->knet_handle, nodes[i], links[j]);
  406. if (res) {
  407. knet_log_printf (LOGSYS_LEVEL_ERROR, "totemknet: knet_link_clear_config(node %d, link %d) failed: %s", nodes[i], links[j], strerror(errno));
  408. }
  409. }
  410. res = knet_host_remove(instance->knet_handle, nodes[i]);
  411. if (res) {
  412. knet_log_printf (LOGSYS_LEVEL_ERROR, "totemknet: knet_host_remove(node %d) failed: %s", nodes[i], strerror(errno));
  413. }
  414. }
  415. finalise_error:
  416. res = knet_handle_setfwd(instance->knet_handle, 0);
  417. if (res) {
  418. knet_log_printf (LOGSYS_LEVEL_CRIT, "totemknet: knet_handle_setfwd failed: %s", strerror(errno));
  419. }
  420. res = knet_handle_free(instance->knet_handle);
  421. if (res) {
  422. knet_log_printf (LOGSYS_LEVEL_CRIT, "totemknet: knet_handle_free failed: %s", strerror(errno));
  423. }
  424. totemknet_stop_merge_detect_timeout(instance);
  425. log_flush_messages(instance);
  426. return (res);
  427. }
  428. static int log_deliver_fn (
  429. int fd,
  430. int revents,
  431. void *data)
  432. {
  433. struct totemknet_instance *instance = (struct totemknet_instance *)data;
  434. char buffer[KNET_MAX_LOG_MSG_SIZE*4];
  435. char *bufptr = buffer;
  436. int done = 0;
  437. int len;
  438. len = read(fd, buffer, sizeof(buffer));
  439. while (done < len) {
  440. struct knet_log_msg *msg = (struct knet_log_msg *)bufptr;
  441. switch (msg->msglevel) {
  442. case KNET_LOG_ERR:
  443. libknet_log_printf (LOGSYS_LEVEL_ERROR, "%s: %s",
  444. knet_log_get_subsystem_name(msg->subsystem),
  445. msg->msg);
  446. break;
  447. case KNET_LOG_WARN:
  448. libknet_log_printf (LOGSYS_LEVEL_WARNING, "%s: %s",
  449. knet_log_get_subsystem_name(msg->subsystem),
  450. msg->msg);
  451. break;
  452. case KNET_LOG_INFO:
  453. libknet_log_printf (LOGSYS_LEVEL_INFO, "%s: %s",
  454. knet_log_get_subsystem_name(msg->subsystem),
  455. msg->msg);
  456. break;
  457. case KNET_LOG_DEBUG:
  458. libknet_log_printf (LOGSYS_LEVEL_DEBUG, "%s: %s",
  459. knet_log_get_subsystem_name(msg->subsystem),
  460. msg->msg);
  461. break;
  462. }
  463. bufptr += KNET_MAX_LOG_MSG_SIZE;
  464. done += KNET_MAX_LOG_MSG_SIZE;
  465. }
  466. return 0;
  467. }
  468. static int data_deliver_fn (
  469. int fd,
  470. int revents,
  471. void *data)
  472. {
  473. struct totemknet_instance *instance = (struct totemknet_instance *)data;
  474. struct msghdr msg_hdr;
  475. struct iovec iov_recv;
  476. struct sockaddr_storage system_from;
  477. ssize_t msg_len;
  478. iov_recv.iov_base = instance->iov_buffer;
  479. iov_recv.iov_len = KNET_MAX_PACKET_SIZE;
  480. msg_hdr.msg_name = &system_from;
  481. msg_hdr.msg_namelen = sizeof (struct sockaddr_storage);
  482. msg_hdr.msg_iov = &iov_recv;
  483. msg_hdr.msg_iovlen = 1;
  484. #ifdef HAVE_MSGHDR_CONTROL
  485. msg_hdr.msg_control = 0;
  486. #endif
  487. #ifdef HAVE_MSGHDR_CONTROLLEN
  488. msg_hdr.msg_controllen = 0;
  489. #endif
  490. #ifdef HAVE_MSGHDR_FLAGS
  491. msg_hdr.msg_flags = 0;
  492. #endif
  493. #ifdef HAVE_MSGHDR_ACCRIGHTS
  494. msg_hdr.msg_accrights = NULL;
  495. #endif
  496. #ifdef HAVE_MSGHDR_ACCRIGHTSLEN
  497. msg_hdr.msg_accrightslen = 0;
  498. #endif
  499. msg_len = recvmsg (fd, &msg_hdr, MSG_NOSIGNAL | MSG_DONTWAIT);
  500. if (msg_len <= 0) {
  501. return (0);
  502. }
  503. /*
  504. * Handle incoming message
  505. */
  506. instance->totemknet_deliver_fn (
  507. instance->context,
  508. instance->iov_buffer,
  509. msg_len);
  510. return (0);
  511. }
  512. static void timer_function_netif_check_timeout (
  513. void *data)
  514. {
  515. struct totemknet_instance *instance = (struct totemknet_instance *)data;
  516. int i;
  517. for (i=0; i < INTERFACE_MAX; i++) {
  518. if (!instance->totem_config->interfaces[i].configured) {
  519. continue;
  520. }
  521. instance->totemknet_iface_change_fn (instance->context,
  522. &instance->my_ids[i],
  523. i);
  524. }
  525. }
  526. static void totemknet_refresh_config(
  527. int32_t event,
  528. const char *key_name,
  529. struct icmap_notify_value new_val,
  530. struct icmap_notify_value old_val,
  531. void *user_data)
  532. {
  533. uint8_t reloading;
  534. uint32_t value;
  535. uint32_t link_no;
  536. size_t num_nodes;
  537. knet_node_id_t host_ids[KNET_MAX_HOST];
  538. int i;
  539. int err;
  540. char path[ICMAP_KEYNAME_MAXLEN];
  541. struct totemknet_instance *instance = (struct totemknet_instance *)user_data;
  542. ENTER();
  543. /*
  544. * If a full reload is in progress then don't do anything until it's done and
  545. * can reconfigure it all atomically
  546. */
  547. if (icmap_get_uint8("config.totemconfig_reload_in_progress", &reloading) == CS_OK && reloading) {
  548. return;
  549. }
  550. if (icmap_get_uint32("totem.knet_pmtud_interval", &value) == CS_OK) {
  551. instance->totem_config->knet_pmtud_interval = value;
  552. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet_pmtud_interval now %d", value);
  553. err = knet_handle_pmtud_setfreq(instance->knet_handle, instance->totem_config->knet_pmtud_interval);
  554. if (err) {
  555. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_WARNING, "knet_handle_pmtud_setfreq failed");
  556. }
  557. }
  558. /* Get link parameters */
  559. for (i = 0; i < INTERFACE_MAX; i++) {
  560. if (!instance->totem_config->interfaces[i].configured) {
  561. continue;
  562. }
  563. sprintf(path, "totem.interface.%d.knet_link_priority", i);
  564. if (icmap_get_uint32(path, &value) == CS_OK) {
  565. instance->totem_config->interfaces[i].knet_link_priority = value;
  566. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet_link_priority on link %d now %d", i, value);
  567. }
  568. sprintf(path, "totem.interface.%d.knet_ping_interval", i);
  569. if (icmap_get_uint32(path, &value) == CS_OK) {
  570. instance->totem_config->interfaces[i].knet_ping_interval = value;
  571. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet_ping_interval on link %d now %d", i, value);
  572. }
  573. sprintf(path, "totem.interface.%d.knet_ping_timeout", i);
  574. if (icmap_get_uint32(path, &value) == CS_OK) {
  575. instance->totem_config->interfaces[i].knet_ping_timeout = value;
  576. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet_ping_timeout on link %d now %d", i, value);
  577. }
  578. sprintf(path, "totem.interface.%d.knet_ping_precision", i);
  579. if (icmap_get_uint32(path, &value) == CS_OK) {
  580. instance->totem_config->interfaces[i].knet_ping_precision = value;
  581. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet_ping_precision on link %d now %d", i, value);
  582. }
  583. sprintf(path, "totem.interface.%d.knet_pong_count", i);
  584. if (icmap_get_uint32(path, &value) == CS_OK) {
  585. instance->totem_config->interfaces[i].knet_pong_count = value;
  586. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet_pong_count on link %d now %d", i, value);
  587. }
  588. }
  589. /* Configure link parameters for each node */
  590. err = knet_host_get_host_list(instance->knet_handle, host_ids, &num_nodes);
  591. if (err != 0) {
  592. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_host_get_host_list failed");
  593. }
  594. for (i=0; i<num_nodes; i++) {
  595. for (link_no = 0; link_no < INTERFACE_MAX; link_no++) {
  596. if (host_ids[i] == instance->our_nodeid || !instance->totem_config->interfaces[link_no].configured) {
  597. continue;
  598. }
  599. err = knet_link_set_ping_timers(instance->knet_handle, host_ids[i], link_no,
  600. instance->totem_config->interfaces[link_no].knet_ping_interval,
  601. instance->totem_config->interfaces[link_no].knet_ping_timeout,
  602. instance->totem_config->interfaces[link_no].knet_ping_precision);
  603. if (err) {
  604. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_ping_timers for node %d link %d failed", host_ids[i], link_no);
  605. }
  606. err = knet_link_set_pong_count(instance->knet_handle, host_ids[i], link_no,
  607. instance->totem_config->interfaces[link_no].knet_pong_count);
  608. if (err) {
  609. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_pong_count for node %d link %d failed",host_ids[i], link_no);
  610. }
  611. err = knet_link_set_priority(instance->knet_handle, host_ids[i], link_no,
  612. instance->totem_config->interfaces[link_no].knet_link_priority);
  613. if (err) {
  614. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_priority for node %d link %d failed", host_ids[i], link_no);
  615. }
  616. }
  617. }
  618. LEAVE();
  619. }
  620. static void totemknet_add_config_notifications(struct totemknet_instance *instance)
  621. {
  622. icmap_track_t icmap_track_totem = NULL;
  623. icmap_track_t icmap_track_reload = NULL;
  624. ENTER();
  625. icmap_track_add("totem.",
  626. ICMAP_TRACK_ADD | ICMAP_TRACK_DELETE | ICMAP_TRACK_MODIFY | ICMAP_TRACK_PREFIX,
  627. totemknet_refresh_config,
  628. instance,
  629. &icmap_track_totem);
  630. icmap_track_add("config.totemconfig_reload_in_progress",
  631. ICMAP_TRACK_ADD | ICMAP_TRACK_MODIFY,
  632. totemknet_refresh_config,
  633. instance,
  634. &icmap_track_reload);
  635. LEAVE();
  636. }
  637. /*
  638. * Create an instance
  639. */
  640. int totemknet_initialize (
  641. qb_loop_t *poll_handle,
  642. void **knet_context,
  643. struct totem_config *totem_config,
  644. totemsrp_stats_t *stats,
  645. void *context,
  646. void (*deliver_fn) (
  647. void *context,
  648. const void *msg,
  649. unsigned int msg_len),
  650. void (*iface_change_fn) (
  651. void *context,
  652. const struct totem_ip_address *iface_address,
  653. unsigned int link_no),
  654. void (*mtu_changed) (
  655. void *context,
  656. int net_mtu),
  657. void (*target_set_completed) (
  658. void *context))
  659. {
  660. struct totemknet_instance *instance;
  661. int8_t channel=0;
  662. int res;
  663. int i;
  664. instance = malloc (sizeof (struct totemknet_instance));
  665. if (instance == NULL) {
  666. return (-1);
  667. }
  668. totemknet_instance_initialize (instance);
  669. instance->totem_config = totem_config;
  670. /*
  671. * Configure logging
  672. */
  673. instance->totemknet_log_level_security = 1; //totem_config->totem_logging_configuration.log_level_security;
  674. instance->totemknet_log_level_error = totem_config->totem_logging_configuration.log_level_error;
  675. instance->totemknet_log_level_warning = totem_config->totem_logging_configuration.log_level_warning;
  676. instance->totemknet_log_level_notice = totem_config->totem_logging_configuration.log_level_notice;
  677. instance->totemknet_log_level_debug = totem_config->totem_logging_configuration.log_level_debug;
  678. instance->totemknet_subsys_id = totem_config->totem_logging_configuration.log_subsys_id;
  679. instance->totemknet_log_printf = totem_config->totem_logging_configuration.log_printf;
  680. instance->knet_subsys_id = _logsys_subsys_create("KNET", "libknet.h");
  681. /*
  682. * Initialize local variables for totemknet
  683. */
  684. instance->our_nodeid = instance->totem_config->node_id;
  685. for (i=0; i< INTERFACE_MAX; i++) {
  686. totemip_copy(&instance->my_ids[i], &totem_config->interfaces[i].bindnet);
  687. instance->my_ids[i].nodeid = instance->our_nodeid;
  688. instance->ip_port[i] = totem_config->interfaces[i].ip_port;
  689. /* Needed for totemsrp */
  690. totem_config->interfaces[i].boundto.nodeid = instance->our_nodeid;
  691. }
  692. instance->poll_handle = poll_handle;
  693. instance->context = context;
  694. instance->totemknet_deliver_fn = deliver_fn;
  695. instance->totemknet_iface_change_fn = iface_change_fn;
  696. instance->totemknet_mtu_changed = mtu_changed;
  697. instance->totemknet_target_set_completed = target_set_completed;
  698. res = pipe(instance->logpipes);
  699. if (res == -1) {
  700. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_CRIT, "failed to create pipe for instance->logpipes");
  701. goto exit_error;
  702. }
  703. fcntl(instance->logpipes[0], F_SETFL, O_NONBLOCK);
  704. fcntl(instance->logpipes[1], F_SETFL, O_NONBLOCK);
  705. instance->knet_handle = knet_handle_new(instance->totem_config->node_id, instance->logpipes[1], KNET_LOG_DEBUG);
  706. if (!instance->knet_handle) {
  707. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_CRIT, "knet_handle_new failed");
  708. goto exit_error;
  709. }
  710. res = knet_handle_pmtud_setfreq(instance->knet_handle, instance->totem_config->knet_pmtud_interval);
  711. if (res) {
  712. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_WARNING, "knet_handle_pmtud_setfreq failed");
  713. }
  714. res = knet_handle_enable_filter(instance->knet_handle, instance, dst_host_filter_callback_fn);
  715. if (res) {
  716. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_WARNING, "knet_handle_enable_filter failed");
  717. }
  718. res = knet_handle_enable_sock_notify(instance->knet_handle, instance, socket_error_callback_fn);
  719. if (res) {
  720. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_WARNING, "knet_handle_enable_sock_notify failed");
  721. }
  722. res = knet_host_enable_status_change_notify(instance->knet_handle, instance, host_change_callback_fn);
  723. if (res) {
  724. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_WARNING, "knet_host_enable_status_change_notify failed");
  725. }
  726. res = knet_handle_enable_pmtud_notify(instance->knet_handle, instance, pmtu_change_callback_fn);
  727. if (res) {
  728. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_WARNING, "knet_handle_enable_pmtud_notify failed");
  729. }
  730. global_instance = instance;
  731. /* Get an fd into knet */
  732. instance->knet_fd = 0;
  733. res = knet_handle_add_datafd(instance->knet_handle, &instance->knet_fd, &channel);
  734. if (res) {
  735. knet_log_printf(LOG_DEBUG, "knet_handle_add_datafd failed: %s", strerror(errno));
  736. goto exit_error;
  737. }
  738. /* Enable crypto if requested */
  739. if (strcmp(instance->totem_config->crypto_cipher_type, "none") != 0) {
  740. struct knet_handle_crypto_cfg crypto_cfg;
  741. strcpy(crypto_cfg.crypto_model, "nss");
  742. strcpy(crypto_cfg.crypto_cipher_type, instance->totem_config->crypto_cipher_type);
  743. strcpy(crypto_cfg.crypto_hash_type, instance->totem_config->crypto_hash_type);
  744. memcpy(crypto_cfg.private_key, instance->totem_config->private_key, instance->totem_config->private_key_len);
  745. crypto_cfg.private_key_len = instance->totem_config->private_key_len;
  746. res = knet_handle_crypto(instance->knet_handle, &crypto_cfg);
  747. if (res == -1) {
  748. knet_log_printf(LOGSYS_LEVEL_ERROR, "knet_handle_crypto failed: %s", strerror(errno));
  749. goto exit_error;
  750. }
  751. if (res == -2) {
  752. knet_log_printf(LOGSYS_LEVEL_ERROR, "knet_handle_crypto failed: -2");
  753. goto exit_error;
  754. }
  755. knet_log_printf(LOG_INFO, "kronosnet crypto initialized: %s/%s", crypto_cfg.crypto_cipher_type, crypto_cfg.crypto_hash_type);
  756. }
  757. /* Set up compression */
  758. totemknet_reconfigure(instance, instance->totem_config);
  759. knet_handle_setfwd(instance->knet_handle, 1);
  760. instance->link_mode = KNET_LINK_POLICY_PASSIVE;
  761. if (strcmp(instance->totem_config->link_mode, "active")==0) {
  762. instance->link_mode = KNET_LINK_POLICY_ACTIVE;
  763. }
  764. if (strcmp(instance->totem_config->link_mode, "rr")==0) {
  765. instance->link_mode = KNET_LINK_POLICY_RR;
  766. }
  767. for (i=0; i<INTERFACE_MAX; i++) {
  768. instance->link_status[i] = malloc(CFG_INTERFACE_STATUS_MAX_LEN);
  769. if (!instance->link_status[i]) {
  770. goto exit_error;
  771. }
  772. }
  773. qb_loop_poll_add (instance->poll_handle,
  774. QB_LOOP_MED,
  775. instance->logpipes[0],
  776. POLLIN, instance, log_deliver_fn);
  777. qb_loop_poll_add (instance->poll_handle,
  778. QB_LOOP_HIGH,
  779. instance->knet_fd,
  780. POLLIN, instance, data_deliver_fn);
  781. /*
  782. * Upper layer isn't ready to receive message because it hasn't
  783. * initialized yet. Add short timer to check the interfaces.
  784. */
  785. qb_loop_timer_add (instance->poll_handle,
  786. QB_LOOP_MED,
  787. 100*QB_TIME_NS_IN_MSEC,
  788. (void *)instance,
  789. timer_function_netif_check_timeout,
  790. &instance->timer_netif_check_timeout);
  791. totemknet_start_merge_detect_timeout(instance);
  792. /* Start listening for config changes */
  793. totemknet_add_config_notifications(instance);
  794. knet_log_printf (LOGSYS_LEVEL_INFO, "totemknet initialized");
  795. *knet_context = instance;
  796. return (0);
  797. exit_error:
  798. log_flush_messages(instance);
  799. return (-1);
  800. }
  801. void *totemknet_buffer_alloc (void)
  802. {
  803. /* Need to have space for a message AND a struct mcast in case of encapsulated messages */
  804. return malloc(KNET_MAX_PACKET_SIZE + 512);
  805. }
  806. void totemknet_buffer_release (void *ptr)
  807. {
  808. return free (ptr);
  809. }
  810. int totemknet_processor_count_set (
  811. void *knet_context,
  812. int processor_count)
  813. {
  814. return (0);
  815. }
  816. int totemknet_recv_flush (void *knet_context)
  817. {
  818. return (0);
  819. }
  820. int totemknet_send_flush (void *knet_context)
  821. {
  822. return (0);
  823. }
  824. int totemknet_token_send (
  825. void *knet_context,
  826. const void *msg,
  827. unsigned int msg_len)
  828. {
  829. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  830. int res = 0;
  831. ucast_sendmsg (instance, &instance->token_target, msg, msg_len);
  832. return (res);
  833. }
  834. int totemknet_mcast_flush_send (
  835. void *knet_context,
  836. const void *msg,
  837. unsigned int msg_len)
  838. {
  839. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  840. int res = 0;
  841. mcast_sendmsg (instance, msg, msg_len, 0);
  842. return (res);
  843. }
  844. int totemknet_mcast_noflush_send (
  845. void *knet_context,
  846. const void *msg,
  847. unsigned int msg_len)
  848. {
  849. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  850. int res = 0;
  851. mcast_sendmsg (instance, msg, msg_len, 1);
  852. return (res);
  853. }
  854. extern int totemknet_iface_check (void *knet_context)
  855. {
  856. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  857. int res = 0;
  858. knet_log_printf(LOG_DEBUG, "totemknet: iface_check");
  859. return (res);
  860. }
  861. extern void totemknet_net_mtu_adjust (void *knet_context, struct totem_config *totem_config)
  862. {
  863. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  864. knet_log_printf(LOG_DEBUG, "totemknet: Returning MTU of %d", totem_config->net_mtu);
  865. }
  866. int totemknet_token_target_set (
  867. void *knet_context,
  868. const struct totem_ip_address *token_target)
  869. {
  870. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  871. int res = 0;
  872. memcpy (&instance->token_target, token_target,
  873. sizeof (struct totem_ip_address));
  874. instance->totemknet_target_set_completed (instance->context);
  875. return (res);
  876. }
  877. extern int totemknet_recv_mcast_empty (
  878. void *knet_context)
  879. {
  880. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  881. unsigned int res;
  882. struct sockaddr_storage system_from;
  883. struct msghdr msg_hdr;
  884. struct iovec iov_recv;
  885. struct pollfd ufd;
  886. int nfds;
  887. int msg_processed = 0;
  888. iov_recv.iov_base = instance->iov_buffer;
  889. iov_recv.iov_len = KNET_MAX_PACKET_SIZE;
  890. msg_hdr.msg_name = &system_from;
  891. msg_hdr.msg_namelen = sizeof (struct sockaddr_storage);
  892. msg_hdr.msg_iov = &iov_recv;
  893. msg_hdr.msg_iovlen = 1;
  894. #ifdef HAVE_MSGHDR_CONTROL
  895. msg_hdr.msg_control = 0;
  896. #endif
  897. #ifdef HAVE_MSGHDR_CONTROLLEN
  898. msg_hdr.msg_controllen = 0;
  899. #endif
  900. #ifdef HAVE_MSGHDR_FLAGS
  901. msg_hdr.msg_flags = 0;
  902. #endif
  903. #ifdef HAVE_MSGHDR_ACCRIGHTS
  904. msg_msg_hdr.msg_accrights = NULL;
  905. #endif
  906. #ifdef HAVE_MSGHDR_ACCRIGHTSLEN
  907. msg_msg_hdr.msg_accrightslen = 0;
  908. #endif
  909. do {
  910. ufd.fd = instance->knet_fd;
  911. ufd.events = POLLIN;
  912. nfds = poll (&ufd, 1, 0);
  913. if (nfds == 1 && ufd.revents & POLLIN) {
  914. res = recvmsg (instance->knet_fd, &msg_hdr, MSG_NOSIGNAL | MSG_DONTWAIT);
  915. if (res != -1) {
  916. msg_processed = 1;
  917. } else {
  918. msg_processed = -1;
  919. }
  920. }
  921. } while (nfds == 1);
  922. return (msg_processed);
  923. }
  924. int totemknet_iface_set (void *knet_context,
  925. const struct totem_ip_address *local_addr,
  926. unsigned short ip_port,
  927. unsigned int iface_no)
  928. {
  929. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  930. totemip_copy(&instance->my_ids[iface_no], local_addr);
  931. knet_log_printf(LOG_INFO, "Configured link number %d: local addr: %s, port=%d", iface_no, totemip_print(local_addr), ip_port);
  932. instance->ip_port[iface_no] = ip_port;
  933. return 0;
  934. }
  935. int totemknet_member_add (
  936. void *knet_context,
  937. const struct totem_ip_address *local,
  938. const struct totem_ip_address *member,
  939. int link_no)
  940. {
  941. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  942. int err;
  943. int port = instance->ip_port[link_no];
  944. struct sockaddr_storage remote_ss;
  945. struct sockaddr_storage local_ss;
  946. int addrlen;
  947. /* Only create 1 loopback link */
  948. if (member->nodeid == instance->our_nodeid && link_no > 0) {
  949. return 0;
  950. }
  951. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet: member_add: %d (%s), link=%d", member->nodeid, totemip_print(member), link_no);
  952. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet: local: %d (%s)", local->nodeid, totemip_print(local));
  953. if (link_no == 0) {
  954. if (knet_host_add(instance->knet_handle, member->nodeid)) {
  955. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_host_add");
  956. return -1;
  957. }
  958. if (knet_host_set_policy(instance->knet_handle, member->nodeid, instance->link_mode)) {
  959. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_set_policy failed");
  960. return -1;
  961. }
  962. }
  963. memset(&local_ss, 0, sizeof(local_ss));
  964. /* Casts to remove const */
  965. totemip_totemip_to_sockaddr_convert((struct totem_ip_address *)member, port+link_no, &remote_ss, &addrlen);
  966. totemip_totemip_to_sockaddr_convert((struct totem_ip_address *)local, port+link_no, &local_ss, &addrlen);
  967. if (member->nodeid == instance->our_nodeid) {
  968. err = knet_link_set_config(instance->knet_handle, member->nodeid, link_no,
  969. KNET_TRANSPORT_LOOPBACK,
  970. &local_ss, &remote_ss, KNET_LINK_FLAG_TRAFFICHIPRIO);
  971. }
  972. else {
  973. err = knet_link_set_config(instance->knet_handle, member->nodeid, link_no,
  974. instance->totem_config->interfaces[link_no].knet_transport,
  975. &local_ss, &remote_ss, KNET_LINK_FLAG_TRAFFICHIPRIO);
  976. }
  977. if (err) {
  978. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_config failed");
  979. return -1;
  980. }
  981. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet: member_add: Setting link prio to %d",
  982. instance->totem_config->interfaces[link_no].knet_link_priority);
  983. err = knet_link_set_priority(instance->knet_handle, member->nodeid, link_no,
  984. instance->totem_config->interfaces[link_no].knet_link_priority);
  985. if (err) {
  986. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_priority for nodeid %d, link %d failed", member->nodeid, link_no);
  987. }
  988. err = knet_link_set_ping_timers(instance->knet_handle, member->nodeid, link_no,
  989. instance->totem_config->interfaces[link_no].knet_ping_interval,
  990. instance->totem_config->interfaces[link_no].knet_ping_timeout,
  991. instance->totem_config->interfaces[link_no].knet_ping_precision);
  992. if (err) {
  993. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_ping_timers for nodeid %d, link %d failed", member->nodeid, link_no);
  994. }
  995. err = knet_link_set_pong_count(instance->knet_handle, member->nodeid, link_no,
  996. instance->totem_config->interfaces[link_no].knet_pong_count);
  997. if (err) {
  998. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_pong_count for nodeid %d, link %d failed", member->nodeid, link_no);
  999. }
  1000. err = knet_link_set_enable(instance->knet_handle, member->nodeid, link_no, 1);
  1001. if (err) {
  1002. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set_enable for nodeid %d, link %d failed", member->nodeid, link_no);
  1003. return -1;
  1004. }
  1005. /* register stats */
  1006. stats_knet_add_member(member->nodeid, link_no);
  1007. return (0);
  1008. }
  1009. int totemknet_member_remove (
  1010. void *knet_context,
  1011. const struct totem_ip_address *token_target,
  1012. int link_no)
  1013. {
  1014. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  1015. int res;
  1016. uint8_t link_list[KNET_MAX_LINK];
  1017. size_t num_links;
  1018. knet_log_printf (LOGSYS_LEVEL_DEBUG, "knet: member_remove: %d, link=%d", token_target->nodeid, link_no);
  1019. /* Only link 0 is valid for localhost */
  1020. if (token_target->nodeid == instance->our_nodeid && link_no > 0) {
  1021. return 0;
  1022. }
  1023. /* Tidy stats */
  1024. stats_knet_del_member(token_target->nodeid, link_no);
  1025. /* Remove the link first */
  1026. res = knet_link_set_enable(instance->knet_handle, token_target->nodeid, link_no, 0);
  1027. if (res != 0) {
  1028. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_set enable(off) for nodeid %d, link %d failed", token_target->nodeid, link_no);
  1029. return res;
  1030. }
  1031. res = knet_link_clear_config(instance->knet_handle, token_target->nodeid, link_no);
  1032. if (res != 0) {
  1033. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_link_clear_config for nodeid %d, link %d failed", token_target->nodeid, link_no);
  1034. return res;
  1035. }
  1036. /* If this is the last link, then remove the node */
  1037. res = knet_link_get_link_list(instance->knet_handle,
  1038. token_target->nodeid, link_list, &num_links);
  1039. if (res) {
  1040. return (0); /* not really failure */
  1041. }
  1042. if (num_links == 0) {
  1043. res = knet_host_remove(instance->knet_handle, token_target->nodeid);
  1044. }
  1045. return res;
  1046. }
  1047. int totemknet_member_list_rebind_ip (
  1048. void *knet_context)
  1049. {
  1050. return (0);
  1051. }
  1052. int totemknet_reconfigure (
  1053. void *knet_context,
  1054. struct totem_config *totem_config)
  1055. {
  1056. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  1057. struct knet_handle_compress_cfg compress_cfg;
  1058. int res = 0;
  1059. if (totem_config->knet_compression_model) {
  1060. strcpy(compress_cfg.compress_model, totem_config->knet_compression_model);
  1061. compress_cfg.compress_threshold = totem_config->knet_compression_threshold;
  1062. compress_cfg.compress_level = totem_config->knet_compression_level;
  1063. res = knet_handle_compress(instance->knet_handle, &compress_cfg);
  1064. if (res) {
  1065. KNET_LOGSYS_PERROR(errno, LOGSYS_LEVEL_ERROR, "knet_handle_compress failed");
  1066. }
  1067. }
  1068. return (res);
  1069. }
  1070. void totemknet_stats_clear (
  1071. void *knet_context)
  1072. {
  1073. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  1074. (void) knet_handle_clear_stats(instance->knet_handle, KNET_CLEARSTATS_HANDLE_AND_LINK);
  1075. }
  1076. /* For the stats module */
  1077. int totemknet_link_get_status (
  1078. knet_node_id_t node, uint8_t link_no,
  1079. struct knet_link_status *status)
  1080. {
  1081. int res;
  1082. int ret = CS_OK;
  1083. /* We are probably not using knet */
  1084. if (!global_instance) {
  1085. return CS_ERR_NOT_EXIST;
  1086. }
  1087. if (link_no >= INTERFACE_MAX) {
  1088. return CS_ERR_NOT_EXIST; /* Invalid link number */
  1089. }
  1090. res = knet_link_get_status(global_instance->knet_handle, node, link_no, status, sizeof(struct knet_link_status));
  1091. if (res) {
  1092. switch (errno) {
  1093. case EINVAL:
  1094. ret = CS_ERR_INVALID_PARAM;
  1095. break;
  1096. case EBUSY:
  1097. ret = CS_ERR_BUSY;
  1098. break;
  1099. case EDEADLK:
  1100. ret = CS_ERR_TRY_AGAIN;
  1101. break;
  1102. default:
  1103. ret = CS_ERR_LIBRARY;
  1104. break;
  1105. }
  1106. }
  1107. return (ret);
  1108. }
  1109. int totemknet_handle_get_stats (
  1110. struct knet_handle_stats *stats)
  1111. {
  1112. /* We are probably not using knet */
  1113. if (!global_instance) {
  1114. return CS_ERR_NOT_EXIST;
  1115. }
  1116. return knet_handle_get_stats(global_instance->knet_handle, stats, sizeof(struct knet_handle_stats));
  1117. }
  1118. static void timer_function_merge_detect_timeout (
  1119. void *data)
  1120. {
  1121. struct totemknet_instance *instance = (struct totemknet_instance *)data;
  1122. if (instance->merge_detect_messages_sent_before_timeout == 0) {
  1123. instance->send_merge_detect_message = 1;
  1124. }
  1125. instance->merge_detect_messages_sent_before_timeout = 0;
  1126. totemknet_start_merge_detect_timeout(instance);
  1127. }
  1128. static void totemknet_start_merge_detect_timeout(
  1129. void *knet_context)
  1130. {
  1131. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  1132. qb_loop_timer_add(instance->poll_handle,
  1133. QB_LOOP_MED,
  1134. instance->totem_config->merge_timeout * 2 * QB_TIME_NS_IN_MSEC,
  1135. (void *)instance,
  1136. timer_function_merge_detect_timeout,
  1137. &instance->timer_merge_detect_timeout);
  1138. }
  1139. static void totemknet_stop_merge_detect_timeout(
  1140. void *knet_context)
  1141. {
  1142. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  1143. qb_loop_timer_del(instance->poll_handle,
  1144. instance->timer_merge_detect_timeout);
  1145. }
  1146. static void log_flush_messages (void *knet_context)
  1147. {
  1148. struct pollfd pfd;
  1149. struct totemknet_instance *instance = (struct totemknet_instance *)knet_context;
  1150. int cont;
  1151. cont = 1;
  1152. while (cont) {
  1153. pfd.fd = instance->logpipes[0];
  1154. pfd.events = POLLIN;
  1155. pfd.revents = 0;
  1156. if ((poll(&pfd, 1, 0) > 0) &&
  1157. (pfd.revents & POLLIN) &&
  1158. (log_deliver_fn(instance->logpipes[0], POLLIN, instance) == 0)) {
  1159. cont = 1;
  1160. } else {
  1161. cont = 0;
  1162. }
  1163. }
  1164. }