55 lines
1.8 KiB
Nim
55 lines
1.8 KiB
Nim
|
import whisky
|
||
|
|
import times, tables
|
||
|
|
import ../views/[sessions, listeners, console, eventlog]
|
||
|
|
import ../../common/[types, utils, serialize, event]
|
||
|
|
export sendHeartbeat
|
||
|
|
|
||
|
|
#[
|
||
|
|
Client -> Server
|
||
|
|
]#
|
||
|
|
proc sendStartListener*(ws: WebSocket, listener: UIListener) =
|
||
|
|
var packer = Packer.init()
|
||
|
|
|
||
|
|
packer.add(cast[uint8](CLIENT_LISTENER_START))
|
||
|
|
packer.add(string.toUUid(listener.listenerId))
|
||
|
|
packer.addDataWithLengthPrefix(string.toBytes(listener.address))
|
||
|
|
packer.add(cast[uint16](listener.port))
|
||
|
|
packer.add(cast[uint8](listener.protocol))
|
||
|
|
|
||
|
|
let data = packer.pack()
|
||
|
|
|
||
|
|
ws.send(Bytes.toString(data), BinaryMessage)
|
||
|
|
|
||
|
|
proc sendStopListener*(ws: WebSocket, listenerId: string) =
|
||
|
|
discard
|
||
|
|
# var packer = Packer.init()
|
||
|
|
|
||
|
|
# packer.add(cast[uint8](CLIENT_LISTENER_STOP))
|
||
|
|
# packer.add(string.toUuid(listenerId))
|
||
|
|
# let data = packer.pack()
|
||
|
|
|
||
|
|
# ws.send(Bytes.toString(data), BinaryMessage)
|
||
|
|
|
||
|
|
# proc sendAgentCommand*(ws: WebSocket, agentId: string, command: string) =
|
||
|
|
# var packer = Packer.init()
|
||
|
|
|
||
|
|
# packer.add(cast[uint8](CLIENT_AGENT_COMMAND))
|
||
|
|
# packer.add(string.toUuid(agentId))
|
||
|
|
# packer.addDataWithLengthPrefix(string.toBytes(command))
|
||
|
|
# let data = packer.pack()
|
||
|
|
|
||
|
|
# ws.send(Bytes.toString(data), BinaryMessage)
|
||
|
|
|
||
|
|
# proc sendAgentBuild*(ws: WebSocket, listenerId: string, sleepDelay: int, sleepMask: SleepObfuscationTechnique, spoofStack: bool, modules: uint32) =
|
||
|
|
# var packer = Packer.init()
|
||
|
|
|
||
|
|
# packer.add(cast[uint8](CLIENT_AGENT_BUILD))
|
||
|
|
# packer.add(string.toUuid(listenerId))
|
||
|
|
# packer.add(cast[uint32](sleepDelay))
|
||
|
|
# packer.add(cast[uint8](sleepMask))
|
||
|
|
# packer.add(cast[uint8](spoofStack))
|
||
|
|
# packer.add(modules)
|
||
|
|
# let data = packer.pack()
|
||
|
|
|
||
|
|
# ws.send(Bytes.toString(data), BinaryMessage)
|
||
|
|
|