| Index: tools/perf/update_reference_build.py
|
| diff --git a/tools/perf/update_reference_build.py b/tools/perf/update_reference_build.py
|
| deleted file mode 100755
|
| index 9123de29745b8a5ded3b0a509a6f3cf1d7f9f09a..0000000000000000000000000000000000000000
|
| --- a/tools/perf/update_reference_build.py
|
| +++ /dev/null
|
| @@ -1,333 +0,0 @@
|
| -#!/usr/bin/env python
|
| -#
|
| -# Copyright 2013 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.
|
| -
|
| -"""Updates the Chrome reference builds.
|
| -
|
| -Before running this script, you should first verify that you are authenticated
|
| -for SVN. You can do this by running:
|
| - $ svn ls svn://svn.chromium.org/chrome/trunk/deps/reference_builds
|
| -You may need to get your SVN password from https://chromium-access.appspot.com/.
|
| -
|
| -Usage:
|
| - $ cd /tmp
|
| - $ /path/to/update_reference_build.py
|
| - $ cd reference_builds/reference_builds
|
| - $ gcl change
|
| - $ gcl upload <change>
|
| - $ gcl commit <change>
|
| -"""
|
| -
|
| -import collections
|
| -import logging
|
| -import os
|
| -import shutil
|
| -import subprocess
|
| -import sys
|
| -import urllib2
|
| -import zipfile
|
| -
|
| -
|
| -# Google storage location (no public web URL's), example:
|
| -# gs://chrome-unsigned/desktop-*/30.0.1595.0/precise32/chrome-precise32.zip
|
| -CHROME_GS_URL_FMT = ('gs://chrome-unsigned/desktop-*/%s/%s/%s')
|
| -
|
| -
|
| -def _ReportValueError(error_string):
|
| - # TODO(aiolos): alert sheriffs via email when an error is seen.
|
| - # This should be added when alerts are added when updating the build.
|
| - raise ValueError(error_string)
|
| -
|
| -
|
| -class BuildUpdater(object):
|
| - # Remove a platform name from this list to disable updating it.
|
| - _REF_BUILD_PLATFORMS = ['Mac64', 'Win', 'Linux', 'Linux_x64']
|
| -
|
| - # Omaha is Chrome's autoupdate server. It reports the current versions used
|
| - # by each platform on each channel.
|
| - _OMAHA_PLATFORMS = ['mac', 'linux', 'win']
|
| -
|
| - # All of the information we need to update each platform.
|
| - # omaha: name omaha uses for the platforms.
|
| - # zip_name: name of the zip file to be retrieved from cloud storage.
|
| - # gs_build: name of the Chrome build platform used in cloud storage.
|
| - # destination: Name of the folder to download the reference build to.
|
| - UpdateInfo = collections.namedtuple('UpdateInfo',
|
| - 'omaha, gs_build, zip_name, destination')
|
| - _PLATFORM_MAP = {'Mac64': UpdateInfo(omaha='mac',
|
| - gs_build='mac64',
|
| - zip_name='chrome-mac.zip',
|
| - destination='chrome_mac'),
|
| - 'Win': UpdateInfo(omaha='win',
|
| - gs_build='win',
|
| - zip_name='chrome-win.zip',
|
| - destination='chrome_win'),
|
| - 'Linux': UpdateInfo(omaha='linux',
|
| - gs_build='precise32',
|
| - zip_name='chrome-precise32.zip',
|
| - destination='chrome_linux'),
|
| - 'Linux_x64': UpdateInfo(omaha='linux',
|
| - gs_build='precise64',
|
| - zip_name='chrome-precise64.zip',
|
| - destination='chrome_linux64')}
|
| -
|
| - def __init__(self):
|
| - stable_versions = self._StableVersionsMap()
|
| - current_versions = self._CurrentRefBuildsMap()
|
| - self._platform_to_version_map = {}
|
| - for platform in stable_versions:
|
| - if (platform not in current_versions or
|
| - stable_versions[platform] != current_versions[platform]):
|
| - self._platform_to_version_map[platform] = stable_versions[platform]
|
| -
|
| - @classmethod
|
| - def _StableVersionsMap(cls):
|
| - omaha_versions_map = cls._OmahaVersionsMap()
|
| - versions_map = {}
|
| - for platform in cls._REF_BUILD_PLATFORMS:
|
| - omaha_platform = cls._PLATFORM_MAP[platform].omaha
|
| - if omaha_platform in omaha_versions_map:
|
| - versions_map[platform] = omaha_versions_map[omaha_platform]
|
| - return versions_map
|
| -
|
| - @classmethod
|
| - def _OmahaReport(cls):
|
| - url = 'https://omahaproxy.appspot.com/all?channel=stable'
|
| - lines = urllib2.urlopen(url).readlines()
|
| - return [l.split(',') for l in lines]
|
| -
|
| - @classmethod
|
| - def _OmahaVersionsMap(cls):
|
| - platforms = cls._OMAHA_PLATFORMS
|
| - rows = cls._OmahaReport()
|
| - if (len(rows) < 1 or
|
| - not rows[0][0:3] == ['os', 'channel', 'current_version']):
|
| - _ReportValueError('Omaha report is not in the expected form: %s.'
|
| - % rows)
|
| - versions_map = {}
|
| - for row in rows[1:]:
|
| - if row[1] != 'stable':
|
| - _ReportValueError('Omaha report contains a line with the channel %s'
|
| - % row[1])
|
| - if row[0] in platforms:
|
| - versions_map[row[0]] = row[2]
|
| -
|
| - if not all(platform in versions_map for platform in platforms):
|
| - _ReportValueError('Omaha report did not contain all desired platforms')
|
| - return versions_map
|
| -
|
| - @classmethod
|
| - def _CurrentRefBuildsMap(cls):
|
| - # TODO(aiolos): Add logic for pulling the current reference build versions.
|
| - # Return an empty dictionary to force an update until we store the builds in
|
| - # in cloud storage.
|
| - return {}
|
| -
|
| - @staticmethod
|
| - def _GetCmdStatusAndOutput(args, cwd=None, shell=False):
|
| - """Executes a subprocess and returns its exit code and output.
|
| -
|
| - Args:
|
| - args: A string or a sequence of program arguments.
|
| - cwd: If not None, the subprocess's current directory will be changed to
|
| - |cwd| before it's executed.
|
| - shell: Whether to execute args as a shell command.
|
| -
|
| - Returns:
|
| - The tuple (exit code, output).
|
| - """
|
| - logging.info(str(args) + ' ' + (cwd or ''))
|
| - p = subprocess.Popen(args=args, cwd=cwd, stdout=subprocess.PIPE,
|
| - stderr=subprocess.PIPE, shell=shell)
|
| - stdout, stderr = p.communicate()
|
| - exit_code = p.returncode
|
| - if stderr:
|
| - logging.critical(stderr)
|
| - logging.info(stdout)
|
| - return (exit_code, stdout)
|
| -
|
| - def _GetBuildUrl(self, platform, version, filename):
|
| - """Returns the URL for fetching one file.
|
| -
|
| - Args:
|
| - platform: Platform name, must be a key in |self._PLATFORM_MAP|.
|
| - version: A Chrome version number, e.g. 30.0.1600.1.
|
| - filename: Name of the file to fetch.
|
| -
|
| - Returns:
|
| - The URL for fetching a file. This may be a GS or HTTP URL.
|
| - """
|
| - return CHROME_GS_URL_FMT % (
|
| - version, self._PLATFORM_MAP[platform].gs_build, filename)
|
| -
|
| - def _FindBuildVersion(self, platform, version, filename):
|
| - """Searches for a version where a filename can be found.
|
| -
|
| - Args:
|
| - platform: Platform name.
|
| - version: A Chrome version number, e.g. 30.0.1600.1.
|
| - filename: Filename to look for.
|
| -
|
| - Returns:
|
| - A version where the file could be found, or None.
|
| - """
|
| - # TODO(shadi): Iterate over official versions to find a valid one.
|
| - return (version
|
| - if self._DoesBuildExist(platform, version, filename) else None)
|
| -
|
| - def _DoesBuildExist(self, platform, version, filename):
|
| - """Checks whether a file can be found for the given Chrome version.
|
| -
|
| - Args:
|
| - platform: Platform name.
|
| - version: Chrome version number, e.g. 30.0.1600.1.
|
| - filename: Filename to look for.
|
| -
|
| - Returns:
|
| - True if the file could be found, False otherwise.
|
| - """
|
| - url = self._GetBuildUrl(platform, version, filename)
|
| - return self._DoesGSFileExist(url)
|
| -
|
| - def _DoesGSFileExist(self, gs_file_name):
|
| - """Returns True if the GS file can be found, False otherwise."""
|
| - exit_code = BuildUpdater._GetCmdStatusAndOutput(
|
| - ['gsutil', 'ls', gs_file_name])[0]
|
| - return not exit_code
|
| -
|
| - def _GetPlatformFiles(self, platform):
|
| - """Returns the name of the zip file to fetch for |platform|."""
|
| - return BuildUpdater._PLATFORM_MAP[platform].zip_name
|
| -
|
| - def _DownloadBuilds(self):
|
| - for platform in self._platform_to_version_map:
|
| - version = self._platform_to_version_map[platform]
|
| - filename = self._GetPlatformFiles(platform)
|
| - output = os.path.join('dl', platform,
|
| - '%s_%s_%s' % (platform,
|
| - version,
|
| - filename))
|
| - if os.path.exists(output):
|
| - logging.info('%s alread exists, skipping download', output)
|
| - continue
|
| - build_version = self._FindBuildVersion(platform, version, filename)
|
| - if not build_version:
|
| - logging.critical('Failed to find %s build for r%s\n', platform,
|
| - version)
|
| - sys.exit(1)
|
| - dirname = os.path.dirname(output)
|
| - if dirname and not os.path.exists(dirname):
|
| - os.makedirs(dirname)
|
| - url = self._GetBuildUrl(platform, build_version, filename)
|
| - self._DownloadFile(url, output)
|
| -
|
| - def _DownloadFile(self, url, output):
|
| - logging.info('Downloading %s, saving to %s', url, output)
|
| - BuildUpdater._GetCmdStatusAndOutput(['gsutil', 'cp', url, output])
|
| -
|
| - def _FetchSvnRepos(self):
|
| - if not os.path.exists('reference_builds'):
|
| - os.makedirs('reference_builds')
|
| - BuildUpdater._GetCmdStatusAndOutput(
|
| - ['gclient', 'config',
|
| - 'svn://svn.chromium.org/chrome/trunk/deps/reference_builds'],
|
| - 'reference_builds')
|
| - BuildUpdater._GetCmdStatusAndOutput(
|
| - ['gclient', 'sync'], 'reference_builds')
|
| -
|
| - def _UnzipFile(self, dl_file, dest_dir):
|
| - """Unzips a file if it is a zip file.
|
| -
|
| - Args:
|
| - dl_file: The downloaded file to unzip.
|
| - dest_dir: The destination directory to unzip to.
|
| -
|
| - Returns:
|
| - True if the file was unzipped. False if it wasn't a zip file.
|
| - """
|
| - if not zipfile.is_zipfile(dl_file):
|
| - return False
|
| - logging.info('Opening %s', dl_file)
|
| - with zipfile.ZipFile(dl_file, 'r') as z:
|
| - for content in z.namelist():
|
| - dest = os.path.join(dest_dir, content[content.find('/') + 1:])
|
| - # Create dest parent dir if it does not exist.
|
| - if not os.path.isdir(os.path.dirname(dest)):
|
| - os.makedirs(os.path.dirname(dest))
|
| - # If dest is just a dir listing, do nothing.
|
| - if not os.path.basename(dest):
|
| - continue
|
| - if not os.path.isdir(os.path.dirname(dest)):
|
| - os.makedirs(os.path.dirname(dest))
|
| - with z.open(content) as unzipped_content:
|
| - logging.info('Extracting %s to %s (%s)', content, dest, dl_file)
|
| - with file(dest, 'wb') as dest_file:
|
| - dest_file.write(unzipped_content.read())
|
| - permissions = z.getinfo(content).external_attr >> 16
|
| - if permissions:
|
| - os.chmod(dest, permissions)
|
| - return True
|
| -
|
| - def _ClearDir(self, directory):
|
| - """Clears all files in |directory| except for hidden files and folders."""
|
| - for root, dirs, files in os.walk(directory):
|
| - # Skip hidden files and folders (like .svn and .git).
|
| - files = [f for f in files if f[0] != '.']
|
| - dirs[:] = [d for d in dirs if d[0] != '.']
|
| -
|
| - for f in files:
|
| - os.remove(os.path.join(root, f))
|
| -
|
| - def _ExtractBuilds(self):
|
| - for platform in self._platform_to_version_map:
|
| - if os.path.exists('tmp_unzip'):
|
| - os.path.unlink('tmp_unzip')
|
| - dest_dir = os.path.join(
|
| - 'reference_builds', 'reference_builds',
|
| - BuildUpdater._PLATFORM_MAP[platform].destination)
|
| - self._ClearDir(dest_dir)
|
| - for root, _, dl_files in os.walk(os.path.join('dl', platform)):
|
| - for dl_file in dl_files:
|
| - dl_file = os.path.join(root, dl_file)
|
| - if not self._UnzipFile(dl_file, dest_dir):
|
| - logging.info('Copying %s to %s', dl_file, dest_dir)
|
| - shutil.copy(dl_file, dest_dir)
|
| -
|
| - def _SvnAddAndRemove(self):
|
| - svn_dir = os.path.join('reference_builds', 'reference_builds')
|
| - # List all changes without ignoring any files.
|
| - stat = BuildUpdater._GetCmdStatusAndOutput(['svn', 'stat', '--no-ignore'],
|
| - svn_dir)[1]
|
| - for line in stat.splitlines():
|
| - action, filename = line.split(None, 1)
|
| - # Add new and ignored files.
|
| - if action == '?' or action == 'I':
|
| - BuildUpdater._GetCmdStatusAndOutput(
|
| - ['svn', 'add', filename], svn_dir)
|
| - elif action == '!':
|
| - BuildUpdater._GetCmdStatusAndOutput(
|
| - ['svn', 'delete', filename], svn_dir)
|
| - filepath = os.path.join(svn_dir, filename)
|
| - if not os.path.isdir(filepath) and os.access(filepath, os.X_OK):
|
| - BuildUpdater._GetCmdStatusAndOutput(
|
| - ['svn', 'propset', 'svn:executable', 'true', filename], svn_dir)
|
| -
|
| - def DownloadAndUpdateBuilds(self):
|
| - self._DownloadBuilds()
|
| - self._FetchSvnRepos()
|
| - self._ExtractBuilds()
|
| - self._SvnAddAndRemove()
|
| -
|
| -
|
| -def main():
|
| - logging.getLogger().setLevel(logging.DEBUG)
|
| - # TODO(aiolos): check that there are no options passed (argparse).
|
| - b = BuildUpdater()
|
| - b.DownloadAndUpdateBuilds()
|
| - logging.info('Successfully updated reference builds. Move to '
|
| - 'reference_builds/reference_builds and make a change with gcl.')
|
| -
|
| -if __name__ == '__main__':
|
| - main()
|
|
|