#!/usr/bin/env python # Copyright 2014 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. """An auto-roller for GN binaries into Chromium. This script is used to update the GN binaries that a Chromium checkout uses. In order to update the binaries, one must follow four steps in order: 1. Trigger try jobs to build a new GN binary at tip-of-tree and upload the newly-built binaries into the right Google CloudStorage bucket. 2. Wait for the try jobs to complete. 3. Update the buildtools repo with the .sha1 hashes of the newly built binaries. 4. Update Chromium's DEPS file to the new version of the buildtools repo. The script has four commands that correspond to the four steps above: 'build', 'wait', 'roll_buildtools', and 'roll_deps'. The script has a fifth command, 'roll', that runs the four in order. If given no arguments, the script will run the 'roll' command. It can only be run on linux in a clean Chromium checkout; it should error out in most cases if something bad happens, but the error checking isn't yet foolproof. """ from __future__ import print_function import argparse import json import os import re import subprocess import sys import tempfile import time import urllib2 depot_tools_path = os.path.abspath(os.path.normpath(os.path.join( os.path.dirname(__file__), '..', '..', '..', 'third_party', 'depot_tools'))) if not depot_tools_path in sys.path: sys.path.insert(0, depot_tools_path) CHROMIUM_REPO = 'https://chromium.googlesource.com/chromium/src.git' COMMITISH_DIGITS = 10 UNKNOWN, PENDING, STARTED, SUCCESS = ( 'unknown', 'pending', 'started', 'success') class BuildResult(object): def __init__(self): self.masterName = '-' self.builderName = '-' self.buildNumber = '-' self.state = UNKNOWN self.sha1 = '-' self.url = '-' class GNRoller(object): def __init__(self): self.chromium_src_dir = None self.buildtools_dir = None self.old_gn_commitish = None self.new_gn_commitish = None self.old_gn_version = None self.new_gn_version = None self.reviewer = 'dpranke@chromium.org' if os.getenv('USER') == 'dpranke': self.reviewer = 'brettw@chromium.org' def Roll(self): parser = argparse.ArgumentParser() parser.usage = __doc__ parser.add_argument('command', nargs='?', default='roll', help='build|roll|roll_buildtools|roll_deps|wait' ' (%(default)s is the default)') args = parser.parse_args() command = args.command ret = self.SetUp() if not ret and command in ('roll', 'build'): ret = self.TriggerBuild() if not ret and command in ('roll', 'wait'): ret = self.WaitForBuildToFinish() if not ret and command in ('roll', 'roll_buildtools'): ret = self.RollBuildtools() if not ret and command in ('roll', 'roll_deps'): ret = self.RollDEPS() return ret def SetUp(self): if sys.platform not in ('darwin', 'linux2'): print('roll_gn is only tested and working on Linux and Mac for now.') return 1 ret, out, _ = self.Call('git config --get remote.origin.url') origin = out.strip() if ret or origin != CHROMIUM_REPO: print('Not in a Chromium repo? git config --get remote.origin.url ' 'returned %d: %s' % (ret, origin)) return 1 ret, _, _ = self.Call('git diff -q') if ret: print("Checkout is dirty, exiting") return 1 _, out, _ = self.Call('git rev-parse --show-toplevel', cwd=os.getcwd()) self.chromium_src_dir = out.strip() self.buildtools_dir = os.path.join(self.chromium_src_dir, 'buildtools') self.new_gn_commitish, self.new_gn_version = self.GetNewVersions() _, out, _ = self.Call('gn --version') self.old_gn_version = out.strip() _, out, _ = self.Call('git crrev-parse %s' % self.old_gn_version) self.old_gn_commitish = out.strip() return 0 def GetNewVersions(self): _, out, _ = self.Call('git log -1 --grep Cr-Commit-Position') commit_msg = out.splitlines() first_line = commit_msg[0] new_gn_commitish = first_line.split()[1] last_line = commit_msg[-1] new_gn_version = re.sub('.*master@{#(\d+)}', '\\1', last_line) return new_gn_commitish, new_gn_version def TriggerBuild(self): ret, _, _ = self.Call('git new-branch build_gn_%s' % self.new_gn_version) if ret: print('Failed to create a new branch for build_gn_%s' % self.new_gn_version) return 1 self.MakeDummyDepsChange() ret, out, err = self.Call('git commit -a -m "Build gn at %s"' % self.new_gn_version) if ret: print('git commit failed: %s' % out + err) return 1 print('Uploading CL to build GN at {#%s} - %s' % (self.new_gn_version, self.new_gn_commitish)) ret, out, err = self.Call('git cl upload -f') if ret: print('git-cl upload failed: %s' % out + err) return 1 print('Starting try jobs') self.Call('git-cl try -m tryserver.chromium.linux ' '-b linux_chromium_gn_upload -r %s' % self.new_gn_commitish) self.Call('git-cl try -m tryserver.chromium.mac ' '-b mac_chromium_gn_upload -r %s' % self.new_gn_commitish) self.Call('git-cl try -m tryserver.chromium.win ' '-b win_chromium_gn_upload -r %s' % self.new_gn_commitish) return 0 def MakeDummyDepsChange(self): with open('DEPS') as fp: deps_content = fp.read() new_deps = deps_content.replace("'buildtools_revision':", "'buildtools_revision': ") with open('DEPS', 'w') as fp: fp.write(new_deps) def WaitForBuildToFinish(self): ret = self.CheckoutBuildBranch() if ret: return ret print('Checking build') results = self.CheckBuild() while (any(r.state in (PENDING, STARTED) for r in results.values())): print() print('Sleeping for 30 seconds') time.sleep(30) print('Checking build') results = self.CheckBuild() ret = 0 if all(r.state == SUCCESS for r in results.values()) else 1 if ret: print('Build failed.') else: print('Builds ready.') # Close the build CL and move off of the build branch back to whatever # we were on before. self.Call('git-cl set-close') self.MoveToLastHead() return ret def CheckoutBuildBranch(self): ret, out, err = self.Call('git checkout build_gn_%s' % self.new_gn_version) if ret: print('Failed to check out build_gn_%s' % self.new_gn_version) if out: print(out) if err: print(err, file=sys.stderr) return ret def CheckBuild(self): _, out, _ = self.Call('git-cl try-results') builders = { 'linux_chromium_gn_upload': 'linux64', 'mac_chromium_gn_upload': 'mac', 'win_chromium_gn_upload': 'win' } results = {} for platform in ('linux64', 'mac', 'win'): results[platform] = BuildResult() state = PENDING for line in out.splitlines(): fields = line.strip().split() if fields[0] == 'Started:': state = STARTED if fields[0] == 'Successes:': state = SUCCESS elif fields[0] == 'Total': pass elif fields[0] in builders: builder = fields[0] platform = builders[builder] result = results[platform] result.masterName = ('tryserver.chromium.%s' % platform.replace('linux64', 'linux')) result.builderName = builder result.url = fields[1] if result.url.startswith('id'): result.state = PENDING else: result.state = state result.buildNumber = int(result.url[result.url.rfind('/')+1:]) for result in results.values(): if result.state == SUCCESS: url = 'https://luci-milo.appspot.com/prpc/milo.BuildInfo/Get' data = json.dumps({"buildbot": { 'masterName': result.masterName, 'builderName': result.builderName, 'buildNumber': result.buildNumber, }}) headers = { 'content-type': 'application/json', 'accept': 'application/json', } req = urllib2.Request(url, data, headers) resp = urllib2.urlopen(req) data = resp.read() resp.close() # The first line of the response is garbage; skip it. js = json.loads(data.splitlines()[1]) sha1_step_name = 'gn sha1' for step in js['step']['substep']: if step['step']['name'] == sha1_step_name: sha1 = step['step']['text'][-1] result.sha1 = sha1 for platform, r in results.items(): print(platform) print(' sha1: %s' % r.sha1) print(' state: %s' % r.state) print(' build: %s' % r.buildNumber) print(' url: %s' % r.url) print() return results def RollBuildtools(self): ret = self.CheckoutBuildBranch() if ret: return ret results = self.CheckBuild() if (len(results) < 3 or not all(r.state == SUCCESS for r in results.values()) or not all(r.sha1 != '-' for r in results.values())): print("Roll isn't done or didn't succeed, exiting:") return 1 desc = self.GetBuildtoolsDesc() self.Call('git new-branch roll_buildtools_gn_%s' % self.new_gn_version, cwd=self.buildtools_dir) for platform in results: fname = 'gn.exe.sha1' if platform == 'win' else 'gn.sha1' path = os.path.join(self.buildtools_dir, platform, fname) with open(path, 'w') as fp: fp.write('%s\n' % results[platform].sha1) desc_file = tempfile.NamedTemporaryFile(delete=False) try: desc_file.write(desc) desc_file.close() self.Call('git commit -a -F %s' % desc_file.name, cwd=self.buildtools_dir) self.Call('git-cl upload -f --send-mail', cwd=self.buildtools_dir) finally: os.remove(desc_file.name) ret, out, err = self.Call('git cl land', cwd=self.buildtools_dir) if ret: print("buildtools git cl land failed: %d" % ret) if out: print(out) if err: print(err) return ret # Fetch the revision we just committed so that RollDEPS will find it. self.Call('git fetch', cwd=self.buildtools_dir) # Reset buildtools to the new commit so that we're not still on the # merged branch. self.Call('git checkout origin/master', cwd=self.buildtools_dir) _, out, _ = self.Call('git rev-parse origin/master', cwd=self.buildtools_dir) new_buildtools_commitish = out.strip() print('Ready to roll buildtools to %s in DEPS' % new_buildtools_commitish) return 0 def RollDEPS(self): ret, _, _ = self.Call('git new-branch roll_gn_%s' % self.new_gn_version) if ret: print('Failed to create a new branch for roll_gn_%s' % self.new_gn_version) return 1 _, out, _ = self.Call('git rev-parse origin/master', cwd=self.buildtools_dir) new_buildtools_commitish = out.strip() new_deps_lines = [] old_buildtools_commitish = '' with open(os.path.join(self.chromium_src_dir, 'DEPS')) as fp: for l in fp.readlines(): m = re.match(".*'buildtools_revision':.*'(.+)',", l) if m: old_buildtools_commitish = m.group(1) new_deps_lines.append(" 'buildtools_revision': '%s',\n" % new_buildtools_commitish) else: new_deps_lines.append(l) if not old_buildtools_commitish: print('Could not update DEPS properly, exiting') return 1 with open('DEPS', 'w') as fp: fp.write(''.join(new_deps_lines)) desc = self.GetDEPSRollDesc(old_buildtools_commitish, new_buildtools_commitish) desc_file = tempfile.NamedTemporaryFile(delete=False) try: desc_file.write(desc) desc_file.close() self.Call('git commit -a -F %s' % desc_file.name) self.Call('git-cl upload -f --send-mail --use-commit-queue') finally: os.remove(desc_file.name) # Move off of the roll branch onto whatever we were on before. # Do not explicitly close the roll CL issue, however; the CQ # will close it when the roll lands, assuming it does so. self.MoveToLastHead() return 0 def MoveToLastHead(self): # When this is called, there will be a commit + a checkout as # the two most recent entries in the reflog, assuming nothing as # modified the repo while this script has been running. _, out, _ = self.Call('git reflog -2') m = re.search('moving from ([^\s]+)', out) last_head = m.group(1) self.Call('git checkout %s' % last_head) def GetBuildtoolsDesc(self): gn_changes = self.GetGNChanges() return ( 'Roll gn %s..%s (r%s:r%s)\n' '\n' '%s' '\n' 'TBR=%s\n' % ( self.old_gn_commitish[:COMMITISH_DIGITS], self.new_gn_commitish[:COMMITISH_DIGITS], self.old_gn_version, self.new_gn_version, gn_changes, self.reviewer, )) def GetDEPSRollDesc(self, old_buildtools_commitish, new_buildtools_commitish): gn_changes = self.GetGNChanges() return ( 'Roll buildtools %s..%s\n' '\n' ' In order to roll GN %s..%s (r%s:r%s) and pick up\n' ' the following changes:\n' '\n' '%s' '\n' 'TBR=%s\n' % ( old_buildtools_commitish[:COMMITISH_DIGITS], new_buildtools_commitish[:COMMITISH_DIGITS], self.old_gn_commitish[:COMMITISH_DIGITS], self.new_gn_commitish[:COMMITISH_DIGITS], self.old_gn_version, self.new_gn_version, gn_changes, self.reviewer, )) def GetGNChanges(self): _, out, _ = self.Call( "git log --pretty=' %h %s' " + "%s..%s tools/gn" % (self.old_gn_commitish, self.new_gn_commitish)) return out def Call(self, cmd, cwd=None): proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True, cwd=(cwd or self.chromium_src_dir)) out, err = proc.communicate() return proc.returncode, out or '', err or '' if __name__ == '__main__': roller = GNRoller() sys.exit(roller.Roll())