projects/linux·blob master · refs

86 lines · 2.2 KiB

1/* SPDX-License-Identifier: GPL-2.0 */
2/*
3 * Copyright (C) 2015-2019 Jason A. Donenfeld <[email protected]>. 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
19struct wg_device;
20
21struct endpoint {
22 union {
23 struct sockaddr_inet addr; /* Large enough for both address families */
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
37struct 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
69struct 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
73struct wg_peer *__must_check wg_peer_get_maybe_zero(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}
79void wg_peer_put(struct wg_peer *peer);
80void wg_peer_remove(struct wg_peer *peer);
81void wg_peer_remove_all(struct wg_device *wg);
82
83int wg_peer_init(void);
84void wg_peer_uninit(void);
85
86#endif /* _WG_PEER_H */