#!/usr/bin/python2.6 # # The Qubes OS Project, http://www.qubes-os.org # # Copyright (C) 2010 Joanna Rutkowska # Copyright (C) 2010 Rafal Wojtczuk # # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; either version 2 # of the License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. # # from qubes.qubes import QubesVmCollection from qubes.qubes import QubesException from optparse import OptionParser import subprocess import socket import errno import dbus import time qubes_guid_path = "/usr/bin/qubes_guid" qubes_clipd_path = "/usr/bin/qclipd" qubes_qfilexchgd_path= "/usr/bin/qfilexchgd" qrexec_daemon_path = "/usr/lib/qubes/qrexec_daemon" notify_object = None # how long (in sec) to wait for VMs to shutdown # before killing them (when used with --wait option) shutdown_counter_max = 30 def tray_notify(str, label, timeout = 3000): notify_object.Notify("Qubes", 0, label.icon, "Qubes", str, [], [], timeout, dbus_interface="org.freedesktop.Notifications") def tray_notify_error(str, timeout = 3000): notify_object.Notify("Qubes", 0, "dialog-error", "Qubes", str, [], [], timeout, dbus_interface="org.freedesktop.Notifications") def vm_run_cmd(vm, cmd, options): if options.shutdown: if options.verbose: print "Shutting down VM: '{0}'...".format(vm.name) subprocess.call (["/usr/sbin/xm", "shutdown", vm.name]) return if options.pause: if options.verbose: print "Pausing VM: '{0}'...".format(vm.name) subprocess.call (["/usr/sbin/xm", "pause", vm.name]) return if options.unpause: if options.verbose: print "UnPausing VM: '{0}'...".format(vm.name) subprocess.call (["/usr/sbin/xm", "unpause", vm.name]) return if options.verbose: print "Running command on VM: '{0}'...".format(vm.name) if not vm.is_running(): if not options.auto: print "VM '{0}' is not running, please start it first, or use the '--auto' switch" exit (1) try: if options.verbose: print "Starting the VM '{0}'...".format(vm.name) if options.tray: tray_notify ("Starting the '{0}' VM...".format(vm.name), label=vm.label) xid = vm.start(verbose=options.verbose) retcode = subprocess.call ([qrexec_daemon_path, str(xid)]) if (retcode != 0) : print "ERROR: Cannot start qrexec_daemon!" exit (1) except (IOError, OSError, QubesException) as err: print "ERROR: {0}".format(err) if options.tray: tray_notify_error ("Error while starting the '{0}' VM: {1}".format(vm.name, err)) exit (1) except (MemoryError) as err: print "ERROR: {0}".format(err) print "Close one or more running VMs and try again." if options.tray: subprocess.call(["kdialog", "--error", "Not enough memory to start '{0}' VM! Close one or more running VMs and try again.".format(vm.name)]) exit (1) if options.verbose: print "--> Starting Qubes GUId..." retcode = subprocess.call ([qubes_guid_path, "-d", str(xid), "-c", vm.label.color, "-e", cmd, "-i", vm.label.icon, "-l", str(vm.label.index)]) if (retcode != 0) : print "ERROR: Cannot start qubes_guid!" if options.tray: tray_notify_error ("ERROR: Cannot start qubes_guid!") exit (1) else: # VM already running... guid_is_running = True xid = vm.get_xid() s = socket.socket (socket.AF_UNIX) try: s.connect ("/var/run/qubes/cmd_socket.{0}".format(xid)) except (IOError, OSError) as e: if e.errno in [errno.ENOENT,errno.ECONNREFUSED]: guid_is_running = False else: print "ERROR: unix-connect: {0}".format(e) if options.tray: tray_notify_error ("ERROR: Cannot connect to GUI daemon for this VM!") exit(1) if guid_is_running: s.send (cmd) s.close() else: retcode = subprocess.call ([qubes_guid_path, "-d", str(xid), "-c", vm.label.color, "-e", cmd, "-i", vm.label.icon, "-l", str(vm.label.index)]) if (retcode != 0) : print "ERROR: Cannot start qubes_guid!" if options.tray: tray_notify_error ("ERROR: Cannot start the GUI daemon for this VM!") exit (1) def main(): usage = "usage: %prog [options] [] []" parser = OptionParser (usage) parser.add_option ("-q", "--quiet", action="store_false", dest="verbose", default=True) parser.add_option ("-a", "--auto", action="store_true", dest="auto", default=False, help="Auto start the VM if not running") parser.add_option ("-u", "--user", action="store", dest="user", default="user", help="Run command in a VM as a specified user") parser.add_option ("--tray", action="store_true", dest="tray", default=False, help="Use tray notifications instead of stdout" ) parser.add_option ("--all", action="store_true", dest="run_on_all_running", default=False, help="Run command on all currently running VMs (or all paused, in case of --unpause)") parser.add_option ("--exclude", action="append", dest="exclude_list", help="When --all is used: exclude this VM name (might be repeated)") parser.add_option ("--wait", action="store_true", dest="wait_for_shutdown", default=False, help="Wait for the VM(s) to shutdown") parser.add_option ("--shutdown", action="store_true", dest="shutdown", default=False, help="Do 'xm shutdown' for the VM(s) (can be combined this with --all and --wait)") parser.add_option ("--pause", action="store_true", dest="pause", default=False, help="Do 'xm pause' for the VM(s) (can be combined this with --all and --wait)") parser.add_option ("--unpause", action="store_true", dest="unpause", default=False, help="Do 'xm unpause' for the VM(s) (can be combined this with --all and --wait)") (options, args) = parser.parse_args () if (options.shutdown or options.pause or options.unpause): takes_cmd_argument = False else: takes_cmd_argument = True if options.run_on_all_running: if len(args) < 1 and takes_cmd_argument: parser.error ("You must provide a command to execute on all the VMs.") if len(args) > 1 or ((not takes_cmd_argument) and len(args) > 0): parser.error ("To many arguments...") cmdstr = args[0] if takes_cmd_argument else None else: if len (args) < 1 and not takes_cmd_argument: parser.error ("You must specify the VM name to shutdown/pause/unpause.") if len (args) < 2 and takes_cmd_argument: parser.error ("You must specify the VM name and the command to execute in the VM.") if len (args) > 2 or ((not takes_cmd_argument) and len(args) > 1): parser.error ("To many arguments...") vmname = args[0] cmdstr = args[1] if takes_cmd_argument else None if options.tray: global notify_object notify_object = dbus.SessionBus().get_object("org.freedesktop.Notifications", "/org/freedesktop/Notifications") qvm_collection = QubesVmCollection() qvm_collection.lock_db_for_reading() qvm_collection.load() qvm_collection.unlock_db() vms_list = [] if options.run_on_all_running: all_vms = [vm for vm in qvm_collection.values()] for vm in all_vms: if options.exclude_list is not None and vm.name in options.exclude_list: continue if vm.qid == 0: continue if (options.unpause and vm.is_paused()) or (not options.unpause and vm.is_running()): vms_list.append (vm) else: vm = qvm_collection.get_vm_by_name(vmname) if vm is None: print "A VM with the name '{0}' does not exist in the system!".format(vmname) exit(1) vms_list.append(vm) if takes_cmd_argument: cmd = "{user}:{cmd}".format(user=options.user, cmd=cmdstr) else: cmd = None for vm in vms_list: vm_run_cmd(vm, cmd, options) if options.wait_for_shutdown: if options.verbose: print "Waiting for the VM(s) to shutdown..." shutdown_counter = 0 while len (vms_list): if options.verbose: print "Waiting for VMs: ", [vm.name for vm in vms_list] for vm in vms_list: if not vm.is_running(): vms_list.remove (vm) if shutdown_counter > shutdown_counter_max: # kill the VM if options.verbose: print "Killing the (apparently hanging) VM '{0}'...".format(vm.name) vm.force_shutdown() #vms_list.remove(vm) shutdown_counter += 1 time.sleep (1) exit (0) # there is no point in executing the other daemons in the case of --wait retcode = subprocess.call([qubes_clipd_path]) if retcode != 0: print "ERROR: Cannot start qclipd!" if options.tray: tray_notify ("ERROR: Cannot start the Qubes Clipboard Notifier!") retcode = subprocess.call([qubes_qfilexchgd_path]) if retcode != 0: print "ERROR: Cannot start qfilexchgd!" if options.tray: tray_notify ("ERROR: Cannot start the Qubes Inter-VM File Exchange Daemon!") main()