add update ca and curl py file

7 files changed, 74 insertions, 3 deletions

diff --git a/install_curl.py b/install_curl.py
new file mode 100644
index 0000000..b49558f
--- /dev/null
+++ b/install_curl.py
@@ -0,0 +1,20 @@
+# -*- coding: UTF-8 -*-
+import os
+import subprocess
+
+def install_curl():
+    try:
+        subprocess.check_call(["wget", "https://curl.se/download/curl-7.68.0.tar.gz"])
+        subprocess.check_call(["tar", "-xzvf", "curl-7.68.0.tar.gz"])
+        os.chdir("/opt/curl-7.68.0")
+        subprocess.check_call(["export", "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/ssl/lib"])
+        subprocess.check_call(["./configure", "--prefix=/usr/local/curl/", "--without-nss", "--with-ssl=/usr/local/ssl/"])
+        subprocess.check_call(["make && make install"])
+        subprocess.check_call(["mv", "/usr/bin/curl", "/usr/bin/curl.bak"])
+        subprocess.check_call(["ln", "-s", "/usr/local/curl/bin/curl", "/usr/bin/curl"])
+        print("Install curl successfully.")
+    except:
+        print("Fail to install curl.")
+        
+if __name__ == '__main__':
+    install_curl()
\ No newline at end of file
diff --git a/mail_file/mail_test_english.txt b/mail_file/mail_test_english.txt
new file mode 100644
index 0000000..707d5e1
--- /dev/null
+++ b/mail_file/mail_test_english.txt
@@ -0,0 +1 @@
+halashaogusina
diff --git a/mail_file/subjectEnglish.txt b/mail_file/subjectEnglish.txt
new file mode 100644
index 0000000..850d049
--- /dev/null
+++ b/mail_file/subjectEnglish.txt
@@ -0,0 +1,5 @@
+Cc: [email protected]
+Bcc: [email protected]
+From: [email protected]
+To: [email protected]
+Subject: Bestman
diff --git a/start.sh b/start.sh
index a55d221..1c9625f 100644
--- a/start.sh
+++ b/start.sh
@@ -5,7 +5,7 @@
 if [[ "$(command -v python3)" == "" ]]; then
     echo "No python3, start to installing."
     yum -y update
-    yum install -y net-tools wget python3-devel python3 python3-pip pciutils bind-utils
+    yum install -y net-tools wget python3-devel python3 python3-pip pciutils bind-utils curl
     pip3 install --no-cache-dir scapy
     pip3 install --no-cache-dir PyYAML
     pip3 install --no-cache-dir requests
@@ -27,6 +27,8 @@ git checkout develop
 find . -type f -name "*.py" -exec chmod 777 {} \;
 
 python3 /opt/traffic_replay/install_trex.py
+python3 /opt/traffic_replay/install_curl.py
+python3 /opt/traffic_replay/update_tsg_ca.py
 python3 /opt/traffic_replay/kill_gunicorn.py
 # python3 /opt/traffic_replay/kill_start.py
 # python3 /opt/traffic_replay/app.py
diff --git a/start_gunicorn.sh b/start_gunicorn.sh
index 2a71ee4..a62594f 100644
--- a/start_gunicorn.sh
+++ b/start_gunicorn.sh
@@ -1,6 +1,5 @@
 #!/bin/sh
 
-python3 /opt/traffic_replay/install_trex.py
-# python3 /opt/traffic_replay/app.py
+python3 /opt/traffic_replay/kill_gunicorn.py
 gunicorn -c gunicorn.conf.py app:app &
 # /usr/local/bin/supervisord -c /opt/traffic_replay/supervisor.conf
