Fix formatting for multi-line command output and delete tasks after completion.

This commit is contained in:
Jakob Friedl
2025-05-23 16:02:16 +02:00
parent 5ab9cd302c
commit a8a32668d1
6 changed files with 33 additions and 21 deletions

View File

@@ -26,7 +26,7 @@ proc register*(listener: string): string =
# Register agent to the Conquest server
return waitFor client.postContent(fmt"http://localhost:5555/{listener}/register", $body)
except CatchableError as err:
echo "[-] [REGISTER FAILED]:", err.msg
echo "[-] [register]:", err.msg
quit(0)
finally:
client.close()
@@ -42,7 +42,7 @@ proc getTasks*(listener: string, agent: string): seq[Task] =
except CatchableError as err:
# When the listener is not reachable, don't kill the application, but check in at the next time
echo "[-] [TASK-RETRIEVAL FAILED]:", err.msg
echo "[-] [getTasks]:", err.msg
finally:
client.close()
@@ -57,12 +57,14 @@ proc postResults*(listener, agent: string, task: Task): bool =
let taskJson = %task
echo $taskJson
try:
# Register agent to the Conquest server
discard waitFor client.postContent(fmt"http://localhost:5555/{listener}/{agent}/{task.id}/results", $taskJson)
except CatchableError as err:
# When the listener is not reachable, don't kill the application, but check in at the next time
echo "[-] [RESULTS FAILED]:", err.msg
echo "[-] [postResults]: ", err.msg
finally:
client.close()