ToB企服应用市场:ToB评测及商务社交产业平台

标题: Havoc插件编写 [打印本页]
作者: 火影    时间: 2024-3-23 03:57
标题: Havoc插件编写
‍
配置文件的webhook支持discord，所以尝试使用钉钉和企业微信。
  1. WebHook {
  2.         Discord {
  3.                 Url = ""
  4.                 AvatarUrl = ""
  5.                 User = "announcer"
  6.         }
  7. }
复制代码
服务端中判断如果配置了webhook会在自身添加agent之前就转发给discord了。
  1. func (t *Teamserver) AgentAdd(Agent *agent.Agent) []*agent.Agent {
  2.         if Agent != nil {
  3.                 if t.WebHooks != nil {
  4.                         t.WebHooks.NewAgent(Agent.ToMap())
  5.                 }
  6.         }
  7. ...
  8. }
复制代码
可以看到在上线时，server的控制台上会有上线的信息。
​​
借鉴一个老哥的做法：起个子程序来开服务端，同时监听并捕获这个信息：
  1.     process = subprocess.Popen(['./havoc', 'server', '--profile', './profiles/havoc.yaotl', '-v', '--debug'],
  2.                                stdout=subprocess.PIPE,
  3.                                stderr=subprocess.STDOUT,
  4.                                text=True)
  5.   
  6.     capture = False
  8. // 获取到前四行即可
  9.         if "[DBUG] [agent.ParseDemonRegisterRequest:382]" in line:
  10.             capture = True
  11.             captured_text = ""
  12.             line_count = 0
  13.             continue
  15.         if capture:
  16.             if line_count < 5:  
  17.                 captured_text += line + '\n'
  18.                 line_count += 1
  19.             else: #
  20.                 send_messages('New connection!\n'+captured_text.strip())
  21.                 capture = False
复制代码
然后根据官方文档 发送text类型的消息，markdown可以但是在微信中不能正常显示。
​​
上面是markdown下面是text文本，代码也上传了：gayhub
​​​​
但这样并不是很好，而翻官方文档，里面有提供对客户端api的详细说明，主要涉及到havoc和havocui这两个。
​​
对于ui可以直接在客户端的console中尝试他的效果：
​​
像获取demons的数量可以用havoc.GetDemons()​
文档中介绍了一些比较常用的api，在\client\src\Havoc\PythonApi​有更多的调用方向，比如下面的：
  1. PyMemberDef PyDemonClass_members[] = {
  3.         { "Listener",       T_STRING, offsetof( PyDemonClass, Listener ),    0, "Listener name" },
  4.         { "DemonID",        T_STRING, offsetof( PyDemonClass, DemonID ),     0, "Listener name" },
  5.         { "ExternalIP",     T_STRING, offsetof( PyDemonClass, ExternalIP ),  0, "External IP" },
  6.         { "InternalIP",     T_STRING, offsetof( PyDemonClass, InternalIP ),  0, "Internal IP" },
  7.         { "User",           T_STRING, offsetof( PyDemonClass, User ),        0, "Username" },
  8.         { "Computer",       T_STRING, offsetof( PyDemonClass, Computer ),    0, "Computer" },
  9.         { "Domain",         T_STRING, offsetof( PyDemonClass, Domain ),      0, "Domain" },
  10.         { "OS",             T_STRING, offsetof( PyDemonClass, OS ),          0, "Windows Version" },
  11.         { "OSBuild",        T_STRING, offsetof( PyDemonClass, OSBuild ),     0, "Windows OS Build" },
  12.         { "OSArch",         T_STRING, offsetof( PyDemonClass, OSArch ),      0, "Windows Architecture" },
  13.         { "ProcessName",    T_STRING, offsetof( PyDemonClass, ProcessName ), 0, "Process Name" },
  14.         { "ProcessID",      T_STRING, offsetof( PyDemonClass, ProcessID ),   0, "Process ID" },
  15.         { "ProcessArch",    T_STRING, offsetof( PyDemonClass, ProcessArch ), 0, "Process Architecture" },
  17.         { "CONSOLE_INFO",   T_INT, offsetof( PyDemonClass, CONSOLE_INFO ),   0, "Console message type info" },
  18.         { "CONSOLE_ERROR",  T_INT, offsetof( PyDemonClass, CONSOLE_ERROR ),  0, "Console message type error" },
  19.         { "CONSOLE_TASK",   T_INT, offsetof( PyDemonClass, CONSOLE_TASK ),   0, "Console message type task" },
  21.         { NULL },
  22. };
  24. PyMethodDef PyDemonClass_methods[] = {
  26.         { "ConsoleWrite",           ( PyCFunction ) DemonClass_ConsoleWrite,           METH_VARARGS, "Prints messages to the demon sessions console" },
  27.         { "ProcessCreate",          ( PyCFunction ) DemonClass_ProcessCreate,          METH_VARARGS, "Creates a Process" },
  28.         { "InlineExecute",          ( PyCFunction ) DemonClass_InlineExecute,          METH_VARARGS, "Executes a coff file in the context of the demon sessions" },
  29.         { "InlineExecuteGetOutput", ( PyCFunction ) DemonClass_InlineExecuteGetOutput, METH_VARARGS, "Executes a coff file in the context of the demon sessions and get the output via a callback" },
  30.         { "DllSpawn",               ( PyCFunction ) DemonClass_DllSpawn,               METH_VARARGS, "Spawn and injects a reflective dll and get output from it" },
  31.         { "DllInject",              ( PyCFunction ) DemonClass_DllInject,              METH_VARARGS, "Injects a reflective dll into a specified process" },
  32.         { "DotnetInlineExecute",    ( PyCFunction ) DemonClass_DotnetInlineExecute,    METH_VARARGS, "Executes a dotnet assembly in the context of the demon sessions" },
  33.         { "Command",                ( PyCFunction ) DemonClass_Command,                METH_VARARGS, "Run a command" },
  34.         { "CommandGetOutput",       ( PyCFunction ) DemonClass_CommandGetOutput,                METH_VARARGS, "Run a command and retreive the output" },
  35.         { "ShellcodeSpawn",         ( PyCFunction ) DemonClass_ShellcodeSpawn,         METH_VARARGS, "Executes shellcode spawning a new process" },
  37.         { NULL },
  38. };
复制代码
代码的逻辑也很简单，就是通过havoc.Demon(demon_id)​获取到这个对象，抽出这里面的对象发送即可。代码可以去仓库看看。最终完成的效果如下：
​​
最后也是正常能提示了，传送门：gayhub（一起交流）
​​

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！



欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/) Powered by Discuz! X3.4