/* * Compact Elastic Binary Trees - exported functions operating on u32 keys * * Copyright (C) 2014-2025 Willy Tarreau - w@1wt.eu * * Permission is hereby granted, free of charge, to any person obtaining * a copy of this software and associated documentation files (the * "Software"), to deal in the Software without restriction, including * without limitation the rights to use, copy, modify, merge, publish, * distribute, sublicense, and/or sell copies of the Software, and to * permit persons to whom the Software is furnished to do so, subject to * the following conditions: * * The above copyright notice and this permission notice shall be * included in all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR * OTHER DEALINGS IN THE SOFTWARE. */ #ifndef _CEB32_TREE_H #define _CEB32_TREE_H #include "cebtree.h" #include /* simpler version */ struct ceb_node *ceb32_imm_insert(struct ceb_root **root, struct ceb_node *node); struct ceb_node *ceb32_imm_first(struct ceb_root *const *root); struct ceb_node *ceb32_imm_last(struct ceb_root *const *root); struct ceb_node *ceb32_imm_lookup(struct ceb_root *const *root, uint32_t key); struct ceb_node *ceb32_imm_lookup_le(struct ceb_root *const *root, uint32_t key); struct ceb_node *ceb32_imm_lookup_lt(struct ceb_root *const *root, uint32_t key); struct ceb_node *ceb32_imm_lookup_ge(struct ceb_root *const *root, uint32_t key); struct ceb_node *ceb32_imm_lookup_gt(struct ceb_root *const *root, uint32_t key); struct ceb_node *ceb32_imm_next_unique(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *ceb32_imm_prev_unique(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *ceb32_imm_next_dup(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *ceb32_imm_prev_dup(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *ceb32_imm_next(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *ceb32_imm_prev(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *ceb32_imm_delete(struct ceb_root **root, struct ceb_node *node); struct ceb_node *ceb32_imm_pick(struct ceb_root **root, uint32_t key); struct ceb_node *cebu32_imm_insert(struct ceb_root **root, struct ceb_node *node); struct ceb_node *cebu32_imm_first(struct ceb_root *const *root); struct ceb_node *cebu32_imm_last(struct ceb_root *const *root); struct ceb_node *cebu32_imm_lookup(struct ceb_root *const *root, uint32_t key); struct ceb_node *cebu32_imm_lookup_le(struct ceb_root *const *root, uint32_t key); struct ceb_node *cebu32_imm_lookup_lt(struct ceb_root *const *root, uint32_t key); struct ceb_node *cebu32_imm_lookup_ge(struct ceb_root *const *root, uint32_t key); struct ceb_node *cebu32_imm_lookup_gt(struct ceb_root *const *root, uint32_t key); struct ceb_node *cebu32_imm_next(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *cebu32_imm_prev(struct ceb_root *const *root, struct ceb_node *node); struct ceb_node *cebu32_imm_delete(struct ceb_root **root, struct ceb_node *node); struct ceb_node *cebu32_imm_pick(struct ceb_root **root, uint32_t key); /* generic dump function */ void ceb32_imm_default_dump(struct ceb_root **ceb_root, const char *label, const void *ctx, int sub); /* returns the pointer to the uint32_t key */ static inline uint32_t *ceb32_imm_key(const struct ceb_node *node) { return (uint32_t *)ceb_key_ptr(node, sizeof(struct ceb_node)); } /* version taking a key offset */ struct ceb_node *ceb32_ofs_insert(struct ceb_root **root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_first(struct ceb_root *const *root, ptrdiff_t kofs); struct ceb_node *ceb32_ofs_last(struct ceb_root *const *root, ptrdiff_t kofs); struct ceb_node *ceb32_ofs_lookup(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *ceb32_ofs_lookup_le(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *ceb32_ofs_lookup_lt(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *ceb32_ofs_lookup_ge(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *ceb32_ofs_lookup_gt(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *ceb32_ofs_next_unique(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_prev_unique(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_next_dup(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_prev_dup(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_next(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_prev(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_delete(struct ceb_root **root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *ceb32_ofs_pick(struct ceb_root **root, ptrdiff_t kofs, uint32_t key); struct ceb_node *cebu32_ofs_insert(struct ceb_root **root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *cebu32_ofs_first(struct ceb_root *const *root, ptrdiff_t kofs); struct ceb_node *cebu32_ofs_last(struct ceb_root *const *root, ptrdiff_t kofs); struct ceb_node *cebu32_ofs_lookup(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *cebu32_ofs_lookup_le(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *cebu32_ofs_lookup_lt(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *cebu32_ofs_lookup_ge(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *cebu32_ofs_lookup_gt(struct ceb_root *const *root, ptrdiff_t kofs, uint32_t key); struct ceb_node *cebu32_ofs_next(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *cebu32_ofs_prev(struct ceb_root *const *root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *cebu32_ofs_delete(struct ceb_root **root, ptrdiff_t kofs, struct ceb_node *node); struct ceb_node *cebu32_ofs_pick(struct ceb_root **root, ptrdiff_t kofs, uint32_t key); /* generic dump function taking a key offset */ void ceb32_ofs_default_dump(struct ceb_root *const *root, ptrdiff_t kofs, const char *label, const void *ctx, int sub); /* returns the pointer to the uint32_t key */ static inline uint32_t *ceb32_ofs_key(const struct ceb_node *node, ptrdiff_t kofs) { return (uint32_t *)ceb_key_ptr(node, kofs); } /* insert at root , the item using node member and * key member , and returns either the inserted item, or the one that * was found there. */ #define ceb32_item_insert(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_insert(root, _kofs, &(item_ptr)->nname); \ (typeof(item_ptr))((char *)(_node) - _nofs); \ }) /* descend root , and return the first item of type , using node * member and key member , or NULL if the tree is empty. */ #define ceb32_item_first(root, nname, kname, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_first(root, _kofs); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* descend root , and return the last item of type , using node * member and key member , or NULL if the tree is empty. */ #define ceb32_item_last(root, nname, kname, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_last(root, _kofs); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the first item of type which contains the * key , using node member and key member . Returns the * pointer to the item, or NULL if not found. */ #define ceb32_item_lookup(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_lookup(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the last of * the greatest keys that are lower than or equal to , using node member * and key member . Returns the pointer to the item, or NULL if * not found. */ #define ceb32_item_lookup_le(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_lookup_le(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the last of * the greatest keys that are strictly lower than , using node member * and key member . Returns the pointer to the item, or NULL if * not found. */ #define ceb32_item_lookup_lt(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_lookup_lt(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the first of * the lowest keys key that are greater than or equal to , using node * member and key member . Returns the pointer to the item, or * NULL if not found. */ #define ceb32_item_lookup_ge(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_lookup_ge(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the first of * the lowest keys that are strictly greater than , using node member * and key member . Returns the pointer to the item, or NULL if * not found. */ #define ceb32_item_lookup_gt(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_lookup_gt(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the first item after item that has a * strictly greater key, using node member and key member , and * returns either the found item, or NULL if none exists. */ #define ceb32_item_next_unique(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_next_unique(root, _kofs, &(item_ptr)->nname);\ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the last item before item that has a * strictly lower key, using node member and key member , and * returns either the found item, or NULL if none exists. */ #define ceb32_item_prev_unique(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_prev_unique(root, _kofs, &(item_ptr)->nname);\ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the first item after item that has * exactly the same key, using node member and key member , and * returns either the found item, or NULL if none exists. */ #define ceb32_item_next_dup(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_next_dup(root, _kofs, &(item_ptr)->nname); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the last item before item that has * exactly the same key, using node member and key member , and * returns either the found item, or NULL if none exists. */ #define ceb32_item_prev_dup(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_prev_dup(root, _kofs, &(item_ptr)->nname); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the first item immediately after item , * using node member and key member , and returns either the * found item, or NULL if none exists. */ #define ceb32_item_next(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_next(root, _kofs, &(item_ptr)->nname); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the last item immediately before item , * using node member and key member , and returns either the * found item, or NULL if none exists. */ #define ceb32_item_prev(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_prev(root, _kofs, &(item_ptr)->nname); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item using node member and * key member , deletes it and returns it. If the item is NULL or absent * from the tree, NULL is returned. */ #define ceb32_item_delete(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ typeof(item_ptr) _item = (item_ptr); \ struct ceb_node *_node = ceb32_ofs_delete(root, _kofs, _item ? &_item->nname : NULL); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the first item of type which contains the * key , using node member and key member , deletes it and * returns it. If the key is not found in the tree, NULL is returned. */ #define ceb32_item_pick(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = ceb32_ofs_pick(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /*** versions using unique keys below ***/ /* insert at root , the item using node member and key * member , and returns either the inserted item, or the one that was * found there. */ #define cebu32_item_insert(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_insert(root, _kofs, &(item_ptr)->nname); \ (typeof(item_ptr))((char *)(_node) - _nofs); \ }) /* descend root , and return the first item of type , using node * member and key member , or NULL if the tree is empty. */ #define cebu32_item_first(root, nname, kname, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_first(root, _kofs); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* descend root , and return the last item of type , using node * member and key member , or NULL if the tree is empty. */ #define cebu32_item_last(root, nname, kname, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_last(root, _kofs); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the key * , using node member and key member . Returns the pointer * to the item, or NULL if not found. */ #define cebu32_item_lookup(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_lookup(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the greatest * key that is lower than or equal to , using node member and key * member . Returns the pointer to the item, or NULL if not found. */ #define cebu32_item_lookup_le(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_lookup_le(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the greatest * key that is strictly lower than , using node member and key * member . Returns the pointer to the item, or NULL if not found. */ #define cebu32_item_lookup_lt(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_lookup_lt(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the lowest * key key that is greater than or equal to , using node member * and key member . Returns the pointer to the item, or NULL if not * found. */ #define cebu32_item_lookup_ge(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_lookup_ge(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item of type which contains the lowest * key that is strictly greater than , using node member and key * member . Returns the pointer to the item, or NULL if not found. */ #define cebu32_item_lookup_gt(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_lookup_gt(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the first item immediately after item , * using node member and key member , and returns either the * found item, or NULL if none exists. */ #define cebu32_item_next(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_next(root, _kofs, &(item_ptr)->nname); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the last item immediately before item , * using node member and key member , and returns either the * found item, or NULL if none exists. */ #define cebu32_item_prev(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_prev(root, _kofs, &(item_ptr)->nname); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the item using node member and * key member , deletes it and returns it. If the item is NULL or absent * from the tree, NULL is returned. */ #define cebu32_item_delete(root, nname, kname, item_ptr) ({ \ ptrdiff_t _nofs = offsetof(typeof(*(item_ptr)), nname); \ ptrdiff_t _kofs = offsetof(typeof(*(item_ptr)), kname) - _nofs; \ typeof(item_ptr) _item = (item_ptr); \ struct ceb_node *_node = cebu32_ofs_delete(root, _kofs, _item ? &_item->nname : NULL); \ _node ? (typeof(item_ptr))((char *)(_node) - _nofs) : NULL; \ }) /* lookup from root , the first item of type which contains the * key , using node member and key member , deletes it and * returns it. If the key is not found in the tree, NULL is returned. */ #define cebu32_item_pick(root, nname, kname, key, type) ({ \ ptrdiff_t _nofs = offsetof(type, nname); \ ptrdiff_t _kofs = offsetof(type, kname) - _nofs; \ struct ceb_node *_node = cebu32_ofs_pick(root, _kofs, key); \ _node ? (type *)((char *)(_node) - _nofs) : NULL; \ }) #endif /* _CEB32_TREE_H */