1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
4  */
5 
6 #ifndef _WG_PEERLOOKUP_H
7 #define _WG_PEERLOOKUP_H
8 
9 #include "messages.h"
10 
11 #include <linux/hashtable.h>
12 #include <linux/mutex.h>
13 #include <linux/siphash.h>
14 
15 struct wg_peer;
16 
17 struct pubkey_hashtable {
18 	/* TODO: move to rhashtable */
19 	DECLARE_HASHTABLE(hashtable, 11);
20 	siphash_key_t key;
21 	struct mutex lock;
22 };
23 
24 struct pubkey_hashtable *wg_pubkey_hashtable_alloc(void);
25 void wg_pubkey_hashtable_add(struct pubkey_hashtable *table,
26 			     struct wg_peer *peer);
27 void wg_pubkey_hashtable_remove(struct pubkey_hashtable *table,
28 				struct wg_peer *peer);
29 struct wg_peer *
30 wg_pubkey_hashtable_lookup(struct pubkey_hashtable *table,
31 			   const u8 pubkey[NOISE_PUBLIC_KEY_LEN]);
32 
33 struct index_hashtable {
34 	/* TODO: move to rhashtable */
35 	DECLARE_HASHTABLE(hashtable, 13);
36 	spinlock_t lock;
37 };
38 
39 enum index_hashtable_type {
40 	INDEX_HASHTABLE_HANDSHAKE = 1U << 0,
41 	INDEX_HASHTABLE_KEYPAIR = 1U << 1
42 };
43 
44 struct index_hashtable_entry {
45 	struct wg_peer *peer;
46 	struct hlist_node index_hash;
47 	enum index_hashtable_type type;
48 	__le32 index;
49 };
50 
51 struct index_hashtable *wg_index_hashtable_alloc(void);
52 __le32 wg_index_hashtable_insert(struct index_hashtable *table,
53 				 struct index_hashtable_entry *entry);
54 bool wg_index_hashtable_replace(struct index_hashtable *table,
55 				struct index_hashtable_entry *old,
56 				struct index_hashtable_entry *new);
57 void wg_index_hashtable_remove(struct index_hashtable *table,
58 			       struct index_hashtable_entry *entry);
59 struct index_hashtable_entry *
60 wg_index_hashtable_lookup(struct index_hashtable *table,
61 			  const enum index_hashtable_type type_mask,
62 			  const __le32 index, struct wg_peer **peer);
63 
64 #endif /* _WG_PEERLOOKUP_H */
65