1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
'''
DNSv6工具包,注意tomorrow3无法在arm架构处理器上使用(m1)
'''
import dns.resolver
import pandas as pd
import random as rd
import tomorrow3 as tm
from tqdm import tqdm
alphabet = "abcdefghijklmnopqrstuvwxyz1234567890"
result = pd.DataFrame(columns=["rdns", "result"])
dot_ressult = pd.DataFrame(columns=["dot", "result"])
# val负责定位,dataframe指定数据来源
@tm.threads(200)
def dnsresolver(val,dataframe):
characters = "".join(rd.sample(alphabet, 10)) # 生成子域名
test = dataframe.loc[val, "rdns"]
reso = dns.resolver.Resolver()
reso.nameservers = [test]
reso.timeout = 10
try:
AAAA = reso.resolve(characters + ".v4.testv4-v6.live", "AAAA").response
# result = result.append([[test, AAAA.rcode()]], ignore_index=True)
return [test,AAAA.rcode()]
except:
# result=result.append([[test, 1]], ignore_index=True)
return [test,1]
# if __name__=="__main__":
# mode="main"
# rdns = pd.read_csv("./res_data/rdns-shandong.csv", names=["rdns"])
# dot = pd.read_csv("./res_data/853-shandong.csv", names=["dot"])
# ch_rdns = pd.read_excel("./res_data/全国-递归DNS测量结果.xlsx", names=["rdns", "loc", "company"])
#
# for i in tqdm(range(ch_rdns.shape[0])):
# dnsresolver(i,ch_rdns,result)
#
# if (mode == "rdns"):
# result.to_csv("./result/"+str(2)+"-ch_rdns.csv")
# else:
# dot_ressult.to_csv("./result/dot.csv")
|