Updated directory structure

2025-07-16 10:33:13 +02:00
parent 668a4984d1
commit aae35ef59d
11 changed files with 30 additions and 31 deletions
--- a/src/server/core/server.nim
+++ b/src/server/core/server.nim
@@ -0,0 +1,156 @@
+import prompt, terminal, argparse
+import strutils, strformat, times, system, tables
+
+import ./[agent, listener]
+import ../db/database
+import ../globals
+import ../../types
+
+#[
+    Argument parsing
+]# 
+var parser = newParser: 
+    help("Conquest Command & Control")
+    nohelpflag()
+
+    command("listener"):
+        help("Manage, start and stop listeners.")
+
+        command("list"): 
+            help("List all active listeners.")
+        command("start"): 
+            help("Starts a new HTTP listener.")
+            option("-i", "--ip", default=some("127.0.0.1"), help="IPv4 address to listen on.", required=false)
+            option("-p", "--port", help="Port to listen on.", required=true)
+            
+            # TODO: Future features:
+            # flag("--dns", help="Use the DNS protocol for C2 communication.")
+            # flag("--doh", help="Use DNS over HTTPS for C2 communication.)
+        command("stop"):
+            help("Stop an active listener.")
+            option("-n", "--name", help="Name of the listener.", required=true)
+    
+    command("agent"): 
+        help("Manage, build and interact with agents.")
+
+        command("list"):
+            help("List all agents.")
+            option("-l", "--listener", help="Name of the listener.")
+
+        command("info"): 
+            help("Display details for a specific agent.")
+            option("-n", "--name", help="Name of the agent.", required=true)
+
+        command("kill"):
+            help("Terminate the connection of an active listener and remove it from the interface.")
+            option("-n", "--name", help="Name of the agent.", required=true)
+            # flag("--self-delete", help="Remove agent executable from target system.")
+
+        command("interact"):
+            help("Interact with an active agent.")
+            option("-n", "--name", help="Name of the agent.", required=true)
+
+        command("build"): 
+            help("Generate a new agent to connect to an active listener.")
+            option("-l", "--listener", help="Name of the listener.", required=true)
+            option("-s", "--sleep", help="Sleep delay in seconds.", default=some("10") )
+            option("-p", "--payload", help="Agent type.\n\t\t\t    ", default=some("monarch"), choices = @["monarch"],)
+
+    command("help"):
+        nohelpflag()
+
+    command("exit"):
+        nohelpflag()
+
+proc handleConsoleCommand(cq: Conquest, args: string) = 
+
+    # Return if no command (or just whitespace) is entered
+    if args.replace(" ", "").len == 0: return
+
+    let date: string = now().format("dd-MM-yyyy HH:mm:ss")
+    cq.writeLine(fgBlue, styleBright, fmt"[{date}] ", resetStyle, styleBright, args)
+
+    try:
+        let opts = parser.parse(args.split(" ").filterIt(it.len > 0))
+
+        case opts.command
+        
+        of "exit": # Exit program 
+            echo "\n"
+            quit(0) 
+
+        of "help": # Display help menu
+            cq.writeLine(parser.help())
+
+        of "listener": 
+            case opts.listener.get.command
+            of "list":
+                cq.listenerList()
+            of "start": 
+                cq.listenerStart(opts.listener.get.start.get.ip, opts.listener.get.start.get.port)
+            of "stop": 
+                cq.listenerStop(opts.listener.get.stop.get.name)
+            else: 
+                cq.listenerUsage()
+
+        of "agent":
+            case opts.agent.get.command
+            of "list":    
+                cq.agentList(opts.agent.get.list.get.listener)
+            of "info":
+                cq.agentInfo(opts.agent.get.info.get.name)
+            of "kill": 
+                cq.agentKill(opts.agent.get.kill.get.name)
+            of "interact":
+                cq.agentInteract(opts.agent.get.interact.get.name) 
+            of "build": 
+                cq.agentBuild(opts.agent.get.build.get.listener, opts.agent.get.build.get.sleep, opts.agent.get.build.get.payload)
+            else: 
+                cq.agentUsage()
+
+    # Handle help flag
+    except ShortCircuit as err:
+        if err.flag == "argparse_help":
+            cq.writeLine(err.help)
+    
+    # Handle invalid arguments
+    except CatchableError: 
+        cq.writeLine(fgRed, styleBright, "[-] ", getCurrentExceptionMsg())
+    
+    cq.writeLine("")
+
+proc header(cq: Conquest) = 
+    cq.writeLine("")
+    cq.writeLine("┏┏┓┏┓┏┓┓┏┏┓┏╋")
+    cq.writeLine("┗┗┛┛┗┗┫┗┻┗ ┛┗ V0.1")
+    cq.writeLine("      ┗  @jakobfriedl")  
+    cq.writeLine("─".repeat(21))
+    cq.writeLine("")
+
+proc startServer*() =
+    # Handle CTRL+C,  
+    proc exit() {.noconv.} = 
+        echo "Received CTRL+C. Type \"exit\" to close the application.\n"    
+
+    setControlCHook(exit)
+
+    # Initialize framework
+    let dbPath: string = "../data/conquest.db"
+    cq = initConquest(dbPath) 
+
+    # Print header
+    cq.header()
+    
+    # Initialize database
+    cq.dbInit()
+    cq.restartListeners()
+    cq.addMultiple(cq.dbGetAllAgents())
+
+    # Main loop
+    while true: 
+        cq.setIndicator("[conquest]> ")
+        cq.setStatusBar(@[("[mode]", "manage"), ("[listeners]", $len(cq.listeners)), ("[agents]", $len(cq.agents))])    
+        cq.showPrompt() 
+ 
+        var command: string = cq.readLine()
+        cq.withOutput(handleConsoleCommand, command)