@contrast/assess 1.48.0 → 1.50.0

package/lib/dataflow/propagation/install/fastify-send.js

@@ -21,7 +21,6 @@ module.exports = function (core) {
   const {
     depHooks,
     patcher,
-    assess: { getPropagatorContext }
   } = core;
 
   return core.assess.dataflow.propagation.fastifySend = {
@@ -33,9 +32,10 @@ module.exports = function (core) {
           usePerf: 'sync',
           pre(data) {
             const { args } = data;
-
-            if (!getPropagatorContext()) return;
-
+            // (†) This is a minimal propagator that just untracks the argument. There are
+            // no propagation events generated and no expensive tag range computations. We
+            // don't get the propagator context before proceeding since it could lead to false
+            // positives e.g. if the number of propagation events exceeds the configured limit.
             const untrackedPath = StringPrototypeSlice.call(` ${args[0]}`, 1);
             args[0] = untrackedPath;
           },
@@ -49,9 +49,7 @@ module.exports = function (core) {
           usePerf: 'sync',
           pre(data) {
             const { args } = data;
-
-            if (!getPropagatorContext()) return;
-
+            // (†)
             const untrackedPath = StringPrototypeSlice.call(` ${args[1]}`, 1);
             args[1] = untrackedPath;
           },

package/lib/dataflow/propagation/install/send.js

@@ -21,7 +21,6 @@ module.exports = function (core) {
   const {
     depHooks,
     patcher,
-    assess: { getPropagatorContext }
   } = core;
 
   const send = {};
@@ -38,8 +37,10 @@ module.exports = function (core) {
           patchType,
           pre(data) {
             const { args } = data;
-            if (!getPropagatorContext()) return;
-
+            // This is a minimal propagator that just untracks the argument. There are
+            // no propagation events generated and no expensive tag range computations. We
+            // don't get the propagator context before proceeding since it could lead to false
+            // positives e.g. if the number of propagation events exceeds the configured limit.
             const untrackedPath = StringPrototypeSlice.call(` ${args[0]}`, 1);
             args[0] = untrackedPath;
           },

package/lib/dataflow/propagation/install/string/concat.js

@@ -32,30 +32,48 @@ module.exports = function(core) {
   return core.assess.dataflow.propagation.stringInstrumentation.concat = {
     install() {
       const name = 'String.prototype.concat';
+      const store = { name: `${name}:${patchType}`, lock: true };
 
       patcher.patch(String.prototype, 'concat', {
         name,
         patchType,
         usePerf: 'sync',
+        around(next, data) {
+          let hasArray = false;
+          for (let i = 0; i < data.args.length; i++) {
+            if (Array.isArray(data.args[i])) {
+              hasArray = true;
+              break;
+            }
+          }
+
+          // prevent propagation from happening in original concat method
+          // if arg is Array, since arg.join will be called on it.
+          if (hasArray && !core.scopes.instrumentation.isLocked())
+            return core.scopes.instrumentation.run(store, next);
+
+          return next();
+        },
         post(data) {
-          const { obj, result, hooked, orig } = data;
-          if (!result || !getPropagatorContext()) return;
+          if (!data.result || !getPropagatorContext()) return;
 
-          const rInfo = tracker.getData(result);
+          const rInfo = tracker.getData(data.result);
           if (rInfo) {
             // this may happen w/ trackedStr.concat('') => trackedStr
             return;
           }
 
-          const objInfo = tracker.getData(obj);
+          const objInfo = tracker.getData(data.obj);
           const history = objInfo ? new Set([objInfo]) : new Set();
-          let globalOffset = typeof obj !== 'function' ? obj.length : 0;
+
+          let globalOffset = typeof data.obj !== 'function' ? data.obj.length : 0;
           const args = [];
           let tags = objInfo?.tags;
 
           for (const arg of data.args) {
+            const isArray = Array.isArray(arg);
+            // TODO NODE-3748: handle tag ranges when arg is an Array
             const strInfo = tracker.getData(arg);
-
             if (strInfo) {
               args.push({ tracked: true, value: arg });
               history.add(strInfo);
@@ -64,12 +82,12 @@ module.exports = function(core) {
               args.push({ tracked: false, value: getAdjustedUntrackedValue(arg) });
             }
 
-            globalOffset += `${arg}`.length;
+            // if arg is an Array, then `${arg}` causes arg.join to get called and cause unwanted propagation
+            globalOffset += isArray ? ArrayPrototypeJoin.call(arg).length : `${arg}`.length;
           }
 
-
           if (history.size) {
-            const objVal = objInfo ? `'${objInfo.value}'` : getAdjustedUntrackedValue(obj);
+            const objVal = objInfo ? `'${objInfo.value}'` : getAdjustedUntrackedValue(data.obj);
             const context = `${objVal}.concat(${ArrayPrototypeJoin.call(args.map((a) => a.value))})`;
             const event = createPropagationEvent({
               name,
@@ -77,11 +95,11 @@ module.exports = function(core) {
               methodName: 'prototype.concat',
               context,
               object: {
-                value: objInfo?.value ?? getAdjustedUntrackedValue(obj),
+                value: objInfo?.value ?? getAdjustedUntrackedValue(data.obj),
                 tracked: !!objInfo
               },
               result: {
-                value: result,
+                value: data.result,
                 tracked: true
               },
               args,
@@ -90,19 +108,19 @@ module.exports = function(core) {
               source: objInfo ? (history.size > 1 ? 'A' : 'O') : 'P',
               target: 'R',
               stacktraceOpts: {
-                constructorOpt: hooked,
-                prependFrames: [orig]
+                constructorOpt: data.hooked,
+                prependFrames: [data.orig]
               },
             });
 
             if (!event) return;
-            const { extern } = tracker.track(result, event);
+            const { extern } = tracker.track(data.result, event);
 
             if (extern) {
               data.result = extern;
             }
           }
-        },
+        }
       });
     },
     uninstall() {

package/lib/dataflow/propagation/install/util-format.js

@@ -15,7 +15,7 @@
 
 'use strict';
 
-const { isString, primordials: { StringPrototypeMatch, StringPrototypeToLowerCase } } = require('@contrast/common');
+const { isString, primordials: { StringPrototypeMatch } } = require('@contrast/common');
 const { createAppendTags } = require('../../tag-utils');
 const { patchType } = require('../common');
 
@@ -47,7 +47,7 @@ module.exports = function(core) {
             let newTags = {};
             const history = [];
             const eventArgs = [];
-            const formatChars = args[0].includes('%') ? StringPrototypeMatch.call(args[0], /[^%]+/g).map((x) => x[0]) : [];
+            const formatChars = args[0].includes('%') ? StringPrototypeMatch.call(args[0], /%[sdifjoOc]/g).map((x) => x[1]) : [];
             let i = 0;
 
             if (formatChars.length > 0) {
@@ -58,14 +58,44 @@ module.exports = function(core) {
             for (i; i < args.length; i++) {
               let arg = args[i];
               const formatChar = formatChars[i - 1];
-              if (formatChar && typeof arg === 'object') {
-
-                if (formatChar === 'j') {
-                  arg = JSON.stringify(arg);
-                } else if (StringPrototypeToLowerCase.call(formatChar) === 'o') {
-                  // TO-DO: NODE-3235
+              if (formatChar) {
+                switch (formatChar) {
+                  case 's':
+                    if (typeof arg === 'object') {
+                      // util.inspect instrumentation NYI
+                      arg = arg?.toString ? arg.toString() : util.inspect(arg, { depth: 0, colors: false, compact: 3 });
+                    } else {
+                      arg = String(arg);
+                    }
+                    break;
+                  case 'd':
+                  case 'i':
+                  case 'f':
+                    // won't be tracked
+                    break;
+                  case 'j':
+                    arg = JSON.stringify(arg) ?? 'undefined';
+                    break;
+                  case 'o':
+                    // util.inspect instrumentation NYI
+                    arg = util.inspect(arg, { showHidden: true, showProxy: true });
+                    break;
+                  case 'O':
+                    // util.inspect instrumentation NYI
+                    arg = util.inspect(arg);
+                    break;
+                  case 'c':
+                    // c is ignored and skipped
+                    arg = '';
+                    break;
                 }
+              } else if (typeof arg !== 'string') {
+                arg = util.inspect(arg);
               }
+
+              const argInfo = tracker.getData(arg);
+              if (!argInfo) continue;
+
               const currIdx = result.indexOf(arg, idx);
               if (currIdx > -1) {
                 idx = currIdx + arg.length;
@@ -73,9 +103,6 @@ module.exports = function(core) {
                 continue;
               }
 
-              const argInfo = tracker.getData(arg);
-              if (!argInfo) continue;
-
               newTags = createAppendTags(newTags, argInfo.tags, currIdx);
 
               history.push({ ...argInfo });
@@ -120,8 +147,7 @@ module.exports = function(core) {
             }
           }
         });
-      }
-      );
+      });
     },
   };
 };

package/lib/dataflow/sources/handler.js

@@ -24,178 +24,182 @@ const {
     StringPrototypeToLowerCase
   }
 } = require('@contrast/common');
+const { Core } = require('@contrast/core/lib/ioc/core');
+
+module.exports = Core.makeComponent({
+  name: 'assess.dataflow.sources.handle',
+  factory(core) {
+    const {
+      assess: {
+        eventFactory,
+        dataflow: {
+          sources,
+          tracker
+        }
+      },
+      config,
+      createSnapshot,
+    } = core;
 
-module.exports = function (core) {
-  const {
-    assess: {
-      eventFactory,
-      dataflow: {
-        sources,
-        tracker
-      }
-    },
-    config,
-    createSnapshot,
-  } = core;
-
-  const logger = core.logger.child({ name: 'contrast:sources' });
+    const logger = core.logger.child({ name: 'contrast:sources' });
 
-  const emptyStack = Object.freeze([]);
+    const emptyStack = Object.freeze([]);
 
-  sources.createTags = function createTags({ inputType, fieldName = '', value, tagNames }) {
-    if (!value?.length) {
-      return null;
-    }
+    sources.createTags = function createTags({ inputType, fieldName = '', value, tagNames }) {
+      if (!value?.length) {
+        return null;
+      }
 
-    const stop = value.length - 1;
-    const tags = {
-      [DataflowTag.UNTRUSTED]: [0, stop]
-    };
+      const stop = value.length - 1;
+      const tags = {
+        [DataflowTag.UNTRUSTED]: [0, stop]
+      };
 
-    if (tagNames) {
-      for (const tag of tagNames) {
-        tags[tag] = [0, stop];
+      if (tagNames) {
+        for (const tag of tagNames) {
+          tags[tag] = [0, stop];
+        }
       }
-    }
 
-    if (inputType === InputType.HEADER && StringPrototypeToLowerCase.call(fieldName) === 'referer') {
-      tags[DataflowTag.HEADER] = [0, stop];
-    }
+      if (inputType === InputType.HEADER && StringPrototypeToLowerCase.call(fieldName) === 'referer') {
+        tags[DataflowTag.HEADER] = [0, stop];
+      }
 
-    return tags;
-  };
-
-  sources.createStacktrace = function (stacktraceOpts) {
-    return config.assess.stacktraces === 'NONE' || config.assess.stacktraces === 'SINK'
-      ? emptyStack
-      : createSnapshot(stacktraceOpts)();
-  };
-
-  sources.handle = function ({
-    context,
-    keys,
-    name,
-    inputType = InputType.UNKNOWN,
-    stacktraceOpts,
-    data,
-    sourceContext,
-  }) {
-    if (!data) return;
-
-    if (!sourceContext) {
-      logger.trace({ inputType, sourceName: name }, 'skipping assess source handling - no request context');
-      return null;
-    }
+      return tags;
+    };
 
-    // url exclusion
-    if (!sourceContext.policy) {
-      return null;
-    }
+    sources.createStacktrace = function(stacktraceOpts) {
+      return config.assess.stacktraces === 'NONE' || config.assess.stacktraces === 'SINK'
+        ? emptyStack
+        : createSnapshot(stacktraceOpts)();
+    };
 
-    if (!context) {
-      context = inputType;
-    }
+    sources.handle = function({
+      context,
+      keys,
+      name,
+      inputType = InputType.UNKNOWN,
+      stacktraceOpts,
+      data,
+      sourceContext,
+    }) {
+      if (!data) return;
+
+      if (!sourceContext) {
+        logger.trace({ inputType, sourceName: name }, 'skipping assess source handling - no request context');
+        return null;
+      }
 
-    const { policy: requestPolicy } = sourceContext;
-    const max = config.assess.max_context_source_events;
-    let _data = data;
-    let stack;
+      // url exclusion
+      if (!sourceContext.policy) {
+        return null;
+      }
 
-    if (keys) {
-      _data = {};
-      for (const key of keys) {
-        _data[key] = data[key];
+      if (!context) {
+        context = inputType;
       }
-    }
 
-    function createEvent({ fieldName, pathName, value, excludedRules }) {
-      const tagNames = Array.from(excludedRules).map((ruleId) => `excluded:${ruleId}`);
-      // create the stacktrace once per call to .handle()
-      stack || (stack = sources.createStacktrace(stacktraceOpts));
-      return eventFactory.createSourceEvent({
-        context: `${context}.${pathName}`,
-        name,
-        fieldName,
-        pathName,
-        stack,
-        inputType,
-        tags: sources.createTags({ inputType, fieldName, value, tagNames }),
-        result: { tracked: true, value },
-      });
-    }
+      const { policy: requestPolicy } = sourceContext;
+      const max = config.assess.max_context_source_events;
+      let _data = data;
+      let stack;
 
-    if (Buffer.isBuffer(data) && !tracker.getData(data)) {
-      const { track, excludedRules } = requestPolicy.getInputPolicy(InputType.BODY);
-      if (!track) {
-        core.logger.debug({ inputType }, 'assess input exclusion disabled tracking');
-        return;
+      if (keys) {
+        _data = {};
+        for (const key of keys) {
+          _data[key] = data[key];
+        }
       }
 
-      const event = createEvent({ pathName: 'body', value: data, fieldName: '', excludedRules });
-      if (event) {
-        tracker.track(data, event);
+      function createEvent({ fieldName, pathName, value, excludedRules }) {
+        const tagNames = Array.from(excludedRules).map((ruleId) => `excluded:${ruleId}`);
+        // create the stacktrace once per call to .handle()
+        stack || (stack = sources.createStacktrace(stacktraceOpts));
+        return eventFactory.createSourceEvent({
+          context: `${context}.${pathName}`,
+          name,
+          fieldName,
+          pathName,
+          stack,
+          inputType,
+          tags: sources.createTags({ inputType, fieldName, value, tagNames }),
+          result: { tracked: true, value },
+        });
       }
 
-      return;
-    }
-
-    traverse(_data, (path, fieldName, value, obj) => {
-      const pathName = ArrayPrototypeJoin.call(path, '.');
+      if (Buffer.isBuffer(data) && !tracker.getData(data)) {
+        const { track, excludedRules } = requestPolicy.getInputPolicy(InputType.BODY);
+        if (!track) {
+          core.logger.debug({ inputType }, 'assess input exclusion disabled tracking');
+          return;
+        }
 
-      if (sourceContext.sourceEventsCount >= max) {
-        logger.trace({ inputType, sourceName: name }, 'exiting assess source handling - %s max events exceeded', max);
-        return true;
-      }
+        const event = createEvent({ pathName: 'body', value: data, fieldName: '', excludedRules });
+        if (event) {
+          tracker.track(data, event);
+        }
 
-      const { track, excludedRules } = sourceContext.policy.getInputPolicy(inputType, fieldName);
-      if (!track) {
-        core.logger.debug({ fieldName, inputType }, 'assess input exclusion disabling tracking');
         return;
       }
 
-      if (isString(value) && value.length) {
-        const strInfo = tracker.getData(value);
-        if (strInfo) {
-          // TODO: confirm this "layering-on" approach is what we want
-          // when a value is already tracked, the handler will "re-track" the value with new source
-          // event metadata. without this step tracker would complain about value already being tracked.
-          // alternatively we could treat this more like a propagation event and update existing metadata.
-          value = strInfo.value;
+      traverse(_data, (path, fieldName, value, obj) => {
+        const pathName = ArrayPrototypeJoin.call(path, '.');
+
+        if (sourceContext.sourceEventsCount >= max) {
+          logger.trace({ inputType, sourceName: name }, 'exiting assess source handling - %s max events exceeded', max);
+          return true;
         }
-        const event = createEvent({ pathName, value, fieldName, excludedRules });
-        if (!event) {
-          logger.warn({ inputType, sourceName: name, pathName, value }, 'unable to create source event');
+
+        const { track, excludedRules } = sourceContext.policy.getInputPolicy(inputType, fieldName);
+        if (!track) {
+          core.logger.debug({ fieldName, inputType }, 'assess input exclusion disabling tracking');
           return;
         }
 
-        const { extern } = tracker.track(value, event);
-        if (extern) {
-          logger.trace({ extern, fieldName, sourceName: name, inputType }, 'tracked');
-          obj[fieldName] = extern;
-
-          sourceContext.sourceEventsCount++;
+        if (isString(value) && value.length) {
+          const strInfo = tracker.getData(value);
+          if (strInfo) {
+            // TODO: confirm this "layering-on" approach is what we want
+            // when a value is already tracked, the handler will "re-track" the value with new source
+            // event metadata. without this step tracker would complain about value already being tracked.
+            // alternatively we could treat this more like a propagation event and update existing metadata.
+            value = strInfo.value;
+          }
+          const event = createEvent({ pathName, value, fieldName, excludedRules });
+          if (!event) {
+            logger.warn({ inputType, sourceName: name, pathName, value }, 'unable to create source event');
+            return;
+          }
+
+          const { extern } = tracker.track(value, event);
+          if (extern) {
+            logger.trace({ extern, fieldName, sourceName: name, inputType }, 'tracked');
+            obj[fieldName] = extern;
+
+            sourceContext.sourceEventsCount++;
+          }
+        } else if (Buffer.isBuffer(value) && !tracker.getData(value)) {
+          const event = createEvent({ pathName, value, fieldName, excludedRules });
+          if (event) {
+            tracker.track(value, event);
+          } else {
+            logger.warn({ inputType, sourceName: name, pathName, value }, 'unable to create source event');
+          }
         }
-      } else if (Buffer.isBuffer(value) && !tracker.getData(value)) {
-        const event = createEvent({ pathName, value, fieldName, excludedRules });
-        if (event) {
-          tracker.track(value, event);
-        } else {
-          logger.warn({ inputType, sourceName: name, pathName, value }, 'unable to create source event');
-        }
-      }
-    });
+      });
 
-    if (keys) {
-      for (const key of keys) {
-        data[key] = _data[key];
+      if (keys) {
+        for (const key of keys) {
+          data[key] = _data[key];
+        }
       }
-    }
 
-    return data;
-  };
+      return data;
+    };
 
-  return sources;
-};
+    return sources;
+  }
+});
 
 /**
  * A custom traversal function for handling source value tracking efficiently.

package/lib/dataflow/sources/index.js

@@ -20,8 +20,7 @@ const { callChildComponentMethodsSync } = require('@contrast/common');
 module.exports = function (core) {
   const sources = core.assess.dataflow.sources = {};
 
-  require('./handler')(core);
-
+  core.initComponentSync(require('./handler'));
   require('./install/express')(core);
   require('./install/fastify')(core);
   require('./install/hapi')(core);

package/lib/dataflow/tag-utils.js

@@ -14,7 +14,7 @@
  */
 'use strict';
 
-const { primordials: { StringPrototypeSplit } } = require('@contrast/common');
+const { empties, primordials: { StringPrototypeSplit } } = require('@contrast/common');
 
 //
 // This module implements tag range manipulation functions. There are generally
@@ -295,7 +295,7 @@ function createMergedTags(firstTags, secondTags) {
 
 function mergeCore(firstTagRanges, secondTagRanges) {
   if (!firstTagRanges && !secondTagRanges) {
-    return [];
+    return empties.ARRAY;
   } else if (!firstTagRanges) {
     return [...secondTagRanges];
   } else if (!secondTagRanges) {
@@ -463,7 +463,7 @@ function createAdjustedQueryTags(path, tags, value, argString) {
       break;
     }
   }
-  return idx >= 0 ? createAppendTags([], tags, idx) : { ...tags };
+  return idx >= 0 ? createAppendTags(empties.OBJECT, tags, idx) : { ...tags };
 }
 
 /**

package/lib/event-factory.js

@@ -15,309 +15,205 @@
 
 'use strict';
 
-const { InputType, primordials: { StringPrototypeMatch } } = require('@contrast/common');
+const { InputType, empties, primordials: { StringPrototypeMatch } } = require('@contrast/common');
+const { Core } = require('@contrast/core/lib/ioc/core');
 const ANNOTATION_REGEX = /^(A|O|R|P|P\d+)$/;
 const SOURCE_EVENT_MSG = 'Source event not created: %s';
 const PROPAGATION_EVENT_MSG = 'Propagation event not created: %s';
 
-module.exports = function (core) {
-  const {
-    createSnapshot,
-    config,
-    logger,
-    scopes: { sources },
-  } = core;
-
-  const eventFactory = core.assess.eventFactory = {};
-
-  eventFactory.createdEvents = new WeakSet();
-
-  eventFactory.createSourceEvent = function (data = {}) {
+module.exports = Core.makeComponent({
+  name: 'assess.eventFactory',
+  factory(core) {
     const {
-      name,
-      result = { value: null, tracked: false },
-      tags,
-      inputType,
-      stack,
-    } = data;
-
-    if (!result.value) {
-      logger.debug({ name }, SOURCE_EVENT_MSG, 'invalid result');
-      return null;
-    }
-
-    if (!name) {
-      logger.debug({ name }, SOURCE_EVENT_MSG, 'invalid name');
-      return null;
-    }
-
-    if (!(inputType in InputType)) {
-      logger.debug({ name }, SOURCE_EVENT_MSG, 'invalid inputType');
-      return null;
-    }
-
-    if (!tags) {
-      logger.debug({ name }, SOURCE_EVENT_MSG, 'event has no tags');
-      return null;
-    }
-
-
-    if (!stack || !Array.isArray(stack)) {
-      logger.debug({ name }, SOURCE_EVENT_MSG, 'invalid stack');
-      return null;
-    }
-
-    data.time = Date.now();
-    eventFactory.createdEvents.add(data);
-
-    return data;
-  };
-
-  eventFactory.createPropagationEvent = function (data) {
-    const {
-      name = '',
-      moduleName,
-      methodName,
-      history = [],
-      object = { value: null, tracked: false },
-      args = [],
-      context,
-      result = { value: null, tracked: false },
-      tags = {},
-      addedTags = [],
-      removedTags = [],
-      source,
-      target,
-      stacktraceOpts
-    } = data;
-
-    const sourceContext = sources.getStore()?.assess;
-
-    if (!sourceContext) {
-      logger.debug({ name }, 'No sourceContext found during Propagation event creation');
-      return null;
-    }
-
-    if (sourceContext.propagationEventsCount >= config.assess.max_propagation_events) {
-      logger.debug({ name }, 'Maximum number of Propagation events reached. Event not created');
-      return null;
-    }
-
-    if (!name) {
-      logger.debug({ name }, PROPAGATION_EVENT_MSG, 'invalid name');
-      return null;
-    }
-
-    if (!history.length) {
-      logger.debug({ name }, PROPAGATION_EVENT_MSG, 'invalid history');
-      return null;
-    }
-
-    if (!source || !StringPrototypeMatch.call(source, ANNOTATION_REGEX)) {
-      logger.debug({ name }, PROPAGATION_EVENT_MSG, 'invalid source');
-      return null;
-    }
-
-    if (!target || !StringPrototypeMatch.call(target, ANNOTATION_REGEX)) {
-      logger.debug({ name }, PROPAGATION_EVENT_MSG, 'invalid target');
-      return null;
-    }
-
-    let stack;
-    if (config.assess.stacktraces === 'ALL') {
-      stack = createSnapshot(stacktraceOpts)();
-    } else {
-      stack = [];
-    }
-
-    const event = {
-      addedTags,
-      args,
-      context,
-      history,
-      name,
-      moduleName,
-      methodName,
-      object,
-      removedTags,
-      result,
-      source,
-      stack,
-      tags,
-      target,
-      time: Date.now(),
+      createSnapshot,
+      config,
+      logger,
+      scopes: { sources },
+    } = core;
+
+    const eventFactory = core.assess.eventFactory = {};
+
+    eventFactory.createdEvents = new WeakSet();
+
+    eventFactory.createSourceEvent = function(data = {}) {
+      if (!data.result?.value) {
+        logger.debug(SOURCE_EVENT_MSG, `invalid result: ${data.name}`);
+        return null;
+      }
+      if (!data.name) {
+        logger.debug(SOURCE_EVENT_MSG, `invalid name: ${data.name}`);
+        return null;
+      }
+      if (!(data.inputType in InputType)) {
+        logger.debug(SOURCE_EVENT_MSG, `invalid inputType: ${data.name}`);
+        return null;
+      }
+      if (!data.tags) {
+        logger.debug(SOURCE_EVENT_MSG, `event has no tags: ${data.name}`);
+        return null;
+      }
+      if (!data.stack || !Array.isArray(data.stack)) {
+        logger.debug(SOURCE_EVENT_MSG, `invalid stack: ${data.name}`);
+        return null;
+      }
+
+      data.time = Date.now();
+      eventFactory.createdEvents.add(data);
+
+      return data;
     };
 
-    eventFactory.createdEvents.add(event);
-    sourceContext.propagationEventsCount++;
-
-    return event;
-  };
-
-  eventFactory.createSinkEvent = function (data) {
-    const {
-      context,
-      name = '',
-      moduleName,
-      methodName,
-      history = [],
-      object = { value: null, tracked: false },
-      args = [],
-      result = { value: null, tracked: false },
-      tags = {},
-      source,
-      stacktraceOpts
-    } = data;
-
-    const sourceContext = sources.getStore()?.assess;
-    if (!sourceContext) {
-      logger.debug({ name }, 'no sourceContext found during sink event creation');
-      return null;
-    }
-    if (!name) {
-      logger.debug({ name }, 'no sink event name');
-      return null;
-    }
-    if (!history.length) {
-      logger.debug({ name }, 'empty history for sink event');
-      return null;
-    }
-    if (
-      (!source || !StringPrototypeMatch.call(source, ANNOTATION_REGEX))
-    ) {
-      logger.debug({ name }, 'malformed or missing sink event source field');
-      return null;
-    }
-
-    let stack;
-    if (config.assess.stacktraces !== 'NONE') {
-      stack = createSnapshot(stacktraceOpts)();
-    } else {
-      stack = [];
-    }
-
-    const event = {
-      args,
-      context,
-      history,
-      name,
-      moduleName,
-      methodName,
-      object,
-      result,
-      source,
-      stack,
-      tags,
-      time: Date.now(),
+    eventFactory.createPropagationEvent = function(data) {
+      const sourceContext = sources.getStore()?.assess;
+
+      if (!sourceContext) {
+        logger.debug('No sourceContext found during Propagation event creation: %s', data.name);
+        return null;
+      }
+      if (sourceContext.propagationEventsCount >= config.assess.max_propagation_events) {
+        logger.debug('Maximum number of Propagation events reached. Event not created: %s', data.name);
+        return null;
+      }
+      if (!data.name) {
+        logger.debug(PROPAGATION_EVENT_MSG, `invalid name (${data.name})`);
+        return null;
+      }
+      if (!data.history.length) {
+        logger.debug(PROPAGATION_EVENT_MSG, `invalid history (${data.name})`);
+        return null;
+      }
+      if (!data.source || !StringPrototypeMatch.call(data.source, ANNOTATION_REGEX)) {
+        logger.debug(PROPAGATION_EVENT_MSG, `invalid source (${data.name})`);
+        return null;
+      }
+      if (!data.target || !StringPrototypeMatch.call(data.target, ANNOTATION_REGEX)) {
+        logger.debug(PROPAGATION_EVENT_MSG, `invalid target (${data.name})`);
+        return null;
+      }
+
+      if (config.assess.stacktraces === 'ALL') {
+        data.stack = createSnapshot(data.stacktraceOpts)();
+      } else {
+        data.stack = empties.ARRAY;
+      }
+
+      data.args ??= empties.ARRAY;
+      data.addedTags ??= empties.ARRAY;
+      data.history ??= empties.ARRAY;
+      data.object ??= empties.UNTRACKED_VALUE_OBJ;
+      data.result ??= empties.UNTRACKED_VALUE_OBJ;
+      data.removedTags ??= empties.ARRAY;
+      data.time = Date.now();
+
+      eventFactory.createdEvents.add(data);
+      sourceContext.propagationEventsCount++;
+
+      return data;
     };
 
-    eventFactory.createdEvents.add(event);
-
-    return event;
-  };
-
-  eventFactory.createSessionEvent = function (data) {
-    const {
-      context,
-      name = '',
-      moduleName,
-      methodName,
-      object = { value: null, tracked: false },
-      args = [],
-      result = { value: null, tracked: false },
-      source,
-      stacktraceOpts,
-      framework,
-      options
-    } = data;
-
-    if (!name) {
-      logger.debug({ name }, 'no sink event name');
-      return null;
-    }
-
-    if (
-      (!source || !StringPrototypeMatch.call(source, ANNOTATION_REGEX))
-    ) {
-      logger.debug({ name }, 'malformed or missing sink event source field');
-      return null;
-    }
-
-    let stack;
-    if (config.assess.stacktraces !== 'NONE') {
-      stack = createSnapshot(stacktraceOpts)();
-    } else {
-      stack = [];
-    }
-
-    const event = {
-      args,
-      context,
-      history: [],
-      name,
-      moduleName,
-      methodName,
-      object,
-      result,
-      source,
-      stack,
-      tags: {},
-      time: Date.now(),
-      framework,
-      options,
+    eventFactory.createSinkEvent = function(data) {
+      const sourceContext = sources.getStore()?.assess;
+
+      if (!sourceContext) {
+        logger.debug('no sourceContext found during sink event creation: %s', data.name);
+        return null;
+      }
+      if (!data.name) {
+        logger.debug('no sink event name: %s', data.name);
+        return null;
+      }
+      if (!data.history.length) {
+        logger.debug('empty history for sink event: %s', data.name);
+        return null;
+      }
+      if (
+        (!data.source || !StringPrototypeMatch.call(data.source, ANNOTATION_REGEX))
+      ) {
+        logger.debug('malformed or missing sink event source field: %s', data.name);
+        return null;
+      }
+
+      if (config.assess.stacktraces !== 'NONE') {
+        data.stack = createSnapshot(data.stacktraceOpts)();
+      } else {
+        data.stack = empties.ARRAY;
+      }
+
+      data.args ??= empties.ARRAY;
+      data.history ??= empties.ARRAY;
+      data.object ??= empties.ARRAY;
+      data.result ??= empties.UNTRACKED_VALUE_OBJ;
+      data.time = Date.now();
+
+      eventFactory.createdEvents.add(data);
+
+      return data;
     };
 
-    eventFactory.createdEvents.add(event);
-
-    return event;
-  };
-
-
-  /**
-   * @param {{
-   *  context: string,
-   *  name: string,
-   *  moduleName: string,
-   *  methodName: string,
-   *  object: { value: any, tracked: boolean },
-   *  args: any[],
-   *  result: { value: vany, tracked: boolean },
-   *  source: string,
-   *  stacktraceOpts: { constructorOpt?: Function},
-   * }} data
-   * @returns {any}
-   */
-  eventFactory.createCryptoAnalysisEvent = function (data) {
-    const {
-      name = '',
-      source,
-      stacktraceOpts,
-    } = data;
-
-    if (!name) {
-      logger.debug({ name }, 'no sink event name');
-      return null;
-    }
-
-    if (!source || !StringPrototypeMatch.call(source, ANNOTATION_REGEX)) {
-      logger.debug({ name }, 'malformed or missing sink event source field');
-      return null;
-    }
-
-    let stack;
-    if (config.assess.stacktraces !== 'NONE') {
-      stack = createSnapshot(stacktraceOpts)();
-    } else {
-      stack = [];
-    }
-
-    data.stack = stack;
-    data.time = Date.now();
-
-    eventFactory.createdEvents.add(data);
+    eventFactory.createSessionEvent = function(data) {
+      if (!data.name) {
+        logger.debug('no sink event name: %s', data.name);
+        return null;
+      }
+      if (
+        (!data.source || !StringPrototypeMatch.call(data.source, ANNOTATION_REGEX))
+      ) {
+        logger.debug('malformed or missing sink event source field: %s', data.name);
+        return null;
+      }
+      if (config.assess.stacktraces !== 'NONE') {
+        data.stack = createSnapshot(data.stacktraceOpts)();
+      } else {
+        data.stack = empties.ARRAY;
+      }
+
+      data.args ??= empties.ARRAY;
+      data.history ??= empties.ARRAY;
+      data.obj ??= empties.UNTRACKED_VALUE_OBJ;
+      data.result ??= empties.UNTRACKED_VALUE_OBJ;
+      data.tags ??= empties.OBJECT;
+      data.time = Date.now();
+
+      eventFactory.createdEvents.add(data);
+
+      return data;
+    };
 
-    return data;
-  };
+    /**
+     * @param {{
+     *  context: string,
+     *  name: string,
+     *  moduleName: string,
+     *  methodName: string,
+     *  object: { value: any, tracked: boolean },
+     *  args: any[],
+     *  result: { value: vany, tracked: boolean },
+     *  source: string,
+     *  stacktraceOpts: { constructorOpt?: Function},
+     * }} data
+     * @returns {any}
+     */
+    eventFactory.createCryptoAnalysisEvent = function(data) {
+      if (!data.name) {
+        logger.debug('no sink event name: %s', data.name);
+        return null;
+      }
+      if (!data.source || !StringPrototypeMatch.call(data.source, ANNOTATION_REGEX)) {
+        logger.debug('malformed or missing sink event source field: %s', data.name);
+        return null;
+      }
+      if (config.assess.stacktraces !== 'NONE') {
+        data.stack = createSnapshot(data.stacktraceOpts)();
+      } else {
+        data.stack = empties.ARRAY;
+      }
+
+      data.time = Date.now();
+      eventFactory.createdEvents.add(data);
+
+      return data;
+    };
 
-  return eventFactory;
-};
+    return eventFactory;
+  }
+});

package/lib/index.js

@@ -52,10 +52,10 @@ module.exports = function assess(core) {
   // ancillary tools used by different features
   require('./sampler')(core);
   require('./get-policy')(core);
-  require('./make-source-context')(core);
+  core.initComponentSync(require('./make-source-context'));
   require('./rule-scopes')(core);
   require('./get-source-context')(core);
-  require('./event-factory')(core);
+  core.initComponentSync(require('./event-factory'));
 
   // various Assess features
   require('./dataflow')(core);

package/lib/make-source-context.js

@@ -16,75 +16,80 @@
 'use strict';
 
 const { primordials: { StringPrototypeToLowerCase, StringPrototypeSlice } } = require('@contrast/common');
+const { Core } = require('@contrast/core/lib/ioc/core');
+
 /**
  * @param {{
  *  assess: import('@contrast/assess').Assess,
  *  logger: import('@contrast/logger').Logger,
  * }} core
  */
-module.exports = function(core) {
-  const { assess, logger } = core;
+module.exports = Core.makeComponent({
+  name: 'assess.makeSourceContext',
+  factory(core) {
+    const { assess, logger } = core;
 
-  /**
-   * @returns {import('@contrast/assess').SourceContext}
-   */
-  return core.assess.makeSourceContext = function (sourceData) {
-    try {
-      // todo: how to handle non-HTTP sources
-      const { incomingMessage: req } = sourceData;
+    /**
+     * @returns {import('@contrast/assess').SourceContext}
+     */
+    return core.assess.makeSourceContext = function(sourceData) {
+      try {
+        // todo: how to handle non-HTTP sources
+        const { incomingMessage: req } = sourceData;
 
-      // minimally process the request data for sampling and exclusions.
-      // more request fields will be appended in final result below.
-      let uriPath;
-      let queries;
-      const idx = req.url.indexOf('?');
-      if (idx >= 0) {
-        uriPath = StringPrototypeSlice.call(req.url, 0, idx);
-        queries = StringPrototypeSlice.call(req.url, idx + 1);
-      } else {
-        uriPath = req.url;
-        queries = '';
-      }
+        // minimally process the request data for sampling and exclusions.
+        // more request fields will be appended in final result below.
+        let uriPath;
+        let queries;
+        const idx = req.url.indexOf('?');
+        if (idx >= 0) {
+          uriPath = StringPrototypeSlice.call(req.url, 0, idx);
+          queries = StringPrototypeSlice.call(req.url, idx + 1);
+        } else {
+          uriPath = req.url;
+          queries = '';
+        }
 
-      const ctx = sourceData.store.assess = {
-        // default policy to `null` until it is set later below. this will cause
-        // all instrumentation to short-circuit, see `./get-source-context.js`.
-        policy: null,
-        reqData: {
-          method: req.method,
-          uriPath,
-          queries,
-        },
-      };
+        const ctx = sourceData.store.assess = {
+          // default policy to `null` until it is set later below. this will cause
+          // all instrumentation to short-circuit, see `./get-source-context.js`.
+          policy: null,
+          reqData: {
+            method: req.method,
+            uriPath,
+            queries,
+          },
+        };
 
-      // check whether sampling allows processing
-      ctx.sampleInfo = assess.sampler?.getSampleInfo(sourceData) ?? null;
-      if (ctx.sampleInfo?.canSample === false) return ctx;
+        // check whether sampling allows processing
+        ctx.sampleInfo = assess.sampler?.getSampleInfo(sourceData) ?? null;
+        if (ctx.sampleInfo?.canSample === false) return ctx;
 
-      // set policy - can be returned as `null` if request is url-excluded.
-      ctx.policy = assess.getPolicy(ctx.reqData);
-      if (!ctx.policy) return ctx;
+        // set policy - can be returned as `null` if request is url-excluded.
+        ctx.policy = assess.getPolicy(ctx.reqData);
+        if (!ctx.policy) return ctx;
 
-      // build remaining reqData
-      ctx.reqData.headers = { ...req.headers }; // copy to avoid storing tracked values
-      ctx.reqData.ip = req.socket.remoteAddress;
-      ctx.reqData.httpVersion = req.httpVersion;
-      if (ctx.reqData.headers['content-type'])
-        ctx.reqData.contentType = StringPrototypeToLowerCase.call(ctx.reqData.headers['content-type']);
+        // build remaining reqData
+        ctx.reqData.headers = { ...req.headers }; // copy to avoid storing tracked values
+        ctx.reqData.ip = req.socket.remoteAddress;
+        ctx.reqData.httpVersion = req.httpVersion;
+        if (ctx.reqData.headers['content-type'])
+          ctx.reqData.contentType = StringPrototypeToLowerCase.call(ctx.reqData.headers['content-type']);
 
-      return {
-        ...ctx,
-        propagationEventsCount: 0,
-        sourceEventsCount: 0,
-        responseData: {},
-        ruleState: {},
-      };
-    } catch (err) {
-      logger.error(
-        { err },
-        'unable to construct assess store. assess will be disabled for request.'
-      );
-      return null;
-    }
-  };
-};
+        return {
+          ...ctx,
+          propagationEventsCount: 0,
+          sourceEventsCount: 0,
+          responseData: {},
+          ruleState: {},
+        };
+      } catch (err) {
+        logger.error(
+          { err },
+          'unable to construct assess store. assess will be disabled for request.'
+        );
+        return null;
+      }
+    };
+  }
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@contrast/assess",
-  "version": "1.48.0",
+  "version": "1.50.0",
   "description": "Contrast service providing framework-agnostic Assess support",
   "license": "SEE LICENSE IN LICENSE",
   "author": "Contrast Security <nodejs@contrastsecurity.com> (https://www.contrastsecurity.com)",
@@ -20,17 +20,17 @@
     "test": "../scripts/test.sh"
   },
   "dependencies": {
-    "@contrast/common": "1.30.0",
-    "@contrast/config": "1.41.0",
-    "@contrast/core": "1.46.0",
-    "@contrast/dep-hooks": "1.15.0",
+    "@contrast/common": "1.32.0",
+    "@contrast/config": "1.43.0",
+    "@contrast/core": "1.48.0",
+    "@contrast/dep-hooks": "1.17.0",
     "@contrast/distringuish": "^5.1.0",
-    "@contrast/instrumentation": "1.25.0",
-    "@contrast/logger": "1.19.0",
-    "@contrast/patcher": "1.18.0",
-    "@contrast/rewriter": "1.22.0",
-    "@contrast/route-coverage": "1.36.0",
-    "@contrast/scopes": "1.16.0",
+    "@contrast/instrumentation": "1.27.0",
+    "@contrast/logger": "1.21.0",
+    "@contrast/patcher": "1.20.0",
+    "@contrast/rewriter": "1.24.0",
+    "@contrast/route-coverage": "1.38.0",
+    "@contrast/scopes": "1.18.0",
     "semver": "^7.6.0"
   }
 }