\ No newline at end of file
diff --git a/tsg_ca.txt b/tsg_ca.txt
new file mode 100644
index 0000000..0615ea1
--- /dev/null
+++ b/tsg_ca.txt
@@ -0,0 +1,22 @@
+#TSG CA
+-----BEGIN CERTIFICATE-----
+MIIDeTCCAmGgAwIBAgIJAKkZA2wHOYFDMA0GCSqGSIb3DQEBCwUAMC0xDzANBgNV
+BAMMBlRTRyBDQTEaMBgGA1UECgwRTWFzZXJhdGkgU29sdXRpb24wHhcNMjIwMjE2
+MDg0OTMyWhcNNDIwMjExMDg0OTMyWjAtMQ8wDQYDVQQDDAZUU0cgQ0ExGjAYBgNV
+BAoMEU1hc2VyYXRpIFNvbHV0aW9uMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIB
+CgKCAQEAyAfT3h/At7OaSH7/hqvh5H+NFMJAsuM6RsB812idxrNdnEldIJb1sh4J
+JbPViI/hhCrB6e3rx27oeyFsxFPRmFK+dEpIHI9pQra+ZT8BN1G2seRMOuiZ7p/J
+tKtyff9ejwpVzk3nGrRRsFJ7lRHaXZy7vE2oS+ne9re8eYcSg1aBv34uOljwHjPH
+k/8iM1pPcHLUdUdRrH1R+SutIx3PciwI0FjBGIB+yUTFpxlcGTySdFOHiSiJESdD
+4k8/2bkSy80Iu8mgmeEL4zPxcLb4nGyxvi/SZNceVBiT27Y+Ry89Ml83C0ERnApY
+a9XULndiCgHaGla1BLOOiEjAM71mrwIDAQABo4GbMIGYMBEGCWCGSAGG+EIBAQQE
+AwICBDBJBgNVHSUEQjBABggrBgEFBQcDAQYIKwYBBQUHAwIGCCsGAQUFBwMEBggr
+BgEFBQcDCAYKKwYBBAGCNwoDAQYKKwYBBAGCNwoDBDAdBgNVHQ4EFgQUmZAagCLQ
+uMZ/T7bKkhL/3jzlgacwDAYDVR0TBAUwAwEB/zALBgNVHQ8EBAMCAQYwDQYJKoZI
+hvcNAQELBQADggEBACIxLmfZdlncpBdv5q+aBFXaoPacrcQ+Sp6+mzkeasgmW3PU
+dfgJlTqJXNdG9D3VWCuWMzVsDDlP0tKKOXCXJ2mU3WwbfVKddQ/AP4b4jQannH4A
+6bkZXm+kwJH40oOymPT+AwVXKcrmHor31j5yIKiffvimEtH511i1iPMxE4KmwN/o
+ntHfww14Pk3ICCt7dMJsiTHHp9GU+XlcjwmzGEgBoD3B+8CYhYOk5kT3XbuopSXw
+98XTsZHy83xPOsJ1p6nhOP9dlh1/2G0KBOe1kvn8s1gP5ojUPpYvf2MzHw90Pnpy
+LhiVVAWOET2MANXMLaJisLI3/XPfmPqQUFza5os=
+-----END CERTIFICATE-----
\ No newline at end of file
diff --git a/update_tsg_ca.py b/update_tsg_ca.py
new file mode 100644
index 0000000..ec2523a
--- /dev/null
+++ b/update_tsg_ca.py
@@ -0,0 +1,22 @@
+# -*- coding: UTF-8 -*-
+import os
+import subprocess
+
+def update_tsg_ca():
+    try:
+        tsg_ca_file = 'tsg_ca.txt'  
+        # 读取证书文件内容  
+        with open(tsg_ca_file, 'r') as cert_file:  
+            cert_content = cert_file.read()  
+        # 将证书内容追加到ca-bundle.crt文件中
+        with open('/etc/pki/tls/certs/ca-bundle.crt', 'a') as ca_bundle:
+            ca_bundle.write(cert_content)  
+            ca_bundle.write('\n')
+        # 更新证书信任库
+        # subprocess.run(['sudo', 'update-ca-trust', 'extract'])
+        print("Update TSG CA successfully.")
+    except:
+        print("Fail to update TSG CA.")
+        
+if __name__ == '__main__':
+    update_tsg_ca()
\ No newline at end of file