纪录片：Websocket长连接接口渗透测试过程

# 1 前言

最近接到公司的一个API接口测试的项目，除了常规的HTTP协议的API接口外，其中还有一部分是需要测试Websocket接口的。

回想了一下除了之前复现Jumpserver漏洞时候用过Websocket，以外就没有对它有过什么研究了。

这次也算是第一次与它交手，开始测试的时候就不是很顺利，但是好在最后解决了，故在此做下简单的记录。

---

# 2 希望达到的目标

能够使用Burpsuite来测试Websocket接口。

---

# 3 测试过程

1、使用Burpsuite自带的Websocket建立连接（失败）

---

2、使用浏览器插件测试（失败）

这里使用的是一款叫做“WebSocket King”的chrome插件进行测试的。

但是使用插件多次连接WS接口发现都是403，研究了一番后发现，目标会检测Origin头，如果不是自己站点，就直接返回403，因此需要用Burp来修改请求头。

修改了请求头后，浏览器插件终于可以正常连接到目标了，可以看到已经收到了来自服务器的Ping包。

这里要说明一下，该Websocket接口存在心跳检测的机制，服务端会定期向客户端发送ping包，需要客户端回复pong包来证明存活。

然而在我寻找了插件的各个角落，也没有发现存在自动回复pong包的功能，很遗憾这种方法失败。

---

3、既然没有现成的工具（也许是我没找到），只能自己造了，思路非常简单，如下：

• 能够自动给服务器回复pong包

• 能够将流量打到Burpsuite上

• 使用Python编写代码

基于以上思路，开始搜索python相关的websocket的教程，最后发现利用“websocket-client”库，可以很便捷的实现我的需求，代码如下：

import websocket``from websocket._exceptions import *``from urllib.parse import urlparse``import json`

`url = "wss://xxxxxxxx.com/xxxxxxxx"`
`# proxy = ""``proxy = "http://127.0.0.1:8080"`
`options = {}`
`if proxy != "":`    `p = urlparse(proxy)`    `options["http_proxy_host"] = p.hostname`    `options["http_proxy_port"] = p.port`
`ws = websocket.create_connection(url, **options)`
`while True:`    `try :`        `data = ws.recv()`        `print(data)`        `j = json.loads(data)`        `if j["type"] == "ping":`            `send_data = '{"type":"pong","data":"%s"}' % j["data"]`            `print(send_data)`            `ws.send(send_data)`
    `except WebSocketConnectionClosedException:`        `print("[!] Connection to remote host was lost.")`        `break`
`ws.close()

ps：临时使用的代码，请不要在意编写细节

使用起来还是很方便的，脚本中的代理服务器指向Burpsuite，然后开始运行，发现数据包ping、pong正常交互。

打开bp，找到repeater，新建一个websocket会话，选择刚刚已经建立连接的Websocket会话，并点击下方的Attach。

接下来就可以开始发送数据，测试这个接口了，自此问题得到解决。

今天的内容就分享到这里，如果你喜欢这篇文章的话，麻烦请动动你的手点个赞或者点个关注吧，这将是我最大的动力！谢谢，这里说点题外话，想学习黑客技术的，这里有我以前整理的大量自学资料。

需要的可以点这里自取:网安入门到进阶资源