diff --git a/catvod/src/main/java/com/github/catvod/net/OkDns.java b/catvod/src/main/java/com/github/catvod/net/OkDns.java index 3b24b52a6..7c3faccdd 100644 --- a/catvod/src/main/java/com/github/catvod/net/OkDns.java +++ b/catvod/src/main/java/com/github/catvod/net/OkDns.java @@ -2,9 +2,12 @@ package com.github.catvod.net; import androidx.annotation.NonNull; +import com.github.catvod.utils.Util; + import java.net.InetAddress; import java.net.UnknownHostException; import java.util.List; +import java.util.Map; import java.util.concurrent.ConcurrentHashMap; import okhttp3.Dns; @@ -40,7 +43,7 @@ public class OkDns implements Dns { @NonNull @Override public List lookup(@NonNull String hostname) throws UnknownHostException { - return (doh != null ? doh : Dns.SYSTEM).lookup(map.containsKey(hostname) ? map.get(hostname) : hostname); + for (Map.Entry entry : map.entrySet()) if (Util.containOrMatch(hostname, entry.getKey())) hostname = entry.getValue(); + return (doh != null ? doh : Dns.SYSTEM).lookup(hostname); } } -