admin/conquest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added more websocket commands and started agent generation modal window.

Browse Source
This commit is contained in:
Jakob Friedl
2025-09-23 15:51:57 +02:00
parent d3b37aa4a1
commit d4bdf56224
13 changed files with 326 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
src/client/layout.ini
View File

@@ -1,43 +1,43 @@
[Window][Sessions [Table View]]
Pos=10,43
Size=2016,548
Size=2101,474
Collapsed=0
DockId=0x00000003,0
[Window][Listeners]
Pos=10,593
Size=2528,804
Pos=10,519
Size=2848,1166
Collapsed=0
DockId=0x00000006,0
[Window][Eventlog]
Pos=2028,43
Size=510,548
Pos=2113,43
Size=745,474
Collapsed=0
DockId=0x00000004,0
[Window][Dear ImGui Demo]
Pos=2028,43
Size=510,548
Pos=2113,43
Size=745,474
Collapsed=0
DockId=0x00000004,1
[Window][Dockspace]
Pos=0,0
Size=2548,1407
Size=2868,1695
Collapsed=0
[Window][[FACEDEAD] bob@LAPTOP-02]
Pos=10,593
Size=2528,804
Pos=10,661
Size=2848,1024
Collapsed=0
DockId=0x00000006,1
DockId=0x00000006,3
[Window][[C9D8E7F6] charlie@SERVER-03]
Pos=10,593
Size=2528,804
Pos=10,661
Size=2848,1024
Collapsed=0
DockId=0x00000006,1
DockId=0x00000006,4
[Window][Debug##Default]
Pos=60,60
@@ -45,22 +45,22 @@ Size=400,400
Collapsed=0
[Window][[G1H2I3J5] diana@WORKSTATION-04]
Pos=10,593
Size=2528,804
Pos=10,372
Size=1888,617
Collapsed=0
DockId=0x00000006,1
[Window][[DEADBEEF] alice@DESKTOP-01]
Pos=10,716
Size=2848,969
Pos=10,661
Size=2848,1024
Collapsed=0
DockId=0x00000006,2
[Window][Example: Console]
Pos=10,572
Size=2848,1113
Pos=10,661
Size=2848,1024
Collapsed=0
DockId=0x00000005,2
DockId=0x00000006,2
[Window][Example: Assets Browser]
Pos=60,60
@@ -110,8 +110,17 @@ Size=76,76
Collapsed=0
[Window][Start Listener]
Pos=955,591
Size=637,225
Pos=713,369
Size=500,225
Collapsed=0
[Window][Dear ImGui Demo/0_E1BADA21]
IsChild=1
Size=1363,540
[Window][Generate Payload]
Pos=704,337
Size=500,325
Collapsed=0
[Table][0x32886A44,8]
@@ -136,9 +145,9 @@ Column 3 Weight=0.9746
[Docking][Data]
DockNode ID=0x00000009 Pos=100,200 Size=754,103 Selected=0x64D005CF
DockSpace ID=0x85940918 Window=0x260A4489 Pos=10,43 Size=2528,1354 Split=Y
DockNode ID=0x00000005 Parent=0x85940918 SizeRef=1888,548 Split=X
DockNode ID=0x00000003 Parent=0x00000005 SizeRef=1376,159 CentralNode=1 Selected=0x61E02D75
DockNode ID=0x00000004 Parent=0x00000005 SizeRef=510,159 Selected=0x5E5F7166
DockNode ID=0x00000006 Parent=0x85940918 SizeRef=1888,804 Selected=0x6BE22050
DockSpace ID=0x85940918 Window=0x260A4489 Pos=10,43 Size=2848,1642 Split=Y
DockNode ID=0x00000005 Parent=0x85940918 SizeRef=1888,474 Split=X
DockNode ID=0x00000003 Parent=0x00000005 SizeRef=2101,159 CentralNode=1 Selected=0x61E02D75
DockNode ID=0x00000004 Parent=0x00000005 SizeRef=745,159 Selected=0x0FA43D88
DockNode ID=0x00000006 Parent=0x85940918 SizeRef=1888,1166 Selected=0x6BE22050

2
src/client/main.nim
View File

@@ -62,7 +62,7 @@ proc main() =
case message.getMessageType()
of CLIENT_EVENT_LOG:
message.receiveEventlogItem(addr eventlog)
else: discard
else: discard
# Draw/update UI components/views

22
src/client/views/listeners.nim
View File

@@ -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)

0
src/client/views/modals/buildAgent.nim
View File

107
src/client/views/modals/generatePayload.nim Normal file
View File

@@ -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()

1
src/client/views/modals/startListener.nim
View File

@@ -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()

4
src/client/websocket.nim
View File

@@ -65,7 +65,7 @@ proc sendAgentBuild*(ws: WebSocket, listenerId: string, sleepDelay: int, sleepMa
ws.send(Bytes.toString(data), BinaryMessage)
#[
#[
[ Retrieval Functions ]
Server -> Client
]#
@@ -97,7 +97,7 @@ proc receiveAgentConnection*(message: Message, sessions: ptr SessionsTableCompon
sleep: int(unpacker.getUint32()),
tasks: @[],
firstCheckin: cast[int64](unpacker.getUint32()).fromUnix().utc(),
latestCheckin: now(),
latestCheckin: cast[int64](unpacker.getUint32()).fromUnix().utc(),
)
sessions.agents.add(agent)