summaryrefslogtreecommitdiffstats
path: root/testing/mochitest/leaks.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 /testing/mochitest/leaks.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 'testing/mochitest/leaks.py')
-rw-r--r--testing/mochitest/leaks.py437
1 files changed, 437 insertions, 0 deletions
diff --git a/testing/mochitest/leaks.py b/testing/mochitest/leaks.py
new file mode 100644
index 0000000000..d94478dd52
--- /dev/null
+++ b/testing/mochitest/leaks.py
@@ -0,0 +1,437 @@
+# 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/
+
+# The content of this file comes orginally from automationutils.py
+# and *should* be revised.
+
+from __future__ import absolute_import, division
+
+import re
+from operator import itemgetter
+
+RE_DOCSHELL = re.compile("I\/DocShellAndDOMWindowLeak ([+\-]{2})DOCSHELL")
+RE_DOMWINDOW = re.compile("I\/DocShellAndDOMWindowLeak ([+\-]{2})DOMWINDOW")
+
+
+class ShutdownLeaks(object):
+
+ """
+ Parses the mochitest run log when running a debug build, assigns all leaked
+ DOM windows (that are still around after test suite shutdown, despite running
+ the GC) to the tests that created them and prints leak statistics.
+ """
+
+ def __init__(self, logger):
+ self.logger = logger
+ self.tests = []
+ self.leakedWindows = {}
+ self.hiddenWindowsCount = 0
+ self.leakedDocShells = set()
+ self.hiddenDocShellsCount = 0
+ self.numDocShellCreatedLogsSeen = 0
+ self.numDocShellDestroyedLogsSeen = 0
+ self.numDomWindowCreatedLogsSeen = 0
+ self.numDomWindowDestroyedLogsSeen = 0
+ self.currentTest = None
+ self.seenShutdown = set()
+
+ def log(self, message):
+ action = message["action"]
+
+ # Remove 'log' when clipboard is gone and/or structured.
+ if action in ("log", "process_output"):
+ line = message["message"] if action == "log" else message["data"]
+
+ m = RE_DOMWINDOW.search(line)
+ if m:
+ self._logWindow(line, m.group(1) == "++")
+ return
+
+ m = RE_DOCSHELL.search(line)
+ if m:
+ self._logDocShell(line, m.group(1) == "++")
+ return
+
+ if line.startswith("Completed ShutdownLeaks collections in process"):
+ pid = int(line.split()[-1])
+ self.seenShutdown.add(pid)
+ elif action == "test_start":
+ fileName = message["test"].replace(
+ "chrome://mochitests/content/browser/", ""
+ )
+ self.currentTest = {
+ "fileName": fileName,
+ "windows": set(),
+ "docShells": set(),
+ }
+ elif action == "test_end":
+ # don't track a test if no windows or docShells leaked
+ if self.currentTest and (
+ self.currentTest["windows"] or self.currentTest["docShells"]
+ ):
+ self.tests.append(self.currentTest)
+ self.currentTest = None
+
+ def process(self):
+ failures = 0
+
+ if not self.seenShutdown:
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | ShutdownLeaks | process() called before end of test suite"
+ )
+ failures += 1
+
+ if (
+ self.numDocShellCreatedLogsSeen == 0
+ or self.numDocShellDestroyedLogsSeen == 0
+ ):
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | did not see DOCSHELL log strings."
+ " this occurs if the DOCSHELL logging gets disabled by"
+ " something. %d created seen %d destroyed seen"
+ % (self.numDocShellCreatedLogsSeen, self.numDocShellDestroyedLogsSeen)
+ )
+ failures += 1
+ else:
+ self.logger.info(
+ "TEST-INFO | Confirming we saw %d DOCSHELL created and %d destroyed log"
+ " strings."
+ % (self.numDocShellCreatedLogsSeen, self.numDocShellDestroyedLogsSeen)
+ )
+
+ if (
+ self.numDomWindowCreatedLogsSeen == 0
+ or self.numDomWindowDestroyedLogsSeen == 0
+ ):
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | did not see DOMWINDOW log strings."
+ " this occurs if the DOMWINDOW logging gets disabled by"
+ " something%d created seen %d destroyed seen"
+ % (self.numDomWindowCreatedLogsSeen, self.numDomWindowDestroyedLogsSeen)
+ )
+ failures += 1
+ else:
+ self.logger.info(
+ "TEST-INFO | Confirming we saw %d DOMWINDOW created and %d destroyed log"
+ " strings."
+ % (self.numDomWindowCreatedLogsSeen, self.numDomWindowDestroyedLogsSeen)
+ )
+
+ for test in self._parseLeakingTests():
+ for url, count in self._zipLeakedWindows(test["leakedWindows"]):
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | %s | leaked %d window(s) until shutdown "
+ "[url = %s]" % (test["fileName"], count, url)
+ )
+ failures += 1
+
+ if test["leakedWindowsString"]:
+ self.logger.info(
+ "TEST-INFO | %s | windows(s) leaked: %s"
+ % (test["fileName"], test["leakedWindowsString"])
+ )
+
+ if test["leakedDocShells"]:
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | %s | leaked %d docShell(s) until "
+ "shutdown" % (test["fileName"], len(test["leakedDocShells"]))
+ )
+ failures += 1
+ self.logger.info(
+ "TEST-INFO | %s | docShell(s) leaked: %s"
+ % (
+ test["fileName"],
+ ", ".join(
+ [
+ "[pid = %s] [id = %s]" % x
+ for x in test["leakedDocShells"]
+ ]
+ ),
+ )
+ )
+
+ if test["hiddenWindowsCount"] > 0:
+ # Note: to figure out how many hidden windows were created, we divide
+ # this number by 2, because 1 hidden window creation implies in
+ # 1 outer window + 1 inner window.
+ # pylint --py3k W1619
+ self.logger.info(
+ "TEST-INFO | %s | This test created %d hidden window(s)"
+ % (test["fileName"], test["hiddenWindowsCount"] / 2)
+ )
+
+ if test["hiddenDocShellsCount"] > 0:
+ self.logger.info(
+ "TEST-INFO | %s | This test created %d hidden docshell(s)"
+ % (test["fileName"], test["hiddenDocShellsCount"])
+ )
+
+ return failures
+
+ def _logWindow(self, line, created):
+ pid = self._parseValue(line, "pid")
+ serial = self._parseValue(line, "serial")
+ self.numDomWindowCreatedLogsSeen += 1 if created else 0
+ self.numDomWindowDestroyedLogsSeen += 0 if created else 1
+
+ # log line has invalid format
+ if not pid or not serial:
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | ShutdownLeaks | failed to parse line <%s>"
+ % line
+ )
+ return
+
+ key = (pid, serial)
+
+ if self.currentTest:
+ windows = self.currentTest["windows"]
+ if created:
+ windows.add(key)
+ else:
+ windows.discard(key)
+ elif int(pid) in self.seenShutdown and not created:
+ url = self._parseValue(line, "url")
+ if not self._isHiddenWindowURL(url):
+ self.leakedWindows[key] = url
+ else:
+ self.hiddenWindowsCount += 1
+
+ def _logDocShell(self, line, created):
+ pid = self._parseValue(line, "pid")
+ id = self._parseValue(line, "id")
+ self.numDocShellCreatedLogsSeen += 1 if created else 0
+ self.numDocShellDestroyedLogsSeen += 0 if created else 1
+
+ # log line has invalid format
+ if not pid or not id:
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | ShutdownLeaks | failed to parse line <%s>"
+ % line
+ )
+ return
+
+ key = (pid, id)
+
+ if self.currentTest:
+ docShells = self.currentTest["docShells"]
+ if created:
+ docShells.add(key)
+ else:
+ docShells.discard(key)
+ elif int(pid) in self.seenShutdown and not created:
+ url = self._parseValue(line, "url")
+ if not self._isHiddenWindowURL(url):
+ self.leakedDocShells.add(key)
+ else:
+ self.hiddenDocShellsCount += 1
+
+ def _parseValue(self, line, name):
+ match = re.search("\[%s = (.+?)\]" % name, line)
+ if match:
+ return match.group(1)
+ return None
+
+ def _parseLeakingTests(self):
+ leakingTests = []
+
+ for test in self.tests:
+ leakedWindows = [id for id in test["windows"] if id in self.leakedWindows]
+ test["leakedWindows"] = [self.leakedWindows[id] for id in leakedWindows]
+ test["hiddenWindowsCount"] = self.hiddenWindowsCount
+ test["leakedWindowsString"] = ", ".join(
+ ["[pid = %s] [serial = %s]" % x for x in leakedWindows]
+ )
+ test["leakedDocShells"] = [
+ id for id in test["docShells"] if id in self.leakedDocShells
+ ]
+ test["hiddenDocShellsCount"] = self.hiddenDocShellsCount
+ test["leakCount"] = len(test["leakedWindows"]) + len(
+ test["leakedDocShells"]
+ )
+
+ if (
+ test["leakCount"]
+ or test["hiddenWindowsCount"]
+ or test["hiddenDocShellsCount"]
+ ):
+ leakingTests.append(test)
+
+ return sorted(leakingTests, key=itemgetter("leakCount"), reverse=True)
+
+ def _zipLeakedWindows(self, leakedWindows):
+ counts = []
+ counted = set()
+
+ for url in leakedWindows:
+ if url not in counted:
+ counts.append((url, leakedWindows.count(url)))
+ counted.add(url)
+
+ return sorted(counts, key=itemgetter(1), reverse=True)
+
+ def _isHiddenWindowURL(self, url):
+ return (
+ url == "resource://gre-resources/hiddenWindow.html"
+ or url == "chrome://browser/content/hiddenWindowMac.xhtml" # Win / Linux
+ ) # Mac
+
+
+class LSANLeaks(object):
+
+ """
+ Parses the log when running an LSAN build, looking for interesting stack frames
+ in allocation stacks, and prints out reports.
+ """
+
+ def __init__(self, logger):
+ self.logger = logger
+ self.inReport = False
+ self.fatalError = False
+ self.symbolizerError = False
+ self.foundFrames = set([])
+ self.recordMoreFrames = None
+ self.currStack = None
+ self.maxNumRecordedFrames = 4
+
+ # Don't various allocation-related stack frames, as they do not help much to
+ # distinguish different leaks.
+ unescapedSkipList = [
+ "malloc",
+ "js_malloc",
+ "js_arena_malloc",
+ "malloc_",
+ "__interceptor_malloc",
+ "moz_xmalloc",
+ "calloc",
+ "js_calloc",
+ "js_arena_calloc",
+ "calloc_",
+ "__interceptor_calloc",
+ "moz_xcalloc",
+ "realloc",
+ "js_realloc",
+ "js_arena_realloc",
+ "realloc_",
+ "__interceptor_realloc",
+ "moz_xrealloc",
+ "new",
+ "js::MallocProvider",
+ ]
+ self.skipListRegExp = re.compile(
+ "^" + "|".join([re.escape(f) for f in unescapedSkipList]) + "$"
+ )
+
+ self.startRegExp = re.compile(
+ "==\d+==ERROR: LeakSanitizer: detected memory leaks"
+ )
+ self.fatalErrorRegExp = re.compile(
+ "==\d+==LeakSanitizer has encountered a fatal error."
+ )
+ self.symbolizerOomRegExp = re.compile(
+ "LLVMSymbolizer: error reading file: Cannot allocate memory"
+ )
+ self.stackFrameRegExp = re.compile(" #\d+ 0x[0-9a-f]+ in ([^(</]+)")
+ self.sysLibStackFrameRegExp = re.compile(
+ " #\d+ 0x[0-9a-f]+ \(([^+]+)\+0x[0-9a-f]+\)"
+ )
+
+ def log(self, line):
+ if re.match(self.startRegExp, line):
+ self.inReport = True
+ return
+
+ if re.match(self.fatalErrorRegExp, line):
+ self.fatalError = True
+ return
+
+ if re.match(self.symbolizerOomRegExp, line):
+ self.symbolizerError = True
+ return
+
+ if not self.inReport:
+ return
+
+ if line.startswith("Direct leak") or line.startswith("Indirect leak"):
+ self._finishStack()
+ self.recordMoreFrames = True
+ self.currStack = []
+ return
+
+ if line.startswith("SUMMARY: AddressSanitizer"):
+ self._finishStack()
+ self.inReport = False
+ return
+
+ if not self.recordMoreFrames:
+ return
+
+ stackFrame = re.match(self.stackFrameRegExp, line)
+ if stackFrame:
+ # Split the frame to remove any return types.
+ frame = stackFrame.group(1).split()[-1]
+ if not re.match(self.skipListRegExp, frame):
+ self._recordFrame(frame)
+ return
+
+ sysLibStackFrame = re.match(self.sysLibStackFrameRegExp, line)
+ if sysLibStackFrame:
+ # System library stack frames will never match the skip list,
+ # so don't bother checking if they do.
+ self._recordFrame(sysLibStackFrame.group(1))
+
+ # If we don't match either of these, just ignore the frame.
+ # We'll end up with "unknown stack" if everything is ignored.
+
+ def process(self):
+ failures = 0
+
+ if self.fatalError:
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | LeakSanitizer | LeakSanitizer "
+ "has encountered a fatal error."
+ )
+ failures += 1
+
+ if self.symbolizerError:
+ self.logger.error(
+ "TEST-UNEXPECTED-FAIL | LeakSanitizer | LLVMSymbolizer "
+ "was unable to allocate memory."
+ )
+ failures += 1
+ self.logger.info(
+ "TEST-INFO | LeakSanitizer | This will cause leaks that "
+ "should be ignored to instead be reported as an error"
+ )
+
+ if self.foundFrames:
+ self.logger.info(
+ "TEST-INFO | LeakSanitizer | To show the "
+ "addresses of leaked objects add report_objects=1 to LSAN_OPTIONS"
+ )
+ self.logger.info(
+ "TEST-INFO | LeakSanitizer | This can be done "
+ "in testing/mozbase/mozrunner/mozrunner/utils.py"
+ )
+
+ for f in self.foundFrames:
+ self.logger.error("TEST-UNEXPECTED-FAIL | LeakSanitizer | leak at " + f)
+ failures += 1
+
+ return failures
+
+ def _finishStack(self):
+ if self.recordMoreFrames and len(self.currStack) == 0:
+ self.currStack = ["unknown stack"]
+ if self.currStack:
+ self.foundFrames.add(", ".join(self.currStack))
+ self.currStack = None
+ self.recordMoreFrames = False
+ self.numRecordedFrames = 0
+
+ def _recordFrame(self, frame):
+ self.currStack.append(frame)
+ self.numRecordedFrames += 1
+ if self.numRecordedFrames >= self.maxNumRecordedFrames:
+ self.recordMoreFrames = False