def get_architecture(host,guest): if not is_acpi_state(host,guest,0): guest_start(host,guest) guest_switch(host,guest) wait_for_guest(host,guest) os = get_system_type(host,guest) if os == 'windows': unparsed = run_via_exec_daemon(['systeminfo'], host=domain_address(host,guest),line_split=True) for line in unparsed: if "System Type" in line: correct_line = line.split(" ") break if correct_line == None: print "ERR: Architecture not returned" raise InformationCollectionFailure() temp = correct_line correct_line = None for seg in temp: if any(char.isdigit() for char in seg): correct_line = seg break if correct_line == None: print "ERR: Architecture not returned" raise InformaitonCollectionFailure() Arch = re.sub("\D", "", correct_line) return Arch.strip() elif os == 'linux': return run(['uname','-m'],host=domain_address(host,guest),ignore_failure=True)[0].strip() else: raise UnexpectedOs()
def vm_reinstate_hibernate(dut,who='all'): """used to reinstate hibernate after vm_sleep_self call""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(dut,domain['name']) call_exec_daemon('run', [r'powercfg -h on'], vmip)
def wait_for_vms(machine, who='all', timeout=120): """wait for vm/vms to come up after a power operation""" domlist = find_guest_vms(machine) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(machine, domain['name']) wait_for_windows_to_come_up(vmip, timeout=timeout)
def vm_sleep_self(dut, who='all'): """sleep vm/vms from within the VM""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(dut, domain['name']) print "turning off hibernate :" + str(call_exec_daemon(command='run', args=[r'powercfg -h off'], host=vmip, timeout=60)) sleep(4) call_exec_daemon('run', [r'C:\Windows\System32\rundll32.exe powrprof.dll,SetSuspendState sleep'], vmip)
def vm_reboot_dom0(dut, who='all'): """reboot vm/vms from dom0""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: run(['xec-vm', '-n', domain['name'], 'reboot'], host=dut) vmip = domains.domain_address(dut, domain['name']) wait_for_windows_to_go_down(vmip) wait_for_windows_to_come_up(vmip)
def vm_shutdown_self(dut, who='all'): """shutdown vm/vms from within the VM""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(dut, domain['name']) call_exec_daemon('run', ['shutdown -s'], host=vmip, timeout=60) for domain in domlist: if who == 'all' or domain['name'] == who: wait_for_guest_to_go_down(dut, domain['name'])
def vm_hibernate_self(dut, who='all'): """hibernate vm/vms from within the VM""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(dut, domain['name']) call_exec_daemon('run', [r'C:\Windows\System32\rundll32.exe powrprof.dll,SetSuspendState hibernate'], host=vmip, timeout=60) for domain in domlist: if who == 'all' or domain['name'] == who: wait_for_guest_to_go_down(dut, domain['name'])
def vm_sleep_dom0(dut, who='all'): """sleep vm/vms from dom0""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: run(['xec-vm', '-n', domain['name'], 'sleep'], host=dut) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(dut, domain['name']) wait_for_windows_to_go_down(vmip)
def soft_shutdown_guest(host, guest, timeout=DEFAULT_SHUTDOWN_TIMEOUT, method="exec_daemon"): """Shutdown guest using platform-specific method""" if is_stopped(host, guest): return True vm_address = domain_address(host, guest, timeout=5) if method == "exec_daemon": call_exec_daemon('shutdown', host=vm_address, timeout=20) elif method == "ssh": run(['shutdown', '-h', 'now'], host=vm_address, timeout=20, check_host_key=False) else: raise Exception("Unknown method %s" % method) wait_for_shutdown(host, guest, timeout=timeout) return True
def vm_hibernate_self(dut, who='all'): """hibernate vm/vms from within the VM""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(dut, domain['name']) call_exec_daemon('run', [ r'C:\Windows\System32\rundll32.exe powrprof.dll,SetSuspendState hibernate' ], host=vmip, timeout=60) for domain in domlist: if who == 'all' or domain['name'] == who: wait_for_guest_to_go_down(dut, domain['name'])
def tools_install_problems(host, guest): """Return a string describing reasons why the tools install is bad or None if it is good for VM identified by host, guest pair""" bad = [] vm_address = domain_address(host, guest, timeout=5) for m in XENCLIENT_MODULES: if not check_module_loaded(vm_address, m): bad.append("Module %s not loaded" % m) if bad == []: return message = ('tools not considered installed at this point because '+ ' and '.join(bad)) print 'INSTALL_TOOLS:', message return message
def vm_sleep_self(dut, who='all'): """sleep vm/vms from within the VM""" domlist = find_guest_vms(dut) for domain in domlist: if who == 'all' or domain['name'] == who: vmip = domains.domain_address(dut, domain['name']) print "turning off hibernate :" + str( call_exec_daemon(command='run', args=[r'powercfg -h off'], host=vmip, timeout=60)) sleep(4) call_exec_daemon('run', [ r'C:\Windows\System32\rundll32.exe powrprof.dll,SetSuspendState sleep' ], vmip)
def get_architecture(host, guest): if not is_acpi_state(host, guest, 0): guest_start(host, guest) guest_switch(host, guest) wait_for_guest(host, guest) os = get_system_type(host, guest) if os == 'windows': unparsed = run_via_exec_daemon(['systeminfo'], host=domain_address(host, guest), line_split=True) for line in unparsed: if "System Type" in line: correct_line = line.split(" ") break if correct_line == None: print "ERR: Architecture not returned" raise InformationCollectionFailure() temp = correct_line correct_line = None for seg in temp: if any(char.isdigit() for char in seg): correct_line = seg break if correct_line == None: print "ERR: Architecture not returned" raise InformaitonCollectionFailure() Arch = re.sub("\D", "", correct_line) return Arch.strip() elif os == 'linux': return run(['uname', '-m'], host=domain_address(host, guest), ignore_failure=True)[0].strip() else: raise UnexpectedOs()
def disableSnapshot(machine,vms,steps,build,test_path,caller): dummy = str(random.randint(10000,20000)) vm_address = str(domain_address(machine, vms)) r = run(['xec-vm','-n',vms,'--disk','1','get','snapshot'], host=machine) if 'temporary' in r: tslib.forcepowerstate(machine, 'off', vms) r = run(['xec-vm','-n',vms,'--disk','1','set','snapshot','none'], host=machine) tslib.cmdResponse(r, 'xec-vm -n '+vms+' --disk 1 set snapshot none') snapshot = run(['xec-vm','-n',vms,'--disk','1','get','snapshot'], host=machine) tslib.cmdResponse(r, 'xec-vm -n '+vms+' --disk 1 get snapshot') if 'none' not in snapshot: tslib.pauseToDebug("xec failed to disable snapshot") windows_transitions.vm_poweron(machine, vms) for i in range(2): windows_transitions.vm_reboot_dom0(machine, vms) res = run_via_exec_daemon(['mkdir','C:\\Users\\Administrator\\Desktop\\'+dummy],host=vm_address, wait=True) windows_transitions.vm_reboot_dom0(machine, vms) res = run_via_exec_daemon(['dir','C:\\Users\\Administrator\\Desktop\\'],host=vm_address, wait=True) if dummy not in res: tslib.pauseToDebug("Snapshot disabled: Old folders lost after reboot")
def stubdom_boot(dut, guest): """Test guest still boots on dut when running with studbom""" _, name = name_split(guest) vm_address = domain_address(dut, guest) run(['xec-vm', '-n', name, 'start'], host=dut) run(['xec-vm', '-n', name, 'switch'], host=dut) wait_for_windows_to_come_up(vm_address) shutdown_windows(vm_address) wait_for_vm_to_stop(dut, guest) status = find_domain(dut, guest) print 'INFO: domain status', status run(['xec-vm', '-n', name, 'set', 'cd', 'xc-tools.iso'], host=dut) run(['xec-vm', '-n', name, 'set', 'stubdom', 'true'], host=dut) print 'INFO: booting', guest, 'with stubdom' run(['xec-vm', '-n', name, 'start'], host=dut) run(['xec-vm', '-n', name, 'switch'], host=dut) wait_for_windows_to_come_up(vm_address) print 'HEADLINE: verified', guest, 'works with stubdom=true'
def stubdom_boot(dut, guest): """Test guest still boots on dut when running with studbom""" _, name = name_split(guest) vm_address = domain_address(dut, guest) run(['xec-vm', '-n', name, 'start'], host=dut) run(['xec-vm', '-n', name, 'switch'], host=dut) wait_for_windows_to_come_up(vm_address) shutdown_windows(vm_address) wait_for_vm_to_stop(dut, guest) status = find_domain(dut, guest) print 'INFO: domain status', status run(['xec-vm', '-n', name, 'set', 'cd', 'xc-tools.iso'], host=dut) run(['xec-vm', '-n', name, 'set', 'stubdom', 'true'], host=dut) print 'INFO: booting', guest, 'with stubdom' run(['xec-vm', '-n', name, 'start'], host=dut) run(['xec-vm', '-n', name, 'switch'], host=dut) wait_for_windows_to_come_up(vm_address) print 'HEADLINE: verified', guest, 'works with stubdom=true'
def soft_shutdown_guest(host, guest, timeout=DEFAULT_SHUTDOWN_TIMEOUT, method="exec_daemon"): """Shutdown guest using platform-specific method""" if is_stopped(host, guest): return True vm_address = domain_address(host, guest, timeout=5) if method == "exec_daemon": call_exec_daemon('shutdown', host=vm_address, timeout=20) elif method == "ssh": run(['shutdown', '-h', 'now'], host=vm_address, timeout=20, check_host_key=False) else: raise Exception("Unknown method %s" % method) wait_for_shutdown(host, guest, timeout=timeout) return True
def guest_disk_count(host,guest): """function to get the count of 'disks' the guest sees""" total_disk_count = 0 os = get_system_type(host,guest) if os == 'windows': unparsed = run_via_exec_daemon(['echo', 'list', 'disk', '|', 'diskpart','|','find','"Disk"','|','find','"Online"'],host=domain_address(host, guest, timeout=5),line_split=True) #necessary as windows sometimes puts random blank lines at the end print "DEBUG: "+',,'.join(unparsed) for entry in unparsed: if "Disk" in entry: total_disk_count += 1 elif os == 'linux': unparsed = run(['lsblk','-l','-o','NAME,TYPE'], host=domain_address(host, guest, timeout=5), timeout=20, check_host_key=False, line_split=True) for entry in unparsed: if "disk" in entry: total_disk_count += 1 else: raise UnexpectedOs() print "DEBUG: returning guest: \"%s\" has %d disks" % (guest,total_disk_count) return total_disk_count
def tools_install_problems(host, guest): """Return a string describing reasons why the tools install is bad or None if it is good for VM identified by host, guest pair""" bad = [] vm_address = domain_address(host, guest, timeout=5) msi_versions = get_msi_versions(vm_address) for msi in msi_versions: print 'INFO:', msi, 'is installed' for msi in EXPECTED_MSIS: if msi not in msi_versions: bad.append(msi + ' MSI not installed') services = get_running_services(vm_address) for service in services: print 'TOOLS: service', service, 'running' for service in EXPECTED_SERVICES: if service not in services: bad.append(service + ' not running') if bad == []: return message = ('tools not considered installed at this point because ' + ' and '.join(bad)) print 'INSTALL_TOOLS:', message return message
def tools_install_problems(host, guest): """Return a string describing reasons why the tools install is bad or None if it is good for VM identified by host, guest pair""" bad = [] vm_address = domain_address(host, guest, timeout=5) msi_versions = get_msi_versions(vm_address) for msi in msi_versions: print 'INFO:', msi, 'is installed' for msi in EXPECTED_MSIS: if msi not in msi_versions: bad.append(msi+' MSI not installed') services = get_running_services(vm_address) for service in services: print 'TOOLS: service', service, 'running' for service in EXPECTED_SERVICES: if service not in services: bad.append(service+' not running') if bad == []: return message = ('tools not considered installed at this point because '+ ' and '.join(bad)) print 'INSTALL_TOOLS:', message return message
def get_address(): """get address for VM """ check_running() return domain_address(host, guest, timeout=5)
def get_address(): """get address for VM """ check_running() return domain_address(host, guest, timeout=5)
def soft_reboot(dut, vm_name): domain = find_domain(dut, vm_name) print 'INFO: rebooting guest', domain vm_address = domain_address(dut, vm_name) run(['shutdown', '-r', 'now'], host=vm_address, check_host_key=False) wait_for_domid_change(dut, vm_name, domain['dom_id'])
def soft_reboot(dut, vm_name): domain = find_domain(dut, vm_name) print 'INFO: rebooting guest', domain vm_address = domain_address(dut, vm_name) run(['shutdown', '-r', 'now'], host=vm_address, check_host_key=False) wait_for_domid_change(dut, vm_name, domain['dom_id'])