mirror of
https://github.com/actions/setup-python
synced 2024-11-30 21:22:28 +08:00
402 lines
13 KiB
JavaScript
402 lines
13 KiB
JavaScript
// Copyright 2012 Joyent, Inc. All rights reserved.
|
|
|
|
var assert = require('assert-plus');
|
|
var crypto = require('crypto');
|
|
var http = require('http');
|
|
var util = require('util');
|
|
var sshpk = require('sshpk');
|
|
var jsprim = require('jsprim');
|
|
var utils = require('./utils');
|
|
|
|
var sprintf = require('util').format;
|
|
|
|
var HASH_ALGOS = utils.HASH_ALGOS;
|
|
var PK_ALGOS = utils.PK_ALGOS;
|
|
var InvalidAlgorithmError = utils.InvalidAlgorithmError;
|
|
var HttpSignatureError = utils.HttpSignatureError;
|
|
var validateAlgorithm = utils.validateAlgorithm;
|
|
|
|
///--- Globals
|
|
|
|
var AUTHZ_FMT =
|
|
'Signature keyId="%s",algorithm="%s",headers="%s",signature="%s"';
|
|
|
|
///--- Specific Errors
|
|
|
|
function MissingHeaderError(message) {
|
|
HttpSignatureError.call(this, message, MissingHeaderError);
|
|
}
|
|
util.inherits(MissingHeaderError, HttpSignatureError);
|
|
|
|
function StrictParsingError(message) {
|
|
HttpSignatureError.call(this, message, StrictParsingError);
|
|
}
|
|
util.inherits(StrictParsingError, HttpSignatureError);
|
|
|
|
/* See createSigner() */
|
|
function RequestSigner(options) {
|
|
assert.object(options, 'options');
|
|
|
|
var alg = [];
|
|
if (options.algorithm !== undefined) {
|
|
assert.string(options.algorithm, 'options.algorithm');
|
|
alg = validateAlgorithm(options.algorithm);
|
|
}
|
|
this.rs_alg = alg;
|
|
|
|
/*
|
|
* RequestSigners come in two varieties: ones with an rs_signFunc, and ones
|
|
* with an rs_signer.
|
|
*
|
|
* rs_signFunc-based RequestSigners have to build up their entire signing
|
|
* string within the rs_lines array and give it to rs_signFunc as a single
|
|
* concat'd blob. rs_signer-based RequestSigners can add a line at a time to
|
|
* their signing state by using rs_signer.update(), thus only needing to
|
|
* buffer the hash function state and one line at a time.
|
|
*/
|
|
if (options.sign !== undefined) {
|
|
assert.func(options.sign, 'options.sign');
|
|
this.rs_signFunc = options.sign;
|
|
|
|
} else if (alg[0] === 'hmac' && options.key !== undefined) {
|
|
assert.string(options.keyId, 'options.keyId');
|
|
this.rs_keyId = options.keyId;
|
|
|
|
if (typeof (options.key) !== 'string' && !Buffer.isBuffer(options.key))
|
|
throw (new TypeError('options.key for HMAC must be a string or Buffer'));
|
|
|
|
/*
|
|
* Make an rs_signer for HMACs, not a rs_signFunc -- HMACs digest their
|
|
* data in chunks rather than requiring it all to be given in one go
|
|
* at the end, so they are more similar to signers than signFuncs.
|
|
*/
|
|
this.rs_signer = crypto.createHmac(alg[1].toUpperCase(), options.key);
|
|
this.rs_signer.sign = function () {
|
|
var digest = this.digest('base64');
|
|
return ({
|
|
hashAlgorithm: alg[1],
|
|
toString: function () { return (digest); }
|
|
});
|
|
};
|
|
|
|
} else if (options.key !== undefined) {
|
|
var key = options.key;
|
|
if (typeof (key) === 'string' || Buffer.isBuffer(key))
|
|
key = sshpk.parsePrivateKey(key);
|
|
|
|
assert.ok(sshpk.PrivateKey.isPrivateKey(key, [1, 2]),
|
|
'options.key must be a sshpk.PrivateKey');
|
|
this.rs_key = key;
|
|
|
|
assert.string(options.keyId, 'options.keyId');
|
|
this.rs_keyId = options.keyId;
|
|
|
|
if (!PK_ALGOS[key.type]) {
|
|
throw (new InvalidAlgorithmError(key.type.toUpperCase() + ' type ' +
|
|
'keys are not supported'));
|
|
}
|
|
|
|
if (alg[0] !== undefined && key.type !== alg[0]) {
|
|
throw (new InvalidAlgorithmError('options.key must be a ' +
|
|
alg[0].toUpperCase() + ' key, was given a ' +
|
|
key.type.toUpperCase() + ' key instead'));
|
|
}
|
|
|
|
this.rs_signer = key.createSign(alg[1]);
|
|
|
|
} else {
|
|
throw (new TypeError('options.sign (func) or options.key is required'));
|
|
}
|
|
|
|
this.rs_headers = [];
|
|
this.rs_lines = [];
|
|
}
|
|
|
|
/**
|
|
* Adds a header to be signed, with its value, into this signer.
|
|
*
|
|
* @param {String} header
|
|
* @param {String} value
|
|
* @return {String} value written
|
|
*/
|
|
RequestSigner.prototype.writeHeader = function (header, value) {
|
|
assert.string(header, 'header');
|
|
header = header.toLowerCase();
|
|
assert.string(value, 'value');
|
|
|
|
this.rs_headers.push(header);
|
|
|
|
if (this.rs_signFunc) {
|
|
this.rs_lines.push(header + ': ' + value);
|
|
|
|
} else {
|
|
var line = header + ': ' + value;
|
|
if (this.rs_headers.length > 0)
|
|
line = '\n' + line;
|
|
this.rs_signer.update(line);
|
|
}
|
|
|
|
return (value);
|
|
};
|
|
|
|
/**
|
|
* Adds a default Date header, returning its value.
|
|
*
|
|
* @return {String}
|
|
*/
|
|
RequestSigner.prototype.writeDateHeader = function () {
|
|
return (this.writeHeader('date', jsprim.rfc1123(new Date())));
|
|
};
|
|
|
|
/**
|
|
* Adds the request target line to be signed.
|
|
*
|
|
* @param {String} method, HTTP method (e.g. 'get', 'post', 'put')
|
|
* @param {String} path
|
|
*/
|
|
RequestSigner.prototype.writeTarget = function (method, path) {
|
|
assert.string(method, 'method');
|
|
assert.string(path, 'path');
|
|
method = method.toLowerCase();
|
|
this.writeHeader('(request-target)', method + ' ' + path);
|
|
};
|
|
|
|
/**
|
|
* Calculate the value for the Authorization header on this request
|
|
* asynchronously.
|
|
*
|
|
* @param {Func} callback (err, authz)
|
|
*/
|
|
RequestSigner.prototype.sign = function (cb) {
|
|
assert.func(cb, 'callback');
|
|
|
|
if (this.rs_headers.length < 1)
|
|
throw (new Error('At least one header must be signed'));
|
|
|
|
var alg, authz;
|
|
if (this.rs_signFunc) {
|
|
var data = this.rs_lines.join('\n');
|
|
var self = this;
|
|
this.rs_signFunc(data, function (err, sig) {
|
|
if (err) {
|
|
cb(err);
|
|
return;
|
|
}
|
|
try {
|
|
assert.object(sig, 'signature');
|
|
assert.string(sig.keyId, 'signature.keyId');
|
|
assert.string(sig.algorithm, 'signature.algorithm');
|
|
assert.string(sig.signature, 'signature.signature');
|
|
alg = validateAlgorithm(sig.algorithm);
|
|
|
|
authz = sprintf(AUTHZ_FMT,
|
|
sig.keyId,
|
|
sig.algorithm,
|
|
self.rs_headers.join(' '),
|
|
sig.signature);
|
|
} catch (e) {
|
|
cb(e);
|
|
return;
|
|
}
|
|
cb(null, authz);
|
|
});
|
|
|
|
} else {
|
|
try {
|
|
var sigObj = this.rs_signer.sign();
|
|
} catch (e) {
|
|
cb(e);
|
|
return;
|
|
}
|
|
alg = (this.rs_alg[0] || this.rs_key.type) + '-' + sigObj.hashAlgorithm;
|
|
var signature = sigObj.toString();
|
|
authz = sprintf(AUTHZ_FMT,
|
|
this.rs_keyId,
|
|
alg,
|
|
this.rs_headers.join(' '),
|
|
signature);
|
|
cb(null, authz);
|
|
}
|
|
};
|
|
|
|
///--- Exported API
|
|
|
|
module.exports = {
|
|
/**
|
|
* Identifies whether a given object is a request signer or not.
|
|
*
|
|
* @param {Object} object, the object to identify
|
|
* @returns {Boolean}
|
|
*/
|
|
isSigner: function (obj) {
|
|
if (typeof (obj) === 'object' && obj instanceof RequestSigner)
|
|
return (true);
|
|
return (false);
|
|
},
|
|
|
|
/**
|
|
* Creates a request signer, used to asynchronously build a signature
|
|
* for a request (does not have to be an http.ClientRequest).
|
|
*
|
|
* @param {Object} options, either:
|
|
* - {String} keyId
|
|
* - {String|Buffer} key
|
|
* - {String} algorithm (optional, required for HMAC)
|
|
* or:
|
|
* - {Func} sign (data, cb)
|
|
* @return {RequestSigner}
|
|
*/
|
|
createSigner: function createSigner(options) {
|
|
return (new RequestSigner(options));
|
|
},
|
|
|
|
/**
|
|
* Adds an 'Authorization' header to an http.ClientRequest object.
|
|
*
|
|
* Note that this API will add a Date header if it's not already set. Any
|
|
* other headers in the options.headers array MUST be present, or this
|
|
* will throw.
|
|
*
|
|
* You shouldn't need to check the return type; it's just there if you want
|
|
* to be pedantic.
|
|
*
|
|
* The optional flag indicates whether parsing should use strict enforcement
|
|
* of the version draft-cavage-http-signatures-04 of the spec or beyond.
|
|
* The default is to be loose and support
|
|
* older versions for compatibility.
|
|
*
|
|
* @param {Object} request an instance of http.ClientRequest.
|
|
* @param {Object} options signing parameters object:
|
|
* - {String} keyId required.
|
|
* - {String} key required (either a PEM or HMAC key).
|
|
* - {Array} headers optional; defaults to ['date'].
|
|
* - {String} algorithm optional (unless key is HMAC);
|
|
* default is the same as the sshpk default
|
|
* signing algorithm for the type of key given
|
|
* - {String} httpVersion optional; defaults to '1.1'.
|
|
* - {Boolean} strict optional; defaults to 'false'.
|
|
* @return {Boolean} true if Authorization (and optionally Date) were added.
|
|
* @throws {TypeError} on bad parameter types (input).
|
|
* @throws {InvalidAlgorithmError} if algorithm was bad or incompatible with
|
|
* the given key.
|
|
* @throws {sshpk.KeyParseError} if key was bad.
|
|
* @throws {MissingHeaderError} if a header to be signed was specified but
|
|
* was not present.
|
|
*/
|
|
signRequest: function signRequest(request, options) {
|
|
assert.object(request, 'request');
|
|
assert.object(options, 'options');
|
|
assert.optionalString(options.algorithm, 'options.algorithm');
|
|
assert.string(options.keyId, 'options.keyId');
|
|
assert.optionalArrayOfString(options.headers, 'options.headers');
|
|
assert.optionalString(options.httpVersion, 'options.httpVersion');
|
|
|
|
if (!request.getHeader('Date'))
|
|
request.setHeader('Date', jsprim.rfc1123(new Date()));
|
|
if (!options.headers)
|
|
options.headers = ['date'];
|
|
if (!options.httpVersion)
|
|
options.httpVersion = '1.1';
|
|
|
|
var alg = [];
|
|
if (options.algorithm) {
|
|
options.algorithm = options.algorithm.toLowerCase();
|
|
alg = validateAlgorithm(options.algorithm);
|
|
}
|
|
|
|
var i;
|
|
var stringToSign = '';
|
|
for (i = 0; i < options.headers.length; i++) {
|
|
if (typeof (options.headers[i]) !== 'string')
|
|
throw new TypeError('options.headers must be an array of Strings');
|
|
|
|
var h = options.headers[i].toLowerCase();
|
|
|
|
if (h === 'request-line') {
|
|
if (!options.strict) {
|
|
/**
|
|
* We allow headers from the older spec drafts if strict parsing isn't
|
|
* specified in options.
|
|
*/
|
|
stringToSign +=
|
|
request.method + ' ' + request.path + ' HTTP/' +
|
|
options.httpVersion;
|
|
} else {
|
|
/* Strict parsing doesn't allow older draft headers. */
|
|
throw (new StrictParsingError('request-line is not a valid header ' +
|
|
'with strict parsing enabled.'));
|
|
}
|
|
} else if (h === '(request-target)') {
|
|
stringToSign +=
|
|
'(request-target): ' + request.method.toLowerCase() + ' ' +
|
|
request.path;
|
|
} else {
|
|
var value = request.getHeader(h);
|
|
if (value === undefined || value === '') {
|
|
throw new MissingHeaderError(h + ' was not in the request');
|
|
}
|
|
stringToSign += h + ': ' + value;
|
|
}
|
|
|
|
if ((i + 1) < options.headers.length)
|
|
stringToSign += '\n';
|
|
}
|
|
|
|
/* This is just for unit tests. */
|
|
if (request.hasOwnProperty('_stringToSign')) {
|
|
request._stringToSign = stringToSign;
|
|
}
|
|
|
|
var signature;
|
|
if (alg[0] === 'hmac') {
|
|
if (typeof (options.key) !== 'string' && !Buffer.isBuffer(options.key))
|
|
throw (new TypeError('options.key must be a string or Buffer'));
|
|
|
|
var hmac = crypto.createHmac(alg[1].toUpperCase(), options.key);
|
|
hmac.update(stringToSign);
|
|
signature = hmac.digest('base64');
|
|
|
|
} else {
|
|
var key = options.key;
|
|
if (typeof (key) === 'string' || Buffer.isBuffer(key))
|
|
key = sshpk.parsePrivateKey(options.key);
|
|
|
|
assert.ok(sshpk.PrivateKey.isPrivateKey(key, [1, 2]),
|
|
'options.key must be a sshpk.PrivateKey');
|
|
|
|
if (!PK_ALGOS[key.type]) {
|
|
throw (new InvalidAlgorithmError(key.type.toUpperCase() + ' type ' +
|
|
'keys are not supported'));
|
|
}
|
|
|
|
if (alg[0] !== undefined && key.type !== alg[0]) {
|
|
throw (new InvalidAlgorithmError('options.key must be a ' +
|
|
alg[0].toUpperCase() + ' key, was given a ' +
|
|
key.type.toUpperCase() + ' key instead'));
|
|
}
|
|
|
|
var signer = key.createSign(alg[1]);
|
|
signer.update(stringToSign);
|
|
var sigObj = signer.sign();
|
|
if (!HASH_ALGOS[sigObj.hashAlgorithm]) {
|
|
throw (new InvalidAlgorithmError(sigObj.hashAlgorithm.toUpperCase() +
|
|
' is not a supported hash algorithm'));
|
|
}
|
|
options.algorithm = key.type + '-' + sigObj.hashAlgorithm;
|
|
signature = sigObj.toString();
|
|
assert.notStrictEqual(signature, '', 'empty signature produced');
|
|
}
|
|
|
|
var authzHeaderName = options.authorizationHeaderName || 'Authorization';
|
|
|
|
request.setHeader(authzHeaderName, sprintf(AUTHZ_FMT,
|
|
options.keyId,
|
|
options.algorithm,
|
|
options.headers.join(' '),
|
|
signature));
|
|
|
|
return true;
|
|
}
|
|
|
|
};
|