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

Improved module selection in payload generation modal with tooltips from the module manager.

Browse Source
This commit is contained in:
Jakob Friedl
2025-09-27 12:36:59 +02:00
parent 9de4ac4838
commit 933a72f920
13 changed files with 55 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -32,14 +32,16 @@ proc draw*(component: ListenersTableComponent, showComponent: ptr bool, ws: WebS
if igButton("Start Listener", vec2(0.0f, 0.0f)):
igOpenPopup_str("Start Listener", ImGui_PopupFlags_None.int32)
igSameLine(0.0f, textSpacing)
# Payload generation modal
# Payload generation modal (only enabled when at least one listener is active)
igBeginDisabled(component.listeners.len() <= 0)
if igButton("Generate Payload", vec2(0.0f, 0.0f)):
igOpenPopup_str("Generate Payload", ImGui_PopupFlags_None.int32)
igEndDisabled()
let listener = component.startListenerModal.draw()
if listener != nil:
ws.sendStartListener(listener)
component.listeners.add(listener)
component.generatePayloadModal.draw(component.listeners)

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

@@ -2,6 +2,7 @@ import strutils, sequtils
import imguin/[cimgui, glfw_opengl, simple]
import ../../utils/[appImGui, colors]
import ../../../common/[types, profile, utils]
import ../../../modules/manager
import ../widgets/dualListSelection
type
@@ -11,7 +12,7 @@ type
sleepMask: int32
spoofStack: bool
sleepMaskTechniques: seq[string]
moduleSelection: DualListSelectionComponent[ModuleType]
moduleSelection: DualListSelectionComponent[Module]
proc AgentModal*(): AgentModalComponent =
result = new AgentModalComponent
@@ -23,14 +24,17 @@ proc AgentModal*(): AgentModalComponent =
for technique in SleepObfuscationTechnique.low .. SleepObfuscationTechnique.high:
result.sleepMaskTechniques.add($technique)
var modules: seq[ModuleType]
for module in ModuleType:
modules.add(module)
let modules = getModules()
proc moduleName(module: Module): string =
return module.name
proc moduleDesc(module: Module): string =
result = module.description & "\nModule commands:\n"
for cmd in module.commands:
result &= " - " & cmd.name & "\n"
proc compareModules(x, y: Module): int =
return cmp(x.name, y.name)
proc moduleName(module: ModuleType): string =
return ($module).split("_")[1..^1].mapIt(it.toLowerAscii().capitalizeAscii()).join("")
result.moduleSelection = DualListSelection(modules, moduleName)
result.moduleSelection = DualListSelection(modules, moduleName, compareModules, moduleDesc)
proc resetModalValues(component: AgentModalComponent) =
component.listener = 0
@@ -119,7 +123,7 @@ proc draw*(component: AgentModalComponent, listeners: seq[UIListener]) =
# Iterate over modules
var module: uint32 = 0
for m in component.moduleSelection.items[1]:
module = module or uint32(m)
module = module or uint32(m.moduleType)
echo module
component.resetModalValues()

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

@@ -1,6 +1,6 @@
import strutils, sequtils, algorithm
import imguin/[cimgui, glfw_opengl, simple]
import ../../utils/[appImGui, colors]
import ../../utils/[appImGui, colors, utils]
import ../../../common/[types, utils]
type
@@ -8,36 +8,40 @@ type
items*: array[2, seq[T]]
selection: array[2, ptr ImGuiSelectionBasicStorage]
display: proc(item: T): string
compare: proc(x, y: T): int
tooltip: proc(item: T): string
proc defaultDisplay[T](item: T): string =
return $item
proc DualListSelection*[T](items: seq[T], display: proc(item: T): string = defaultDisplay): DualListSelectionComponent[T] =
proc DualListSelection*[T](items: seq[T], display: proc(item: T): string = defaultDisplay, compare: proc(x, y: T): int, tooltip: proc(item: T): string = nil): 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
result.compare = compare
result.tooltip = tooltip
proc moveAll[T](component: DualListSelectionComponent[T], src, dst: int) =
for m in component.items[src]:
component.items[dst].add(m)
component.items[dst].sort()
component.items[dst].sort(component.compare)
component.items[src].setLen(0)
ImGuiSelectionBasicStorage_Swap(component.selection[src], component.selection[dst])
ImGuiSelectionBasicStorage_Clear(component.selection[src])
proc moveSelection[T](component: DualListSelectionComponent[T], src, dst: int) =
var keep: seq[ModuleType]
var keep: seq[T]
for i in 0 ..< component.items[src].len():
let item = component.items[src][i]
if not component.selection[src].ImGuiSelectionBasicStorage_Contains(cast[ImGuiID](i)):
keep.add(item)
continue
component.items[dst].add(item)
component.items[dst].sort()
component.items[dst].sort(component.compare)
component.items[src] = keep
ImGuiSelectionBasicStorage_Swap(component.selection[src], component.selection[dst])
@@ -83,6 +87,9 @@ proc draw*[T](component: DualListSelectionComponent[T]) =
var isSelected = ImGuiSelectionBasicStorage_Contains(selection, cast[ImGuiID](row))
igSetNextItemSelectionUserData(row)
discard igSelectable_Bool(component.display(modules[row]), isSelected, ImGuiSelectableFlags_AllowDoubleClick.int32, vec2(0.0f, 0.0f))
if not component.tooltip.isNil():
setTooltip(component.tooltip(modules[row]))
# Move on Enter and double-click
if igIsItemFocused():
@@ -134,6 +141,9 @@ proc draw*[T](component: DualListSelectionComponent[T]) =
igSetNextItemSelectionUserData(row)
discard igSelectable_Bool(component.display(modules[row]), isSelected, ImGuiSelectableFlags_AllowDoubleClick.int32, vec2(0.0f, 0.0f))
if not component.tooltip.isNil():
setTooltip(component.tooltip(modules[row]))
# Move on Enter and double-click
if igIsItemFocused():
if igIsKeyPressed_Bool(ImGuiKey_Enter, false) or igIsKeyPressed_Bool(ImGuiKey_KeypadEnter, false):

1
src/common/types.nim
View File

@@ -318,6 +318,7 @@ type
Module* = object
name*: string
description*: string
moduleType*: ModuleType
commands*: seq[Command]
# Definitions for ImGui User interface

3
src/modules/bof.nim
View File

@@ -7,6 +7,7 @@ proc executeBof(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("bof"),
description: protect("Load and execute BOF/COFF files in memory."),
moduleType: MODULE_BOF,
commands: @[
Command(
name: protect("bof"),
@@ -23,7 +24,7 @@ let module* = Module(
)
# Implement execution functions
when defined(server):
when not defined(agent):
proc executeBof(ctx: AgentCtx, task: Task): TaskResult = nil
when defined(agent):

3
src/modules/dotnet.nim
View File

@@ -7,6 +7,7 @@ proc executeAssembly(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("dotnet"),
description: protect("Load and execute .NET assemblies in memory."),
moduleType: MODULE_DOTNET,
commands: @[
Command(
name: protect("dotnet"),
@@ -23,7 +24,7 @@ let module* = Module(
)
# Implement execution functions
when defined(server):
when not defined(agent):
proc executeAssembly(ctx: AgentCtx, task: Task): TaskResult = nil
when defined(agent):

3
src/modules/filesystem.nim
View File

@@ -13,6 +13,7 @@ proc executeCopy(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("filesystem"),
description: protect("Conduct simple filesystem operations via Windows API."),
moduleType: MODULE_DOTNET,
commands: @[
Command(
name: protect("pwd"),
@@ -88,7 +89,7 @@ let module* = Module(
)
# Implementation of the execution functions
when defined(server):
when not defined(agent):
proc executePwd(ctx: AgentCtx, task: Task): TaskResult = nil
proc executeCd(ctx: AgentCtx, task: Task): TaskResult = nil
proc executeDir(ctx: AgentCtx, task: Task): TaskResult = nil

3
src/modules/filetransfer.nim
View File

@@ -8,6 +8,7 @@ proc executeUpload(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("filetransfer"),
description: protect("Upload/download files to/from the target system."),
moduleType: MODULE_FILESYSTEM,
commands: @[
Command(
name: protect("download"),
@@ -33,7 +34,7 @@ let module* = Module(
)
# Implement execution functions
when defined(server):
when not defined(agent):
proc executeDownload(ctx: AgentCtx, task: Task): TaskResult = nil
proc executeUpload(ctx: AgentCtx, task: Task): TaskResult = nil

7
src/modules/manager.nim
View File

@@ -5,12 +5,14 @@ const MODULES {.intdefine.} = 0
type
ModuleManager* = object
modules*: seq[Module]
commandsByType*: Table[CommandType, Command]
commandsByName*: Table[string, Command]
var manager: ModuleManager
proc registerModule(module: Module) {.discardable.} =
manager.modules.add(module)
for cmd in module.commands:
manager.commandsByType[cmd.commandType] = cmd
manager.commandsByName[cmd.name] = cmd
@@ -71,4 +73,7 @@ proc getCommandByName*(cmdName: string): Command =
raise newException(ValueError, fmt"The command '{cmdName}' does not exist.")
proc getAvailableCommands*(): Table[string, Command] =
return manager.commandsByName
return manager.commandsByName
proc getModules*(): seq[Module] =
return manager.modules

3
src/modules/screenshot.nim
View File

@@ -7,6 +7,7 @@ proc executeScreenshot(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("screenshot"),
description: protect("Take and retrieve a screenshot of the target desktop."),
moduleType: MODULE_SCREENSHOT,
commands: @[
Command(
name: protect("screenshot"),
@@ -20,7 +21,7 @@ let module* = Module(
)
# Implement execution functions
when defined(server):
when not defined(agent):
proc executeScreenshot(ctx: AgentCtx, task: Task): TaskResult = nil
when defined(agent):

3
src/modules/shell.nim
View File

@@ -7,6 +7,7 @@ proc executeShell(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("shell"),
description: protect("Execute shell commands or programs."),
moduleType: MODULE_SHELL,
commands: @[
Command(
name: protect("shell"),
@@ -23,7 +24,7 @@ let module* = Module(
)
# Implement execution functions
when defined(server):
when not defined(agent):
proc executeShell(ctx: AgentCtx, task: Task): TaskResult = nil
when defined(agent):

3
src/modules/situationalAwareness.nim
View File

@@ -9,6 +9,7 @@ proc executeWhoami(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("situational-awareness"),
description: protect("Retrieve information about the target system and environment."),
moduleType: MODULE_SITUATIONAL_AWARENESS,
commands: @[
Command(
name: protect("ps"),
@@ -38,7 +39,7 @@ let module* = Module(
)
# Implement execution functions
when defined(server):
when not defined(agent):
proc executePs(ctx: AgentCtx, task: Task): TaskResult = nil
proc executeEnv(ctx: AgentCtx, task: Task): TaskResult = nil
proc executeWhoami(ctx: AgentCtx, task: Task): TaskResult = nil

3
src/modules/sleep.nim
View File

@@ -8,6 +8,7 @@ proc executeSleepmask(ctx: AgentCtx, task: Task): TaskResult
let module* = Module(
name: protect("sleep"),
description: protect("Change sleep settings."),
moduleType: MODULE_SLEEP,
commands: @[
Command(
name: protect("sleep"),
@@ -34,7 +35,7 @@ let module* = Module(
)
# Implement execution functions
when defined(server):
when not defined(agent):
proc executeSleep(ctx: AgentCtx, task: Task): TaskResult = nil
proc executeSleepmask(ctx: AgentCtx, task: Task): TaskResult = nil