From ccdff3bbffe84a7502c6ab136f9b75ebca1abeb3 Mon Sep 17 00:00:00 2001
From: fumingwei <fumingwei@geedgenetworks.com>
Date: Sat, 12 Sep 2020 12:59:09 +0800
Subject: [PATCH] =?UTF-8?q?1=E3=80=81=E5=88=A0=E9=99=A4=E6=89=93=E5=8D=B0?=
 =?UTF-8?q?=E4=BF=A1=E6=81=AF=202=E3=80=81=E6=B7=BB=E5=8A=A0=E9=85=8D?=
 =?UTF-8?q?=E7=BD=AE=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 unittest_python/unittest/tsg_diagnose.conf | 120 +++++++++++++++++++++
 unittest_python/unittest/tsg_diagnose.py   |   6 +-
 2 files changed, 121 insertions(+), 5 deletions(-)
 create mode 100644 unittest_python/unittest/tsg_diagnose.conf

diff --git a/unittest_python/unittest/tsg_diagnose.conf b/unittest_python/unittest/tsg_diagnose.conf
new file mode 100644
index 0000000..58b4556
--- /dev/null
+++ b/unittest_python/unittest/tsg_diagnose.conf
@@ -0,0 +1,120 @@
+[test_securityPolicy_bypass]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_securityPolicy_intercept]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_securityPolicy_intercept_certerrExpired]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  
+
+# 50M bps
+
+[test_securityPolicy_intercept_certerrSelf_signed]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_securityPolicy_intercept_certerrUntrusted_root]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_ssl_redirect]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_ssl_block]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_ssl_replace]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_ssl_hijack]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_ssl_insert]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_http_redirect]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_http_block]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_http_replace]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_http_hijack]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_proxyPolicy_http_insert]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_1k]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_4k]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_16k]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_64k]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_256k]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_1M]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_4M]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+[test_https_con_traffic_16M]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
+
+[test_https_con_traffic_64M]
+enabled = 1
+conn_timeout = 1
+max_recv_speed_large = 6553600  # 50M bps
diff --git a/unittest_python/unittest/tsg_diagnose.py b/unittest_python/unittest/tsg_diagnose.py
index 43cf68b..294e3d6 100644
--- a/unittest_python/unittest/tsg_diagnose.py
+++ b/unittest_python/unittest/tsg_diagnose.py
@@ -11,6 +11,7 @@ import ciunittest
 import argparse
 from telegraf.client import TelegrafClient
 import hashlib
+#from configparser import ConfigParser
 
 URLBypass          = 'https://sha384.badssl.self-test.geedge.net'
 URLIntercept       = 'https://sha256.badssl.self-test.geedge.net'
@@ -224,11 +225,6 @@ class ProxyRequestBuild:
         if isSsl == True:      
             certs = self.conn.getinfo(self.conn.INFO_CERTINFO)
         body = self.bodyBuf.getvalue().decode('utf-8')
-        print(url)
-        print(proxy_info_re)
-        print(isSsl)
-        self.conn.close()
-        print (body)
         self._cert_verify(ssl_replace_info_re,certs, isSsl)
         if not re.search(r'EnglishSearchShared', body, 0) and \
                 re.search(r'03C174CD9D809789CCEC18D6F585DF3E', body, 0):