Update proxycheck.py

inc/proxycheck.py

@@ -1,15 +1,17 @@
 #!/usr/bin/env python
 # coding=utf-8
-  ################
- #   AabyssZG   #
+################
+#   AabyssZG   #
 ################
 
-from inc import output,run,vul,console
+from inc import output, run, vul, console
 import requests, sys, json
 from termcolor import cprint
 import requests.packages.urllib3
+
 requests.packages.urllib3.disable_warnings()
 
+
 # 检查代理的使用
 def SpringBoot_Scan_Proxy(args):
     if args.proxy:
@@ -28,7 +30,10 @@ def SpringBoot_Scan_Proxy(args):
             if res.status_code == 200:
                 print("GET www.baidu.com 状态码为:" + str(res.status_code))
                 cprint(f"[+] 代理可用，马上执行！", "cyan")
-                SpringBoot_Scan_Header(args,proxies)
+                if args.urlfile:
+                    proxies = f'http://{args.proxy}'
+                SpringBoot_Scan_Header(args, proxies)
+
         except KeyboardInterrupt:
             print("Ctrl + C 手动终止了进程")
             sys.exit()
@@ -37,10 +42,11 @@ def SpringBoot_Scan_Proxy(args):
             sys.exit()
     else:
         proxies = ''
-        SpringBoot_Scan_Header(args,proxies)
+        SpringBoot_Scan_Header(args, proxies)
+
 
 # 导入自定义HTTP头部
-def SpringBoot_Scan_Header(args,proxies):
+def SpringBoot_Scan_Header(args, proxies):
     if args.newheader:
         cprint(f"=====正在导入自定义HTTP头部=====", "cyan")
         filename = args.newheader
@@ -62,6 +68,7 @@ def SpringBoot_Scan_Header(args,proxies):
         header_new = '{}'
         SpringBoot_Scan_Main(args, proxies, header_new)
 
+
 def SpringBoot_Scan_Main(args, proxies, header_new):
     if (args.url or args.urlfile or args.vul or args.vulfile or args.dump or args.zoomeye or args.fofa or args.hunter):
         console.SpringBoot_Scan_console(args, proxies, header_new)