1*e7096c13SJason A. Donenfeld /* SPDX-License-Identifier: GPL-2.0 */ 2*e7096c13SJason A. Donenfeld /* 3*e7096c13SJason A. Donenfeld * Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved. 4*e7096c13SJason A. Donenfeld */ 5*e7096c13SJason A. Donenfeld 6*e7096c13SJason A. Donenfeld #ifndef _WG_SOCKET_H 7*e7096c13SJason A. Donenfeld #define _WG_SOCKET_H 8*e7096c13SJason A. Donenfeld 9*e7096c13SJason A. Donenfeld #include <linux/netdevice.h> 10*e7096c13SJason A. Donenfeld #include <linux/udp.h> 11*e7096c13SJason A. Donenfeld #include <linux/if_vlan.h> 12*e7096c13SJason A. Donenfeld #include <linux/if_ether.h> 13*e7096c13SJason A. Donenfeld 14*e7096c13SJason A. Donenfeld int wg_socket_init(struct wg_device *wg, u16 port); 15*e7096c13SJason A. Donenfeld void wg_socket_reinit(struct wg_device *wg, struct sock *new4, 16*e7096c13SJason A. Donenfeld struct sock *new6); 17*e7096c13SJason A. Donenfeld int wg_socket_send_buffer_to_peer(struct wg_peer *peer, void *data, 18*e7096c13SJason A. Donenfeld size_t len, u8 ds); 19*e7096c13SJason A. Donenfeld int wg_socket_send_skb_to_peer(struct wg_peer *peer, struct sk_buff *skb, 20*e7096c13SJason A. Donenfeld u8 ds); 21*e7096c13SJason A. Donenfeld int wg_socket_send_buffer_as_reply_to_skb(struct wg_device *wg, 22*e7096c13SJason A. Donenfeld struct sk_buff *in_skb, 23*e7096c13SJason A. Donenfeld void *out_buffer, size_t len); 24*e7096c13SJason A. Donenfeld 25*e7096c13SJason A. Donenfeld int wg_socket_endpoint_from_skb(struct endpoint *endpoint, 26*e7096c13SJason A. Donenfeld const struct sk_buff *skb); 27*e7096c13SJason A. Donenfeld void wg_socket_set_peer_endpoint(struct wg_peer *peer, 28*e7096c13SJason A. Donenfeld const struct endpoint *endpoint); 29*e7096c13SJason A. Donenfeld void wg_socket_set_peer_endpoint_from_skb(struct wg_peer *peer, 30*e7096c13SJason A. Donenfeld const struct sk_buff *skb); 31*e7096c13SJason A. Donenfeld void wg_socket_clear_peer_endpoint_src(struct wg_peer *peer); 32*e7096c13SJason A. Donenfeld 33*e7096c13SJason A. Donenfeld #if defined(CONFIG_DYNAMIC_DEBUG) || defined(DEBUG) 34*e7096c13SJason A. Donenfeld #define net_dbg_skb_ratelimited(fmt, dev, skb, ...) do { \ 35*e7096c13SJason A. Donenfeld struct endpoint __endpoint; \ 36*e7096c13SJason A. Donenfeld wg_socket_endpoint_from_skb(&__endpoint, skb); \ 37*e7096c13SJason A. Donenfeld net_dbg_ratelimited(fmt, dev, &__endpoint.addr, \ 38*e7096c13SJason A. Donenfeld ##__VA_ARGS__); \ 39*e7096c13SJason A. Donenfeld } while (0) 40*e7096c13SJason A. Donenfeld #else 41*e7096c13SJason A. Donenfeld #define net_dbg_skb_ratelimited(fmt, skb, ...) 42*e7096c13SJason A. Donenfeld #endif 43*e7096c13SJason A. Donenfeld 44*e7096c13SJason A. Donenfeld #endif /* _WG_SOCKET_H */ 45