Websocket长连接接口渗透测试过程随记

关注

原创 2023-12-13 19:24:30 1.7k 阅读 AI 写同款· GEO 优化 ›

WebSocket接口测试方法 WebSocket 是一种基于在单个 TCP 连接上进行全双工通信的协议，是从HTML5开始提供的一种浏览器与服务器之间进行全双工通讯的网络技术，解决了HTTP协议不适用于实时通信的缺点，相较于 HTTP 协议，WebSocket 协议实现了持久化网络通信，可以实现客户端和服务端的长连接，能够进行双向实时通信，协议名为"ws"。二者关系在线工具有很多，自行百度吧！个人习惯用这个工具Postman支持测试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 websocketfrom websocket._exceptions import *from urllib.parse import urlparseimport 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()