用python实现的websocket代码

.用python实现的websocket代码ubuntu下python2.76windowsPython2.79,chrome37firefox35通过代码是在别人(cddn有人提问)基础上改的,主要改动了parsedata和sendmessage这2个函数.改代码参考下面了这段文档.主要是第5条,发送的数据长度分别是8bit和16bit和64bit(即127,65535,和2^64-1)三种情况发送和收取是一样的,例如1.长度小于125时(由于使用126,127用作标志位.)2.数据长度在128-65525之间时,PayloadLength位设为126,后面额外使用16bit 表 关于同志近三年现实表现材料材料类招标技术评分表图表与交易pdf视力表打印pdf用图表说话 pdf 示长度(前面的126不再是长度的一部分)3.数据长度在65526-2^64-1之间时,PayloadLength位设为127,后面额外使用64bit表示长度(前面的127不再是长度的一部分)1.Fin(bit0):determinesifthisisthelastframeinthemessage.Thiswouldbesetto1ontheendofaseriesofframes,orinasingle-framemessage,itwouldbesetto1asitisboththefirstandlastframe.2.RSV1,RSV2,RSV3(bits1-3):thesethreebitsarereservedforwebsocketextensions,andshouldbe0unlessaspecificextensionrequirestheuseofanyofthesebytes.3.Opcode(bits4-7):thesefourbitsdeteriminethetypeoftheframe.ControlframescommunicateWebSocketstate,whilenon-controlframescommunicatedata.Thevarioustypesofcodesinclude:.1.x0:continuationframe;thisframecontainsdatathatshouldbeappendedtothepreviousframe2.x1:textframe;thisframe(andanyfollowing)containstext3.x2:binaryframe;thisframe(andanyfollowing)containsbinarydata4.x3-x7:non-controlreservedframes;thesearereservedforpossiblewebsocketextensions5.x8:closeframe;thisframeshouldendtheconnection6.x9:pingframe7.xA:pongframe8.xB-xF:controlreservedframes4.Mask(bit8):thisbitdetermineswhetherthisspecificframeusesamaskornot.5.PayloadLength(bits9-15,or16-31,or16-79):thesesevenbytesdeterminethepayloadlength.Ifthelengthis126,thelengthisactuallydeterminedbybits16through31(thatis,thefollowingtwobytes).Ifthelengthis127,thelengthisactuallydeterminedbybits16through79(thatis,thefollowingeightbytes).6.MaskingKey(thefollowingfourbytes):thisrepresentsthemask,iftheMaskbitissetto1.7.PayloadData(thefollowingdata):finally,thedata.Thepayloaddatamaybesentovermultipleframes;weknowthesizeoftheentire.messagebythepayloadlengththatwassent,andcanappenddatatogethertoformasinglemessageuntilwereceivethemessagewiththeFinflag.Eachconsecutivepayload,ifitexists,willcontainthe0“continuationframe”opcode.具体代码：[python]viewplaincopy1.#coding=utf82.#!/usr/bin/python3.4.5.importstruct,socket6.importhashlib7.importthreading,random8.importtime9.importstruct10.frombase64importb64encode,b64decode11.12.13.connectionlist={}14.g_code_length=0.15.g_header_length=016.17.18.defhex2dec(string_num):19.returnstr(int(string_num.upper(),16))20.21.22.23.24.defget_datalength(msg):25.globalg_code_length26.globalg_header_length27.28.print(len(msg))29.g_code_length=ord(msg[1])&12730.received_length=0;31.ifg_code_length==126:32.#g_code_length=msg[2:4]33.#g_code_length=(ord(msg[2])<<8)+(ord(msg[3]))34.g_code_length=struct.unpack('>H',str(msg[2:4]))[0]35.g_header_length=836.elifg_code_length==127:.37.#g_code_length=msg[2:10]38.g_code_length=struct.unpack('>Q',str(msg[2:10]))[0]39.g_header_length=1440.else:41.g_header_length=642.g_code_length=int(g_code_length)43.returng_code_length44.45.defparse_data(msg):46.globalg_code_length47.g_code_length=ord(msg[1])&12748.received_length=0;49.ifg_code_length==126:50.g_code_length=struct.unpack('>H',str(msg[2:4]))[0]51.masks=msg[4:8]52.data=msg[8:]53.elifg_code_length==127:54.g_code_length=struct.unpack('>Q',str(msg[2:10]))[0]55.masks=msg[10:14]56.data=msg[14:]57.else:58.masks=msg[2:6].59.data=msg[6:]60.61.62.i=063.raw_str=''64.65.66.fordindata:67.raw_str+=chr(ord(d)^ord(masks[i%4]))68.i+=169.70.71.print(u"总长度是：%d"%int(g_code_length))72.returnraw_str73.74.75.defsendMessage(message):76.globalconnectionlist77.78.message_utf_8=message.encode('utf-8')79.forconnectioninconnectionlist.values():80.back_str=[].81.back_str.append('\x81')82.data_length=len(message_utf_8)83.84.85.ifdata_length<=125:86.back_str.append(chr(data_length))87.elifdata_length<=65535:88.back_str.append(struct.pack('b',126))89.back_str.append(struct.pack('>h',data_length))90.#back_str.append(chr(data_length>>8))91.#back_str.append(chr(data_length&0xFF))92.#a=struct.pack('>h',data_length)93.#b=chr(data_length>>8)94.#c=chr(data_length&0xFF)95.elifdata_length<=(2^64-1):96.#back_str.append(chr(127))97.back_str.append(struct.pack('b',127))98.back_str.append(struct.pack('>q',data_length))99.#back_str.append(chr(data_length>>8))100.#back_str.append(chr(data_length&0xFF))101.else:102.print(u'太长了').103.msg=''104.forcinback_str:105.msg+=c;106.back_str=str(msg)+message_utf_8#.encode('utf-8')107.#connection.send(str.encode(str(u"\x00%s\xFF\n\n"%message)))#这个是旧版108.#print(u'sendmessage:'+message)109.ifback_str!=Noneandlen(back_str)>0:110.print(back_str)111.connection.send(back_str)112.113.114.defdeleteconnection(item):115.globalconnectionlist116.delconnectionlist['connection'+item]117.118.119.classWebSocket(threading.Thread):#继承Thread120.121.122.GUID="258EAFA5-E914-47DA-95CA-C5AB0DC85B11"123..124.125.def__init__(self,conn,index,name,remote,path="/"):126.threading.Thread.__init__(self)#初始化父类Thread127.self.conn=conn128.self.index=index129.self.name=name130.self.remote=remote131.self.path=path132.self.buffer=""133.self.buffer_utf8=""134.self.length_buffer=0135.defrun(self):#重载Thread的run136.print('Socket%sStart!'%self.index)137.headers={}138.self.handshaken=False139.140.141.whileTrue:142.ifself.handshaken==False:143.print('Socket%sStartHandshakenwith%s!'%(self.index,self.remote))144.self.buffer+=bytes.decode(self.conn.recv(1024)).145.146.147.ifself.buffer.find('\r\n\r\n')!=-1:148.header,data=self.buffer.split('\r\n\r\n',1)149.forlineinheader.split("\r\n")[1:]:150.key,value=line.split(":",1)151.headers[key]=value152.153.154.headers["Location"]=("ws://%s%s"%(headers["Host"],self.path))155.key=headers['Sec-WebSocket-Key']156.token=b64encode(hashlib.sha1(str.encode(str(key+self.GUID))).digest())157.158.159.handshake="HTTP/1.1101SwitchingProtocols\r\n"\160."Upgrade:websocket\r\n"\161."Connection:Upgrade\r\n"\162."Sec-WebSocket-Accept:"+bytes.decode(token)+"\r\n"\.163."WebSocket-Origin:"+str(headers["Origin"])+"\r\n"\164."WebSocket-Location:"+str(headers["Location"])+"\r\n\r\n"165.166.167.self.conn.send(str.encode(str(handshake)))168.self.handshaken=True169.print('Socket%sHandshakenwith%ssuccess!'%(self.index,self.remote))170.sendMessage(u'Welcome,'+self.name+'!')171.self.buffer_utf8=""172.g_code_length=0173.174.175.else:176.globalg_code_length177.globalg_header_length178.mm=self.conn.recv(128)179.iflen(mm)<=0:180.continue181.ifg_code_length==0:.182.get_datalength(mm)183.#接受的长度184.self.length_buffer=self.length_buffer+len(mm)185.self.buffer=self.buffer+mm186.ifself.length_buffer-g_header_length2.3.4.5.6.22.23.87.88.89.90.91.92.

WebSocket

93.

94.95.