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 import sys
  23
  24 from datetime import datetime
  25
  26
  27 def compress(filename):
  28     command = [
  29         'tar', '-c', '-z',
  30         '-f', filename + ".tar.gz",
  31         '-C', filename,
  32         './'
  33     ]
  34     subprocess.run(command, check=True)
  35     shutil.rmtree(filename)
  36
  37
  38 packages = [
  39     'autoconf',
  40     'automake',
  41     'bash-completion',
  42     'bison',
  43     'build-essential',
  44     'chrpath',
  45     'clang',
  46     'cloc',
  47     'curl',
  48     'elfutils',
  49     'flex',
  50     'gettext',
  51     'git',
  52     'htop',
  53     'jq',
  54     'libarchive-tools',
  55     'libdw-dev',
  56     'libelf-dev',
  57     'libffi-dev',
  58     'libglib2.0-dev',
  59     'libmount-dev',
  60     'libnuma-dev',
  61     'libpfm4-dev',
  62     'libpopt-dev',
  63     'libtap-harness-archive-perl',
  64     'libtool',
  65     'libxml2',
  66     'libxml2-dev',
  67     'netcat-traditional',
  68     'openssh-server',
  69     'psmisc',
  70     'python3-virtualenv',
  71     'python3',
  72     'python3-dev',
  73     'python3-numpy',
  74     'python3-pandas',
  75     'python3-pip',
  76     'python3-setuptools',
  77     'python3-sphinx',
  78     'stress',
  79     'swig',
  80     'texinfo',
  81     'tree',
  82     'uuid-dev',
  83     'vim',
  84     'wget',
  85 ]
  86
  87
  88 def main():
  89     parser = argparse.ArgumentParser(description='Generate lava lttng rootfs')
  90     parser.add_argument("--arch", default='amd64')
  91     parser.add_argument("--distribution", default='jammy')
  92     parser.add_argument("--mirror", default='http://archive.ubuntu.com/ubuntu')
  93     parser.add_argument(
  94         "--component", default='universe,multiverse,main,restricted')
  95     args = parser.parse_args()
  96
  97     name = "rootfs_{}_{}_{}".format(args.arch, args.distribution,
  98                                         datetime.now().strftime("%Y-%m-%d"))
  99
 100     hostname = "linaro-server"
 101     user = "linaro/linaro"
 102     root_password = "root"
 103     print(name)
 104     command = [
 105         "debootstrap",
 106         "--arch={}".format(args.arch),
 107         "--components={}".format(args.component),
 108         "--verbose",
 109         args.distribution,  # SUITE
 110         name,  # TARGET (directory is created)
 111         args.mirror,  # MIRROR
 112     ]
 113     completed_command = subprocess.run(command, check=True)
 114
 115     # packages
 116     command = [
 117         'chroot', name,
 118         'apt-get', 'install', '-y', ] + packages
 119     completed_command = subprocess.run(command, check=True)
 120
 121     # hostname
 122     with open(os.path.join(name, 'etc', 'hostname'), 'w', encoding='utf-8') as f:
 123         f.write(hostname + "\n")
 124
 125     # user
 126     command = [
 127         'chroot', name,
 128         'adduser', '--gecos', '', '--disabled-password', 'linaro',
 129     ]
 130     completed_process = subprocess.run(command, check=True)
 131
 132     command = [
 133         'chroot', name, 'chpasswd',
 134         ]
 135     process = subprocess.Popen(command, stdin=subprocess.PIPE, text=True)
 136     process.communicate(input='linaro:linaro')
 137
 138     # root password
 139     process = subprocess.Popen(command, stdin=subprocess.PIPE, text=True)
 140     process.communicate(input="root:root")
 141
 142     compress(name)
 143
 144
 145 if __name__ == "__main__":
 146     if os.getuid() != 0:
 147         print("This script should be run as root: this is required by deboostrap", file=sys.stderr)
 148         sys.exit(1)
 149     main()