websocket建立连接

websocket建立连接

客户端消息格式

• “ws:” “//” host [ “:” port ] path [ “?” query ] 默认 port 端口 80
• “wss:” “//” host [ “:” port ] path [ “?” query ] 默认 port 端口 443
• 客户端提供信息（放在http或https头headers里面）
  • host 与 port：主机名与端口
  • shema：是否基于 SSL
  • 访问资源：URI
  • 握手随机数：Sec-WebSocket-Key
  • 选择子协议： Sec-WebSocket-Protocol
  • 扩展协议： Sec-WebSocket-Extensions
  • CORS 跨域：Origin

服务端消息格式

• HTTP/1.1 101 Web Socket Protocol Handshake // 101连接建立成功
• Connection: upgrade
• 握手随机数：Sec-WebSocket-Key
• Upgrade: websocket

import os
import socket
from base64 import encodebytes as base64encode
import websocket


def do_request_():
    # ws = websocket.create_connection("wss://echo.websocket.org")
    ws = websocket.create_connection("ws://echo.websocket.org")
    ws.send("test aaa!")
    ws.send("test bbb!")
    print(ws.recv())
    print(ws.recv())


def _create_sec_websocket_key():
    randomness = os.urandom(16)
    return base64encode(randomness).decode('utf-8').strip()


def get_headers():
    websocket_key = _create_sec_websocket_key()
    print(websocket_key)
    print("----")
    headers = {"Host": "echo.websocket.org", "Sec-WebSocket-Version": "13",
               "Sec-WebSocket-Key": websocket_key, "Connection": "keep-alive, Upgrade",
               "Sec-WebSocket-Extensions": "permessage-deflate", "Upgrade": "websocket"}
    headers = (key + ": " + value for key, value in headers.items())
    return "GET / HTTP/1.1\r\n" + "\r\n".join(headers)


def do_request():
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        headers = get_headers()
        # client.connect(("echo.websocket.org", 80))
        client.connect(("121.40.165.18", 8800))
        client.send(headers.encode('utf-8'))
        data = client.recv(1024)
        print(data.decode())
    except Exception as e:
        print(f'error {e}!')
    finally:
        client.close()


if __name__ == "__main__":
    do_request()

   
# 输出
# Sec-WebSocket-Accept:  gk6AiBD48FKZC2RUFMH/ow==
# ----
# HTTP/1.1 101 Switching Protocols
# Upgrade: websocket
# Connection: Upgrade
# Sec-WebSocket-Accept: e5/mWaJy8r13Fsq0la9Rql8R27k=