Home | History | Annotate | Download | only in binutils 
      1 #!/usr/bin/env python
      2 # Copyright 2014 The Chromium Authors. All rights reserved.
      3 # Use of this source code is governed by a BSD-style license that can be
      4 # found in the LICENSE file.
      5 # vim: set ts=2 sw=2 et sts=2 ai:
      6 
      7 """Minimal tool to download binutils from Google storage.
      8 
      9 TODO(mithro): Replace with generic download_and_extract tool.
     10 """
     11 
     12 import os
     13 import platform
     14 import re
     15 import shutil
     16 import subprocess
     17 import sys
     18 
     19 
     20 BINUTILS_DIR = os.path.abspath(os.path.dirname(__file__))
     21 BINUTILS_FILE = 'binutils.tar.bz2'
     22 BINUTILS_TOOLS = ['bin/ld.gold', 'bin/objcopy', 'bin/objdump']
     23 BINUTILS_OUT = 'Release'
     24 
     25 DETECT_HOST_ARCH = os.path.abspath(os.path.join(
     26     BINUTILS_DIR, '../../build/detect_host_arch.py'))
     27 
     28 
     29 def ReadFile(filename):
     30   with file(filename, 'r') as f:
     31     return f.read().strip()
     32 
     33 
     34 def WriteFile(filename, content):
     35   assert not os.path.exists(filename)
     36   with file(filename, 'w') as f:
     37     f.write(content)
     38     f.write('\n')
     39 
     40 
     41 def GetArch():
     42   gyp_host_arch = re.search(
     43       'host_arch=(\S*)', os.environ.get('GYP_DEFINES', ''))
     44   if gyp_host_arch:
     45     arch = gyp_host_arch.group(1)
     46     # This matches detect_host_arch.py.
     47     if arch == 'x86_64':
     48       return 'x64'
     49     return arch
     50 
     51   return subprocess.check_output(['python', DETECT_HOST_ARCH]).strip()
     52 
     53 
     54 def FetchAndExtract(arch):
     55   archdir = os.path.join(BINUTILS_DIR, 'Linux_' + arch)
     56   tarball = os.path.join(archdir, BINUTILS_FILE)
     57   outdir = os.path.join(archdir, BINUTILS_OUT)
     58 
     59   sha1file = tarball + '.sha1'
     60   if not os.path.exists(sha1file):
     61     print "WARNING: No binutils found for your architecture (%s)!" % arch
     62     return 0
     63 
     64   checksum = ReadFile(sha1file)
     65 
     66   stampfile = tarball + '.stamp'
     67   if os.path.exists(stampfile):
     68     if (os.path.exists(tarball) and
     69         os.path.exists(outdir) and
     70         checksum == ReadFile(stampfile)):
     71       return 0
     72     else:
     73       os.unlink(stampfile)
     74 
     75   print "Downloading", tarball
     76   subprocess.check_call([
     77       'download_from_google_storage',
     78       '--no_resume',
     79       '--no_auth',
     80       '--bucket', 'chromium-binutils',
     81       '-s', sha1file])
     82   assert os.path.exists(tarball)
     83 
     84   if os.path.exists(outdir):
     85     shutil.rmtree(outdir)
     86   assert not os.path.exists(outdir)
     87   os.makedirs(outdir)
     88   assert os.path.exists(outdir)
     89 
     90   print "Extracting", tarball
     91   subprocess.check_call(['tar', 'axf', tarball], cwd=outdir)
     92 
     93   for tool in BINUTILS_TOOLS:
     94     assert os.path.exists(os.path.join(outdir, tool))
     95 
     96   WriteFile(stampfile, checksum)
     97   return 0
     98 
     99 
    100 def main(args):
    101   if not sys.platform.startswith('linux'):
    102     return 0
    103 
    104   arch = GetArch()
    105   if arch == 'x64':
    106     return FetchAndExtract(arch)
    107   if arch == 'ia32':
    108     ret = FetchAndExtract(arch)
    109     if ret != 0:
    110       return ret
    111     # Fetch the x64 toolchain as well for official bots with 64-bit kernels.
    112     return FetchAndExtract('x64')
    113   print "Host architecture %s is not supported." % arch
    114   return 1
    115 
    116 
    117 if __name__ == '__main__':
    118   sys.exit(main(sys.argv))
    119