def vmware_pre_run(_VirtualMachine): ######################################################## # Mount the second partition of the Virtual Machine # Check the User Manual for a reference to VM partitions ######################################################## vm_util.mount_vmdk(_VirtualMachine, 2, _VirtualMachine._m_mount_directory) ######################################################## # sudo echo <url assigned to the current Virtual Machine> <path to the mounted directory> url.txt # The selenium script expects the file to be read to be "url.txt" ######################################################## if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "echo " + _VirtualMachine._m_url + " > " + "\"" + _VirtualMachine._m_mount_directory + "url.txt" + "\"" else: _Command = "sudo echo " + _VirtualMachine._m_url + " > " + "\"" + _VirtualMachine._m_mount_directory + "url.txt" + "\"" print _Command os.system(_Command) ######################################################## # Unmount the partition and return to start running ######################################################## vm_util.umount_vmdk(_VirtualMachine, _VirtualMachine._m_mount_directory)
def vmware_post_run(_VirtualMachine): if _VirtualMachine._m_startup_file != None: _Command = "sudo vmrun -T ws snapshot " + "\"" + _VirtualMachine._m_directory + _VirtualMachine._m_config_file + "\"" + " NewMachineSnapShot2" print "Creating Snapshot" os.system(_Command) time.sleep(15) # make sure the snapshot finishes before starting _Vmdk = vm_util.find_last_snapshot(_VirtualMachine) _Command = "sudo cp " + "\"" + _VirtualMachine._m_directory + _Vmdk + "\"" + " " + "\"" + _VirtualMachine._m_collection + "\"" print _Command os.system(_Command) print "Deleting snapshot" _Command = "sudo vmrun -T ws deleteSnapshot " + "\"" + _VirtualMachine._m_directory + _VirtualMachine._m_config_file + "\"" + " NewMachineSnapshot2" os.system(_Command) time.sleep(20) vm_util.mount_vmdk(_VirtualMachine, 3, _VirtualMachine._m_mount_directory) _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "pictures" "\"" + " "+ "\"" + _VirtualMachine._m_collection + "/pictures" + "\"" print _Command os.system(_Command) time.sleep(.1) _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "pcaps" "\"" + " "+ "\"" + _VirtualMachine._m_collection + "/pcaps" + "\"" print _Command os.system(_Command) time.sleep(.1) _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "cookies" "\"" + " "+ "\"" + _VirtualMachine._m_collection + "/cookies" + "\"" print _Command os.system(_Command) time.sleep(.1) _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "source" "\"" + " "+ "\"" + _VirtualMachine._m_collection + "/source" + "\"" print _Command os.system(_Command) time.sleep(5) vm_util.umount_vmdk(_VirtualMachine, _VirtualMachine._m_mount_directory)
def vmware_pre_run(_VirtualMachine): if _VirtualMachine._m_startup_file != None: vm_util.mount_vmdk(_VirtualMachine, 3, _VirtualMachine._m_mount_directory) _Command = "sudo echo " + URLList[int(float(get_url_index()) % len(URLList))][:-1] + " > " + "\"" + _VirtualMachine._m_mount_directory + "url.txt" + "\"" print _Command os.system(_Command) vm_util.umount_vmdk(_VirtualMachine)
def vmware_pre_run(_VirtualMachine): if _VirtualMachine._m_startup_file != None: vm_util.mount_vmdk(_VirtualMachine, 2, _VirtualMachine._m_mount_directory) _Command = "sudo echo " + _VirtualMachine._m_url + " > " + "\"" + _VirtualMachine._m_mount_directory + "url.txt" + "\"" print _Command os.system(_Command) vm_util.umount_vmdk(_VirtualMachine, _VirtualMachine._m_mount_directory)
def vmware_post_run(_VirtualMachine): if _VirtualMachine._m_startup_file != None: vm_util.mount_vmdk(_VirtualMachine, 2, _VirtualMachine._m_mount_directory) _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "pictures" "\"" + " "+ "\"" + _VirtualMachine._m_collection + "/pictures" + "\"" print _Command os.system(_Command) _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "pcaps" "\"" + " "+ "\"" + _VirtualMachine._m_collection + "/pcaps" + "\"" print _Command os.system(_Command) vm_util.umount_vmdk(_VirtualMachine)
def vmware_post_run(_VirtualMachine): if _VirtualMachine._m_snap_shot: ##################################################### # Logic: # # Create a new Snapshot # Copy it to the output destination # delete the Snapshot so it looks like it was never made. ##################################################### if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "vmrun -T ws snapshot " + "\"" + _VirtualMachine._m_directory + _VirtualMachine._m_config_file + "\"" + " NewMachineSnapShot2" else: _Command = "sudo vmrun -T ws snapshot " + "\"" + _VirtualMachine._m_directory + _VirtualMachine._m_config_file + "\"" + " NewMachineSnapShot2" print "Creating Snapshot" os.system(_Command) time.sleep(15) # make sure the snapshot finishes before starting _Vmdk = vm_util.find_last_snapshot(_VirtualMachine) if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "copy " + "\"" + _VirtualMachine._m_directory + _Vmdk + "\"" + " " + "\"" + _VirtualMachine._m_collection + "\"" else: _Command = "sudo cp " + "\"" + _VirtualMachine._m_directory + _Vmdk + "\"" + " " + "\"" + _VirtualMachine._m_collection + "\"" print _Command os.system(_Command) print "Deleting snapshot" if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "vmrun -T ws deleteSnapshot " + "\"" + _VirtualMachine._m_directory + _VirtualMachine._m_config_file + "\"" + " NewMachineSnapshot2" else: _Command = "sudo vmrun -T ws deleteSnapshot " + "\"" + _VirtualMachine._m_directory + _VirtualMachine._m_config_file + "\"" + " NewMachineSnapshot2" os.system(_Command) time.sleep(20) # make sure there is enough time for the snapshot to be deleted ######################################################## # Mount the third partition, which contains output data # Check the User Manual for a reference to VM partitions ######################################################## vm_util.mount_vmdk(_VirtualMachine, 3, _VirtualMachine._m_mount_directory) ######################################################## # sudo cp -r <directory of mounted VM filesystem>/* <directory of output> ######################################################## if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "copy " + "\"" + _VirtualMachine._m_mount_directory + "pcaps" + "\"" + " " +"\"" + _VirtualMachine._m_collection + "/" + "\"" else: _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "pcaps" + "\"" + " " +"\"" + _VirtualMachine._m_collection + "/" + "\"" print _Command os.system(_Command) if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "copy " + "\"" + _VirtualMachine._m_mount_directory + "cookies" + "\"" + " " + "\"" + _VirtualMachine._m_collection + "/" + "\"" else: _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "cookies" + "\"" + " " + "\"" + _VirtualMachine._m_collection + "/" + "\"" print _Command os.system(_Command) if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "copy " + "\"" + _VirtualMachine._m_mount_directory + "pictures" + "\"" + " " + "\"" + _VirtualMachine._m_collection + "/" + "\"" else: _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "pictures" + "\"" + " " + "\"" + _VirtualMachine._m_collection + "/" + "\"" print _Command os.system(_Command) if os.environ.get('OS','') == 'Windows_NT': ############################################ # Win32 Support ############################################ _Command = "copy " + "\"" + _VirtualMachine._m_mount_directory + "source" + "\"" + " " +"\"" + _VirtualMachine._m_collection + "/" + "\"" else: _Command = "sudo cp -r " + "\"" + _VirtualMachine._m_mount_directory + "source" + "\"" + " " +"\"" + _VirtualMachine._m_collection + "/" + "\"" print _Command os.system(_Command) time.sleep(5) ######################################################## # Unmount the partition and return to continue execution ######################################################## vm_util.umount_vmdk(_VirtualMachine, _VirtualMachine._m_mount_directory)