HTTP、WebSocket 等應用層協議,都是基於 TCP 協議來傳輸數據的。咱們能夠把這些高級協議理解成對 TCP 的封裝。
既然你們都使用 TCP 協議,那麼你們的鏈接和斷開,都要遵循 TCP 協議中的三次握手和四次揮手 ,只是在鏈接以後發送的內容不一樣,或者是斷開的時間不一樣.
咱們知道http是基於tcp實現的,瀏覽器能夠看作是客戶端socket,這個socket在開發的時候遵循http規範,數據在發送並接收成功的時候就斷開socket了,那麼若是咱們的socket一直鏈接不斷開,而且發送數據的時候都把數據進行加密,這樣是否是也是能夠的呢?這種想法就能夠引出websocket了,值得一提的是,websocket的兼容性不是全部版本的瀏覽器都支持(取決因而否可以new WebSocket), 可是隨着技術的發展,瀏覽器對websocket的支持確定會愈來愈好。
對於 WebSocket 來講,它必須依賴 HTTP 協議進行一次握手 ,握手成功後,數據就直接從 TCP 通道傳輸,與 HTTP 無關了。
Websocket和socket不一樣,Websocket工做在應用層,而socket是基於門面模式的一種封裝,可讓程序員簡便地寫網絡通訊程序html
安裝gevent-websocket模塊jquery
from flask import Flask,render_template,request from geventwebsocket.handler import WebSocketHandler from gevent.pywsgi import WSGIServer import json app = Flask(__name__) USERS = { '1':{'name':'jack','count':0}, '2':{'name':'tuple','count':0}, '3':{'name':'lily','count':100}, } @app.route('/index') def index(): return render_template('index.html',users=USERS) # http://127.0.0.1:5000/message 既能夠接受http,也能夠接受websocket WEBSOCKET_LIST = [] @app.route('/message') def message(): ws = request.environ.get('wsgi.websocket') if not ws: return '您使用的是Http協議' WEBSOCKET_LIST.append(ws) while True: # 當關閉webscoekt的時候cid 獲得None cid = ws.receive() if not cid: WEBSOCKET_LIST.remove(ws) ws.close() break old_count = USERS[cid]['count'] new_count = old_count + 1 USERS[cid]['count'] = new_count for client in WEBSOCKET_LIST: client.send(json.dumps({'cid':cid,'count':new_count})) return '' if __name__ == '__main__': # 當請求是websocket的時候,那麼就用WebSocketHandler進行處理 http_server = WSGIServer(('0.0.0.0', 5000), app, handler_class=WebSocketHandler) http_server.serve_forever()
index.html程序員
<!DOCTYPE html> <html lang="zh-CN"> <head> <meta charset="UTF-8"> <title>Title</title> <meta name="viewport" content="width=device-width, initial-scale=1"> </head> <body> <h1>投票系統</h1> <ul> {% for k,v in users.items() %} <li onclick="vote({{k}})" id="id_{{k}}">{{v.name}}<span>{{v.count}}</span></li> {% endfor %} </ul> <script src="{{ url_for('static',filename='jquery-3.3.1.min.js')}}"></script> <script> var ws = new WebSocket('ws://127.0.0.1:5000/message'); // 事件監聽綁定回調函數,一旦有消息發送過來就會調用函數 ws.onmessage = function (event) { /* 服務器端向客戶端發送數據時,自動執行 */ // {'cid':cid,'count':new} var response = JSON.parse(event.data); $('#id_'+response.cid).find('span').text(response.count); }; function vote(cid) { ws.send(cid) } </script> </body> </html>
websocket的握手須要藉助http, new WebSocket的時候,服務端接收到的的請求頭信息有有三個比較特殊:web
Sec-WebSocket-Key:FUH2Nige4Npq/InFS0OoJQ== Upgrade:websocket Connection: Upgrade
表示此次請求是用於升級http爲websocket,並把一個隨機字符串發送過去,此時服務端應該拿到這個字符串進行加密而後發送過去,瀏覽器對這個字符串也加密,比較加密後的字符串和服務端發送過來的字符串是否一致,一致就表名握手成功,此時response header爲json
Connection:Upgrade Sec-WebSocket-Accept:+lZCKceYfJsmNj6q0GYPa9r9LXE= Upgrade:websocket
狀態碼爲:101 Switching Protocols,表示協議切換成功,此時,後面的通訊就和http沒有半毛錢關係了。flask
def get_headers(data): """ 將請求頭格式化成字典 :param data: :return: """ header_dict = {} data = str(data, encoding='utf-8') header, body = data.split('\r\n\r\n', 1) header_list = header.split('\r\n') for i in range(0, len(header_list)): if i == 0: if len(header_list[i].split(' ')) == 3: header_dict['method'], header_dict['url'], header_dict['protocol'] = header_list[i].split(' ') else: k, v = header_list[i].split(':', 1) header_dict[k] = v.strip() return header_dict def send_msg(conn, msg_bytes): """ WebSocket服務端向客戶端發送消息 :param conn: 客戶端鏈接到服務器端的socket對象,即: conn,address = socket.accept() :param msg_bytes: 向客戶端發送的字節 :return: """ import struct 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.send(msg) import base64 import hashlib import socket sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) sock.bind(('127.0.0.1', 8002)) sock.listen(5) while 1: # 1. 等待用戶鏈接 conn, address = sock.accept() # 2. 接收驗證消息 msg = conn.recv(8096) msg_dict = get_headers(msg) # 3. 對數據加密, 其中魔法字符串'258EAFA5-E914-47DA-95CA-C5AB0DC85B11' 是固定的值 value = msg_dict['Sec-WebSocket-Key'] + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11' ac = base64.b64encode(hashlib.sha1(value.encode('utf-8')).digest()) # 4. 將加密後的結果返回給瀏覽器,發送的時候是用http通訊的,回消息的時候也要封裝成http格式 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://127.0.0.1:8002\r\n\r\n" response = response_tpl %(ac.decode('utf-8'),) conn.send(response.encode('utf-8')) # 5. 接受瀏覽器發送過來的加密數據,並進行解密,發送數據也須要進行加密,加密和解密規則能夠參考官方 while True: info = conn.recv(8096) # 10111111 去掉第一位,和 01111111 相與就行 payload_len = info[1] & 127 if payload_len == 127: # 若是payload_len是127,那麼須要日後面取8個字節做爲頭信息,頭部總共10個字節 extend_payload_len = info[2:10] mask = info[10:14] decoded = info[14:] elif payload_len == 126: # 若是payload_len是126,那麼須要日後面取2個字節做爲頭信息,頭部總共4個字節 extend_payload_len = info[2:4] mask = info[4:8] decoded = info[8:] else: # 若是payload_len小於126,那麼頭部總共2個字節 extend_payload_len = None mask = info[2:6] decoded = info[6:] bytes_list = bytearray() # decoded 是真實的數據 for i in range(len(decoded)): chunk = decoded[i] ^ mask[i % 4] bytes_list.append(chunk) body = str(bytes_list, encoding='utf-8') print('瀏覽器發送過來的數據',body) # # import time # time.sleep(2) send_msg(conn,body.encode('utf8'))