python实现websocket服务端

2019独角兽企业重金招聘Python工程师标准>>>

ws.py是服务端，文件内容如下

#!/usr/bin/env python
import socket
import base64
import hashlib
import struct
from select import select
class dryWebSocket():
    #####
    @staticmethod
    def handShake(serverHandle):
        key = ''
        data = serverHandle.recv(8192)
        for line in data.split('\r\n\r\n')[0].split('\r\n')[1:]:
            k, v = line.split(': ')
            if k == 'Sec-WebSocket-Key':
                key = base64.b64encode(hashlib.sha1(v + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11').digest())
        response = 'HTTP/1.1 101 Switching Protocols\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept:%s\r\n\r\n' % key
        serverHandle.send(response)
        return True
    #####
    @staticmethod
    def recv(serverHandle, size=8192):
        data = serverHandle.recv(size)
        if not len(data):
            return False
        length = ord(data[1]) & 127
        if length == 126:
            mask = data[4:8]
            raw = data[8:]
        elif length == 127:
            mask = data[10:14]
            raw = data[14:]
        else:
            mask = data[2:6]
            raw = data[6:]
        ret = ''
        for cnt, d in enumerate(raw):
            ret += chr(ord(d) ^ ord(mask[cnt%4]))
        return ret
    #####
    @staticmethod
    def send(serverHandle, data):
        head = '\x81'
        if len(data) < 126:
            head += struct.pack('B', len(data))
        elif len(data) <= 0xFFFF:
            head += struct.pack('!BH', 126, len(data))
        else:
            head += struct.pack('!BQ', 127, len(data))
        serverHandle.send(head + data)
##########################################################################################
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
socketList = set()
##########
def run(ip, port, count, function, debug):
    server.bind((ip, port))
    server.listen(count)
    socketList.add(server)
    print 'websocket server start on port %d' % port
    while True:
        r, w, e = select(socketList, [], [])
        for sock in r:
            if sock == server:
                serverHandle, addr = sock.accept()
                if dryWebSocket.handShake(serverHandle):
                    socketList.add(serverHandle)
            else:
                data = dryWebSocket.recv(sock)
                if not data:
                    socketList.remove(sock)
                else:
                    function(sock, data, debug)
##########
def dataDo(serverHandle, data, debug):
    if debug:
        print data
    dryWebSocket.send(serverHandle, data)
##########
if __name__ == '__main__':
    run('0.0.0.0', 7000, 100, dataDo, True)

ws.html是客户端，文件内容如下

 

转载于:https://my.oschina.net/mickelfeng/blog/903837