summaryrefslogtreecommitdiff
path: root/DNSv6/Code/dnsfound_util.py
blob: 15e3da9285846be31b85cbefb7b9c5d21cff150b (plain)
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")