admin/conquest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated ps command output.

Browse Source
This commit is contained in:
Jakob Friedl
2025-08-06 12:46:53 +02:00
parent dfcafa9c24
commit ea00e67e80
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/modules/environment.nim
View File

@@ -58,7 +58,6 @@ when defined(agent):
echo fmt" [>] Listing running processes." echo fmt" [>] Listing running processes."
try: try:
var processes: seq[DWORD] = @[] var processes: seq[DWORD] = @[]
var procMap = initTable[DWORD, ProcessInfo]() var procMap = initTable[DWORD, ProcessInfo]()
var output: string = "" var output: string = ""
@@ -98,6 +97,11 @@ when defined(agent):
else: else:
processes.add(pid) processes.add(pid)
# Add header row
let headers = @["PID", "PPID", "Process"]
output &= fmt"{headers[0]:<10}{headers[1]:<10}{headers[2]:<25}" & "\n"
output &= "-".repeat(len(headers[0])).alignLeft(10) & "-".repeat(len(headers[1])).alignLeft(10) & "-".repeat(len(headers[2])).alignLeft(25) & "\n"
# Format and print process # Format and print process
proc printProcess(pid: DWORD, indentSpaces: int = 0) = proc printProcess(pid: DWORD, indentSpaces: int = 0) =
if not procMap.contains(pid): if not procMap.contains(pid):
@@ -106,9 +110,7 @@ when defined(agent):
var process = procMap[pid] var process = procMap[pid]
let indent = " ".repeat(indentSpaces) let indent = " ".repeat(indentSpaces)
let tree = (indent & fmt"[{process.pid}]").alignLeft(30) output &= fmt"{process.pid:<10}{process.ppid:<10}{indent}{process.name:<25}" & "\n"
output &= fmt"{tree}{process.name:<25}" & "\n"
# Recursively print child processes with indentation # Recursively print child processes with indentation
process.children.sort() process.children.sort()
@@ -125,6 +127,7 @@ when defined(agent):
except CatchableError as err: except CatchableError as err:
return createTaskResult(task, STATUS_FAILED, RESULT_STRING, err.msg.toBytes()) return createTaskResult(task, STATUS_FAILED, RESULT_STRING, err.msg.toBytes())
import sugar
proc executeEnv(config: AgentConfig, task: Task): TaskResult = proc executeEnv(config: AgentConfig, task: Task): TaskResult =
echo fmt" [>] Displaying environment variables." echo fmt" [>] Displaying environment variables."
@@ -132,7 +135,8 @@ when defined(agent):
try: try:
var envVars: string = "" var envVars: string = ""
for key, value in envPairs(): for key, value in envPairs():
envVars &= fmt"{key}: {value}" & '\n' envVars &= fmt"{key}: {value}" & '\n'
return createTaskResult(task, STATUS_COMPLETED, RESULT_STRING, envVars.toBytes()) return createTaskResult(task, STATUS_COMPLETED, RESULT_STRING, envVars.toBytes())
except CatchableError as err: except CatchableError as err:
@@ -143,8 +147,9 @@ when defined(agent):
echo fmt" [>] Getting user information." echo fmt" [>] Getting user information."
try: try:
echo "whoami"
let message = "Not implemented"
return createTaskResult(task, STATUS_FAILED, RESULT_STRING, message.toBytes())
except CatchableError as err: except CatchableError as err:
return createTaskResult(task, STATUS_FAILED, RESULT_STRING, err.msg.toBytes()) return createTaskResult(task, STATUS_FAILED, RESULT_STRING, err.msg.toBytes())