icmap.c 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344
  1. /*
  2. * Copyright (c) 2011 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 <config.h>
  35. #include <string.h>
  36. #include <stdio.h>
  37. #include <corosync/corotypes.h>
  38. #include <qb/qbdefs.h>
  39. #include <corosync/list.h>
  40. #include <corosync/icmap.h>
  41. #define ICMAP_MAX_VALUE_LEN (16*1024)
  42. struct icmap_item {
  43. char *key_name;
  44. icmap_value_types_t type;
  45. size_t value_len;
  46. char value[];
  47. };
  48. struct icmap_map {
  49. qb_map_t *qb_map;
  50. };
  51. static icmap_map_t icmap_global_map;
  52. struct icmap_track {
  53. char *key_name;
  54. int32_t track_type;
  55. icmap_notify_fn_t notify_fn;
  56. void *user_data;
  57. struct list_head list;
  58. };
  59. struct icmap_ro_access_item {
  60. char *key_name;
  61. int prefix;
  62. struct list_head list;
  63. };
  64. DECLARE_LIST_INIT(icmap_ro_access_item_list_head);
  65. DECLARE_LIST_INIT(icmap_track_list_head);
  66. /*
  67. * Static functions declarations
  68. */
  69. /*
  70. * Check if key_name is valid icmap key name. Returns 0 on success, and -1 on fail
  71. */
  72. static int icmap_check_key_name(const char *key_name);
  73. /*
  74. * Check that value with given type has correct length value_len. Returns 0 on success,
  75. * and -1 on fail
  76. */
  77. static int icmap_check_value_len(const void *value, size_t value_len, icmap_value_types_t type);
  78. /*
  79. * Returns length of value of given type, or 0 for string and binary data type
  80. */
  81. static size_t icmap_get_valuetype_len(icmap_value_types_t type);
  82. /*
  83. * Converts track type of icmap to qb
  84. */
  85. static int32_t icmap_tt_to_qbtt(int32_t track_type);
  86. /*
  87. * Convert track type of qb to icmap
  88. */
  89. static int32_t icmap_qbtt_to_tt(int32_t track_type);
  90. /*
  91. * Checks if item has same value as value with value_len and given type. Returns 0 if not, otherwise !0.
  92. */
  93. static int icmap_item_eq(const struct icmap_item *item, const void *value, size_t value_len, icmap_value_types_t type);
  94. /*
  95. * Checks if given character is valid in key name. Returns 0 if not, otherwise !0.
  96. */
  97. static int icmap_is_valid_name_char(char c);
  98. /*
  99. * Helper for getting integer and float value with given type for key key_name and store it in value.
  100. */
  101. static cs_error_t icmap_get_int_r(
  102. const icmap_map_t map,
  103. const char *key_name,
  104. void *value,
  105. icmap_value_types_t type);
  106. /*
  107. * Return raw item value data. Internal function used by icmap_get_r which does most
  108. * of arguments validity checks but doesn't copy data (it returns raw item data
  109. * pointer). It's not very safe tho it's static.
  110. */
  111. static cs_error_t icmap_get_ref_r(
  112. const icmap_map_t map,
  113. const char *key_name,
  114. void **value,
  115. size_t *value_len,
  116. icmap_value_types_t *type);
  117. /*
  118. * Function implementation
  119. */
  120. static int32_t icmap_tt_to_qbtt(int32_t track_type)
  121. {
  122. int32_t res = 0;
  123. if (track_type & ICMAP_TRACK_DELETE) {
  124. res |= QB_MAP_NOTIFY_DELETED;
  125. }
  126. if (track_type & ICMAP_TRACK_MODIFY) {
  127. res |= QB_MAP_NOTIFY_REPLACED;
  128. }
  129. if (track_type & ICMAP_TRACK_ADD) {
  130. res |= QB_MAP_NOTIFY_INSERTED;
  131. }
  132. if (track_type & ICMAP_TRACK_PREFIX) {
  133. res |= QB_MAP_NOTIFY_RECURSIVE;
  134. }
  135. return (res);
  136. }
  137. static int32_t icmap_qbtt_to_tt(int32_t track_type)
  138. {
  139. int32_t res = 0;
  140. if (track_type & QB_MAP_NOTIFY_DELETED) {
  141. res |= ICMAP_TRACK_DELETE;
  142. }
  143. if (track_type & QB_MAP_NOTIFY_REPLACED) {
  144. res |= ICMAP_TRACK_MODIFY;
  145. }
  146. if (track_type & QB_MAP_NOTIFY_INSERTED) {
  147. res |= ICMAP_TRACK_ADD;
  148. }
  149. if (track_type & QB_MAP_NOTIFY_RECURSIVE) {
  150. res |= ICMAP_TRACK_PREFIX;
  151. }
  152. return (res);
  153. }
  154. static void icmap_map_free_cb(uint32_t event,
  155. char* key, void* old_value,
  156. void* value, void* user_data)
  157. {
  158. struct icmap_item *item = (struct icmap_item *)old_value;
  159. /*
  160. * value == old_value -> fast_adjust_int was used, don't free data
  161. */
  162. if (item != NULL && value != old_value) {
  163. free(item->key_name);
  164. free(item);
  165. }
  166. }
  167. cs_error_t icmap_init_r(icmap_map_t *result)
  168. {
  169. int32_t err;
  170. *result = malloc(sizeof(struct icmap_map));
  171. if (*result == NULL) {
  172. return (CS_ERR_NO_MEMORY);
  173. }
  174. (*result)->qb_map = qb_trie_create();
  175. if ((*result)->qb_map == NULL)
  176. return (CS_ERR_INIT);
  177. err = qb_map_notify_add((*result)->qb_map, NULL, icmap_map_free_cb, QB_MAP_NOTIFY_FREE, NULL);
  178. return (qb_to_cs_error(err));
  179. }
  180. cs_error_t icmap_init(void)
  181. {
  182. return (icmap_init_r(&icmap_global_map));
  183. }
  184. static void icmap_set_ro_access_free(void)
  185. {
  186. struct list_head *iter = icmap_ro_access_item_list_head.next;
  187. struct icmap_ro_access_item *icmap_ro_ai;
  188. while (iter != &icmap_ro_access_item_list_head) {
  189. icmap_ro_ai = list_entry(iter, struct icmap_ro_access_item, list);
  190. list_del(&icmap_ro_ai->list);
  191. free(icmap_ro_ai->key_name);
  192. free(icmap_ro_ai);
  193. iter = icmap_ro_access_item_list_head.next;
  194. }
  195. }
  196. static void icmap_del_all_track(void)
  197. {
  198. struct list_head *iter = icmap_track_list_head.next;
  199. struct icmap_track *icmap_track;
  200. while (iter != &icmap_track_list_head) {
  201. icmap_track = list_entry(iter, struct icmap_track, list);
  202. icmap_track_delete(icmap_track);
  203. iter = icmap_track_list_head.next;
  204. }
  205. }
  206. void icmap_fini_r(const icmap_map_t map)
  207. {
  208. qb_map_destroy(map->qb_map);
  209. free(map);
  210. return;
  211. }
  212. void icmap_fini(void)
  213. {
  214. icmap_del_all_track();
  215. /*
  216. * catch 22 warning:
  217. * We need to drop this notify but we can't because it calls icmap_map_free_cb
  218. * while destroying the tree to free icmap_item(s).
  219. * -> qb_map_notify_del_2(icmap_map, NULL, icmap_map_free_cb, QB_MAP_NOTIFY_FREE, NULL);
  220. * and we cannot call it after map_destroy. joy! :)
  221. */
  222. icmap_fini_r(icmap_global_map);
  223. icmap_set_ro_access_free();
  224. return ;
  225. }
  226. icmap_map_t icmap_get_global_map(void)
  227. {
  228. return (icmap_global_map);
  229. }
  230. static int icmap_is_valid_name_char(char c)
  231. {
  232. return ((c >= 'a' && c <= 'z') ||
  233. (c >= 'A' && c <= 'Z') ||
  234. (c >= '0' && c <= '9') ||
  235. c == '.' || c == '_' || c == '-' || c == '/' || c == ':');
  236. }
  237. void icmap_convert_name_to_valid_name(char *key_name)
  238. {
  239. int i;
  240. for (i = 0; i < strlen(key_name); i++) {
  241. if (!icmap_is_valid_name_char(key_name[i])) {
  242. key_name[i] = '_';
  243. }
  244. }
  245. }
  246. static int icmap_check_key_name(const char *key_name)
  247. {
  248. int i;
  249. if ((strlen(key_name) < ICMAP_KEYNAME_MINLEN) || strlen(key_name) > ICMAP_KEYNAME_MAXLEN) {
  250. return (-1);
  251. }
  252. for (i = 0; i < strlen(key_name); i++) {
  253. if (!icmap_is_valid_name_char(key_name[i])) {
  254. return (-1);
  255. }
  256. }
  257. return (0);
  258. }
  259. static size_t icmap_get_valuetype_len(icmap_value_types_t type)
  260. {
  261. size_t res = 0;
  262. switch (type) {
  263. case ICMAP_VALUETYPE_INT8: res = sizeof(int8_t); break;
  264. case ICMAP_VALUETYPE_UINT8: res = sizeof(uint8_t); break;
  265. case ICMAP_VALUETYPE_INT16: res = sizeof(int16_t); break;
  266. case ICMAP_VALUETYPE_UINT16: res = sizeof(uint16_t); break;
  267. case ICMAP_VALUETYPE_INT32: res = sizeof(int32_t); break;
  268. case ICMAP_VALUETYPE_UINT32: res = sizeof(uint32_t); break;
  269. case ICMAP_VALUETYPE_INT64: res = sizeof(int64_t); break;
  270. case ICMAP_VALUETYPE_UINT64: res = sizeof(uint64_t); break;
  271. case ICMAP_VALUETYPE_FLOAT: res = sizeof(float); break;
  272. case ICMAP_VALUETYPE_DOUBLE: res = sizeof(double); break;
  273. case ICMAP_VALUETYPE_STRING:
  274. case ICMAP_VALUETYPE_BINARY:
  275. res = 0;
  276. break;
  277. }
  278. return (res);
  279. }
  280. static int icmap_check_value_len(const void *value, size_t value_len, icmap_value_types_t type)
  281. {
  282. if (value_len > ICMAP_MAX_VALUE_LEN) {
  283. return (-1);
  284. }
  285. if (type != ICMAP_VALUETYPE_STRING && type != ICMAP_VALUETYPE_BINARY) {
  286. if (icmap_get_valuetype_len(type) == value_len) {
  287. return (0);
  288. } else {
  289. return (-1);
  290. }
  291. }
  292. if (type == ICMAP_VALUETYPE_STRING) {
  293. /*
  294. * value_len can be shorter then real string length, but never
  295. * longer (+ 1 is because of 0 at the end of string)
  296. */
  297. if (value_len > strlen((const char *)value) + 1) {
  298. return (-1);
  299. } else {
  300. return (0);
  301. }
  302. }
  303. return (0);
  304. }
  305. static int icmap_item_eq(const struct icmap_item *item, const void *value, size_t value_len, icmap_value_types_t type)
  306. {
  307. size_t ptr_len;
  308. if (item->type != type) {
  309. return (0);
  310. }
  311. if (item->type == ICMAP_VALUETYPE_STRING) {
  312. ptr_len = strlen((const char *)value);
  313. if (ptr_len > value_len) {
  314. ptr_len = value_len;
  315. }
  316. ptr_len++;
  317. } else {
  318. ptr_len = value_len;
  319. }
  320. if (item->value_len == ptr_len) {
  321. return (memcmp(item->value, value, value_len) == 0);
  322. };
  323. return (0);
  324. }
  325. int icmap_key_value_eq(
  326. const icmap_map_t map1,
  327. const char *key_name1,
  328. const icmap_map_t map2,
  329. const char *key_name2)
  330. {
  331. struct icmap_item *item1, *item2;
  332. if (map1 == NULL || key_name1 == NULL || map2 == NULL || key_name2 == NULL) {
  333. return (0);
  334. }
  335. item1 = qb_map_get(map1->qb_map, key_name1);
  336. item2 = qb_map_get(map2->qb_map, key_name2);
  337. if (item1 == NULL || item2 == NULL) {
  338. return (0);
  339. }
  340. return (icmap_item_eq(item1, item2->value, item2->value_len, item2->type));
  341. }
  342. cs_error_t icmap_set_r(
  343. const icmap_map_t map,
  344. const char *key_name,
  345. const void *value,
  346. size_t value_len,
  347. icmap_value_types_t type)
  348. {
  349. struct icmap_item *item;
  350. struct icmap_item *new_item;
  351. size_t new_value_len;
  352. size_t new_item_size;
  353. if (value == NULL || key_name == NULL) {
  354. return (CS_ERR_INVALID_PARAM);
  355. }
  356. if (icmap_check_value_len(value, value_len, type) != 0) {
  357. return (CS_ERR_INVALID_PARAM);
  358. }
  359. item = qb_map_get(map->qb_map, key_name);
  360. if (item != NULL) {
  361. /*
  362. * Check that key is really changed
  363. */
  364. if (icmap_item_eq(item, value, value_len, type)) {
  365. return (CS_OK);
  366. }
  367. } else {
  368. if (icmap_check_key_name(key_name) != 0) {
  369. return (CS_ERR_NAME_TOO_LONG);
  370. }
  371. }
  372. if (type == ICMAP_VALUETYPE_BINARY || type == ICMAP_VALUETYPE_STRING) {
  373. if (type == ICMAP_VALUETYPE_STRING) {
  374. new_value_len = strlen((const char *)value);
  375. if (new_value_len > value_len) {
  376. new_value_len = value_len;
  377. }
  378. new_value_len++;
  379. } else {
  380. new_value_len = value_len;
  381. }
  382. } else {
  383. new_value_len = icmap_get_valuetype_len(type);
  384. }
  385. new_item_size = sizeof(struct icmap_item) + new_value_len;
  386. new_item = malloc(new_item_size);
  387. if (new_item == NULL) {
  388. return (CS_ERR_NO_MEMORY);
  389. }
  390. memset(new_item, 0, new_item_size);
  391. if (item == NULL) {
  392. new_item->key_name = strdup(key_name);
  393. if (new_item->key_name == NULL) {
  394. free(new_item);
  395. return (CS_ERR_NO_MEMORY);
  396. }
  397. } else {
  398. new_item->key_name = item->key_name;
  399. item->key_name = NULL;
  400. }
  401. new_item->type = type;
  402. new_item->value_len = new_value_len;
  403. memcpy(new_item->value, value, new_value_len);
  404. if (new_item->type == ICMAP_VALUETYPE_STRING) {
  405. ((char *)new_item->value)[new_value_len - 1] = 0;
  406. }
  407. qb_map_put(map->qb_map, new_item->key_name, new_item);
  408. return (CS_OK);
  409. }
  410. cs_error_t icmap_set(
  411. const char *key_name,
  412. const void *value,
  413. size_t value_len,
  414. icmap_value_types_t type)
  415. {
  416. return (icmap_set_r(icmap_global_map, key_name, value, value_len, type));
  417. }
  418. cs_error_t icmap_set_int8_r(const icmap_map_t map, const char *key_name, int8_t value)
  419. {
  420. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_INT8));
  421. }
  422. cs_error_t icmap_set_uint8_r(const icmap_map_t map, const char *key_name, uint8_t value)
  423. {
  424. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_UINT8));
  425. }
  426. cs_error_t icmap_set_int16_r(const icmap_map_t map, const char *key_name, int16_t value)
  427. {
  428. return (icmap_set_r(map,key_name, &value, sizeof(value), ICMAP_VALUETYPE_INT16));
  429. }
  430. cs_error_t icmap_set_uint16_r(const icmap_map_t map, const char *key_name, uint16_t value)
  431. {
  432. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_UINT16));
  433. }
  434. cs_error_t icmap_set_int32_r(const icmap_map_t map, const char *key_name, int32_t value)
  435. {
  436. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_INT32));
  437. }
  438. cs_error_t icmap_set_uint32_r(const icmap_map_t map, const char *key_name, uint32_t value)
  439. {
  440. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_UINT32));
  441. }
  442. cs_error_t icmap_set_int64_r(const icmap_map_t map, const char *key_name, int64_t value)
  443. {
  444. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_INT64));
  445. }
  446. cs_error_t icmap_set_uint64_r(const icmap_map_t map, const char *key_name, uint64_t value)
  447. {
  448. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_UINT64));
  449. }
  450. cs_error_t icmap_set_float_r(const icmap_map_t map, const char *key_name, float value)
  451. {
  452. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_FLOAT));
  453. }
  454. cs_error_t icmap_set_double_r(const icmap_map_t map, const char *key_name, double value)
  455. {
  456. return (icmap_set_r(map, key_name, &value, sizeof(value), ICMAP_VALUETYPE_DOUBLE));
  457. }
  458. cs_error_t icmap_set_string_r(const icmap_map_t map, const char *key_name, const char *value)
  459. {
  460. if (value == NULL) {
  461. return (CS_ERR_INVALID_PARAM);
  462. }
  463. return (icmap_set_r(map, key_name, value, strlen(value), ICMAP_VALUETYPE_STRING));
  464. }
  465. cs_error_t icmap_set_int8(const char *key_name, int8_t value)
  466. {
  467. return (icmap_set_int8_r(icmap_global_map, key_name, value));
  468. }
  469. cs_error_t icmap_set_uint8(const char *key_name, uint8_t value)
  470. {
  471. return (icmap_set_uint8_r(icmap_global_map, key_name, value));
  472. }
  473. cs_error_t icmap_set_int16(const char *key_name, int16_t value)
  474. {
  475. return (icmap_set_int16_r(icmap_global_map, key_name, value));
  476. }
  477. cs_error_t icmap_set_uint16(const char *key_name, uint16_t value)
  478. {
  479. return (icmap_set_uint16_r(icmap_global_map, key_name, value));
  480. }
  481. cs_error_t icmap_set_int32(const char *key_name, int32_t value)
  482. {
  483. return (icmap_set_int32_r(icmap_global_map, key_name, value));
  484. }
  485. cs_error_t icmap_set_uint32(const char *key_name, uint32_t value)
  486. {
  487. return (icmap_set_uint32_r(icmap_global_map, key_name, value));
  488. }
  489. cs_error_t icmap_set_int64(const char *key_name, int64_t value)
  490. {
  491. return (icmap_set_int64_r(icmap_global_map, key_name, value));
  492. }
  493. cs_error_t icmap_set_uint64(const char *key_name, uint64_t value)
  494. {
  495. return (icmap_set_uint64_r(icmap_global_map, key_name, value));
  496. }
  497. cs_error_t icmap_set_float(const char *key_name, float value)
  498. {
  499. return (icmap_set_float_r(icmap_global_map, key_name, value));
  500. }
  501. cs_error_t icmap_set_double(const char *key_name, double value)
  502. {
  503. return (icmap_set_double_r(icmap_global_map, key_name, value));
  504. }
  505. cs_error_t icmap_set_string(const char *key_name, const char *value)
  506. {
  507. return (icmap_set_string_r(icmap_global_map, key_name, value));
  508. }
  509. cs_error_t icmap_delete_r(const icmap_map_t map, const char *key_name)
  510. {
  511. struct icmap_item *item;
  512. if (key_name == NULL) {
  513. return (CS_ERR_INVALID_PARAM);
  514. }
  515. item = qb_map_get(map->qb_map, key_name);
  516. if (item == NULL) {
  517. return (CS_ERR_NOT_EXIST);
  518. }
  519. if (qb_map_rm(map->qb_map, item->key_name) != QB_TRUE) {
  520. return (CS_ERR_NOT_EXIST);
  521. }
  522. return (CS_OK);
  523. }
  524. cs_error_t icmap_delete(const char *key_name)
  525. {
  526. return (icmap_delete_r(icmap_global_map, key_name));
  527. }
  528. static cs_error_t icmap_get_ref_r(
  529. const icmap_map_t map,
  530. const char *key_name,
  531. void **value,
  532. size_t *value_len,
  533. icmap_value_types_t *type)
  534. {
  535. struct icmap_item *item;
  536. if (key_name == NULL) {
  537. return (CS_ERR_INVALID_PARAM);
  538. }
  539. item = qb_map_get(map->qb_map, key_name);
  540. if (item == NULL) {
  541. return (CS_ERR_NOT_EXIST);
  542. }
  543. if (type != NULL) {
  544. *type = item->type;
  545. }
  546. if (value_len != NULL) {
  547. *value_len = item->value_len;
  548. }
  549. if (value != NULL) {
  550. *value = item->value;
  551. }
  552. return (CS_OK);
  553. }
  554. cs_error_t icmap_get_r(
  555. const icmap_map_t map,
  556. const char *key_name,
  557. void *value,
  558. size_t *value_len,
  559. icmap_value_types_t *type)
  560. {
  561. cs_error_t res;
  562. void *tmp_value;
  563. size_t tmp_value_len;
  564. res = icmap_get_ref_r(map, key_name, &tmp_value, &tmp_value_len, type);
  565. if (res != CS_OK) {
  566. return (res);
  567. }
  568. if (value == NULL) {
  569. if (value_len != NULL) {
  570. *value_len = tmp_value_len;
  571. }
  572. } else {
  573. if (value_len == NULL || *value_len < tmp_value_len) {
  574. return (CS_ERR_INVALID_PARAM);
  575. }
  576. *value_len = tmp_value_len;
  577. memcpy(value, tmp_value, tmp_value_len);
  578. }
  579. return (CS_OK);
  580. }
  581. cs_error_t icmap_get(
  582. const char *key_name,
  583. void *value,
  584. size_t *value_len,
  585. icmap_value_types_t *type)
  586. {
  587. return (icmap_get_r(icmap_global_map, key_name, value, value_len, type));
  588. }
  589. cs_error_t icmap_get_string_r(icmap_map_t map, const char *key_name, char **str)
  590. {
  591. cs_error_t res;
  592. size_t str_len;
  593. icmap_value_types_t type;
  594. res = icmap_get_r(map, key_name, NULL, &str_len, &type);
  595. if (res != CS_OK || type != ICMAP_VALUETYPE_STRING) {
  596. if (res == CS_OK) {
  597. res = CS_ERR_INVALID_PARAM;
  598. }
  599. goto return_error;
  600. }
  601. *str = malloc(str_len);
  602. if (*str == NULL) {
  603. res = CS_ERR_NO_MEMORY;
  604. goto return_error;
  605. }
  606. res = icmap_get_r(map, key_name, *str, &str_len, &type);
  607. if (res != CS_OK) {
  608. free(*str);
  609. goto return_error;
  610. }
  611. return (CS_OK);
  612. return_error:
  613. return (res);
  614. }
  615. static cs_error_t icmap_get_int_r(
  616. const icmap_map_t map,
  617. const char *key_name,
  618. void *value,
  619. icmap_value_types_t type)
  620. {
  621. char key_value[16];
  622. size_t key_size;
  623. cs_error_t err;
  624. icmap_value_types_t key_type;
  625. key_size = sizeof(key_value);
  626. memset(key_value, 0, key_size);
  627. err = icmap_get_r(map, key_name, key_value, &key_size, &key_type);
  628. if (err != CS_OK)
  629. return (err);
  630. if (key_type != type) {
  631. return (CS_ERR_INVALID_PARAM);
  632. }
  633. memcpy(value, key_value, icmap_get_valuetype_len(key_type));
  634. return (CS_OK);
  635. }
  636. cs_error_t icmap_get_int8_r(const icmap_map_t map, const char *key_name, int8_t *i8)
  637. {
  638. return (icmap_get_int_r(map, key_name, i8, ICMAP_VALUETYPE_INT8));
  639. }
  640. cs_error_t icmap_get_uint8_r(const icmap_map_t map, const char *key_name, uint8_t *u8)
  641. {
  642. return (icmap_get_int_r(map, key_name, u8, ICMAP_VALUETYPE_UINT8));
  643. }
  644. cs_error_t icmap_get_int16_r(const icmap_map_t map, const char *key_name, int16_t *i16)
  645. {
  646. return (icmap_get_int_r(map, key_name, i16, ICMAP_VALUETYPE_INT16));
  647. }
  648. cs_error_t icmap_get_uint16_r(const icmap_map_t map, const char *key_name, uint16_t *u16)
  649. {
  650. return (icmap_get_int_r(map, key_name, u16, ICMAP_VALUETYPE_UINT16));
  651. }
  652. cs_error_t icmap_get_int32_r(const icmap_map_t map, const char *key_name, int32_t *i32)
  653. {
  654. return (icmap_get_int_r(map, key_name, i32, ICMAP_VALUETYPE_INT32));
  655. }
  656. cs_error_t icmap_get_uint32_r(const icmap_map_t map, const char *key_name, uint32_t *u32)
  657. {
  658. return (icmap_get_int_r(map, key_name, u32, ICMAP_VALUETYPE_UINT32));
  659. }
  660. cs_error_t icmap_get_int64_r(const icmap_map_t map, const char *key_name, int64_t *i64)
  661. {
  662. return(icmap_get_int_r(map, key_name, i64, ICMAP_VALUETYPE_INT64));
  663. }
  664. cs_error_t icmap_get_uint64_r(const icmap_map_t map, const char *key_name, uint64_t *u64)
  665. {
  666. return (icmap_get_int_r(map, key_name, u64, ICMAP_VALUETYPE_UINT64));
  667. }
  668. cs_error_t icmap_get_float_r(const icmap_map_t map, const char *key_name, float *flt)
  669. {
  670. return (icmap_get_int_r(map, key_name, flt, ICMAP_VALUETYPE_FLOAT));
  671. }
  672. cs_error_t icmap_get_double_r(const icmap_map_t map, const char *key_name, double *dbl)
  673. {
  674. return (icmap_get_int_r(map, key_name, dbl, ICMAP_VALUETYPE_DOUBLE));
  675. }
  676. cs_error_t icmap_get_string(const char *key_name, char **str)
  677. {
  678. return (icmap_get_string_r(icmap_global_map, key_name, str));
  679. }
  680. cs_error_t icmap_get_int8(const char *key_name, int8_t *i8)
  681. {
  682. return (icmap_get_int8_r(icmap_global_map, key_name, i8));
  683. }
  684. cs_error_t icmap_get_uint8(const char *key_name, uint8_t *u8)
  685. {
  686. return (icmap_get_uint8_r(icmap_global_map, key_name, u8));
  687. }
  688. cs_error_t icmap_get_int16(const char *key_name, int16_t *i16)
  689. {
  690. return (icmap_get_int16_r(icmap_global_map, key_name, i16));
  691. }
  692. cs_error_t icmap_get_uint16(const char *key_name, uint16_t *u16)
  693. {
  694. return (icmap_get_uint16_r(icmap_global_map, key_name, u16));
  695. }
  696. cs_error_t icmap_get_int32(const char *key_name, int32_t *i32)
  697. {
  698. return (icmap_get_int32_r(icmap_global_map, key_name, i32));
  699. }
  700. cs_error_t icmap_get_uint32(const char *key_name, uint32_t *u32)
  701. {
  702. return (icmap_get_uint32_r(icmap_global_map, key_name, u32));
  703. }
  704. cs_error_t icmap_get_int64(const char *key_name, int64_t *i64)
  705. {
  706. return(icmap_get_int64_r(icmap_global_map, key_name, i64));
  707. }
  708. cs_error_t icmap_get_uint64(const char *key_name, uint64_t *u64)
  709. {
  710. return (icmap_get_uint64_r(icmap_global_map, key_name, u64));
  711. }
  712. cs_error_t icmap_get_float(const char *key_name, float *flt)
  713. {
  714. return (icmap_get_float_r(icmap_global_map, key_name, flt));
  715. }
  716. cs_error_t icmap_get_double(const char *key_name, double *dbl)
  717. {
  718. return (icmap_get_double_r(icmap_global_map, key_name, dbl));
  719. }
  720. cs_error_t icmap_adjust_int_r(
  721. const icmap_map_t map,
  722. const char *key_name,
  723. int32_t step)
  724. {
  725. struct icmap_item *item;
  726. uint8_t u8;
  727. uint16_t u16;
  728. uint32_t u32;
  729. uint64_t u64;
  730. cs_error_t err = CS_OK;
  731. if (key_name == NULL) {
  732. return (CS_ERR_INVALID_PARAM);
  733. }
  734. item = qb_map_get(map->qb_map, key_name);
  735. if (item == NULL) {
  736. return (CS_ERR_NOT_EXIST);
  737. }
  738. switch (item->type) {
  739. case ICMAP_VALUETYPE_INT8:
  740. case ICMAP_VALUETYPE_UINT8:
  741. memcpy(&u8, item->value, sizeof(u8));
  742. u8 += step;
  743. err = icmap_set(key_name, &u8, sizeof(u8), item->type);
  744. break;
  745. case ICMAP_VALUETYPE_INT16:
  746. case ICMAP_VALUETYPE_UINT16:
  747. memcpy(&u16, item->value, sizeof(u16));
  748. u16 += step;
  749. err = icmap_set(key_name, &u16, sizeof(u16), item->type);
  750. break;
  751. case ICMAP_VALUETYPE_INT32:
  752. case ICMAP_VALUETYPE_UINT32:
  753. memcpy(&u32, item->value, sizeof(u32));
  754. u32 += step;
  755. err = icmap_set(key_name, &u32, sizeof(u32), item->type);
  756. break;
  757. case ICMAP_VALUETYPE_INT64:
  758. case ICMAP_VALUETYPE_UINT64:
  759. memcpy(&u64, item->value, sizeof(u64));
  760. u64 += step;
  761. err = icmap_set(key_name, &u64, sizeof(u64), item->type);
  762. break;
  763. case ICMAP_VALUETYPE_FLOAT:
  764. case ICMAP_VALUETYPE_DOUBLE:
  765. case ICMAP_VALUETYPE_STRING:
  766. case ICMAP_VALUETYPE_BINARY:
  767. err = CS_ERR_INVALID_PARAM;
  768. break;
  769. }
  770. return (err);
  771. }
  772. cs_error_t icmap_adjust_int(
  773. const char *key_name,
  774. int32_t step)
  775. {
  776. return (icmap_adjust_int_r(icmap_global_map, key_name, step));
  777. }
  778. cs_error_t icmap_fast_adjust_int_r(
  779. const icmap_map_t map,
  780. const char *key_name,
  781. int32_t step)
  782. {
  783. struct icmap_item *item;
  784. cs_error_t err = CS_OK;
  785. if (key_name == NULL) {
  786. return (CS_ERR_INVALID_PARAM);
  787. }
  788. item = qb_map_get(map->qb_map, key_name);
  789. if (item == NULL) {
  790. return (CS_ERR_NOT_EXIST);
  791. }
  792. switch (item->type) {
  793. case ICMAP_VALUETYPE_INT8:
  794. case ICMAP_VALUETYPE_UINT8:
  795. *(uint8_t *)item->value += step;
  796. break;
  797. case ICMAP_VALUETYPE_INT16:
  798. case ICMAP_VALUETYPE_UINT16:
  799. *(uint16_t *)item->value += step;
  800. break;
  801. case ICMAP_VALUETYPE_INT32:
  802. case ICMAP_VALUETYPE_UINT32:
  803. *(uint32_t *)item->value += step;
  804. break;
  805. case ICMAP_VALUETYPE_INT64:
  806. case ICMAP_VALUETYPE_UINT64:
  807. *(uint64_t *)item->value += step;
  808. break;
  809. case ICMAP_VALUETYPE_FLOAT:
  810. case ICMAP_VALUETYPE_DOUBLE:
  811. case ICMAP_VALUETYPE_STRING:
  812. case ICMAP_VALUETYPE_BINARY:
  813. err = CS_ERR_INVALID_PARAM;
  814. break;
  815. }
  816. if (err == CS_OK) {
  817. qb_map_put(map->qb_map, item->key_name, item);
  818. }
  819. return (err);
  820. }
  821. cs_error_t icmap_fast_adjust_int(
  822. const char *key_name,
  823. int32_t step)
  824. {
  825. return (icmap_fast_adjust_int_r(icmap_global_map, key_name, step));
  826. }
  827. cs_error_t icmap_inc_r(const icmap_map_t map, const char *key_name)
  828. {
  829. return (icmap_adjust_int_r(map, key_name, 1));
  830. }
  831. cs_error_t icmap_inc(const char *key_name)
  832. {
  833. return (icmap_inc_r(icmap_global_map, key_name));
  834. }
  835. cs_error_t icmap_dec_r(const icmap_map_t map, const char *key_name)
  836. {
  837. return (icmap_adjust_int_r(map, key_name, -1));
  838. }
  839. cs_error_t icmap_dec(const char *key_name)
  840. {
  841. return (icmap_dec_r(icmap_global_map, key_name));
  842. }
  843. cs_error_t icmap_fast_inc_r(const icmap_map_t map, const char *key_name)
  844. {
  845. return (icmap_fast_adjust_int_r(map, key_name, 1));
  846. }
  847. cs_error_t icmap_fast_inc(const char *key_name)
  848. {
  849. return (icmap_fast_inc_r(icmap_global_map, key_name));
  850. }
  851. cs_error_t icmap_fast_dec_r(const icmap_map_t map, const char *key_name)
  852. {
  853. return (icmap_fast_adjust_int_r(map, key_name, -1));
  854. }
  855. cs_error_t icmap_fast_dec(const char *key_name)
  856. {
  857. return (icmap_fast_dec_r(icmap_global_map, key_name));
  858. }
  859. icmap_iter_t icmap_iter_init_r(const icmap_map_t map, const char *prefix)
  860. {
  861. return (qb_map_pref_iter_create(map->qb_map, prefix));
  862. }
  863. icmap_iter_t icmap_iter_init(const char *prefix)
  864. {
  865. return (icmap_iter_init_r(icmap_global_map, prefix));
  866. }
  867. const char *icmap_iter_next(icmap_iter_t iter, size_t *value_len, icmap_value_types_t *type)
  868. {
  869. struct icmap_item *item;
  870. const char *res;
  871. res = qb_map_iter_next(iter, (void **)&item);
  872. if (res == NULL) {
  873. return (res);
  874. }
  875. if (value_len != NULL) {
  876. *value_len = item->value_len;
  877. }
  878. if (type != NULL) {
  879. *type = item->type;
  880. }
  881. return (res);
  882. }
  883. void icmap_iter_finalize(icmap_iter_t iter)
  884. {
  885. qb_map_iter_free(iter);
  886. }
  887. static void icmap_notify_fn(uint32_t event, char *key, void *old_value, void *value, void *user_data)
  888. {
  889. icmap_track_t icmap_track = (icmap_track_t)user_data;
  890. struct icmap_item *new_item = (struct icmap_item *)value;
  891. struct icmap_item *old_item = (struct icmap_item *)old_value;
  892. struct icmap_notify_value new_val;
  893. struct icmap_notify_value old_val;
  894. if (value == NULL && old_value == NULL) {
  895. return ;
  896. }
  897. if (new_item != NULL) {
  898. new_val.type = new_item->type;
  899. new_val.len = new_item->value_len;
  900. new_val.data = new_item->value;
  901. } else {
  902. memset(&new_val, 0, sizeof(new_val));
  903. }
  904. /*
  905. * old_item == new_item if fast functions are used -> don't fill old value
  906. */
  907. if (old_item != NULL && old_item != new_item) {
  908. old_val.type = old_item->type;
  909. old_val.len = old_item->value_len;
  910. old_val.data = old_item->value;
  911. } else {
  912. memset(&old_val, 0, sizeof(old_val));
  913. }
  914. icmap_track->notify_fn(icmap_qbtt_to_tt(event),
  915. key,
  916. new_val,
  917. old_val,
  918. icmap_track->user_data);
  919. }
  920. cs_error_t icmap_track_add(
  921. const char *key_name,
  922. int32_t track_type,
  923. icmap_notify_fn_t notify_fn,
  924. void *user_data,
  925. icmap_track_t *icmap_track)
  926. {
  927. int32_t err;
  928. if (notify_fn == NULL || icmap_track == NULL) {
  929. return (CS_ERR_INVALID_PARAM);
  930. }
  931. if ((track_type & ~(ICMAP_TRACK_ADD | ICMAP_TRACK_DELETE | ICMAP_TRACK_MODIFY | ICMAP_TRACK_PREFIX)) != 0) {
  932. return (CS_ERR_INVALID_PARAM);
  933. }
  934. *icmap_track = malloc(sizeof(**icmap_track));
  935. if (*icmap_track == NULL) {
  936. return (CS_ERR_NO_MEMORY);
  937. }
  938. memset(*icmap_track, 0, sizeof(**icmap_track));
  939. if (key_name != NULL) {
  940. (*icmap_track)->key_name = strdup(key_name);
  941. };
  942. (*icmap_track)->track_type = track_type;
  943. (*icmap_track)->notify_fn = notify_fn;
  944. (*icmap_track)->user_data = user_data;
  945. if ((err = qb_map_notify_add(icmap_global_map->qb_map, (*icmap_track)->key_name, icmap_notify_fn,
  946. icmap_tt_to_qbtt(track_type), *icmap_track)) != 0) {
  947. free((*icmap_track)->key_name);
  948. free(*icmap_track);
  949. return (qb_to_cs_error(err));
  950. }
  951. list_init(&(*icmap_track)->list);
  952. list_add (&(*icmap_track)->list, &icmap_track_list_head);
  953. return (CS_OK);
  954. }
  955. cs_error_t icmap_track_delete(icmap_track_t icmap_track)
  956. {
  957. int32_t err;
  958. if ((err = qb_map_notify_del_2(icmap_global_map->qb_map, icmap_track->key_name,
  959. icmap_notify_fn, icmap_tt_to_qbtt(icmap_track->track_type), icmap_track)) != 0) {
  960. return (qb_to_cs_error(err));
  961. }
  962. list_del(&icmap_track->list);
  963. free(icmap_track->key_name);
  964. free(icmap_track);
  965. return (CS_OK);
  966. }
  967. void *icmap_track_get_user_data(icmap_track_t icmap_track)
  968. {
  969. return (icmap_track->user_data);
  970. }
  971. cs_error_t icmap_set_ro_access(const char *key_name, int prefix, int ro_access)
  972. {
  973. struct list_head *iter;
  974. struct icmap_ro_access_item *icmap_ro_ai;
  975. for (iter = icmap_ro_access_item_list_head.next; iter != &icmap_ro_access_item_list_head; iter = iter->next) {
  976. icmap_ro_ai = list_entry(iter, struct icmap_ro_access_item, list);
  977. if (icmap_ro_ai->prefix == prefix && strcmp(key_name, icmap_ro_ai->key_name) == 0) {
  978. /*
  979. * We found item
  980. */
  981. if (ro_access) {
  982. return (CS_ERR_EXIST);
  983. } else {
  984. list_del(&icmap_ro_ai->list);
  985. free(icmap_ro_ai->key_name);
  986. free(icmap_ro_ai);
  987. return (CS_OK);
  988. }
  989. }
  990. }
  991. if (!ro_access) {
  992. return (CS_ERR_NOT_EXIST);
  993. }
  994. icmap_ro_ai = malloc(sizeof(*icmap_ro_ai));
  995. if (icmap_ro_ai == NULL) {
  996. return (CS_ERR_NO_MEMORY);
  997. }
  998. memset(icmap_ro_ai, 0, sizeof(*icmap_ro_ai));
  999. icmap_ro_ai->key_name = strdup(key_name);
  1000. if (icmap_ro_ai->key_name == NULL) {
  1001. free(icmap_ro_ai);
  1002. return (CS_ERR_NO_MEMORY);
  1003. }
  1004. icmap_ro_ai->prefix = prefix;
  1005. list_init(&icmap_ro_ai->list);
  1006. list_add (&icmap_ro_ai->list, &icmap_ro_access_item_list_head);
  1007. return (CS_OK);
  1008. }
  1009. int icmap_is_key_ro(const char *key_name)
  1010. {
  1011. struct list_head *iter;
  1012. struct icmap_ro_access_item *icmap_ro_ai;
  1013. for (iter = icmap_ro_access_item_list_head.next; iter != &icmap_ro_access_item_list_head; iter = iter->next) {
  1014. icmap_ro_ai = list_entry(iter, struct icmap_ro_access_item, list);
  1015. if (icmap_ro_ai->prefix) {
  1016. if (strlen(icmap_ro_ai->key_name) > strlen(key_name))
  1017. continue;
  1018. if (strncmp(icmap_ro_ai->key_name, key_name, strlen(icmap_ro_ai->key_name)) == 0) {
  1019. return (CS_TRUE);
  1020. }
  1021. } else {
  1022. if (strcmp(icmap_ro_ai->key_name, key_name) == 0) {
  1023. return (CS_TRUE);
  1024. }
  1025. }
  1026. }
  1027. return (CS_FALSE);
  1028. }
  1029. cs_error_t icmap_copy_map(icmap_map_t dst_map, const icmap_map_t src_map)
  1030. {
  1031. icmap_iter_t iter;
  1032. size_t value_len;
  1033. icmap_value_types_t value_type;
  1034. const char *key_name;
  1035. cs_error_t err;
  1036. void *value;
  1037. iter = icmap_iter_init_r(src_map, NULL);
  1038. if (iter == NULL) {
  1039. return (CS_ERR_NO_MEMORY);
  1040. }
  1041. err = CS_OK;
  1042. while ((key_name = icmap_iter_next(iter, &value_len, &value_type)) != NULL) {
  1043. err = icmap_get_ref_r(src_map, key_name, &value, &value_len, &value_type);
  1044. if (err != CS_OK) {
  1045. goto exit_iter_finalize;
  1046. }
  1047. err = icmap_set_r(dst_map, key_name, value, value_len, value_type);
  1048. if (err != CS_OK) {
  1049. goto exit_iter_finalize;
  1050. }
  1051. }
  1052. exit_iter_finalize:
  1053. icmap_iter_finalize(iter);
  1054. return (err);
  1055. }