4
0

icmap.c 30 KB

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