summaryrefslogtreecommitdiffstats
path: root/js/src/devtools/rootAnalysis/analyze.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-28 14:29:10 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-28 14:29:10 +0000
commit2aa4a82499d4becd2284cdb482213d541b8804dd (patch)
treeb80bf8bf13c3766139fbacc530efd0dd9d54394c /js/src/devtools/rootAnalysis/analyze.py
parentInitial commit. (diff)
downloadfirefox-2aa4a82499d4becd2284cdb482213d541b8804dd.tar.xz
firefox-2aa4a82499d4becd2284cdb482213d541b8804dd.zip
Adding upstream version 86.0.1.upstream/86.0.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'js/src/devtools/rootAnalysis/analyze.py')
-rwxr-xr-xjs/src/devtools/rootAnalysis/analyze.py414
1 files changed, 414 insertions, 0 deletions
diff --git a/js/src/devtools/rootAnalysis/analyze.py b/js/src/devtools/rootAnalysis/analyze.py
new file mode 100755
index 0000000000..594e65a1c5
--- /dev/null
+++ b/js/src/devtools/rootAnalysis/analyze.py
@@ -0,0 +1,414 @@
+#!/usr/bin/python
+
+#
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+"""
+Runs the static rooting analysis
+"""
+
+from subprocess import Popen
+import argparse
+import os
+import subprocess
+import sys
+import re
+
+try:
+ from shlex import quote
+except ImportError:
+ from pipes import quote
+
+# Python 2/3 version independence polyfills
+
+anystring_t = str if sys.version_info[0] > 2 else basestring
+
+try:
+ execfile
+except Exception:
+
+ def execfile(thefile, globals):
+ exec(compile(open(thefile).read(), filename=thefile, mode="exec"), globals)
+
+
+def env(config):
+ e = dict(os.environ)
+ e["PATH"] = ":".join(
+ p for p in (config.get("gcc_bin"), config.get("sixgill_bin"), e["PATH"]) if p
+ )
+ e["XDB"] = "%(sixgill_bin)s/xdb.so" % config
+ e["SOURCE"] = config["source"]
+ e["ANALYZED_OBJDIR"] = config["objdir"]
+ return e
+
+
+def fill(command, config):
+ try:
+ return tuple(s % config for s in command)
+ except Exception:
+ print("Substitution failed:")
+ problems = []
+ for fragment in command:
+ try:
+ fragment % config
+ except Exception:
+ problems.append(fragment)
+ raise Exception(
+ "\n".join(["Substitution failed:"] + [" %s" % s for s in problems])
+ )
+
+
+def print_command(command, outfile=None, env=None):
+ output = " ".join(quote(s) for s in command)
+ if outfile:
+ output += " > " + outfile
+ if env:
+ changed = {}
+ e = os.environ
+ for key, value in env.items():
+ if (key not in e) or (e[key] != value):
+ changed[key] = value
+ if changed:
+ outputs = []
+ for key, value in changed.items():
+ if key in e and e[key] in value:
+ start = value.index(e[key])
+ end = start + len(e[key])
+ outputs.append(
+ '%s="%s${%s}%s"' % (key, value[:start], key, value[end:])
+ )
+ else:
+ outputs.append("%s='%s'" % (key, value))
+ output = " ".join(outputs) + " " + output
+
+ print(output)
+
+
+def generate_hazards(config, outfilename):
+ jobs = []
+ for i in range(int(config["jobs"])):
+ command = fill(
+ (
+ "%(js)s",
+ "%(analysis_scriptdir)s/analyzeRoots.js",
+ "%(gcFunctions_list)s",
+ "%(gcEdges)s",
+ "%(limitedFunctions_list)s",
+ "%(gcTypes)s",
+ "%(typeInfo)s",
+ str(i + 1),
+ "%(jobs)s",
+ "tmp.%s" % (i + 1,),
+ ),
+ config,
+ )
+ outfile = "rootingHazards.%s" % (i + 1,)
+ output = open(outfile, "w")
+ if config["verbose"]:
+ print_command(command, outfile=outfile, env=env(config))
+ jobs.append((command, Popen(command, stdout=output, env=env(config))))
+
+ final_status = 0
+ while jobs:
+ pid, status = os.wait()
+ jobs = [job for job in jobs if job[1].pid != pid]
+ final_status = final_status or status
+
+ if final_status:
+ raise subprocess.CalledProcessError(final_status, "analyzeRoots.js")
+
+ with open(outfilename, "w") as output:
+ command = ["cat"] + [
+ "rootingHazards.%s" % (i + 1,) for i in range(int(config["jobs"]))
+ ]
+ if config["verbose"]:
+ print_command(command, outfile=outfilename)
+ subprocess.call(command, stdout=output)
+
+
+JOBS = {
+ "dbs": (
+ (
+ "%(analysis_scriptdir)s/run_complete",
+ "--foreground",
+ "--no-logs",
+ "--build-root=%(objdir)s",
+ "--wrap-dir=%(sixgill)s/scripts/wrap_gcc",
+ "--work-dir=work",
+ "-b",
+ "%(sixgill_bin)s",
+ "--buildcommand=%(buildcommand)s",
+ ".",
+ ),
+ (),
+ ),
+ "list-dbs": (("ls", "-l"), ()),
+ "callgraph": (
+ (
+ "%(js)s",
+ "%(analysis_scriptdir)s/computeCallgraph.js",
+ "%(typeInfo)s",
+ "[callgraph]",
+ ),
+ ("callgraph.txt",),
+ ),
+ "gcFunctions": (
+ (
+ "%(js)s",
+ "%(analysis_scriptdir)s/computeGCFunctions.js",
+ "%(callgraph)s",
+ "[gcFunctions]",
+ "[gcFunctions_list]",
+ "[gcEdges]",
+ "[limitedFunctions_list]",
+ ),
+ ("gcFunctions.txt", "gcFunctions.lst", "gcEdges.txt", "limitedFunctions.lst"),
+ ),
+ "gcTypes": (
+ (
+ "%(js)s",
+ "%(analysis_scriptdir)s/computeGCTypes.js",
+ "[gcTypes]",
+ "[typeInfo]",
+ ),
+ ("gcTypes.txt", "typeInfo.txt"),
+ ),
+ "allFunctions": (
+ (
+ "%(sixgill_bin)s/xdbkeys",
+ "src_body.xdb",
+ ),
+ "allFunctions.txt",
+ ),
+ "hazards": (generate_hazards, "rootingHazards.txt"),
+ "explain": (
+ (
+ os.environ.get("PYTHON", "python2.7"),
+ "%(analysis_scriptdir)s/explain.py",
+ "%(hazards)s",
+ "%(gcFunctions)s",
+ "[explained_hazards]",
+ "[unnecessary]",
+ "[refs]",
+ ),
+ ("hazards.txt", "unnecessary.txt", "refs.txt"),
+ ),
+ "heapwrites": (
+ ("%(js)s", "%(analysis_scriptdir)s/analyzeHeapWrites.js"),
+ "heapWriteHazards.txt",
+ ),
+}
+
+
+def out_indexes(command):
+ for i in range(len(command)):
+ m = re.match(r"^\[(.*)\]$", command[i])
+ if m:
+ yield (i, m.group(1))
+
+
+def run_job(name, config):
+ cmdspec, outfiles = JOBS[name]
+ print("Running " + name + " to generate " + str(outfiles))
+ if hasattr(cmdspec, "__call__"):
+ cmdspec(config, outfiles)
+ else:
+ temp_map = {}
+ cmdspec = fill(cmdspec, config)
+ if isinstance(outfiles, anystring_t):
+ stdout_filename = "%s.tmp" % name
+ temp_map[stdout_filename] = outfiles
+ if config["verbose"]:
+ print_command(cmdspec, outfile=outfiles, env=env(config))
+ else:
+ stdout_filename = None
+ pc = list(cmdspec)
+ outfile = 0
+ for (i, name) in out_indexes(cmdspec):
+ pc[i] = outfiles[outfile]
+ outfile += 1
+ if config["verbose"]:
+ print_command(pc, env=env(config))
+
+ command = list(cmdspec)
+ outfile = 0
+ for (i, name) in out_indexes(cmdspec):
+ command[i] = "%s.tmp" % name
+ temp_map[command[i]] = outfiles[outfile]
+ outfile += 1
+
+ sys.stdout.flush()
+ if stdout_filename is None:
+ subprocess.check_call(command, env=env(config))
+ else:
+ with open(stdout_filename, "w") as output:
+ subprocess.check_call(command, stdout=output, env=env(config))
+ for (temp, final) in temp_map.items():
+ try:
+ os.rename(temp, final)
+ except OSError:
+ print("Error renaming %s -> %s" % (temp, final))
+ raise
+
+
+config = {"analysis_scriptdir": os.path.dirname(__file__)}
+
+defaults = [
+ "%s/defaults.py" % config["analysis_scriptdir"],
+ "%s/defaults.py" % os.getcwd(),
+]
+
+parser = argparse.ArgumentParser(
+ description="Statically analyze build tree for rooting hazards."
+)
+parser.add_argument(
+ "step", metavar="STEP", type=str, nargs="?", help="run starting from this step"
+)
+parser.add_argument(
+ "--source", metavar="SOURCE", type=str, nargs="?", help="source code to analyze"
+)
+parser.add_argument(
+ "--objdir",
+ metavar="DIR",
+ type=str,
+ nargs="?",
+ help="object directory of compiled files",
+)
+parser.add_argument(
+ "--js",
+ metavar="JSSHELL",
+ type=str,
+ nargs="?",
+ help="full path to ctypes-capable JS shell",
+)
+parser.add_argument(
+ "--upto", metavar="UPTO", type=str, nargs="?", help="last step to execute"
+)
+parser.add_argument(
+ "--jobs",
+ "-j",
+ default=None,
+ metavar="JOBS",
+ type=int,
+ help="number of simultaneous analyzeRoots.js jobs",
+)
+parser.add_argument(
+ "--list", const=True, nargs="?", type=bool, help="display available steps"
+)
+parser.add_argument(
+ "--buildcommand",
+ "--build",
+ "-b",
+ type=str,
+ nargs="?",
+ help="command to build the tree being analyzed",
+)
+parser.add_argument(
+ "--tag",
+ "-t",
+ type=str,
+ nargs="?",
+ help='name of job, also sets build command to "build.<tag>"',
+)
+parser.add_argument(
+ "--expect-file",
+ type=str,
+ nargs="?",
+ help="deprecated option, temporarily still present for backwards " "compatibility",
+)
+parser.add_argument(
+ "--verbose",
+ "-v",
+ action="count",
+ default=1,
+ help="Display cut & paste commands to run individual steps",
+)
+parser.add_argument("--quiet", "-q", action="count", default=0, help="Suppress output")
+
+args = parser.parse_args()
+args.verbose = max(0, args.verbose - args.quiet)
+
+for default in defaults:
+ try:
+ execfile(default, config)
+ if args.verbose:
+ print("Loaded %s" % default)
+ except Exception:
+ pass
+
+data = config.copy()
+
+for k, v in vars(args).items():
+ if v is not None:
+ data[k] = v
+
+if args.tag and not args.buildcommand:
+ args.buildcommand = "build.%s" % args.tag
+
+if args.jobs is not None:
+ data["jobs"] = args.jobs
+if not data.get("jobs"):
+ data["jobs"] = int(subprocess.check_output(["nproc", "--ignore=1"]).strip())
+
+if args.buildcommand:
+ data["buildcommand"] = args.buildcommand
+elif "BUILD" in os.environ:
+ data["buildcommand"] = os.environ["BUILD"]
+else:
+ data["buildcommand"] = "make -j4 -s"
+
+if "ANALYZED_OBJDIR" in os.environ:
+ data["objdir"] = os.environ["ANALYZED_OBJDIR"]
+
+if "GECKO_PATH" in os.environ:
+ data["source"] = os.environ["GECKO_PATH"]
+if "SOURCE" in os.environ:
+ data["source"] = os.environ["SOURCE"]
+
+steps = [
+ "dbs",
+ "gcTypes",
+ "callgraph",
+ "gcFunctions",
+ "allFunctions",
+ "hazards",
+ "explain",
+ "heapwrites",
+]
+
+if args.list:
+ for step in steps:
+ command, outfilename = JOBS[step]
+ if outfilename:
+ print("%s -> %s" % (step, outfilename))
+ else:
+ print(step)
+ sys.exit(0)
+
+for step in steps:
+ command, outfiles = JOBS[step]
+ if isinstance(outfiles, anystring_t):
+ data[step] = outfiles
+ else:
+ outfile = 0
+ for (i, name) in out_indexes(command):
+ data[name] = outfiles[outfile]
+ outfile += 1
+ assert (
+ len(outfiles) == outfile
+ ), "step '%s': mismatched number of output files (%d) and params (%d)" % (
+ step,
+ outfile,
+ len(outfiles),
+ ) # NOQA: E501
+
+if args.step:
+ steps = steps[steps.index(args.step) :]
+
+if args.upto:
+ steps = steps[: steps.index(args.upto) + 1]
+
+for step in steps:
+ run_job(step, data)