websocket

最新推荐文章于 2022-09-23 16:11:27 发布

苟修今

最新推荐文章于 2022-09-23 16:11:27 发布

阅读量354

点赞数

文章标签： websocket

本文链接：https://blog.csdn.net/DeskyAki/article/details/100106052

版权

一简单介绍

Http socket实现短链接双工通道三次握手四次挥手

websocket socket实现双工通道能推送(服务端可以主动向客户端推送消息) 不断开只用http一次握手

WebSocket 是 HTML5 开始提供的一种在单个 TCP 连接上进行全双工通讯的协议。它又是一种新的应用层协议。

全双工即双方可同时向对方发送消息

WebSocket 使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在 WebSocket API 中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。

在 WebSocket API 中，浏览器和服务器只需要做一个握手的动作，然后，浏览器和服务器之间就形成了一条快速通道。两者之间就直接可以数据互相传送。

现在，很多网站为了实现推送技术，所用的技术都是 Ajax 轮询。轮询是在特定的的时间间隔（如每1秒），由浏览器对服务器发出HTTP请求，然后由服务器返回最新的数据给客户端的浏览器。这种传统的模式带来很明显的缺点，即浏览器需要不断的向服务器发出请求，然而HTTP请求可能包含较长的头部，其中真正有效的数据可能只是很小的一部分，显然这样会浪费很多的带宽等资源。

HTML5 定义的 WebSocket 协议，能更好的节省服务器资源和带宽，并且能够更实时地进行通讯。

当客户端连接服务端的时候，会向服务端发送一个类似下面的http报文，报文中有一个upgrade首部，它的作用是告诉服务端需要将通信协议切换到websocket,如果服务端支持websocket协议，那么它就会将自己的通信协议切换到websocket

同时会给客户端类似的响应头

返回的状态码为101，表示同意客户端协议转换请求，并将它转换为websocket协议。以上过程都是利用http通信完成的，称之为websocket协议握手(websocket Protocol handshake)，进过这握手之后，客户端和服务端就建立了websocket连接，以后的通信走的都是websocket协议了。所以总结为websocket握手需要借助于http协议，建立连接后通信过程使用websocket协议。同时需要了解的是，该websocket连接还是基于我们刚才发起http连接的那个TCP连接。一旦建立连接之后，我们就可以进行数据传输了，websocket提供两种数据传输：文本数据和二进制数据。

二实现流程

服务端开启socket，监听IP和端口
客户端发送连接请求(通过的HTTP请求)
客户端生成一个随机字符串，发送给服务端
服务端收到随机字符串（客户端向服务端发送的请求报文的Sec-WebSocket-Key），根据magic string(魔法字符串就是个固定的随机字符串258EAFA5-E914-47DA-95CA-C5AB0DC85B11)进行加密, 并返回给客户端
客户端根据魔法字符串加密规则和服务端发来的字符串对比

三实现简单的websocket

sever.py

import socket
import base64
import hashlib
 
 
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)
    return True
 
 
def run():
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    sock.bind(('127.0.0.1', 8003))
    sock.listen(5)
 
    conn, address = sock.accept()
    data = conn.recv(1024)
    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%s\r\n\r\n"
 
    value = headers['Sec-WebSocket-Key'] + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11'
    ac = base64.b64encode(hashlib.sha1(value.encode('utf-8')).digest())
    response_str = response_tpl % (ac.decode('utf-8'), headers['Host'], headers['url'])
    conn.send(bytes(response_str, encoding='utf-8'))
 
    while True:
        try:
            info = conn.recv(8096)
        except Exception as e:
            info = None
        if not info:
            break
        payload_len = info[1] & 127
        if payload_len == 126:
            extend_payload_len = info[2:4]
            mask = info[4:8]
            decoded = info[8:]
        elif payload_len == 127:
            extend_payload_len = info[2:10]
            mask = info[10:14]
            decoded = info[14:]
        else:
            extend_payload_len = None
            mask = info[2:6]
            decoded = info[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')
        body = body + "卧槽"
        send_msg(conn,body.encode('utf-8'))
 
    sock.close()
 
if __name__ == '__main__':
    run()

client.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
<script>
    ws = new WebSocket("ws://127.0.0.1:8002");
    // 建立连接时除法
    ws.onopen = function () {
        ws.send("ok")
    };
    // 客户端接收服务端数据时触发
    ws.onmessage = function (event) {
        console.log(event)
    };
    // 服务端主动断开，执行
    ws.onclose = function () {
    };
    // 通信发生错误时触发
    ws.onerror = function () {

    }

</script>
</body>
</html>

启动server.py ，然后打开client.html , 点击控制台

essageEvent {isTrusted: true, data: "ok卧槽", origin: "ws://127.0.0.1:8002", lastEventId: "", source: null, …} 说明成功了

四基于tornado和websecket 的聊天室实例

app.py


import tornado.web
import tornado.ioloop
import tornado.websocket

users = set()


class IndexHandler(tornado.web.RequestHandler):

    def get(self, *args, **kwargs):
        self.render("index.html")


class ChatHandler(tornado.websocket.WebSocketHandler):
    # 因为是websocket发送的请求， 不是http发送的

    def open(self, *args, **kwargs):
        # 客户端和服务端已经建立连接， 连接， 握手
        users.add(self)
        # 加到集合里
        print("接客")

    def on_message(self, message):
        # 客户端发送消息时触发
        content = self.render_string("message.html", msg=message)
        # 写进每一个浏览者的消息框
        for client in users:
            client.write_message(content)

    def on_close(self):
        # 客户端主动关闭连接
        users.remove(self)
    

def run():
    settings = {
        'template_path': 'templates',
        "static_path": "static",
        "static_url_prefix": "/static/",
    }
    application = tornado.web.Application([
        (r"/", IndexHandler),
        (r"/chat", ChatHandler)
    ], **settings)
    application.listen(8888)
    tornado.ioloop.IOLoop.instance().start()


if __name__ == "__main__":
    run()

index.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
    <style>
        .container {
            border: 1px solid #dddddd;
            height: 400px;
            overflow: auto;
        }
    </style>
</head>
<body>
<div style="width: 750px; margin: 0 auto;">
    <h1>瞎聊</h1>
    <div class="container"></div>
    <div class="input">
        <input type="text" id="txt">
        <input type="button" value="发送" id="btn" onclick="sendMessage();">
    </div>
</div>
<script src="/static/jquery-3.3.1.js"></script>
<script>

    ws = new WebSocket("ws://127.0.0.1:8888/chat");
    ws.onmessage = function (event) {
        $(".container").append(event.data);
    };
    // 服务点主动终止除法
    ws.onclose = function () {
    };
    function sendMessage() {
        ws.send($('#txt').val());
    }
</script>
</body>
</html>

message.html

<div style="margin: 20px; background-color: aquamarine">
    {{ msg }}
</div>

参考https://www.cnblogs.com/myzhibie/p/4470065.html

参考https://www.cnblogs.com/wupeiqi/p/6558766.html

苟修今

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

websocket

一 简单介绍

二 实现流程

三 实现简单的websocket

四 基于tornado和websecket 的聊天室实例

一简单介绍

二实现流程

三实现简单的websocket

四基于tornado和websecket 的聊天室实例