Exemplo n.º 1
1
def check_device_compatibility():
    """
    checks for connected device compatibility with the framework. Quit if connected device is not found or compatible
    """
    '''
    DEVICE_CONFIG = "devices.ini"

    config = ConfigParser.ConfigParser()
    config.read(DEVICE_CONFIG)
    serial_numbers = config.sections()
    '''

    serial_numbers = dev.list()

    adb = Adb()
    connected_devices = adb.devices()

    if len(connected_devices) >= 1:
        print "Total of %d devices connected"%len(connected_devices)
    else:
        log.critical("Please connect device before starting droidpot")
        exit(0)

    for device in connected_devices:
        isCompatible = False
        for serial_number in serial_numbers:
            if device == serial_number:
                print "device %s is compatible"%serial_number
                isCompatible = True
                break

        if not isCompatible:
            log.critical("Found device that is not compatible with droidpot. Please use only compatible devices for malware analysis.")
            exit(1)
Exemplo n.º 2
0
class fastboot_testCase(unittest.TestCase):
    def setUp(self):
        self.fastboot = Fastboot()
        self.adb = Adb()
        self.adb.reboot_bootloader()

    def tearDown(self):
        self.fastboot.reboot()

        while len(self.adb.devices()) == 0:
            time.sleep(5)

    @unittest.skip("skip reboot test")
    def test_reboot(self):
        self.fastboot.reboot()
        result = True

        print "[!] waiting for phone to reboot..."
        for i in range(0, 5):
            if len(self.adb.devices()) == 0:
                time.sleep(10)
                result = False
            else:
                result = True
                break

            if i == 4:
                print "[!] Reboot time out"

        self.assertTrue(result, msg="fastboot reboot failed")

    @unittest.skip("skip reboot bootloader test")
    def test_reboot_bootloader(self):
        result = False
        self.fastboot.reboot_bootloader()

        devices = self.fastboot.devices()

        print "[!] waiting for phone to enter bootloader mode..."
        for i in range(0, 5):
            devices = self.fastboot.devices()

            if len(devices) != 0:
                result = True
                break
            else:
                time.sleep(10)

            if i == 4:
                print "[!] Reboot time out"

        self.assertTrue(result, msg="fastboot reboot-bootloader failed")

    def test_devices(self):
        result = False
        self.fastboot.reboot_bootloader()

        time.sleep(5)
        devices = self.fastboot.devices()

        if len(devices) > 0:
            result = True

        self.assertTrue(result, msg="fastboot devices did not detect any devices")
Exemplo n.º 3
0
class adb_testcase(unittest.TestCase):
    def setUp(self):
        self.adb_c = Adb()
        self.adb_c.start_server()

    def tearDown(self):
        #self.adb_c.kill_server()
        pass

    #@unittest.skip("skipping test device")
    def test_devices(self):
        devices = self.adb_c.devices()

        #print devices
        result = True
        if len(devices) == 0:
            result = False

        self.assertTrue(result, msg="No device detected")


    def test_push_file(self):
        path = os.path.dirname(os.path.realpath(__file__))

        src = os.path.join(path, TEST_FILE)
        dest = os.path.join(SDCARD_PATH, TEST_FILE)

        isSuccessful = self.adb_c.push(source=src, dest=dest)
        self.assertTrue(isSuccessful, msg="adb push not successful")

    def test_pull_file(self):
        path = os.path.dirname(os.path.realpath(__file__))

        src = os.path.join(SDCARD_PATH, TEST_FILE)
        dest = os.path.join(path, "extract.txt")

        isSuccessful = self.adb_c.pull(source=src, dest=dest)
        self.assertTrue(isSuccessful, msg="adb pull not successful")

        os.remove(dest)

    def test_shell_no_root(self):
        parameters = "ps"
        result = self.adb_c.shell(command=parameters)

        self.assertTrue(result.isSuccess, msg="adb shell ps not successful")
        print result.std_output

    def test_shell_with_root(self):
        parameters = "strace -p 1"
        result = self.adb_c.shell(needOutput=False, root=True, command=parameters)

        self.assertTrue(result.isSuccess, msg="adb shell with root not successful")
        self.adb_c.reboot()

        sleep(5)
        while len(self.adb_c.devices()) == 0:
            print("waiting for device to reboot...")
            sleep(10)


    @unittest.skip("skipping reboot test")
    def test_reboot_commands(self):
        """
        reboot command testcase should be the last
        :return:
        """
        self.adb_c.reboot()
        print("device rebooting... Please wait...")
        sleep(3)
        devices = self.adb_c.devices()

        foundDevice = False
        if devices:
            foundDevice = True

        self.assertFalse(foundDevice, msg="adb reboot not successful")