forked from zhurui/management
143 lines
4.2 KiB
JavaScript
143 lines
4.2 KiB
JavaScript
'use strict';
|
|
|
|
var Buffer = require('safe-buffer').Buffer,
|
|
crypto = require('crypto'),
|
|
url = require('url'),
|
|
util = require('util'),
|
|
HttpParser = require('../http_parser'),
|
|
Base = require('./base'),
|
|
Hybi = require('./hybi'),
|
|
Proxy = require('./proxy');
|
|
|
|
var Client = function(_url, options) {
|
|
this.version = 'hybi-' + Hybi.VERSION;
|
|
Hybi.call(this, null, _url, options);
|
|
|
|
this.readyState = -1;
|
|
this._key = Client.generateKey();
|
|
this._accept = Hybi.generateAccept(this._key);
|
|
this._http = new HttpParser('response');
|
|
|
|
var uri = url.parse(this.url),
|
|
auth = uri.auth && Buffer.from(uri.auth, 'utf8').toString('base64');
|
|
|
|
if (this.VALID_PROTOCOLS.indexOf(uri.protocol) < 0)
|
|
throw new Error(this.url + ' is not a valid WebSocket URL');
|
|
|
|
this._pathname = (uri.pathname || '/') + (uri.search || '');
|
|
|
|
this._headers.set('Host', uri.host);
|
|
this._headers.set('Upgrade', 'websocket');
|
|
this._headers.set('Connection', 'Upgrade');
|
|
this._headers.set('Sec-WebSocket-Key', this._key);
|
|
this._headers.set('Sec-WebSocket-Version', Hybi.VERSION);
|
|
|
|
if (this._protocols.length > 0)
|
|
this._headers.set('Sec-WebSocket-Protocol', this._protocols.join(', '));
|
|
|
|
if (auth)
|
|
this._headers.set('Authorization', 'Basic ' + auth);
|
|
};
|
|
util.inherits(Client, Hybi);
|
|
|
|
Client.generateKey = function() {
|
|
return crypto.randomBytes(16).toString('base64');
|
|
};
|
|
|
|
var instance = {
|
|
VALID_PROTOCOLS: ['ws:', 'wss:'],
|
|
|
|
proxy: function(origin, options) {
|
|
return new Proxy(this, origin, options);
|
|
},
|
|
|
|
start: function() {
|
|
if (this.readyState !== -1) return false;
|
|
this._write(this._handshakeRequest());
|
|
this.readyState = 0;
|
|
return true;
|
|
},
|
|
|
|
parse: function(chunk) {
|
|
if (this.readyState === 3) return;
|
|
if (this.readyState > 0) return Hybi.prototype.parse.call(this, chunk);
|
|
|
|
this._http.parse(chunk);
|
|
if (!this._http.isComplete()) return;
|
|
|
|
this._validateHandshake();
|
|
if (this.readyState === 3) return;
|
|
|
|
this._open();
|
|
this.parse(this._http.body);
|
|
},
|
|
|
|
_handshakeRequest: function() {
|
|
var extensions = this._extensions.generateOffer();
|
|
if (extensions)
|
|
this._headers.set('Sec-WebSocket-Extensions', extensions);
|
|
|
|
var start = 'GET ' + this._pathname + ' HTTP/1.1',
|
|
headers = [start, this._headers.toString(), ''];
|
|
|
|
return Buffer.from(headers.join('\r\n'), 'utf8');
|
|
},
|
|
|
|
_failHandshake: function(message) {
|
|
message = 'Error during WebSocket handshake: ' + message;
|
|
this.readyState = 3;
|
|
this.emit('error', new Error(message));
|
|
this.emit('close', new Base.CloseEvent(this.ERRORS.protocol_error, message));
|
|
},
|
|
|
|
_validateHandshake: function() {
|
|
this.statusCode = this._http.statusCode;
|
|
this.headers = this._http.headers;
|
|
|
|
if (this._http.error)
|
|
return this._failHandshake(this._http.error.message);
|
|
|
|
if (this._http.statusCode !== 101)
|
|
return this._failHandshake('Unexpected response code: ' + this._http.statusCode);
|
|
|
|
var headers = this._http.headers,
|
|
upgrade = headers['upgrade'] || '',
|
|
connection = headers['connection'] || '',
|
|
accept = headers['sec-websocket-accept'] || '',
|
|
protocol = headers['sec-websocket-protocol'] || '';
|
|
|
|
if (upgrade === '')
|
|
return this._failHandshake("'Upgrade' header is missing");
|
|
if (upgrade.toLowerCase() !== 'websocket')
|
|
return this._failHandshake("'Upgrade' header value is not 'WebSocket'");
|
|
|
|
if (connection === '')
|
|
return this._failHandshake("'Connection' header is missing");
|
|
if (connection.toLowerCase() !== 'upgrade')
|
|
return this._failHandshake("'Connection' header value is not 'Upgrade'");
|
|
|
|
if (accept !== this._accept)
|
|
return this._failHandshake('Sec-WebSocket-Accept mismatch');
|
|
|
|
this.protocol = null;
|
|
|
|
if (protocol !== '') {
|
|
if (this._protocols.indexOf(protocol) < 0)
|
|
return this._failHandshake('Sec-WebSocket-Protocol mismatch');
|
|
else
|
|
this.protocol = protocol;
|
|
}
|
|
|
|
try {
|
|
this._extensions.activate(this.headers['sec-websocket-extensions']);
|
|
} catch (e) {
|
|
return this._failHandshake(e.message);
|
|
}
|
|
}
|
|
};
|
|
|
|
for (var key in instance)
|
|
Client.prototype[key] = instance[key];
|
|
|
|
module.exports = Client;
|