forked from zhurui/management
118 lines
3.1 KiB
JavaScript
118 lines
3.1 KiB
JavaScript
'use strict';
|
|
|
|
var Buffer = require('safe-buffer').Buffer,
|
|
Base = require('./base'),
|
|
Draft75 = require('./draft75'),
|
|
crypto = require('crypto'),
|
|
util = require('util');
|
|
|
|
|
|
var numberFromKey = function(key) {
|
|
return parseInt((key.match(/[0-9]/g) || []).join(''), 10);
|
|
};
|
|
|
|
var spacesInKey = function(key) {
|
|
return (key.match(/ /g) || []).length;
|
|
};
|
|
|
|
|
|
var Draft76 = function(request, url, options) {
|
|
Draft75.apply(this, arguments);
|
|
this._stage = -1;
|
|
this._body = [];
|
|
this.version = 'hixie-76';
|
|
|
|
this._headers.clear();
|
|
|
|
this._headers.set('Upgrade', 'WebSocket');
|
|
this._headers.set('Connection', 'Upgrade');
|
|
this._headers.set('Sec-WebSocket-Origin', this._request.headers.origin);
|
|
this._headers.set('Sec-WebSocket-Location', this.url);
|
|
};
|
|
util.inherits(Draft76, Draft75);
|
|
|
|
var instance = {
|
|
BODY_SIZE: 8,
|
|
|
|
start: function() {
|
|
if (!Draft75.prototype.start.call(this)) return false;
|
|
this._started = true;
|
|
this._sendHandshakeBody();
|
|
return true;
|
|
},
|
|
|
|
close: function() {
|
|
if (this.readyState === 3) return false;
|
|
if (this.readyState === 1) this._write(Buffer.from([0xFF, 0x00]));
|
|
this.readyState = 3;
|
|
this.emit('close', new Base.CloseEvent(null, null));
|
|
return true;
|
|
},
|
|
|
|
_handshakeResponse: function() {
|
|
var headers = this._request.headers,
|
|
key1 = headers['sec-websocket-key1'],
|
|
key2 = headers['sec-websocket-key2'];
|
|
|
|
if (!key1) throw new Error('Missing required header: Sec-WebSocket-Key1');
|
|
if (!key2) throw new Error('Missing required header: Sec-WebSocket-Key2');
|
|
|
|
var number1 = numberFromKey(key1),
|
|
spaces1 = spacesInKey(key1),
|
|
|
|
number2 = numberFromKey(key2),
|
|
spaces2 = spacesInKey(key2);
|
|
|
|
if (number1 % spaces1 !== 0 || number2 % spaces2 !== 0)
|
|
throw new Error('Client sent invalid Sec-WebSocket-Key headers');
|
|
|
|
this._keyValues = [number1 / spaces1, number2 / spaces2];
|
|
|
|
var start = 'HTTP/1.1 101 WebSocket Protocol Handshake',
|
|
headers = [start, this._headers.toString(), ''];
|
|
|
|
return Buffer.from(headers.join('\r\n'), 'binary');
|
|
},
|
|
|
|
_handshakeSignature: function() {
|
|
if (this._body.length < this.BODY_SIZE) return null;
|
|
|
|
var md5 = crypto.createHash('md5'),
|
|
buffer = Buffer.allocUnsafe(8 + this.BODY_SIZE);
|
|
|
|
buffer.writeUInt32BE(this._keyValues[0], 0);
|
|
buffer.writeUInt32BE(this._keyValues[1], 4);
|
|
Buffer.from(this._body).copy(buffer, 8, 0, this.BODY_SIZE);
|
|
|
|
md5.update(buffer);
|
|
return Buffer.from(md5.digest('binary'), 'binary');
|
|
},
|
|
|
|
_sendHandshakeBody: function() {
|
|
if (!this._started) return;
|
|
var signature = this._handshakeSignature();
|
|
if (!signature) return;
|
|
|
|
this._write(signature);
|
|
this._stage = 0;
|
|
this._open();
|
|
|
|
if (this._body.length > this.BODY_SIZE)
|
|
this.parse(this._body.slice(this.BODY_SIZE));
|
|
},
|
|
|
|
_parseLeadingByte: function(octet) {
|
|
if (octet !== 0xFF)
|
|
return Draft75.prototype._parseLeadingByte.call(this, octet);
|
|
|
|
this._closing = true;
|
|
this._length = 0;
|
|
this._stage = 1;
|
|
}
|
|
};
|
|
|
|
for (var key in instance)
|
|
Draft76.prototype[key] = instance[key];
|
|
|
|
module.exports = Draft76;
|