コード例 #1
0
    def get_ns_history_record(self):
        try:
            db = MySQL(SOURCE_CONFIG)
        except:
            pass
	full_list = []
        sql = 'SELECT domain,ns,ns_ip FROM domain_ns WHERE ns!=""'
	try:
            db.query(sql)
            data = db.fetch_all_rows()
        except Exception as e:
            return "read tld Error" + str(e)
	for i in data:
            tld = str(i['domain'])
            ns = str(i['ns'])
            ns_list = ns.split(';')
            ns = ','.join(ns_list)
            ns_ip = str(i['ns_ip'])
            flag = '0'
            full_list.append((tld,ns,ns_ip,flag))

        taskid = str(random.randint(10000000, 99999999))
        self.write_tld(taskid)
        new_file_name = 'full_' + taskid
        with open("domain_dns_data/sec_compared/"+new_file_name, "w") as f:
            for t in full_list:
                f.write(t[0]+'\t'+'NS'+'\t'+t[1]+'\t'+t[3]+'\n')
                ns_list = t[1].split(',')
                ip_list = t[2].split(';')
                for i in range(0,len(ns_list)):
                    f.write(ns_list[i]+'\t'+'A'+'\t'+ip_list[i]+'\t'+'0'+'\n')	
        self.produce(new_file_name)
        self.produce('tld_'+taskid)
コード例 #2
0
def fetch_tld_ns():
    """
    获取顶级域名的权威服务器(ns)IP地址
    """

    tld_ns = defaultdict(set)
    try:
        db = MySQL(SOURCE_CONFIG)
        sql = 'SELECT tld,server_ipv4 from tld_ns_zone'
        db.query(sql)
        tld_ns_query = db.fetch_all_rows()  # 获取已存储的顶级域名的权威服务器信息
    except Exception, e:
        logger.logger.error("获取顶级域名异常:", e)
        return tld_ns