tools/actions/container_manager.py

   1 # Copyright 2021 Erfan Abdi
   2 # SPDX-License-Identifier: GPL-3.0-or-later
   3 from shutil import which
   4 import logging
   5 import os
   6 import time
   7 import glob
   8 import signal
   9 import sys
  10 import tools.config
  11 from tools import helpers
  12 from tools import services
  13
  14
  15 def start(args):
  16     def make_prop(full_props_path):
  17         def add_prop(key, cfg_key):
  18             value = session_cfg["session"][cfg_key]
  19             if value != "None":
  20                 props.append(key + "=" + session_cfg["session"][cfg_key])
  21
  22         if not os.path.isfile(args.work + "/waydroid_base.prop"):
  23             raise RuntimeError("waydroid_base.prop Not found")
  24         with open(args.work + "/waydroid_base.prop") as f:
  25             props = f.read().splitlines()
  26         if not props:
  27             raise RuntimeError("waydroid_base.prop is broken!!?")
  28
  29         add_prop("waydroid.host.user", "user_name")
  30         add_prop("waydroid.host.uid", "user_id")
  31         add_prop("waydroid.host.gid", "group_id")
  32         add_prop("waydroid.xdg_runtime_dir", "xdg_runtime_dir")
  33         add_prop("waydroid.pulse_runtime_path", "pulse_runtime_path")
  34         add_prop("waydroid.wayland_display", "wayland_display")
  35         if which("waydroid-sensord") is None:
  36             props.append("waydroid.stub_sensors_hal=1")
  37         dpi = session_cfg["session"]["lcd_density"]
  38         if dpi != "0":
  39             props.append("ro.sf.lcd_density=" + dpi)
  40
  41         final_props = open(full_props_path, "w")
  42         for prop in props:
  43             final_props.write(prop + "\n")
  44         final_props.close()
  45         os.chmod(full_props_path, 0o644)
  46
  47     def set_permissions(perm_list=None, mode="777"):
  48         def chmod(path, mode):
  49             if os.path.exists(path):
  50                 command = ["chmod", mode, "-R", path]
  51                 tools.helpers.run.root(args, command, check=False)
  52
  53         # Nodes list
  54         if not perm_list:
  55             perm_list = [
  56                 "/dev/ashmem",
  57
  58                 # sw_sync for HWC
  59                 "/dev/sw_sync",
  60                 "/sys/kernel/debug/sync/sw_sync",
  61
  62                 # Media
  63                 "/dev/Vcodec",
  64                 "/dev/MTK_SMI",
  65                 "/dev/mdp_sync",
  66                 "/dev/mtk_cmdq",
  67                 "/dev/video32",
  68                 "/dev/video33",
  69
  70                 # Graphics
  71                 "/dev/dri",
  72                 "/dev/graphics",
  73
  74                 # Wayland and pulse socket permissions
  75                 session_cfg["session"]["pulse_runtime_path"],
  76                 session_cfg["session"]["xdg_runtime_dir"]
  77             ]
  78
  79             # Framebuffers
  80             perm_list.extend(glob.glob("/dev/fb*"))
  81
  82         for path in perm_list:
  83             chmod(path, mode)
  84
  85     def signal_handler(sig, frame):
  86         services.hardware_manager.stop(args)
  87         stop(args)
  88         sys.exit(0)
  89
  90     status = helpers.lxc.status(args)
  91     if status == "STOPPED":
  92         # Load binder and ashmem drivers
  93         cfg = tools.config.load(args)
  94         if cfg["waydroid"]["vendor_type"] == "MAINLINE":
  95             if helpers.drivers.probeBinderDriver(args) != 0:
  96                 logging.error("Failed to load Binder driver")
  97             if helpers.drivers.probeAshmemDriver(args) != 0:
  98                 logging.error("Failed to load Ashmem driver")
  99         helpers.drivers.loadBinderNodes(args)
 100         set_permissions([
 101             "/dev/" + args.BINDER_DRIVER,
 102             "/dev/" + args.VNDBINDER_DRIVER,
 103             "/dev/" + args.HWBINDER_DRIVER
 104         ], "666")
 105
 106         if os.path.exists(tools.config.session_defaults["config_path"]):
 107             session_cfg = tools.config.load_session()
 108             if session_cfg["session"]["state"] != "STOPPED":
 109                 logging.warning("Found session config on state: {}, restart session".format(
 110                     session_cfg["session"]["state"]))
 111                 os.remove(tools.config.session_defaults["config_path"])
 112         logging.debug("Container manager is waiting for session to load")
 113         while not os.path.exists(tools.config.session_defaults["config_path"]):
 114             time.sleep(1)
 115
 116         # Load session configs
 117         session_cfg = tools.config.load_session()
 118
 119         # Generate props
 120         make_prop(args.work + "/waydroid.prop")
 121
 122         # Networking
 123         command = [tools.config.tools_src +
 124                    "/data/scripts/waydroid-net.sh", "start"]
 125         tools.helpers.run.root(args, command, check=False)
 126
 127         # Sensors
 128         tools.helpers.run.root(
 129             args, ["waydroid-sensord", "/dev/" + args.HWBINDER_DRIVER], output="background")
 130
 131         # Mount rootfs
 132         helpers.images.mount_rootfs(args, cfg["waydroid"]["images_path"])
 133
 134         # Mount data
 135         helpers.mount.bind(args, session_cfg["session"]["waydroid_data"],
 136                            tools.config.defaults["data"])
 137
 138         # Cgroup hacks
 139         if which("start"):
 140             command = ["start", "cgroup-lite"]
 141             tools.helpers.run.root(args, command, check=False)
 142         helpers.mount.umount_all(args, "/sys/fs/cgroup/schedtune")
 143
 144         #TODO: remove NFC hacks
 145         if which("stop"):
 146             command = ["stop", "nfcd"]
 147             tools.helpers.run.root(args, command, check=False)
 148
 149         # Set permissions
 150         set_permissions()
 151
 152         helpers.lxc.start(args)
 153         session_cfg["session"]["state"] = helpers.lxc.status(args)
 154         tools.config.save_session(session_cfg)
 155
 156         if not hasattr(args, 'hardwareLoop'):
 157             services.hardware_manager.start(args)
 158
 159         signal.signal(signal.SIGINT, signal_handler)
 160         while os.path.exists(tools.config.session_defaults["config_path"]):
 161             session_cfg = tools.config.load_session()
 162             if session_cfg["session"]["state"] == "STOPPED":
 163                 services.hardware_manager.stop(args)
 164                 sys.exit(0)
 165             elif session_cfg["session"]["state"] == "UNFREEZE":
 166                 session_cfg["session"]["state"] = helpers.lxc.status(args)
 167                 tools.config.save_session(session_cfg)
 168                 unfreeze(args)
 169             time.sleep(1)
 170
 171         logging.warning("session manager stopped, stopping container and waiting...")
 172         stop(args)
 173         start(args)
 174     else:
 175         logging.error("WayDroid container is {}".format(status))
 176
 177 def stop(args):
 178     status = helpers.lxc.status(args)
 179     if status != "STOPPED":
 180         helpers.lxc.stop(args)
 181         if os.path.exists(tools.config.session_defaults["config_path"]):
 182             session_cfg = tools.config.load_session()
 183             session_cfg["session"]["state"] = helpers.lxc.status(args)
 184             tools.config.save_session(session_cfg)
 185
 186         # Networking
 187         command = [tools.config.tools_src +
 188                    "/data/scripts/waydroid-net.sh", "stop"]
 189         tools.helpers.run.root(args, command, check=False)
 190
 191         #TODO: remove NFC hacks
 192         if which("start"):
 193             command = ["start", "nfcd"]
 194             tools.helpers.run.root(args, command, check=False)
 195
 196         # Sensors
 197         if which("waydroid-sensord"):
 198             command = ["pidof", "waydroid-sensord"]
 199             pid = tools.helpers.run.root(args, command, check=False, output_return=True)
 200             if pid:
 201                 command = ["killall", pid]
 202                 tools.helpers.run.root(args, command, check=False)
 203
 204     else:
 205         logging.error("WayDroid container is {}".format(status))
 206
 207 def freeze(args):
 208     status = helpers.lxc.status(args)
 209     if status == "RUNNING":
 210         helpers.lxc.freeze(args)
 211         if os.path.exists(tools.config.session_defaults["config_path"]):
 212             session_cfg = tools.config.load_session()
 213             session_cfg["session"]["state"] = helpers.lxc.status(args)
 214             tools.config.save_session(session_cfg)
 215     else:
 216         logging.error("WayDroid container is {}".format(status))
 217
 218 def unfreeze(args):
 219     status = helpers.lxc.status(args)
 220     if status == "FROZEN":
 221         helpers.lxc.unfreeze(args)
 222         if os.path.exists(tools.config.session_defaults["config_path"]):
 223             session_cfg = tools.config.load_session()
 224             session_cfg["session"]["state"] = helpers.lxc.status(args)
 225             tools.config.save_session(session_cfg)
 226     else:
 227         logging.error("WayDroid container is {}".format(status))