1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun /* Copyright (C) 2006-2020 B.A.T.M.A.N. contributors:
3*4882a593Smuzhiyun *
4*4882a593Smuzhiyun * Simon Wunderlich, Marek Lindner
5*4882a593Smuzhiyun */
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun #include "hash.h"
8*4882a593Smuzhiyun #include "main.h"
9*4882a593Smuzhiyun
10*4882a593Smuzhiyun #include <linux/gfp.h>
11*4882a593Smuzhiyun #include <linux/lockdep.h>
12*4882a593Smuzhiyun #include <linux/slab.h>
13*4882a593Smuzhiyun
14*4882a593Smuzhiyun /* clears the hash */
batadv_hash_init(struct batadv_hashtable * hash)15*4882a593Smuzhiyun static void batadv_hash_init(struct batadv_hashtable *hash)
16*4882a593Smuzhiyun {
17*4882a593Smuzhiyun u32 i;
18*4882a593Smuzhiyun
19*4882a593Smuzhiyun for (i = 0; i < hash->size; i++) {
20*4882a593Smuzhiyun INIT_HLIST_HEAD(&hash->table[i]);
21*4882a593Smuzhiyun spin_lock_init(&hash->list_locks[i]);
22*4882a593Smuzhiyun }
23*4882a593Smuzhiyun
24*4882a593Smuzhiyun atomic_set(&hash->generation, 0);
25*4882a593Smuzhiyun }
26*4882a593Smuzhiyun
27*4882a593Smuzhiyun /**
28*4882a593Smuzhiyun * batadv_hash_destroy() - Free only the hashtable and the hash itself
29*4882a593Smuzhiyun * @hash: hash object to destroy
30*4882a593Smuzhiyun */
batadv_hash_destroy(struct batadv_hashtable * hash)31*4882a593Smuzhiyun void batadv_hash_destroy(struct batadv_hashtable *hash)
32*4882a593Smuzhiyun {
33*4882a593Smuzhiyun kfree(hash->list_locks);
34*4882a593Smuzhiyun kfree(hash->table);
35*4882a593Smuzhiyun kfree(hash);
36*4882a593Smuzhiyun }
37*4882a593Smuzhiyun
38*4882a593Smuzhiyun /**
39*4882a593Smuzhiyun * batadv_hash_new() - Allocates and clears the hashtable
40*4882a593Smuzhiyun * @size: number of hash buckets to allocate
41*4882a593Smuzhiyun *
42*4882a593Smuzhiyun * Return: newly allocated hashtable, NULL on errors
43*4882a593Smuzhiyun */
batadv_hash_new(u32 size)44*4882a593Smuzhiyun struct batadv_hashtable *batadv_hash_new(u32 size)
45*4882a593Smuzhiyun {
46*4882a593Smuzhiyun struct batadv_hashtable *hash;
47*4882a593Smuzhiyun
48*4882a593Smuzhiyun hash = kmalloc(sizeof(*hash), GFP_ATOMIC);
49*4882a593Smuzhiyun if (!hash)
50*4882a593Smuzhiyun return NULL;
51*4882a593Smuzhiyun
52*4882a593Smuzhiyun hash->table = kmalloc_array(size, sizeof(*hash->table), GFP_ATOMIC);
53*4882a593Smuzhiyun if (!hash->table)
54*4882a593Smuzhiyun goto free_hash;
55*4882a593Smuzhiyun
56*4882a593Smuzhiyun hash->list_locks = kmalloc_array(size, sizeof(*hash->list_locks),
57*4882a593Smuzhiyun GFP_ATOMIC);
58*4882a593Smuzhiyun if (!hash->list_locks)
59*4882a593Smuzhiyun goto free_table;
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun hash->size = size;
62*4882a593Smuzhiyun batadv_hash_init(hash);
63*4882a593Smuzhiyun return hash;
64*4882a593Smuzhiyun
65*4882a593Smuzhiyun free_table:
66*4882a593Smuzhiyun kfree(hash->table);
67*4882a593Smuzhiyun free_hash:
68*4882a593Smuzhiyun kfree(hash);
69*4882a593Smuzhiyun return NULL;
70*4882a593Smuzhiyun }
71*4882a593Smuzhiyun
72*4882a593Smuzhiyun /**
73*4882a593Smuzhiyun * batadv_hash_set_lock_class() - Set specific lockdep class for hash spinlocks
74*4882a593Smuzhiyun * @hash: hash object to modify
75*4882a593Smuzhiyun * @key: lockdep class key address
76*4882a593Smuzhiyun */
batadv_hash_set_lock_class(struct batadv_hashtable * hash,struct lock_class_key * key)77*4882a593Smuzhiyun void batadv_hash_set_lock_class(struct batadv_hashtable *hash,
78*4882a593Smuzhiyun struct lock_class_key *key)
79*4882a593Smuzhiyun {
80*4882a593Smuzhiyun u32 i;
81*4882a593Smuzhiyun
82*4882a593Smuzhiyun for (i = 0; i < hash->size; i++)
83*4882a593Smuzhiyun lockdep_set_class(&hash->list_locks[i], key);
84*4882a593Smuzhiyun }
85