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

Made dual list selection widget generic.

Browse Source
This commit is contained in:
Jakob Friedl
2025-09-25 10:01:49 +02:00
parent 8baf65a96d
commit a4456723ce
4 changed files with 75 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
src/client/layout.ini
View File

@@ -1,43 +1,43 @@
[Window][Sessions [Table View]]
Pos=10,43
Size=1141,338
Size=1310,279
Collapsed=0
DockId=0x00000003,0
[Window][Listeners]
Pos=10,383
Size=1888,654
Pos=10,324
Size=1888,665
Collapsed=0
DockId=0x00000006,0
[Window][Eventlog]
Pos=1153,43
Size=745,338
Pos=1322,43
Size=576,279
Collapsed=0
DockId=0x00000004,0
[Window][Dear ImGui Demo]
Pos=1153,43
Size=745,338
Pos=1322,43
Size=576,279
Collapsed=0
DockId=0x00000004,1
[Window][Dockspace]
Pos=0,0
Size=1908,1047
Size=1908,999
Collapsed=0
[Window][[FACEDEAD] bob@LAPTOP-02]
Pos=10,661
Size=2848,1024
Pos=10,395
Size=1888,594
Collapsed=0
DockId=0x00000006,3
DockId=0x00000006,1
[Window][[C9D8E7F6] charlie@SERVER-03]
Pos=10,661
Size=2848,1024
Pos=10,324
Size=1888,665
Collapsed=0
DockId=0x00000006,4
DockId=0x00000006,1
[Window][Debug##Default]
Pos=60,60
@@ -45,16 +45,16 @@ Size=400,400
Collapsed=0
[Window][[G1H2I3J5] diana@WORKSTATION-04]
Pos=10,372
Size=1888,617
Pos=10,125
Size=784,665
Collapsed=0
DockId=0x00000006,1
[Window][[DEADBEEF] alice@DESKTOP-01]
Pos=10,661
Size=2848,1024
Pos=10,324
Size=1888,665
Collapsed=0
DockId=0x00000006,2
DockId=0x00000006,1
[Window][Example: Console]
Pos=10,661
@@ -119,23 +119,23 @@ IsChild=1
Size=1363,540
[Window][Generate Payload]
Pos=704,185
Size=500,677
Pos=1075,509
Size=717,677
Collapsed=0
[Window][Generate Payload/0_B6B17D5F]
IsChild=1
Size=217,310
Size=326,310
[Table][0x32886A44,8]
Column 0 Weight=0.6465
Column 1 Weight=0.9697
Column 2 Weight=0.6696
Column 3 Weight=0.9639
Column 4 Weight=1.7143
Column 5 Weight=1.1544
Column 6 Weight=0.4271
Column 7 Weight=1.4546
Column 0 Weight=0.6432
Column 1 Weight=0.9647
Column 2 Weight=0.6694
Column 3 Weight=1.0960
Column 4 Weight=1.5816
Column 5 Weight=1.1551
Column 6 Weight=0.4331
Column 7 Weight=1.4570
[Table][0xB6880529,2]
RefScale=27
@@ -149,9 +149,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=1888,994 Split=Y
DockNode ID=0x00000005 Parent=0x85940918 SizeRef=1888,338 Split=X
DockNode ID=0x00000003 Parent=0x00000005 SizeRef=2101,159 CentralNode=1 Selected=0x61E02D75
DockNode ID=0x00000004 Parent=0x00000005 SizeRef=745,159 Selected=0x5E5F7166
DockNode ID=0x00000006 Parent=0x85940918 SizeRef=1888,654 Selected=0x6BE22050
DockSpace ID=0x85940918 Window=0x260A4489 Pos=10,43 Size=1888,946 Split=Y
DockNode ID=0x00000005 Parent=0x85940918 SizeRef=1888,279 Split=X
DockNode ID=0x00000003 Parent=0x00000005 SizeRef=1310,159 CentralNode=1 Selected=0x61E02D75
DockNode ID=0x00000004 Parent=0x00000005 SizeRef=576,159 Selected=0x5E5F7166
DockNode ID=0x00000006 Parent=0x85940918 SizeRef=1888,665 Selected=0x65D642C0

35
src/client/views/console.nim
View File

@@ -172,20 +172,33 @@ proc draw*(component: ConsoleComponent, ws: WebSocket) =
Huge thanks to @dinau for implementing ImGuiTextSelect into imguin very rapidly after I requested it.
]#
let consolePadding: float = 10.0f
let footerHeight = (consolePadding * 2) + (igGetStyle().ItemSpacing.y + igGetFrameHeightWithSpacing()) * 1.5f
let footerHeight = (consolePadding * 2) + (igGetStyle().ItemSpacing.y + igGetFrameHeightWithSpacing()) * 0.75f
let textSpacing = igGetStyle().ItemSpacing.x
# Padding
igDummy(vec2(0.0f, consolePadding))
#[
Session information
]#
let domain = if component.agent.domain.isEmptyOrWhitespace(): "" else: fmt".{component.agent.domain}"
let sessionInfo = fmt"{component.agent.username}@{component.agent.hostname}{domain} | {component.agent.ip} | {$component.agent.pid}/{component.agent.process}"
igTextColored(GRAY, sessionInfo)
igSameLine(0.0f, 0.0f)
#[
Filter & Options
]#
var availableSize: ImVec2
igGetContentRegionAvail(addr availableSize)
var labelSize: ImVec2
igCalcTextSize(addr labelSize, ICON_FA_MAGNIFYING_GLASS, nil, false, 0.0f)
igSameLine(0.0f, igGetWindowWidth() - 200.0f - (labelSize.x + textSpacing) - (igGetStyle().WindowPadding.x * 2))
# SHow tooltip when hovering the search icon
let searchBoxWidth: float32 = 200.0f
igSameLine(0.0f, availableSize.x - (labelSize.x + textSpacing) - searchBoxWidth)
# Show tooltip when hovering the search icon
igTextUnformatted(ICON_FA_MAGNIFYING_GLASS.cstring, nil)
if igIsItemHovered(ImGuiHoveredFlags_None.int32):
igBeginTooltip()
@@ -199,7 +212,7 @@ proc draw*(component: ConsoleComponent, ws: WebSocket) =
igSetKeyboardFocusHere(0)
igSameLine(0.0f, textSpacing)
component.filter.ImGuiTextFilter_Draw("##ConsoleSearch", 200.0f)
component.filter.ImGuiTextFilter_Draw("##ConsoleSearch", searchBoxWidth)
try:
# Set styles of the console window
@@ -245,9 +258,8 @@ proc draw*(component: ConsoleComponent, ws: WebSocket) =
igSameLine(0.0f, textSpacing)
# Calculate available width for input
var availableWidth: ImVec2
igGetContentRegionAvail(addr availableWidth)
igSetNextItemWidth(availableWidth.x)
igGetContentRegionAvail(addr availableSize)
igSetNextItemWidth(availableSize.x)
let inputFlags = ImGuiInputTextFlags_EnterReturnsTrue.int32 or ImGuiInputTextFlags_EscapeClearsAll.int32 or ImGuiInputTextFlags_CallbackHistory.int32 or ImGuiInputTextFlags_CallbackCompletion.int32
if igInputText("##Input", addr component.inputBuffer[0], MAX_INPUT_LENGTH, inputFlags, callback, cast[pointer](component)):
@@ -277,11 +289,4 @@ proc draw*(component: ConsoleComponent, ws: WebSocket) =
igSetItemDefaultFocus()
if focusInput:
igSetKeyboardFocusHere(-1)
#[
Session information
]#
let sessionInfo = fmt"{component.agent.username}@{component.agent.hostname}.{component.agent.domain} [{component.agent.ip}] [{component.agent.process}/{$component.agent.pid}]"
igTextColored(vec4(0.75f, 0.75f, 0.75f, 1.0f), sessionInfo)
igSetKeyboardFocusHere(-1)

