@contrast/agent 4.10.0 → 4.10.1

package/bin/VERSION

@@ -1 +1 @@
-2.28.5
+2.28.9

package/lib/assess/propagators/mongoose/helpers.js

@@ -12,9 +12,44 @@ Copyright: 2022 Contrast Security, Inc
   engineered, modified, repackaged, sold, redistributed or otherwise used in a
   way not consistent with the End User License Agreement.
 */
+const TagRange = require('../../models/tag-range');
+const { CallContext, PropagationEvent, Signature } = require('../../models');
+
 const hasUserDefinedValidator = (data) =>
   data.obj.validators.some((validator) => validator.type === 'user defined');
 
+const tagCustomValidatedValues = (values, data, tracker, tagRangeUtil) => {
+  for (const value of values) {
+    if (typeof value !== 'string') continue;
+
+    const trackingData = tracker.track(value);
+
+    if (!trackingData) return;
+
+    const { props } = trackingData;
+    const stringLength = value.length - 1;
+
+    props.tagRanges = tagRangeUtil.add(
+      props.tagRanges,
+      new TagRange(0, stringLength, 'custom-validated-nosql-injection')
+    );
+
+    props.tagRanges = tagRangeUtil.add(
+      props.tagRanges,
+      new TagRange(0, stringLength, 'string-type-checked')
+    );
+
+    props.event = new PropagationEvent({
+      context: new CallContext(data),
+      signature: new Signature('mongoose.mixed.doValidateSync'),
+      tagRanges: props.tagRanges,
+      source: 'P',
+      target: 'A',
+      parents: [props.event]
+    });
+  }
+};
 module.exports = {
-  hasUserDefinedValidator
+  hasUserDefinedValidator,
+  tagCustomValidatedValues
 };

package/lib/assess/propagators/mongoose/index.js

@@ -15,4 +15,5 @@ Copyright: 2022 Contrast Security, Inc
 module.exports.handle = () => {
   require('./map');
   require('./string');
+  require('./mixed');
 };

package/lib/assess/propagators/mongoose/map.js

@@ -21,9 +21,11 @@ const tagRangeUtil = require('../../models/tag-range/util');
 const {
   PATCH_TYPES: { ASSESS_PROPAGATOR }
 } = require('../../../constants');
