123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227 |
- #!/usr/bin/python2.6
- #
- # The Qubes OS Project, http://www.qubes-os.org
- #
- # Copyright (C) 2010 Joanna Rutkowska <joanna@invisiblethingslab.com>
- # Copyright (C) 2010 Rafal Wojtczuk <rafal@invisiblethingslab.com>
- #
- # 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"
- 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.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)
- 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])
- 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])
- 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] [<vm-name>] [<cmd>]"
- 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")
- 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)")
- (options, args) = parser.parse_args ()
- if options.run_on_all_running:
- if len(args) < 1 and not options.shutdown:
- parser.error ("You must provide a command to execute on all the VMs.")
- if len(args) > 1 or (options.shutdown and len(args) > 0):
- parser.error ("To many arguments...")
- cmdstr = args[0] if not options.shutdown else None
- else:
- if len (args) < 1 and options.shutdown:
- parser.error ("You must specify the VM name to shutdown.")
- if len (args) < 2 and not options.shutdown:
- parser.error ("You must specify the VM name and the command to execute in the VM.")
- if len (args) > 2 or (options.shutdown and len(args) > 1):
- parser.error ("To many arguments...")
- vmname = args[0]
- cmdstr = args[1] if not options.shutdown 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 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 options.shutdown:
- cmd = None
- else:
- cmd = "{user}:{cmd}".format(user=options.user, cmd=cmdstr)
- 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()
|