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