Qubes/core-admin
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

dom0/qvm-shutdown: add support of shutting down multiple VMs (#396)

Browse Source 
Just copy code from qvm-run.
This commit is contained in:
Marek Marczykowski 2012-01-16 03:25:52 +01:00
parent ba29b3c7ba
commit cfd22f8887
1 changed files with 52 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
dom0/qvm-tools/qvm-shutdown
View File

@ -35,24 +35,53 @@ def main():
help="Force operation, even if may damage other VMs (eg shutdown of NetVM)") help="Force operation, even if may damage other VMs (eg shutdown of NetVM)")
parser.add_option ("--wait", action="store_true", dest="wait_for_shutdown", default=False, parser.add_option ("--wait", action="store_true", dest="wait_for_shutdown", default=False,
help="Wait for the VM(s) to shutdown") help="Wait for the VM(s) to shutdown")
parser.add_option ("--all", action="store_true", dest="shutdown_all", default=False,
help="Shutdown all running VMs")
parser.add_option ("--exclude", action="append", dest="exclude_list",
help="When --all is used: exclude this VM name (might be repeated)")
(options, args) = parser.parse_args () (options, args) = parser.parse_args ()
if (len (args) != 1): if not options.shutdown_all and (len (args) != 1):
parser.error ("You must specify VM name!") parser.error ("You must specify VM name!")
vmname = args[0]
qvm_collection = QubesVmCollection() qvm_collection = QubesVmCollection()
qvm_collection.lock_db_for_reading() qvm_collection.lock_db_for_reading()
qvm_collection.load() qvm_collection.load()
qvm_collection.unlock_db() qvm_collection.unlock_db()
vms_list = []
if options.shutdown_all:
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:
vmname = args[0]
vm = qvm_collection.get_vm_by_name(vmname) vm = qvm_collection.get_vm_by_name(vmname)
if vm is None: if vm is None:
print >> sys.stderr, "A VM with the name '{0}' does not exist in the system.".format(vmname) print >> sys.stderr, "A VM with the name '{0}' does not exist in the system!".format(vmname)
exit(1)
vms_list.append(vm)
# Check VMs network dependencies
if not options.force:
for vm in vms_list:
if vm.is_netvm():
for connected_vm in vm.connected_vms.values():
if connected_vm.is_running() and not connected_vm in vms_list:
print >> sys.stderr, "ERROR: There are other VMs connected to VM '%s'" % vm.name
exit(1) exit(1)
for vm in vms_list:
try: try:
vm.shutdown(force=options.force) if options.verbose:
print >> sys.stderr, "Shutting down VM: '{0}'...".format(vm.name)
# Dependencies already checked above
vm.shutdown(force=True)
except (IOError, OSError, QubesException) as err: except (IOError, OSError, QubesException) as err:
print >> sys.stderr, "ERROR: {0}".format(err) print >> sys.stderr, "ERROR: {0}".format(err)
exit (1) exit (1)
@ -62,12 +91,18 @@ def main():
print >> sys.stderr, "Waiting for the VM(s) to shutdown..." print >> sys.stderr, "Waiting for the VM(s) to shutdown..."
shutdown_counter = 0 shutdown_counter = 0
while vm.is_running(): 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: if shutdown_counter > shutdown_counter_max:
# kill the VM # kill the VM
if options.verbose: if options.verbose:
print >> sys.stderr, "Killing the (apparently hanging) VM '{0}'...".format(vm.name) print >> sys.stderr, "Killing the (apparently hanging) VM '{0}'...".format(vm.name)
vm.force_shutdown() vm.force_shutdown()
#vms_list.remove(vm)
shutdown_counter += 1 shutdown_counter += 1
time.sleep (1) time.sleep (1)