From 8dcc40b1930e0d10d530277f1aef44e14f9baf98 Mon Sep 17 00:00:00 2001
From: Dhrumil Mistry <56185972+dmdhrumilmistry@users.noreply.github.com>
Date: Sat, 23 Jul 2022 01:54:18 +0530
Subject: [PATCH] create android data harvestor

---
 examples/Android/data-cloner.py               | 11 +++
 .../Android/forensics/data_harvestor.py       | 88 +++++++++++++++++++
 2 files changed, 99 insertions(+)
 create mode 100644 examples/Android/data-cloner.py
 create mode 100644 pyhtools/attackers/Android/forensics/data_harvestor.py

diff --git a/examples/Android/data-cloner.py b/examples/Android/data-cloner.py
new file mode 100644
index 0000000..344cd3f
--- /dev/null
+++ b/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()
\ No newline at end of file
diff --git a/pyhtools/attackers/Android/forensics/data_harvestor.py b/pyhtools/attackers/Android/forensics/data_harvestor.py
new file mode 100644
index 0000000..e6a3885
--- /dev/null
+++ b/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')
\ No newline at end of file