Python模拟websocket登陆-拆包封包
- 解析一个网站
- 获取wss原始数据
- 拆包wss数据
- 封包wss数据
- 发送接收websocket的常驻后台脚本
- 总结
解析一个网站
这里所用的网站是我一个内测的网站,主要手段是chrome devtools,用得很多,但我玩的不深,这次上了点干货.
- 首先在网络那一块,找到js或者index.html,右键点击,选择替换内容,在需要分析的地方写上自己的代码。
- 对于ajax加载的js模块,把js在加载地址,换成离线下载到本地的地址,存入本地的服务,原来是http的,还还用http。原来https的必须也在https,本地web开启跨域允许。然后ajax加载部分js也就替换成可定制的了。
- 虽然代码很难读,但是在适合的地方,可以随便写cosole.log.
- 对于js的调试和中断不太懂,还没用,
- 界面的记录器标签,可以记录一组点击,然后能看到json代码,用于回放。
- 记录的动作可以用js实现插入在任何位置,只要调试通过就行。
获取wss原始数据
获取数据的两种方式,各种优缺点:
- 在上一章中,可以看到websocket建立和sendBytes,onMessage之类的函数,在这里启动cosole.log就能得到逐条纪录。不论ws还是wss。这里可以看到语义,并可定制测试代码。
- 在devtools网络标签下的wss://server:port/url地址,对应的respone可以取得全部的来往数据,和代码块所收发的字节按道理是完全一致的,可以做为一个验证和参考。
拆包wss数据
所谓的拆包是理解数据的意义,我还没有修炼到靠数据读含义的深度,只能靠代码,也就js,顺眼化处理的代码,比如以下的登陆数据代码,从1万行里拽出来的。
- 消息头4表字节
u.prototype.addHeader = function(e, t, n, a) {
return void 0 === n && (n = 0),
void 0 === a && (a = 0),
e | t << 4 | n << 12 | a << 23
4个参数在长度 e->4bit, t ->,8bit,n->11bit a->9bit.
含义e=command ,t=action, n=length,a=ext
- 登陆消息的主体
var a = new z.SyncLogonDto;
a.account = e.account,
a.sn = e.sn,
a.token = e.token,
a.uid = e.userID,
a.localHost = 0,
4 == a.sn.length && 0 < a.uid && "" != a.token ? this.sendMsg(a.getBody(),
z.ServiceType.HALL_CMD, z.ServiceType.HALL_LOGIN_ACT, 2 =
o.prototype.getBody = function() {
var e = new t.BGByteArray;
return e.writeUnsignedInt(this.major),
e.writeUnsignedInt(this.minor),
e.writeUTFBytes(this.sn),
e.writeUnsignedInt(this.localHost),
e.writeLongUint(this.uid),
var a = new z.SyncLogonDto;
a.account = e.account,
a.sn = e.sn,
a.token = e.token,
a.uid = e.userID,
a.localHost = 0,
4 == a.sn.length && 0 < a.uid && "" != a.token ? this.sendMsg(a.getBody(),
z.ServiceType.HALL_CMD, z.ServiceType.HALL_LOGIN_ACT, 2 =
o.prototype.getBody = function() {
var e = new t.BGByteArray;
return e.writeUnsignedInt(this.major),
e.writeUnsignedInt(this.minor),
e.writeUTFBytes(this.sn),
e.writeUnsignedInt(this.localHost),
e.writeLongUint(this.uid),
e.writeFixedLenthString(this.account, 32),
e.writeFixedLenthString(this.token, 32),
e
}
,
a = o,
t.SyncLogonDto = a
e.writeFixedLenthString(this.token, 32),
e
}
,
a = o,
t.SyncLogonDto = a
主要的处理逻辑在·o.prototype.getBody = function()
这里的writeUnsigedInt是四字节无符号整数,而且是小头的。就是低位在前,高位在后
,相同还有,
writeFixedLenthString(this.account, 32),补充位,也是先写入数据,后填充‘\x00’,在python的bytes使用bytes.ljust(),后面有详细介绍。
这是消息的主体
封包wss数据
根据上面在js可以生成一些python代码用于数据组织
def addHeader(bodyl,command,action=0,ext=0):
# Data---\x3e command:" + t + " action:" + n + " size: 4+len(e),ext:a
num= command | action << 4 | bodyl << 12 | ext << 23
#num.to_bytes(length=32,byteorder='little')
return num.to_bytes(length=4,byteorder='little')
def parseHeader(hex4='11c08500'):
# little_byte = b'\x01\x00\x00\x00\x00\x00\x00\x00'
hex4=int.from_bytes(bytes.fromhex(hex4),byteorder='little')
command=hex4 &int('F',16)
action=hex4>>4 & int('FF',16)
l=hex4>>12 & int('7FF',16) #only 11bit
ext=hex4>>23
print (f'command:{command},action:{action},len:{l},ext:{ext}')
def loginbytes(sn,lh,uid,account,token):
re=bytes()
re+=int(b'01',16).to_bytes(4,byteorder='little')
re+=int(b'01',16).to_bytes(4,byteorder='little')
re+=sn.encode()
re+=int(lh).to_bytes(4,byteorder='little')
re+=int(uid).to_bytes(8,byteorder='little')
re+=account.encode().ljust(32,b'\x00')
re+=token.encode().ljust(32,b'\x00')
# print(len(re))
# print (re.hex())
return re
解释
addHeader使用按位或,在返回时byteorder='little')
这是比较原始数据得出的。
parseHeader用按位与,移位来排除前,与来排除后,只留对照有用的。
这两个函数处理4字节32位的头部信息。
loginbytes 是改写的js中的SyncLogonDto getBody其中account.encode().ljust(32,b'\x00')
是对account补足32,int(uid).to_bytes(8,byteorder='little')
是将uid转为长整数8字节长,低位在前高位在后的bytes。
发送接收websocket的常驻后台脚本
本段代码,pip install websocket-client, 版本号1.18,websocket协议13
import websocket
import threading
import time
import os
import login
#os.path+=['../']
# 定义当接收到消息时调用的回调函数
def on_message(ws, message):
print(f"Received message: {message}")
# 定义当连接关闭时调用的回调函数
def on_close(ws, close_status_code, close_msg):
print(f"### Closed ###")
# 定义当出现错误时调用的回调函数
def on_error(ws, error):
print(f"### Error ### {error}")
if __name__ == "__main__":
# websocket服务地址
ws_service_address = "ws://your_websocket_server"
ws_service_address = "wss://alidr-311.klwgt.com/data"
# 创建websocket应用实例
websocket.enableTrace(True)
ws = websocket.WebSocketApp(ws_service_address,
on_message=on_message,
on_close=on_close,
on_error=on_error)
# 创建一个线程用于运行websocket客户端
wst = threading.Thread(target=ws.run_forever)
wst.daemon = True
wst.start()
time.sleep(3)
mss=login.getlogin()
ws.send_bytes(mss)
# 主线程做其他事情...
# 例如,主线程可以发送消息到websocket服务器
# ws.send("Your message here")
# 主线程在此处等待,否则程序会立即退出
# 如果你的程序需要在后台运行,则不需要这一行
wst.join()
其实主要就是来自百度AI的一段代码。稍加调整假入了延时的登陆调用。然后就成功登陆了。实现和浏览器js登陆websocket的一样的效果。
总结
虽然这段代码,没有什么业务功能,细节也是基本的类型转换,但是它是从怀疑中不断产生的。因为我开始时怀疑python在websocket库是否能完全仿真js在websocket。在查看js的建立连接的请求头时,发现协议版本是2011年的13,然后查看了websocket-client在pypi,也是支持到这个版本,只是还没有实现gzip功能的扩展。
既然如此,wss的 建立也没要求cookie和其实token。那就用python跑一下。基于开始处对于网站数据的整理,要是没有原本的js脚本,这也是一个不可能完成的事情了。但是虽然登陆成了,以后的业务逻辑还不知道怎么处理。最少,分步骤,实现批处理,是可以的。这就由python建立 了 一个,js代码的客户端。
好吧纯属有病。
再见