fix unmap 4in6 ip in wireguard

wwqgtxx · Mar 21, 2024 · 5b2a568 · 5b2a568
1 parent 3f493a4
commit 5b2a568
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/adapter/outbound/wireguard.go b/adapter/outbound/wireguard.go
@@ -145,7 +145,12 @@ func NewWireGuard(option WireGuardOption) (*WireGuard, error) {
 			return address.AddrPort(), nil
 		}
 		udpAddr, err := resolveUDPAddr(ctx, "udp", address.String())
-		return udpAddr.AddrPort(), err
+		if err != nil {
+			return netip.AddrPort{}, err
+		}
+		// net.ResolveUDPAddr maybe return 4in6 address, so unmap at here
+		addrPort := udpAddr.AddrPort()
+		return netip.AddrPortFrom(addrPort.Addr().Unmap(), addrPort.Port()), nil
 	}
 
 	var reserved [3]uint8

diff --git a/dns/resolver.go b/dns/resolver.go
@@ -340,7 +340,7 @@ func (r *Resolver) ipExchange(ctx context.Context, m *D.Msg) (msg *D.Msg, err er
 func (r *Resolver) lookupIP(ctx context.Context, host string, dnsType uint16) ([]netip.Addr, error) {
 
 	if ip, err := netip.ParseAddr(host); err == nil {
-		isIPv4 := ip.Is4()
+		isIPv4 := ip.Is4() || ip.Is4In6()
 		if dnsType == D.TypeAAAA && !isIPv4 {
 			return []netip.Addr{ip}, nil
 		} else if dnsType == D.TypeA && isIPv4 {