115 lines
3.4 KiB
Nim
115 lines
3.4 KiB
Nim
import strutils, times
|
|
|
|
import ../../common/[types, sequence, crypto, utils]
|
|
|
|
proc parseInput*(input: string): seq[string] =
|
|
var i = 0
|
|
while i < input.len:
|
|
|
|
# Skip whitespaces/tabs
|
|
while i < input.len and input[i] in {' ', '\t'}:
|
|
inc i
|
|
if i >= input.len:
|
|
break
|
|
|
|
var arg = ""
|
|
if input[i] == '"':
|
|
# Parse quoted argument
|
|
inc i # Skip opening quote
|
|
|
|
# Add parsed argument when quotation is closed
|
|
while i < input.len and input[i] != '"':
|
|
arg.add(input[i])
|
|
inc i
|
|
|
|
if i < input.len:
|
|
inc i # Skip closing quote
|
|
|
|
else:
|
|
while i < input.len and input[i] notin {' ', '\t'}:
|
|
arg.add(input[i])
|
|
inc i
|
|
|
|
# Add argument to returned result
|
|
if arg.len > 0: result.add(arg)
|
|
|
|
proc parseArgument*(argument: Argument, value: string): TaskArg =
|
|
|
|
var arg: TaskArg
|
|
arg.argType = cast[uint8](argument.argumentType)
|
|
|
|
case argument.argumentType:
|
|
|
|
of INT:
|
|
# Length: 4 bytes
|
|
let intValue = cast[uint32](parseUInt(value))
|
|
arg.data = @[byte(intValue and 0xFF), byte((intValue shr 8) and 0xFF), byte((intValue shr 16) and 0xFF), byte((intValue shr 24) and 0xFF)]
|
|
|
|
of LONG:
|
|
# Length: 8 bytes
|
|
var data = newSeq[byte](8)
|
|
let intValue = cast[uint64](parseUInt(value))
|
|
for i in 0..7:
|
|
data[i] = byte((intValue shr (i * 8)) and 0xFF)
|
|
arg.data = data
|
|
|
|
of BOOL:
|
|
# Length: 1 byte
|
|
if value == "true":
|
|
arg.data = @[1'u8]
|
|
elif value == "false":
|
|
arg.data = @[0'u8]
|
|
else:
|
|
raise newException(ValueError, "Invalid value for boolean argument.")
|
|
|
|
of STRING:
|
|
arg.data = cast[seq[byte]](value)
|
|
|
|
of BINARY:
|
|
# Read file as binary stream
|
|
|
|
discard
|
|
|
|
return arg
|
|
|
|
proc createTask*(cq: Conquest, command: Command, arguments: seq[string]): Task =
|
|
|
|
# Construct the task payload prefix
|
|
var task: Task
|
|
task.taskId = string.toUuid(generateUUID())
|
|
task.listenerId = string.toUuid(cq.interactAgent.listenerId)
|
|
task.timestamp = uint32(now().toTime().toUnix())
|
|
task.command = cast[uint16](command.commandType)
|
|
task.argCount = uint8(arguments.len)
|
|
|
|
var taskArgs: seq[TaskArg]
|
|
|
|
# Add the task arguments
|
|
for i, arg in command.arguments:
|
|
if i < arguments.len:
|
|
taskArgs.add(parseArgument(arg, arguments[i]))
|
|
else:
|
|
if arg.isRequired:
|
|
raise newException(ValueError, "Missing required argument.")
|
|
else:
|
|
# Handle optional argument
|
|
taskArgs.add(parseArgument(arg, ""))
|
|
|
|
task.args = taskArgs
|
|
|
|
# Construct the header
|
|
var taskHeader: Header
|
|
taskHeader.magic = MAGIC
|
|
taskHeader.version = VERSION
|
|
taskHeader.packetType = cast[uint8](MSG_TASK)
|
|
taskHeader.flags = cast[uint16](FLAG_ENCRYPTED)
|
|
taskHeader.size = 0'u32
|
|
taskHeader.agentId = string.toUuid(cq.interactAgent.agentId)
|
|
taskHeader.seqNr = nextSequence(taskHeader.agentId)
|
|
taskHeader.iv = generateIV() # Generate a random IV for AES-256 GCM
|
|
taskHeader.gmac = default(AuthenticationTag)
|
|
|
|
task.header = taskHeader
|
|
|
|
# Return the task object for serialization
|
|
return task |