/* 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 */