104 lines
3.0 KiB
Nim
104 lines
3.0 KiB
Nim
import argparse, times, strformat, terminal, nanoid
|
|
import ./task
|
|
import ../[types]
|
|
|
|
#[
|
|
Agent Argument parsing
|
|
]#
|
|
var parser = newParser:
|
|
help("Conquest Command & Control")
|
|
|
|
command("shell"):
|
|
help("Execute a shell command and retrieve the output.")
|
|
arg("command", help="Command", nargs = 1)
|
|
arg("arguments", help="Arguments.", nargs = -1) # Handle 0 or more command-line arguments (seq[string])
|
|
|
|
command("sleep"):
|
|
help("Update sleep delay configuration.")
|
|
arg("delay", help="Delay in seconds.", nargs = 1)
|
|
|
|
command("info"):
|
|
help("Display agent information and current settings.")
|
|
|
|
command("pwd"):
|
|
help("Retrieve current working directory.")
|
|
|
|
command("cd"):
|
|
help("Change current working directory.")
|
|
arg("directory", help="Relative or absolute path of the directory to change to.", nargs = -1)
|
|
|
|
command("ls"):
|
|
help("List files and directories.")
|
|
arg("directory", help="Relative or absolute path. Default: current working directory.", nargs = -1)
|
|
|
|
command("rm"):
|
|
help("Remove file.")
|
|
arg("file", help="Relative or absolute path to the file to delete.", nargs = -1)
|
|
|
|
command("rmdir"):
|
|
help("Remove directory.")
|
|
arg("directory", help="Relative or absolute path to the directory to delete.", nargs = -1)
|
|
|
|
command("help"):
|
|
nohelpflag()
|
|
|
|
command("back"):
|
|
nohelpflag()
|
|
|
|
proc handleAgentCommand*(cq: Conquest, args: varargs[string]) =
|
|
|
|
# Return if no command (or just whitespace) is entered
|
|
if args[0].replace(" ", "").len == 0: return
|
|
|
|
let date: string = now().format("dd-MM-yyyy HH:mm:ss")
|
|
cq.writeLine(fgBlue, styleBright, fmt"[{date}] ", fgYellow, fmt"[{cq.interactAgent.name}] ", resetStyle, styleBright, args[0])
|
|
|
|
try:
|
|
let opts = parser.parse(args[0].split(" ").filterIt(it.len > 0))
|
|
|
|
case opts.command
|
|
|
|
of "back": # Return to management mode
|
|
discard
|
|
|
|
of "help": # Display help menu
|
|
cq.writeLine(parser.help())
|
|
|
|
of "shell":
|
|
var
|
|
command: string = opts.shell.get.command
|
|
arguments: seq[string] = opts.shell.get.arguments
|
|
arguments.insert(command, 0)
|
|
cq.taskExecuteShell(arguments)
|
|
|
|
of "sleep":
|
|
cq.taskExecuteSleep(parseInt(opts.sleep.get.delay))
|
|
|
|
of "info":
|
|
discard
|
|
|
|
of "pwd":
|
|
cq.taskGetWorkingDirectory()
|
|
|
|
of "cd":
|
|
cq.taskSetWorkingDirectory(opts.cd.get.directory)
|
|
|
|
of "ls":
|
|
cq.taskListDirectory(opts.ls.get.directory)
|
|
|
|
of "rm":
|
|
cq.taskRemoveFile(opts.rm.get.file)
|
|
|
|
of "rmdir":
|
|
cq.taskRemoveDirectory(opts.rmdir.get.directory)
|
|
|
|
# Handle help flag
|
|
except ShortCircuit as err:
|
|
if err.flag == "argparse_help":
|
|
cq.writeLine(err.help)
|
|
|
|
# Handle invalid arguments
|
|
except CatchableError:
|
|
cq.writeLine(fgRed, styleBright, "[-] ", getCurrentExceptionMsg(), "\n")
|
|
|