server.on('upgrade', function (req, socket, upgradeHead) {
 var key = req.headers['sec-websocket-key'];
 key = crypto.createHash("sha1").update(key + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11").digest("base64");
 var headers = [
 'HTTP/1.1 101 Switching Protocols',
 'Upgrade: websocket',
 'Connection: Upgrade',
 'Sec-WebSocket-Accept: ' + key
 ];
 socket.setNoDelay(true);
 socket.write(headers.join("\r\n") + "\r\n\r\n", 'ascii');
 var ws = new WebSocket(socket);
 webSocketCollector.push(ws);
 callback(ws);
});

WebSocket.prototype.handleDataStat = function (data) {
 if (!this.stat) {
 var dataIndex = 2; //數(shù)據(jù)索引，因為第一個字節(jié)和第二個字節(jié)肯定不為數(shù)據(jù)，所以初始值為2
 var secondByte = data[1]; //代表masked位和可能是payloadLength位的第二個字節(jié)
 var hasMask = secondByte >= 128; //如果大于或等于128，說明masked位為1
 secondByte -= hasMask ? 128 : 0; //如果有掩碼，需要將掩碼那一位去掉
 var dataLength, maskedData;
 //如果為126，則后面16位長的數(shù)據(jù)為數(shù)據(jù)長度，如果為127，則后面64位長的數(shù)據(jù)為數(shù)據(jù)長度
 if (secondByte == 126) {
  dataIndex += 2;
  dataLength = data.readUInt16BE(2);
 } else if (secondByte == 127) {
  dataIndex += 8;
  dataLength = data.readUInt32BE(2) + data.readUInt32BE(6);
 } else {
  dataLength = secondByte;
 }
 //如果有掩碼，則獲取32位的二進制masking key，同時更新index
 if (hasMask) {
  maskedData = data.slice(dataIndex, dataIndex + 4);
  dataIndex += 4;
 }
 //數(shù)據(jù)量最大為10kb
 if (dataLength > 10240) {
  this.send("Warning : data limit 10kb");
 } else {
  //計算到此處時，dataIndex為數(shù)據(jù)位的起始位置，dataLength為數(shù)據(jù)長度，maskedData為二進制的解密數(shù)據(jù)
  this.stat = {
  index: dataIndex,
  totalLength: dataLength,
  length: dataLength,
  maskedData: maskedData,
  opcode: parseInt(data[0].toString(16).split("")[1] , 16) //獲取第一個字節(jié)的opcode位
  };
 }
 } else {
 this.stat.index = 0;
 }
};

WebSocket.prototype.dataHandle = function (data) {
 this.handleDataStat(data);
 var stat;
 if (!(stat = this.stat)) return;
 //如果opcode為9，則發(fā)送pong響應(yīng)，如果opcode為10則置pingtimes為0
 if (stat.opcode === 9 || stat.opcode === 10) {
 (stat.opcode === 9) ? (this.sendPong()) : (this.pingTimes = 0);
 this.reset();
 return;
 }
 var result;
 if (stat.maskedData) {
 result = new Buffer(data.length-stat.index);
 for (var i = stat.index, j = 0; i < data.length; i++, j++) {
  //對每個字節(jié)進行異或運算，masked是4個字節(jié)，所以%4，借此循環(huán)
  result[j] = data[i] ^ stat.maskedData[j % 4];
 }
 } else {
 result = data.slice(stat.index, data.length);
 }
 this.datas.push(result);
 stat.length -= (data.length - stat.index);
 //當(dāng)長度為0，說明當(dāng)前幀為最后幀
 if (stat.length == 0) {
 var buf = Buffer.concat(this.datas, stat.totalLength);
 if (stat.opcode == 8) {
  this.close(buf.toString());
 } else {
  this.emit("message", buf.toString());
 }
 this.reset();
 }
};

//數(shù)據(jù)發(fā)送
WebSocket.prototype.send = function (message) {
 if(this.state !== "OPEN") return;
 message = String(message);
 var length = Buffer.byteLength(message);
// 數(shù)據(jù)的起始位置，如果數(shù)據(jù)長度16位也無法描述，則用64位，即8字節(jié)，如果16位能描述則用2字節(jié)，否則用第二個字節(jié)描述
 var index = 2 + (length > 65535 ? 8 : (length > 125 ? 2 : 0));
// 定義buffer，長度為描述字節(jié)長度 + message長度
 var buffer = new Buffer(index + length);
// 第一個字節(jié)，fin位為1，opcode為1
 buffer[0] = 129;
// 因為是由服務(wù)端發(fā)至客戶端，所以無需masked掩碼
 if (length > 65535) {
 buffer[1] = 127;
// 長度超過65535的則由8個字節(jié)表示，因為4個字節(jié)能表達(dá)的長度為4294967295，已經(jīng)完全夠用，因此直接將前面4個字節(jié)置0
 buffer.writeUInt32BE(0, 2);
 buffer.writeUInt32BE(length, 6);
 } else if (length > 125) {
 buffer[1] = 126;
// 長度超過125的話就由2個字節(jié)表示
 buffer.writeUInt16BE(length, 2);
 } else {
 buffer[1] = length;
 }
// 寫入正文
 buffer.write(message, index);
 this.socket.write(buffer);
};

//每隔10秒進行一次心跳檢測，若連續(xù)發(fā)出三次心跳卻沒收到響應(yīng)則關(guān)閉socket
WebSocket.prototype.checkHeartBeat = function () {
 var that = this;
 setTimeout(function () {
 if (that.state !== "OPEN") return;
 if (that.pingTimes >= 3) {
  that.close("time out");
  return;
 }
 //記錄心跳次數(shù)
 that.pingTimes++;
 that.sendPing();
 that.checkHeartBeat();
 }, 10000);
};
WebSocket.prototype.sendPing = function () {
 this.socket.write(new Buffer(['0x89', '0x0']))
};
WebSocket.prototype.sendPong = function () {
 this.socket.write(new Buffer(['0x8A', '0x0']))
};