Merge pull request #15 from dmdhrumilmistry/android-data-cloner

create android data harvestor

examples/Android/data-cloner.py

@@ -0,0 +1,11 @@
+from pyhtools.attackers.Android.forensics.data_harvestor import DataHarvestor
+
+cloner = DataHarvestor(
+        dest_path=r'path', # path where cloned data will be stored on host machine
+        device_name='emulator-5554', # device name using `adb devices`
+        host='127.0.0.1', # adb server ip
+        port=5037, # adb server port
+    )
+
+# start cloning
+cloner.start()

pyhtools/attackers/Android/forensics/data_harvestor.py

@@ -0,0 +1,88 @@
+from ppadb.client import Client
+from ppadb.device import Device
+from subprocess import Popen, PIPE, STDOUT
+
+import logging
+import shutil
+import threading
+import os
+logging.basicConfig(level=logging.DEBUG,
+                    format='[%(asctime)s] [%(levelname)s] - %(message)s')
+
+
+class DataHarvestorExceptions:
+    '''
+    Exceptions for DataHarvestor class
+    '''
+    class ServerNotRunning(Exception):
+        pass
+
+    class NoDevicesFound(Exception):
+        pass
+
+
+class DataHarvestor:
+    def __init__(self, dest_path: str, device_name: str, host: str = '127.0.0.1', port: int = 5037) -> None:
+        assert type(dest_path) == str
+        assert type(device_name) == str
+        assert type(host) == str
+        assert type(port) == int
+
+        self.__device_name = device_name
+        self.__dest_path = dest_path
+
+        self._adb = Client(
+            host=host,
+            port=port
+        )
+        self.device: Device = self.get_device()
+        logging.info(f'Using {self.__device_name}')
+
+        # check paths, if present remove all files present
+        if os.path.isdir(dest_path):
+            logging.warning(
+                'Destination Path is already present, complete directory data will be overwritten')
+            shutil.rmtree(dest_path)
+
+        # Create Paths
+        logging.info('Creating Destination Path')
+        os.makedirs(dest_path)
+
+    def get_device(self):
+        _ = self.get_adb_devices()
+        device: Device = self._adb.device(self.__device_name)
+        return device
+
+    def get_adb_devices(self):
+        try:
+            devices: list[Device] = self._adb.devices()
+            if len(devices) == 0:
+                raise DataHarvestorExceptions.NoDevicesFound(
+                    "No ADB Devices Attached")
+
+            return devices
+        except RuntimeError:
+            raise DataHarvestorExceptions.ServerNotRunning(
+                "ADB Server is not running, start using `adb start-server`")
+
+    def __clone_dir(self, phone_src: str):
+        # clone directory to host machine
+        Popen(
+            ['adb', 'pull', str(phone_src), str(self.__dest_path)],
+            shell=True,
+            stdout=PIPE,
+            stderr=STDOUT,
+        )
+
+    def start(self):
+        # get package paths
+        logging.info('Getting packages list')
+        package_paths = [os.path.join('/data/data/', package_path)
+                         for package_path in str(self.device.shell('ls -a /data/data')).splitlines()]
+
+        logging.info('Harvesting Data')
+        for package_path in package_paths:
+            threading.Thread(target=self.__clone_dir,
+                             args=(package_path,)).start()
+
+        logging.info('Data Harvesting Completed')