From bc260310a09c51bde208ce57b2dc7db5a86d6395 Mon Sep 17 00:00:00 2001
From: FongMi <b8892778@gmail.com>
Date: Thu, 20 Feb 2025 20:04:22 +0800
Subject: [PATCH] Support regex

---
 catvod/src/main/java/com/github/catvod/net/OkDns.java | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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<InetAddress> lookup(@NonNull String hostname) throws UnknownHostException {
-        return (doh != null ? doh : Dns.SYSTEM).lookup(map.containsKey(hostname) ? map.get(hostname) : hostname);
+        for (Map.Entry<String, String> entry : map.entrySet()) if (Util.containOrMatch(hostname, entry.getKey())) hostname = entry.getValue();
+        return (doh != null ? doh : Dns.SYSTEM).lookup(hostname);
     }
 }
-