From bb26db3f7f9d68ffffdbd397e7495550af478014 Mon Sep 17 00:00:00 2001 From: Mery-Sanz Date: Fri, 10 Jan 2025 15:53:17 +0000 Subject: [PATCH] Try to remove blank files --- .../command_and_control/command_and_control.py | 0 .../data_exfiltration/data_exfiltration.py | 0 cai/tools/reconnaissance/curl.py | 18 +++++++++++++++++- .../2_picoctf_reversing_python.py | 8 +++++--- 4 files changed, 22 insertions(+), 4 deletions(-) delete mode 100644 cai/tools/command_and_control/command_and_control.py delete mode 100644 cai/tools/data_exfiltration/data_exfiltration.py diff --git a/cai/tools/command_and_control/command_and_control.py b/cai/tools/command_and_control/command_and_control.py deleted file mode 100644 index e69de29b..00000000 diff --git a/cai/tools/data_exfiltration/data_exfiltration.py b/cai/tools/data_exfiltration/data_exfiltration.py deleted file mode 100644 index e69de29b..00000000 diff --git a/cai/tools/reconnaissance/curl.py b/cai/tools/reconnaissance/curl.py index 0191627f..6ec543af 100644 --- a/cai/tools/reconnaissance/curl.py +++ b/cai/tools/reconnaissance/curl.py @@ -1 +1,17 @@ -#curl \ No newline at end of file +from cai.tools.common import run_command +""" +Here are the curl tools. +""" +def curl(args: str, target: str, ctf=None) -> str: + """ + A simple curl tool to make HTTP requests to a specified target. + + Args: + args: Additional arguments to pass to the curl command + target: The target URL to request + + Returns: + str: The output of running the curl command + """ + command = f'curl {args} {target}' + return run_command(command, ctf=ctf) \ No newline at end of file diff --git a/examples/cybersecurity/2_picoctf_reversing_python.py b/examples/cybersecurity/2_picoctf_reversing_python.py index d6ea4f59..6dbb968e 100644 --- a/examples/cybersecurity/2_picoctf_reversing_python.py +++ b/examples/cybersecurity/2_picoctf_reversing_python.py @@ -13,16 +13,16 @@ import os # 2. Init the CTF environment ctf = ptt.ctf( "picoctf_reversing_python", - subnet="192.168.9.0/24", + subnet="192.168.2.0/24", container_name="ancient", - ip_address="192.168.9.52" + ip_address="192.168.2.52" ) ctf.start_ctf() challenges = list(ctf.get_challenges().keys()) challenge = challenges[0] # 3. Instance the cybersecurity AI Framework client -client = CAI(ctf=ctf) +client = CAI() # 4. Instance the agent/agents to solve the CTF ctf_agent = Agent( @@ -39,6 +39,8 @@ messages = [{ "content": "Instructions: " + ctf.get_instructions() + "\nChallenge: " + ctf.get_challenges()[challenge] + "\nTechniques: " + ctf.get_techniques() + + "\nTarget IP: " + ctf.get_ip() + + "\nTarget Port: 5555" "\nExtract and return only the flag" "\nYOU MUST RETURN THE FLAG IN PLAIN TEXT" }]