tools/helpers/lxc.py

   1 # Copyright 2021 Erfan Abdi
   2 # SPDX-License-Identifier: GPL-3.0-or-later
   3 import subprocess
   4 import os
   5 import re
   6 import logging
   7 import glob
   8 import shutil
   9 import platform
  10 import tools.config
  11 import tools.helpers.run
  12
  13
  14 def get_lxc_version(args):
  15     if shutil.which("lxc-info") is not None:
  16         command = ["lxc-info", "--version"]
  17         version_str = tools.helpers.run.user(args, command, output_return=True)
  18         return int(version_str[0])
  19     else:
  20         return 0
  21
  22
  23 def generate_nodes_lxc_config(args):
  24     def make_entry(src, dist=None, mnt_type="none", options="bind,create=file,optional 0 0", check=True):
  25         if check and not os.path.exists(src):
  26             return False
  27         entry = "lxc.mount.entry = "
  28         entry += src + " "
  29         if dist is None:
  30             dist = src[1:]
  31         entry += dist + " "
  32         entry += mnt_type + " "
  33         entry += options
  34         nodes.append(entry)
  35         return True
  36
  37     nodes = []
  38     # Necessary dev nodes
  39     make_entry("tmpfs", "dev", "tmpfs", "nosuid 0 0", False)
  40     make_entry("/dev/zero")
  41     make_entry("/dev/full")
  42     make_entry("/dev/ashmem", check=False)
  43     make_entry("/dev/fuse")
  44     make_entry("/dev/ion")
  45     make_entry("/dev/char", options="bind,create=dir,optional 0 0")
  46
  47     # Graphic dev nodes
  48     make_entry("/dev/kgsl-3d0")
  49     make_entry("/dev/mali0")
  50     make_entry("/dev/pvr_sync")
  51     make_entry("/dev/pmsg0")
  52     make_entry("/dev/fb0")
  53     make_entry("/dev/graphics/fb0")
  54     make_entry("/dev/fb1")
  55     make_entry("/dev/graphics/fb1")
  56     make_entry("/dev/fb2")
  57     make_entry("/dev/graphics/fb2")
  58     make_entry("/dev/dxg")
  59     make_entry("/dev/dri", options="bind,create=dir,optional 0 0")
  60
  61     # Binder dev nodes
  62     make_entry("/dev/" + args.BINDER_DRIVER, "dev/binder", check=False)
  63     make_entry("/dev/" + args.VNDBINDER_DRIVER, "dev/vndbinder", check=False)
  64     make_entry("/dev/" + args.HWBINDER_DRIVER, "dev/hwbinder", check=False)
  65
  66     if args.vendor_type != "MAINLINE":
  67         if not make_entry("/dev/hwbinder", "dev/host_hwbinder"):
  68             raise OSError('Binder node "hwbinder" of host not found')
  69         make_entry("/vendor", "vendor_extra", options="bind,optional 0 0")
  70
  71     # Necessary device nodes for adb
  72     make_entry("none", "dev/pts", "devpts", "defaults,mode=644,ptmxmode=666,create=dir 0 0", False)
  73     make_entry("/dev/uhid")
  74
  75     # Low memory killer sys node
  76     make_entry("/sys/module/lowmemorykiller", options="bind,create=dir,optional 0 0")
  77
  78     # Mount /data
  79     make_entry("tmpfs", "mnt", "tmpfs", "mode=0755,uid=0,gid=1000", False)
  80     make_entry(tools.config.defaults["data"], "data", options="bind 0 0", check=False)
  81
  82     # Mount host permissions
  83     make_entry(tools.config.defaults["host_perms"],
  84                "vendor/etc/host-permissions", options="bind,optional 0 0")
  85
  86     # Recursive mount /run to provide necessary host sockets
  87     make_entry("/run", options="rbind,create=dir 0 0")
  88
  89     # Necessary sw_sync node for HWC
  90     make_entry("/dev/sw_sync")
  91     make_entry("/sys/kernel/debug", options="rbind,create=dir,optional 0 0")
  92
  93     # Media dev nodes (for Mediatek)
  94     make_entry("/dev/Vcodec")
  95     make_entry("/dev/MTK_SMI")
  96     make_entry("/dev/mdp_sync")
  97     make_entry("/dev/mtk_cmdq")
  98
  99     # Media dev nodes (for Qcom)
 100     make_entry("/dev/video32")
 101     make_entry("/dev/video33")
 102
 103     # WSLg
 104     make_entry("tmpfs", "mnt_extra", "tmpfs", "nodev 0 0", False)
 105     make_entry("/mnt/wslg", "mnt_extra/wslg",
 106                options="rbind,create=dir,optional 0 0")
 107
 108     # var
 109     make_entry("tmpfs", "var", "tmpfs", "nodev 0 0", False)
 110     make_entry("/var/run", options="rbind,create=dir,optional 0 0")
 111
 112     return nodes
 113
 114
 115 def set_lxc_config(args):
 116     lxc_path = tools.config.defaults["lxc"] + "/waydroid"
 117     config_file = "config_2"
 118     lxc_ver = get_lxc_version(args)
 119     if lxc_ver == 0:
 120         raise OSError("LXC is not installed")
 121     elif lxc_ver <= 2:
 122         config_file = "config_1"
 123     config_path = tools.config.tools_src + "/data/configs/" + config_file
 124
 125     command = ["mkdir", "-p", lxc_path]
 126     tools.helpers.run.user(args, command)
 127     command = ["cp", "-fpr", config_path, lxc_path + "/config"]
 128     tools.helpers.run.user(args, command)
 129     command = ["sed", "-i", "s/LXCARCH/{}/".format(platform.machine()), lxc_path + "/config"]
 130     tools.helpers.run.user(args, command)
 131
 132     nodes = generate_nodes_lxc_config(args)
 133     config_nodes_tmp_path = args.work + "/config_nodes"
 134     config_nodes = open(config_nodes_tmp_path, "w")
 135     for node in nodes:
 136         config_nodes.write(node + "\n")
 137     config_nodes.close()
 138     command = ["mv", config_nodes_tmp_path, lxc_path]
 139     tools.helpers.run.user(args, command)
 140
 141
 142 def make_base_props(args):
 143     def find_hal(hardware):
 144         hardware_props = [
 145             "ro.hardware." + hardware,
 146             "ro.hardware",
 147             "ro.product.board",
 148             "ro.arch",
 149             "ro.board.platform"]
 150         for p in hardware_props:
 151             prop = tools.helpers.props.host_get(args, p)
 152             hal_prop = ""
 153             if prop != "":
 154                 for lib in ["lib", "lib64"]:
 155                     hal_file = "/vendor/" + lib + "/hw/" + hardware + "." + prop + ".so"
 156                     command = ["readlink", "-f", hal_file]
 157                     hal_file_path = tools.helpers.run.user(args, command, output_return=True).strip()
 158                     if os.path.isfile(hal_file_path):
 159                         hal_prop = re.sub(".*" + hardware + ".", "", hal_file_path)
 160                         hal_prop = re.sub(".so", "", hal_prop)
 161                         if hal_prop != "":
 162                             return hal_prop
 163             if hal_prop != "":
 164                 return hal_prop
 165         return ""
 166
 167     props = []
 168     egl = tools.helpers.props.host_get(args, "ro.hardware.egl")
 169
 170     gralloc = find_hal("gralloc")
 171     if gralloc == "":
 172         if os.path.exists("/dev/dri"):
 173             gralloc = "gbm"
 174             egl = "mesa"
 175         else:
 176             gralloc = "default"
 177             egl = "swiftshader"
 178         props.append("debug.stagefright.ccodec=0")
 179     props.append("ro.hardware.gralloc=" + gralloc)
 180
 181     if egl != "":
 182         props.append("ro.hardware.egl=" + egl)
 183
 184     media_profiles = tools.helpers.props.host_get(args, "media.settings.xml")
 185     if media_profiles != "":
 186         media_profiles = media_profiles.replace("vendor/", "vendor_extra/")
 187         media_profiles = media_profiles.replace("odm/", "odm_extra/")
 188         props.append("media.settings.xml=" + media_profiles)
 189
 190     ccodec = tools.helpers.props.host_get(args, "debug.stagefright.ccodec")
 191     if ccodec != "":
 192         props.append("debug.stagefright.ccodec=" + ccodec)
 193
 194     ext_library = tools.helpers.props.host_get(args, "ro.vendor.extension_library")
 195     if ext_library != "":
 196         ext_library = ext_library.replace("vendor/", "vendor_extra/")
 197         ext_library = ext_library.replace("odm/", "odm_extra/")
 198         props.append("ro.vendor.extension_library=" + ext_library)
 199
 200     vulkan = find_hal("vulkan")
 201     if vulkan != "":
 202         props.append("ro.hardware.vulkan=" + vulkan)
 203
 204     opengles = tools.helpers.props.host_get(args, "ro.opengles.version")
 205     if opengles == "":
 206         opengles = "196608"
 207     props.append("ro.opengles.version=" + opengles)
 208
 209     props.append("waydroid.system_ota=" + args.system_ota)
 210     props.append("waydroid.vendor_ota=" + args.vendor_ota)
 211     props.append("waydroid.tools_version=" + tools.config.version)
 212
 213     base_props = open(args.work + "/waydroid_base.prop", "w")
 214     for prop in props:
 215         base_props.write(prop + "\n")
 216     base_props.close()
 217
 218
 219 def setup_host_perms(args):
 220     sku = tools.helpers.props.host_get(args, "ro.boot.product.hardware.sku")
 221     copy_list = []
 222     copy_list.extend(
 223         glob.glob("/vendor/etc/permissions/android.hardware.nfc.*"))
 224     if os.path.exists("/vendor/etc/permissions/android.hardware.consumerir.xml"):
 225         copy_list.append("/vendor/etc/permissions/android.hardware.consumerir.xml")
 226     copy_list.extend(
 227         glob.glob("/odm/etc/permissions/android.hardware.nfc.*"))
 228     if os.path.exists("/odm/etc/permissions/android.hardware.consumerir.xml"):
 229         copy_list.append("/odm/etc/permissions/android.hardware.consumerir.xml")
 230     if sku != "":
 231         copy_list.extend(
 232             glob.glob("/odm/etc/permissions/sku_{}/android.hardware.nfc.*".format(sku)))
 233         if os.path.exists("/odm/etc/permissions/sku_{}/android.hardware.consumerir.xml".format(sku)):
 234             copy_list.append(
 235                 "/odm/etc/permissions/sku_{}/android.hardware.consumerir.xml".format(sku))
 236
 237     if not os.path.exists(tools.config.defaults["host_perms"]):
 238         os.mkdir(tools.config.defaults["host_perms"])
 239
 240     for filename in copy_list:
 241         shutil.copy(filename, tools.config.defaults["host_perms"])
 242
 243 def status(args):
 244     command = ["lxc-info", "-P", tools.config.defaults["lxc"], "-n", "waydroid", "-sH"]
 245     return subprocess.run(command, stdout=subprocess.PIPE).stdout.decode('utf-8').strip()
 246
 247 def start(args):
 248     command = ["lxc-start", "-P", tools.config.defaults["lxc"],
 249                "-F", "-n", "waydroid", "--", "/init"]
 250     tools.helpers.run.user(args, command, output="background")
 251
 252 def stop(args):
 253     command = ["lxc-stop", "-P",
 254                tools.config.defaults["lxc"], "-n", "waydroid", "-k"]
 255     tools.helpers.run.user(args, command)
 256
 257 def freeze(args):
 258     command = ["lxc-freeze", "-P", tools.config.defaults["lxc"], "-n", "waydroid"]
 259     tools.helpers.run.user(args, command)
 260
 261 def unfreeze(args):
 262     command = ["lxc-unfreeze", "-P",
 263                tools.config.defaults["lxc"], "-n", "waydroid"]
 264     tools.helpers.run.user(args, command)
 265
 266 def shell(args):
 267     if status(args) != "RUNNING":
 268         logging.error("WayDroid container is {}".format(status(args)))
 269         return
 270     command = ["lxc-attach", "-P", tools.config.defaults["lxc"],
 271                "-n", "waydroid", "--"]
 272     if args.COMMAND:
 273         command.append(args.COMMAND)
 274     else:
 275         command.append("/system/bin/sh")
 276     subprocess.run(command, env={"PATH": os.environ['PATH'] + "/system/bin:/vendor/bin"})
 277
 278 def logcat(args):
 279     if status(args) != "RUNNING":
 280         logging.error("WayDroid container is {}".format(status(args)))
 281         return
 282     command = ["lxc-attach", "-P", tools.config.defaults["lxc"],
 283                "-n", "waydroid", "--", "/system/bin/logcat"]
 284     subprocess.run(command)