From 32c02b558c296b824ea0e01eae18ad4126120602 Mon Sep 17 00:00:00 2001
From: dadevel <dadevel@disroot.org>
Date: Wed, 31 May 2023 16:35:08 +0200
Subject: [PATCH] wg-resolve: support ipv6 dns lookup

---
 extras/wg-resolve/wg-resolve.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/extras/wg-resolve/wg-resolve.sh b/extras/wg-resolve/wg-resolve.sh
index 0245f62..2c32a48 100755
--- a/extras/wg-resolve/wg-resolve.sh
+++ b/extras/wg-resolve/wg-resolve.sh
@@ -1,12 +1,12 @@
 #!/bin/sh
 set -eu
 
-# dependencies: dig, ip, ping and wg
+# dependencies: cut, getent, ip, ping and wg
 
 WG_ENDPOINT_DOMAIN="${WG_ENDPOINT%%:*}"
 WG_ENDPOINT_PORT="${WG_ENDPOINT##*:}"
 
 if ! ip netns exec "$WG_NAMESPACE" ping -q -c 1 -W "${WG_TIMEOUT:-5}" "$WG_GATEWAY"; then
     echo 'probe failed, resolving endpoint'
-    ip netns exec "$WG_NAMESPACE" wg set "$WG_INTERFACE" peer "$WG_PEER" endpoint "$(dig +short "$WG_ENDPOINT_DOMAIN"):$WG_ENDPOINT_PORT"
+    ip netns exec "$WG_NAMESPACE" wg set "$WG_INTERFACE" peer "$WG_PEER" endpoint "$(getent hosts -- "$WG_ENDPOINT_DOMAIN" | cut -d ' ' -f 1):$WG_ENDPOINT_PORT"
 fi