-const TagRange = require('../../models/tag-range');
-const { CallContext, PropagationEvent, Signature } = require('../../models');
-const { hasUserDefinedValidator } = require('./helpers');
+const {
+  hasUserDefinedValidator,
+  tagCustomValidatedValues
+} = require('./helpers');
+const agent = require('../../../agent');
 
 const doValidateSyncPatcher = (SchemaMap) => {
   patcher.patch(SchemaMap.prototype, 'doValidateSync', {
@@ -31,37 +33,16 @@ const doValidateSyncPatcher = (SchemaMap) => {
     name: 'mongoose.map.doValidateSync',
     patchType: ASSESS_PROPAGATOR,
     post(data) {
-      if (data.result || data.obj.options.of.name !== 'String') return;
+      if (data.result) return;
 
       if (!hasUserDefinedValidator(data)) return;
 
-      for (const value of data.args[0].values()) {
-        const trackingData = tracker.track(value);
-
-        if (!trackingData) return;
-
-        const { props } = trackingData;
-        const stringLength = value.length - 1;
-
-        props.tagRanges = tagRangeUtil.add(
-          props.tagRanges,
-          new TagRange(0, stringLength, 'custom-validated-nosql-injection')
-        );
-
-        props.tagRanges = tagRangeUtil.add(
-          props.tagRanges,
-          new TagRange(0, stringLength, 'string-type-checked')
-        );
-
-        props.event = new PropagationEvent({
-          context: new CallContext(data),
-          signature: new Signature('mongoose.map.doValidateSync'),
-          tagRanges: props.tagRanges,
-          source: 'P',
-          target: 'A',
-          parents: [props.event]
-        });
-      }
+      tagCustomValidatedValues(
+        data.args[0].values(),
+        data,
+        tracker,
+        tagRangeUtil
+      );
     }
   });
 };
@@ -69,6 +50,13 @@ const doValidateSyncPatcher = (SchemaMap) => {
 requireHook.resolve(
   { name: 'mongoose', file: 'lib/schema/map.js', version: '>=5.0.0' },
   (SchemaMap) => {
+    if (
+      !agent.config ||
+      (agent.config && !agent.config.agent.trust_custom_validators)
+    ) {
+      return;
+    }
+
     doValidateSyncPatcher(SchemaMap);
   }
 );

package/lib/assess/propagators/mongoose/mixed.js

@@ -0,0 +1,71 @@
+/**
+Copyright: 2022 Contrast Security, Inc
+  Contact: support@contrastsecurity.com
+  License: Commercial
+
+  NOTICE: This Software and the patented inventions embodied within may only be
+  used as part of Contrast Security’s commercial offerings. Even though it is
+  made available through public repositories, use of this Software is subject to
+  the applicable End User Licensing Agreement found at
+  https://www.contrastsecurity.com/enduser-terms-0317a or as otherwise agreed
+  between Contrast Security and the End User. The Software may not be reverse
+  engineered, modified, repackaged, sold, redistributed or otherwise used in a
+  way not consistent with the End User License Agreement.
+*/
+'use strict';
+
+const tracker = require('../../../tracker');
+const patcher = require('../../../hooks/patcher');
+const requireHook = require('../../../hooks/require');
+const tagRangeUtil = require('../../models/tag-range/util');
+const {
+  PATCH_TYPES: { ASSESS_PROPAGATOR }
+} = require('../../../constants');
+const {
+  hasUserDefinedValidator,
+  tagCustomValidatedValues
+} = require('./helpers');
+const agent = require('../../../agent');
+
+const doValidateSyncPatcher = (SchemaMap) => {
+  patcher.patch(SchemaMap.prototype, 'doValidateSync', {
+    alwaysRun: true,
+    name: 'mongoose.mixed.doValidateSync',
+    patchType: ASSESS_PROPAGATOR,
+    post(data) {
+      if (data.result || !hasUserDefinedValidator(data)) return;
+
+      const input = data.args[0];
+      const inputType = typeof input;
+
+      if (inputType !== 'string' && inputType !== 'object') return;
+
+      let values;
+      if (inputType === 'string') {
+        values = [input];
+      } else if (Array.isArray(input)) {
+        values = input;
+      } else if (input instanceof Map) {
+        values = input.values();
+      } else {
+        values = Object.values(input);
+      }
+
+      tagCustomValidatedValues(values, data, tracker, tagRangeUtil);
+    }
+  });
+};
+
+requireHook.resolve(
+  { name: 'mongoose', file: 'lib/schema/mixed.js', version: '>=5.0.0' },
+  (SchemaMap) => {
+    if (
+      !agent.config ||
+      (agent.config && !agent.config.agent.trust_custom_validators)
+    ) {
+      return;
+    }
+
+    doValidateSyncPatcher(SchemaMap);
+  }
+);

package/lib/core/arch-components/mysql.js

@@ -19,6 +19,7 @@ const ModuleHook = require('../../hooks/require');
 const agentEmitter = require('../../agent-emitter');
 const { PATCH_TYPES } = require('../../constants');
 const logger = require('../logger')('contrast:arch-component');
+const waitToConnect = require('./util');
 
 ModuleHook.resolve(
   { name: 'mysql', file: 'lib/Connection.js' },
@@ -28,22 +29,31 @@ ModuleHook.resolve(
       patchType: PATCH_TYPES.ARCH_COMPONENT,
       alwaysRun: true,
       post(wrapCtx) {
-        try {
-          let url = 'mysql://';
-          if (this.config.socketPath) {
-            url += this.config.socketPath;
-          } else {
-            url += `${this.config.host}`;
-          }
-          agentEmitter.emit('architectureComponent', {
-            vendor: 'MySQL',
-            url: new URL(url).toString(),
-            remoteHost: '',
-            remotePort: !this.config.socketPath ? this.config.port : -1
+        waitToConnect(wrapCtx)
+          .then(() => {
+            try {
+              let url = 'mysql://';
+              if (this.config.socketPath) {
+                url += this.config.socketPath;
+              } else {
+                url += `${this.config.host}`;
+              }
+              agentEmitter.emit('architectureComponent', {
+                vendor: 'MySQL',
+                url: new URL(url).toString(),
+                remoteHost: '',
+                remotePort: !this.config.socketPath ? this.config.port : -1
+              });
+            } catch (err) {
+              logger.warn(
+                'unable to report MySQL architecture component\n',
+                err
+              );
+            }
+          })
+          .catch((err) => {
+            logger.warn('unable to report MySQL architecture component\n', err);
           });
-        } catch (err) {
-          logger.warn('unable to report MySQL architecture component\n', err);
-        }
       }
     });
   }

package/lib/core/arch-components/postgres.js

@@ -18,6 +18,7 @@ const ModuleHook = require('../../hooks/require');
 const agentEmitter = require('../../agent-emitter');
 const { PATCH_TYPES } = require('../../constants');
 const logger = require('../logger')('contrast:arch-component');
+const waitToConnect = require('./util');
 
 ModuleHook.resolve({ name: 'pg', file: 'lib/client.js' }, (pgClient) =>
   patcher.patch(pgClient, {
@@ -25,37 +26,46 @@ ModuleHook.resolve({ name: 'pg', file: 'lib/client.js' }, (pgClient) =>
     patchType: PATCH_TYPES.ARCH_COMPONENT,
     alwaysRun: true,
     post(wrapCtx) {
-      try {
-        const {
-          host = process.env.PGHOST,
-          port = process.env.PGPORT
-        } = wrapCtx.result;
+      waitToConnect(wrapCtx)
+        .then(() => {
+          try {
+            const {
+              host = process.env.PGHOST,
+              port = process.env.PGPORT
+            } = wrapCtx.result;
 
-        if (!host) {
-          return;
-        }
+            if (!host) {
+              return;
+            }
 
-        let url = host;
+            let url = host;
 
-        // build protocol and port into url prior to parsing
-        if (url.indexOf('://') === -1) {
-          url = `postgresql://${url}`;
-        }
-        if (port !== undefined) {
-          url = `${url}:${port}`;
-        }
+            // build protocol and port into url prior to parsing
+            if (url.indexOf('://') === -1) {
+              url = `postgresql://${url}`;
+            }
+            if (port !== undefined) {
+              url = `${url}:${port}`;
+            }
 
-        agentEmitter.emit('architectureComponent', {
-          vendor: 'PostgreSQL',
-          remotePort: port || 0,
-          url: new URL(url).toString()
+            agentEmitter.emit('architectureComponent', {
+              vendor: 'PostgreSQL',
+              remotePort: port || 0,
+              url: new URL(url).toString()
+            });
+          } catch (err) {
+            logger.warn(
+              'unable to report PostgreSQL architecture component\n%o',
+              err
+            );
+          }
+        })
+        .catch((err) => {
+          logger.warn(
+            'unable to report PostgreSQL architecture component\n%o',
+            err
+          );
         });
-      } catch (err) {
-        logger.warn(
-          'unable to report PostgreSQL architecture component\n%o',
-          err
-        );
-      }
     }
   })
 );

package/lib/core/arch-components/util.js

@@ -0,0 +1,49 @@
+/**
+Copyright: 2022 Contrast Security, Inc
+  Contact: support@contrastsecurity.com
+  License: Commercial
+
+  NOTICE: This Software and the patented inventions embodied within may only be
+  used as part of Contrast Security’s commercial offerings. Even though it is
+  made available through public repositories, use of this Software is subject to
+  the applicable End User Licensing Agreement found at
+  https://www.contrastsecurity.com/enduser-terms-0317a or as otherwise agreed
+  between Contrast Security and the End User. The Software may not be reverse
+  engineered, modified, repackaged, sold, redistributed or otherwise used in a
+  way not consistent with the End User License Agreement.
+*/
+const MYSQL = 'mysql.connect.arch_component';
+const POSTGRES = 'pg.Client.arch_component';
+
+module.exports = function waitToConnect(ctx, count = 0) {
+  return new Promise((resolve, reject) => {
+    const maxAttempts = 10 * 60; // i.e., 1 min.
+    const checkConnection = setInterval(
+      function(ctx, resolve, reject) {
+        if (count >= maxAttempts) {
+          clearInterval(checkConnection);
+          reject();
+        }
+        switch (ctx.name) {
+          case MYSQL:
+            if (ctx.obj.state === 'authenticated') {
+              clearInterval(checkConnection);
+              resolve();
+            }
+            break;
+          case POSTGRES:
+            if (ctx.result._connected) {
+              clearInterval(checkConnection);
+              resolve();
+            }
+            break;
+        }
+        count++;
+      },
+      100,
+      ctx,
+      resolve,
+      reject
+    );
+  });
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@contrast/agent",
-  "version": "4.10.0",
+  "version": "4.10.1",
   "description": "Node.js security instrumentation by Contrast Security",
   "keywords": [
     "security",