本文为CS 4.4 二开笔记系列第二篇，主要是针对aggressor客户端进行功能分析与二开。感觉现在的查到的文章较多在于beacon的协议的分析和针对特征的修改，在aggressor的分析还是比较少的，而aggressor又是使用者进行一系列操作的入口，只掌握sleep来编写cna脚本是远远不够的，因此本文将着重讨论针对aggressor功能的分析与扩充，文中可能会对部分步骤进行省略，主要是提供足够的思路来分析aggressor的源码和增强。

之前在实战中遇到的问题是经常忘了连接的是哪台服务器的teamserver，后来分析了一下cscat 4.5的源码，他的设置方式为通过aggressor进行设置，因为我在开发时没有打算在aggressor处设置配置文件，因此使用了不同的方式实现显示，先看一下效果：

此处涉及到cna脚本的编写及aggressor的源码，cna脚本的编写可以参考下面的官方链接及狼组的中文文档，本文中不再赘述：
cs插件开发 - 先知社区
Aggressor-Script | 狼组安全团队公开知识库
https://hstechdocs.helpsystems.com/manuals/cobaltstrike/current/userguide/content/topics_aggressor-scripts/as_cobalt-strike.htm
cna脚本会在点击时检索脚本中是否存在相应的函数，如果不存在则进行函数提交，交给java端进行判断及相应，在这里我修改了默认的default.cna脚本，新增了serverip()函数来进行显示：

这个函数的实现并不在cna脚本中，因此在触发时脚本会向aggressor发送serverip这个字符串，然后aggressor会判断是否等于这个字符串来进行响应。aggressor的代码与下面两部分有关：

钉钉提醒网上常用的方案为调用agscript来执行python操作，再进行提醒。我感觉这样过于麻烦，因此直接将此功能集成至server端中，只需要在server端的配置文件中填入sever端的配置文件，就可以默认启动，类似

server端日志显示：

创建方法如下：
新建一个类：

接着在server/Beacons.java中调用这个类，我这里是通过配置文件进行的设置，其他方式大家可以自己研究：

上线效果就是这样：

这个功能需要涉及到数据库的保存，这里使用的是jdbc，效果如下：

在新增这个功能时，发现了cscat 4.5的一个bug，就是无法正确的显示last，后来发现主要原因在于server/Beacons.java的checkin函数循环导致无法正确刷新map。目前更改后的checkin代码如下：

数据库的实现：

在护网时遇到比较多的一个问题，就是开着热点上的cs，中途去吃饭了结果回来再连上热点就会显示该用户已使用，导致就要重启客户端，因此在此处新增用户名后缀防止重复登录。
主要修改点在server/ManageUser.java中，在process函数中新增：

这个随机字符串我控制在长度为3，对应实现的类为：

最终效果就是这样，终于解决了这个烦人的问题：

在原版CS中，每次新增监听器时默认值为本地的IP地址，还需要手动改为远程IP地址，因此在这里对这部分代码进行修改：
Connect.java：新增public static String aggressorRemoteIp = "";


然后在ScListenerDialog.java修改显示参数：

show_http和show_https都改掉：

最后更改效果为：

这个操作就跟上面 Event log显示远程IP差不多，主要单独列出来方便直接看。
default.cna:

调用的是openConnectDialog()，这个在java端的aggressor/bridges/AggressorBridges.java中进行判断，如果点击了这个，则调用这个文件中的代码。AggressorBridges.java先是注册，再是响应：
注册：

响应：

最后执行的代码为：

这个代码是在aggressor/dialogs/ConnectDialog.java中的类。实际上aggressor界面上的那些按钮的触发都是这个套路，因此我们后续如果不想用纯粹的cna脚本，就可以用cna脚本触发，函数写死在java端，好处就是把一些不改动的cna脚本写死在里面，不用拷贝给队友时还需要额外附带脚本。cna脚本写死并加载的地方在aggressor/AggressorClient.java中：

这个没什么绕的地方，主要就是分析代码，我直接贴提取方式了：

我之所以弄这个，是之前想将免杀模块放在本地来进行操作，后来这个方案被我废弃了，打算将免杀流程放在teamserver端进行操作，再回传至aggressor。

发送信号我是分析的钓鱼模块，aggressor交互都在aggressor/dialogs里面，发送信号使用的是TeamQueue类，CS的通信主要用的就是这个类：

其中"SendSign.test"代表的是发送数据的标签，服务端需要根据这个标签进行判断，后面的几个都是传递的参数。我自己实现的调用逻辑为：
dialogAction() -> send() -> final_send()

与dialog.java成对存在的就是server目录中的文件，我这自己新建了一个类，具体接收信号的函数为：

其中比较重要的就是var1.is("SendSign.test", 4)，这个与aggressor传来的数据是相对应的，4为四个参数。

广播的代码为：

这个true的含义貌似是aggressor重连后还会接收到这个信息，我改为false也没什么变化，广播的数据为Map类型。

接收广播部分有个巨坑，如果直接使用TeamQueue提供的setSubscriber回调函数来接收广播，则会导致CS的UI不会刷新。存在问题的代码为：

这个代码虽然可以正确接收，但是UI不刷新肯定是存在问题的，后来换了一个方案：

在使用这个时，一定要注意是否data中存在这个键值对，如果不存在就会空指针异常，因此我在dialogAction中去检索data数据前先使用this.data.put("key","value");进行了置空操作，问题就解决了。

以上是基于下面的模型进行讲解的，掌握这个模型一系列操作方式后CS的可玩性还是非常高的。

本人也是刚刚学习java开发，文中可能存在诸多问题，欢迎大家指出，最后祝大家国庆快乐~

[注意]传递专业知识、拓宽行业人脉——看雪讲师团队等你加入！