| import os |
| import platform |
| import signal |
| import subprocess |
| import sys |
| |
| import mozinfo |
| import mozleak |
| from mozprocess import ProcessHandler |
| from mozprofile import FirefoxProfile, Preferences |
| from mozprofile.permissions import ServerLocations |
| from mozrunner import FirefoxRunner |
| from mozrunner.utils import get_stack_fixer_function |
| from mozcrash import mozcrash |
| |
| from .base import (get_free_port, |
| Browser, |
| ExecutorBrowser, |
| require_arg, |
| cmd_arg, |
| browser_command) |
| from ..executors import executor_kwargs as base_executor_kwargs |
| from ..executors.executormarionette import (MarionetteTestharnessExecutor, |
| MarionetteRefTestExecutor, |
| MarionetteWdspecExecutor) |
| from ..environment import hostnames |
| |
| |
| here = os.path.join(os.path.split(__file__)[0]) |
| |
| __wptrunner__ = {"product": "firefox", |
| "check_args": "check_args", |
| "browser": "FirefoxBrowser", |
| "executor": {"testharness": "MarionetteTestharnessExecutor", |
| "reftest": "MarionetteRefTestExecutor", |
| "wdspec": "MarionetteWdspecExecutor"}, |
| "browser_kwargs": "browser_kwargs", |
| "executor_kwargs": "executor_kwargs", |
| "env_extras": "env_extras", |
| "env_options": "env_options", |
| "run_info_extras": "run_info_extras", |
| "update_properties": "update_properties"} |
| |
| |
| def get_timeout_multiplier(test_type, run_info_data, **kwargs): |
| if kwargs["timeout_multiplier"] is not None: |
| return kwargs["timeout_multiplier"] |
| if test_type == "reftest": |
| if run_info_data["debug"] or run_info_data.get("asan"): |
| return 4 |
| else: |
| return 2 |
| elif run_info_data["debug"] or run_info_data.get("asan"): |
| return 3 |
| return 1 |
| |
| |
| def check_args(**kwargs): |
| require_arg(kwargs, "binary") |
| if kwargs["ssl_type"] != "none": |
| require_arg(kwargs, "certutil_binary") |
| |
| |
| def browser_kwargs(test_type, run_info_data, **kwargs): |
| return {"binary": kwargs["binary"], |
| "prefs_root": kwargs["prefs_root"], |
| "extra_prefs": kwargs["extra_prefs"], |
| "test_type": test_type, |
| "debug_info": kwargs["debug_info"], |
| "symbols_path": kwargs["symbols_path"], |
| "stackwalk_binary": kwargs["stackwalk_binary"], |
| "certutil_binary": kwargs["certutil_binary"], |
| "ca_certificate_path": kwargs["ssl_env"].ca_cert_path(), |
| "e10s": kwargs["gecko_e10s"], |
| "stackfix_dir": kwargs["stackfix_dir"], |
| "binary_args": kwargs["binary_args"], |
| "timeout_multiplier": get_timeout_multiplier(test_type, |
| run_info_data, |
| **kwargs), |
| "leak_check": kwargs["leak_check"], |
| "stylo_threads": kwargs["stylo_threads"]} |
| |
| |
| def executor_kwargs(test_type, server_config, cache_manager, run_info_data, |
| **kwargs): |
| executor_kwargs = base_executor_kwargs(test_type, server_config, |
| cache_manager, **kwargs) |
| executor_kwargs["close_after_done"] = test_type != "reftest" |
| executor_kwargs["timeout_multiplier"] = get_timeout_multiplier(test_type, |
| run_info_data, |
| **kwargs) |
| if test_type == "reftest": |
| executor_kwargs["reftest_internal"] = kwargs["reftest_internal"] |
| executor_kwargs["reftest_screenshot"] = kwargs["reftest_screenshot"] |
| if test_type == "wdspec": |
| executor_kwargs["binary"] = kwargs["binary"] |
| executor_kwargs["webdriver_binary"] = kwargs.get("webdriver_binary") |
| executor_kwargs["webdriver_args"] = kwargs.get("webdriver_args") |
| fxOptions = {} |
| if kwargs["binary"]: |
| fxOptions["binary"] = kwargs["binary"] |
| if kwargs["binary_args"]: |
| fxOptions["args"] = kwargs["binary_args"] |
| fxOptions["prefs"] = { |
| "network.dns.localDomains": ",".join(hostnames) |
| } |
| capabilities = {"moz:firefoxOptions": fxOptions} |
| executor_kwargs["capabilities"] = capabilities |
| return executor_kwargs |
| |
| |
| def env_extras(**kwargs): |
| return [] |
| |
| |
| def env_options(): |
| return {"host": "127.0.0.1", |
| "external_host": "web-platform.test", |
| "bind_hostname": "false", |
| "certificate_domain": "web-platform.test", |
| "supports_debugger": True} |
| |
| |
| def run_info_extras(**kwargs): |
| return {"e10s": kwargs["gecko_e10s"], |
| "headless": "MOZ_HEADLESS" in os.environ} |
| |
| |
| def update_properties(): |
| return (["debug", "stylo", "e10s", "os", "version", "processor", "bits"], |
| {"debug", "e10s", "stylo"}) |
| |
| |
| class FirefoxBrowser(Browser): |
| used_ports = set() |
| init_timeout = 60 |
| shutdown_timeout = 60 |
| |
| def __init__(self, logger, binary, prefs_root, test_type, extra_prefs=None, debug_info=None, |
| symbols_path=None, stackwalk_binary=None, certutil_binary=None, |
| ca_certificate_path=None, e10s=False, stackfix_dir=None, |
| binary_args=None, timeout_multiplier=None, leak_check=False, stylo_threads=1): |
| Browser.__init__(self, logger) |
| self.binary = binary |
| self.prefs_root = prefs_root |
| self.test_type = test_type |
| self.extra_prefs = extra_prefs |
| self.marionette_port = None |
| self.runner = None |
| self.debug_info = debug_info |
| self.profile = None |
| self.symbols_path = symbols_path |
| self.stackwalk_binary = stackwalk_binary |
| self.ca_certificate_path = ca_certificate_path |
| self.certutil_binary = certutil_binary |
| self.e10s = e10s |
| self.binary_args = binary_args |
| if self.symbols_path and stackfix_dir: |
| self.stack_fixer = get_stack_fixer_function(stackfix_dir, |
| self.symbols_path) |
| else: |
| self.stack_fixer = None |
| |
| if timeout_multiplier: |
| self.init_timeout = self.init_timeout * timeout_multiplier |
| |
| self.leak_report_file = None |
| self.leak_check = leak_check |
| self.stylo_threads = stylo_threads |
| |
| def settings(self, test): |
| return {"check_leaks": self.leak_check and not test.leaks} |
| |
| def start(self, **kwargs): |
| if self.marionette_port is None: |
| self.marionette_port = get_free_port(2828, exclude=self.used_ports) |
| self.used_ports.add(self.marionette_port) |
| |
| env = os.environ.copy() |
| env["MOZ_CRASHREPORTER"] = "1" |
| env["MOZ_CRASHREPORTER_SHUTDOWN"] = "1" |
| env["MOZ_DISABLE_NONLOCAL_CONNECTIONS"] = "1" |
| env["STYLO_THREADS"] = str(self.stylo_threads) |
| |
| locations = ServerLocations(filename=os.path.join(here, "server-locations.txt")) |
| |
| preferences = self.load_prefs() |
| |
| self.profile = FirefoxProfile(locations=locations, |
| preferences=preferences) |
| self.profile.set_preferences({"marionette.port": self.marionette_port, |
| "dom.disable_open_during_load": False, |
| "network.dns.localDomains": ",".join(hostnames), |
| "network.proxy.type": 0, |
| "places.history.enabled": False, |
| "dom.send_after_paint_to_content": True}) |
| if self.e10s: |
| self.profile.set_preferences({"browser.tabs.remote.autostart": True}) |
| |
| if self.test_type == "reftest": |
| self.profile.set_preferences({"layout.interruptible-reflow.enabled": False}) |
| |
| if self.leak_check and kwargs.get("check_leaks", True): |
| self.leak_report_file = os.path.join(self.profile.profile, "runtests_leaks.log") |
| if os.path.exists(self.leak_report_file): |
| os.remove(self.leak_report_file) |
| env["XPCOM_MEM_BLOAT_LOG"] = self.leak_report_file |
| else: |
| self.leak_report_file = None |
| |
| # Bug 1262954: winxp + e10s, disable hwaccel |
| if (self.e10s and platform.system() in ("Windows", "Microsoft") and |
| '5.1' in platform.version()): |
| self.profile.set_preferences({"layers.acceleration.disabled": True}) |
| |
| if self.ca_certificate_path is not None: |
| self.setup_ssl() |
| |
| debug_args, cmd = browser_command(self.binary, |
| self.binary_args if self.binary_args else [] + |
| [cmd_arg("marionette"), "about:blank"], |
| self.debug_info) |
| |
| self.runner = FirefoxRunner(profile=self.profile, |
| binary=cmd[0], |
| cmdargs=cmd[1:], |
| env=env, |
| process_class=ProcessHandler, |
| process_args={"processOutputLine": [self.on_output]}) |
| |
| self.logger.debug("Starting Firefox") |
| |
| self.runner.start(debug_args=debug_args, interactive=self.debug_info and self.debug_info.interactive) |
| self.logger.debug("Firefox Started") |
| |
| def load_prefs(self): |
| prefs = Preferences() |
| |
| prefs_path = os.path.join(self.prefs_root, "prefs_general.js") |
| if os.path.exists(prefs_path): |
| prefs.add(Preferences.read_prefs(prefs_path)) |
| else: |
| self.logger.warning("Failed to find base prefs file in %s" % prefs_path) |
| |
| # Add any custom preferences |
| prefs.add(self.extra_prefs, cast=True) |
| |
| return prefs() |
| |
| def stop(self, force=False): |
| if self.runner is not None and self.runner.is_running(): |
| try: |
| # For Firefox we assume that stopping the runner prompts the |
| # browser to shut down. This allows the leak log to be written |
| for clean, stop_f in [(True, lambda: self.runner.wait(self.shutdown_timeout)), |
| (False, lambda: self.runner.stop(signal.SIGTERM)), |
| (False, lambda: self.runner.stop(signal.SIGKILL))]: |
| if not force or not clean: |
| retcode = stop_f() |
| if retcode is not None: |
| self.logger.info("Browser exited with return code %s" % retcode) |
| break |
| except OSError: |
| # This can happen on Windows if the process is already dead |
| pass |
| self.logger.debug("stopped") |
| |
| def process_leaks(self): |
| self.logger.debug("PROCESS LEAKS %s" % self.leak_report_file) |
| if self.leak_report_file is None: |
| return |
| mozleak.process_leak_log( |
| self.leak_report_file, |
| leak_thresholds={ |
| "default": 0, |
| "tab": 10000, # See dependencies of bug 1051230. |
| # GMP rarely gets a log, but when it does, it leaks a little. |
| "geckomediaplugin": 20000, |
| }, |
| ignore_missing_leaks=["geckomediaplugin"], |
| log=self.logger, |
| stack_fixer=self.stack_fixer |
| ) |
| |
| def pid(self): |
| if self.runner.process_handler is None: |
| return None |
| |
| try: |
| return self.runner.process_handler.pid |
| except AttributeError: |
| return None |
| |
| def on_output(self, line): |
| """Write a line of output from the firefox process to the log""" |
| data = line.decode("utf8", "replace") |
| if self.stack_fixer: |
| data = self.stack_fixer(data) |
| self.logger.process_output(self.pid(), |
| data, |
| command=" ".join(self.runner.command)) |
| |
| def is_alive(self): |
| if self.runner: |
| return self.runner.is_running() |
| return False |
| |
| def cleanup(self): |
| self.stop() |
| self.process_leaks() |
| |
| def executor_browser(self): |
| assert self.marionette_port is not None |
| return ExecutorBrowser, {"marionette_port": self.marionette_port} |
| |
| def check_for_crashes(self): |
| dump_dir = os.path.join(self.profile.profile, "minidumps") |
| |
| return bool(mozcrash.check_for_crashes(dump_dir, |
| symbols_path=self.symbols_path, |
| stackwalk_binary=self.stackwalk_binary, |
| quiet=True)) |
| |
| def log_crash(self, process, test): |
| dump_dir = os.path.join(self.profile.profile, "minidumps") |
| |
| mozcrash.log_crashes(self.logger, |
| dump_dir, |
| symbols_path=self.symbols_path, |
| stackwalk_binary=self.stackwalk_binary, |
| process=process, |
| test=test) |
| |
| def setup_ssl(self): |
| """Create a certificate database to use in the test profile. This is configured |
| to trust the CA Certificate that has signed the web-platform.test server |
| certificate.""" |
| |
| self.logger.info("Setting up ssl") |
| |
| # Make sure the certutil libraries from the source tree are loaded when using a |
| # local copy of certutil |
| # TODO: Maybe only set this if certutil won't launch? |
| env = os.environ.copy() |
| certutil_dir = os.path.dirname(self.binary) |
| if mozinfo.isMac: |
| env_var = "DYLD_LIBRARY_PATH" |
| elif mozinfo.isUnix: |
| env_var = "LD_LIBRARY_PATH" |
| else: |
| env_var = "PATH" |
| |
| |
| env[env_var] = (os.path.pathsep.join([certutil_dir, env[env_var]]) |
| if env_var in env else certutil_dir).encode( |
| sys.getfilesystemencoding() or 'utf-8', 'replace') |
| |
| def certutil(*args): |
| cmd = [self.certutil_binary] + list(args) |
| self.logger.process_output("certutil", |
| subprocess.check_output(cmd, |
| env=env, |
| stderr=subprocess.STDOUT), |
| " ".join(cmd)) |
| |
| pw_path = os.path.join(self.profile.profile, ".crtdbpw") |
| with open(pw_path, "w") as f: |
| # Use empty password for certificate db |
| f.write("\n") |
| |
| cert_db_path = self.profile.profile |
| |
| # Create a new certificate db |
| certutil("-N", "-d", cert_db_path, "-f", pw_path) |
| |
| # Add the CA certificate to the database and mark as trusted to issue server certs |
| certutil("-A", "-d", cert_db_path, "-f", pw_path, "-t", "CT,,", |
| "-n", "web-platform-tests", "-i", self.ca_certificate_path) |
| |
| # List all certs in the database |
| certutil("-L", "-d", cert_db_path) |
