def _process_command(config, msg):
try:
command = json.loads(msg)
except json.decoder.JSONDecodeError:
print("Invalid command \"%s\" ! (JSON decode error)" % msg)
return
try:
if command["type"] == "switch":
print("Writing requested GPU mode %s" % command["args"]["mode"])
var.write_requested_mode(command["args"]["mode"])
elif command["type"] == "startup":
print("Writing startup mode %s" % command["args"]["mode"])
var.write_startup_mode(command["args"]["mode"])
elif command["type"] == "temp_config":
print("Writing temporary config file path %s" %
command["args"]["path"])
var.write_temp_conf_path_var(command["args"]["path"])
else:
print("Invalid command \"%s\" ! Unknown type %s" %
(msg, command["type"]))
except KeyError as e:
print("Invalid command \"%s\" ! Key error : %s" % (msg, str(e)))
def set_startup(mode):
try:
write_startup_mode(mode)
except VarError as e:
print("Cannot write startup mode : %s" % str(e))
def main():
# Arguments parsing
parser = argparse.ArgumentParser(
description=
"Display Manager setup service for the Optimus Manager tool.\n"
"https://github.com/Askannz/optimus-manager")
parser.add_argument('--setup-start',
action='store_true',
help='Setup Optimus before the login manager starts.')
parser.add_argument('--setup-stop',
action='store_true',
help='Cleanup Optimus after the login manager stops.')
args = parser.parse_args()
print("Optimus Manager (DM setup) version %s" % envs.VERSION)
# Config
try:
config = load_config()
except ConfigError as e:
print("Error loading config file : %s" % str(e))
sys.exit(1)
if args.setup_start:
print("Setting up Optimus configuration")
# Cleanup
clean_autogenerated()
try:
requested_mode = read_requested_mode()
except VarError as e:
print(
"Cannot read requested mode : %s.\nUsing startup mode instead."
% str(e))
try:
startup_mode = read_startup_mode()
except VarError as e:
print(
"Cannot read startup mode : %s.\nUsing default startup mode %s instead."
% (str(e), envs.DEFAULT_STARTUP_MODE))
startup_mode = envs.DEFAULT_STARTUP_MODE
print("Startup mode :", startup_mode)
if startup_mode == "nvidia_once":
requested_mode = "nvidia"
write_startup_mode("intel")
else:
requested_mode = startup_mode
# We are done reading the command
remove_requested_mode_var()
print("Requested mode :", requested_mode)
try:
if requested_mode == "nvidia":
switch_to_nvidia(config)
elif requested_mode == "intel":
switch_to_intel(config)
except SwitchError as e:
print("Cannot switch GPUS : %s" % str(e))
sys.exit(0)
elif args.setup_stop:
print("Cleaning up Optimus configuration")
clean_autogenerated()
#
print("Terminating X11 sessions")
try:
terminate_current_x11_sessions()
except SessionError as e:
print("Error terminating sessions : %s" % str(e))
#
print("Killing remaining X11 servers")
try:
kill_current_xorg_servers()
except XorgError as e:
print("Error killing X servers : %s" % str(e))
#
# There is a known bug causing systemd-logind to keep ownership of the GPU
# and prevents module unloading
print("Killing systemd-logind")
try:
exec_bash("pkill systemd-logind")
except BashError:
pass
print("Unloading kernel modules")
try:
exec_bash(
"modprobe -r nvidia_drm nvidia_modeset nvidia_uvm nvidia nouveau"
)
except BashError as e:
print("Cannot unload modules : %s" % str(e))
sys.exit(1)
# Reset the PCI device corresponding to the Nvidia GPU
if config["optimus"]["pci_reset"] == "yes":
if config["optimus"]["switching"] == "bbswitch":
print("bbswitch is enabled, pci_reset option ignored.")
else:
print("Resetting the GPU")
try:
pci.reset_gpu()
except pci.PCIError as e:
print("Error resetting the PCI device : %s" % str(e))
else:
print("Invalid argument")