diff --git a/lib/saml11.js b/lib/saml11.js index bc3512a..bb73197 100644 --- a/lib/saml11.js +++ b/lib/saml11.js @@ -57,17 +57,34 @@ function extractSaml11Options(opts) { * @param [options.encryptionAlgorithm] {string} * @param [options.keyEncryptionAlgorithm] {string} * + * @param [strategies] + * + * // Signing strategy + * @param {Function} [strategies.signXml] + * + * // Encryption strategy + * @param {Function} [strategies.encryptXml] + * * @param {Function} [callback] required if encrypting * @return {String|*} */ -exports.create = function(options, callback) { - return createAssertion(extractSaml11Options(options), { +exports.create = function(options, strategies, callback) { + var defaultStrategy = { signXml: SignXml.fromSignXmlOptions(Object.assign({ xpathToNodeBeforeSignature: "//*[local-name(.)='AuthenticationStatement']", signatureIdAttribute: 'AssertionID' }, options)), encryptXml: EncryptXml.fromEncryptXmlOptions(options) - }, callback); + } + + if (typeof strategies === 'function' && callback == null) { + callback = strategies + strategies = defaultStrategy + } else { + strategies = strategies || defaultStrategy + } + + return createAssertion(extractSaml11Options(options), strategies, callback); } /** @@ -177,9 +194,19 @@ function createAssertion(options, strategies, callback) { nameIDs[1].setAttribute('Format', options.nameIdentifierFormat); } + if (callback) { + signAndEncryptAsync(doc, strategies, options, callback) + } else { + return signAndEncryptSync(doc, strategies) + } +} + +function signAndEncryptAsync(doc, strategies, options, callback) { if (strategies.encryptXml === EncryptXml.unencrypted) { - var signed = strategies.signXml(doc); - return strategies.encryptXml(signed, callback); + strategies.signXml(doc, function(err, signed){ + strategies.encryptXml(signed, callback); + }); + return } // encryption is turned on, @@ -206,6 +233,13 @@ function createAssertion(options, strategies, callback) { }); } +function signAndEncryptSync(doc, strategies) { + if (strategies.encryptXml === EncryptXml.unencrypted) { + var signed = strategies.signXml(doc); + return strategies.encryptXml(signed); + } +} + function addSubjectConfirmation(encryptOptions, doc, randomBytes, callback) { xmlenc.encryptKeyInfo(randomBytes, encryptOptions, function(err, keyinfo) { if (err) return callback(err); diff --git a/lib/saml20.js b/lib/saml20.js index 9db8141..7a9d1b0 100644 --- a/lib/saml20.js +++ b/lib/saml20.js @@ -96,17 +96,34 @@ function extractSaml20Options(opts) { * @param [options.encryptionAlgorithm] {string} * @param [options.keyEncryptionAlgorithm] {string} * + * @param [strategies] + * + * // Signing strategy + * @param {Function} [strategies.signXml] + * + * // Encryption strategy + * @param {Function} [strategies.encryptXml] + * * @param {Function} [callback] required if encrypting * @return {*} */ -exports.create = function createSignedAssertion(options, callback) { - return createAssertion(extractSaml20Options(options), { +exports.create = function createSignedAssertion(options, strategies, callback) { + var defaultStrategy = { signXml: SignXml.fromSignXmlOptions(Object.assign({ xpathToNodeBeforeSignature: "//*[local-name(.)='Issuer']", signatureIdAttribute: 'ID' }, options)), encryptXml: EncryptXml.fromEncryptXmlOptions(options) - }, callback); + } + + if (typeof strategies === 'function' && callback == null) { + callback = strategies + strategies = defaultStrategy + } else { + strategies = strategies || defaultStrategy + } + + return createAssertion(extractSaml20Options(options), strategies, callback); }; /** @@ -251,6 +268,40 @@ function createAssertion(options, strategies, callback) { authnCtxClassRef.textContent = options.authnContextClassRef; } + if (callback) { + signAndEncryptAsync(doc, strategies, callback) + } else { + return signAndEncryptSync(doc, strategies) + } +} + +function signAndEncryptAsync(doc, strategies, callback) { + try { + strategies.signXml(doc, function(err, signed){ + if (err || !signed) { + return utils.reportError(err, callback); + } + + if (strategies.encryptXml === EncryptXml.unencrypted) { + return strategies.encryptXml(signed, callback); + } + + async.waterfall([ + function (cb) { + strategies.encryptXml(signed, cb) + }, + function (encrypted, cb) { + var assertion = '' + encrypted + ''; + cb(null, utils.removeWhitespace(assertion)); + }, + ], callback); + }); + } catch(err) { + utils.reportError(err, callback); + } +} + +function signAndEncryptSync(doc, strategies) { var signed; try { signed = strategies.signXml(doc); @@ -259,16 +310,6 @@ function createAssertion(options, strategies, callback) { } if (strategies.encryptXml === EncryptXml.unencrypted) { - return strategies.encryptXml(signed, callback); + return strategies.encryptXml(signed); } - - async.waterfall([ - function (cb) { - strategies.encryptXml(signed, cb) - }, - function (encrypted, cb) { - var assertion = '' + encrypted + ''; - cb(null, utils.removeWhitespace(assertion)); - }, - ], callback); } diff --git a/test/saml11.tests.js b/test/saml11.tests.js index 0e7a3c8..d37dada 100644 --- a/test/saml11.tests.js +++ b/test/saml11.tests.js @@ -8,6 +8,9 @@ var xmlenc = require('xml-encryption'); var utils = require('./utils'); var saml11 = require('../lib/saml11'); +var EncryptXml = require('../lib/xml/encrypt'); +var SignXml = require('../lib/xml/sign'); + describe('saml 1.1', function () { saml11TestSuite({ @@ -347,6 +350,39 @@ describe('saml 1.1', function () { }); }); + it('should create a saml 1.1 encrypted assertion using async strategy', function (done) { + var options = { + cert: fs.readFileSync(__dirname + '/test-auth0.pem'), + key: fs.readFileSync(__dirname + '/test-auth0.key'), + encryptionPublicKey: fs.readFileSync(__dirname + '/test-auth0_rsa.pub'), + encryptionCert: fs.readFileSync(__dirname + '/test-auth0.pem') + }; + + var strategies = { + signXml: SignXml.fromSignXmlOptions(Object.assign({ + xpathToNodeBeforeSignature: "//*[local-name(.)='AuthenticationStatement']", + signatureIdAttribute: 'AssertionID' + }, options)), + encryptXml: EncryptXml.fromEncryptXmlOptions(options) + } + + var callback = function(err, encrypted) { + if (err) return done(err); + + xmlenc.decrypt(encrypted, { key: fs.readFileSync(__dirname + '/test-auth0.key')}, function(err, decrypted) { + if (err) return done(err); + assertSignature(decrypted, options); + done(); + }); + } + + if (createAssertion === 'create'){ + saml11[createAssertion](options, strategies, callback); + } else { + saml11[createAssertion](options, callback); + } + }); + it('should support holder-of-key suject confirmationmethod', function (done) { var options = { cert: fs.readFileSync(__dirname + '/test-auth0.pem'), diff --git a/test/saml20.tests.js b/test/saml20.tests.js index b55a7b6..dd51e23 100644 --- a/test/saml20.tests.js +++ b/test/saml20.tests.js @@ -7,6 +7,8 @@ var xmldom = require('xmldom'); var xmlenc = require('xml-encryption'); var saml = require('../lib/saml20'); +var EncryptXml = require('../lib/xml/encrypt'); +var SignXml = require('../lib/xml/sign'); describe('saml 2.0', function () { saml20TestSuite({ @@ -513,6 +515,40 @@ describe('saml 2.0', function () { }); }); + it('should create a saml 2.0 signed and encrypted assertion with async strategy', function (done) { + var options = { + cert: fs.readFileSync(__dirname + '/test-auth0.pem'), + key: fs.readFileSync(__dirname + '/test-auth0.key'), + encryptionPublicKey: fs.readFileSync(__dirname + '/test-auth0_rsa.pub'), + encryptionCert: fs.readFileSync(__dirname + '/test-auth0.pem') + }; + + var strategies = { + signXml: SignXml.fromSignXmlOptions(Object.assign({ + xpathToNodeBeforeSignature: "//*[local-name(.)='Issuer']", + signatureIdAttribute: 'ID' + }, options)), + encryptXml: EncryptXml.fromEncryptXmlOptions(options) + } + + var callback = function(err, encrypted){ + if (err) return done(err); + var encryptedData = utils.getEncryptedData(encrypted); + + xmlenc.decrypt(encryptedData.toString(), { key: fs.readFileSync(__dirname + '/test-auth0.key') }, function (err, decrypted) { + if (err) return done(err); + assertSignature(decrypted, options); + done(); + }); + } + + if (createAssertion === 'create'){ + saml[createAssertion](options, strategies, callback) + } else { + saml[createAssertion](options, callback) + } + }); + it('should set attributes', function (done) { var options = { cert: fs.readFileSync(__dirname + '/test-auth0.pem'),