msg.c 23 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018
  1. /*
  2. * Copyright (c) 2015-2016 Red Hat, Inc.
  3. *
  4. * All rights reserved.
  5. *
  6. * Author: Jan Friesse (jfriesse@redhat.com)
  7. *
  8. * This software licensed under BSD license, the text of which follows:
  9. *
  10. * Redistribution and use in source and binary forms, with or without
  11. * modification, are permitted provided that the following conditions are met:
  12. *
  13. * - Redistributions of source code must retain the above copyright notice,
  14. * this list of conditions and the following disclaimer.
  15. * - Redistributions in binary form must reproduce the above copyright notice,
  16. * this list of conditions and the following disclaimer in the documentation
  17. * and/or other materials provided with the distribution.
  18. * - Neither the name of the Red Hat, Inc. nor the names of its
  19. * contributors may be used to endorse or promote products derived from this
  20. * software without specific prior written permission.
  21. *
  22. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  23. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  26. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  27. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  28. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  29. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  30. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  31. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  32. * THE POSSIBILITY OF SUCH DAMAGE.
  33. */
  34. #include <sys/types.h>
  35. #include <arpa/inet.h>
  36. #include <inttypes.h>
  37. #include <stdlib.h>
  38. #include <string.h>
  39. #include "msg.h"
  40. #define MSG_TYPE_LENGTH 2
  41. #define MSG_LENGTH_LENGTH 4
  42. #define MSG_STATIC_SUPPORTED_MESSAGES_SIZE 16
  43. enum msg_type msg_static_supported_messages[MSG_STATIC_SUPPORTED_MESSAGES_SIZE] = {
  44. MSG_TYPE_PREINIT,
  45. MSG_TYPE_PREINIT_REPLY,
  46. MSG_TYPE_STARTTLS,
  47. MSG_TYPE_INIT,
  48. MSG_TYPE_INIT_REPLY,
  49. MSG_TYPE_SERVER_ERROR,
  50. MSG_TYPE_SET_OPTION,
  51. MSG_TYPE_SET_OPTION_REPLY,
  52. MSG_TYPE_ECHO_REQUEST,
  53. MSG_TYPE_ECHO_REPLY,
  54. MSG_TYPE_NODE_LIST,
  55. MSG_TYPE_NODE_LIST_REPLY,
  56. MSG_TYPE_ASK_FOR_VOTE,
  57. MSG_TYPE_ASK_FOR_VOTE_REPLY,
  58. MSG_TYPE_VOTE_INFO,
  59. MSG_TYPE_VOTE_INFO_REPLY,
  60. };
  61. size_t
  62. msg_get_header_length(void)
  63. {
  64. return (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH);
  65. }
  66. static void
  67. msg_add_type(struct dynar *msg, enum msg_type type)
  68. {
  69. uint16_t ntype;
  70. ntype = htons((uint16_t)type);
  71. dynar_cat(msg, &ntype, sizeof(ntype));
  72. }
  73. enum msg_type
  74. msg_get_type(const struct dynar *msg)
  75. {
  76. uint16_t ntype;
  77. uint16_t type;
  78. memcpy(&ntype, dynar_data(msg), sizeof(ntype));
  79. type = ntohs(ntype);
  80. return (type);
  81. }
  82. /*
  83. * We don't know size of message before call of this function, so zero is
  84. * added. Real value is set afterwards by msg_set_len.
  85. */
  86. static void
  87. msg_add_len(struct dynar *msg)
  88. {
  89. uint32_t len;
  90. len = 0;
  91. dynar_cat(msg, &len, sizeof(len));
  92. }
  93. static void
  94. msg_set_len(struct dynar *msg, uint32_t len)
  95. {
  96. uint32_t nlen;
  97. nlen = htonl(len);
  98. memcpy(dynar_data(msg) + MSG_TYPE_LENGTH, &nlen, sizeof(nlen));
  99. }
  100. /*
  101. * Used only for echo reply msg. All other messages should use msg_add_type.
  102. */
  103. static void
  104. msg_set_type(struct dynar *msg, enum msg_type type)
  105. {
  106. uint16_t ntype;
  107. ntype = htons((uint16_t)type);
  108. memcpy(dynar_data(msg), &ntype, sizeof(ntype));
  109. }
  110. uint32_t
  111. msg_get_len(const struct dynar *msg)
  112. {
  113. uint32_t nlen;
  114. uint32_t len;
  115. memcpy(&nlen, dynar_data(msg) + MSG_TYPE_LENGTH, sizeof(nlen));
  116. len = ntohl(nlen);
  117. return (len);
  118. }
  119. size_t
  120. msg_create_preinit(struct dynar *msg, const char *cluster_name, int add_msg_seq_number,
  121. uint32_t msg_seq_number)
  122. {
  123. dynar_clean(msg);
  124. msg_add_type(msg, MSG_TYPE_PREINIT);
  125. msg_add_len(msg);
  126. if (add_msg_seq_number) {
  127. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  128. goto small_buf_err;
  129. }
  130. }
  131. if (tlv_add_cluster_name(msg, cluster_name) == -1) {
  132. goto small_buf_err;
  133. }
  134. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  135. return (dynar_size(msg));
  136. small_buf_err:
  137. return (0);
  138. }
  139. size_t
  140. msg_create_preinit_reply(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number,
  141. enum tlv_tls_supported tls_supported, int tls_client_cert_required)
  142. {
  143. dynar_clean(msg);
  144. msg_add_type(msg, MSG_TYPE_PREINIT_REPLY);
  145. msg_add_len(msg);
  146. if (add_msg_seq_number) {
  147. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  148. goto small_buf_err;
  149. }
  150. }
  151. if (tlv_add_tls_supported(msg, tls_supported) == -1) {
  152. goto small_buf_err;
  153. }
  154. if (tlv_add_tls_client_cert_required(msg, tls_client_cert_required) == -1) {
  155. goto small_buf_err;
  156. }
  157. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  158. return (dynar_size(msg));
  159. small_buf_err:
  160. return (0);
  161. }
  162. size_t
  163. msg_create_starttls(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number)
  164. {
  165. dynar_clean(msg);
  166. msg_add_type(msg, MSG_TYPE_STARTTLS);
  167. msg_add_len(msg);
  168. if (add_msg_seq_number) {
  169. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  170. goto small_buf_err;
  171. }
  172. }
  173. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  174. return (dynar_size(msg));
  175. small_buf_err:
  176. return (0);
  177. }
  178. size_t
  179. msg_create_server_error(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number,
  180. enum tlv_reply_error_code reply_error_code)
  181. {
  182. dynar_clean(msg);
  183. msg_add_type(msg, MSG_TYPE_SERVER_ERROR);
  184. msg_add_len(msg);
  185. if (add_msg_seq_number) {
  186. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  187. goto small_buf_err;
  188. }
  189. }
  190. if (tlv_add_reply_error_code(msg, reply_error_code) == -1) {
  191. goto small_buf_err;
  192. }
  193. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  194. return (dynar_size(msg));
  195. small_buf_err:
  196. return (0);
  197. }
  198. static uint16_t *
  199. msg_convert_msg_type_array_to_u16_array(const enum msg_type *msg_type_array, size_t array_size)
  200. {
  201. uint16_t *u16a;
  202. size_t i;
  203. u16a = malloc(sizeof(*u16a) * array_size);
  204. if (u16a == NULL) {
  205. return (NULL);
  206. }
  207. for (i = 0; i < array_size; i++) {
  208. u16a[i] = (uint16_t)msg_type_array[i];
  209. }
  210. return (u16a);
  211. }
  212. size_t
  213. msg_create_init(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number,
  214. enum tlv_decision_algorithm_type decision_algorithm,
  215. const enum msg_type *supported_msgs, size_t no_supported_msgs,
  216. const enum tlv_opt_type *supported_opts, size_t no_supported_opts, uint32_t node_id,
  217. uint32_t heartbeat_interval, const struct tlv_tie_breaker *tie_breaker,
  218. const struct tlv_ring_id *ring_id)
  219. {
  220. uint16_t *u16a;
  221. int res;
  222. u16a = NULL;
  223. dynar_clean(msg);
  224. msg_add_type(msg, MSG_TYPE_INIT);
  225. msg_add_len(msg);
  226. if (add_msg_seq_number) {
  227. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  228. goto small_buf_err;
  229. }
  230. }
  231. if (supported_msgs != NULL && no_supported_msgs > 0) {
  232. u16a = msg_convert_msg_type_array_to_u16_array(supported_msgs, no_supported_msgs);
  233. if (u16a == NULL) {
  234. goto small_buf_err;
  235. }
  236. res = tlv_add_u16_array(msg, TLV_OPT_SUPPORTED_MESSAGES, u16a, no_supported_msgs);
  237. free(u16a);
  238. if (res == -1) {
  239. goto small_buf_err;
  240. }
  241. }
  242. if (supported_opts != NULL && no_supported_opts > 0) {
  243. if (tlv_add_supported_options(msg, supported_opts, no_supported_opts) == -1) {
  244. goto small_buf_err;
  245. }
  246. }
  247. if (tlv_add_node_id(msg, node_id) == -1) {
  248. goto small_buf_err;
  249. }
  250. if (tlv_add_decision_algorithm(msg, decision_algorithm) == -1) {
  251. goto small_buf_err;
  252. }
  253. if (tlv_add_heartbeat_interval(msg, heartbeat_interval) == -1) {
  254. goto small_buf_err;
  255. }
  256. if (tlv_add_tie_breaker(msg, tie_breaker) == -1) {
  257. goto small_buf_err;
  258. }
  259. if (tlv_add_ring_id(msg, ring_id) == -1) {
  260. goto small_buf_err;
  261. }
  262. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  263. return (dynar_size(msg));
  264. small_buf_err:
  265. return (0);
  266. }
  267. size_t
  268. msg_create_init_reply(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number,
  269. enum tlv_reply_error_code reply_error_code,
  270. const enum msg_type *supported_msgs, size_t no_supported_msgs,
  271. const enum tlv_opt_type *supported_opts, size_t no_supported_opts,
  272. size_t server_maximum_request_size, size_t server_maximum_reply_size,
  273. const enum tlv_decision_algorithm_type *supported_decision_algorithms,
  274. size_t no_supported_decision_algorithms)
  275. {
  276. uint16_t *u16a;
  277. int res;
  278. u16a = NULL;
  279. dynar_clean(msg);
  280. msg_add_type(msg, MSG_TYPE_INIT_REPLY);
  281. msg_add_len(msg);
  282. if (tlv_add_reply_error_code(msg, reply_error_code) == -1) {
  283. goto small_buf_err;
  284. }
  285. if (supported_msgs != NULL && no_supported_msgs > 0) {
  286. u16a = msg_convert_msg_type_array_to_u16_array(supported_msgs, no_supported_msgs);
  287. if (u16a == NULL) {
  288. goto small_buf_err;
  289. }
  290. res = tlv_add_u16_array(msg, TLV_OPT_SUPPORTED_MESSAGES, u16a, no_supported_msgs);
  291. free(u16a);
  292. if (res == -1) {
  293. goto small_buf_err;
  294. }
  295. }
  296. if (supported_opts != NULL && no_supported_opts > 0) {
  297. if (tlv_add_supported_options(msg, supported_opts, no_supported_opts) == -1) {
  298. goto small_buf_err;
  299. }
  300. }
  301. if (add_msg_seq_number) {
  302. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  303. goto small_buf_err;
  304. }
  305. }
  306. if (tlv_add_server_maximum_request_size(msg, server_maximum_request_size) == -1) {
  307. goto small_buf_err;
  308. }
  309. if (tlv_add_server_maximum_reply_size(msg, server_maximum_reply_size) == -1) {
  310. goto small_buf_err;
  311. }
  312. if (supported_decision_algorithms != NULL && no_supported_decision_algorithms > 0) {
  313. if (tlv_add_supported_decision_algorithms(msg, supported_decision_algorithms,
  314. no_supported_decision_algorithms) == -1) {
  315. goto small_buf_err;
  316. }
  317. }
  318. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  319. return (dynar_size(msg));
  320. small_buf_err:
  321. return (0);
  322. }
  323. size_t
  324. msg_create_set_option(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number,
  325. int add_heartbeat_interval, uint32_t heartbeat_interval)
  326. {
  327. dynar_clean(msg);
  328. msg_add_type(msg, MSG_TYPE_SET_OPTION);
  329. msg_add_len(msg);
  330. if (add_msg_seq_number) {
  331. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  332. goto small_buf_err;
  333. }
  334. }
  335. if (add_heartbeat_interval) {
  336. if (tlv_add_heartbeat_interval(msg, heartbeat_interval) == -1) {
  337. goto small_buf_err;
  338. }
  339. }
  340. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  341. return (dynar_size(msg));
  342. small_buf_err:
  343. return (0);
  344. }
  345. size_t
  346. msg_create_set_option_reply(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number,
  347. uint32_t heartbeat_interval)
  348. {
  349. dynar_clean(msg);
  350. msg_add_type(msg, MSG_TYPE_SET_OPTION_REPLY);
  351. msg_add_len(msg);
  352. if (add_msg_seq_number) {
  353. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  354. goto small_buf_err;
  355. }
  356. }
  357. if (tlv_add_heartbeat_interval(msg, heartbeat_interval) == -1) {
  358. goto small_buf_err;
  359. }
  360. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  361. return (dynar_size(msg));
  362. small_buf_err:
  363. return (0);
  364. }
  365. size_t
  366. msg_create_echo_request(struct dynar *msg, int add_msg_seq_number, uint32_t msg_seq_number)
  367. {
  368. dynar_clean(msg);
  369. msg_add_type(msg, MSG_TYPE_ECHO_REQUEST);
  370. msg_add_len(msg);
  371. if (add_msg_seq_number) {
  372. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  373. goto small_buf_err;
  374. }
  375. }
  376. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  377. return (dynar_size(msg));
  378. small_buf_err:
  379. return (0);
  380. }
  381. size_t
  382. msg_create_echo_reply(struct dynar *msg, const struct dynar *echo_request_msg)
  383. {
  384. dynar_clean(msg);
  385. if (dynar_cat(msg, dynar_data(echo_request_msg), dynar_size(echo_request_msg)) == -1) {
  386. goto small_buf_err;
  387. }
  388. msg_set_type(msg, MSG_TYPE_ECHO_REPLY);
  389. return (dynar_size(msg));
  390. small_buf_err:
  391. return (0);
  392. }
  393. size_t
  394. msg_create_node_list(struct dynar *msg,
  395. uint32_t msg_seq_number, enum tlv_node_list_type node_list_type,
  396. int add_ring_id, const struct tlv_ring_id *ring_id,
  397. int add_config_version, uint64_t config_version,
  398. int add_quorate, enum tlv_quorate quorate,
  399. const struct node_list *nodes)
  400. {
  401. struct node_list_entry *node_info;
  402. struct tlv_node_info tlv_ni;
  403. dynar_clean(msg);
  404. msg_add_type(msg, MSG_TYPE_NODE_LIST);
  405. msg_add_len(msg);
  406. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  407. goto small_buf_err;
  408. }
  409. if (tlv_add_node_list_type(msg, node_list_type) == -1) {
  410. goto small_buf_err;
  411. }
  412. if (add_ring_id) {
  413. if (tlv_add_ring_id(msg, ring_id) == -1) {
  414. goto small_buf_err;
  415. }
  416. }
  417. if (add_config_version) {
  418. if (tlv_add_config_version(msg, config_version) == -1) {
  419. goto small_buf_err;
  420. }
  421. }
  422. if (add_quorate) {
  423. if (tlv_add_quorate(msg, quorate) == -1) {
  424. goto small_buf_err;
  425. }
  426. }
  427. TAILQ_FOREACH(node_info, nodes, entries) {
  428. node_list_entry_to_tlv_node_info(node_info, &tlv_ni);
  429. if (tlv_add_node_info(msg, &tlv_ni) == -1) {
  430. goto small_buf_err;
  431. }
  432. }
  433. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  434. return (dynar_size(msg));
  435. small_buf_err:
  436. return (0);
  437. }
  438. size_t
  439. msg_create_node_list_reply(struct dynar *msg, uint32_t msg_seq_number,
  440. enum tlv_node_list_type node_list_type, const struct tlv_ring_id *ring_id,
  441. enum tlv_vote vote)
  442. {
  443. dynar_clean(msg);
  444. msg_add_type(msg, MSG_TYPE_NODE_LIST_REPLY);
  445. msg_add_len(msg);
  446. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  447. goto small_buf_err;
  448. }
  449. if (tlv_add_node_list_type(msg, node_list_type) == -1) {
  450. goto small_buf_err;
  451. }
  452. if (tlv_add_ring_id(msg, ring_id) == -1) {
  453. goto small_buf_err;
  454. }
  455. if (tlv_add_vote(msg, vote) == -1) {
  456. goto small_buf_err;
  457. }
  458. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  459. return (dynar_size(msg));
  460. small_buf_err:
  461. return (0);
  462. }
  463. size_t
  464. msg_create_ask_for_vote(struct dynar *msg, uint32_t msg_seq_number)
  465. {
  466. dynar_clean(msg);
  467. msg_add_type(msg, MSG_TYPE_ASK_FOR_VOTE);
  468. msg_add_len(msg);
  469. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  470. goto small_buf_err;
  471. }
  472. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  473. return (dynar_size(msg));
  474. small_buf_err:
  475. return (0);
  476. }
  477. size_t
  478. msg_create_ask_for_vote_reply(struct dynar *msg, uint32_t msg_seq_number,
  479. const struct tlv_ring_id *ring_id, enum tlv_vote vote)
  480. {
  481. dynar_clean(msg);
  482. msg_add_type(msg, MSG_TYPE_ASK_FOR_VOTE_REPLY);
  483. msg_add_len(msg);
  484. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  485. goto small_buf_err;
  486. }
  487. if (tlv_add_vote(msg, vote) == -1) {
  488. goto small_buf_err;
  489. }
  490. if (tlv_add_ring_id(msg, ring_id) == -1) {
  491. goto small_buf_err;
  492. }
  493. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  494. return (dynar_size(msg));
  495. small_buf_err:
  496. return (0);
  497. }
  498. size_t
  499. msg_create_vote_info(struct dynar *msg, uint32_t msg_seq_number, const struct tlv_ring_id *ring_id,
  500. enum tlv_vote vote)
  501. {
  502. dynar_clean(msg);
  503. msg_add_type(msg, MSG_TYPE_VOTE_INFO);
  504. msg_add_len(msg);
  505. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  506. goto small_buf_err;
  507. }
  508. if (tlv_add_vote(msg, vote) == -1) {
  509. goto small_buf_err;
  510. }
  511. if (tlv_add_ring_id(msg, ring_id) == -1) {
  512. goto small_buf_err;
  513. }
  514. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  515. return (dynar_size(msg));
  516. small_buf_err:
  517. return (0);
  518. }
  519. size_t
  520. msg_create_vote_info_reply(struct dynar *msg, uint32_t msg_seq_number)
  521. {
  522. dynar_clean(msg);
  523. msg_add_type(msg, MSG_TYPE_VOTE_INFO_REPLY);
  524. msg_add_len(msg);
  525. if (tlv_add_msg_seq_number(msg, msg_seq_number) == -1) {
  526. goto small_buf_err;
  527. }
  528. msg_set_len(msg, dynar_size(msg) - (MSG_TYPE_LENGTH + MSG_LENGTH_LENGTH));
  529. return (dynar_size(msg));
  530. small_buf_err:
  531. return (0);
  532. }
  533. int
  534. msg_is_valid_msg_type(const struct dynar *msg)
  535. {
  536. enum msg_type type;
  537. size_t i;
  538. type = msg_get_type(msg);
  539. for (i = 0; i < MSG_STATIC_SUPPORTED_MESSAGES_SIZE; i++) {
  540. if (msg_static_supported_messages[i] == type) {
  541. return (1);
  542. }
  543. }
  544. return (0);
  545. }
  546. void
  547. msg_decoded_init(struct msg_decoded *decoded_msg)
  548. {
  549. memset(decoded_msg, 0, sizeof(*decoded_msg));
  550. node_list_init(&decoded_msg->nodes);
  551. }
  552. void
  553. msg_decoded_destroy(struct msg_decoded *decoded_msg)
  554. {
  555. free(decoded_msg->cluster_name);
  556. free(decoded_msg->supported_messages);
  557. free(decoded_msg->supported_options);
  558. free(decoded_msg->supported_decision_algorithms);
  559. node_list_free(&decoded_msg->nodes);
  560. msg_decoded_init(decoded_msg);
  561. }
  562. /*
  563. * 0 - No error
  564. * -1 - option with invalid length
  565. * -2 - Unable to allocate memory
  566. * -3 - Inconsistent msg (tlv len > msg size)
  567. * -4 - invalid option content
  568. */
  569. int
  570. msg_decode(const struct dynar *msg, struct msg_decoded *decoded_msg)
  571. {
  572. struct tlv_iterator tlv_iter;
  573. uint16_t *u16a;
  574. uint32_t u32;
  575. uint64_t u64;
  576. struct tlv_ring_id ring_id;
  577. struct tlv_node_info node_info;
  578. struct tlv_tie_breaker tie_breaker;
  579. size_t zi;
  580. enum tlv_opt_type opt_type;
  581. int iter_res;
  582. int res;
  583. msg_decoded_destroy(decoded_msg);
  584. decoded_msg->type = msg_get_type(msg);
  585. tlv_iter_init(msg, msg_get_header_length(), &tlv_iter);
  586. while ((iter_res = tlv_iter_next(&tlv_iter)) > 0) {
  587. opt_type = tlv_iter_get_type(&tlv_iter);
  588. switch (opt_type) {
  589. case TLV_OPT_MSG_SEQ_NUMBER:
  590. if ((res = tlv_iter_decode_u32(&tlv_iter, &u32)) != 0) {
  591. return (res);
  592. }
  593. decoded_msg->seq_number_set = 1;
  594. decoded_msg->seq_number = u32;
  595. break;
  596. case TLV_OPT_CLUSTER_NAME:
  597. if ((res = tlv_iter_decode_str(&tlv_iter, &decoded_msg->cluster_name,
  598. &decoded_msg->cluster_name_len)) != 0) {
  599. return (-2);
  600. }
  601. break;
  602. case TLV_OPT_TLS_SUPPORTED:
  603. if ((res = tlv_iter_decode_tls_supported(&tlv_iter,
  604. &decoded_msg->tls_supported)) != 0) {
  605. return (res);
  606. }
  607. decoded_msg->tls_supported_set = 1;
  608. break;
  609. case TLV_OPT_TLS_CLIENT_CERT_REQUIRED:
  610. if ((res = tlv_iter_decode_client_cert_required(&tlv_iter,
  611. &decoded_msg->tls_client_cert_required)) != 0) {
  612. return (res);
  613. }
  614. decoded_msg->tls_client_cert_required_set = 1;
  615. break;
  616. case TLV_OPT_SUPPORTED_MESSAGES:
  617. free(decoded_msg->supported_messages);
  618. if ((res = tlv_iter_decode_u16_array(&tlv_iter, &u16a,
  619. &decoded_msg->no_supported_messages)) != 0) {
  620. return (res);
  621. }
  622. decoded_msg->supported_messages =
  623. malloc(sizeof(enum msg_type) * decoded_msg->no_supported_messages);
  624. if (decoded_msg->supported_messages == NULL) {
  625. free(u16a);
  626. return (-2);
  627. }
  628. for (zi = 0; zi < decoded_msg->no_supported_messages; zi++) {
  629. decoded_msg->supported_messages[zi] = (enum msg_type)u16a[zi];
  630. }
  631. free(u16a);
  632. break;
  633. case TLV_OPT_SUPPORTED_OPTIONS:
  634. free(decoded_msg->supported_options);
  635. if ((res = tlv_iter_decode_supported_options(&tlv_iter,
  636. &decoded_msg->supported_options,
  637. &decoded_msg->no_supported_options)) != 0) {
  638. return (res);
  639. }
  640. break;
  641. case TLV_OPT_REPLY_ERROR_CODE:
  642. if ((res = tlv_iter_decode_reply_error_code(&tlv_iter,
  643. &decoded_msg->reply_error_code)) != 0) {
  644. return (res);
  645. }
  646. decoded_msg->reply_error_code_set = 1;
  647. break;
  648. case TLV_OPT_SERVER_MAXIMUM_REQUEST_SIZE:
  649. if ((res = tlv_iter_decode_u32(&tlv_iter, &u32)) != 0) {
  650. return (res);
  651. }
  652. decoded_msg->server_maximum_request_size_set = 1;
  653. decoded_msg->server_maximum_request_size = u32;
  654. break;
  655. case TLV_OPT_SERVER_MAXIMUM_REPLY_SIZE:
  656. if ((res = tlv_iter_decode_u32(&tlv_iter, &u32)) != 0) {
  657. return (res);
  658. }
  659. decoded_msg->server_maximum_reply_size_set = 1;
  660. decoded_msg->server_maximum_reply_size = u32;
  661. break;
  662. case TLV_OPT_NODE_ID:
  663. if ((res = tlv_iter_decode_u32(&tlv_iter, &u32)) != 0) {
  664. return (res);
  665. }
  666. decoded_msg->node_id_set = 1;
  667. decoded_msg->node_id = u32;
  668. break;
  669. case TLV_OPT_SUPPORTED_DECISION_ALGORITHMS:
  670. free(decoded_msg->supported_decision_algorithms);
  671. if ((res = tlv_iter_decode_supported_decision_algorithms(&tlv_iter,
  672. &decoded_msg->supported_decision_algorithms,
  673. &decoded_msg->no_supported_decision_algorithms)) != 0) {
  674. return (res);
  675. }
  676. break;
  677. case TLV_OPT_DECISION_ALGORITHM:
  678. if ((res = tlv_iter_decode_decision_algorithm(&tlv_iter,
  679. &decoded_msg->decision_algorithm)) != 0) {
  680. return (res);
  681. }
  682. decoded_msg->decision_algorithm_set = 1;
  683. break;
  684. case TLV_OPT_HEARTBEAT_INTERVAL:
  685. if ((res = tlv_iter_decode_u32(&tlv_iter, &u32)) != 0) {
  686. return (res);
  687. }
  688. decoded_msg->heartbeat_interval_set = 1;
  689. decoded_msg->heartbeat_interval = u32;
  690. break;
  691. case TLV_OPT_RING_ID:
  692. if ((res = tlv_iter_decode_ring_id(&tlv_iter, &ring_id)) != 0) {
  693. return (res);
  694. }
  695. decoded_msg->ring_id_set = 1;
  696. memcpy(&decoded_msg->ring_id, &ring_id, sizeof(ring_id));
  697. break;
  698. case TLV_OPT_CONFIG_VERSION:
  699. if ((res = tlv_iter_decode_u64(&tlv_iter, &u64)) != 0) {
  700. return (res);
  701. }
  702. decoded_msg->config_version_set = 1;
  703. decoded_msg->config_version = u64;
  704. break;
  705. case TLV_OPT_DATA_CENTER_ID:
  706. if ((res = tlv_iter_decode_u32(&tlv_iter, &u32)) != 0) {
  707. return (res);
  708. }
  709. decoded_msg->data_center_id = u32;
  710. break;
  711. case TLV_OPT_NODE_STATE:
  712. if ((res = tlv_iter_decode_node_state(&tlv_iter,
  713. &decoded_msg->node_state)) != 0) {
  714. return (res);
  715. }
  716. break;
  717. case TLV_OPT_NODE_INFO:
  718. if ((res = tlv_iter_decode_node_info(&tlv_iter, &node_info)) != 0) {
  719. return (res);
  720. }
  721. if (node_list_add_from_node_info(&decoded_msg->nodes, &node_info) == NULL) {
  722. return (-2);
  723. }
  724. break;
  725. case TLV_OPT_NODE_LIST_TYPE:
  726. if ((res = tlv_iter_decode_node_list_type(&tlv_iter,
  727. &decoded_msg->node_list_type)) != 0) {
  728. return (res);
  729. }
  730. decoded_msg->node_list_type_set = 1;
  731. break;
  732. case TLV_OPT_VOTE:
  733. if ((res = tlv_iter_decode_vote(&tlv_iter, &decoded_msg->vote)) != 0) {
  734. return (res);
  735. }
  736. decoded_msg->vote_set = 1;
  737. break;
  738. case TLV_OPT_QUORATE:
  739. if ((res = tlv_iter_decode_quorate(&tlv_iter,
  740. &decoded_msg->quorate)) != 0) {
  741. return (res);
  742. }
  743. decoded_msg->quorate_set = 1;
  744. break;
  745. case TLV_OPT_TIE_BREAKER:
  746. if ((res = tlv_iter_decode_tie_breaker(&tlv_iter, &tie_breaker)) != 0) {
  747. return (res);
  748. }
  749. decoded_msg->tie_breaker_set = 1;
  750. memcpy(&decoded_msg->tie_breaker, &tie_breaker, sizeof(tie_breaker));
  751. break;
  752. /*
  753. * Default is not defined intentionally. Compiler shows warning when
  754. * new tlv option is added. Also protocol ignores unknown options so
  755. * no extra work is needed.
  756. */
  757. }
  758. }
  759. if (iter_res != 0) {
  760. return (-3);
  761. }
  762. return (0);
  763. }
  764. void
  765. msg_get_supported_messages(enum msg_type **supported_messages, size_t *no_supported_messages)
  766. {
  767. *supported_messages = msg_static_supported_messages;
  768. *no_supported_messages = MSG_STATIC_SUPPORTED_MESSAGES_SIZE;
  769. }
  770. const char *
  771. msg_type_to_str(enum msg_type type)
  772. {
  773. switch (type) {
  774. case MSG_TYPE_PREINIT: return ("Preinit"); break;
  775. case MSG_TYPE_PREINIT_REPLY: return ("Preinit reply"); break;
  776. case MSG_TYPE_STARTTLS: return ("StartTLS"); break;
  777. case MSG_TYPE_INIT: return ("Init"); break;
  778. case MSG_TYPE_INIT_REPLY: return ("Init reply"); break;
  779. case MSG_TYPE_SERVER_ERROR: return ("Server error"); break;
  780. case MSG_TYPE_SET_OPTION: return ("Set option"); break;
  781. case MSG_TYPE_SET_OPTION_REPLY: return ("Set option reply"); break;
  782. case MSG_TYPE_ECHO_REQUEST: return ("Echo request"); break;
  783. case MSG_TYPE_ECHO_REPLY: return ("Echo reply"); break;
  784. case MSG_TYPE_NODE_LIST: return ("Node list"); break;
  785. case MSG_TYPE_NODE_LIST_REPLY: return ("Node list reply"); break;
  786. case MSG_TYPE_ASK_FOR_VOTE: return ("Ask for vote"); break;
  787. case MSG_TYPE_ASK_FOR_VOTE_REPLY: return ("Ask for vote reply"); break;
  788. case MSG_TYPE_VOTE_INFO: return ("Vote info"); break;
  789. case MSG_TYPE_VOTE_INFO_REPLY: return ("Vote info reply"); break;
  790. }
  791. return ("Unknown message type");
  792. }