feat(server): add wildcard domain name resolution

Author: duanhongyi

smartdns/server.py

@@ -72,26 +72,31 @@ def query(self, query, timeout=None, addr=None, edns=None):
             return defer.fail(failure.Failure(NotImplementedError(str(self.__class__) + " " + str(query.type))))
 
     def lookupAddress(self, name, timeout=None, addr=None, edns=None):
-        if name in self.a_mapping:
-            ttl = self.a_mapping[name]['ttl']
 
-            def packResult(value):
-                ret = []
-                add = []
-                for x in value:
-                    ret.append(dns.RRHeader(name, dns.A, dns.IN, ttl, dns.Record_A(x, ttl), True))
+        def packResult(value, ttl):
+            ret = []
+            add = []
+            for x in value:
+                ret.append(dns.RRHeader(name, dns.A, dns.IN, ttl, dns.Record_A(x, ttl), True))
 
-                if edns is not None:
-                    if edns.rdlength > 8:
-                        add.append(dns.RRHeader('', sdns.EDNS, 4096, edns.ttl, edns.payload, True))
-                    else:
-                        add.append(dns.RRHeader('', sdns.EDNS, 4096, 0, sdns.Record_EDNS(None, 0), True))
-                return [ret, (), add]
+            if edns is not None:
+                if edns.rdlength > 8:
+                    add.append(dns.RRHeader('', sdns.EDNS, 4096, edns.ttl, edns.payload, True))
+                else:
+                    add.append(dns.RRHeader('', sdns.EDNS, 4096, 0, sdns.Record_EDNS(None, 0), True))
+            return [ret, (), add]
 
+        wildcard = name[name.index("."):]
+        if name in self.a_mapping:
+            ttl = self.a_mapping[name]['ttl']
             result = self.finder.findIP(str(addr[0]), name)
-            # 返回的IP数组乱序
-            random.shuffle(result)
-            return packResult(result)
+            random.shuffle(result)  # 返回的IP数组乱序
+            return packResult(result, ttl)
+        elif wildcard in self.a_mapping:
+            ttl = self.a_mapping[wildcard]['ttl']
+            result = self.finder.findIP(str(addr[0]), wildcard)
+            random.shuffle(result)  # 返回的IP数组乱序
+            return packResult(result, ttl)
         else:
             return self._lookup(name, dns.IN, dns.A, timeout)
 

templates/a.yaml

@@ -1,4 +1,10 @@
 ---
+.test.com:
+  ttl: 3600
+  default: 3.3.3.7 1.1.1.1
+  中国,广东,,联通: 1.1.1.1 3.3.3.1
+  中国,广东,,电信: 1.1.1.2 3.3.3.2
+
 test.test.com:
   ttl: 3600
   default: 3.3.3.7 1.1.1.1
@@ -12,4 +18,4 @@ test.test.com:
   中国,内蒙古,,: 1.1.1.8 3.3.3.8
   中国,,,: 1.1.1.9 3.3.3.9
   美国,,,: 1.1.1.10 3.3.3.10
-  马来西亚,,,: 1.1.1.11 3.3.3.11
+  马来西亚,,,: 1.1.1.11 3.3.3.11