8
src/client/views/modals/generatePayload.nim
View File

@@ -12,7 +12,7 @@ type
spoofStack: bool
listeners: seq[string]
sleepMaskTechniques: seq[string]
moduleSelection: DualListSelectionComponent
moduleSelection: DualListSelectionComponent[ModuleType]
proc AgentModal*(listeners: seq[Listener]): AgentModalComponent =
result = new AgentModalComponent
@@ -30,7 +30,11 @@ proc AgentModal*(listeners: seq[Listener]): AgentModalComponent =
var modules: seq[ModuleType]
for module in ModuleType:
modules.add(module)
result.moduleSelection = DualListSelection(modules)
proc moduleName(module: ModuleType): string =
return ($module).split("_")[1..^1].mapIt(it.toLowerAscii().capitalizeAscii()).join("")
result.moduleSelection = DualListSelection(modules, moduleName)
proc resetModalValues(component: AgentModalComponent) =
discard

26
src/client/views/widgets/dualListSelection.nim
View File

@@ -4,18 +4,23 @@ import ../../utils/[appImGui, colors]
import ../../../common/[types, utils]
type
DualListSelectionComponent* = ref object of RootObj
items*: array[2, seq[ModuleType]]
DualListSelectionComponent*[T] = ref object of RootObj
items*: array[2, seq[T]]
selection: array[2, ptr ImGuiSelectionBasicStorage]
display: proc(item: T): string
proc DualListSelection*(items: seq[ModuleType]): DualListSelectionComponent =
result = new DualListSelectionComponent
proc defaultDisplay[T](item: T): string =
return $item
proc DualListSelection*[T](items: seq[T], display: proc(item: T): string = defaultDisplay): DualListSelectionComponent[T] =
result = new DualListSelectionComponent[T]
result.items[0] = items
result.items[1] = @[]
result.selection[0] = ImGuiSelectionBasicStorage_ImGuiSelectionBasicStorage()
result.selection[1] = ImGuiSelectionBasicStorage_ImGuiSelectionBasicStorage()
result.display = display
proc moveAll(component: DualListSelectionComponent, src, dst: int) =
proc moveAll[T](component: DualListSelectionComponent[T], src, dst: int) =
for m in component.items[src]:
component.items[dst].add(m)
component.items[dst].sort()
@@ -24,7 +29,7 @@ proc moveAll(component: DualListSelectionComponent, src, dst: int) =
ImGuiSelectionBasicStorage_Swap(component.selection[src], component.selection[dst])
ImGuiSelectionBasicStorage_Clear(component.selection[src])
proc moveSelection(component: DualListSelectionComponent, src, dst: int) =
proc moveSelection[T](component: DualListSelectionComponent[T], src, dst: int) =
var keep: seq[ModuleType]
for i in 0 ..< component.items[src].len():
let item = component.items[src][i]
@@ -38,10 +43,7 @@ proc moveSelection(component: DualListSelectionComponent, src, dst: int) =
ImGuiSelectionBasicStorage_Swap(component.selection[src], component.selection[dst])
ImGuiSelectionBasicStorage_Clear(component.selection[src])
proc moduleName(module: ModuleType): string =
return ($module).split("_")[1..^1].mapIt(it.toLowerAscii().capitalizeAscii()).join("")
proc draw*(component: DualListSelectionComponent) =
proc draw*[T](component: DualListSelectionComponent[T]) =
if igBeginTable("split", 3, ImGuiTableFlags_None.int32, vec2(0.0f, 0.0f), 0.0f):
@@ -77,7 +79,7 @@ proc draw*(component: DualListSelectionComponent) =
for row in 0 ..< modules.len().int32:
var isSelected = ImGuiSelectionBasicStorage_Contains(selection, cast[ImGuiID](row))
igSetNextItemSelectionUserData(row)
discard igSelectable_Bool(modules[row].moduleName(), isSelected, ImGuiSelectableFlags_AllowDoubleClick.int32, vec2(0.0f, 0.0f))
discard igSelectable_Bool(component.display(modules[row]), isSelected, ImGuiSelectableFlags_AllowDoubleClick.int32, vec2(0.0f, 0.0f))
# Move on Enter and double-click
if igIsItemFocused():
@@ -127,7 +129,7 @@ proc draw*(component: DualListSelectionComponent) =
for row in 0 ..< modules.len().int32:
var isSelected = ImGuiSelectionBasicStorage_Contains(selection, cast[ImGuiID](row))
igSetNextItemSelectionUserData(row)
discard igSelectable_Bool(modules[row].moduleName(), isSelected, ImGuiSelectableFlags_AllowDoubleClick.int32, vec2(0.0f, 0.0f))
discard igSelectable_Bool(component.display(modules[row]), isSelected, ImGuiSelectableFlags_AllowDoubleClick.int32, vec2(0.0f, 0.0f))
# Move on Enter and double-click
if igIsItemFocused():