From 46de6f340ae179741fb475ad6858a7a56a34af4a Mon Sep 17 00:00:00 2001 From: "Jason A. Donenfeld" Date: Thu, 10 Nov 2016 23:28:35 +0100 Subject: socket: keep track of src address in sending packets --- src/socket.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/socket.h') diff --git a/src/socket.h b/src/socket.h index 36cb4ed..ce668ad 100644 --- a/src/socket.h +++ b/src/socket.h @@ -9,6 +9,7 @@ #include struct wireguard_device; +struct endpoint; #define SKB_HEADER_LEN (max(sizeof(struct iphdr), sizeof(struct ipv6hdr)) + sizeof(struct udphdr) + NET_SKB_PAD) @@ -18,7 +19,7 @@ int socket_send_buffer_to_peer(struct wireguard_peer *peer, void *data, size_t l int socket_send_skb_to_peer(struct wireguard_peer *peer, struct sk_buff *skb, uint8_t ds); int socket_send_buffer_as_reply_to_skb(struct wireguard_device *wg, struct sk_buff *in_skb, void *out_buffer, size_t len); -int socket_addr_from_skb(struct sockaddr_storage *sockaddr, struct sk_buff *skb); -void socket_set_peer_addr(struct wireguard_peer *peer, struct sockaddr_storage *sockaddr); +int socket_endpoint_from_skb(struct endpoint *endpoint, struct sk_buff *skb); +void socket_set_peer_endpoint(struct wireguard_peer *peer, struct endpoint *endpoint); #endif -- cgit v1.2.3-59-g8ed1b