| 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 | |
|---|
