3 /* dict - a generic dictionnary, variation over the B-tree concept */
5 /* Authors: Martin Quinson */
6 /* Copyright (C) 2003 the OURAGAN project. */
8 /* This program is free software; you can redistribute it and/or modify it
9 under the terms of the license (GNU LGPL) which comes with this package. */
11 #include "dict_private.h" /* prototypes of this module */
13 XBT_LOG_EXTERNAL_CATEGORY(dict);
14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(dict_elm,dict,"Dictionaries internals");
16 XBT_LOG_NEW_SUBCATEGORY(dict_add,dict,"Dictionaries internals: elements addition");
17 XBT_LOG_NEW_SUBCATEGORY(dict_search,dict,"Dictionaries internals: searching");
18 XBT_LOG_NEW_SUBCATEGORY(dict_remove,dict,"Dictionaries internals: elements removal");
19 XBT_LOG_NEW_SUBCATEGORY(dict_collapse,dict,"Dictionaries internals: post-removal cleanup");
20 XBT_LOG_NEW_SUBCATEGORY(dict_multi,dict,"Dictionaries internals: dictionaries of dictionaries");
22 /*####[ Private prototypes ]#################################################*/
24 static _XBT_INLINE void _xbt_dictelm_alloc(char *key,
28 void_f_pvoid_t *free_f,
29 /*OUT*/s_xbt_dictelm_t **where);
30 static void _dictelm_wrapper_free(void*);
32 static _XBT_INLINE void _str_prefix_lgr(const char *key1,
40 static void _xbt_dictelm_dump_rec(s_xbt_dictelm_t *head,
42 void_f_pvoid_t *output);
46 static void _xbt_dictelm_set_rec(s_xbt_dictelm_t *head,
51 void_f_pvoid_t *free_f);
52 static xbt_error_t _xbt_dictelm_get_rec(s_xbt_dictelm_t *head,
56 /* OUT */void **data);
57 static xbt_error_t _xbt_dictelm_remove_rec(s_xbt_dictelm_t *head,
64 _collapse_if_need(s_xbt_dictelm_t *p_head,
72 xbt_memdup(const void * const ptr,
73 const size_t length) {
74 void * new_ptr = NULL;
76 new_ptr = xbt_malloc(length);
77 memcpy(new_ptr, ptr, length);
83 * _xbt_nibble_to_char:
85 * Change any byte to a printable char
90 _xbt_nibble_to_char(unsigned char c) {
92 return c>9 ? c-10+'a' : c + '0';
96 * _xbt_bytes_to_string:
98 * Change any byte array to a printable string
99 * The length of string_container should at least be data_len*2+1
103 _xbt_bytes_to_string(char * const ptr,
105 char * const string_container) {
106 unsigned char *src = (unsigned char *)ptr;
107 char *dst = string_container;
110 *dst++ = _xbt_nibble_to_char(*src & 0x0f );
111 *dst++ = _xbt_nibble_to_char(*src++ & 0xf0 >> 4);
122 * _xbt_dictelm_alloc:
124 * Alloc a dict element with no child.
128 _xbt_dictelm_alloc(char *key,
132 void_f_pvoid_t *free_f,
133 /*OUT*/s_xbt_dictelm_t **pp_elm) {
134 xbt_error_t errcode = no_error;
135 s_xbt_dictelm_t *p_elm = NULL;
137 p_elm = xbt_new(s_xbt_dictelm_t,1);
140 p_elm->key_len = key_len;
141 p_elm->offset = offset;
142 p_elm->content = data;
143 p_elm->free_f = free_f;
144 p_elm->sub = xbt_dynar_new(sizeof(s_xbt_dictelm_t*), _dictelm_wrapper_free);
153 * @pp_elm: the dict elem to be freed
155 * Frees a dictionnary element with all its childs.
158 xbt_dictelm_free(s_xbt_dictelm_t **pp_elm) {
160 s_xbt_dictelm_t *p_elm = *pp_elm;
162 xbt_dynar_free(&(p_elm->sub));
165 xbt_free(p_elm->key);
168 if (p_elm->free_f && p_elm->content) {
169 p_elm->free_f(p_elm->content);
178 * _dictelm_wrapper_free:
180 * a wrapper to free dictelm with the right prototype to be usable within dynar
184 _dictelm_wrapper_free(void *pp_elm) {
185 DEBUG3("Free dictelm '%.*s' %p",
186 (*(s_xbt_dictelm_t**)pp_elm)->key_len, (*(s_xbt_dictelm_t**)pp_elm)->key,
188 xbt_dictelm_free((s_xbt_dictelm_t**)pp_elm);
191 /*####[ utility functions ]##################################################*/
196 * Returns the length of the common prefix of @str1 and @str2.
197 * Do make sure the strings are not null
201 _str_prefix_lgr(const char *key1,
207 const int old_offset = *p_offset;
213 /*CDEBUG5(dict_search, "%s: [%.*s] <=> [%.*s]", __FUNCTION__,
214 key1,key_len1,key2,key_len2);*/
216 if (o < key_len1 && o < key_len2) {
218 while (key1[o] == key2[o]) {
221 if (!(o < key_len1 && o < key_len2))
229 if (o != old_offset) {
239 } else if (o >= key_len2) {
252 * _dictelm_child_cmp:
254 * Compares two dictelm keys and return their matching (using the same
255 * convention than @_xbt_dict_child_search() )
259 _dict_child_cmp(s_xbt_dictelm_t *p_dict,
266 s_xbt_dictelm_t *p_child = NULL;
271 p_child = xbt_dynar_get_as(p_dict->sub, pos, s_xbt_dictelm_t*);
273 /* Compute the length of the prefix
274 and if the searched key is before or after cur */
275 _str_prefix_lgr(p_child->key, p_child->key_len,
280 if (m) /* found, get out */
283 if (o < p_child->key_len && (o >= key_len || key[o] < p_child->key[o])) {
289 CDEBUG6(dict_search, "Cmp '%.*s' and '%.*s' (offset=%d) => %d",
290 p_child->key_len - *p_offset, p_child->key + *p_offset,
291 key_len - *p_offset, key + *p_offset,
301 * _xbt_dict_child_search:
303 * Search where would be inserted @key between the childs of @p_elm.
305 * Returns position of the child having a common prefix with this key
306 * If *match==0, no child have a common prefix
307 * *pos is where to add the key
308 * If *match==1, A child (located at *pos) have exactly this key
309 * If *match==2, A child (located at *pos) constitutes a prefix of the key
310 * the recursion have to go on that guy
311 * *prefix = the size of the key eaten at this level
312 * If *match==3 The key is a prefix of the child at *pos
313 * If *match==4, A child (loc. at *pos) share a common prefix with this key
314 * *prefix = size of the prefix.
315 * If searching, that's a mismatch.
316 * If inserting, you have to break the child and create an
317 * internal node having {child, key} as childs
318 * offset is used in input and output. In input, that's the length of the key
319 * handled by previous levels of recursion. In output, that the one counting
324 _xbt_dictelm_child_search(s_xbt_dictelm_t *p_elm,
337 CDEBUG5(dict_search, "search child [%.*s] under [%.*s] (len=%lu)",
339 p_elm?p_elm->key_len:6, p_elm?p_elm->key:"(head)",
340 (p_elm&&p_elm->sub)?xbt_dynar_length(p_elm->sub):0);
343 len = xbt_dynar_length(p_elm->sub);
345 for (p = 0; p < len; p++) {
348 _dict_child_cmp(p_elm, p, key, key_len, &o, &m, &cmp);
360 CDEBUG5(dict_search, "search [%.*s] in [%.*s] => %s",
362 p_elm?p_elm->key_len:6, p_elm?p_elm->key:"(head)",
363 ( m == 0 ? "no child have a common prefix" :
364 ( m == 1 ? "selected child have exactly this key" :
365 ( m == 2 ? "selected child constitutes a prefix" :
366 ( m == 3 ? "key is a prefix of selected child" :
367 (m == 4 ? "selected child share a prefix" :
368 "internal error")))))
373 * _xbt_dictelm_change_value:
375 * Change the value of the dictelm, making sure to free the old one, if any.
379 _xbt_dictelm_change_value(s_xbt_dictelm_t *p_elm,
381 void_f_pvoid_t *free_f) {
383 if (p_elm->content && p_elm->free_f) {
384 p_elm->free_f(p_elm->content);
387 p_elm->free_f = free_f;
388 p_elm->content = data;
392 * _xbt_dictelm_set_rec:
394 * @head: the head of the dict
395 * @key: the key to set the new data
396 * @offset: offset on key.
397 * @data: the data to add in the dict
398 * @Returns: a gras_error
400 * set the @data in the structure under the @key. The @key is destroyed
401 * in the process. Think to strdup it before.
403 * This is a helper function to xbt_dict_set which locks the struct and
404 * strdup the key before action.
407 _xbt_dictelm_set_rec(s_xbt_dictelm_t *p_head,
412 void_f_pvoid_t *free_f) {
415 const int old_offset = offset;
417 CDEBUG6(dict_add, "--> Insert '%.*s' after '%.*s' (offset=%d) in tree %p",
419 ((p_head && p_head->key) ? p_head->key_len : 6),
420 ((p_head && p_head->key) ? p_head->key : "(head)"),
421 offset, (void*)p_head);
423 /*** The trivial cases first ***/
425 /* there is no key (we did enough recursion), change the value of head */
426 if (offset >= key_len) {
428 CDEBUG0(dict_add, "--> Change the value of head");
430 _xbt_dictelm_change_value(p_head, data, free_f);
431 xbt_free(key); /* Keep the key used in the tree */
436 /*** Search where to add this child, and how ***/
437 _xbt_dictelm_child_search(p_head, key, key_len, &pos, &offset, &match);
439 CDEBUG3(dict_add, "child_search => pos=%d, offset=%d, match=%d",
444 case 0: /* no child have a common prefix */
446 s_xbt_dictelm_t *p_child = NULL;
448 _xbt_dictelm_alloc(key, key_len, offset, data, free_f, &p_child);
449 CDEBUG1(dict_add, "-> Add a child %p", (void*)p_child);
450 xbt_dynar_insert_at(p_head->sub, pos, &p_child);
455 case 1: /* A child have exactly this key => change its value*/
457 s_xbt_dictelm_t *p_child = NULL;
459 p_child = xbt_dynar_get_as(p_head->sub, pos, s_xbt_dictelm_t*);
460 CDEBUG1(dict_add, "-> Change the value of the child %p", (void*)p_child);
461 _xbt_dictelm_change_value(p_child, data, free_f);
468 case 2: /* A child constitutes a prefix of the key => recurse */
470 s_xbt_dictelm_t *p_child = NULL;
472 p_child=xbt_dynar_get_as(p_head->sub, pos, s_xbt_dictelm_t*);
473 CDEBUG2(dict_add,"-> Recurse on %p (offset=%d)", (void*)p_child, offset);
475 _xbt_dictelm_set_rec(p_child, key, key_len,
476 offset, data, free_f);
480 case 3: /* The key is a prefix of the child => child becomes child of p_new */
482 s_xbt_dictelm_t *p_new = NULL;
483 s_xbt_dictelm_t *p_child = NULL;
485 p_child=xbt_dynar_get_as(p_head->sub, pos, s_xbt_dictelm_t*);
486 _xbt_dictelm_alloc(key, key_len, old_offset, data, free_f, &p_new);
488 CDEBUG2(dict_add, "-> The child %p become child of new dict (%p)",
489 (void*)p_child, (void*)p_new);
491 xbt_dynar_push(p_new->sub, &p_child);
492 p_child->offset = offset;
493 xbt_dynar_set(p_head->sub, pos, &p_new);
498 case 4: /* A child share a common prefix with this key => Common ancestor */
500 s_xbt_dictelm_t *p_new = NULL;
501 s_xbt_dictelm_t *p_child = NULL;
502 s_xbt_dictelm_t *p_anc = NULL;
503 char *anc_key = NULL;
504 int anc_key_len = offset;
506 _xbt_dictelm_alloc(key, key_len, offset, data, free_f, &p_new);
507 p_child=xbt_dynar_get_as(p_head->sub, pos, s_xbt_dictelm_t*);
509 anc_key = xbt_memdup(key, anc_key_len);
511 _xbt_dictelm_alloc(anc_key, anc_key_len, old_offset, NULL, NULL, &p_anc);
513 CDEBUG3(dict_add, "-> Make a common ancestor %p (%.*s)",
514 (void*)p_anc, anc_key_len, anc_key);
516 if (key[offset] < p_child->key[offset]) {
517 xbt_dynar_push(p_anc->sub, &p_new);
518 xbt_dynar_push(p_anc->sub, &p_child);
520 xbt_dynar_push(p_anc->sub, &p_child);
521 xbt_dynar_push(p_anc->sub, &p_new);
524 p_child->offset = offset;
526 xbt_dynar_set(p_head->sub, pos, &p_anc);
537 * xbt_dictelm_set_ext:
539 * @head: the head of the dict
540 * @key: the key to set the new data
541 * @data: the data to add in the dict
542 * @Returns: a gras_error
544 * set the @data in the structure under the @key, which can be any kind
545 * of data, as long as its length is provided in @key_len.
548 xbt_dictelm_set_ext(s_xbt_dictelm_t **pp_head,
552 void_f_pvoid_t *free_f) {
553 s_xbt_dictelm_t *p_head = *pp_head;
556 key = xbt_memdup(_key, key_len);
558 /* there is no head, create it */
560 s_xbt_dictelm_t *p_child = NULL;
562 CDEBUG0(dict_add, "Create an head");
564 /* The head is priviledged by being the only one with a NULL key */
565 _xbt_dictelm_alloc(NULL, 0, 0, NULL, NULL, &p_head);
567 _xbt_dictelm_alloc(key, key_len, 0, data, free_f, &p_child);
568 xbt_dynar_insert_at(p_head->sub, 0, &p_child);
575 _xbt_dictelm_set_rec(p_head, key, key_len, 0, data, free_f);
581 * @head: the head of the dict
582 * @key: the key to set the new data
583 * @data: the data to add in the dict
584 * @Returns: a gras_error
586 * set the @data in the structure under the @key, which is a
587 * null terminated string.
590 xbt_dictelm_set(s_xbt_dictelm_t **pp_head,
593 void_f_pvoid_t *free_f) {
595 xbt_dictelm_set_ext(pp_head, _key, 1+strlen(_key), data, free_f);
601 * @head: the head of the dict
602 * @key: the key to find data
603 * @offset: offset on the key
604 * @data: the data that we are looking for
605 * @Returns: gras_error
607 * Search the given @key. mismatch_error when not found.
611 _xbt_dictelm_get_rec(s_xbt_dictelm_t *p_head,
618 CDEBUG3(dict_search, "Search %.*s in %p", key_len, key, (void*)p_head);
620 /*** The trivial case first ***/
622 /* we did enough recursion, we're done */
623 if (offset >= key_len) {
624 *data = p_head->content;
633 *data = NULL; /* Make it ready to answer 'not found' in one operation */
635 /*** Search where is the good child, and how good it is ***/
636 _xbt_dictelm_child_search(p_head, key, key_len, &pos, &offset, &match);
640 case 0: /* no child have a common prefix */
641 return mismatch_error;
643 case 1: /* A child have exactly this key => Got it */
645 s_xbt_dictelm_t *p_child = NULL;
647 p_child = xbt_dynar_get_as(p_head->sub, pos, s_xbt_dictelm_t*);
648 *data = p_child->content;
653 case 2: /* A child constitutes a prefix of the key => recurse */
655 s_xbt_dictelm_t *p_child = NULL;
657 p_child = xbt_dynar_get_as(p_head->sub, pos, s_xbt_dictelm_t*);
659 return _xbt_dictelm_get_rec(p_child, key, key_len, offset, data);
662 case 3: /* The key is a prefix of the child => not found */
663 return mismatch_error;
665 case 4: /* A child share a common prefix with this key => not found */
666 return mismatch_error;
675 * xbt_dictelm_get_ext:
677 * @head: the head of the dict
678 * @key: the key to find data
679 * @data: the data that we are looking for
680 * @Returns: gras_error
682 * Search the given @key. mismatch_error when not found.
685 xbt_dictelm_get_ext(s_xbt_dictelm_t *p_head,
688 /* OUT */void **data) {
689 /* there is no head, go to hell */
691 return mismatch_error;
694 return _xbt_dictelm_get_rec(p_head, key, key_len, 0, data);
700 * @head: the head of the dict
701 * @key: the key to find data
702 * @data: the data that we are looking for
703 * @Returns: gras_error
705 * Search the given @key. mismatch_error when not found.
708 xbt_dictelm_get(s_xbt_dictelm_t *p_head,
710 /* OUT */void **data) {
712 return xbt_dictelm_get_ext(p_head, key, 1+strlen(key), data);
715 /*----[ _xbt_dict_collapse ]------------------------------------------------*/
718 _collapse_if_need(xbt_dictelm_t head,
721 xbt_dictelm_t child = NULL;
723 CDEBUG2(dict_collapse, "Collapse %d of %p... ", pos, (void*)head);
726 /* Remove the child if |it's key| == 0 (meaning it's dead) */
727 child = xbt_dynar_get_as(head->sub, pos, xbt_dictelm_t);
729 if (offset >= child->key_len) {
731 xbt_assert0(xbt_dynar_length(child->sub) == 0,
732 "Found a dead child with grand childs. Internal error");
734 CDEBUG1(dict_collapse, "Remove dead child %p.... ", (void*)child);
735 xbt_dynar_remove_at(head->sub, pos, &child);
736 xbt_dictelm_free(&child);
741 CDEBUG0(dict_collapse, "Do not collapse the head, you stupid programm");
745 if (head->content || head->free_f ||
746 xbt_dynar_length(head->sub) != 1) {
747 CDEBUG0(dict_collapse, "Cannot collapse");
748 return; /* cannot collapse */
751 child = xbt_dynar_get_as(head->sub, 0, xbt_dictelm_t);
753 /* Get the child's key as new key */
754 CDEBUG2(dict_collapse,
755 "Do collapse with only child %.*s", child->key_len, child->key);
757 head->content = child->content;
758 head->free_f = child->free_f;
760 head->key = child->key;
761 head->key_len = child->key_len;
763 xbt_dynar_free_container(&(head->sub)) ;
765 head->sub = child->sub;
770 * _xbt_dict_remove_rec:
772 * @head: the head of the dict
773 * @key: the key of the data to be removed
774 * @offset: offset on the key
775 * @Returns: xbt_error_t
777 * Remove the entry associated with the given @key
780 _xbt_dictelm_remove_rec(xbt_dictelm_t head,
784 xbt_error_t errcode = no_error;
786 /* there is no key to search, we did enough recursion => kill current */
787 if (offset >= key_len) {
788 int killme = 0; /* do I need to suicide me ? */
790 if (head->content && head->free_f) {
791 head->free_f(head->content);
794 killme = !xbt_dynar_length(head->sub);
795 head->content = NULL;
797 _collapse_if_need(head, -1, offset);
800 DEBUG0("kill this node");
801 head->key_len = 0; /* killme. Cleanup done one step higher in recursion */
809 int old_offset = offset;
811 /*** Search where is the good child, and how good it is ***/
812 _xbt_dictelm_child_search(head, key, key_len, &pos, &offset, &match);
816 case 1: /* A child have exactly this key => recurse */
817 case 2: /* A child constitutes a prefix of the key => recurse */
820 s_xbt_dictelm_t *p_child = NULL;
822 p_child = xbt_dynar_get_as(head->sub, pos, s_xbt_dictelm_t*);
823 /*DEBUG5("Recurse on child %d of %p to remove %.*s (prefix=%d)",
824 pos, (void*)p_child, key+offset, key_len-offset,offset);*/
825 TRY(_xbt_dictelm_remove_rec(p_child, key, key_len, offset));
827 _collapse_if_need(head, pos, old_offset);
832 case 0: /* no child have a common prefix */
833 case 3: /* The key is a prefix of the child => not found */
834 case 4: /* A child share a common prefix with this key => not found */
836 return mismatch_error;
847 * xbt_dictelm_remove_ext:
849 * @head: the head of the dict
850 * @key: the key of the data to be removed
851 * @Returns: xbt_error_t
853 * Remove the entry associated with the given @key
856 xbt_dictelm_remove_ext(xbt_dictelm_t head,
859 /* there is no head, go to hell */
861 RAISE0(mismatch_error, "there is no head, go to hell");
864 return _xbt_dictelm_remove_rec(head, key, key_len, 0);
868 * xbt_dictelm_remove:
870 * @head: the head of the dict
871 * @key: the key of the data to be removed
872 * @Returns: xbt_error_t
874 * Remove the entry associated with the given @key
877 xbt_dictelm_remove(xbt_dictelm_t head,
879 return _xbt_dictelm_remove_rec(head, key, 1+strlen(key),0);
882 /*----[ _xbt_dict_dump_rec ]------------------------------------------------*/
883 /* private function to do the job of xbt_dict_dump recursively */
884 /*---------------------------------------------------------------------------*/
887 _xbt_dictelm_dump_rec(xbt_dictelm_t head,
889 void_f_pvoid_t *output) {
890 xbt_dictelm_t child = NULL;
898 printf("[%p] ", (void*)head);
901 key_len = head->key_len;
906 for (i = 0; i < offset; i++)
916 char *key_string = NULL;
918 key_string = xbt_malloc(key_len*2+1);
919 _xbt_bytes_to_string(key, key_len, key_string);
921 printf("%.*s|(%d)", key_len-offset, key_string + offset, offset);
923 xbt_free(key_string);
933 output(head->content);
942 printf(" \t\t\t[ %lu child(s) ]\n", xbt_dynar_length(head->sub));
944 xbt_dynar_foreach(head->sub, i, child)
945 _xbt_dictelm_dump_rec(child, child->offset, output);
952 * @head: the head of the dict
953 * @output: a function to dump each data in the tree
954 * @Returns: xbt_error_t
956 * Ouputs the content of the structure. (for debuging purpose). @ouput is a
957 * function to output the data. If NULL, data won't be displayed.
961 xbt_dictelm_dump(xbt_dictelm_t head,
962 void_f_pvoid_t *output) {
963 _xbt_dictelm_dump_rec(head, 0, output);
967 * xbt_dictelm_print_fct:
971 * To dump multidict, this function dumps a dict
975 xbt_dictelm_print_fct(void *data) {
976 printf("tree %p", (void*)data);