You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
3.5 KiB
87 lines
3.5 KiB
4 years ago
|
import re
|
||
|
|
import time
|
||
|
|
import threading
|
||
|
|
import requests
|
||
|
|
import libs.core as cores
|
||
|
|
|
||
|
|
class NetThreads(threading.Thread):
|
||
|
|
|
||
|
|
def __init__(self,threadID,name,domain_queue,worksheet):
|
||
|
|
threading.Thread.__init__(self)
|
||
|
|
self.name = name
|
||
|
|
self.threadID = threadID
|
||
|
|
self.lock = threading.Lock()
|
||
|
|
self.domain_queue = domain_queue
|
||
|
|
self.worksheet = worksheet
|
||
|
|
|
||
|
|
def __get_Http_info__(self,threadLock):
|
||
|
|
while True:
|
||
|
|
domains = self.domain_queue.get(timeout=5)
|
||
|
|
domain = domains["domain"]
|
||
|
|
url_ip = domains["url_ip"]
|
||
|
|
time.sleep(2)
|
||
|
|
result = self.__get_request_result__(url_ip)
|
||
|
|
if result != "error":
|
||
|
|
if self.lock.acquire(True):
|
||
|
|
cores.excel_row = cores.excel_row + 1
|
||
|
|
self.worksheet.write(cores.excel_row, 0, label = cores.excel_row)
|
||
|
|
self.worksheet.write(cores.excel_row, 1, label = url_ip)
|
||
|
|
self.worksheet.write(cores.excel_row, 2, label = domain)
|
||
|
|
if result != "timeout":
|
||
|
|
self.worksheet.write(cores.excel_row, 3, label = result["status"])
|
||
|
|
self.worksheet.write(cores.excel_row, 4, label = result["des_ip"])
|
||
|
|
self.worksheet.write(cores.excel_row, 5, label = result["server"])
|
||
|
|
self.worksheet.write(cores.excel_row, 6, label = result["title"])
|
||
|
|
self.worksheet.write(cores.excel_row, 7, label = result["cdn"])
|
||
|
|
self.worksheet.write(cores.excel_row, 8, label = "")
|
||
|
|
self.lock.release()
|
||
|
|
if self.domain_queue.empty():
|
||
|
|
break
|
||
|
|
|
||
|
|
def __get_request_result__(self,url):
|
||
|
|
result={"status":"","server":"","cookie":"","cdn":"","des_ip":"","sou_ip":"","title":""}
|
||
|
|
cdn = ""
|
||
|
|
try:
|
||
|
|
# python3 app.py ios -i C:\Users\Administrator\Desktop\frida-ipa-dump\三晋通\Payload\三晋通.app\三晋通
|
||
|
|
#
|
||
|
|
with requests.get(url, timeout=5,stream=True) as rsp:
|
||
|
|
status_code = rsp.status_code
|
||
|
|
result["status"] = status_code
|
||
|
|
headers = rsp.headers
|
||
|
|
if "Server" in headers:
|
||
|
|
result["server"] = headers['Server']
|
||
|
|
if "Cookie" in headers:
|
||
|
|
result["cookie"] = headers['Cookie']
|
||
|
|
if "X-Via" in headers:
|
||
|
|
cdn = cdn + headers['X-Via']
|
||
|
|
if "Via" in headers:
|
||
|
|
cdn = cdn + headers['Via']
|
||
|
|
result["cdn"] = cdn
|
||
|
|
sock = rsp.raw._connection.sock
|
||
|
|
|
||
|
|
if sock:
|
||
|
|
des_ip = sock.getpeername()[0]
|
||
|
|
sou_ip = sock.getsockname()[0]
|
||
|
|
if des_ip:
|
||
|
|
result["des_ip"] = des_ip
|
||
|
|
if sou_ip:
|
||
|
|
result["sou_ip"] = sou_ip
|
||
|
|
sock.close()
|
||
|
|
html = rsp.text
|
||
|
|
title = re.findall('<title>(.+)</title>',html)
|
||
|
|
if title:
|
||
|
|
result["title"] = title[0]
|
||
|
|
rsp.close()
|
||
|
|
return result
|
||
|
|
except requests.exceptions.InvalidURL as e:
|
||
|
|
return "error"
|
||
|
|
except requests.exceptions.ConnectionError as e1:
|
||
|
|
return "timeout"
|
||
|
|
except requests.exceptions.ReadTimeout as e2:
|
||
|
|
return "timeout"
|
||
|
|
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
threadLock = threading.Lock()
|
||
|
|
self.__get_Http_info__(threadLock)
|