Added more websocket commands and started agent generation modal window.

src/client/views/listeners.nim

@@ -2,7 +2,8 @@ import strutils
 import imguin/[cimgui, glfw_opengl, simple]
 import ../utils/appImGui
 import ../../common/[types, utils]
-import ./modals/startListener
+import ./modals/[startListener, generatePayload]
+import ../websocket
 import whisky
 
 type 
@@ -11,6 +12,7 @@ type
         listeners*: seq[Listener]
         selection: ptr ImGuiSelectionBasicStorage
         startListenerModal: ListenerModalComponent
+        generatePayloadModal: AgentModalComponent
 
 let exampleListeners: seq[Listener] = @[
     Listener(
@@ -33,6 +35,8 @@ proc ListenersTable*(title: string): ListenersTableComponent =
     result.listeners = exampleListeners
     result.selection = ImGuiSelectionBasicStorage_ImGuiSelectionBasicStorage()
     result.startListenerModal = ListenerModal()
+    result.generatePayloadModal = AgentModal(result.listeners)
+
 
 proc draw*(component: ListenersTableComponent, showComponent: ptr bool, ws: WebSocket) = 
     igBegin(component.title, showComponent, 0)
@@ -43,13 +47,21 @@ proc draw*(component: ListenersTableComponent, showComponent: ptr bool, ws: WebS
     # Listener creation modal
     if igButton("Start Listener", vec2(0.0f, 0.0f)):          
         igOpenPopup_str("Start Listener", ImGui_PopupFlags_None.int32) 
+    igSameLine(0.0f, textSpacing)
+    # Payload generation modal
+    if igButton("Generate Payload", vec2(0.0f, 0.0f)):          
+        igOpenPopup_str("Generate Payload", ImGui_PopupFlags_None.int32) 
 
     let listener = component.startListenerModal.draw()
     if listener != nil: 
-        # TODO: Start listener
-        ws.send("Starting listener: " & listener.listenerId)
+        ws.sendStartListener(listener)
         component.listeners.add(listener)    
 
+    component.generatePayloadModal.draw()
+
+
+
+
     #[
         Listener table
     ]#
@@ -109,8 +121,8 @@ proc draw*(component: ListenersTableComponent, showComponent: ptr bool, ws: WebS
                 for i in 0 ..< component.listeners.len():
                     if not ImGuiSelectionBasicStorage_Contains(component.selection, cast[ImGuiID](i)):
                         newListeners.add(component.listeners[i])
-
-                # TODO: Stop/kill listener
+                    else: 
+                        ws.sendStopListener(component.listeners[i].listenerId)
 
                 component.listeners = newListeners
                 ImGuiSelectionBasicStorage_Clear(component.selection)

src/client/views/modals/generatePayload.nim

@@ -0,0 +1,107 @@
+import strutils
+import imguin/[cimgui, glfw_opengl, simple]
+import ../../utils/appImGui
+import ../../../common/[types, utils]
+
+
+type 
+    AgentModalComponent* = ref object of RootObj
+        listener: int32 
+        sleepDelay: uint32 
+        sleepMask: int32 
+        spoofStack: bool 
+        listeners: seq[string]
+        sleepMaskTechniques: seq[string]
+
+proc AgentModal*(listeners: seq[Listener]): AgentModalComponent =
+    result = new AgentModalComponent
+    result.listener = 0
+    result.sleepDelay = 5
+    result.sleepMask = 0
+    result.spoofStack = false
+
+    for l in listeners: 
+        result.listeners.add(l.listenerId)
+
+    for s in SleepObfuscationTechnique.low .. SleepObfuscationTechnique.high:
+        result.sleepMaskTechniques.add($s)
+
+proc resetModalValues(component: AgentModalComponent) = 
+    discard 
+
+proc draw*(component: AgentModalComponent) =
+    let textSpacing = igGetStyle().ItemSpacing.x    
+    
+    # Center modal
+    let vp = igGetMainViewport()
+    var center: ImVec2
+    ImGuiViewport_GetCenter(addr center, vp)
+    igSetNextWindowPos(center, ImGuiCond_Appearing.int32, vec2(0.5f, 0.5f))
+
+    let modalWidth = max(500.0f, vp.Size.x * 0.25)
+    igSetNextWindowSize(vec2(modalWidth, 0.0f), ImGuiCond_Always.int32)
+    
+    var show = true
+    let windowFlags = ImGuiWindowFlags_None.int32 # or ImGuiWindowFlags_NoMove.int32
+    if igBeginPopupModal("Generate Payload", addr show, windowFlags):
+        defer: igEndPopup()
+        
+        var availableSize: ImVec2
+        igGetContentRegionAvail(addr availableSize)
+
+        # Listener selection
+        igText("Listener:       ")
+        igSameLine(0.0f, textSpacing)
+        igGetContentRegionAvail(addr availableSize)
+        igSetNextItemWidth(availableSize.x)
+        igCombo_Str("##InputListener", addr component.listener, (component.listeners.join("\0") & "\0").cstring , component.listeners.len().int32)
+
+        # Sleep delay
+        let step: uint32 = 1
+        igText("Sleep delay:    ")
+        igSameLine(0.0f, textSpacing)
+        igSetNextItemWidth(availableSize.x)
+        igInputScalar("##InputSleepDelay", ImGuiDataType_U32.int32, addr component.sleepDelay, addr step, nil, "%hu", ImGui_InputTextFlags_CharsDecimal.int32)
+
+        # Agent sleep obfuscation technique dropdown selection
+        igText("Sleep mask:     ")
+        igSameLine(0.0f, textSpacing)
+        igSetNextItemWidth(availableSize.x)
+        igCombo_Str("##InputSleepMask", addr component.sleepMask, (component.sleepMaskTechniques.join("\0") & "\0").cstring , component.sleepMaskTechniques.len().int32)
+
+        # Stack spoofing checkbox (only for EKKO/ZILEAN)
+        igText("Stack spoofing: ")
+        igSameLine(0.0f, textSpacing)
+        igSetNextItemWidth(availableSize.x)
+
+        igBeginDisabled((component.sleepMaskTechniques[component.sleepMask] != $EKKO and component.sleepMaskTechniques[component.sleepMask] != $ZILEAN))
+        if (component.sleepMaskTechniques[component.sleepMask] != $EKKO and component.sleepMaskTechniques[component.sleepMask] != $ZILEAN):
+            component.spoofStack = false
+        igCheckbox("##InputSpoofStack", addr component.spoofStack)
+        igEndDisabled()
+
+        igDummy(vec2(0.0f, 10.0f))
+        igSeparator()
+        igDummy(vec2(0.0f, 10.0f))
+
+        igText("Modules: ")
+        
+
+
+        igGetContentRegionAvail(addr availableSize)
+
+        igDummy(vec2(0.0f, 10.0f))
+        igSeparator()
+        igDummy(vec2(0.0f, 10.0f))
+
+        if igButton("Build", vec2(availableSize.x * 0.5 - textSpacing * 0.5, 0.0f)):
+            
+            
+            component.resetModalValues()
+            igCloseCurrentPopup() 
+        
+        igSameLine(0.0f, textSpacing)
+
+        if igButton("Close", vec2(availableSize.x * 0.5 - textSpacing * 0.5, 0.0f)):
+            component.resetModalValues()
+            igCloseCurrentPopup()

src/client/views/modals/startListener.nim

@@ -89,6 +89,5 @@ proc draw*(component: ListenerModalComponent): Listener =
         igSameLine(0.0f, textSpacing)
 
         if igButton("Close", vec2(availableSize.x * 0.5 - textSpacing * 0.5, 0.0f)):
-            
             component.resetModalValues()
             igCloseCurrentPopup()