python web socket_Python 实现WebSocket通信 [转载]

在网上转载的一个使用Python实现的WebSocket通信案例，放在这里只作为留档。

import socket,struct,hashlib,base64

def get_headers(data):

headers = {}

data = str(data, encoding="utf-8")

header, body = data.split("\r\n\r\n", 1)

header_list = header.split("\r\n")

for i in header_list:

i_list = i.split(":", 1)

if len(i_list) >= 2:

headers[i_list[0]] = "".join(i_list[1::]).strip()

else:

i_list = i.split(" ", 1)

if i_list and len(i_list) == 2:

headers["method"] = i_list[0]

headers["protocol"] = i_list[1]

return headers

def parse_payload(payload):

payload_len = payload[1] & 127

if payload_len == 126:

mask = payload[4:8]

decoded = payload[8:]

elif payload_len == 127:

mask = payload[10:14]

decoded = payload[14:]

else:

mask = payload[2:6]

decoded = payload[6:]

bytes_list = bytearray()

for i in range(len(decoded)):

chunk = decoded[i] ^ mask[i % 4]

bytes_list.append(chunk)

body = str(bytes_list, encoding='utf-8')

return body

def send_msg(conn, msg_bytes):

token = b"\x81"

length = len(msg_bytes)

if length < 126:

token += struct.pack("B", length)

elif length <= 0xFFFF:

token += struct.pack("!BH", 126, length)

else:

token += struct.pack("!BQ", 127, length)

msg = token + msg_bytes

conn.sendall(msg)

return True

def recv_msg(conn):

recv = conn.recv(8096)

ref_string = parse_payload(recv)

return ref_string

if __name__ == "__main__":

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

sock.bind(("127.0.0.1", 10083))

sock.listen(5)

conn, addr = sock.accept()

data = conn.recv(8096)

headers = get_headers(data)

# 对请求头中的sec-websocket-key进行加密

response_tpl = "HTTP/1.1 101 Switching Protocols\r\n" \

"Upgrade:websocket\r\n" \

"Connection: Upgrade\r\n" \

"Sec-WebSocket-Accept: %s\r\n" \

"WebSocket-Location: ws://%s\r\n\r\n"

magic_string = '258EAFA5-E914-47DA-95CA-C5AB0DC85B11'

if headers.get('Sec-WebSocket-Key'):

value = headers['Sec-WebSocket-Key'] + magic_string

ac = base64.b64encode(hashlib.sha1(value.encode('utf-8')).digest())

response_str = response_tpl % (ac.decode('utf-8'), headers.get("Host"))

conn.sendall(bytes(response_str, encoding="utf-8"))

while True:

print(recv_msg(conn))

send_msg(conn, b"hello lyshark")

多线程

import socket,struct,hashlib,base64

import threading

def get_headers(data):

headers = {}

data = str(data, encoding="utf-8")

header, body = data.split("\r\n\r\n", 1)

header_list = header.split("\r\n")

for i in header_list:

i_list = i.split(":", 1)

if len(i_list) >= 2:

headers[i_list[0]] = "".join(i_list[1::]).strip()

else:

i_list = i.split(" ", 1)

if i_list and len(i_list) == 2:

headers["method"] = i_list[0]

headers["protocol"] = i_list[1]

return headers

def parse_payload(payload):

payload_len = payload[1] & 127

if payload_len == 126:

mask = payload[4:8]

decoded = payload[8:]

elif payload_len == 127:

mask = payload[10:14]

decoded = payload[14:]

else:

mask = payload[2:6]

decoded = payload[6:]

bytes_list = bytearray()

for i in range(len(decoded)):

chunk = decoded[i] ^ mask[i % 4]

bytes_list.append(chunk)

body = str(bytes_list, encoding='utf-8')

return body

def send_msg(conn, msg_bytes):

first_byte = b"\x81"

length = len(msg_bytes)

if length < 126:

first_byte += struct.pack("B", length)

elif length <= 0xFFFF:

first_byte += struct.pack("!BH", 126, length)

else:

first_byte += struct.pack("!BQ", 127, length)

msg = first_byte + msg_bytes

conn.sendall(msg)

return True

def recv_msg(conn):

data_recv = conn.recv(8096)

if data_recv[0:1] == b"\x81":

data_parse = parse_payload(data_recv)

return data_parse

return False

def handler_accept(sock):

while True:

conn, addr = sock.accept()

data = conn.recv(8096)

headers = get_headers(data)

response_tpl = "HTTP/1.1 101 Switching Protocols\r\n" \

"Upgrade:websocket\r\n" \

"Connection: Upgrade\r\n" \

"Sec-WebSocket-Accept: %s\r\n" \

"WebSocket-Location: ws://%s\r\n\r\n"

magic_string = '258EAFA5-E914-47DA-95CA-C5AB0DC85B11'

if headers.get('Sec-WebSocket-Key'):

value = headers['Sec-WebSocket-Key'] + magic_string

ac = base64.b64encode(hashlib.sha1(value.encode('utf-8')).digest())

response_str = response_tpl % (ac.decode('utf-8'), headers.get("Host"))

conn.sendall(bytes(response_str, encoding="utf-8"))

t = threading.Thread(target=handler_msg, args=(conn, ))

t.start()

def handler_msg(conn):

with conn as c:

while True:

try:

recv = recv_msg(c)

print("接收数据: {}".format(recv))

send_msg(c, bytes("hello lyshark", encoding="utf-8"))

except Exception:

exit(0)

if __name__ == "__main__":

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

sock.bind(("127.0.0.1", 10083))

sock.listen(5)

t = threading.Thread(target=handler_accept(sock))

t.start()

前端

var oUl=document.getElementById('content');

var oConnect=document.getElementById('connect');

var oSend=document.getElementById('send');

var oInput=document.getElementById('message');

var ws=null;

oConnect.οnclick=function(){

ws=new WebSocket('ws://127.0.0.1:10083');

ws.onopen=function(){

oUl.innerHTML+="

客户端已连接";

}

ws.onmessage=function(evt){

console.log("fdsa")

oUl.innerHTML+="

"+evt.data+"";

}

ws.onclose=function(){

oUl.innerHTML+="

客户端已断开连接";

};

ws.οnerrοr=function(evt){

oUl.innerHTML+="

"+evt.data+"";

};

};

oSend.οnclick=function(){

if(ws){

ws.send($("#message").val())

}

}