Update urlcheck.py

urlcheck.py

@@ -10,10 +10,11 @@
 END = '\033[0m'  # 结束
 
 
-
 async def check_url(url, alive_urls, timeout):
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'}
     try:
-        async with aiohttp.ClientSession() as session:
+        async with aiohttp.ClientSession(headers=headers) as session:
             async with session.get(url, timeout=timeout) as response:
                 if response.status == 200:
                     print(GREEN + '[+] ' + url + " is alive" + END)
@@ -31,7 +32,8 @@ def main():
     parser = argparse.ArgumentParser(description='欢迎使用urlcheck  作者：月影')
     parser.add_argument('-f', '--file', type=str, help='指定文件批量探测')
     parser.add_argument('-u', '--url', type=str, help='指定一个URL进行探测')
-    parser.add_argument('-t', '--timeout', type=int, default=20, help='设置超时时间，默认为20秒')
+    parser.add_argument('-t', '--timeout', type=int, default=20, help='设置超时时间，默认为30秒，建议50秒')
+    parser.add_argument('-k', '--key-file', type=str, default='key.txt', help='指定字典文件，默认为key.txt')
     parser.add_argument('-H', '--help-info', action='help', help='显示所有参数及描述\n\n')
 
     args = parser.parse_args()
@@ -47,24 +49,29 @@ def main():
 
     urls = ['http://' + url + '/' if not url.startswith(('http://', 'https://')) else url + '/' for url in urls]
 
-    with open("key.txt") as f:
+    with open(args.key_file) as f:
         keys = f.read().splitlines()
 
-    urls_with_keys = [url + key + '/' for url in urls for key in keys]
+    urls_with_keys = [url + key for url in urls for key in keys]
 
     start_time = time.time()
 
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     alive_urls = set()
+    count = 0
     tasks = [loop.create_task(check_url(url, alive_urls, args.timeout)) for url in urls_with_keys]
     loop.run_until_complete(asyncio.wait(tasks))
     loop.close()
 
     end_time = time.time()
 
+    for url in alive_urls:
+        count += 1
+    print("存活的URL数量为: %d" % count)
     print("扫描完毕，总耗时: %.2f 秒" % (end_time - start_time))
 
+
     with open('alive.txt', 'w') as f:
         for url in alive_urls:
             f.write(url + '\n')