shenhan | b7ff88b | 2013-02-19 20:42:48 +0000 | [diff] [blame] | 1 | #!/usr/bin/python |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 2 | # |
Rahul Chaudhry | c1635a8 | 2014-10-31 14:59:39 -0700 | [diff] [blame] | 3 | # Copyright 2010 The Chromium OS Authors. All rights reserved. |
| 4 | # Use of this source code is governed by a BSD-style license that can be |
| 5 | # found in the LICENSE file. |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 6 | """Script to checkout the ChromeOS source. |
| 7 | |
| 8 | This script sets up the ChromeOS source in the given directory, matching a |
| 9 | particular release of ChromeOS. |
| 10 | """ |
| 11 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 12 | __author__ = 'raymes@google.com (Raymes Khoury)' |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 13 | |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 14 | from datetime import datetime |
asharif | ca35b77 | 2013-02-15 04:56:41 +0000 | [diff] [blame] | 15 | import getpass |
Luis Lozano | 8cf5308 | 2013-08-01 12:35:11 -0700 | [diff] [blame] | 16 | |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 17 | import optparse |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 18 | import os |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 19 | import pickle |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 20 | import sys |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 21 | import tempfile |
| 22 | import time |
raymes | 01959ae | 2013-02-15 04:50:07 +0000 | [diff] [blame] | 23 | from utils import command_executer |
raymes | 69c8d72 | 2013-02-15 17:55:36 +0000 | [diff] [blame] | 24 | from utils import logger |
Luis Lozano | 8cf5308 | 2013-08-01 12:35:11 -0700 | [diff] [blame] | 25 | from utils import manifest_versions |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 26 | |
raymes | bfb5799 | 2013-02-15 04:35:45 +0000 | [diff] [blame] | 27 | GCLIENT_FILE = """solutions = [ |
| 28 | { "name" : "CHROME_DEPS", |
| 29 | "url" : |
| 30 | "svn://svn.chromium.org/chrome-internal/trunk/tools/buildspec/releases/%s", |
| 31 | "custom_deps" : { |
| 32 | "src/third_party/WebKit/LayoutTests": None, |
| 33 | "src-pdf": None, |
| 34 | "src/pdf": None, |
| 35 | }, |
| 36 | "safesync_url": "", |
| 37 | }, |
| 38 | ] |
| 39 | """ |
| 40 | |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 41 | # List of stable versions used for common team image |
| 42 | # Sheriff must update this list when a new common version becomes available |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 43 | COMMON_VERSIONS = '/home/mobiletc-prebuild/common_images/common_list.txt' |
| 44 | |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 45 | |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 46 | def Usage(parser): |
| 47 | parser.print_help() |
| 48 | sys.exit(0) |
| 49 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 50 | |
shenhan | f6617ae | 2013-03-01 17:17:01 +0000 | [diff] [blame] | 51 | # Get version spec file, either from "paladin" or "buildspec" directory. |
| 52 | def GetVersionSpecFile(version, versions_git): |
| 53 | temp = tempfile.mkdtemp() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 54 | commands = ['cd {0}'.format(temp), \ |
| 55 | 'git clone {0} versions'.format(versions_git)] |
shenhan | f6617ae | 2013-03-01 17:17:01 +0000 | [diff] [blame] | 56 | cmd_executer = command_executer.GetCommandExecuter() |
| 57 | ret = cmd_executer.RunCommands(commands) |
| 58 | err_msg = None |
| 59 | if ret: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 60 | err_msg = 'Failed to checkout versions_git - {0}'.format(versions_git) |
shenhan | f6617ae | 2013-03-01 17:17:01 +0000 | [diff] [blame] | 61 | ret = None |
| 62 | else: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 63 | v, m = version.split('.', 1) |
| 64 | paladin_spec = 'paladin/buildspecs/{0}/{1}.xml'.format(v, m) |
| 65 | generic_spec = 'buildspecs/{0}/{1}.xml'.format(v, m) |
| 66 | paladin_path = '{0}/versions/{1}'.format(temp, paladin_spec) |
| 67 | generic_path = '{0}/versions/{1}'.format(temp, generic_spec) |
shenhan | f6617ae | 2013-03-01 17:17:01 +0000 | [diff] [blame] | 68 | if os.path.exists(paladin_path): |
| 69 | ret = paladin_spec |
| 70 | elif os.path.exists(generic_path): |
| 71 | ret = generic_spec |
| 72 | else: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 73 | err_msg = 'No spec found for version {0}'.format(version) |
shenhan | f6617ae | 2013-03-01 17:17:01 +0000 | [diff] [blame] | 74 | ret = None |
| 75 | # Fall through to clean up. |
| 76 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 77 | commands = ['rm -rf {0}'.format(temp)] |
shenhan | f6617ae | 2013-03-01 17:17:01 +0000 | [diff] [blame] | 78 | cmd_executer.RunCommands(commands) |
| 79 | if err_msg: |
| 80 | logger.GetLogger().LogFatal(err_msg) |
| 81 | return ret |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 82 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 83 | |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 84 | def TimeToCommonVersion(timestamp): |
| 85 | """Convert timestamp to common image version.""" |
| 86 | tdt = datetime.fromtimestamp(float(timestamp)) |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 87 | with open(COMMON_VERSIONS, 'r') as f: |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 88 | common_list = pickle.load(f) |
| 89 | for sv in common_list: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 90 | sdt = datetime.strptime(sv['date'], '%Y-%m-%d %H:%M:%S.%f') |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 91 | if tdt >= sdt: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 92 | return '%s.%s' % (sv['chrome_major_version'], sv['chromeos_version']) |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 93 | # should never reach here |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 94 | logger.GetLogger().LogFatal('No common version for timestamp') |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 95 | return None |
| 96 | |
| 97 | |
asharif | 0d3535a | 2013-02-15 04:50:33 +0000 | [diff] [blame] | 98 | def Main(argv): |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 99 | """Checkout the ChromeOS source.""" |
| 100 | parser = optparse.OptionParser() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 101 | parser.add_option('--dir', |
| 102 | dest='directory', |
| 103 | help='Target directory for ChromeOS installation.') |
| 104 | parser.add_option('--version', |
| 105 | dest='version', |
| 106 | default='latest_lkgm', |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 107 | help="""ChromeOS version. Can be: |
| 108 | (1) A release version in the format: 'X.X.X.X' |
| 109 | (2) 'top' for top of trunk |
| 110 | (3) 'latest_lkgm' for the latest lkgm version |
| 111 | (4) 'lkgm' for the lkgm release before timestamp |
| 112 | (5) 'latest_common' for the latest team common stable version |
| 113 | (6) 'common' for the team common stable version before timestamp |
| 114 | Default is 'latest_lkgm'.""") |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 115 | parser.add_option('--timestamp', |
| 116 | dest='timestamp', |
| 117 | default=None, |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 118 | help="""Timestamps in epoch format. It will check out the |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 119 | latest LKGM or the latest COMMON version of ChromeOS before the timestamp. |
| 120 | Use in combination with --version=latest or --version=common. Use |
| 121 | 'date -d <date string> +%s' to find epoch time""") |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 122 | parser.add_option('--minilayout', |
| 123 | dest='minilayout', |
| 124 | default=False, |
| 125 | action='store_true', |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 126 | help="""Whether to checkout the minilayout |
raymes | 01959ae | 2013-02-15 04:50:07 +0000 | [diff] [blame] | 127 | (smaller checkout).'""") |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 128 | parser.add_option('--jobs', |
| 129 | '-j', |
| 130 | dest='jobs', |
| 131 | help='Number of repo sync threads to use.') |
| 132 | parser.add_option('--public', |
| 133 | '-p', |
| 134 | dest='public', |
| 135 | default=False, |
| 136 | action='store_true', |
| 137 | help='Use the public checkout instead of the private one.') |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 138 | |
asharif | 0d3535a | 2013-02-15 04:50:33 +0000 | [diff] [blame] | 139 | options = parser.parse_args(argv)[0] |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 140 | |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 141 | if not options.version: |
| 142 | parser.print_help() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 143 | logger.GetLogger().LogFatal('No version specified.') |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 144 | else: |
| 145 | version = options.version.strip() |
| 146 | |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 147 | if not options.timestamp: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 148 | timestamp = '' |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 149 | else: |
| 150 | timestamp = options.timestamp.strip() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 151 | if version not in ('lkgm', 'common'): |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 152 | parser.print_help() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 153 | logger.GetLogger().LogFatal('timestamp option only applies for ' |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 154 | "versions \"lkgm\" or \"common\"") |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 155 | |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 156 | if not options.directory: |
| 157 | parser.print_help() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 158 | logger.GetLogger().LogFatal('No directory specified.') |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 159 | |
| 160 | directory = options.directory.strip() |
| 161 | |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 162 | if options.public: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 163 | manifest_repo = 'https://chromium.googlesource.com/chromiumos/manifest.git' |
| 164 | versions_repo = ('https://chromium.googlesource.com/' |
| 165 | 'chromiumos/manifest-versions.git') |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 166 | else: |
| 167 | manifest_repo = ( |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 168 | 'https://chrome-internal.googlesource.com/chromeos/manifest-internal.git' |
| 169 | ) |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 170 | versions_repo = ( |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 171 | 'https://chrome-internal.googlesource.com/chromeos/manifest-versions.git' |
| 172 | ) |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 173 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 174 | if version == 'top': |
| 175 | init = 'repo init -u %s' % manifest_repo |
| 176 | elif version == 'latest_lkgm': |
Luis Lozano | 8cf5308 | 2013-08-01 12:35:11 -0700 | [diff] [blame] | 177 | manifests = manifest_versions.ManifestVersions() |
| 178 | version = manifests.TimeToVersion(time.mktime(time.gmtime())) |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 179 | version, manifest = version.split('.', 1) |
| 180 | logger.GetLogger().LogOutput('found version %s.%s for latest LKGM' % |
| 181 | (version, manifest)) |
| 182 | init = ('repo init -u %s -m paladin/buildspecs/%s/%s.xml' % |
| 183 | (versions_repo, version, manifest)) |
Luis Lozano | 8cf5308 | 2013-08-01 12:35:11 -0700 | [diff] [blame] | 184 | del manifests |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 185 | elif version == 'lkgm': |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 186 | if not timestamp: |
| 187 | parser.print_help() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 188 | logger.GetLogger().LogFatal('No timestamp specified for version=lkgm') |
Luis Lozano | 8cf5308 | 2013-08-01 12:35:11 -0700 | [diff] [blame] | 189 | manifests = manifest_versions.ManifestVersions() |
| 190 | version = manifests.TimeToVersion(timestamp) |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 191 | version, manifest = version.split('.', 1) |
| 192 | logger.GetLogger().LogOutput('found version %s.%s for LKGM at timestamp %s' |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 193 | % (version, manifest, timestamp)) |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 194 | init = ('repo init -u %s -m paladin/buildspecs/%s/%s.xml' % |
| 195 | (versions_repo, version, manifest)) |
Luis Lozano | 8cf5308 | 2013-08-01 12:35:11 -0700 | [diff] [blame] | 196 | del manifests |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 197 | elif version == 'latest_common': |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 198 | version = TimeToCommonVersion(time.mktime(time.gmtime())) |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 199 | version, manifest = version.split('.', 1) |
| 200 | logger.GetLogger().LogOutput('found version %s.%s for latest Common image' % |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 201 | (version, manifest)) |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 202 | init = ('repo init -u %s -m buildspecs/%s/%s.xml' % (versions_repo, version, |
| 203 | manifest)) |
| 204 | elif version == 'common': |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 205 | if not timestamp: |
| 206 | parser.print_help() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 207 | logger.GetLogger().LogFatal('No timestamp specified for version=lkgm') |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 208 | version = TimeToCommonVersion(timestamp) |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 209 | version, manifest = version.split('.', 1) |
| 210 | logger.GetLogger().LogOutput('found version %s.%s for latest common image ' |
| 211 | 'at timestamp %s' % (version, manifest, |
| 212 | timestamp)) |
| 213 | init = ('repo init -u %s -m buildspecs/%s/%s.xml' % (versions_repo, version, |
| 214 | manifest)) |
raymes | 69c8d72 | 2013-02-15 17:55:36 +0000 | [diff] [blame] | 215 | else: |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 216 | # user specified a specific version number |
shenhan | f6617ae | 2013-03-01 17:17:01 +0000 | [diff] [blame] | 217 | version_spec_file = GetVersionSpecFile(version, versions_repo) |
| 218 | if not version_spec_file: |
| 219 | return 1 |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 220 | init = 'repo init -u %s -m %s' % (versions_repo, version_spec_file) |
llozano | 9efafde | 2013-02-23 01:37:04 +0000 | [diff] [blame] | 221 | |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 222 | if options.minilayout: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 223 | init += ' -g minilayout' |
yunlian | 3802fbf | 2013-02-19 19:58:44 +0000 | [diff] [blame] | 224 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 225 | init += ' --repo-url=https://chromium.googlesource.com/external/repo.git' |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 226 | |
Rahul Chaudhry | c1635a8 | 2014-10-31 14:59:39 -0700 | [diff] [blame] | 227 | # crosbug#31837 - "Sources need to be world-readable to properly |
| 228 | # function inside the chroot" |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 229 | sync = 'umask 022 && repo sync' |
Rahul Chaudhry | c1635a8 | 2014-10-31 14:59:39 -0700 | [diff] [blame] | 230 | if options.jobs: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 231 | sync += ' -j %s' % options.jobs |
Rahul Chaudhry | c1635a8 | 2014-10-31 14:59:39 -0700 | [diff] [blame] | 232 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 233 | commands = ['mkdir -p %s' % directory, 'cd %s' % directory, init, sync] |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 234 | cmd_executer = command_executer.GetCommandExecuter() |
| 235 | ret = cmd_executer.RunCommands(commands) |
| 236 | if ret: |
| 237 | return ret |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 238 | |
asharif | bf6899d | 2013-02-15 21:42:35 +0000 | [diff] [blame] | 239 | return cmd_executer.RunCommand( |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 240 | 'git ls-remote ' |
| 241 | 'https://chrome-internal.googlesource.com/chrome/src-internal.git ' |
| 242 | '> /dev/null') |
bjanakiraman | 7f4a485 | 2013-02-15 04:35:28 +0000 | [diff] [blame] | 243 | |
| 244 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 245 | if __name__ == '__main__': |
asharif | 2198c51 | 2013-02-15 09:21:35 +0000 | [diff] [blame] | 246 | retval = Main(sys.argv) |
| 247 | sys.exit(retval) |