2
1
mirror of https://github.com/Thream/socketio-jwt.git synced 2024-07-21 09:38:31 +02:00
socketio-jwt/lib/index.js

153 lines
4.1 KiB
JavaScript
Raw Normal View History

2013-06-05 13:38:33 +02:00
var xtend = require('xtend');
var jwt = require('jsonwebtoken');
var UnauthorizedError = require('./UnauthorizedError');
2014-01-14 12:30:39 +01:00
function noQsMethod(options) {
2015-05-29 13:52:14 +02:00
var defaults = { required: true };
options = xtend(defaults, options);
2014-01-13 22:41:10 +01:00
return function (socket) {
2015-08-31 16:04:04 +02:00
var server = this.server || socket.server;
2014-01-14 12:30:39 +01:00
if (!server.$emit) {
//then is socket.io 1.0
2015-08-31 16:04:04 +02:00
var Namespace = Object.getPrototypeOf(server.sockets).constructor;
if (!~Namespace.events.indexOf('authenticated')) {
Namespace.events.push('authenticated');
}
}
2015-05-29 13:52:14 +02:00
if(options.required){
var auth_timeout = setTimeout(function () {
socket.disconnect('unauthorized');
}, options.timeout || 5000);
}
2014-01-13 22:41:10 +01:00
socket.on('authenticate', function (data) {
2015-05-29 13:52:14 +02:00
if(options.required){
clearTimeout(auth_timeout);
}
2014-01-13 22:41:10 +01:00
jwt.verify(data.token, options.secret, options, function(err, decoded) {
// error handler
var onError = function(err, code) {
if (err) {
code = code || 'unknown';
var error = new UnauthorizedError(code, {
message: (Object.prototype.toString.call(err) === '[object Object]' && err.message) ? err.message : err
});
socket.emit('unauthorized', error, function() {
socket.disconnect('unauthorized');
});
return; // stop logic, socket will be close on next tick
}
};
2015-05-29 13:52:14 +02:00
if (err) {
return onError(err, 'invalid_token');
2014-01-13 22:41:10 +01:00
}
2015-05-29 13:52:14 +02:00
// success handler
var onSuccess = function(){
socket.decoded_token = decoded;
socket.emit('authenticated');
if (server.$emit) {
server.$emit('authenticated', socket);
} else {
2015-08-31 16:04:04 +02:00
//try getting the current namespace otherwise fallback to all sockets.
var namespace = (server.nsps && socket.nsp &&
server.nsps[socket.nsp.name]) ||
server.sockets;
// explicit namespace
2015-08-31 16:04:04 +02:00
namespace.emit('authenticated', socket);
}
};
2015-05-29 13:52:14 +02:00
if(options.additional_auth && typeof options.additional_auth === 'function') {
2015-07-05 18:57:24 +02:00
options.additional_auth(decoded, onSuccess, onError);
} else {
onSuccess();
}
2014-01-13 22:41:10 +01:00
});
});
};
}
function authorize(options, onConnection) {
2015-05-29 13:52:14 +02:00
if (!options.handshake) {
return noQsMethod(options);
}
var defaults = {
success: function(data, accept){
2014-06-03 13:12:07 +02:00
if (data.request) {
accept();
} else {
accept(null, true);
}
},
fail: function(error, data, accept){
2014-06-03 13:12:07 +02:00
if (data.request) {
accept(error);
2014-06-03 13:12:07 +02:00
} else {
accept(null, false);
}
}
};
2012-09-05 20:14:36 +02:00
2013-11-15 10:47:51 +01:00
var auth = xtend(defaults, options);
2013-06-30 21:06:21 +02:00
2012-09-05 20:14:36 +02:00
return function(data, accept){
var token, error;
var req = data.request || data;
var authorization_header = (req.headers || {}).authorization;
2014-06-03 13:12:07 +02:00
if (authorization_header) {
var parts = authorization_header.split(' ');
if (parts.length == 2) {
var scheme = parts[0],
credentials = parts[1];
2012-09-05 20:14:36 +02:00
2015-05-18 03:05:00 +02:00
if (scheme.toLowerCase() === 'bearer') {
token = credentials;
}
} else {
error = new UnauthorizedError('credentials_bad_format', {
message: 'Format is Authorization: Bearer [token]'
});
return auth.fail(error, data, accept);
}
}
2014-06-03 13:12:07 +02:00
//get the token from query string
if (req._query && req._query.token) {
token = req._query.token;
}
else if (req.query && req.query.token) {
token = req.query.token;
}
2012-09-05 20:14:36 +02:00
if (!token) {
error = new UnauthorizedError('credentials_required', {
message: 'No Authorization header was found'
2012-09-05 20:14:36 +02:00
});
return auth.fail(error, data, accept);
}
jwt.verify(token, options.secret, options, function(err, decoded) {
if (err) {
error = new UnauthorizedError('invalid_token', err);
return auth.fail(error, data, accept);
}
2014-01-14 21:44:03 +01:00
data.decoded_token = decoded;
2012-09-05 20:14:36 +02:00
return auth.success(data, accept);
2012-09-05 20:14:36 +02:00
});
};
}
exports.authorize = authorize;