drivers/net/wireguard/socket.h

   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_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 */