lava/rootfs/vmdeboostrap/generate-root.py

   1 #!/usr/bin/python3
   2 # Copyright (C) 2018 - Jonathan Rajotte-Julien <jonathan.rajotte-julien@efficios.com>
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU General Public License as published by
   6 # the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 import argparse
  18 import gzip
  19 import os
  20 import shutil
  21 import subprocess
  22
  23 from datetime import datetime
  24
  25
  26 def compress(filename):
  27     with open(filename, 'rb') as f_in:
  28         with gzip.open('{}.gz'.format(filename), 'wb') as f_out:
  29             shutil.copyfileobj(f_in, f_out)
  30     os.remove(filename)
  31
  32
  33 packages = [
  34     'autoconf',
  35     'automake',
  36     'bash-completion',
  37     'bison',
  38     'bsdtar',
  39     'build-essential',
  40     'chrpath',
  41     'clang',
  42     'cloc',
  43     'cppcheck',
  44     'curl',
  45     'elfutils',
  46     'flex',
  47     'gettext',
  48     'git',
  49     'htop',
  50     'jq',
  51     'libdw-dev',
  52     'libelf-dev',
  53     'libffi-dev',
  54     'libglib2.0-dev',
  55     'libmount-dev',
  56     'libnuma-dev',
  57     'libpfm4-dev',
  58     'libpopt-dev',
  59     'libtap-harness-archive-perl',
  60     'libtool',
  61     'libxml2',
  62     'libxml2-dev',
  63     'netcat-traditional',
  64     'openssh-server',
  65     'psmisc',
  66     'python-virtualenv',
  67     'python3',
  68     'python3-dev',
  69     'python3-numpy',
  70     'python3-pandas',
  71     'python3-pip',
  72     'python3-setuptools',
  73     'python3-sphinx',
  74     'stress',
  75     'swig',
  76     'texinfo',
  77     'tree',
  78     'uuid-dev',
  79     'vim',
  80     'wget',
  81 ]
  82
  83
  84 def main():
  85     parser = argparse.ArgumentParser(description='Generate lava lttng rootfs')
  86     parser.add_argument("--arch", default='amd64')
  87     parser.add_argument("--distribution", default='bionic')
  88     parser.add_argument("--mirror", default='http://archive.ubuntu.com/ubuntu')
  89     parser.add_argument(
  90         "--component", default='universe,multiverse,main,restricted')
  91     args = parser.parse_args()
  92
  93     name = "rootfs_{}_{}_{}.tar".format(args.arch, args.distribution,
  94                                         datetime.now().strftime("%Y-%m-%d"))
  95
  96     hostname = "linaro-server"
  97     user = "linaro/linaro"
  98     root_password = "root"
  99     print(name)
 100     command = [
 101         "sudo",
 102         "vmdebootstrap",
 103         "--arch={}".format(args.arch),
 104         "--distribution={}".format(args.distribution),
 105         "--mirror={}".format(args.mirror),
 106         "--debootstrapopts=components={}".format(args.component),
 107         "--tarball={}".format(name),
 108         "--package={}".format(",".join(packages)),
 109         "--hostname={}".format(hostname),
 110         "--user={}".format(user),
 111         "--root-password={}".format(root_password),
 112         "--no-kernel",
 113         "--verbose",
 114     ]
 115
 116     completed_command = subprocess.run(command, check=True)
 117
 118     compress(name)
 119
 120
 121 if __name__ == "__main__":
 122     main()