def keypress_get_generator(self):
"""
Return a generator to convert scripts into a language this sensor
understands
@return: KeypressGenerator for virtual machines
"""
return KeypressGeneratorPhysical()
def main(options):
"""
Main function
"""
if options.machine_config is None:
logger.error("No config file given.")
return
if options.command_file is None:
logger.error("No script file provided.")
return
# This isn't the class we use in practice, but fake it here for simplicity
# Get list of machine objects
machines = CONF.import_from_config(options.machine_config, "machine")
if options.machine not in machines:
logger.error("%s is not a valid machine from the config file."%options.machine)
logger.error("Valid targets are: %s"%machines.keys())
return
machine = machines[options.machine]
# Add a sensors to physical machines if needed
if machine.type == G.MACHINE_TYPES.PHYSICAL:
has_memory = has_disk = False
# Ensure that a sensor config is defined
if options.sensor_config is None:
logger.error("A sensor config file must be defined for physical analysis")
return
# Get the list of sensors
sensors = CONF.import_from_config(options.sensor_config, "sensor")
# Add sensors to our machine
print "Trying to find physical sensors for %s..."%options.machine
added_sensors = machine.add_sensors(sensors)
# Check that we can do both memory and disk analysis
if not machine.memory:
logger.error("No memory sensor available for analysis! Quitting.")
return
if not machine.disk:
logger.error("No disk sensor available for analysis! Quitting.")
return
if not machine.control:
logger.error("No control sensor available for analysis! Quitting.")
return
# load the command script
if not os.path.exists(options.command_file):
logger.error("File (%s) does not exist!" % options.command_file)
sys.exit(0)
# prepare the command script parser
parser = None
if machine.type == G.MACHINE_TYPES.PHYSICAL:
parser = KeypressGeneratorPhysical()
else:
parser = KeypressGeneratorVirtual()
# open file
f = open(options.command_file, 'r')
script_text = f.read()
f.close()
script = parser.text_to_script(script_text)
# Start the trials
for trial_num in range(options.trials):
print "Running trial %d" % trial_num
# Prep the machine -- reset it
if machine.type != G.MACHINE_TYPES.PHYSICAL:
machine.machine_reset()
else:
machine.machine_reset(options.pxe_server)
machine.power_off()
# Wait for machine to shutdown
time.sleep(15)
# Wait until machine has an ip address
logger.info("Waiting to get IP address of machine from PXE Server.")
start_time = time.time()
timeout = 360
while True:
#machine.ip_addr = get_ip(options.pxe_server, machine.get_mac_addr())
if (time.time() - start_time) > timeout:
logger.error("Could not get ip address for test machine from PXE Server for %d s" % timeout)
break
ip = machine.get_ip_addr(options.pxe_server)
if ip:
logger.info("Machine has IP address %s" % ip)
break
# wait until machine is up
logger.info("Waiting for machine to be up on the network.")
start_time = time.time()
timeout = 360
while True:
if (time.time() - start_time) > timeout:
logger.error("Timed out while waiting for machine to come back up (e.g. waiting for system to boot)")
break
if machine.get_net_status():
break
logger.info("Machine is back up. Commencing analysis.")
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Pausing Virtual Machine!")
machine.machine_pause()
else:
# pass
logger.info("Pausing Physical Machine Not Implemented Yet!")
# Take memory snapshot #1
logger.info("Taking start memory dump")
#memory_dump(machine, os.path.join(options.output_dir, "mem_dump_start" + str(trial_num)))
# TODO: Spawn data consumers for disk and memory?
logger.info("TODO: Starting disk analysis")
# Resume machine
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Resuming Virtual Machine!")
machine.machine_resume()
else:
# pass
logger.info("Resuming Physical Machine Not Implemented Yet!")
# Run command script and wait runtime seconds
logger.info("Running %s script for %d seconds." % (options.command_file, options.runtime))
machine.keypress_send(script)
time.sleep(options.runtime)
# pause machine if VM
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Pausing Virtual Machine!")
machine.machine_pause()
else:
# pass
logger.info("Pausing Physical Machine Not Implemented Yet!")
logger.info("TODO: Stopping disk analysis")
#disk_analysis.stop()
# Take memory snapshot #2
logger.info("Taking end memory dump")
#memory_dump(machine, os.path.join(options.output_dir, "mem_dump_end" + str(trial_num)))
# Resume machine
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Resuming Virtual Machine!")
machine.machine_resume()
else:
# pass
logger.info("Resuming Physical Machine Not Implemented Yet!")
print "Completed trial %d" % trial_num
print "Completed all trials."
def main(options):
# Define our control sensor and parser
if options.sensor_type == G.MACHINE_TYPES.PHYSICAL:
control_sensor = ControlSensorPhysical(options.target,options.port)
parser = KeypressGeneratorPhysical()
else:
control_sensor = ControlSensorVirtual(options.target,vm_type=options.sensor_type)
parser = KeypressGeneratorVirtual()
if options.status:
print "Getting status of machine..."
print "Status:", control_sensor.power_status()
elif options.shutdown:
print "Shutting down machine..."
control_sensor.power_shutdown()
elif options.poweron:
print "Starting machine..."
control_sensor.power_on()
elif options.poweroff:
print "Turing off machine..."
control_sensor.power_off()
elif options.reset:
print "Resetting machine..."
control_sensor.power_reset()
elif options.reboot:
print "Rebooting machine..."
control_sensor.power_reboot()
elif options.mouse_click:
(x,y) = options.mouse_click.split(",")
x = int(x)
y = int(y)
print "Sending mouse click to (%d,%d)"%(x,y)
control_sensor.mouse_click(x, y)
elif options.script:
print "Running script %s on machine..."%options.script
if not os.path.exists(options.script):
logging.error("File (%s) does not exist!" % options.script)
sys.exit(0)
# open file
f = open(options.script, 'r')
script_text = f.read()
f.close()
script = parser.text_to_script(script_text)
control_sensor.keypress_send(script)
elif options.mouse_wiggle is not None:
control_sensor.mouse_wiggle(options.mouse_wiggle)
else:
print "No action taken."
opts.print_help()
def main(options):
"""
Main function
"""
if options.machine_config is None:
logger.error("No config file given.")
return
if options.command_file is None:
logger.error("No script file provided.")
return
# This isn't the class we use in practice, but fake it here for simplicity
# Get list of machine objects
machines = CONF.import_from_config(options.machine_config, "machine")
if options.machine not in machines:
logger.error("%s is not a valid machine from the config file." %
options.machine)
logger.error("Valid targets are: %s" % machines.keys())
return
machine = machines[options.machine]
# Add a sensors to physical machines if needed
if machine.type == G.MACHINE_TYPES.PHYSICAL:
has_memory = has_disk = False
# Ensure that a sensor config is defined
if options.sensor_config is None:
logger.error(
"A sensor config file must be defined for physical analysis")
return
# Get the list of sensors
sensors = CONF.import_from_config(options.sensor_config, "sensor")
# Add sensors to our machine
print "Trying to find physical sensors for %s..." % options.machine
added_sensors = machine.add_sensors(sensors)
# Check that we can do both memory and disk analysis
if not machine.memory:
logger.error("No memory sensor available for analysis! Quitting.")
return
if not machine.disk:
logger.error("No disk sensor available for analysis! Quitting.")
return
if not machine.control:
logger.error("No control sensor available for analysis! Quitting.")
return
# load the command script
if not os.path.exists(options.command_file):
logger.error("File (%s) does not exist!" % options.command_file)
sys.exit(0)
# prepare the command script parser
parser = None
if machine.type == G.MACHINE_TYPES.PHYSICAL:
parser = KeypressGeneratorPhysical()
else:
parser = KeypressGeneratorVirtual()
# open file
f = open(options.command_file, 'r')
script_text = f.read()
f.close()
script = parser.text_to_script(script_text)
# Start the trials
for trial_num in range(options.trials):
print "Running trial %d" % trial_num
# Prep the machine -- reset it
if machine.type != G.MACHINE_TYPES.PHYSICAL:
machine.machine_reset()
else:
machine.machine_reset(options.pxe_server)
machine.power_off()
# Wait for machine to shutdown
time.sleep(15)
# Wait until machine has an ip address
logger.info("Waiting to get IP address of machine from PXE Server.")
start_time = time.time()
timeout = 360
while True:
#machine.ip_addr = get_ip(options.pxe_server, machine.get_mac_addr())
if (time.time() - start_time) > timeout:
logger.error(
"Could not get ip address for test machine from PXE Server for %d s"
% timeout)
break
ip = machine.get_ip_addr(options.pxe_server)
if ip:
logger.info("Machine has IP address %s" % ip)
break
# wait until machine is up
logger.info("Waiting for machine to be up on the network.")
start_time = time.time()
timeout = 360
while True:
if (time.time() - start_time) > timeout:
logger.error(
"Timed out while waiting for machine to come back up (e.g. waiting for system to boot)"
)
break
if machine.get_net_status():
break
logger.info("Machine is back up. Commencing analysis.")
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Pausing Virtual Machine!")
machine.machine_pause()
else:
# pass
logger.info("Pausing Physical Machine Not Implemented Yet!")
# Take memory snapshot #1
logger.info("Taking start memory dump")
#memory_dump(machine, os.path.join(options.output_dir, "mem_dump_start" + str(trial_num)))
# TODO: Spawn data consumers for disk and memory?
logger.info("TODO: Starting disk analysis")
# Resume machine
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Resuming Virtual Machine!")
machine.machine_resume()
else:
# pass
logger.info("Resuming Physical Machine Not Implemented Yet!")
# Run command script and wait runtime seconds
logger.info("Running %s script for %d seconds." %
(options.command_file, options.runtime))
machine.keypress_send(script)
time.sleep(options.runtime)
# pause machine if VM
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Pausing Virtual Machine!")
machine.machine_pause()
else:
# pass
logger.info("Pausing Physical Machine Not Implemented Yet!")
logger.info("TODO: Stopping disk analysis")
#disk_analysis.stop()
# Take memory snapshot #2
logger.info("Taking end memory dump")
#memory_dump(machine, os.path.join(options.output_dir, "mem_dump_end" + str(trial_num)))
# Resume machine
if machine.type != G.MACHINE_TYPES.PHYSICAL:
logger.info("Resuming Virtual Machine!")
machine.machine_resume()
else:
# pass
logger.info("Resuming Physical Machine Not Implemented Yet!")
print "Completed trial %d" % trial_num
print "Completed all trials."