102 lines
5.1 KiB
Python
102 lines
5.1 KiB
Python
import json
|
|
import pycurl
|
|
import os
|
|
import re
|
|
import time
|
|
from io import BytesIO
|
|
|
|
class SSLBuild:
|
|
def __init__(self):
|
|
self.urldict = {"0k":"https://downloadfile.self-test.geedge.net/0k", \
|
|
"1k":"https://downloadfile.self-test.geedge.net/1k", \
|
|
"2k":"https://downloadfile.self-test.geedge.net/2k", \
|
|
"4k":"https://downloadfile.self-test.geedge.net/4k", \
|
|
"8k":"https://downloadfile.self-test.geedge.net/8k", \
|
|
"16k":"https://downloadfile.self-test.geedge.net/16k", \
|
|
"32k":"https://downloadfile.self-test.geedge.net/32k", \
|
|
"64k":"https://downloadfile.self-test.geedge.net/64k", \
|
|
"128k":"https://downloadfile.self-test.geedge.net/128k", \
|
|
"256k":"https://downloadfile.self-test.geedge.net/256k", \
|
|
"512k":"https://downloadfile.self-test.geedge.net/512k", \
|
|
"1M":"https://downloadfile.self-test.geedge.net/1M", \
|
|
"2M":"https://downloadfile.self-test.geedge.net/2M", \
|
|
"4M":"https://downloadfile.self-test.geedge.net/4M", \
|
|
"8M":"https://downloadfile.self-test.geedge.net/8M", \
|
|
"16M":"https://downloadfile.self-test.geedge.net/16M", \
|
|
"32M":"https://downloadfile.self-test.geedge.net/32M", \
|
|
"64M":"https://downloadfile.self-test.geedge.net/64M"}
|
|
self.sizelist = ["0k","1k","2k","4k","8k","16k","32k","64k","128k","256k","512k","1M","2M","4M","8M","16M","32M","64M"]
|
|
self.resultlist = []
|
|
|
|
def build_conninfo_json(self,conn):
|
|
dictconninfo = {}
|
|
dictconninfo["status"] = conn.getinfo(pycurl.HTTP_CODE)
|
|
dictconninfo["dns_time"] = conn.getinfo(pycurl.NAMELOOKUP_TIME)
|
|
dictconninfo["conn_time"] = conn.getinfo(pycurl.CONNECT_TIME)
|
|
dictconninfo["app_time"] = conn.getinfo(pycurl.APPCONNECT_TIME)
|
|
dictconninfo["start_transfer_time"] = conn.getinfo(pycurl.STARTTRANSFER_TIME)
|
|
dictconninfo["total_time"] = conn.getinfo(pycurl.TOTAL_TIME)
|
|
dictconninfo["redirect_count"] = conn.getinfo(pycurl.REDIRECT_COUNT)
|
|
dictconninfo["size_upload"] = conn.getinfo(pycurl.SIZE_UPLOAD)
|
|
dictconninfo["size_download"] = conn.getinfo(pycurl.SIZE_DOWNLOAD)
|
|
dictconninfo["header_size"] = conn.getinfo(pycurl.HEADER_SIZE)
|
|
dictconninfo["request_size"] = conn.getinfo(pycurl.REQUEST_SIZE)
|
|
dictconninfo["speed_upload"] = conn.getinfo(pycurl.SPEED_UPLOAD)
|
|
dictconninfo["speed_download"] = conn.getinfo(pycurl.SPEED_DOWNLOAD)
|
|
dictconninfo["time_connect"] = conn.getinfo(pycurl.CONNECT_TIME)
|
|
dictconninfo["time_pretransfer"] =conn.getinfo(pycurl.PRETRANSFER_TIME)
|
|
return dictconninfo
|
|
|
|
def intecept_succ_get_value(self,urlkey,url,conn):
|
|
dictinfo = {}
|
|
dictinfo["downloadsize"] = urlkey
|
|
dictinfo["url"] = url
|
|
dictinfo["time"] = time.asctime( time.localtime(time.time()))
|
|
dictinfo["result"] = self.build_conninfo_json(conn)
|
|
self.resultlist.append(dictinfo)
|
|
|
|
|
|
def ssl_conn(self,urlkey,url):
|
|
issuer = ()
|
|
conn = pycurl.Curl()
|
|
errdict = {}
|
|
conn.setopt(conn.WRITEFUNCTION, BytesIO().write)
|
|
conn.setopt(conn.SSL_VERIFYPEER, False)
|
|
conn.setopt(conn.OPT_CERTINFO, 1)
|
|
conn.setopt(conn.URL,url)
|
|
conn.perform()
|
|
certs = conn.getinfo(conn.INFO_CERTINFO)
|
|
for cert_info in certs[0]:
|
|
if cert_info[0].lower() == "issuer":
|
|
issuer = cert_info
|
|
break
|
|
if len(issuer) <= 0:
|
|
errdict["status"] = "error"
|
|
errdict["errinfo"] = "Get certificate info error"
|
|
errdict["url"] = url
|
|
errdict["time"] = time.asctime( time.localtime(time.time()))
|
|
self.resultlist.append(errdict)
|
|
elif not re.search(r'CN = Tango[\s\S]*UNTRUST',issuer[1],0):
|
|
errdict["status"] = "error"
|
|
errdict["errinfo"] = "Intercept fail: no Tango cert"
|
|
errdict["url"] = url
|
|
errdict["time"] = time.asctime( time.localtime(time.time()))
|
|
self.resultlist.append(errdict)
|
|
else:
|
|
self.intecept_succ_get_value(urlkey,url,conn)
|
|
conn.close()
|
|
def ssl_intercept(self):
|
|
|
|
for sizefield in self.sizelist:
|
|
self.ssl_conn(sizefield,self.urldict[sizefield])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
while True:
|
|
ssl = SSLBuild()
|
|
ssl.ssl_intercept()
|
|
logpath = "/root/result_self_test/con_traffic_inject/con_traffic_inject_result.log." + time.strftime("%Y-%m-%d",time.localtime())
|
|
with open(logpath,"w+") as f:
|
|
f.write(json.dumps(ssl.resultlist, sort_keys=True, indent=4, separators=(',', ': ')))
|
|
time.sleep(1)
|