diff --git a/lib/saml11.js b/lib/saml11.js index 48c5988..d23c392 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/lib/xml/sign.js b/lib/xml/sign.js index 7c01fd0..d3de3a3 100644 --- a/lib/xml/sign.js +++ b/lib/xml/sign.js @@ -39,7 +39,7 @@ exports.fromSignXmlOptions = function (options) { * @return {string} */ return function signXmlDocument(doc, callback) { - function sign(key) { + function sign(key, signCallback) { const unsigned = exports.unsigned(doc); const cert = utils.pemToCert(pem); @@ -60,32 +60,59 @@ exports.fromSignXmlOptions = function (options) { } }; - sig.computeSignature(unsigned, { - location: {reference: xpathToNodeBeforeSignature, action: 'after'}, - prefix: signatureNamespacePrefix - }); + if (signCallback == null) { + sig.computeSignature(unsigned, { + location: {reference: xpathToNodeBeforeSignature, action: 'after'}, + prefix: signatureNamespacePrefix + }); - return sig.getSignedXml(); + return sig.getSignedXml(); + } else { + sig.computeSignature(unsigned, { + location: {reference: xpathToNodeBeforeSignature, action: 'after'}, + prefix: signatureNamespacePrefix + }, function (err) { + if (err != null) { + signCallback(err, null) + } else { + signCallback(null, sig.getSignedXml()) + } + }); + } } - let signed - try { - try { - signed = sign(key) - } catch (err) { - signed = sign(utils.fixPemFormatting(key)) - } + if (callback != null) { + sign(key, function(err, signed) { + if (err != null) { + sign(utils.fixPemFormatting(key), function(err, signed) { + setImmediate(callback, err, signed); + }) + return; + } - if (callback) { setImmediate(callback, null, signed); - } else { - return signed; - } - } catch (e) { - if (callback) { - setImmediate(callback, e) + }) + } else { + let signed + + try { + try { + signed = sign(key) + } catch (err) { + signed = sign(utils.fixPemFormatting(key)) + } + + if (callback) { + setImmediate(callback, null, signed); + } else { + return signed; + } + } catch (e) { + if (callback) { + setImmediate(callback, e) + } + throw e } - throw e } }; }; diff --git a/test/saml11.tests.js b/test/saml11.tests.js index 6d2841d..b9f37d4 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({ @@ -376,6 +379,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 not error when encryptionPublicKey is missing newlines', function (done) { var options = { cert: fs.readFileSync(__dirname + '/test-auth0.pem'), diff --git a/test/saml20.tests.js b/test/saml20.tests.js index b386341..75b8e9c 100644 --- a/test/saml20.tests.js +++ b/test/saml20.tests.js @@ -7,6 +7,8 @@ var xmldom = require('@xmldom/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({ @@ -603,6 +605,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 not error when encryptionPublicKey is missing newline', function (done) { var options = { cert: fs.readFileSync(__dirname + '/test-auth0.pem'),