microblaze: use asm-generic for seccomp.h
[linux-2.6-block.git] / lib / rhashtable.c
CommitLineData
7e1e7763
TG
1/*
2 * Resizable, Scalable, Concurrent Hash Table
3 *
02fd97c3 4 * Copyright (c) 2015 Herbert Xu <herbert@gondor.apana.org.au>
a5ec68e3 5 * Copyright (c) 2014-2015 Thomas Graf <tgraf@suug.ch>
7e1e7763
TG
6 * Copyright (c) 2008-2014 Patrick McHardy <kaber@trash.net>
7 *
7e1e7763 8 * Code partially derived from nft_hash
02fd97c3
HX
9 * Rewritten with rehash code from br_multicast plus single list
10 * pointer as suggested by Josh Triplett
7e1e7763
TG
11 *
12 * This program is free software; you can redistribute it and/or modify
13 * it under the terms of the GNU General Public License version 2 as
14 * published by the Free Software Foundation.
15 */
16
17#include <linux/kernel.h>
18#include <linux/init.h>
19#include <linux/log2.h>
5beb5c90 20#include <linux/sched.h>
7e1e7763
TG
21#include <linux/slab.h>
22#include <linux/vmalloc.h>
23#include <linux/mm.h>
87545899 24#include <linux/jhash.h>
7e1e7763
TG
25#include <linux/random.h>
26#include <linux/rhashtable.h>
61d7b097 27#include <linux/err.h>
7e1e7763
TG
28
29#define HASH_DEFAULT_SIZE 64UL
c2e213cf 30#define HASH_MIN_SIZE 4U
97defe1e
TG
31#define BUCKET_LOCKS_PER_CPU 128UL
32
988dfbd7 33static u32 head_hashfn(struct rhashtable *ht,
8d24c0b4
TG
34 const struct bucket_table *tbl,
35 const struct rhash_head *he)
7e1e7763 36{
02fd97c3 37 return rht_head_hashfn(ht, tbl, he, ht->p);
7e1e7763
TG
38}
39
a03eaec0 40#ifdef CONFIG_PROVE_LOCKING
a03eaec0 41#define ASSERT_RHT_MUTEX(HT) BUG_ON(!lockdep_rht_mutex_is_held(HT))
a03eaec0
TG
42
43int lockdep_rht_mutex_is_held(struct rhashtable *ht)
44{
45 return (debug_locks) ? lockdep_is_held(&ht->mutex) : 1;
46}
47EXPORT_SYMBOL_GPL(lockdep_rht_mutex_is_held);
48
49int lockdep_rht_bucket_is_held(const struct bucket_table *tbl, u32 hash)
50{
02fd97c3 51 spinlock_t *lock = rht_bucket_lock(tbl, hash);
a03eaec0
TG
52
53 return (debug_locks) ? lockdep_is_held(lock) : 1;
54}
55EXPORT_SYMBOL_GPL(lockdep_rht_bucket_is_held);
56#else
57#define ASSERT_RHT_MUTEX(HT)
a03eaec0
TG
58#endif
59
60
b9ecfdaa
HX
61static int alloc_bucket_locks(struct rhashtable *ht, struct bucket_table *tbl,
62 gfp_t gfp)
97defe1e
TG
63{
64 unsigned int i, size;
65#if defined(CONFIG_PROVE_LOCKING)
66 unsigned int nr_pcpus = 2;
67#else
68 unsigned int nr_pcpus = num_possible_cpus();
69#endif
70
71 nr_pcpus = min_t(unsigned int, nr_pcpus, 32UL);
72 size = roundup_pow_of_two(nr_pcpus * ht->p.locks_mul);
73
a5ec68e3
TG
74 /* Never allocate more than 0.5 locks per bucket */
75 size = min_t(unsigned int, size, tbl->size >> 1);
97defe1e
TG
76
77 if (sizeof(spinlock_t) != 0) {
78#ifdef CONFIG_NUMA
b9ecfdaa
HX
79 if (size * sizeof(spinlock_t) > PAGE_SIZE &&
80 gfp == GFP_KERNEL)
97defe1e
TG
81 tbl->locks = vmalloc(size * sizeof(spinlock_t));
82 else
83#endif
84 tbl->locks = kmalloc_array(size, sizeof(spinlock_t),
b9ecfdaa 85 gfp);
97defe1e
TG
86 if (!tbl->locks)
87 return -ENOMEM;
88 for (i = 0; i < size; i++)
89 spin_lock_init(&tbl->locks[i]);
90 }
91 tbl->locks_mask = size - 1;
92
93 return 0;
94}
95
96static void bucket_table_free(const struct bucket_table *tbl)
97{
98 if (tbl)
99 kvfree(tbl->locks);
100
101 kvfree(tbl);
102}
103
9d901bc0
HX
104static void bucket_table_free_rcu(struct rcu_head *head)
105{
106 bucket_table_free(container_of(head, struct bucket_table, rcu));
107}
108
97defe1e 109static struct bucket_table *bucket_table_alloc(struct rhashtable *ht,
b9ecfdaa
HX
110 size_t nbuckets,
111 gfp_t gfp)
7e1e7763 112{
eb6d1abf 113 struct bucket_table *tbl = NULL;
7e1e7763 114 size_t size;
f89bd6f8 115 int i;
7e1e7763
TG
116
117 size = sizeof(*tbl) + nbuckets * sizeof(tbl->buckets[0]);
b9ecfdaa
HX
118 if (size <= (PAGE_SIZE << PAGE_ALLOC_COSTLY_ORDER) ||
119 gfp != GFP_KERNEL)
120 tbl = kzalloc(size, gfp | __GFP_NOWARN | __GFP_NORETRY);
121 if (tbl == NULL && gfp == GFP_KERNEL)
7e1e7763 122 tbl = vzalloc(size);
7e1e7763
TG
123 if (tbl == NULL)
124 return NULL;
125
126 tbl->size = nbuckets;
127
b9ecfdaa 128 if (alloc_bucket_locks(ht, tbl, gfp) < 0) {
97defe1e
TG
129 bucket_table_free(tbl);
130 return NULL;
131 }
7e1e7763 132
eddee5ba
HX
133 INIT_LIST_HEAD(&tbl->walkers);
134
5269b53d
HX
135 get_random_bytes(&tbl->hash_rnd, sizeof(tbl->hash_rnd));
136
f89bd6f8
TG
137 for (i = 0; i < nbuckets; i++)
138 INIT_RHT_NULLS_HEAD(tbl->buckets[i], ht, i);
139
97defe1e 140 return tbl;
7e1e7763
TG
141}
142
b824478b
HX
143static struct bucket_table *rhashtable_last_table(struct rhashtable *ht,
144 struct bucket_table *tbl)
145{
146 struct bucket_table *new_tbl;
147
148 do {
149 new_tbl = tbl;
150 tbl = rht_dereference_rcu(tbl->future_tbl, ht);
151 } while (tbl);
152
153 return new_tbl;
154}
155
299e5c32 156static int rhashtable_rehash_one(struct rhashtable *ht, unsigned int old_hash)
a5ec68e3 157{
aa34a6cb 158 struct bucket_table *old_tbl = rht_dereference(ht->tbl, ht);
b824478b
HX
159 struct bucket_table *new_tbl = rhashtable_last_table(ht,
160 rht_dereference_rcu(old_tbl->future_tbl, ht));
aa34a6cb
HX
161 struct rhash_head __rcu **pprev = &old_tbl->buckets[old_hash];
162 int err = -ENOENT;
163 struct rhash_head *head, *next, *entry;
164 spinlock_t *new_bucket_lock;
299e5c32 165 unsigned int new_hash;
aa34a6cb
HX
166
167 rht_for_each(entry, old_tbl, old_hash) {
168 err = 0;
169 next = rht_dereference_bucket(entry->next, old_tbl, old_hash);
170
171 if (rht_is_a_nulls(next))
172 break;
a5ec68e3 173
aa34a6cb
HX
174 pprev = &entry->next;
175 }
a5ec68e3 176
aa34a6cb
HX
177 if (err)
178 goto out;
97defe1e 179
aa34a6cb 180 new_hash = head_hashfn(ht, new_tbl, entry);
7e1e7763 181
02fd97c3 182 new_bucket_lock = rht_bucket_lock(new_tbl, new_hash);
7e1e7763 183
8f2484bd 184 spin_lock_nested(new_bucket_lock, SINGLE_DEPTH_NESTING);
aa34a6cb
HX
185 head = rht_dereference_bucket(new_tbl->buckets[new_hash],
186 new_tbl, new_hash);
97defe1e 187
aa34a6cb
HX
188 if (rht_is_a_nulls(head))
189 INIT_RHT_NULLS_HEAD(entry->next, ht, new_hash);
190 else
191 RCU_INIT_POINTER(entry->next, head);
a5ec68e3 192
aa34a6cb
HX
193 rcu_assign_pointer(new_tbl->buckets[new_hash], entry);
194 spin_unlock(new_bucket_lock);
97defe1e 195
aa34a6cb 196 rcu_assign_pointer(*pprev, next);
7e1e7763 197
aa34a6cb
HX
198out:
199 return err;
200}
97defe1e 201
299e5c32
TG
202static void rhashtable_rehash_chain(struct rhashtable *ht,
203 unsigned int old_hash)
aa34a6cb
HX
204{
205 struct bucket_table *old_tbl = rht_dereference(ht->tbl, ht);
206 spinlock_t *old_bucket_lock;
207
02fd97c3 208 old_bucket_lock = rht_bucket_lock(old_tbl, old_hash);
a5ec68e3 209
aa34a6cb
HX
210 spin_lock_bh(old_bucket_lock);
211 while (!rhashtable_rehash_one(ht, old_hash))
212 ;
63d512d0 213 old_tbl->rehash++;
aa34a6cb 214 spin_unlock_bh(old_bucket_lock);
97defe1e
TG
215}
216
b824478b
HX
217static int rhashtable_rehash_attach(struct rhashtable *ht,
218 struct bucket_table *old_tbl,
219 struct bucket_table *new_tbl)
97defe1e 220{
b824478b
HX
221 /* Protect future_tbl using the first bucket lock. */
222 spin_lock_bh(old_tbl->locks);
223
224 /* Did somebody beat us to it? */
225 if (rcu_access_pointer(old_tbl->future_tbl)) {
226 spin_unlock_bh(old_tbl->locks);
227 return -EEXIST;
228 }
7cd10db8 229
aa34a6cb
HX
230 /* Make insertions go into the new, empty table right away. Deletions
231 * and lookups will be attempted in both tables until we synchronize.
aa34a6cb 232 */
c4db8848 233 rcu_assign_pointer(old_tbl->future_tbl, new_tbl);
aa34a6cb 234
9497df88
HX
235 /* Ensure the new table is visible to readers. */
236 smp_wmb();
237
b824478b
HX
238 spin_unlock_bh(old_tbl->locks);
239
240 return 0;
241}
242
243static int rhashtable_rehash_table(struct rhashtable *ht)
244{
245 struct bucket_table *old_tbl = rht_dereference(ht->tbl, ht);
246 struct bucket_table *new_tbl;
247 struct rhashtable_walker *walker;
299e5c32 248 unsigned int old_hash;
b824478b
HX
249
250 new_tbl = rht_dereference(old_tbl->future_tbl, ht);
251 if (!new_tbl)
252 return 0;
253
aa34a6cb
HX
254 for (old_hash = 0; old_hash < old_tbl->size; old_hash++)
255 rhashtable_rehash_chain(ht, old_hash);
256
257 /* Publish the new table pointer. */
258 rcu_assign_pointer(ht->tbl, new_tbl);
259
ba7c95ea 260 spin_lock(&ht->lock);
eddee5ba
HX
261 list_for_each_entry(walker, &old_tbl->walkers, list)
262 walker->tbl = NULL;
ba7c95ea 263 spin_unlock(&ht->lock);
eddee5ba 264
aa34a6cb
HX
265 /* Wait for readers. All new readers will see the new
266 * table, and thus no references to the old table will
267 * remain.
268 */
9d901bc0 269 call_rcu(&old_tbl->rcu, bucket_table_free_rcu);
b824478b
HX
270
271 return rht_dereference(new_tbl->future_tbl, ht) ? -EAGAIN : 0;
7e1e7763
TG
272}
273
274/**
275 * rhashtable_expand - Expand hash table while allowing concurrent lookups
276 * @ht: the hash table to expand
7e1e7763 277 *
aa34a6cb 278 * A secondary bucket array is allocated and the hash entries are migrated.
7e1e7763
TG
279 *
280 * This function may only be called in a context where it is safe to call
281 * synchronize_rcu(), e.g. not within a rcu_read_lock() section.
282 *
97defe1e
TG
283 * The caller must ensure that no concurrent resizing occurs by holding
284 * ht->mutex.
285 *
286 * It is valid to have concurrent insertions and deletions protected by per
287 * bucket locks or concurrent RCU protected lookups and traversals.
7e1e7763 288 */
b824478b 289static int rhashtable_expand(struct rhashtable *ht)
7e1e7763
TG
290{
291 struct bucket_table *new_tbl, *old_tbl = rht_dereference(ht->tbl, ht);
b824478b 292 int err;
7e1e7763
TG
293
294 ASSERT_RHT_MUTEX(ht);
295
b824478b
HX
296 old_tbl = rhashtable_last_table(ht, old_tbl);
297
b9ecfdaa 298 new_tbl = bucket_table_alloc(ht, old_tbl->size * 2, GFP_KERNEL);
7e1e7763
TG
299 if (new_tbl == NULL)
300 return -ENOMEM;
301
b824478b
HX
302 err = rhashtable_rehash_attach(ht, old_tbl, new_tbl);
303 if (err)
304 bucket_table_free(new_tbl);
305
306 return err;
7e1e7763 307}
7e1e7763
TG
308
309/**
310 * rhashtable_shrink - Shrink hash table while allowing concurrent lookups
311 * @ht: the hash table to shrink
7e1e7763 312 *
18093d1c
HX
313 * This function shrinks the hash table to fit, i.e., the smallest
314 * size would not cause it to expand right away automatically.
7e1e7763 315 *
97defe1e
TG
316 * The caller must ensure that no concurrent resizing occurs by holding
317 * ht->mutex.
318 *
7e1e7763
TG
319 * The caller must ensure that no concurrent table mutations take place.
320 * It is however valid to have concurrent lookups if they are RCU protected.
97defe1e
TG
321 *
322 * It is valid to have concurrent insertions and deletions protected by per
323 * bucket locks or concurrent RCU protected lookups and traversals.
7e1e7763 324 */
b824478b 325static int rhashtable_shrink(struct rhashtable *ht)
7e1e7763 326{
a5b6846f 327 struct bucket_table *new_tbl, *old_tbl = rht_dereference(ht->tbl, ht);
299e5c32 328 unsigned int size;
b824478b 329 int err;
7e1e7763
TG
330
331 ASSERT_RHT_MUTEX(ht);
332
299e5c32 333 size = roundup_pow_of_two(atomic_read(&ht->nelems) * 3 / 2);
18093d1c
HX
334 if (size < ht->p.min_size)
335 size = ht->p.min_size;
336
337 if (old_tbl->size <= size)
338 return 0;
339
b824478b
HX
340 if (rht_dereference(old_tbl->future_tbl, ht))
341 return -EEXIST;
342
b9ecfdaa 343 new_tbl = bucket_table_alloc(ht, size, GFP_KERNEL);
97defe1e 344 if (new_tbl == NULL)
7e1e7763
TG
345 return -ENOMEM;
346
b824478b
HX
347 err = rhashtable_rehash_attach(ht, old_tbl, new_tbl);
348 if (err)
349 bucket_table_free(new_tbl);
350
351 return err;
7e1e7763 352}
7e1e7763 353
97defe1e
TG
354static void rht_deferred_worker(struct work_struct *work)
355{
356 struct rhashtable *ht;
357 struct bucket_table *tbl;
b824478b 358 int err = 0;
97defe1e 359
57699a40 360 ht = container_of(work, struct rhashtable, run_work);
97defe1e 361 mutex_lock(&ht->mutex);
28134a53 362
97defe1e 363 tbl = rht_dereference(ht->tbl, ht);
b824478b 364 tbl = rhashtable_last_table(ht, tbl);
97defe1e 365
a5b6846f 366 if (rht_grow_above_75(ht, tbl))
97defe1e 367 rhashtable_expand(ht);
b5e2c150 368 else if (ht->p.automatic_shrinking && rht_shrink_below_30(ht, tbl))
97defe1e 369 rhashtable_shrink(ht);
b824478b
HX
370
371 err = rhashtable_rehash_table(ht);
372
97defe1e 373 mutex_unlock(&ht->mutex);
b824478b
HX
374
375 if (err)
376 schedule_work(&ht->run_work);
97defe1e
TG
377}
378
ccd57b1b
HX
379static bool rhashtable_check_elasticity(struct rhashtable *ht,
380 struct bucket_table *tbl,
299e5c32 381 unsigned int hash)
ccd57b1b 382{
299e5c32 383 unsigned int elasticity = ht->elasticity;
ccd57b1b
HX
384 struct rhash_head *head;
385
386 rht_for_each(head, tbl, hash)
387 if (!--elasticity)
388 return true;
389
390 return false;
391}
392
393int rhashtable_insert_rehash(struct rhashtable *ht)
394{
395 struct bucket_table *old_tbl;
396 struct bucket_table *new_tbl;
397 struct bucket_table *tbl;
398 unsigned int size;
399 int err;
400
401 old_tbl = rht_dereference_rcu(ht->tbl, ht);
402 tbl = rhashtable_last_table(ht, old_tbl);
403
404 size = tbl->size;
405
406 if (rht_grow_above_75(ht, tbl))
407 size *= 2;
408 /* More than two rehashes (not resizes) detected. */
409 else if (WARN_ON(old_tbl != tbl && old_tbl->size == size))
410 return -EBUSY;
411
412 new_tbl = bucket_table_alloc(ht, size, GFP_ATOMIC);
413 if (new_tbl == NULL)
414 return -ENOMEM;
415
416 err = rhashtable_rehash_attach(ht, tbl, new_tbl);
417 if (err) {
418 bucket_table_free(new_tbl);
419 if (err == -EEXIST)
420 err = 0;
421 } else
422 schedule_work(&ht->run_work);
423
424 return err;
425}
426EXPORT_SYMBOL_GPL(rhashtable_insert_rehash);
427
02fd97c3
HX
428int rhashtable_insert_slow(struct rhashtable *ht, const void *key,
429 struct rhash_head *obj,
430 struct bucket_table *tbl)
431{
432 struct rhash_head *head;
299e5c32 433 unsigned int hash;
ccd57b1b 434 int err;
02fd97c3 435
b824478b 436 tbl = rhashtable_last_table(ht, tbl);
02fd97c3
HX
437 hash = head_hashfn(ht, tbl, obj);
438 spin_lock_nested(rht_bucket_lock(tbl, hash), SINGLE_DEPTH_NESTING);
439
ccd57b1b 440 err = -EEXIST;
02fd97c3
HX
441 if (key && rhashtable_lookup_fast(ht, key, ht->p))
442 goto exit;
443
ccd57b1b
HX
444 err = -EAGAIN;
445 if (rhashtable_check_elasticity(ht, tbl, hash) ||
446 rht_grow_above_100(ht, tbl))
447 goto exit;
448
02fd97c3
HX
449 err = 0;
450
451 head = rht_dereference_bucket(tbl->buckets[hash], tbl, hash);
452
453 RCU_INIT_POINTER(obj->next, head);
454
455 rcu_assign_pointer(tbl->buckets[hash], obj);
456
457 atomic_inc(&ht->nelems);
458
459exit:
460 spin_unlock(rht_bucket_lock(tbl, hash));
461
462 return err;
463}
464EXPORT_SYMBOL_GPL(rhashtable_insert_slow);
465
f2dba9c6
HX
466/**
467 * rhashtable_walk_init - Initialise an iterator
468 * @ht: Table to walk over
469 * @iter: Hash table Iterator
470 *
471 * This function prepares a hash table walk.
472 *
473 * Note that if you restart a walk after rhashtable_walk_stop you
474 * may see the same object twice. Also, you may miss objects if
475 * there are removals in between rhashtable_walk_stop and the next
476 * call to rhashtable_walk_start.
477 *
478 * For a completely stable walk you should construct your own data
479 * structure outside the hash table.
480 *
481 * This function may sleep so you must not call it from interrupt
482 * context or with spin locks held.
483 *
484 * You must call rhashtable_walk_exit if this function returns
485 * successfully.
486 */
487int rhashtable_walk_init(struct rhashtable *ht, struct rhashtable_iter *iter)
488{
489 iter->ht = ht;
490 iter->p = NULL;
491 iter->slot = 0;
492 iter->skip = 0;
493
494 iter->walker = kmalloc(sizeof(*iter->walker), GFP_KERNEL);
495 if (!iter->walker)
496 return -ENOMEM;
497
498 mutex_lock(&ht->mutex);
eddee5ba
HX
499 iter->walker->tbl = rht_dereference(ht->tbl, ht);
500 list_add(&iter->walker->list, &iter->walker->tbl->walkers);
f2dba9c6
HX
501 mutex_unlock(&ht->mutex);
502
503 return 0;
504}
505EXPORT_SYMBOL_GPL(rhashtable_walk_init);
506
507/**
508 * rhashtable_walk_exit - Free an iterator
509 * @iter: Hash table Iterator
510 *
511 * This function frees resources allocated by rhashtable_walk_init.
512 */
513void rhashtable_walk_exit(struct rhashtable_iter *iter)
514{
515 mutex_lock(&iter->ht->mutex);
eddee5ba
HX
516 if (iter->walker->tbl)
517 list_del(&iter->walker->list);
f2dba9c6
HX
518 mutex_unlock(&iter->ht->mutex);
519 kfree(iter->walker);
520}
521EXPORT_SYMBOL_GPL(rhashtable_walk_exit);
522
523/**
524 * rhashtable_walk_start - Start a hash table walk
525 * @iter: Hash table iterator
526 *
527 * Start a hash table walk. Note that we take the RCU lock in all
528 * cases including when we return an error. So you must always call
529 * rhashtable_walk_stop to clean up.
530 *
531 * Returns zero if successful.
532 *
533 * Returns -EAGAIN if resize event occured. Note that the iterator
534 * will rewind back to the beginning and you may use it immediately
535 * by calling rhashtable_walk_next.
536 */
537int rhashtable_walk_start(struct rhashtable_iter *iter)
db4374f4 538 __acquires(RCU)
f2dba9c6 539{
eddee5ba
HX
540 struct rhashtable *ht = iter->ht;
541
542 mutex_lock(&ht->mutex);
543
544 if (iter->walker->tbl)
545 list_del(&iter->walker->list);
546
f2dba9c6
HX
547 rcu_read_lock();
548
eddee5ba
HX
549 mutex_unlock(&ht->mutex);
550
551 if (!iter->walker->tbl) {
552 iter->walker->tbl = rht_dereference_rcu(ht->tbl, ht);
f2dba9c6
HX
553 return -EAGAIN;
554 }
555
556 return 0;
557}
558EXPORT_SYMBOL_GPL(rhashtable_walk_start);
559
560/**
561 * rhashtable_walk_next - Return the next object and advance the iterator
562 * @iter: Hash table iterator
563 *
564 * Note that you must call rhashtable_walk_stop when you are finished
565 * with the walk.
566 *
567 * Returns the next object or NULL when the end of the table is reached.
568 *
569 * Returns -EAGAIN if resize event occured. Note that the iterator
570 * will rewind back to the beginning and you may continue to use it.
571 */
572void *rhashtable_walk_next(struct rhashtable_iter *iter)
573{
eddee5ba 574 struct bucket_table *tbl = iter->walker->tbl;
f2dba9c6
HX
575 struct rhashtable *ht = iter->ht;
576 struct rhash_head *p = iter->p;
577 void *obj = NULL;
578
f2dba9c6
HX
579 if (p) {
580 p = rht_dereference_bucket_rcu(p->next, tbl, iter->slot);
581 goto next;
582 }
583
584 for (; iter->slot < tbl->size; iter->slot++) {
585 int skip = iter->skip;
586
587 rht_for_each_rcu(p, tbl, iter->slot) {
588 if (!skip)
589 break;
590 skip--;
591 }
592
593next:
594 if (!rht_is_a_nulls(p)) {
595 iter->skip++;
596 iter->p = p;
597 obj = rht_obj(ht, p);
598 goto out;
599 }
600
601 iter->skip = 0;
602 }
603
d88252f9
HX
604 /* Ensure we see any new tables. */
605 smp_rmb();
606
c4db8848
HX
607 iter->walker->tbl = rht_dereference_rcu(tbl->future_tbl, ht);
608 if (iter->walker->tbl) {
f2dba9c6
HX
609 iter->slot = 0;
610 iter->skip = 0;
f2dba9c6
HX
611 return ERR_PTR(-EAGAIN);
612 }
613
eddee5ba
HX
614 iter->p = NULL;
615
616out:
617
f2dba9c6
HX
618 return obj;
619}
620EXPORT_SYMBOL_GPL(rhashtable_walk_next);
621
622/**
623 * rhashtable_walk_stop - Finish a hash table walk
624 * @iter: Hash table iterator
625 *
626 * Finish a hash table walk.
627 */
628void rhashtable_walk_stop(struct rhashtable_iter *iter)
db4374f4 629 __releases(RCU)
f2dba9c6 630{
eddee5ba
HX
631 struct rhashtable *ht;
632 struct bucket_table *tbl = iter->walker->tbl;
633
eddee5ba 634 if (!tbl)
963ecbd4 635 goto out;
eddee5ba
HX
636
637 ht = iter->ht;
638
ba7c95ea 639 spin_lock(&ht->lock);
c4db8848 640 if (tbl->rehash < tbl->size)
eddee5ba
HX
641 list_add(&iter->walker->list, &tbl->walkers);
642 else
643 iter->walker->tbl = NULL;
ba7c95ea 644 spin_unlock(&ht->lock);
eddee5ba 645
f2dba9c6 646 iter->p = NULL;
963ecbd4
HX
647
648out:
649 rcu_read_unlock();
f2dba9c6
HX
650}
651EXPORT_SYMBOL_GPL(rhashtable_walk_stop);
652
488fb86e 653static size_t rounded_hashtable_size(const struct rhashtable_params *params)
7e1e7763 654{
94000176 655 return max(roundup_pow_of_two(params->nelem_hint * 4 / 3),
e2e21c1c 656 (unsigned long)params->min_size);
7e1e7763
TG
657}
658
31ccde2d
HX
659static u32 rhashtable_jhash2(const void *key, u32 length, u32 seed)
660{
661 return jhash2(key, length, seed);
662}
663
7e1e7763
TG
664/**
665 * rhashtable_init - initialize a new hash table
666 * @ht: hash table to be initialized
667 * @params: configuration parameters
668 *
669 * Initializes a new hash table based on the provided configuration
670 * parameters. A table can be configured either with a variable or
671 * fixed length key:
672 *
673 * Configuration Example 1: Fixed length keys
674 * struct test_obj {
675 * int key;
676 * void * my_member;
677 * struct rhash_head node;
678 * };
679 *
680 * struct rhashtable_params params = {
681 * .head_offset = offsetof(struct test_obj, node),
682 * .key_offset = offsetof(struct test_obj, key),
683 * .key_len = sizeof(int),
87545899 684 * .hashfn = jhash,
f89bd6f8 685 * .nulls_base = (1U << RHT_BASE_SHIFT),
7e1e7763
TG
686 * };
687 *
688 * Configuration Example 2: Variable length keys
689 * struct test_obj {
690 * [...]
691 * struct rhash_head node;
692 * };
693 *
49f7b33e 694 * u32 my_hash_fn(const void *data, u32 len, u32 seed)
7e1e7763
TG
695 * {
696 * struct test_obj *obj = data;
697 *
698 * return [... hash ...];
699 * }
700 *
701 * struct rhashtable_params params = {
702 * .head_offset = offsetof(struct test_obj, node),
87545899 703 * .hashfn = jhash,
7e1e7763 704 * .obj_hashfn = my_hash_fn,
7e1e7763
TG
705 * };
706 */
488fb86e
HX
707int rhashtable_init(struct rhashtable *ht,
708 const struct rhashtable_params *params)
7e1e7763
TG
709{
710 struct bucket_table *tbl;
711 size_t size;
712
713 size = HASH_DEFAULT_SIZE;
714
31ccde2d 715 if ((!params->key_len && !params->obj_hashfn) ||
02fd97c3 716 (params->obj_hashfn && !params->obj_cmpfn))
7e1e7763
TG
717 return -EINVAL;
718
f89bd6f8
TG
719 if (params->nulls_base && params->nulls_base < (1U << RHT_BASE_SHIFT))
720 return -EINVAL;
721
7e1e7763 722 if (params->nelem_hint)
94000176 723 size = rounded_hashtable_size(params);
7e1e7763 724
97defe1e
TG
725 memset(ht, 0, sizeof(*ht));
726 mutex_init(&ht->mutex);
ba7c95ea 727 spin_lock_init(&ht->lock);
97defe1e
TG
728 memcpy(&ht->p, params, sizeof(*params));
729
a998f712
TG
730 if (params->min_size)
731 ht->p.min_size = roundup_pow_of_two(params->min_size);
732
733 if (params->max_size)
734 ht->p.max_size = rounddown_pow_of_two(params->max_size);
735
488fb86e 736 ht->p.min_size = max(ht->p.min_size, HASH_MIN_SIZE);
a998f712 737
27ed44a5
HX
738 /* The maximum (not average) chain length grows with the
739 * size of the hash table, at a rate of (log N)/(log log N).
740 * The value of 16 is selected so that even if the hash
741 * table grew to 2^32 you would not expect the maximum
742 * chain length to exceed it unless we are under attack
743 * (or extremely unlucky).
744 *
745 * As this limit is only to detect attacks, we don't need
746 * to set it to a lower value as you'd need the chain
747 * length to vastly exceed 16 to have any real effect
748 * on the system.
749 */
ccd57b1b
HX
750 if (!params->insecure_elasticity)
751 ht->elasticity = 16;
752
97defe1e
TG
753 if (params->locks_mul)
754 ht->p.locks_mul = roundup_pow_of_two(params->locks_mul);
755 else
756 ht->p.locks_mul = BUCKET_LOCKS_PER_CPU;
757
31ccde2d
HX
758 ht->key_len = ht->p.key_len;
759 if (!params->hashfn) {
760 ht->p.hashfn = jhash;
761
762 if (!(ht->key_len & (sizeof(u32) - 1))) {
763 ht->key_len /= sizeof(u32);
764 ht->p.hashfn = rhashtable_jhash2;
765 }
766 }
767
b9ecfdaa 768 tbl = bucket_table_alloc(ht, size, GFP_KERNEL);
7e1e7763
TG
769 if (tbl == NULL)
770 return -ENOMEM;
771
545a148e 772 atomic_set(&ht->nelems, 0);
a5b6846f 773
7e1e7763
TG
774 RCU_INIT_POINTER(ht->tbl, tbl);
775
4c4b52d9 776 INIT_WORK(&ht->run_work, rht_deferred_worker);
97defe1e 777
7e1e7763
TG
778 return 0;
779}
780EXPORT_SYMBOL_GPL(rhashtable_init);
781
782/**
6b6f302c 783 * rhashtable_free_and_destroy - free elements and destroy hash table
7e1e7763 784 * @ht: the hash table to destroy
6b6f302c
TG
785 * @free_fn: callback to release resources of element
786 * @arg: pointer passed to free_fn
7e1e7763 787 *
6b6f302c
TG
788 * Stops an eventual async resize. If defined, invokes free_fn for each
789 * element to releasal resources. Please note that RCU protected
790 * readers may still be accessing the elements. Releasing of resources
791 * must occur in a compatible manner. Then frees the bucket array.
792 *
793 * This function will eventually sleep to wait for an async resize
794 * to complete. The caller is responsible that no further write operations
795 * occurs in parallel.
7e1e7763 796 */
6b6f302c
TG
797void rhashtable_free_and_destroy(struct rhashtable *ht,
798 void (*free_fn)(void *ptr, void *arg),
799 void *arg)
7e1e7763 800{
6b6f302c
TG
801 const struct bucket_table *tbl;
802 unsigned int i;
97defe1e 803
4c4b52d9 804 cancel_work_sync(&ht->run_work);
97defe1e 805
57699a40 806 mutex_lock(&ht->mutex);
6b6f302c
TG
807 tbl = rht_dereference(ht->tbl, ht);
808 if (free_fn) {
809 for (i = 0; i < tbl->size; i++) {
810 struct rhash_head *pos, *next;
811
812 for (pos = rht_dereference(tbl->buckets[i], ht),
813 next = !rht_is_a_nulls(pos) ?
814 rht_dereference(pos->next, ht) : NULL;
815 !rht_is_a_nulls(pos);
816 pos = next,
817 next = !rht_is_a_nulls(pos) ?
818 rht_dereference(pos->next, ht) : NULL)
819 free_fn(rht_obj(ht, pos), arg);
820 }
821 }
822
823 bucket_table_free(tbl);
97defe1e 824 mutex_unlock(&ht->mutex);
7e1e7763 825}
6b6f302c
TG
826EXPORT_SYMBOL_GPL(rhashtable_free_and_destroy);
827
828void rhashtable_destroy(struct rhashtable *ht)
829{
830 return rhashtable_free_and_destroy(ht, NULL, NULL);
831}
7e1e7763 832EXPORT_SYMBOL_GPL(rhashtable_destroy);