node.js 使用 net 模塊模擬 websocket 握手進(jìn)行數(shù)據(jù)傳遞操作示例

Request URL: ws://localhost:8888/
Request Method: GET
Connection: Upgrade
Upgrade: websocket
Sec-WebSocket-Version: 13
Sec-WebSocket-Key: uR5YP/BMO6M24tAFcmHeXw==
Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits

Status Code: 101 Switching Protocols
Connection: Upgrade
Sec-WebSocket-Accept: eS92kXpBNI6fWsCkj6WxH6QeoHs=
Upgrade: websocket

const crypto = require('crypto');
function getSecWebSocketAccept(key) {
  return crypto.createHash('sha1')
    .update(key + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')
    .digest('base64');
}
console.log(getSecWebSocketAccept('uR5YP/BMO6M24tAFcmHeXw=='));

0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7
+-+-+-+-+-------+-+-------------+-------------------------------+
|F|R|R|R| opcode|M| Payload len |  Extended payload length  |
|I|S|S|S| (4) |A|   (7)   |       (16/64)      |
|N|V|V|V|    |S|       |  (if payload len==126/127)  |
| |1|2|3|    |K|       |                |
+-+-+-+-+-------+-+-------------+-------------------------------+
|   Extended payload length continued, if payload len == 127 |
+-------------------------------+-------------------------------+
|                |Masking-key, if MASK set to 1 |
+-------------------------------+-------------------------------+
| Masking-key (continued)    |     Payload Data     |
+-------------------------------+-------------------------------+
|           Payload Data continued ...        |
+---------------------------------------------------------------+
|           Payload Data continued ...        |
+---------------------------------------------------------------+

function unmask(buffer, mask) {
  const length = buffer.length;
  for (var i = 0; i < length; i++) {
    buffer[i] ^= mask[i & 3];
  }
}

const crypto = require('crypto');
const net = require('net');
//計(jì)算websocket校驗(yàn)
function getSecWebSocketAccept(key) {
  return crypto.createHash('sha1')
    .update(key + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')
    .digest('base64');
}
//掩碼操作
function unmask(buffer, mask) {
  const length = buffer.length;
  for (var i = 0; i < length; i++) {
    buffer[i] ^= mask[i & 3];
  }
}
//創(chuàng)建一個(gè)tcp服務(wù)器
let server = net.createServer(function (socket) {
  socket.once('data', function (data) {
    data = data.toString();
    //查看請(qǐng)求頭中是否有升級(jí)websocket協(xié)議的頭信息
    if (data.match(/Upgrade: websocket/)) {
      let rows = data.split('\r\n');
      //去掉第一行的請(qǐng)求行
      //去掉請(qǐng)求頭的尾部?jī)蓚€(gè)空行
      rows = rows.slice(1, -2);
      let headers = {};
      rows.forEach(function (value) {
        let [k, v] = value.split(': ');
        headers[k] = v;
      });
      //判斷websocket的版本
      if (headers['Sec-WebSocket-Version'] == 13) {
        let secWebSocketKey = headers['Sec-WebSocket-Key'];
        //計(jì)算websocket校驗(yàn)
        let secWebSocketAccept = getSecWebSocketAccept(secWebSocketKey);
        //服務(wù)端響應(yīng)的內(nèi)容
        let res = [
          'HTTP/1.1 101 Switching Protocols',
          'Upgrade: websocket',
          `Sec-WebSocket-Accept: ${secWebSocketAccept}`,
          'Connection: Upgrade',
          '\r\n'
        ].join('\r\n');
        //給客戶端發(fā)送響應(yīng)內(nèi)容
        socket.write(res);
        //注意這里不要斷開連接，繼續(xù)監(jiān)聽'data'事件
        socket.on('data', function (buffer) {
          //注意buffer的最小單位是一個(gè)字節(jié)
          //取第一個(gè)字節(jié)的第一位，判斷是否是結(jié)束位
          let fin = (buffer[0] & 0b10000000) === 0b10000000;
          //取第一個(gè)字節(jié)的后四位，得到的一個(gè)是十進(jìn)制數(shù)
          let opcode = buffer[0] & 0b00001111;
          //取第二個(gè)字節(jié)的第一位是否是1，判斷是否掩碼操作
          let mask = buffer[1] & 0b100000000 === 0b100000000;
          //載荷數(shù)據(jù)的長(zhǎng)度
          let payloadLength = buffer[1] & 0b01111111;
          //掩碼鍵，占4個(gè)字節(jié)
          let maskingKey = buffer.slice(2, 6);
          //載荷數(shù)據(jù)，就是客戶端發(fā)送的實(shí)際數(shù)據(jù)
          let payloadData = buffer.slice(6);
          //對(duì)數(shù)據(jù)進(jìn)行解碼處理
          unmask(payloadData, maskingKey);
          //向客戶端響應(yīng)數(shù)據(jù)
          let send = Buffer.alloc(2 + payloadData.length);
          //0b10000000表示發(fā)送結(jié)束
          send[0] = opcode | 0b10000000;
          //載荷數(shù)據(jù)的長(zhǎng)度
          send[1] = payloadData.length;
          payloadData.copy(send, 2);
          socket.write(send);
        });
      }
    }
  });
  socket.on('error', function (err) {
    console.log(err);
  });
  socket.on('end', function () {
    console.log('連接結(jié)束');
  });
  socket.on('close', function () {
    console.log('連接關(guān)閉');
  });
});
//監(jiān)聽8888端口
server.listen(8888);

<!doctype html>
<html lang="zh-CN">
<head>
  <meta charset="UTF-8">
  <title>Document</title>
</head>
<body>
<script>
  var ws = new WebSocket('ws://localhost:8888');
  ws.onopen = function () {
    console.log('連接成功');
    ws.send('你好服務(wù)端');
  };
  ws.onmessage = function (ev) {
    console.log('接收數(shù)據(jù)', ev.data);
  };
  ws.onclose = function () {
    console.log('連接斷開');
  };
</script>
</body>
</html>