Skip to content

Commit

Permalink
lib: refactor to reuse validators
Browse files Browse the repository at this point in the history
  • Loading branch information
pd4d10 committed May 9, 2021
1 parent 7c8a608 commit f55af36
Show file tree
Hide file tree
Showing 10 changed files with 41 additions and 75 deletions.
12 changes: 6 additions & 6 deletions lib/_http_agent.js
Original file line number Diff line number Diff line change
Expand Up @@ -51,12 +51,15 @@ const { AsyncResource } = require('async_hooks');
const { async_id_symbol } = require('internal/async_hooks').symbols;
const {
codes: {
ERR_INVALID_ARG_TYPE,
ERR_OUT_OF_RANGE,
},
} = require('internal/errors');
const { once } = require('internal/util');
const { validateNumber, validateOneOf } = require('internal/validators');
const {
validateNumber,
validateOneOf,
validateString
} = require('internal/validators');

const kOnKeylog = Symbol('onkeylog');
const kRequestOptions = Symbol('requestOptions');
Expand Down Expand Up @@ -344,10 +347,7 @@ function calculateServerName(options, req) {
let servername = options.host;
const hostHeader = req.getHeader('host');
if (hostHeader) {
if (typeof hostHeader !== 'string') {
throw new ERR_INVALID_ARG_TYPE('options.headers.host',
'String', hostHeader);
}
validateString(hostHeader, 'options.headers.host');

// abc => abc
// abc:123 => abc
Expand Down
28 changes: 8 additions & 20 deletions lib/_tls_wrap.js
Original file line number Diff line number Diff line change
Expand Up @@ -90,6 +90,9 @@ const {
validateObject,
validateString,
validateUint32,
validateBoolean,
validateNumber,
validateFunction
} = require('internal/validators');
const {
InternalX509Certificate
Expand Down Expand Up @@ -468,10 +471,8 @@ function TLSSocket(socket, opts) {
process.emitWarning('Enabling --trace-tls can expose sensitive data in ' +
'the resulting log.');
}
} else if (typeof enableTrace !== 'boolean') {
throw new ERR_INVALID_ARG_TYPE(
'options.enableTrace', 'boolean', enableTrace);
}
} else
validateBoolean(enableTrace, 'options.enableTrace');

if (tlsOptions.ALPNProtocols)
tls.convertALPNProtocols(tlsOptions.ALPNProtocols, tlsOptions);
Expand Down Expand Up @@ -783,11 +784,7 @@ TLSSocket.prototype._init = function(socket, wrap) {
}

if (options.pskCallback && ssl.enablePskCallback) {
if (typeof options.pskCallback !== 'function') {
throw new ERR_INVALID_ARG_TYPE('pskCallback',
'function',
options.pskCallback);
}
validateFunction(options.pskCallback, 'pskCallback');

ssl[kOnPskExchange] = options.isServer ?
onPskServerCallback : onPskClientCallback;
Expand All @@ -796,13 +793,7 @@ TLSSocket.prototype._init = function(socket, wrap) {
ssl.enablePskCallback();

if (options.pskIdentityHint) {
if (typeof options.pskIdentityHint !== 'string') {
throw new ERR_INVALID_ARG_TYPE(
'options.pskIdentityHint',
'string',
options.pskIdentityHint
);
}
validateString(options.pskIdentityHint, 'options.pskIdentityHint');
ssl.setPskIdentityHint(options.pskIdentityHint);
}
}
Expand Down Expand Up @@ -1215,10 +1206,7 @@ function Server(options, listener) {
this[kPskCallback] = options.pskCallback;
this[kPskIdentityHint] = options.pskIdentityHint;

if (typeof this[kHandshakeTimeout] !== 'number') {
throw new ERR_INVALID_ARG_TYPE(
'options.handshakeTimeout', 'number', options.handshakeTimeout);
}
validateNumber(this[kHandshakeTimeout], 'options.handshakeTimeout');

if (this[kSNICallback] && typeof this[kSNICallback] !== 'function') {
throw new ERR_INVALID_ARG_TYPE(
Expand Down
11 changes: 3 additions & 8 deletions lib/events.js
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@ const {
const {
validateAbortSignal,
validateFunction,
validateBoolean
} = require('internal/validators');

const kCapture = Symbol('kCapture');
Expand All @@ -89,10 +90,7 @@ ObjectDefineProperty(EventEmitter, 'captureRejections', {
return EventEmitter.prototype[kCapture];
},
set(value) {
if (typeof value !== 'boolean') {
throw new ERR_INVALID_ARG_TYPE('EventEmitter.captureRejections',
'boolean', value);
}
validateBoolean(value, 'EventEmitter.captureRejections');

EventEmitter.prototype[kCapture] = value;
},
Expand Down Expand Up @@ -190,10 +188,7 @@ EventEmitter.init = function(opts) {


if (opts?.captureRejections) {
if (typeof opts.captureRejections !== 'boolean') {
throw new ERR_INVALID_ARG_TYPE('options.captureRejections',
'boolean', opts.captureRejections);
}
validateBoolean(opts.captureRejections, 'options.captureRejections');
this[kCapture] = Boolean(opts.captureRejections);
} else {
// Assigning the kCapture property directly saves an expensive
Expand Down
6 changes: 3 additions & 3 deletions lib/internal/perf/timerify.js
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,8 @@ const {
} = require('internal/perf/perf');

const {
validateObject
validateObject,
validateFunction
} = require('internal/validators');

const {
Expand Down Expand Up @@ -57,8 +58,7 @@ function processComplete(name, start, args, histogram) {
}

function timerify(fn, options = {}) {
if (typeof fn !== 'function')
throw new ERR_INVALID_ARG_TYPE('fn', 'function', fn);
validateFunction(fn, 'fn');

validateObject(options, 'options');
const {
Expand Down
11 changes: 3 additions & 8 deletions lib/internal/process/per_thread.js
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@ const format = require('internal/util/inspect').format;
const {
validateArray,
validateObject,
validateNumber
} = require('internal/validators');
const constants = internalBinding('constants').os.signals;

Expand Down Expand Up @@ -122,19 +123,13 @@ function wrapProcessMethods(binding) {
if (!previousValueIsValid(prevValue.user)) {
validateObject(prevValue, 'prevValue');

if (typeof prevValue.user !== 'number') {
throw new ERR_INVALID_ARG_TYPE('prevValue.user',
'number', prevValue.user);
}
validateNumber(prevValue.user, 'prevValue.user');
throw new ERR_INVALID_ARG_VALUE.RangeError('prevValue.user',
prevValue.user);
}

if (!previousValueIsValid(prevValue.system)) {
if (typeof prevValue.system !== 'number') {
throw new ERR_INVALID_ARG_TYPE('prevValue.system',
'number', prevValue.system);
}
validateNumber(prevValue.system, 'prevValue.system');
throw new ERR_INVALID_ARG_VALUE.RangeError('prevValue.system',
prevValue.system);
}
Expand Down
8 changes: 3 additions & 5 deletions lib/internal/util.js
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ const {
uvErrmapGet,
overrideStackTrace,
} = require('internal/errors');
const { validateFunction } = require('internal/validators');
const { signals } = internalBinding('constants').os;
const {
getHiddenValue,
Expand Down Expand Up @@ -300,14 +301,11 @@ const kCustomPromisifiedSymbol = SymbolFor('nodejs.util.promisify.custom');
const kCustomPromisifyArgsSymbol = Symbol('customPromisifyArgs');

function promisify(original) {
if (typeof original !== 'function')
throw new ERR_INVALID_ARG_TYPE('original', 'Function', original);
validateFunction(original, 'original');

if (original[kCustomPromisifiedSymbol]) {
const fn = original[kCustomPromisifiedSymbol];
if (typeof fn !== 'function') {
throw new ERR_INVALID_ARG_TYPE('util.promisify.custom', 'Function', fn);
}
validateFunction(fn, 'util.promisify.custom');
return ObjectDefineProperty(fn, kCustomPromisifiedSymbol, {
value: fn, enumerable: false, writable: false, configurable: true
});
Expand Down
13 changes: 6 additions & 7 deletions lib/util.js
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,10 @@ const {
inspect
} = require('internal/util/inspect');
const { debuglog } = require('internal/util/debuglog');
const { validateNumber } = require('internal/validators');
const {
validateNumber,
validateFunction
} = require('internal/validators');
const { TextDecoder, TextEncoder } = require('internal/encoding');
const { isBuffer } = require('buffer').Buffer;
const types = require('internal/util/types');
Expand Down Expand Up @@ -285,18 +288,14 @@ const callbackifyOnRejected = hideStackFrames((reason, cb) => {
* }
*/
function callbackify(original) {
if (typeof original !== 'function') {
throw new ERR_INVALID_ARG_TYPE('original', 'Function', original);
}
validateFunction(original, 'original');

// We DO NOT return the promise as it gives the user a false sense that
// the promise is actually somehow related to the callback's execution
// and that the callback throwing will reject the promise.
function callbackified(...args) {
const maybeCb = ArrayPrototypePop(args);
if (typeof maybeCb !== 'function') {
throw new ERR_INVALID_ARG_TYPE('last argument', 'Function', maybeCb);
}
validateFunction(maybeCb, 'last argument');
const cb = FunctionPrototypeBind(maybeCb, this);
// In true node style we process the callback on `nextTick` with all the
// implications (stack, `uncaughtException`, `async_hooks`)
Expand Down
15 changes: 5 additions & 10 deletions lib/vm.js
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@ const {
validateBuffer,
validateObject,
validateOneOf,
validateFunction
} = require('internal/validators');
const {
kVmBreakFirstLineSymbol,
Expand Down Expand Up @@ -108,11 +109,8 @@ class Script extends ContextifyScript {
}

if (importModuleDynamically !== undefined) {
if (typeof importModuleDynamically !== 'function') {
throw new ERR_INVALID_ARG_TYPE('options.importModuleDynamically',
'function',
importModuleDynamically);
}
validateFunction(importModuleDynamically,
'options.importModuleDynamically');
const { importModuleDynamicallyWrap } =
require('internal/vm/module');
const { callbackMap } = internalBinding('module_wrap');
Expand Down Expand Up @@ -373,11 +371,8 @@ function compileFunction(code, params, options = {}) {
}

if (importModuleDynamically !== undefined) {
if (typeof importModuleDynamically !== 'function') {
throw new ERR_INVALID_ARG_TYPE('options.importModuleDynamically',
'function',
importModuleDynamically);
}
validateFunction(importModuleDynamically,
'options.importModuleDynamically');
const { importModuleDynamicallyWrap } =
require('internal/vm/module');
const { callbackMap } = internalBinding('module_wrap');
Expand Down
6 changes: 2 additions & 4 deletions lib/wasi.js
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ const {
validateBoolean,
validateInt32,
validateObject,
validateFunction
} = require('internal/validators');
const { WASI: _WASI } = internalBinding('wasi');
const kExitCode = Symbol('kExitCode');
Expand Down Expand Up @@ -118,10 +119,7 @@ class WASI {

const { _start, _initialize } = this[kInstance].exports;

if (typeof _start !== 'function') {
throw new ERR_INVALID_ARG_TYPE(
'instance.exports._start', 'function', _start);
}
validateFunction(_start, 'instance.exports._start');
if (_initialize !== undefined) {
throw new ERR_INVALID_ARG_TYPE(
'instance.exports._initialize', 'undefined', _initialize);
Expand Down
6 changes: 2 additions & 4 deletions lib/zlib.js
Original file line number Diff line number Diff line change
Expand Up @@ -71,6 +71,7 @@ const {
const { owner_symbol } = require('internal/async_hooks').symbols;
const {
validateFunction,
validateNumber
} = require('internal/validators');

const kFlushFlag = Symbol('kFlushFlag');
Expand Down Expand Up @@ -212,10 +213,7 @@ const checkFiniteNumber = hideStackFrames((number, name) => {
return false;
}

// Other non-numbers
if (typeof number !== 'number') {
throw new ERR_INVALID_ARG_TYPE(name, 'number', number);
}
validateNumber(number, name);

// Infinite numbers
throw new ERR_OUT_OF_RANGE(name, 'a finite number', number);
Expand Down

0 comments on commit f55af36

Please sign in to comment.