Added --limit-file-checks command line option.

Author: kirk-sayre-work

analyze.js

@@ -911,7 +911,7 @@ var wscript_proxy = new Proxy({
  quit: function() { 
  lib.logIOC("WScript", "Quit()", "The sample explicitly called WScript.Quit().");
  //console.trace()
- if (!argv["ignore-wscript-quit"]) {
+ if ((!argv["ignore-wscript-quit"]) || lib.doWscriptQuit()) {
  process.exit(0);
  }
  },

emulator/FileSystemObject.js

@@ -174,18 +174,38 @@ function FileSystemObject() {
 return true;
  };
  this.fileexists = (path) => {
-const value = !argv["no-file-exists"];
+var value = !argv["no-file-exists"];
 if (value) {
  lib.info(`Returning true for FileSystemObject.FileExists(${path}); use --no-file-exists if nothing happens`);
 }
+if (typeof(this._fileCheckCount) == "undefined") this._fileCheckCount = 0;
+this._fileCheckCount++;
+if (argv["limit-file-checks"] && (this._fileCheckCount > 10)) {
+ // Flip whether the file exists or not to see if that
+ // breaks a loop.
+ value = !value;
+ // Might break emulation based on WScript.quit(). Stop
+ // ignoring WScript.quit().
+ lib.doWscriptQuit(true);
+}
  lib.logIOC("FileExists", path, "The script checked to see if a file exists.");
 return value;
  };
  this.folderexists = (path) => {
-const value = !argv["no-folder-exists"];
+var value = !argv["no-folder-exists"];
 if (value) {
  lib.info(`Returning true for FileSystemObject.FolderExists(${path}); use --no-folder-exists if nothing happens`);
 }
+if (typeof(this._fileCheckCount) == "undefined") this._fileCheckCount = 0;
+this._fileCheckCount++;
+if (argv["limit-file-checks"] && (this._fileCheckCount > 500)) {
+ // Flip whether the file exists or not to see if that
+ // breaks a loop.
+ value = !value;
+ // Might break emulation based on WScript.quit(). Stop
+ // ignoring WScript.quit().
+ lib.doWscriptQuit(true);
+}
  lib.logIOC("FolderExists", path, "The script checked to see if a folder exists.");
 return value;
  };

flags.json

@@ -122,6 +122,11 @@
  "type": "Boolean",
  "description": "Return `false` for Scripting.FileSystemObject.FileExists(x)"
  },
+{
+ "name": "limit-file-checks",
+ "type": "Boolean",
+ "description": "Switch default value for folder/file exists checks if many checks are performed (try to break infinite file check loops)."
+ },
  {
  "name": "no-folder-exists",
  "type": "Boolean",

lib.js

@@ -184,13 +184,20 @@ function noCasePropObj(obj)
  return newObj; // object with upper cased keys
 };
 
+_doWscriptQuit = false;
+function doWscriptQuit(flag) {
+ if (typeof(flag) != "undefined") _doWscriptQuit = flag;
+ return _doWscriptQuit;
+}
+
 module.exports = {
  argv,
  kill,
  getUUID,
  throttleFileWrites,
  throttleCommands,
  noCasePropObj,
+ doWscriptQuit,
 
  debug: log.bind(null, "debug"),
  verbose: log.bind(null, "verb"),

patch.js

@@ -3,8 +3,10 @@ let __PATCH_CODE_ADDED__ = true;
 window = this;
 
 _globalTimeOffset = 0;
+_sleepCount = 0;
 WScript.sleep = function(delay) {
  _globalTimeOffset += delay;
+ _sleepCount++;
 }
 
 let fullYearGetter = Date.prototype.getFullYear;