sandwich: Adds web page replay support for HTTPS

tools/android/loading/device_setup.py

 # Copyright 2016 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import contextlib
 import logging
 import os
 import shutil
 import subprocess
 import sys
@@ skipping 11 matching lines @@
 sys.path.append(os.path.join(_SRC_DIR, 'build', 'android'))
 from pylib import constants
 from pylib import flag_changer
 
 sys.path.append(os.path.join(_SRC_DIR, 'tools', 'perf'))
 from chrome_telemetry_build import chromium_config
 
 sys.path.append(chromium_config.GetTelemetryDir())
 from telemetry.internal.util import webpagereplay
 
+sys.path.append(os.path.join(_SRC_DIR, 'third_party', 'webpagereplay'))
+import adb_install_cert
+import certutils
+
 import devtools_monitor
 
 DEVTOOLS_PORT = 9222
 DEVTOOLS_HOSTNAME = 'localhost'
 DEFAULT_CHROME_PACKAGE = 'chrome'
 
 
 @contextlib.contextmanager
 def TemporaryDirectory():
   """Returns a freshly-created directory that gets automatically deleted after
@@ skipping 86 matching lines @@
     yield []
     return
 
   wpr_server_args = ['--use_closest_match']
   if record:
     wpr_server_args.append('--record')
     if os.path.exists(wpr_archive_path):
       os.remove(wpr_archive_path)
   else:
     assert os.path.exists(wpr_archive_path)
+
+  # Deploy certification authority to the device.
+  temp_certificate_dir = tempfile.mkdtemp()
+  wpr_ca_cert_path = os.path.join(temp_certificate_dir, 'testca.pem')
+  certutils.write_dummy_ca_cert(*certutils.generate_dummy_ca_cert(),
+                                cert_path=wpr_ca_cert_path)
+
+  device_cert_util = adb_install_cert.AndroidCertInstaller(
+      device.adb.GetDeviceSerial(), None, wpr_ca_cert_path)
+  device_cert_util.install_cert(overwrite_cert=True)
+  wpr_server_args.extend(['--should_generate_certs',
+                          '--https_root_ca_cert_path=' + wpr_ca_cert_path])
+
+  # Set up WPR server and device forwarder.
   wpr_server = webpagereplay.ReplayServer(wpr_archive_path,
       '127.0.0.1', 0, 0, None, wpr_server_args)
   ports = wpr_server.StartServer()[:-1]
   host_http_port = ports[0]
   host_https_port = ports[1]
 
   forwarder.Forwarder.Map([(0, host_http_port), (0, host_https_port)], device)
   device_http_port = forwarder.Forwarder.DevicePortForHostPort(host_http_port)
   device_https_port = forwarder.Forwarder.DevicePortForHostPort(host_https_port)
 
   try:
     yield [
       '--host-resolver-rules="MAP * 127.0.0.1,EXCLUDE localhost"',
       '--testing-fixed-http-port={}'.format(device_http_port),
       '--testing-fixed-https-port={}'.format(device_https_port)]
   finally:
     forwarder.Forwarder.UnmapDevicePort(device_http_port, device)
     forwarder.Forwarder.UnmapDevicePort(device_https_port, device)
     wpr_server.StopServer()
 
+    # Remove certification authority from the device.
+    device_cert_util.remove_cert()
+    shutil.rmtree(temp_certificate_dir)
+
 @contextlib.contextmanager
 def DeviceConnection(device,
                      package=DEFAULT_CHROME_PACKAGE,
                      hostname=DEVTOOLS_HOSTNAME,
                      port=DEVTOOLS_PORT,
                      host_exe='out/Release/chrome',
                      host_profile_dir=None,
                      additional_flags=None):
   """Context for starting recording on a device.
 
@@ skipping 63 matching lines @@
     package: the key for chrome package info.
     fn: the function to execute that launches chrome and performs the
         appropriate instrumentation. The function will receive a
         DevToolsConnection as its sole parameter.
 
   Returns:
     As fn() returns.
   """
   with DeviceConnection(device, package) as connection:
     return fn(connection)