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 	bool is_dead;
43 	struct noise_keypairs keypairs;
44 	struct endpoint endpoint;
45 	struct dst_cache endpoint_cache;
46 	rwlock_t endpoint_lock;
47 	struct noise_handshake handshake;
48 	atomic64_t last_sent_handshake;
49 	struct work_struct transmit_handshake_work, clear_peer_work, transmit_packet_work;
50 	struct cookie latest_cookie;
51 	struct hlist_node pubkey_hash;
52 	u64 rx_bytes, tx_bytes;
53 	struct timer_list timer_retransmit_handshake, timer_send_keepalive;
54 	struct timer_list timer_new_handshake, timer_zero_key_material;
55 	struct timer_list timer_persistent_keepalive;
56 	unsigned int timer_handshake_attempts;
57 	u16 persistent_keepalive_interval;
58 	bool timer_need_another_keepalive;
59 	bool sent_lastminute_handshake;
60 	struct timespec64 walltime_last_handshake;
61 	struct kref refcount;
62 	struct rcu_head rcu;
63 	struct list_head peer_list;
64 	struct list_head allowedips_list;
65 	struct napi_struct napi;
66 	u64 internal_id;
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)74 static 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