87 lines · c
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_H7#define _WG_PEER_H8 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 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 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 */87