mirror of https://github.com/aliasrobotics/cai.git
Add new tools with @lidia9 and @luijait2
This commit is contained in:
parent
311280c6c6
commit
92215e06a5
|
|
@ -33,3 +33,12 @@ def cat_file(file_path: str, args: str = "", ctf=None) -> str:
|
|||
"""
|
||||
command = f'cat {args} {file_path} '
|
||||
return run_command(command, ctf=ctf)
|
||||
|
||||
#FileSearchTool
|
||||
#ListDirTool
|
||||
#TextSearchTool
|
||||
#FileAnalysisTool
|
||||
#StringExtractionTool
|
||||
#ReadFileTool
|
||||
#FilePermissionsTool
|
||||
#FileCompressionTool
|
||||
|
|
|
|||
|
|
@ -0,0 +1 @@
|
|||
#GenericLinuxCommandTool
|
||||
|
|
@ -0,0 +1,35 @@
|
|||
from cai.tools.common import run_command
|
||||
"""
|
||||
Here are the netcat tools.
|
||||
"""
|
||||
def netcat(args: str, host: str, port: int, data: str = '', ctf=None) -> str:
|
||||
"""
|
||||
A simple netcat tool to connect to a specified host and port.
|
||||
Args:
|
||||
args: Additional arguments to pass to the netcat command
|
||||
host: The target host to connect to
|
||||
port: The target port to connect to
|
||||
data: Data to send to the host (optional)
|
||||
|
||||
Returns:
|
||||
str: The output of running the netcat command
|
||||
"""
|
||||
if data:
|
||||
command = f'echo "{data}" | nc -w 3 {host} {port} {args}'
|
||||
else:
|
||||
command = f'nc -w 3 {host} {port} {args}'
|
||||
return run_command(command, ctf=ctf)
|
||||
|
||||
def nmap(args: str, target: str, ctf=None) -> str:
|
||||
"""
|
||||
A simple nmap tool to scan a specified target.
|
||||
|
||||
Args:
|
||||
args: Additional arguments to pass to the nmap command
|
||||
target: The target host or IP address to scan
|
||||
|
||||
Returns:
|
||||
str: The output of running the nmap command
|
||||
"""
|
||||
command = f'nmap {args} {target}'
|
||||
return run_command(command, ctf=ctf)
|
||||
|
|
@ -0,0 +1,18 @@
|
|||
from cai.tools.common import run_command
|
||||
"""
|
||||
Here are the nmap tools.
|
||||
"""
|
||||
def nmap(args: str, target: str, ctf=None) -> str:
|
||||
"""
|
||||
A simple nmap tool to scan a specified target.
|
||||
|
||||
Args:
|
||||
args: Additional arguments to pass to the nmap command
|
||||
target: The target host or IP address to scan
|
||||
|
||||
Returns:
|
||||
str: The output of running the nmap command
|
||||
"""
|
||||
command = f'nmap {args} {target}'
|
||||
return run_command(command, ctf=ctf)
|
||||
|
||||
Loading…
Reference in New Issue