forked from Har-Kuun/DomainMegaBot
-
Notifications
You must be signed in to change notification settings - Fork 37
/
domain_scan.py
123 lines (107 loc) · 3.13 KB
/
domain_scan.py
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
import socket
import select
import time
import os
import threading
import sys
# 并发最大线程数
max_thread = 10
timeout = 10
sleep_time = 0.1
socket.setdefaulttimeout(timeout)
# [print(chr(i)) for i in range(ord("a"),ord("z")+1)]
def get_tld():
tld = list()
with open("tld", "r") as f:
for line in f:
if not line.startswith("//"):
tld.append(line)
return tld
# 域名信息,域名后缀,whois服务器
def whois_query(domain_name, name_server, whois_server):
retry = 3
domain = domain_name + "." + name_server
info = ""
while not info and retry > 0:
try:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.connect((whois_server, 43))
s.send(f"{domain} \r\n".encode())
while True:
res = s.recv(1024)
if not len(res):
break
info += str(res)
s.close()
retry -= 1
time.sleep(sleep_time)
except:
pass
return info
def get_reginfo(name, tld_info):
can_reg = False
info = whois_query(name, tld_info[0], tld_info[1])
reg = tld_info[2]
# print(reg)
if info is "":
print(f"{name}.{tld_info[0]} => 查询失败")
return
# print(info)
if info.find(reg) >= 0:
print(f"{name}.{tld_info[0]} => 未注册")
can_reg = True
else:
print(f"{name}.{tld_info[0]} => 已注册", end="\r", flush=True)
can_reg = False
if can_reg:
with open(f"result.txt", "a") as f:
f.write(f"{name}.{tld_info[0]} \n")
def get_domain_name(name):
tld_list = get_tld()
tld_array = [x.split("=")[:-1] for x in tld_list][1:]
# [print(y) for y in tld_array]
for domain in tld_array:
while threading.active_count() > max_thread:
time.sleep(sleep_time)
t = threading.Thread(
target=get_reginfo,
args=(
name,
domain,
),
)
t.start()
time.sleep(sleep_time)
def get_domain_free(name, domain):
name_list = []
with open(name, "r") as f:
for line in f:
if line:
name_list.append(line.strip())
tld_list = get_tld()[1:]
tld_array = [x.split("=")[0] for x in tld_list]
if domain not in tld_array:
print(f"域名 {domain} 不在tld列表中")
tld_index = tld_array.index(domain)
tld_par_list = [x.split("=")[:-1] for x in tld_list]
for name in name_list:
while threading.active_count() > max_thread:
time.sleep(sleep_time)
t = threading.Thread(
target=get_reginfo,
args=(
name,
tld_par_list[tld_index],
),
)
t.start()
time.sleep(sleep_time)
if __name__ == "__main__":
args = sys.argv[1:]
if len(args) == 1:
## python asura
get_domain_name(args[0])
else:
## python lu 2letter
get_domain_free(args[1], args[0])