websocket 原理

自己写一个websocket

原理更多参考--   https://www.cnblogs.com/zunzunQ/p/8549180.html

原理简介:
    client端 握手请求中 包含一个base64 encode的随机字符串,
    server端 magic_string+str    sha1摘要之后base64加密, 发给client 
    服务器完成协议升级后（HTTP->Websocket），服务端就可以主动推送信息给客户端啦  (回调机制)

import socket, hashlib, base64

sock = socket.socket()
sock.bind(('127.0.0.1', 9000))
sock.listen(5)
'''
b'GET / HTTP/1.1

Host: 127.0.0.1:9000

Connection: Upgrade

Pragma: no-cache

Cache-Control: no-cache

User-Agent: Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.81 Safari/537.36

Upgrade: websocket
   --建立--
Origin: http://localhost:63342

Sec-WebSocket-Version: 13

Accept-Encoding: gzip, deflate, br

Accept-Language: zh-CN,zh;q=0.9

Sec-WebSocket-Key: vNtQ8bVSNRPL5p+xo1X5CA==
  验证 一按钥匙
Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits


'''
'''
b'GET / HTTP/1.1
   -- 单纯http地址栏访问
Host: 127.0.0.1:9000

Connection: keep-alive

Pragma: no-cache

Cache-Control: no-cache

Upgrade-Insecure-Requests: 1

User-Agent: Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.81 Safari/537.36

Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8

Accept-Encoding: gzip, deflate, br

Accept-Language: zh-CN,zh;q=0.9

'
'''
conn, addr = sock.accept()
data = conn.recv(1024)
# print(data)

def get_header(data): # 获取头部信息
    header_dict = {}
    header_str = data.decode('utf8')
    for i in header_str.split('
'):
        if str(i).startswith('Sec-WebSocket-Key'):
            header_dict['Sec-WebSocket-Key'] = i.split(':')[1].strip()
    return header_dict


headers = get_header(data)
magic_string = '258EAFA5-E914-47DA-95CA-C5AB0DC85B11'  # magic_string = 魔法字符串
value = headers['Sec-WebSocket-Key'] + magic_string
print(value)  #mzrsVOiYogmzB8fYcyMBpw==258EAFA5-E914-47DA-95CA-C5AB0DC85B11
ac = base64.b64encode(hashlib.sha1(value.encode('utf-8')).digest())  # 固定加密算法
print(ac) #b'0Cu1dAA7U6tFsJS4EVkIDLcvKDU='
response_tpl = "HTTP/1.1 101 Switching Protocols
" 
               "Upgrade:websocket
" 
               "Connection: Upgrade
" 
               "Sec-WebSocket-Accept: %s
" 
               "WebSocket-Location: ws://127.0.0.1:9527

"
response_str = response_tpl % (ac.decode('utf-8'))
conn.send(response_str.encode('utf8'))  # 响应握手信息

while True:
    msg = conn.recv(8096)
    print(msg)

解密过程  分析

hashstr = b'x81x89xbcxb8,xb5Y6xa5Px12x0bxc8x0f:'
# b'x81    x83    xceHxb6x85xffzx85'

# 将第二个字节也就是 x83 第9-16位 进行与127进行 与位 运算
payload = hashstr[1] & 127
print(payload)
if payload == 127:
    extend_payload_len = hashstr[2:10]  # 顾头不顾尾
    mask = hashstr[10:14]
    decoded = hashstr[14:]
# 当位运算结果等于127时,则第3-10个字节为数据长度
# 第11-14字节为mask 解密所需字符串
# 则数据为第15字节至结尾

if payload == 126:
    extend_payload_len = hashstr[2:4]
    mask = hashstr[4:8]
    decoded = hashstr[8:]
# 当位运算结果等于126时,则第3-4个字节为数据长度  255*255+510=65535 21845汉字
# 第5-8字节为mask 解密所需字符串
# 则数据为第9字节至结尾

if payload <= 125:
    extend_payload_len = None
    mask = hashstr[2:6]
    decoded = hashstr[6:]

# 当位运算结果小于等于125时,则这个数字就是数据的长度
# 第3-6字节为mask 解密所需字符串
# 则数据为第7字节至结尾

str_byte = bytearray()  # 字节流

for i in range(len(decoded)):
    byte = decoded[i] ^ mask[i % 4]  # 异或运算  上下为0就是0  上下有1就是1
    str_byte.append(byte)

print(str_byte.decode("utf8"))  # 主要解决中文  厉害了

# # 1字节 = 8 bit  与位运算
# 01111111
# 11111111  最大255
# 2/4/8/16/32/64/138/256

加密过程 -- 分析

import struct
msg_bytes = "hello".encode("utf8")  #转bytes
token = b"x81"       # 数据标准格式
length = len(msg_bytes)  #第二位就是数据长度

if length < 126:
    token += struct.pack("B", length)
elif length == 126:
    token += struct.pack("!BH", 126, length)
else:
    token += struct.pack("!BQ", 127, length)

msg = token + msg_bytes

print(msg)

总结

WebSocket的工作原理
1.握手
    客户端   Sec-WebSocket-Key
    响应     base64(sha1(Sec-WebSocket-Key + magic_string))
    
2.解密：
    与127 进行"与"位运算 结果是两个数的最小值
    1. == 127 第3-10个字节表示该数据的长度 
    2. == 126 第3-4个字节表示该数据的长度 00000010 00000000
    3. <= 125 当前这个数字就是Websocket发送的数据长度