开启域传送漏洞插件，ip2domain增加容错处理

fooying · Jan 12, 2015 · 25a3325 · 25a3325
1 parent f26a438
commit 25a3325
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/plugins/dnszonetransfer/config.yaml b/plugins/dnszonetransfer/config.yaml
@@ -1,4 +1,4 @@
-enable: False
+enable: True
 onerepeat: true
 input: ['root_domain']
 output: ['ip', 'domain']

diff --git a/plugins/dnszonetransfer/dnshelper.py b/plugins/dnszonetransfer/dnshelper.py
@@ -328,7 +328,8 @@ def zone_transfer(self):
         ns_records = list(set(ns_records))
         # Test each NS Server
         for ns_srv in ns_records:
-            if self.check_tcp_dns(ns_srv):
+            #if self.check_tcp_dns(ns_srv):
+            if True:
 
                 try:
                     zone = self.from_wire(dns.query.xfr(ns_srv, self._domain))

diff --git a/plugins/ip2domain/work.py b/plugins/ip2domain/work.py
@@ -69,9 +69,13 @@ def start(self, domain, domain_type, level):
                 for n in range(total_num-1):
                     url = ('http://cn.bing.com/search?q='
                            'ip:%s&first=%s1&FORM=PERE3' % (domain, n))
-                    html = self.req.request('GET', url).text
-                    new_domain_list = re.findall(domain_regx, html, re.X)
-                    domain_list.extend(new_domain_list)
+                    try:
+                        html = self.req.request('GET', url).text
+                    except:
+                        html = ''
+                    else:
+                        new_domain_list = re.findall(domain_regx, html, re.X)
+                        domain_list.extend(new_domain_list)
                     time.sleep(1)
 
             domains, root_domains, ips = self.__classify_result(domain_list)