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_SOCKET_H
7 #define _WG_SOCKET_H
8 
9 #include <linux/netdevice.h>
10 #include <linux/udp.h>
11 #include <linux/if_vlan.h>
12 #include <linux/if_ether.h>
13 
14 int wg_socket_init(struct wg_device *wg, u16 port);
15 void wg_socket_reinit(struct wg_device *wg, struct sock *new4,
16 		      struct sock *new6);
17 int wg_socket_send_buffer_to_peer(struct wg_peer *peer, void *data,
18 				  size_t len, u8 ds);
19 int wg_socket_send_skb_to_peer(struct wg_peer *peer, struct sk_buff *skb,
20 			       u8 ds);
21 int wg_socket_send_buffer_as_reply_to_skb(struct wg_device *wg,
22 					  struct sk_buff *in_skb,
23 					  void *out_buffer, size_t len);
24 
25 int wg_socket_endpoint_from_skb(struct endpoint *endpoint,
26 				const struct sk_buff *skb);
27 void wg_socket_set_peer_endpoint(struct wg_peer *peer,
28 				 const struct endpoint *endpoint);
29 void wg_socket_set_peer_endpoint_from_skb(struct wg_peer *peer,
30 					  const struct sk_buff *skb);
31 void wg_socket_clear_peer_endpoint_src(struct wg_peer *peer);
32 
33 #if defined(CONFIG_DYNAMIC_DEBUG) || defined(DEBUG)
34 #define net_dbg_skb_ratelimited(fmt, dev, skb, ...) do {                       \
35 		struct endpoint __endpoint;                                    \
36 		wg_socket_endpoint_from_skb(&__endpoint, skb);                 \
37 		net_dbg_ratelimited(fmt, dev, &__endpoint.addr,                \
38 				    ##__VA_ARGS__);                            \
39 	} while (0)
40 #else
41 #define net_dbg_skb_ratelimited(fmt, skb, ...)
42 #endif
43 
44 #endif /* _WG_SOCKET_H */
45