385 lines
12 KiB
JavaScript
385 lines
12 KiB
JavaScript
'use strict';
|
|
|
|
require('./shims');
|
|
|
|
var URL = require('url-parse')
|
|
, inherits = require('inherits')
|
|
, JSON3 = require('json3')
|
|
, random = require('./utils/random')
|
|
, escape = require('./utils/escape')
|
|
, urlUtils = require('./utils/url')
|
|
, eventUtils = require('./utils/event')
|
|
, transport = require('./utils/transport')
|
|
, objectUtils = require('./utils/object')
|
|
, browser = require('./utils/browser')
|
|
, log = require('./utils/log')
|
|
, Event = require('./event/event')
|
|
, EventTarget = require('./event/eventtarget')
|
|
, loc = require('./location')
|
|
, CloseEvent = require('./event/close')
|
|
, TransportMessageEvent = require('./event/trans-message')
|
|
, InfoReceiver = require('./info-receiver')
|
|
;
|
|
|
|
var debug = function() {};
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
debug = require('debug')('sockjs-client:main');
|
|
}
|
|
|
|
var transports;
|
|
|
|
// follow constructor steps defined at http://dev.w3.org/html5/websockets/#the-websocket-interface
|
|
function SockJS(url, protocols, options) {
|
|
if (!(this instanceof SockJS)) {
|
|
return new SockJS(url, protocols, options);
|
|
}
|
|
if (arguments.length < 1) {
|
|
throw new TypeError("Failed to construct 'SockJS: 1 argument required, but only 0 present");
|
|
}
|
|
EventTarget.call(this);
|
|
|
|
this.readyState = SockJS.CONNECTING;
|
|
this.extensions = '';
|
|
this.protocol = '';
|
|
|
|
// non-standard extension
|
|
options = options || {};
|
|
if (options.protocols_whitelist) {
|
|
log.warn("'protocols_whitelist' is DEPRECATED. Use 'transports' instead.");
|
|
}
|
|
this._transportsWhitelist = options.transports;
|
|
this._transportOptions = options.transportOptions || {};
|
|
|
|
var sessionId = options.sessionId || 8;
|
|
if (typeof sessionId === 'function') {
|
|
this._generateSessionId = sessionId;
|
|
} else if (typeof sessionId === 'number') {
|
|
this._generateSessionId = function() {
|
|
return random.string(sessionId);
|
|
};
|
|
} else {
|
|
throw new TypeError('If sessionId is used in the options, it needs to be a number or a function.');
|
|
}
|
|
|
|
this._server = options.server || random.numberString(1000);
|
|
|
|
// Step 1 of WS spec - parse and validate the url. Issue #8
|
|
var parsedUrl = new URL(url);
|
|
if (!parsedUrl.host || !parsedUrl.protocol) {
|
|
throw new SyntaxError("The URL '" + url + "' is invalid");
|
|
} else if (parsedUrl.hash) {
|
|
throw new SyntaxError('The URL must not contain a fragment');
|
|
} else if (parsedUrl.protocol !== 'http:' && parsedUrl.protocol !== 'https:') {
|
|
throw new SyntaxError("The URL's scheme must be either 'http:' or 'https:'. '" + parsedUrl.protocol + "' is not allowed.");
|
|
}
|
|
|
|
var secure = parsedUrl.protocol === 'https:';
|
|
// Step 2 - don't allow secure origin with an insecure protocol
|
|
if (loc.protocol === 'https:' && !secure) {
|
|
throw new Error('SecurityError: An insecure SockJS connection may not be initiated from a page loaded over HTTPS');
|
|
}
|
|
|
|
// Step 3 - check port access - no need here
|
|
// Step 4 - parse protocols argument
|
|
if (!protocols) {
|
|
protocols = [];
|
|
} else if (!Array.isArray(protocols)) {
|
|
protocols = [protocols];
|
|
}
|
|
|
|
// Step 5 - check protocols argument
|
|
var sortedProtocols = protocols.sort();
|
|
sortedProtocols.forEach(function(proto, i) {
|
|
if (!proto) {
|
|
throw new SyntaxError("The protocols entry '" + proto + "' is invalid.");
|
|
}
|
|
if (i < (sortedProtocols.length - 1) && proto === sortedProtocols[i + 1]) {
|
|
throw new SyntaxError("The protocols entry '" + proto + "' is duplicated.");
|
|
}
|
|
});
|
|
|
|
// Step 6 - convert origin
|
|
var o = urlUtils.getOrigin(loc.href);
|
|
this._origin = o ? o.toLowerCase() : null;
|
|
|
|
// remove the trailing slash
|
|
parsedUrl.set('pathname', parsedUrl.pathname.replace(/\/+$/, ''));
|
|
|
|
// store the sanitized url
|
|
this.url = parsedUrl.href;
|
|
debug('using url', this.url);
|
|
|
|
// Step 7 - start connection in background
|
|
// obtain server info
|
|
// http://sockjs.github.io/sockjs-protocol/sockjs-protocol-0.3.3.html#section-26
|
|
this._urlInfo = {
|
|
nullOrigin: !browser.hasDomain()
|
|
, sameOrigin: urlUtils.isOriginEqual(this.url, loc.href)
|
|
, sameScheme: urlUtils.isSchemeEqual(this.url, loc.href)
|
|
};
|
|
|
|
this._ir = new InfoReceiver(this.url, this._urlInfo);
|
|
this._ir.once('finish', this._receiveInfo.bind(this));
|
|
}
|
|
|
|
inherits(SockJS, EventTarget);
|
|
|
|
function userSetCode(code) {
|
|
return code === 1000 || (code >= 3000 && code <= 4999);
|
|
}
|
|
|
|
SockJS.prototype.close = function(code, reason) {
|
|
// Step 1
|
|
if (code && !userSetCode(code)) {
|
|
throw new Error('InvalidAccessError: Invalid code');
|
|
}
|
|
// Step 2.4 states the max is 123 bytes, but we are just checking length
|
|
if (reason && reason.length > 123) {
|
|
throw new SyntaxError('reason argument has an invalid length');
|
|
}
|
|
|
|
// Step 3.1
|
|
if (this.readyState === SockJS.CLOSING || this.readyState === SockJS.CLOSED) {
|
|
return;
|
|
}
|
|
|
|
// TODO look at docs to determine how to set this
|
|
var wasClean = true;
|
|
this._close(code || 1000, reason || 'Normal closure', wasClean);
|
|
};
|
|
|
|
SockJS.prototype.send = function(data) {
|
|
// #13 - convert anything non-string to string
|
|
// TODO this currently turns objects into [object Object]
|
|
if (typeof data !== 'string') {
|
|
data = '' + data;
|
|
}
|
|
if (this.readyState === SockJS.CONNECTING) {
|
|
throw new Error('InvalidStateError: The connection has not been established yet');
|
|
}
|
|
if (this.readyState !== SockJS.OPEN) {
|
|
return;
|
|
}
|
|
this._transport.send(escape.quote(data));
|
|
};
|
|
|
|
SockJS.version = require('./version');
|
|
|
|
SockJS.CONNECTING = 0;
|
|
SockJS.OPEN = 1;
|
|
SockJS.CLOSING = 2;
|
|
SockJS.CLOSED = 3;
|
|
|
|
SockJS.prototype._receiveInfo = function(info, rtt) {
|
|
debug('_receiveInfo', rtt);
|
|
this._ir = null;
|
|
if (!info) {
|
|
this._close(1002, 'Cannot connect to server');
|
|
return;
|
|
}
|
|
|
|
// establish a round-trip timeout (RTO) based on the
|
|
// round-trip time (RTT)
|
|
this._rto = this.countRTO(rtt);
|
|
// allow server to override url used for the actual transport
|
|
this._transUrl = info.base_url ? info.base_url : this.url;
|
|
info = objectUtils.extend(info, this._urlInfo);
|
|
debug('info', info);
|
|
// determine list of desired and supported transports
|
|
var enabledTransports = transports.filterToEnabled(this._transportsWhitelist, info);
|
|
this._transports = enabledTransports.main;
|
|
debug(this._transports.length + ' enabled transports');
|
|
|
|
this._connect();
|
|
};
|
|
|
|
SockJS.prototype._connect = function() {
|
|
for (var Transport = this._transports.shift(); Transport; Transport = this._transports.shift()) {
|
|
debug('attempt', Transport.transportName);
|
|
if (Transport.needBody) {
|
|
if (!global.document.body ||
|
|
(typeof global.document.readyState !== 'undefined' &&
|
|
global.document.readyState !== 'complete' &&
|
|
global.document.readyState !== 'interactive')) {
|
|
debug('waiting for body');
|
|
this._transports.unshift(Transport);
|
|
eventUtils.attachEvent('load', this._connect.bind(this));
|
|
return;
|
|
}
|
|
}
|
|
|
|
// calculate timeout based on RTO and round trips. Default to 5s
|
|
var timeoutMs = (this._rto * Transport.roundTrips) || 5000;
|
|
this._transportTimeoutId = setTimeout(this._transportTimeout.bind(this), timeoutMs);
|
|
debug('using timeout', timeoutMs);
|
|
|
|
var transportUrl = urlUtils.addPath(this._transUrl, '/' + this._server + '/' + this._generateSessionId());
|
|
var options = this._transportOptions[Transport.transportName];
|
|
debug('transport url', transportUrl);
|
|
var transportObj = new Transport(transportUrl, this._transUrl, options);
|
|
transportObj.on('message', this._transportMessage.bind(this));
|
|
transportObj.once('close', this._transportClose.bind(this));
|
|
transportObj.transportName = Transport.transportName;
|
|
this._transport = transportObj;
|
|
|
|
return;
|
|
}
|
|
this._close(2000, 'All transports failed', false);
|
|
};
|
|
|
|
SockJS.prototype._transportTimeout = function() {
|
|
debug('_transportTimeout');
|
|
if (this.readyState === SockJS.CONNECTING) {
|
|
if (this._transport) {
|
|
this._transport.close();
|
|
}
|
|
|
|
this._transportClose(2007, 'Transport timed out');
|
|
}
|
|
};
|
|
|
|
SockJS.prototype._transportMessage = function(msg) {
|
|
debug('_transportMessage', msg);
|
|
var self = this
|
|
, type = msg.slice(0, 1)
|
|
, content = msg.slice(1)
|
|
, payload
|
|
;
|
|
|
|
// first check for messages that don't need a payload
|
|
switch (type) {
|
|
case 'o':
|
|
this._open();
|
|
return;
|
|
case 'h':
|
|
this.dispatchEvent(new Event('heartbeat'));
|
|
debug('heartbeat', this.transport);
|
|
return;
|
|
}
|
|
|
|
if (content) {
|
|
try {
|
|
payload = JSON3.parse(content);
|
|
} catch (e) {
|
|
debug('bad json', content);
|
|
}
|
|
}
|
|
|
|
if (typeof payload === 'undefined') {
|
|
debug('empty payload', content);
|
|
return;
|
|
}
|
|
|
|
switch (type) {
|
|
case 'a':
|
|
if (Array.isArray(payload)) {
|
|
payload.forEach(function(p) {
|
|
debug('message', self.transport, p);
|
|
self.dispatchEvent(new TransportMessageEvent(p));
|
|
});
|
|
}
|
|
break;
|
|
case 'm':
|
|
debug('message', this.transport, payload);
|
|
this.dispatchEvent(new TransportMessageEvent(payload));
|
|
break;
|
|
case 'c':
|
|
if (Array.isArray(payload) && payload.length === 2) {
|
|
this._close(payload[0], payload[1], true);
|
|
}
|
|
break;
|
|
}
|
|
};
|
|
|
|
SockJS.prototype._transportClose = function(code, reason) {
|
|
debug('_transportClose', this.transport, code, reason);
|
|
if (this._transport) {
|
|
this._transport.removeAllListeners();
|
|
this._transport = null;
|
|
this.transport = null;
|
|
}
|
|
|
|
if (!userSetCode(code) && code !== 2000 && this.readyState === SockJS.CONNECTING) {
|
|
this._connect();
|
|
return;
|
|
}
|
|
|
|
this._close(code, reason);
|
|
};
|
|
|
|
SockJS.prototype._open = function() {
|
|
debug('_open', this._transport.transportName, this.readyState);
|
|
if (this.readyState === SockJS.CONNECTING) {
|
|
if (this._transportTimeoutId) {
|
|
clearTimeout(this._transportTimeoutId);
|
|
this._transportTimeoutId = null;
|
|
}
|
|
this.readyState = SockJS.OPEN;
|
|
this.transport = this._transport.transportName;
|
|
this.dispatchEvent(new Event('open'));
|
|
debug('connected', this.transport);
|
|
} else {
|
|
// The server might have been restarted, and lost track of our
|
|
// connection.
|
|
this._close(1006, 'Server lost session');
|
|
}
|
|
};
|
|
|
|
SockJS.prototype._close = function(code, reason, wasClean) {
|
|
debug('_close', this.transport, code, reason, wasClean, this.readyState);
|
|
var forceFail = false;
|
|
|
|
if (this._ir) {
|
|
forceFail = true;
|
|
this._ir.close();
|
|
this._ir = null;
|
|
}
|
|
if (this._transport) {
|
|
this._transport.close();
|
|
this._transport = null;
|
|
this.transport = null;
|
|
}
|
|
|
|
if (this.readyState === SockJS.CLOSED) {
|
|
throw new Error('InvalidStateError: SockJS has already been closed');
|
|
}
|
|
|
|
this.readyState = SockJS.CLOSING;
|
|
setTimeout(function() {
|
|
this.readyState = SockJS.CLOSED;
|
|
|
|
if (forceFail) {
|
|
this.dispatchEvent(new Event('error'));
|
|
}
|
|
|
|
var e = new CloseEvent('close');
|
|
e.wasClean = wasClean || false;
|
|
e.code = code || 1000;
|
|
e.reason = reason;
|
|
|
|
this.dispatchEvent(e);
|
|
this.onmessage = this.onclose = this.onerror = null;
|
|
debug('disconnected');
|
|
}.bind(this), 0);
|
|
};
|
|
|
|
// See: http://www.erg.abdn.ac.uk/~gerrit/dccp/notes/ccid2/rto_estimator/
|
|
// and RFC 2988.
|
|
SockJS.prototype.countRTO = function(rtt) {
|
|
// In a local environment, when using IE8/9 and the `jsonp-polling`
|
|
// transport the time needed to establish a connection (the time that pass
|
|
// from the opening of the transport to the call of `_dispatchOpen`) is
|
|
// around 200msec (the lower bound used in the article above) and this
|
|
// causes spurious timeouts. For this reason we calculate a value slightly
|
|
// larger than that used in the article.
|
|
if (rtt > 100) {
|
|
return 4 * rtt; // rto > 400msec
|
|
}
|
|
return 300 + rtt; // 300msec < rto <= 400msec
|
|
};
|
|
|
|
module.exports = function(availableTransports) {
|
|
transports = transport(availableTransports);
|
|
require('./iframe-bootstrap')(SockJS, availableTransports);
|
|
return SockJS;
|
|
};
|