Implement listing agents; Add TODOs for further improvements

server/db/database.nim

@@ -21,7 +21,8 @@ proc dbInit*(cq: Conquest) =
 
         CREATE TABLE agents (
             name TEXT PRIMARY KEY,                   
-            listener TEXT NOT NULL,                    
+            listener TEXT NOT NULL, 
+            process TEXT NOT NULL,                   
             pid INTEGER NOT NULL,
             username TEXT NOT NULL,
             hostname TEXT NOT NULL,
@@ -30,6 +31,7 @@ proc dbInit*(cq: Conquest) =
             elevated BOOLEAN NOT NULL,
             sleep INTEGER DEFAULT 10,
             jitter REAL DEFAULT 0.1,
+            firstCheckin DATETIME NOT NULL,
             FOREIGN KEY (listener) REFERENCES listeners(name)
         );
 
@@ -40,6 +42,9 @@ proc dbInit*(cq: Conquest) =
     except SqliteError: 
         cq.writeLine(fgGreen, "[+] ", cq.dbPath, ": Database file found.")
 
+#[
+    Listeners
+]#
 proc dbStoreListener*(cq: Conquest, listener: Listener): bool = 
 
     try: 
@@ -108,15 +113,18 @@ proc listenerExists*(cq: Conquest, listenerName: string): bool =
         cq.writeLine(fgRed, styleBright, "[-] ", getCurrentExceptionMsg())
         return false
 
+#[
+    Agents
+]#
 proc dbStoreAgent*(cq: Conquest, agent: Agent): bool = 
     
     try: 
         let conquestDb = openDatabase(cq.dbPath, mode=dbReadWrite)
 
         conquestDb.exec("""
-        INSERT INTO agents (name, listener, sleep, jitter, pid,username, hostname, ip, os, elevated)
-        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?);
-        """, agent.name, agent.listener, agent.sleep, agent.jitter, agent.pid, agent.username, agent.hostname, agent.ip, agent.os, agent.elevated)
+        INSERT INTO agents (name, listener, sleep, jitter, process, pid, username, hostname, ip, os, elevated, firstCheckin)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);
+        """, agent.name, agent.listener, agent.sleep, agent.jitter, agent.process, agent.pid, agent.username, agent.hostname, agent.ip, agent.os, agent.elevated, $agent.firstCheckin)
 
         conquestDb.close() 
     except: 
@@ -125,3 +133,36 @@ proc dbStoreAgent*(cq: Conquest, agent: Agent): bool =
     
     return true
 
+proc dbGetAllAgents*(cq: Conquest): seq[Agent] = 
+
+    var agents: seq[Agent] = @[]
+
+    try: 
+        let conquestDb = openDatabase(cq.dbPath, mode=dbReadWrite)
+
+        for row in conquestDb.iterate("SELECT name, listener, sleep, jitter, process, pid, username, hostname, ip, os, elevated, firstCheckin FROM agents;"):
+            let (name, listener, sleep, jitter, process, pid, username, hostname, ip, os, elevated, firstCheckin) = row.unpack((string, string, int, float, string, int, string, string, string, string, bool, string))
+
+            let a = Agent(
+                    name: name,
+                    listener: listener,
+                    sleep: sleep,
+                    jitter: jitter,
+                    process: process,
+                    pid: pid,
+                    username: username,
+                    hostname: hostname,
+                    ip: ip,
+                    os: os,
+                    elevated: elevated,
+                    firstCheckin: firstCheckin,
+                    tasks: @[] 
+                )
+
+            agents.add(a)
+
+        conquestDb.close()
+    except: 
+        cq.writeLine(fgRed, styleBright, "[-] ", getCurrentExceptionMsg())
+
+    return agents