@contrast/agentify 1.3.1 → 1.4.1

package/lib/contrast-methods.js

@@ -15,7 +15,7 @@
 
 'use strict';
 
-module.exports = function(core) {
+module.exports = function (core) {
   const { logger, patcher } = core;
 
   const getOrig = (v) => patcher.unwrap(v);
@@ -28,70 +28,123 @@ module.exports = function(core) {
    * 2) patch those functions on `contrastMethods.api` in order to add instrumentation
    */
   const contrastMethods = {
-    // TODO: Assess will require add'l methods
-    // TODO: ESM will require add'l methods
     api: {
       eval(v) {
         return v;
       },
-      plus(left, right) {
+
+      // Binary Operators
+      add(left, right) {
         const ret = getOrig(left) + getOrig(right);
         return ret;
       },
-      doubleEqual(left, right) {
+      eqEq(left, right) {
         left = getOrig(left);
         right = getOrig(right);
 
         return left == right;
       },
-      notTripleEqual(left, right) {
+      eqEqEq(left, right) {
         left = getOrig(left);
         right = getOrig(right);
 
-        return left !== right;
+        return left === right;
       },
-      tripleEqual(left, right) {
+      notEqEq(left, right) {
         left = getOrig(left);
         right = getOrig(right);
 
-        return left === right;
+        return left !== right;
       },
-      notDoubleEqual(left, right) {
+      notEq(left, right) {
         left = getOrig(left);
         right = getOrig(right);
 
         return left != right;
       },
+
+      // Computed Properties
+      // Used to force a copy of handle:
+      // https://stackoverflow.com/a/31733628
+      // This is a workaround for CONTRAST-30333, in which something we
+      // suspect is a v8 bug means that handle will otherwise be clobbered
+      // and if it was externalized, we will lose the data associated with it
       forceCopy(handle) {
-        // Used to force a copy of handle:
-        // https://stackoverflow.com/a/31733628
-        // This is a workaround for CONTRAST-30333, in which something we
-        // suspect is a v8 bug means that handle will otherwise be clobbered
-        // and if it was externalized, we will lose the data associated with it
         if (typeof handle === 'string') {
           handle = toUntrackedString(handle);
         }
         return handle;
       },
-      cast: function __cast(val) {
+
+      // Switch Case
+      cast(val) {
         return getOrig(val);
       },
-      tag: function __contrastTag(strings) {
-        // XXX implemented in lib/assess/propagators/templates.js
+
+      /**
+       * Tagged Templates
+       * The first argument received by the tag function is an array of strings.
+       * For any template literal, its length is equal to the number of
+       * substitutions (occurrences of ${...}) plus one, and is therefore always
+       * non-empty.
+       * see: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Template_literals#tagged_templates
+       * @param {string[]} strings length n+1
+       * @param {string[]} args length n
+       * @returns {string}
+       */
+      tag(strings, ...args) {
+        const [first, ...rest] = strings;
+
+        return rest.reduce(
+          (result, str, i) => result.concat(args[i], str),
+          first,
+        );
+      },
+
+      // Import Statements
+      importDefault(...args) {
+        // noop
       },
 
+      importStarAs(...args) {
+        // noop
+      },
+
+      importNamed(...args) {
+        // noop
+      },
+
+      // Injections
       Function: core.patcher.patch(global.Function, {
         name: 'global.Function',
-        patchType: 'rewrite-injection'
+        patchType: 'rewrite-injection',
+      }),
+      JSON: core.patcher.patch(global.JSON, {
+        name: 'global.JSON',
+        patchType: 'rewrite-injection',
+      }),
+      Number: core.patcher.patch(global.Number, {
+        name: 'global.Number',
+        patchType: 'rewrite-injection',
+      }),
+      Object: core.patcher.patch(global.Object, {
+        name: 'global.Object',
+        patchType: 'rewrite-injection',
+      }),
+      String: core.patcher.patch(global.String, {
+        name: 'global.String',
+        patchType: 'rewrite-injection',
       }),
     },
+
     installed: false,
+
     getGlobal() {
       return global;
-    }
+    },
   };
 
-  contrastMethods.install = function() {
+  contrastMethods.install = function () {
     if (contrastMethods.installed) {
       return;
     }
@@ -102,15 +155,15 @@ module.exports = function(core) {
       Object.defineProperty(global, 'ContrastMethods', {
         enumerable: true,
         configurable: false,
-        value: contrastMethods.api
+        value: contrastMethods.api,
       });
       contrastMethods.installed = true;
     } catch (err) {
       // We should never expect this since the installation process is well
       // controlled, but still we should have the defensive code.
-      logger.error({ err }, 'Unable to install global.ContrastMethods');
+      logger.error({ err }, 'Unable to inject global.ContrastMethods');
     }
   };
 
-  return core.contrastMethods = contrastMethods;
+  return (core.contrastMethods = contrastMethods);
 };

package/lib/function-hooks.js

@@ -81,7 +81,7 @@ module.exports = function (deps) {
       unwritten = rewriter.unwrite(unwritten);
       unwritten = unwritten.replace(METHOD_CONTEXT, '');
       unwritten = unwritten.replace(FUNCTION_CONTEXT, '');
-      unwritten = unwritten.replace(/;$/, ''); // removes trailing semicolon
+      unwritten = unwritten.replace(/;\s*$/, ''); // removes trailing semicolon/whitespace
       return unwritten;
     } catch (err) {
       logger.warn({ err, code }, 'Failed to unwrite function code');

package/lib/index.js

@@ -31,7 +31,8 @@ const defaultOpts = {
     'assess',
     'protect',
     'depHooks',
-    'enhancedLibraryUsage',
+    'routeCoverage',
+    'libraryAnalysis',
     'rewriteHooks',
     'functionHooks'
   ]
@@ -106,7 +107,7 @@ class Agent {
         await self.handleInstallFailure(err);
       }
 
-      self.logEffectiveConfig();
+      self.logDiagnosticFiles();
       return self.runMain.apply(this, args);
     };
   }
@@ -135,29 +136,39 @@ class Agent {
     }
   }
 
-  logEffectiveConfig() {
-    const { config, getEffectiveConfig } = this.core;
+  logDiagnosticFiles() {
+    const { agentName, agentVersion, config, getEffectiveConfig, getSystemInfo } = this.core;
 
-    if (config._flat['agent.diagnostics.enable'] !== false) {
-      const content = JSON.stringify(getEffectiveConfig(), null, 2).concat('\n\n');
+    if (config.agent.diagnostics.enable !== false) {
+      const effectiveConfig = JSON.stringify(getEffectiveConfig(), null, 2).concat('\n\n');
 
-      if (config._flat['agent.diagnostics.quiet'] !== true) {
-        fs.writeFileSync(1, content, 'utf8');
+      let systemInfo = getSystemInfo();
+      systemInfo.Contrast.Agent = {
+        Name: agentName,
+        Version: agentVersion,
+      };
+
+      systemInfo = JSON.stringify(systemInfo, null, 2).concat('\n\n');
+      if (!config.agent.diagnostics.quiet) {
+        fs.writeFileSync(1, effectiveConfig, 'utf8');
+        fs.writeFileSync(1, systemInfo, 'utf8');
       }
 
-      let outputDir = config._flat['agent.diagnostics.report_path'];
+      let outputDir = config.agent.diagnostics.report_path;
       if (!outputDir && config.agent.logger.path) {
-        outputDir = path.join(config.agent.logger.path, '../contrast_effective_config.json');
+        outputDir = path.join(config.agent.logger.path, '..');
       }
 
       try {
-        fs.writeFileSync(outputDir, content, 'utf-8');
+        fs.writeFileSync(path.join(outputDir, 'contrast_effective_config.json'), effectiveConfig, 'utf-8');
+        fs.writeFileSync(path.join(outputDir, 'contrast_system_info.json'), systemInfo, 'utf-8');
       } catch (err) {
-        outputDir = path.join(process.cwd(), 'contrast_effective_config.json');
+        outputDir = path.join(process.cwd());
         try {
-          fs.writeFileSync(outputDir, content, 'utf-8');
+          fs.writeFileSync(path.join(outputDir, 'contrast_effective_config.json'), effectiveConfig, 'utf-8');
+          fs.writeFileSync(path.join(outputDir, 'contrast_system_info.json'), systemInfo, 'utf-8');
         } catch (err) {
-          fs.writeFileSync(1, `Couldn't create effective config file: ${err}`, 'utf-8');
+          fs.writeFileSync(1, `Couldn't create the diagnostic files: ${err}`, 'utf-8');
         }
       }
     }

package/lib/rewrite-hooks.js

@@ -27,7 +27,12 @@ module.exports = function(deps) {
 
       Module.prototype._compile = function(content, filename) {
         let compiled;
-        const { code } = deps.rewriter.rewrite(content, { filename, inject: true, wrap: true });
+        const { code } = deps.rewriter.rewrite(content, {
+          filename,
+          isModule: false,
+          inject: true,
+          wrap: true,
+        });
 
         try {
           compiled = origCompile.call(this, code, filename);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@contrast/agentify",
-  "version": "1.3.1",
+  "version": "1.4.1",
   "description": "Configures Contrast agent services and instrumentation within an application",
   "license": "SEE LICENSE IN LICENSE",
   "author": "Contrast Security <nodejs@contrastsecurity.com> (https://www.contrastsecurity.com)",
@@ -16,4 +16,4 @@
   "scripts": {
     "test": "../scripts/test.sh"
   }
-}
+}