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_PEER_H 7 #define _WG_PEER_H 8 9 #include "device.h" 10 #include "noise.h" 11 #include "cookie.h" 12 13 #include <linux/types.h> 14 #include <linux/netfilter.h> 15 #include <linux/spinlock.h> 16 #include <linux/kref.h> 17 #include <net/dst_cache.h> 18 19 struct wg_device; 20 21 struct endpoint { 22 union { 23 struct sockaddr addr; 24 struct sockaddr_in addr4; 25 struct sockaddr_in6 addr6; 26 }; 27 union { 28 struct { 29 struct in_addr src4; 30 /* Essentially the same as addr6->scope_id */ 31 int src_if4; 32 }; 33 struct in6_addr src6; 34 }; 35 }; 36 37 struct wg_peer { 38 struct wg_device *device; 39 struct prev_queue tx_queue, rx_queue; 40 struct sk_buff_head staged_packet_queue; 41 int serial_work_cpu; 42 struct noise_keypairs keypairs; 43 struct endpoint endpoint; 44 struct dst_cache endpoint_cache; 45 rwlock_t endpoint_lock; 46 struct noise_handshake handshake; 47 atomic64_t last_sent_handshake; 48 struct work_struct transmit_handshake_work, clear_peer_work, transmit_packet_work; 49 struct cookie latest_cookie; 50 struct hlist_node pubkey_hash; 51 u64 rx_bytes, tx_bytes; 52 struct timer_list timer_retransmit_handshake, timer_send_keepalive; 53 struct timer_list timer_new_handshake, timer_zero_key_material; 54 struct timer_list timer_persistent_keepalive; 55 unsigned int timer_handshake_attempts; 56 u16 persistent_keepalive_interval; 57 bool timer_need_another_keepalive; 58 bool sent_lastminute_handshake; 59 struct timespec64 walltime_last_handshake; 60 struct kref refcount; 61 struct rcu_head rcu; 62 struct list_head peer_list; 63 struct list_head allowedips_list; 64 u64 internal_id; 65 struct napi_struct napi; 66 bool is_dead; 67 }; 68 69 struct wg_peer *wg_peer_create(struct wg_device *wg, 70 const u8 public_key[NOISE_PUBLIC_KEY_LEN], 71 const u8 preshared_key[NOISE_SYMMETRIC_KEY_LEN]); 72 73 struct wg_peer *__must_check wg_peer_get_maybe_zero(struct wg_peer *peer); wg_peer_get(struct wg_peer * peer)74static inline struct wg_peer *wg_peer_get(struct wg_peer *peer) 75 { 76 kref_get(&peer->refcount); 77 return peer; 78 } 79 void wg_peer_put(struct wg_peer *peer); 80 void wg_peer_remove(struct wg_peer *peer); 81 void wg_peer_remove_all(struct wg_device *wg); 82 83 int wg_peer_init(void); 84 void wg_peer_uninit(void); 85 86 #endif /* _WG_PEER_H */ 87