websocket实现简单的通信
Posted dragonliu
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了websocket实现简单的通信相关的知识，希望对你有一定的参考价值。
websocket server端
#coding=utf8
#!/usr/bin/python
import struct,socket
import hashlib
import threading,random
import time
import struct
from base64 import b64encode, b64decode
connectionlist = {}
g_code_length = 0
g_header_length = 0
def hex2dec(string_num):
return str(int(string_num.upper(), 16))
def get_datalength(msg):
global g_code_length
global g_header_length
print (len(msg))
g_code_length = ord(msg[1]) & 127
received_length = 0;
if g_code_length == 126:
#g_code_length = msg[2:4]
#g_code_length = (ord(msg[2])<<8) + (ord(msg[3]))
g_code_length = struct.unpack(‘>H‘, str(msg[2:4]))[0]
g_header_length = 8
elif g_code_length == 127:
#g_code_length = msg[2:10]
g_code_length = struct.unpack(‘>Q‘, str(msg[2:10]))[0]
g_header_length = 14
else:
g_header_length = 6
g_code_length = int(g_code_length)
return g_code_length
def parse_data(msg):
global g_code_length
g_code_length = ord(msg[1]) & 127
received_length = 0;
if g_code_length == 126:
g_code_length = struct.unpack(‘>H‘, str(msg[2:4]))[0]
masks = msg[4:8]
data = msg[8:]
elif g_code_length == 127:
g_code_length = struct.unpack(‘>Q‘, str(msg[2:10]))[0]
masks = msg[10:14]
data = msg[14:]
else:
masks = msg[2:6]
data = msg[6:]
i = 0
raw_str = ‘‘
for d in data:
raw_str += chr(ord(d) ^ ord(masks[i%4]))
i += 1
print (u"总长度是：%d" % int(g_code_length))
return raw_str
def sendMessage(message):
global connectionlist
message_utf_8 = message.encode(‘utf-8‘)
for connection in connectionlist.values():
back_str = []
back_str.append(‘\x81‘)
data_length = len(message_utf_8)
if data_length <= 125:
back_str.append(chr(data_length))
elif data_length <= 65535 :
back_str.append(struct.pack(‘b‘, 126))
back_str.append(struct.pack(‘>h‘, data_length))
#back_str.append(chr(data_length >> 8))
#back_str.append(chr(data_length & 0xFF))
#a = struct.pack(‘>h‘, data_length)
#b = chr(data_length >> 8)
#c = chr(data_length & 0xFF)
elif data_length <= (2^64-1):
#back_str.append(chr(127))
back_str.append(struct.pack(‘b‘, 127))
back_str.append(struct.pack(‘>q‘, data_length))
#back_str.append(chr(data_length >> 8))
#back_str.append(chr(data_length & 0xFF))
else :
print (u‘太长了‘)
msg = ‘‘
for c in back_str:
msg += c;
back_str = str(msg) + message_utf_8#.encode(‘utf-8‘)
#connection.send(str.encode(str(u"\x00%s\xFF\n\n" % message))) #这个是旧版
#print (u‘send message:‘ + message)
if back_str != None and len(back_str) > 0:
print (back_str)
connection.send(back_str)
def deleteconnection(item):
global connectionlist
del connectionlist[‘connection‘+item]
class WebSocket(threading.Thread):#继承Thread
GUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
def __init__(self,conn,index,name,remote, path="/"):
threading.Thread.__init__(self)#初始化父类Thread
self.conn = conn
self.index = index
self.name = name
self.remote = remote
self.path = path
self.buffer = ""
self.buffer_utf8 = ""
self.length_buffer = 0
def run(self):#重载Thread的run
print(‘Socket%s Start!‘ % self.index)
headers = {}
self.handshaken = False
while True:
if self.handshaken == False:
print (‘Socket%s Start Handshaken with %s!‘ % (self.index,self.remote))
self.buffer += bytes.decode(self.conn.recv(1024))
if self.buffer.find(‘\r\n\r\n‘) != -1:
header, data = self.buffer.split(‘\r\n\r\n‘, 1)
for line in header.split("\r\n")[1:]:
key, value = line.split(": ", 1)
headers[key] = value
headers["Location"] = ("ws://%s%s" %(headers["Host"], self.path))
key = headers[‘Sec-WebSocket-Key‘]
token = b64encode(hashlib.sha1(str.encode(str(key + self.GUID))).digest())
handshake="HTTP/1.1 101 Switching Protocols\r\n" "Upgrade: websocket\r\n" "Connection: Upgrade\r\n" "Sec-WebSocket-Accept: "+bytes.decode(token)+"\r\n" "WebSocket-Origin: "+str(headers["Origin"])+"\r\n" "WebSocket-Location: "+str(headers["Location"])+"\r\n\r\n"
self.conn.send(str.encode(str(handshake)))
self.handshaken = True
print (‘Socket %s Handshaken with %s success!‘ %(self.index, self.remote))
sendMessage(u‘Welcome, ‘ + self.name + ‘ !‘)
self.buffer_utf8 = ""
g_code_length = 0
else:
global g_code_length
global g_header_length
mm=self.conn.recv(128)
if len(mm) <= 0:
continue
if g_code_length == 0:
get_datalength(mm)
#接受的长度
self.length_buffer = self.length_buffer + len(mm)
self.buffer = self.buffer + mm
if self.length_buffer - g_header_length < g_code_length :
continue
else :
self.buffer_utf8 = parse_data(self.buffer) #utf8
msg_unicode = str(self.buffer_utf8).decode(‘utf-8‘, ‘ignore‘) #unicode
if msg_unicode==‘quit‘:
print (u‘Socket%s Logout!‘ % (self.index))
nowTime = time.strftime(‘%H:%M:%S‘,time.localtime(time.time()))
sendMessage(u‘%s %s say: %s‘ % (nowTime, self.remote, self.name+‘ Logout‘))
deleteconnection(str(self.index))
self.conn.close()
break #退出线程
else:
#print (u‘Socket%s Got msg:%s from %s!‘ % (self.index, msg_unicode, self.remote))
nowTime = time.strftime(u‘%H:%M:%S‘,time.localtime(time.time()))
sendMessage(u‘%s %s say: %s‘ % (nowTime, self.remote, msg_unicode))
#重置buffer和bufferlength
self.buffer_utf8 = ""
self.buffer = ""
g_code_length = 0
self.length_buffer = 0
self.buffer = ""
class WebSocketServer(object):
def __init__(self):
self.socket = None
def begin(self):
print( ‘WebSocketServer Start!‘)
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
self.socket.bind(("0.0.0.0",12345))
self.socket.listen(50)
global connectionlist
i=0
while True:
connection, address = self.socket.accept()
username=address[0]
newSocket = WebSocket(connection,i,username,address)
newSocket.start() #开始线程,执行run函数
connectionlist[‘connection‘+str(i)]=connection
i = i + 1
if __name__ == "__main__":
server = WebSocketServer()
server.begin()
websocket client 浏览器
<!DOCTYPE html> <html> <head> <title>WebSocket</title> <style> html, body { font: normal 0.9em arial, helvetica; } #log { width: 440px; height: 200px; border: 1px solid #7F9DB9; overflow: auto; } #msg { width: 330px; } </style> <script> var socket; function init() { var host = "ws://127.0.0.1:12345/"; try { socket = new WebSocket(host); socket.onopen = function (msg) { log(‘Connected‘); }; socket.onmessage = function (msg) { log(msg.data); }; socket.onclose = function (msg) { log("Lose Connection!"); }; } catch (ex) { log(ex); } $("msg").focus(); } function send() { var txt, msg; txt = $("msg"); msg = txt.value; if (!msg) { alert("Message can not be empty"); return; } txt.value = ""; txt.focus(); try { socket.send(msg); } catch (ex) { log(ex); } } window.onbeforeunload = function () { try { socket.send(‘quit‘); socket.close(); socket = null; } catch (ex) { log(ex); } }; function $(id) { return document.getElementById(id); } function log(msg) { $("log").innerHTML += "<br>" + msg; } function onkey(event) { if (event.keyCode == 13) { send(); } } </script> </head> <body onload="init()"> <h3>WebSocket</h3> <br><br> <div id="log"></div> <input id="msg" type="textbox" onkeypress="onkey(event)"/> <button onclick="send()">发送</button> </body> </html>
以上是关于websocket实现简单的通信的主要内容，如果未能解决你的问题，请参考以下文章