Ejemplo n.º 1
0
 def __init__(self, config):
     super(GadgetKeyboard, self).__init__()
     self.emulator = config["pem_port"] # Initialize path to HID keyboard emulator
     self.filepath = "" # Initialize filepath for send_keystrokes_from_file()
     self.delay_between_keys = 0 # Delay (seconds) between keystrokes
     self.modifier = 0 # On default don't use modifier key
     self.line_number = 0 # Line number we are parsing from filepath
     logger.set_process_prefix()
Ejemplo n.º 2
0
 def __init__(self, emulator_path="/dev/hidg0", write_mode="w"):
     self.emulator = emulator_path  # Initialize path to HID keyboard emulator
     self.write_mode = write_mode  # Change send_key() file write mode
     self.filepath = ""  # Initialize filepath for send_keystrokes_from_file()
     self.delay_between_keys = 0  # Delay (seconds) between keystrokes
     self.modifier = 0  # On default don't use modifier key
     self.line_number = 0  # Line number we are parsing from filepath
     logger.set_process_prefix()
Ejemplo n.º 3
0
 def __init__(self, config):
     super(GadgetKeyboard, self).__init__()
     self.emulator = config[
         "pem_port"]  # Initialize path to HID keyboard emulator
     self.filepath = ""  # Initialize filepath for send_keystrokes_from_file()
     self.delay_between_keys = 0  # Delay (seconds) between keystrokes
     self.modifier = 0  # On default don't use modifier key
     self.line_number = 0  # Line number we are parsing from filepath
     logger.set_process_prefix()
Ejemplo n.º 4
0
def main(argv=None):
    """
    Entry point for library-like use.
    """
    try:
        logger.set_process_prefix()

        config.parse()

        if argv != None:
            backup_argv = sys.argv
            sys.argv = argv

        args = parse_args()

        if args.debug:
            logger.level(logging.DEBUG)

        device_manager = DevicesManager(args)

        if args.device:
            device, tester = device_manager.try_flash_specific(args)
        else:
            device, tester = device_manager.try_flash_model(args)

        if not args.notest:
            print("Testing " + str(device.name) + ".")
            tester.execute()

        if not args.nopoweroff:
            device.detach()

        if args.boot:
            device_manager.boot_device_to_mode(device, args)

        device_manager.release(device)

        if "backup_argv" in locals():
            sys.argv = backup_argv
        return 0

    except KeyboardInterrupt:
        print("Keyboard interrupt, stopping aft")
        logger.error("Keyboard interrupt, stopping aft.")
        sys.exit(0)

    except:
        _err = sys.exc_info()
        logger.error(str(_err[0]).split("'")[1] + ": " + str(_err[1]))
        raise

    finally:
        thread_handler.set_flag(thread_handler.RECORDERS_STOP)
        for thread in thread_handler.get_threads():
            thread.join(5)
Ejemplo n.º 5
0
Archivo: main.py Proyecto: skuusela/AFT
def main(argv=None):
    """
    Entry point for library-like use.
    """
    try:
        logger.set_process_prefix()

        config.parse()

        if argv != None:
            backup_argv = sys.argv
            sys.argv = argv

        args = parse_args()

        if args.debug:
            logger.level(logging.DEBUG)

        device_manager = DevicesManager(args)
        device, tester = device_manager.try_flash_model(args)

        if args.emulateusb:
            device.boot_usb_test_mode()

        if not args.notest:
            if not args.emulateusb:
                device.boot_internal_test_mode()

            print("Testing " + str(device.name) + ".")
            tester.execute()

            if not args.nopoweroff:
                device.detach()
                if args.emulateusb:
                    device_manager.stop_image_usb_emulation(
                                                    device.leases_file_name)

        if not args.emulateusb:
            if args.boot == "test_mode":
                device.boot_internal_test_mode()

            elif args.boot == "service_mode":
                device.boot_usb_service_mode()

        device_manager.release(device)

        if "backup_argv" in locals():
            sys.argv = backup_argv
        return 0

    except KeyboardInterrupt:
        print("Keyboard interrupt, stopping aft")
        logger.error("Keyboard interrupt, stopping aft.")
        sys.exit(0)

    except:
        _err = sys.exc_info()
        logger.error(str(_err[0]).split("'")[1] + ": " + str(_err[1]))
        raise

    finally:
        thread_handler.set_flag(thread_handler.RECORDERS_STOP)
        for thread in thread_handler.get_threads():
            thread.join(5)