python如何實(shí)現(xiàn)的IP存活掃描腳本

小編給大家分享一下python如何實(shí)現(xiàn)的IP存活掃描腳本，希望大家閱讀完這篇文章后大所收獲，下面讓我們一起去探討吧！

下載地址

ActiveOrNot

用于處理 oneforall 等子域名掃描工具的結(jié)果去重 + 主機(jī)存活掃描

參數(shù)

-f --file  指定存放ip或子域名的文件，默認(rèn) ip.txt
-t --thread  設(shè)置線程數(shù)，默認(rèn) 50

python3 ActiveOrNot.py -f ip.txt -t 12

具體代碼 ActiveOrNot.py

from threading import Thread
from queue import Queue
import requests
from time import time
import argparse

headers = {
 "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.116 Safari/537.36"
}


def ping(url, new_ip):
 url = url.strip()
 if (not url.startswith('http://')) and (not url.startswith('https://')):
  url = 'http://' + url
 try:
  req = requests.get(url, headers=headers, timeout=2)
  new_ip.put(url + ' -- ' + str(req.status_code))
  print("%s 存活" % url)
 except:
  print("%s 不存活" % url)


def new_list(file):
 with open(file, 'r') as f:
  new_ip = []
  ip_list = f.readlines()
  for ip in ip_list:
   ip = ip.strip().replace('http://', '').replace('https://', '')
   if ip:
    if not (ip in new_ip):
     new_ip.append(ip)
  return new_ip


def main(file, th):
 begin_time = time()
 new_ip = Queue()
 ip_list = new_list(file)
 j = 0
 length = len(ip_list)
 while j < length:
  threads = []
  for i in range(th):
   t = Thread(target=ping, args=(ip_list[j], new_ip))
   t.start()
   threads.append(t)
   j += 1
   if j == length:
    break
  for thread in threads:
   thread.join()
 with open('NewIP.txt', 'a+') as nf:
  while not new_ip.empty():
   nf.write(new_ip.get()+'\n')
 end_time = time()
 run_time = end_time - begin_time
 print("總共耗時(shí) %s 秒"% run_time)


if __name__ == '__main__':
 parser = argparse.ArgumentParser(description='url active scan')
 parser.add_argument("-f", "--file", help="指定文件", default='ip.txt')
 parser.add_argument("-t", "--thread", help="設(shè)置線程", default=50)
 args = parser.parse_args()
 file = args.file
 th = args.thread
 main(file, th)

看完了這篇文章，相信你對(duì)python如何實(shí)現(xiàn)的IP存活掃描腳本有了一定的了解，想了解更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！