socket: keep track of src address in sending packets

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>

1 files changed, 13 insertions, 1 deletions

diff --git a/src/peer.h b/src/peer.h
index 895eb77..07c4d16 100644
--- a/src/peer.h
+++ b/src/peer.h
@@ -16,9 +16,21 @@
 
 struct wireguard_device;
 
+struct endpoint {
+	union {
+		struct sockaddr_storage addr_storage;
+		struct sockaddr_in addr4;
+		struct sockaddr_in6 addr6;
+	};
+	union {
+		struct in_addr src4;
+		struct in6_addr src6;
+	};
+};
+
 struct wireguard_peer {
 	struct wireguard_device *device;
-	struct sockaddr_storage endpoint_addr;
+	struct endpoint endpoint;
 	struct dst_cache endpoint_cache;
 	rwlock_t endpoint_lock;
 	struct noise_handshake handshake;