#!/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 from qubes import qubesutils import subprocess import socket import errno import dbus import time import sys import os import os.path qubes_clipd_path = "/usr/bin/qclipd" notify_object = None # how long (in sec) to wait for VMs to shutdown # before killing them (when used with --wait option) from qubes.qubes import shutdown_counter_max 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 tray_notify_generic(level, str): if level == "info": tray_notify(str) elif level == "error": tray_notify_error(str) def vm_run_cmd(vm, cmd, options): if options.shutdown: if options.verbose: print >> sys.stderr, "Shutting down VM: '{0}'...".format(vm.name) try: vm.shutdown(force=options.force) except (QubesException) as err: # "There are other VMs connected to this VM:" print >> sys.stderr, "ERROR: {0}".format(err) if str(err).startswith("There are other VMs connected"): print >> sys.stderr, "Shutdown them first or use --force switch" exit(1) return if options.pause: if options.verbose: print >> sys.stderr, "Pausing VM: '{0}'...".format(vm.name) subprocess.call (["/usr/sbin/xl", "pause", vm.name]) return if options.unpause: if options.verbose: print >> sys.stderr, "UnPausing VM: '{0}'...".format(vm.name) subprocess.call (["/usr/sbin/xl", "unpause", vm.name]) return if options.verbose: print >> sys.stderr, "Running command on VM: '{0}'...".format(vm.name) try: return qubesutils.run_in_vm(vm, cmd, autostart = options.auto, verbose = options.verbose, notify_function = tray_notify_generic if options.tray else None, passio = options.passio, localcmd = options.localcmd) except QubesException as err: print >> sys.stderr, "ERROR: %s" % str(err) 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 'xl 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 'xl 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 'xl unpause' for the VM(s) (can be combined this with --all and --wait)") parser.add_option ("-p", "--pass_io", action="store_true", dest="passio", default=False, help="Pass stdin/stdout/stderr from remote program") parser.add_option ("--localcmd", action="store", dest="localcmd", default=None, help="With --pass_io, pass stdin/stdout/stderr to the given program") parser.add_option ("--force", action="store_true", dest="force", default=False, help="Force operation, even if may damage other VMs (eg shutdown of NetVM)") (options, args) = parser.parse_args () if options.passio: options.verbose = False 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) # disable options incompatible with --all options.passio = False else: vm = qvm_collection.get_vm_by_name(vmname) if vm is None: print >> sys.stderr, "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 >> sys.stderr, "Waiting for the VM(s) to shutdown..." shutdown_counter = 0 while len (vms_list): if options.verbose: print >> sys.stderr, "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 >> sys.stderr, "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 main()