Commit | Line | Data |
---|---|---|
a42cc659 MD |
1 | #ifndef _URCU_RCULFHASH_H |
2 | #define _URCU_RCULFHASH_H | |
ab7d5fc6 | 3 | |
f0c29ed7 MD |
4 | /* |
5 | * urcu/rculfhash.h | |
6 | * | |
7 | * Userspace RCU library - Lock-Free RCU Hash Table | |
8 | * | |
9 | * Copyright 2011 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com> | |
0dcf4847 | 10 | * Copyright 2011 - Lai Jiangshan <laijs@cn.fujitsu.com> |
f0c29ed7 MD |
11 | * |
12 | * This library is free software; you can redistribute it and/or | |
13 | * modify it under the terms of the GNU Lesser General Public | |
14 | * License as published by the Free Software Foundation; either | |
15 | * version 2.1 of the License, or (at your option) any later version. | |
16 | * | |
17 | * This library is distributed in the hope that it will be useful, | |
18 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
20 | * Lesser General Public License for more details. | |
21 | * | |
22 | * You should have received a copy of the GNU Lesser General Public | |
23 | * License along with this library; if not, write to the Free Software | |
24 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
cf77d1fa MD |
25 | * |
26 | * Include this file _after_ including your URCU flavor. | |
f0c29ed7 MD |
27 | */ |
28 | ||
674f7a69 | 29 | #include <stdint.h> |
6d320126 | 30 | #include <urcu/compiler.h> |
abc490a1 | 31 | #include <urcu-call-rcu.h> |
7b17c13e | 32 | #include <urcu-flavor.h> |
abc490a1 | 33 | |
01389791 MD |
34 | #ifdef __cplusplus |
35 | extern "C" { | |
36 | #endif | |
37 | ||
7f52427b | 38 | /* |
04db56f8 | 39 | * cds_lfht_node: Contains the next pointers and reverse-hash |
7f52427b | 40 | * value required for lookup and traversal of the hash table. |
04db56f8 LJ |
41 | * |
42 | * struct cds_lfht_node should be aligned on 4-bytes boundaries because | |
43 | * the two lower bits are used as flags. | |
7f52427b | 44 | * |
ae62aa6a MD |
45 | * struct cds_lfht_node can be embedded into a structure (as a field). |
46 | * caa_container_of() can be used to get the structure from the struct | |
47 | * cds_lfht_node after a lookup. | |
04db56f8 LJ |
48 | * |
49 | * The structure which embeds it typically holds the key (or key-value | |
50 | * pair) of the object. The caller code is responsible for calculation | |
51 | * of the hash value for cds_lfht APIs. | |
ae62aa6a | 52 | */ |
14044b37 | 53 | struct cds_lfht_node { |
1ee8f000 | 54 | struct cds_lfht_node *next; /* ptr | BUCKET_FLAG | REMOVED_FLAG */ |
04db56f8 LJ |
55 | unsigned long reverse_hash; |
56 | } __attribute__((aligned(4))); | |
abc490a1 | 57 | |
7f52427b | 58 | /* cds_lfht_iter: Used to track state while traversing a hash chain. */ |
adc0de68 MD |
59 | struct cds_lfht_iter { |
60 | struct cds_lfht_node *node, *next; | |
61 | }; | |
62 | ||
63 | static inline | |
64 | struct cds_lfht_node *cds_lfht_iter_get_node(struct cds_lfht_iter *iter) | |
65 | { | |
66 | return iter->node; | |
67 | } | |
68 | ||
14044b37 | 69 | struct cds_lfht; |
ab7d5fc6 | 70 | |
5e28c532 MD |
71 | /* |
72 | * Caution ! | |
abc490a1 | 73 | * Ensure reader and writer threads are registered as urcu readers. |
5e28c532 MD |
74 | */ |
75 | ||
996ff57c | 76 | typedef int (*cds_lfht_match_fct)(struct cds_lfht_node *node, const void *key); |
abc490a1 | 77 | |
f0c29ed7 | 78 | /* |
14044b37 | 79 | * cds_lfht_node_init - initialize a hash table node |
0422d92c | 80 | * @node: the node to initialize. |
04db56f8 LJ |
81 | * |
82 | * This function is kept to be eventually used for debugging purposes | |
83 | * (detection of memory corruption). | |
f0c29ed7 | 84 | */ |
abc490a1 | 85 | static inline |
04db56f8 | 86 | void cds_lfht_node_init(struct cds_lfht_node *node) |
abc490a1 | 87 | { |
abc490a1 | 88 | } |
674f7a69 | 89 | |
b8af5011 MD |
90 | /* |
91 | * Hash table creation flags. | |
92 | */ | |
93 | enum { | |
94 | CDS_LFHT_AUTO_RESIZE = (1U << 0), | |
5afadd12 | 95 | CDS_LFHT_ACCOUNTING = (1U << 1), |
b8af5011 MD |
96 | }; |
97 | ||
0b6aa001 LJ |
98 | struct cds_lfht_mm_type { |
99 | struct cds_lfht *(*alloc_cds_lfht)(unsigned long min_nr_alloc_buckets, | |
100 | unsigned long max_nr_buckets); | |
101 | void (*alloc_bucket_table)(struct cds_lfht *ht, unsigned long order); | |
102 | void (*free_bucket_table)(struct cds_lfht *ht, unsigned long order); | |
103 | struct cds_lfht_node *(*bucket_at)(struct cds_lfht *ht, | |
104 | unsigned long index); | |
105 | }; | |
106 | ||
107 | extern const struct cds_lfht_mm_type cds_lfht_mm_order; | |
308d1cb3 | 108 | extern const struct cds_lfht_mm_type cds_lfht_mm_chunk; |
0b6aa001 | 109 | |
674f7a69 | 110 | /* |
7a9dcf9b | 111 | * _cds_lfht_new - API used by cds_lfht_new wrapper. Do not use directly. |
674f7a69 | 112 | */ |
0422d92c | 113 | struct cds_lfht *_cds_lfht_new(unsigned long init_size, |
0722081a | 114 | unsigned long min_nr_alloc_buckets, |
747d725c | 115 | unsigned long max_nr_buckets, |
b8af5011 | 116 | int flags, |
0b6aa001 | 117 | const struct cds_lfht_mm_type *mm, |
7b17c13e | 118 | const struct rcu_flavor_struct *flavor, |
b7d619b0 | 119 | pthread_attr_t *attr); |
ab7d5fc6 | 120 | |
7a9dcf9b MD |
121 | /* |
122 | * cds_lfht_new - allocate a hash table. | |
747d725c LJ |
123 | * @init_size: number of buckets to allocate initially. Must be power of two. |
124 | * @min_nr_alloc_buckets: the minimum number of allocated buckets. | |
125 | * (must be power of two) | |
126 | * @max_nr_buckets: the maximum number of hash table buckets allowed. | |
127 | * (must be power of two) | |
f22dd01d MD |
128 | * @flags: hash table creation flags (can be combined with bitwise or: '|'). |
129 | * 0: no flags. | |
130 | * CDS_LFHT_AUTO_RESIZE: automatically resize hash table. | |
44bbe7fa LJ |
131 | * CDS_LFHT_ACCOUNTING: count the number of node addition |
132 | * and removal in the table | |
b7d619b0 | 133 | * @attr: optional resize worker thread attributes. NULL for default. |
7a9dcf9b | 134 | * |
7a9dcf9b MD |
135 | * Return NULL on error. |
136 | * Note: the RCU flavor must be already included before the hash table header. | |
b7d619b0 MD |
137 | * |
138 | * The programmer is responsible for ensuring that resize operation has a | |
139 | * priority equal to hash table updater threads. It should be performed by | |
140 | * specifying the appropriate priority in the pthread "attr" argument, and, | |
141 | * for CDS_LFHT_AUTO_RESIZE, by ensuring that call_rcu worker threads also have | |
142 | * this priority level. Having lower priority for call_rcu and resize threads | |
143 | * does not pose any correctness issue, but the resize operations could be | |
144 | * starved by updates, thus leading to long hash table bucket chains. | |
44bbe7fa LJ |
145 | * Threads calling this API are NOT required to be registered RCU read-side |
146 | * threads. It can be called very early.(before rcu is initialized ...etc.) | |
7a9dcf9b MD |
147 | */ |
148 | static inline | |
0422d92c | 149 | struct cds_lfht *cds_lfht_new(unsigned long init_size, |
0722081a | 150 | unsigned long min_nr_alloc_buckets, |
747d725c | 151 | unsigned long max_nr_buckets, |
b7d619b0 MD |
152 | int flags, |
153 | pthread_attr_t *attr) | |
7a9dcf9b | 154 | { |
7b17c13e | 155 | return _cds_lfht_new(init_size, min_nr_alloc_buckets, max_nr_buckets, |
0b6aa001 | 156 | flags, &cds_lfht_mm_order, &rcu_flavor, attr); |
7a9dcf9b MD |
157 | } |
158 | ||
f0c29ed7 | 159 | /* |
14044b37 | 160 | * cds_lfht_destroy - destroy a hash table. |
b7d619b0 MD |
161 | * @ht: the hash table to destroy. |
162 | * @attr: (output) resize worker thread attributes, as received by cds_lfht_new. | |
163 | * The caller will typically want to free this pointer if dynamically | |
7f52427b MD |
164 | * allocated. The attr point can be NULL if the caller does not |
165 | * need to be informed of the value passed to cds_lfht_new(). | |
6878c72b MD |
166 | * |
167 | * Return 0 on success, negative error value on error. | |
3df0c49c | 168 | * Threads calling this API need to be registered RCU read-side threads. |
f0c29ed7 | 169 | */ |
b7d619b0 | 170 | int cds_lfht_destroy(struct cds_lfht *ht, pthread_attr_t **attr); |
f0c29ed7 MD |
171 | |
172 | /* | |
14044b37 | 173 | * cds_lfht_count_nodes - count the number of nodes in the hash table. |
7f52427b MD |
174 | * @ht: the hash table. |
175 | * @split_count_before: Sample the node count split-counter before traversal. | |
176 | * @count: Traverse the hash table, count the number of nodes observed. | |
177 | * @removed: Number of logically removed nodes observed during traversal. | |
178 | * @split_count_after: Sample the node count split-counter after traversal. | |
0422d92c | 179 | * |
f0c29ed7 | 180 | * Call with rcu_read_lock held. |
3df0c49c | 181 | * Threads calling this API need to be registered RCU read-side threads. |
f0c29ed7 | 182 | */ |
14044b37 | 183 | void cds_lfht_count_nodes(struct cds_lfht *ht, |
7f52427b | 184 | long *split_count_before, |
adc0de68 | 185 | unsigned long *count, |
973e5e1b | 186 | unsigned long *removed, |
7f52427b | 187 | long *split_count_after); |
ab7d5fc6 | 188 | |
f0c29ed7 | 189 | /* |
14044b37 | 190 | * cds_lfht_lookup - lookup a node by key. |
0422d92c | 191 | * @ht: the hash table. |
0422d92c | 192 | * @hash: the key hash. |
6f554439 LJ |
193 | * @match: the key match function. |
194 | * @key: the current node key. | |
0422d92c | 195 | * @iter: Node, if found (output). *iter->node set to NULL if not found. |
f0c29ed7 | 196 | * |
f0c29ed7 | 197 | * Call with rcu_read_lock held. |
3df0c49c | 198 | * Threads calling this API need to be registered RCU read-side threads. |
f0c29ed7 | 199 | */ |
6f554439 | 200 | void cds_lfht_lookup(struct cds_lfht *ht, unsigned long hash, |
996ff57c | 201 | cds_lfht_match_fct match, const void *key, |
6f554439 | 202 | struct cds_lfht_iter *iter); |
ab7d5fc6 | 203 | |
f0c29ed7 | 204 | /* |
3883c0e5 | 205 | * cds_lfht_next_duplicate - get the next item with same key (after a lookup). |
0422d92c MD |
206 | * @ht: the hash table. |
207 | * @match: the key match function. | |
04db56f8 | 208 | * @key: the current node key. |
0422d92c | 209 | * @iter: Node, if found (output). *iter->node set to NULL if not found. |
f0c29ed7 | 210 | * |
adc0de68 MD |
211 | * Uses an iterator initialized by a lookup. |
212 | * Sets *iter-node to the following node with same key. | |
213 | * Sets *iter->node to NULL if no following node exists with same key. | |
214 | * RCU read-side lock must be held across cds_lfht_lookup and | |
215 | * cds_lfht_next calls, and also between cds_lfht_next calls using the | |
216 | * node returned by a previous cds_lfht_next. | |
217 | * Call with rcu_read_lock held. | |
3df0c49c | 218 | * Threads calling this API need to be registered RCU read-side threads. |
f0c29ed7 | 219 | */ |
0422d92c | 220 | void cds_lfht_next_duplicate(struct cds_lfht *ht, |
996ff57c | 221 | cds_lfht_match_fct match, const void *key, |
04db56f8 | 222 | struct cds_lfht_iter *iter); |
4e9b9fbf MD |
223 | |
224 | /* | |
225 | * cds_lfht_first - get the first node in the table. | |
0422d92c MD |
226 | * @ht: the hash table. |
227 | * @iter: First node, if exists (output). *iter->node set to NULL if not found. | |
4e9b9fbf MD |
228 | * |
229 | * Output in "*iter". *iter->node set to NULL if table is empty. | |
230 | * Call with rcu_read_lock held. | |
3df0c49c | 231 | * Threads calling this API need to be registered RCU read-side threads. |
4e9b9fbf MD |
232 | */ |
233 | void cds_lfht_first(struct cds_lfht *ht, struct cds_lfht_iter *iter); | |
234 | ||
235 | /* | |
236 | * cds_lfht_next - get the next node in the table. | |
0422d92c MD |
237 | * @ht: the hash table. |
238 | * @iter: Next node, if exists (output). *iter->node set to NULL if not found. | |
4e9b9fbf MD |
239 | * |
240 | * Input/Output in "*iter". *iter->node set to NULL if *iter was | |
241 | * pointing to the last table node. | |
242 | * Call with rcu_read_lock held. | |
3df0c49c | 243 | * Threads calling this API need to be registered RCU read-side threads. |
4e9b9fbf MD |
244 | */ |
245 | void cds_lfht_next(struct cds_lfht *ht, struct cds_lfht_iter *iter); | |
ab7d5fc6 | 246 | |
18117871 | 247 | /* |
14044b37 | 248 | * cds_lfht_add - add a node to the hash table. |
0422d92c MD |
249 | * @ht: the hash table. |
250 | * @hash: the key hash. | |
251 | * @node: the node to add. | |
f0c29ed7 | 252 | * |
48ed1c18 | 253 | * This function supports adding redundant keys into the table. |
3df0c49c MD |
254 | * Call with rcu_read_lock held. |
255 | * Threads calling this API need to be registered RCU read-side threads. | |
f0c29ed7 | 256 | */ |
0422d92c MD |
257 | void cds_lfht_add(struct cds_lfht *ht, unsigned long hash, |
258 | struct cds_lfht_node *node); | |
f0c29ed7 MD |
259 | |
260 | /* | |
14044b37 | 261 | * cds_lfht_add_unique - add a node to hash table, if key is not present. |
0422d92c | 262 | * @ht: the hash table. |
6f554439 | 263 | * @hash: the node's hash. |
0422d92c | 264 | * @match: the key match function. |
04db56f8 | 265 | * @key: the node's key. |
0422d92c | 266 | * @node: the node to try adding. |
f0c29ed7 | 267 | * |
6878c72b | 268 | * Return the node added upon success. |
860d07e8 MD |
269 | * Return the unique node already present upon failure. If |
270 | * cds_lfht_add_unique fails, the node passed as parameter should be | |
271 | * freed by the caller. | |
f0c29ed7 | 272 | * Call with rcu_read_lock held. |
3df0c49c | 273 | * Threads calling this API need to be registered RCU read-side threads. |
48ed1c18 MD |
274 | * |
275 | * The semantic of this function is that if only this function is used | |
276 | * to add keys into the table, no duplicated keys should ever be | |
277 | * observable in the table. The same guarantee apply for combination of | |
9357c415 | 278 | * add_unique and add_replace (see below). |
18117871 | 279 | */ |
adc0de68 | 280 | struct cds_lfht_node *cds_lfht_add_unique(struct cds_lfht *ht, |
6f554439 | 281 | unsigned long hash, |
0422d92c | 282 | cds_lfht_match_fct match, |
996ff57c | 283 | const void *key, |
adc0de68 | 284 | struct cds_lfht_node *node); |
3eca1b8c | 285 | |
48ed1c18 | 286 | /* |
9357c415 | 287 | * cds_lfht_add_replace - replace or add a node within hash table. |
0422d92c | 288 | * @ht: the hash table. |
6f554439 | 289 | * @hash: the node's hash. |
0422d92c | 290 | * @match: the key match function. |
04db56f8 | 291 | * @key: the node's key. |
0422d92c | 292 | * @node: the node to add. |
48ed1c18 MD |
293 | * |
294 | * Return the node replaced upon success. If no node matching the key | |
295 | * was present, return NULL, which also means the operation succeeded. | |
296 | * This replacement operation should never fail. | |
297 | * Call with rcu_read_lock held. | |
3df0c49c | 298 | * Threads calling this API need to be registered RCU read-side threads. |
48ed1c18 MD |
299 | * After successful replacement, a grace period must be waited for before |
300 | * freeing the memory reserved for the returned node. | |
301 | * | |
302 | * The semantic of replacement vs lookups is the following: if lookups | |
9357c415 MD |
303 | * are performed between a key unique insertion and its removal, we |
304 | * guarantee that the lookups and get next will always find exactly one | |
305 | * instance of the key if it is replaced concurrently with the lookups. | |
48ed1c18 MD |
306 | * |
307 | * Providing this semantic allows us to ensure that replacement-only | |
308 | * schemes will never generate duplicated keys. It also allows us to | |
9357c415 | 309 | * guarantee that a combination of add_replace and add_unique updates |
48ed1c18 MD |
310 | * will never generate duplicated keys. |
311 | */ | |
9357c415 | 312 | struct cds_lfht_node *cds_lfht_add_replace(struct cds_lfht *ht, |
6f554439 | 313 | unsigned long hash, |
0422d92c | 314 | cds_lfht_match_fct match, |
996ff57c | 315 | const void *key, |
adc0de68 | 316 | struct cds_lfht_node *node); |
48ed1c18 | 317 | |
f0c29ed7 | 318 | /* |
9357c415 | 319 | * cds_lfht_replace - replace a node pointer to by iter within hash table. |
0422d92c MD |
320 | * @ht: the hash table. |
321 | * @old_iter: the iterator position of the node to replace. | |
322 | * @now_node: the new node to try using for replacement. | |
f0c29ed7 | 323 | * |
9357c415 MD |
324 | * Return 0 if replacement is successful, negative value otherwise. |
325 | * Replacing a NULL old node or an already removed node will fail with a | |
326 | * negative value. | |
327 | * Old node can be looked up with cds_lfht_lookup and cds_lfht_next. | |
328 | * RCU read-side lock must be held between lookup and replacement. | |
329 | * Call with rcu_read_lock held. | |
3df0c49c | 330 | * Threads calling this API need to be registered RCU read-side threads. |
9357c415 MD |
331 | * After successful replacement, a grace period must be waited for before |
332 | * freeing the memory reserved for the old node (which can be accessed | |
333 | * with cds_lfht_iter_get_node). | |
334 | * | |
335 | * The semantic of replacement vs lookups is the following: if lookups | |
336 | * are performed between a key unique insertion and its removal, we | |
337 | * guarantee that the lookups and get next will always find exactly one | |
338 | * instance of the key if it is replaced concurrently with the lookups. | |
339 | * | |
340 | * Providing this semantic allows us to ensure that replacement-only | |
341 | * schemes will never generate duplicated keys. It also allows us to | |
342 | * guarantee that a combination of add_replace and add_unique updates | |
343 | * will never generate duplicated keys. | |
344 | */ | |
345 | int cds_lfht_replace(struct cds_lfht *ht, struct cds_lfht_iter *old_iter, | |
346 | struct cds_lfht_node *new_node); | |
347 | ||
348 | /* | |
349 | * cds_lfht_del - remove node pointed to by iterator from hash table. | |
0422d92c MD |
350 | * @ht: the hash table. |
351 | * @iter: the iterator position of the node to delete. | |
9357c415 MD |
352 | * |
353 | * Return 0 if the node is successfully removed, negative value | |
354 | * otherwise. | |
355 | * Replacing a NULL node or an already removed node will fail with a | |
356 | * negative value. | |
357 | * Node can be looked up with cds_lfht_lookup and cds_lfht_next. | |
358 | * cds_lfht_iter_get_node. | |
359 | * RCU read-side lock must be held between lookup and removal. | |
f0c29ed7 | 360 | * Call with rcu_read_lock held. |
3df0c49c | 361 | * Threads calling this API need to be registered RCU read-side threads. |
48ed1c18 | 362 | * After successful removal, a grace period must be waited for before |
9357c415 MD |
363 | * freeing the memory reserved for old node (which can be accessed with |
364 | * cds_lfht_iter_get_node). | |
f0c29ed7 | 365 | */ |
9357c415 | 366 | int cds_lfht_del(struct cds_lfht *ht, struct cds_lfht_iter *iter); |
ab7d5fc6 | 367 | |
f0c29ed7 | 368 | /* |
14044b37 | 369 | * cds_lfht_resize - Force a hash table resize |
0422d92c | 370 | * @ht: the hash table. |
1475579c | 371 | * @new_size: update to this hash table size. |
3df0c49c MD |
372 | * |
373 | * Threads calling this API need to be registered RCU read-side threads. | |
f0c29ed7 | 374 | */ |
1475579c | 375 | void cds_lfht_resize(struct cds_lfht *ht, unsigned long new_size); |
464a1ec9 | 376 | |
6d320126 MD |
377 | /* |
378 | * Note: cds_lfht_for_each are safe for element removal during | |
379 | * iteration. | |
380 | */ | |
381 | #define cds_lfht_for_each(ht, iter, node) \ | |
382 | for (cds_lfht_first(ht, iter), \ | |
383 | node = cds_lfht_iter_get_node(iter); \ | |
384 | node != NULL; \ | |
385 | cds_lfht_next(ht, iter), \ | |
386 | node = cds_lfht_iter_get_node(iter)) | |
387 | ||
6f554439 LJ |
388 | #define cds_lfht_for_each_duplicate(ht, hash, match, key, iter, node) \ |
389 | for (cds_lfht_lookup(ht, hash, match, key, iter), \ | |
6d320126 MD |
390 | node = cds_lfht_iter_get_node(iter); \ |
391 | node != NULL; \ | |
392 | cds_lfht_next_duplicate(ht, match, key, iter), \ | |
393 | node = cds_lfht_iter_get_node(iter)) | |
394 | ||
395 | #define cds_lfht_for_each_entry(ht, iter, pos, member) \ | |
396 | for (cds_lfht_first(ht, iter), \ | |
397 | pos = caa_container_of(cds_lfht_iter_get_node(iter), \ | |
398 | typeof(*(pos)), member); \ | |
0b0627bf | 399 | &(pos)->member != NULL; \ |
6d320126 MD |
400 | cds_lfht_next(ht, iter), \ |
401 | pos = caa_container_of(cds_lfht_iter_get_node(iter), \ | |
402 | typeof(*(pos)), member)) | |
403 | ||
6f554439 | 404 | #define cds_lfht_for_each_entry_duplicate(ht, hash, match, key, \ |
6d320126 | 405 | iter, pos, member) \ |
6f554439 | 406 | for (cds_lfht_lookup(ht, hash, match, key, iter), \ |
6d320126 MD |
407 | pos = caa_container_of(cds_lfht_iter_get_node(iter), \ |
408 | typeof(*(pos)), member); \ | |
0b0627bf | 409 | &(pos)->member != NULL; \ |
6d320126 MD |
410 | cds_lfht_next_duplicate(ht, match, key, iter), \ |
411 | pos = caa_container_of(cds_lfht_iter_get_node(iter), \ | |
412 | typeof(*(pos)), member)) | |
413 | ||
01389791 MD |
414 | #ifdef __cplusplus |
415 | } | |
416 | #endif | |
417 | ||
a42cc659 | 418 | #endif /* _URCU_RCULFHASH_H */ |