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