makebuildserver

   1 #!/usr/bin/env python3
   2
   3 import os
   4 import requests
   5 import sys
   6 import subprocess
   7 import time
   8 import hashlib
   9 import yaml
  10 from clint.textui import progress
  11 from optparse import OptionParser
  12
  13
  14 def vagrant(params, cwd=None, printout=False):
  15     """Run vagrant.
  16
  17     :param: list of parameters to pass to vagrant
  18     :cwd: directory to run in, or None for current directory
  19     :printout: True to print output in realtime, False to just
  20                return it
  21     :returns: (ret, out) where ret is the return code, and out
  22                is the stdout (and stderr) from vagrant
  23     """
  24     p = subprocess.Popen(['vagrant'] + params, cwd=cwd,
  25                          stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
  26                          universal_newlines=True)
  27     out = ''
  28     if printout:
  29         while True:
  30             line = p.stdout.readline()
  31             if len(line) == 0:
  32                 break
  33             print(line.rstrip())
  34             out += line
  35         p.wait()
  36     else:
  37         out = p.communicate()[0]
  38     return (p.returncode, out)
  39
  40 boxfile = 'buildserver.box'
  41 serverdir = 'buildserver'
  42
  43 parser = OptionParser()
  44 parser.add_option("-v", "--verbose", action="store_true", default=False,
  45                   help="Spew out even more information than normal")
  46 parser.add_option("-c", "--clean", action="store_true", default=False,
  47                   help="Build from scratch, rather than attempting to update the existing server")
  48 options, args = parser.parse_args()
  49
  50 # set up default config
  51 cachedir = os.path.join(os.getenv('HOME'), '.cache', 'fdroidserver')
  52 config = {
  53     'basebox': 'jessie64',
  54     # TODO in py3, convert this to pathlib.Path(absolute_path_string).as_uri()
  55     'baseboxurl': [
  56         'file://' + os.path.join(cachedir, 'jessie64.box'),
  57         'https://f-droid.org/jessie64.box',
  58     ],
  59     'debian_mirror': 'http://http.debian.net/debian/',
  60     'apt_package_cache': False,
  61     'boot_timeout': 600,
  62     'cachedir': cachedir,
  63     'cpus': 1,
  64     'memory': 1024,
  65     'hwvirtex': 'off',
  66 }
  67
  68 # load config file, if present
  69 if os.path.exists('makebuildserver.config.py'):
  70     exec(compile(open('makebuildserver.config.py').read(), 'makebuildserver.config.py', 'exec'), config)
  71 elif os.path.exists('makebs.config.py'):
  72     # this is the old name for the config file
  73     exec(compile(open('makebs.config.py').read(), 'makebs.config.py', 'exec'), config)
  74 del(config['__builtins__'])  # added by compile/exec
  75
  76 if not os.path.exists('makebuildserver') or not os.path.exists(serverdir):
  77     print('This must be run from the correct directory!')
  78     sys.exit(1)
  79
  80 if os.path.exists(boxfile):
  81     os.remove(boxfile)
  82
  83 if options.clean:
  84     vagrant(['destroy', '-f'], cwd=serverdir, printout=options.verbose)
  85
  86 # Update cached files.
  87 cachedir = config['cachedir']
  88 if not os.path.exists(cachedir):
  89     os.makedirs(cachedir, 0o755)
  90
  91 if config['apt_package_cache']:
  92     config['aptcachedir'] = cachedir + '/apt/archives'
  93
  94 cachefiles = [
  95     ('https://dl.google.com/android/repository/tools_r25.1.7-linux.zip',
  96      '3ca053600a86a5a64d5571edfbb1dad27f2bda3bfd2d38e2fe54322610b1ef0b'),
  97     ('https://dl.google.com/android/repository/platform-tools_r24-linux.zip',
  98      '076368b337d042d163364594dda63b7e778835f636fafb2c8af4d4a604175c32'),
  99     ('https://dl.google.com/android/repository/android_m2repository_r33.zip',
 100      'be9bb4a27aeefb1c9adb0cade8771f764447c4cbde74426303db2ac6bde1879c'),
 101     ('https://dl.google.com/android/repository/android-1.5_r04-linux.zip',
 102      '85b6c8f9797e56aa415d3a282428bb640c96b0acb17c11d41621bb2a5302fe64'),
 103     ('https://dl.google.com/android/repository/android-1.6_r03-linux.zip',
 104      'a8c4e3b32269c6b04c2adeabd112fce42f292dab1a40ef3b08ea7d4212be0df4'),
 105     ('https://dl.google.com/android/repository/android-2.0_r01-linux.zip',
 106      'e70e2151b49613f23f40828c771ab85e241eed361cab037c6312df77f2612f0a'),
 107     ('https://dl.google.com/android/repository/android-2.0.1_r01-linux.zip',
 108      'f47b46177b17f6368461f85bc2a27d0d2c437929f588ea27105712bc3185f664'),
 109     ('https://dl.google.com/android/repository/android-2.1_r03.zip',
 110      'b9cc140a9b879586181b22cfc7d4aa18b979251e16e9b17771c5d0acb71ba940'),
 111     ('https://dl.google.com/android/repository/android-2.2_r03.zip',
 112      '7c9ea1bd7cb225504bd085d7c93ae27d52bd88d29b621d28108f82fef68177c0'),
 113     ('https://dl.google.com/android/repository/android-2.3.1_r02.zip',
 114      'b2ab4896d0a4857e4f688f69eb08b0e1a8074709d4445a92a83ece7ec7cd198c'),
 115     ('https://dl.google.com/android/repository/android-2.3.3_r02.zip',
 116      '54bdb0f1ca06ba5747061ddeea20f431af72c448334fd4d3d7f84ea2ccd29fea'),
 117     ('https://dl.google.com/android/repository/android-3.0_r02.zip',
 118      '1cacae7b6e1b5a5d73c06f5d29d2ea92d16674df8fd5507681290e77d1647a1c'),
 119     ('https://dl.google.com/android/repository/android-3.1_r03.zip',
 120      '7570c86a86488a146aa2141a65a24d81800959c1907ff4f1d2c13bbafab230c5'),
 121     ('https://dl.google.com/android/repository/android-3.2_r01.zip',
 122      'ff6b26ad34d7060a72ba504b0314cef8ba3138005561705adec5ad470a073d9b'),
 123     ('https://dl.google.com/android/repository/android-14_r04.zip',
 124      'da1af15c77ba41d062eb6d0ef5921cc424ab6167587033b830609d65f04802b6'),
 125     ('https://dl.google.com/android/repository/android-15_r05.zip',
 126      '5bc1f93aae86b4336ffc4cae9eb8ec41a9a8fd677582dd86a9629798f019bed9'),
 127     ('https://dl.google.com/android/repository/android-16_r05.zip',
 128      'fd7f269a423d1f1d079eabf9f918ceab49108702a1c6bb2589d57c23393503d3'),
 129     ('https://dl.google.com/android/repository/android-17_r03.zip',
 130      'b66e73fb2639f8c916fde4369aa29012a5c531e156dbb205fe3788fe998fbbe8'),
 131     ('https://dl.google.com/android/repository/android-18_r03.zip',
 132      '166ae9cf299747a5faa8f04168f0ee47cd7466a975d8b44acaaa62a43e767568'),
 133     ('https://dl.google.com/android/repository/android-19_r04.zip',
 134      '5efc3a3a682c1d49128daddb6716c433edf16e63349f32959b6207524ac04039'),
 135     ('https://dl.google.com/android/repository/android-20_r02.zip',
 136      'ef08c453e16ab6e656cf5d9413ef61cb8c650607d33b24ee4ce08dafdfe965a7'),
 137     ('https://dl.google.com/android/repository/android-21_r02.zip',
 138      'a76cd7ad3080ac6ce9f037cb935b399a1bad396c0605d4ff42f693695f1dcefe'),
 139     ('https://dl.google.com/android/repository/android-22_r02.zip',
 140      '45eb581bbe53c9256f34c26b2cea919543c0079140897ac721cf88c0b9f6789e'),
 141     ('https://dl.google.com/android/repository/platform-23_r03.zip',
 142      '4b4bcddead3319708275c54c76294707bfaa953d767e34f1a5b599f3edd0076c'),
 143     ('https://dl.google.com/android/repository/platform-24_r01.zip',
 144      '292e556e9ec84b4134e4126c835e4ac467f6950924fec3684792af0fcf426091'),
 145     ('https://dl.google.com/android/repository/build-tools_r17-linux.zip',
 146      '4c8444972343a19045236f6924bd7f12046287c70dace96ab88b2159c8ec0e74'),
 147     ('https://dl.google.com/android/repository/build-tools_r18.0.1-linux.zip',
 148      'a9b7b1bdfd864780fdd03fa1683f3fe712a4276cf200646833808cb9159bafc0'),
 149     ('https://dl.google.com/android/repository/build-tools_r18.1-linux.zip',
 150      '0753606738f31cc346426db1d46b7d021bc1bdaff63085f9ee9d278ee054d3c9'),
 151     ('https://dl.google.com/android/repository/build-tools_r18.1.1-linux.zip',
 152      '7e4ed326b53078f4f23276ddab52c400011f7593dfbb6508c0a6671954dba8b0'),
 153     ('https://dl.google.com/android/repository/build-tools_r19-linux.zip',
 154      '9442e1c5212ed594e344a231fa93e7a017a5ef8cc661117011f1d3142eca7acc'),
 155     ('https://dl.google.com/android/repository/build-tools_r19.0.1-linux.zip',
 156      'b068edaff05c3253a63e9c8f0e1786429799b7e4b01514a847a8b291beb9232e'),
 157     ('https://dl.google.com/android/repository/build-tools_r19.0.2-linux.zip',
 158      '06124fad0d4bde21191240d61df2059a8546c085064a9a57d024c36fa2c9bebb'),
 159     ('https://dl.google.com/android/repository/build-tools_r19.0.3-linux.zip',
 160      'bc9b3db0de4a3e233a170274293359051a758f1e3f0d0d852ff4ad6d90d0a794'),
 161     ('https://dl.google.com/android/repository/build-tools_r19.1-linux.zip',
 162      '3833b409f78c002a83244e220be380ea6fa44d604e0d47de4b7e5daefe7cd3f4'),
 163     ('https://dl.google.com/android/repository/build-tools_r20-linux.zip',
 164      '296e09d62095d80e6eaa06a64cfa4c6f9f317c2d67ad8da6514523ec66f5c871'),
 165     ('https://dl.google.com/android/repository/build-tools_r21-linux.zip',
 166      '12b818f38fe1b68091b94545988317438efbf41eb61fd36b72cd79f536044065'),
 167     ('https://dl.google.com/android/repository/build-tools_r21.0.1-linux.zip',
 168      'a8922e80d3dd0cf6df14b29a7862448fa111b48086c639168d4b18c92431f559'),
 169     ('https://dl.google.com/android/repository/build-tools_r21.0.2-linux.zip',
 170      '859b17a6b65d063dfd86c163489b736b12bdeecd9173fdddb3e9f32e0fe584b7'),
 171     ('https://dl.google.com/android/repository/build-tools_r21.1-linux.zip',
 172      '022a85b92360272379b2f04b8a4d727e754dbe7eb8ab5a9568190e33e480d8f1'),
 173     ('https://dl.google.com/android/repository/build-tools_r21.1.1-linux.zip',
 174      '29b612484de6b5cde0df6de655e413f7611b0557b440538397afa69b557e2f08'),
 175     ('https://dl.google.com/android/repository/build-tools_r21.1.2-linux.zip',
 176      '3f88efc2d5316fb73f547f35b472610eed5e6f3f56762750ddad1c7d1d81660d'),
 177     ('https://dl.google.com/android/repository/build-tools_r22-linux.zip',
 178      '061c021243f04c80c19568a6e3a027c00d8e269c9311d7bf07fced60fbde7bd5'),
 179     ('https://dl.google.com/android/repository/build-tools_r22.0.1-linux.zip',
 180      '91e5524bf227aad1135ddd10905518ac49f74797d33d48920dcf8364b9fde214'),
 181     ('https://dl.google.com/android/repository/build-tools_r23-linux.zip',
 182      '56bf4fc6c43638c55fef4a0937bad38281945725459841879b436c6922df786c'),
 183     ('https://dl.google.com/android/repository/build-tools_r23.0.1-linux.zip',
 184      'e56b3ef7b760ad06a7cee9b2d52ba7f43133dcecedfa5357f8845b3a80aeeecf'),
 185     ('https://dl.google.com/android/repository/build-tools_r23.0.2-linux.zip',
 186      '82754f551a6e36eaf516fbdd00c95ff0ccd19f81d1e134125b6ac4916f7ed9b6'),
 187     ('https://dl.google.com/android/repository/build-tools_r23.0.3-linux.zip',
 188      'd961663d4a9e128841751c0156548a347c882c081c83942e53788d8949bf34e1'),
 189     ('https://dl.google.com/android/repository/build-tools_r24-linux.zip',
 190      'b4871f357224c5f660fd2bbee04d8c7d1c187eeddfd9702cc84503529e3b3724'),
 191     # the binaries that Google uses are here:
 192     # https://android.googlesource.com/platform/tools/external/gradle/+/studio-1.5/
 193     ('https://services.gradle.org/distributions/gradle-1.4-bin.zip',
 194      'cd99e85fbcd0ae8b99e81c9992a2f10cceb7b5f009c3720ef3a0078f4f92e94e'),
 195     ('https://services.gradle.org/distributions/gradle-1.6-bin.zip',
 196      'de3e89d2113923dcc2e0def62d69be0947ceac910abd38b75ec333230183fac4'),
 197     ('https://services.gradle.org/distributions/gradle-1.7-bin.zip',
 198      '360c97d51621b5a1ecf66748c718594e5f790ae4fbc1499543e0c006033c9d30'),
 199     ('https://services.gradle.org/distributions/gradle-1.8-bin.zip',
 200      'a342bbfa15fd18e2482287da4959588f45a41b60910970a16e6d97959aea5703'),
 201     ('https://services.gradle.org/distributions/gradle-1.9-bin.zip',
 202      '097ddc2bcbc9da2bb08cbf6bf8079585e35ad088bafd42e8716bc96405db98e9'),
 203     ('https://services.gradle.org/distributions/gradle-1.10-bin.zip',
 204      '6e6db4fc595f27ceda059d23693b6f6848583950606112b37dfd0e97a0a0a4fe'),
 205     ('https://services.gradle.org/distributions/gradle-1.11-bin.zip',
 206      '07e235df824964f0e19e73ea2327ce345c44bcd06d44a0123d29ab287fc34091'),
 207     ('https://services.gradle.org/distributions/gradle-1.12-bin.zip',
 208      '8734b13a401f4311ee418173ed6ca8662d2b0a535be8ff2a43ecb1c13cd406ea'),
 209     ('https://services.gradle.org/distributions/gradle-2.1-bin.zip',
 210      '3eee4f9ea2ab0221b89f8e4747a96d4554d00ae46d8d633f11cfda60988bf878'),
 211     ('https://services.gradle.org/distributions/gradle-2.2-bin.zip',
 212      '91e5655fe11ef414449f218c4fa2985b3a49b7903c57556da109c84fa26e1dfb'),
 213     ('https://services.gradle.org/distributions/gradle-2.2.1-bin.zip',
 214      '420aa50738299327b611c10b8304b749e8d3a579407ee9e755b15921d95ff418'),
 215     ('https://services.gradle.org/distributions/gradle-2.3-bin.zip',
 216      '010dd9f31849abc3d5644e282943b1c1c355f8e2635c5789833979ce590a3774'),
 217     ('https://services.gradle.org/distributions/gradle-2.4-bin.zip',
 218      'c4eaecc621a81f567ded1aede4a5ddb281cc02a03a6a87c4f5502add8fc2f16f'),
 219     ('https://services.gradle.org/distributions/gradle-2.5-bin.zip',
 220      '3f953e0cb14bb3f9ebbe11946e84071547bf5dfd575d90cfe9cc4e788da38555'),
 221     ('https://services.gradle.org/distributions/gradle-2.6-bin.zip',
 222      '18a98c560af231dfa0d3f8e0802c20103ae986f12428bb0a6f5396e8f14e9c83'),
 223     ('https://services.gradle.org/distributions/gradle-2.7-bin.zip',
 224      'cde43b90945b5304c43ee36e58aab4cc6fb3a3d5f9bd9449bb1709a68371cb06'),
 225     ('https://services.gradle.org/distributions/gradle-2.8-bin.zip',
 226      'a88db9c2f104defdaa8011c58cf6cda6c114298ae3695ecfb8beb30da3a903cb'),
 227     ('https://services.gradle.org/distributions/gradle-2.9-bin.zip',
 228      'c9159ec4362284c0a38d73237e224deae6139cbde0db4f0f44e1c7691dd3de2f'),
 229     ('https://services.gradle.org/distributions/gradle-2.10-bin.zip',
 230      '66406247f745fc6f05ab382d3f8d3e120c339f34ef54b86f6dc5f6efc18fbb13'),
 231     ('https://services.gradle.org/distributions/gradle-2.11-bin.zip',
 232      '8d7437082356c9fd6309a4479c8db307673965546daea445c6c72759cd6b1ed6'),
 233     ('https://services.gradle.org/distributions/gradle-2.12-bin.zip',
 234      'e77064981906cd0476ff1e0de3e6fef747bd18e140960f1915cca8ff6c33ab5c'),
 235     ('https://services.gradle.org/distributions/gradle-2.13-bin.zip',
 236      '0f665ec6a5a67865faf7ba0d825afb19c26705ea0597cec80dd191b0f2cbb664'),
 237     ('https://services.gradle.org/distributions/gradle-2.14-bin.zip',
 238      '993b4f33b652c689e9721917d8e021cab6bbd3eae81b39ab2fd46fdb19a928d5'),
 239     ('https://pypi.python.org/packages/source/K/Kivy/Kivy-1.7.2.tar.gz',
 240      '0485e2ef97b5086df886eb01f8303cb542183d2d71a159466f99ad6c8a1d03f1'),
 241     ('https://dl.google.com/android/ndk/android-ndk-r10e-linux-x86_64.bin',
 242      '102d6723f67ff1384330d12c45854315d6452d6510286f4e5891e00a5a8f1d5a'),
 243     ('https://dl.google.com/android/ndk/android-ndk-r9b-linux-x86_64.tar.bz2',
 244      '8956e9efeea95f49425ded8bb697013b66e162b064b0f66b5c75628f76e0f532'),
 245     ('https://dl.google.com/android/ndk/android-ndk-r9b-linux-x86_64-legacy-toolchains.tar.bz2',
 246      'de93a394f7c8f3436db44568648f87738a8d09801a52f459dcad3fc047e045a1'),
 247 ]
 248
 249
 250 def sha256_for_file(path):
 251     with open(path, 'rb') as f:
 252         s = hashlib.sha256()
 253         while True:
 254             data = f.read(4096)
 255             if not data:
 256                 break
 257             s.update(data)
 258         return s.hexdigest()
 259
 260 for srcurl, shasum in cachefiles:
 261     filename = os.path.basename(srcurl)
 262     local_filename = os.path.join(cachedir, filename)
 263
 264     if os.path.exists(local_filename):
 265         local_length = os.path.getsize(local_filename)
 266     else:
 267         local_length = -1
 268
 269     resume_header = {}
 270     download = True
 271
 272     try:
 273         r = requests.head(srcurl, allow_redirects=True, timeout=60)
 274         if r.status_code == 200:
 275             content_length = int(r.headers.get('content-length'))
 276         else:
 277             content_length = local_length  # skip the download
 278     except requests.exceptions.RequestException as e:
 279         content_length = local_length  # skip the download
 280         print(e)
 281
 282     if local_length == content_length:
 283         download = False
 284     elif local_length > content_length:
 285         print('deleting corrupt file from cache: ' + local_filename)
 286         os.remove(local_filename)
 287         print("Downloading " + filename + " to cache")
 288     elif local_length > -1 and local_length < content_length:
 289         print("Resuming download of " + local_filename)
 290         resume_header = {'Range': 'bytes=%d-%d' % (local_length, content_length)}
 291     else:
 292         print("Downloading " + filename + " to cache")
 293
 294     if download:
 295         r = requests.get(srcurl, headers=resume_header,
 296                          stream=True, verify=False, allow_redirects=True)
 297         content_length = int(r.headers.get('content-length'))
 298         with open(local_filename, 'ab') as f:
 299             for chunk in progress.bar(r.iter_content(chunk_size=65536),
 300                                       expected_size=(content_length / 65536) + 1):
 301                 if chunk:  # filter out keep-alive new chunks
 302                     f.write(chunk)
 303
 304     v = sha256_for_file(local_filename)
 305     if v == shasum:
 306         print("\t...shasum verified for " + local_filename)
 307     else:
 308         print("Invalid shasum of '" + v + "' detected for " + local_filename)
 309         sys.exit(1)
 310
 311 # use VirtualBox software virtualization if hardware is not available,
 312 # like if this is being run in kvm or some other VM platform, like
 313 # http://jenkins.debian.net, the values are 'on' or 'off'
 314 if sys.platform.startswith('darwin'):
 315     # all < 10 year old Macs work, and OSX servers as VM host are very
 316     # rare, but this could also be auto-detected if someone codes it
 317     config['hwvirtex'] = 'on'
 318 elif os.path.exists('/proc/cpuinfo'):
 319     with open('/proc/cpuinfo') as f:
 320         contents = f.read()
 321     if 'vmx' in contents or 'svm' in contents:
 322         config['hwvirtex'] = 'on'
 323
 324 # Check against the existing Vagrantfile.yaml, and if they differ, we
 325 # need to create a new box:
 326 vf = os.path.join(serverdir, 'Vagrantfile.yaml')
 327 writevf = True
 328 if os.path.exists(vf):
 329     vagrant(['halt'], serverdir)
 330     with open(vf, 'r', encoding='utf-8') as f:
 331         oldconfig = yaml.load(f)
 332     if config != oldconfig:
 333         print("Server configuration has changed, rebuild from scratch is required")
 334         vagrant(['destroy', '-f'], serverdir)
 335     else:
 336         print("Re-provisioning existing server")
 337         writevf = False
 338 else:
 339     print("No existing server - building from scratch")
 340 if writevf:
 341     with open(vf, 'w', encoding='utf-8') as f:
 342         yaml.dump(config, f)
 343
 344 print("Configuring build server VM")
 345 returncode, out = vagrant(['up', '--provision'], serverdir, printout=True)
 346 with open(os.path.join(serverdir, 'up.log'), 'w') as log:
 347     log.write(out)
 348 if returncode != 0:
 349     print("Failed to configure server")
 350     sys.exit(1)
 351
 352 print("Writing buildserver ID")
 353 p = subprocess.Popen(['git', 'rev-parse', 'HEAD'], stdout=subprocess.PIPE,
 354                      universal_newlines=True)
 355 buildserverid = p.communicate()[0].strip()
 356 print("...ID is " + buildserverid)
 357 subprocess.call(
 358     ['vagrant', 'ssh', '-c', 'sh -c "echo {0} >/home/vagrant/buildserverid"'
 359         .format(buildserverid)],
 360     cwd=serverdir)
 361
 362 print("Stopping build server VM")
 363 vagrant(['halt'], serverdir)
 364
 365 print("Waiting for build server VM to be finished")
 366 ready = False
 367 while not ready:
 368     time.sleep(2)
 369     returncode, out = vagrant(['status'], serverdir)
 370     if returncode != 0:
 371         print("Error while checking status")
 372         sys.exit(1)
 373     for line in out.splitlines():
 374         if line.startswith("default"):
 375             if line.find("poweroff") != -1:
 376                 ready = True
 377             else:
 378                 print("Status: " + line)
 379
 380 print("Packaging")
 381 vagrant(['package', '--output', os.path.join('..', boxfile)], serverdir,
 382         printout=options.verbose)
 383 print("Adding box")
 384 vagrant(['box', 'add', 'buildserver', boxfile, '-f'],
 385         printout=options.verbose)
 386
 387 os.remove(boxfile)