-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
271 lines (208 loc) · 12.5 KB
/
main.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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
from module import fofa
from module.read_control import thread_control
from module.shodan_w import shodan_search
from module.vuln_scan import vuln_scan_poc
from module.vuln_scan import vuln_scan_exp
import colorama
import datetime
def picture():
# 图片打印
print(colorama.Fore.YELLOW+r'''
____ ____
/ ___| _ __ ___ _ _ / ___| ___ __ _ _ __ _ __ ___ _ __
| | _ | '__| / _ \ | | | | \___ \ / __| / _` | | '_ \ | '_ \ / _ \ | '__|
| |_| | | | | __/ | |_| | ___) | | (__ | (_| | | | | | | | | | | __/ | |
\____| |_| \___| \__, | |____/ \___| \__,_| |_| |_| |_| |_| \___| |_|
|___/
'''+colorama.Style.RESET_ALL)
if __name__=='__main__':
picture()
# 死循环界面
while True:
print(colorama.Fore.YELLOW+"-----------------------------------------模式选择------------------------------------"+colorama.Style.RESET_ALL)
print(colorama.Fore.RED+r'''
*---------(1)信息收集
*---------(2)漏洞扫描
*---------(3)退出
'''+colorama.Style.RESET_ALL)
# 第一层界面
user_option1=int(input(colorama.Fore.BLUE+">>>"+colorama.Style.RESET_ALL))
if user_option1==1:
while True:
print(colorama.Fore.YELLOW+'''
*---------fofa(1)
*---------shadon(2)
*---------退出(3)
'''+colorama.Style.RESET_ALL)
user_option2=int(input(colorama.Fore.BLUE+">>>"+colorama.Style.RESET_ALL))
# 第二层界面（信息收集）
if user_option2==1:
print(colorama.Fore.RED+"初次使用信息收集模块需要到./module/fofa.py 里面设置好自己的fof api。"+colorama.Style.RESET_ALL)
print(colorama.Fore.YELLOW+"输入query进入查询模式。"+colorama.Style.RESET_ALL)
print(colorama.Fore.GREEN+"exit 退出"+colorama.Style.RESET_ALL)
while True:
user_input=input(colorama.Fore.BLUE+">>>"+colorama.Style.RESET_ALL)
if user_input=='exit':
print("退出......")
break
elif user_input=='query':
user_querry=input(colorama.Fore.BLUE+"(fofa)>>"+colorama.Style.RESET_ALL)
result=fofa.fofa_search(user_querry)
print(colorama.Fore.BLUE+"是否保存？Y/N"+colorama.Style.RESET_ALL)
user_input=input(colorama.Fore.BLUE+">>>"+colorama.Style.RESET_ALL)
if user_input=='Y':
current_time = datetime.datetime.now()
file_name = "fofa_" + current_time.strftime("%Y%m%d_%H%M%S") + ".txt"
with open(f"./search/{file_name}", "w") as file:
print(result[1])
for i in result:
i=str(i)
file.write(i)
file.close
else:
continue
# 信息收集第二层界面
elif user_option2==2:
print(colorama.Fore.RED+"初次使用信息收集模块需要到./module/shodan_w.py 里面设置好自己的shodan api。"+colorama.Style.RESET_ALL)
print(colorama.Fore.YELLOW+"输入querry进入查询模式。"+colorama.Style.RESET_ALL)
print(colorama.Fore.GREEN+"exit 退出"+colorama.Style.RESET_ALL)
user_input=input(colorama.Fore.BLUE+">>>"+colorama.Style.RESET_ALL)
if user_input=='exit':
print("退出......")
break
elif user_input=='query':
user_querry=input(colorama.Fore.BLUE+"(shodan)>>"+colorama.Style.RESET_ALL)
result=shodan_search(user_querry)
print(colorama.Fore.BLUE+"是否保存？Y/N"+colorama.Style.RESET_ALL)
user_input=input(colorama.Fore.BLUE+">>>"+colorama.Style.RESET_ALL)
if user_input=='Y':
current_time = datetime.datetime.now()
file_name = "shodan_" + current_time.strftime("%Y%m%d_%H%M%S") + ".txt"
with open(f"./search/{file_name}", "w") as file:
print(result[1])
for i in result:
i=str(i)
file.write(i)
file.close
else:
continue
# 第二层（信息收集）退出
elif user_option2==3:
print("退出.....")
break
else:
print(colorama.Fore.RED+"没有的选项..."+colorama.Style.RESET_ALL)
continue
# 第一层界面
elif user_option1==2:
print(colorama.Fore.YELLOW+r'''
*---------(1)端口服务爆破
*---------(2)漏洞利用和扫描
*---------(3)退出
'''+colorama.Style.RESET_ALL)
while True:
user_option2=int(input(colorama.Fore.BLUE+"(漏洞利用和扫描)>>>"+colorama.Style.RESET_ALL))
# 第二层端口爆破界面
if user_option2==1:
print(colorama.Fore.YELLOW+'''
*---------(1)ssh
*---------(2)mysql
*---------(3)telnet
*---------(4)ftp
*---------(5)exit
'''+colorama.Style.RESET_ALL)
while True:
# 第三层界面（端口爆破）
user_option3=int(input(colorama.Fore.BLUE+"(端口爆破)>>>"+colorama.Style.RESET_ALL))
if user_option3==1:
user_thread=int(input(colorama.Fore.BLUE+"(ssh,指定爆破线程)>>>"+colorama.Style.RESET_ALL))
user_ip=str(input(colorama.Fore.BLUE+"(指定爆破ip)>>>"+colorama.Style.RESET_ALL))
user_port=int(input(colorama.Fore.BLUE+"(制定端口)>>>"+colorama.Style.RESET_ALL))
thread_control(user_thread,1,user_ip,user_port)
print(colorama.Fore.YELLOW+'''
*---------(1)ssh
*---------(2)mysql
*---------(3)telnet
*---------(4)ftp
*---------(5)exit
'''+colorama.Style.RESET_ALL)
# 地三层界面退出
elif user_option3==2:
user_thread=int(input(colorama.Fore.BLUE+"(mysql,指定爆破线程)>>>"+colorama.Style.RESET_ALL))
user_ip=str(input(colorama.Fore.BLUE+"(指定爆破ip)>>>"+colorama.Style.RESET_ALL))
user_port=int(input(colorama.Fore.BLUE+"(制定端口)>>>"+colorama.Style.RESET_ALL))
thread_control(user_thread,2,user_ip,user_port)
print(colorama.Fore.YELLOW+'''
*---------(1)ssh
*---------(2)mysql
*---------(3)telnet
*---------(4)ftp
*---------(5)exit
'''+colorama.Style.RESET_ALL)
elif user_option3==3:
user_thread=int(input(colorama.Fore.BLUE+"(telnet,指定爆破线程)>>>"+colorama.Style.RESET_ALL))
user_ip=str(input(colorama.Fore.BLUE+"(指定爆破ip)>>>"+colorama.Style.RESET_ALL))
user_port=int(input(colorama.Fore.BLUE+"(制定端口)>>>"+colorama.Style.RESET_ALL))
thread_control(user_thread,3,user_ip,user_port)
print(colorama.Fore.YELLOW+'''
*---------(1)ssh
*---------(2)mysql
*---------(3)telnet
*---------(4)ftp
*---------(5)exit
'''+colorama.Style.RESET_ALL)
elif user_option3==4:
user_thread=int(input(colorama.Fore.BLUE+"(ftp,指定爆破线程)>>>"+colorama.Style.RESET_ALL))
user_ip=str(input(colorama.Fore.BLUE+"(指定爆破ip)>>>"+colorama.Style.RESET_ALL))
user_port=int(input(colorama.Fore.BLUE+"(制定端口)>>>"+colorama.Style.RESET_ALL))
thread_control(user_thread,4,user_ip,user_port)
print(colorama.Fore.YELLOW+'''
*---------(1)ssh
*---------(2)mysql
*---------(3)telnet
*---------(4)ftp
*---------(5)exit
'''+colorama.Style.RESET_ALL)
elif user_option3==5:
print("退出......")
break
else:
print(colorama.Fore.RED+"没有的选项..."+colorama.Style.RESET_ALL)
continue
elif user_option2==2:
while True:
print(colorama.Fore.YELLOW+'''
*---------(1)exp
*---------(2)poc
*---------(3)连接远程数据
*---------(4)退出
'''+colorama.Style.RESET_ALL)
user_option2=int(input(colorama.Fore.BLUE+"(exp)>>>"+colorama.Style.RESET_ALL))
# 第二层界面 exp和poc界面选择
if user_option2==1:
vuln_scan_exp()
continue
elif user_option2==2:
vuln_scan_poc()
continue
elif user_option2==3:
pass
continue
elif user_option2==4:
break
elif user_option2==3:
print(colorama.Fore.YELLOW+r'''
*---------(1)端口服务爆破
*---------(2)漏洞利用和扫描
*---------(3)退出
'''+colorama.Style.RESET_ALL)
print("退出.....")
break
else:
print("没有的选项")
elif user_option1==3:
print("GoodBye~")
break
else:
print(colorama.Fore.RED+"没有的选项..."+colorama.Style.RESET_ALL)
continue