mirror of
https://github.com/hyugogirubato/KeyDive.git
synced 2024-12-22 19:13:35 +00:00
284 lines
10 KiB
Python
284 lines
10 KiB
Python
import logging
|
|
import re
|
|
import shutil
|
|
import subprocess
|
|
|
|
import frida
|
|
import requests
|
|
|
|
from pathlib import Path
|
|
from frida.core import Device
|
|
|
|
# Suppress urllib3 warnings
|
|
logging.getLogger('urllib3.connectionpool').setLevel(logging.ERROR)
|
|
|
|
|
|
def shell(prompt: list) -> subprocess.CompletedProcess:
|
|
"""
|
|
Executes a shell command and returns the completed process.
|
|
|
|
Args:
|
|
prompt (list): The command to be executed as a list of strings.
|
|
|
|
Returns:
|
|
subprocess.CompletedProcess: The completed process object containing return code, stdout, and stderr.
|
|
"""
|
|
prompt = list(map(str, prompt))
|
|
# logging.getLogger('Shell').debug(' '.join(prompt))
|
|
return subprocess.run(prompt, capture_output=True)
|
|
|
|
|
|
class ADB:
|
|
"""
|
|
Class for managing interactions with the Android device via ADB.
|
|
"""
|
|
|
|
def __init__(self, device: str = None, timeout: int = 5):
|
|
"""
|
|
Initializes the ADB connection to a device.
|
|
|
|
Args:
|
|
device (str, optional): The ID of the device to connect to. If None, defaults to the first USB device.
|
|
timeout (int, optional): The timeout for device connection in seconds. Defaults to 5.
|
|
|
|
Raises:
|
|
EnvironmentError: If ADB is not found in the system path.
|
|
Exception: If the connection to the device fails.
|
|
"""
|
|
self.logger = logging.getLogger(self.__class__.__name__)
|
|
|
|
# Ensure ADB is available
|
|
if not shutil.which('adb'):
|
|
raise EnvironmentError(
|
|
'ADB is not recognized as an environment variable. '
|
|
'Ensure ADB is installed and refer to the documentation: '
|
|
'https://github.com/hyugogirubato/KeyDive/blob/main/docs/PACKAGE.md#adb-android-debug-bridge'
|
|
)
|
|
|
|
# Start the ADB server if not already running
|
|
sp = shell(['adb', 'start-server'])
|
|
if sp.returncode != 0:
|
|
self.logger.warning('ADB server startup failed (Error: %s)', sp.stdout.decode('utf-8').strip())
|
|
|
|
# Select device based on provided ID or default to the first USB device
|
|
try:
|
|
self.device: Device = frida.get_device(id=device, timeout=timeout) if device else frida.get_usb_device(timeout=timeout)
|
|
self.logger.info('Connected to device: %s (%s)', self.device.name, self.device.id)
|
|
except Exception as e:
|
|
self.logger.error('Failed to connect to device: %s', e)
|
|
raise e
|
|
|
|
self.prompt = ['adb', '-s', self.device.id, 'shell']
|
|
|
|
# Obtain device properties
|
|
properties = self.device_properties()
|
|
if properties:
|
|
self.logger.info('SDK API: %s', properties.get('ro.build.version.sdk', 'Unknown'))
|
|
self.logger.info('ABI CPU: %s', properties.get('ro.product.cpu.abi', 'Unknown'))
|
|
else:
|
|
self.logger.warning('No device properties were retrieved.')
|
|
|
|
def device_properties(self) -> dict:
|
|
"""
|
|
Retrieves system properties from the connected device using ADB shell commands.
|
|
|
|
Returns:
|
|
dict: A dictionary mapping device property keys to their corresponding values.
|
|
"""
|
|
# https://source.android.com/docs/core/architecture/configuration/add-system-properties?#shell-commands
|
|
properties = {}
|
|
|
|
# Execute shell command to retrieve device properties
|
|
sp = shell([*self.prompt, 'getprop'])
|
|
if sp.returncode != 0:
|
|
self.logger.error('Failed to retrieve device properties (Error: %s)', sp.stdout.decode('utf-8').strip())
|
|
return properties
|
|
|
|
# Parse the output to fill the properties dictionary
|
|
for line in sp.stdout.decode('utf-8').splitlines():
|
|
match = re.match(r'\[(.*?)\]: \[(.*?)\]', line)
|
|
if match:
|
|
key, value = match.groups()
|
|
|
|
# Attempt to cast numeric and boolean values
|
|
if value.isdigit():
|
|
value = int(value)
|
|
elif value.lower() in ('true', 'false'):
|
|
value = value.lower() == 'true'
|
|
|
|
properties[key] = value
|
|
|
|
return properties
|
|
|
|
def list_applications(self, user: bool = True, system: bool = False) -> dict:
|
|
"""
|
|
Lists installed applications on the device, filtering by user and/or system apps.
|
|
|
|
Args:
|
|
user (bool, optional): Whether to include user-installed applications. Defaults to True.
|
|
system (bool, optional): Whether to include system applications. Defaults to False.
|
|
|
|
Returns:
|
|
dict: A dictionary mapping application packages to their file paths.
|
|
"""
|
|
applications = {}
|
|
|
|
# Validate input and set the appropriate prompt
|
|
if not user and not system:
|
|
return applications
|
|
|
|
prompt = [*self.prompt, 'pm', 'list', 'packages', '-f']
|
|
if user and not system:
|
|
prompt.append('-3')
|
|
elif not user and system:
|
|
prompt.append('-s')
|
|
|
|
# Execute shell command to list applications
|
|
sp = shell(prompt)
|
|
if sp.returncode != 0:
|
|
self.logger.error('Failed to retrieve application list (Error: %s)', sp.stdout.decode('utf-8').strip())
|
|
return applications
|
|
|
|
# Parse and store applications in the dictionary
|
|
for line in sp.stdout.decode('utf-8').splitlines():
|
|
try:
|
|
path, package = line.strip().split(':', 1)[1].rsplit('=', 1)
|
|
applications[package] = path
|
|
except Exception as e:
|
|
pass
|
|
|
|
return applications
|
|
|
|
def start_application(self, package: str) -> bool:
|
|
"""
|
|
Starts an application by its package name.
|
|
|
|
Args:
|
|
package (str): The package name of the application to start.
|
|
|
|
Returns:
|
|
bool: True if the application was started successfully, False otherwise.
|
|
"""
|
|
# Get package information
|
|
sp = shell([*self.prompt, 'dumpsys', 'package', package])
|
|
lines = sp.stdout.decode('utf-8').splitlines()
|
|
|
|
# Remove empty lines to ensure backwards compatibility
|
|
lines = [l.strip() for l in lines if l.strip()]
|
|
|
|
# Look for main activity in package information
|
|
for i, line in enumerate(lines):
|
|
if 'android.intent.action.MAIN' in line:
|
|
match = re.search(fr'({package}/[^ ]+)', lines[i + 1])
|
|
if match:
|
|
# Attempt to start the application
|
|
main_activity = match.group()
|
|
sp = shell([*self.prompt, 'am', 'start', '-n', main_activity])
|
|
if sp.returncode == 0:
|
|
return True
|
|
|
|
self.logger.error('Failed to start application %s (Error: %s)', package, sp.stdout.decode('utf-8').strip())
|
|
break
|
|
|
|
self.logger.error('Package %s not found or has no MAIN intent action.', package)
|
|
return False
|
|
|
|
def enumerate_processes(self) -> dict:
|
|
"""
|
|
Lists running processes on the device, mapping process names to PIDs.
|
|
|
|
Returns:
|
|
dict: A dictionary mapping process names to their corresponding PIDs.
|
|
"""
|
|
# https://github.com/frida/frida/issues/1225#issuecomment-604181822
|
|
processes = {}
|
|
|
|
# Try to get the list of processes using `ps -A`
|
|
prompt = [*self.prompt, 'ps']
|
|
sp = shell([*prompt, '-A'])
|
|
lines = sp.stdout.decode('utf-8').splitlines()
|
|
|
|
# If the output has less than 10 lines, try the alternative `ps` command
|
|
if len(lines) < 10:
|
|
sp = shell(prompt)
|
|
if sp.returncode != 0:
|
|
self.logger.error('Failed to execute ps command (Error: %s)', sp.stdout.decode('utf-8').strip())
|
|
return processes
|
|
lines = sp.stdout.decode('utf-8').splitlines()
|
|
|
|
# Iterate through lines starting from the second line (skipping header)
|
|
for line in lines[1:]:
|
|
try:
|
|
parts = line.split() # USER,PID,PPID,VSZ,RSS,WCHAN,ADDR,S,NAME
|
|
pid = int(parts[1]) # Extract PID
|
|
name = ' '.join(parts[8:]).strip() # Extract process name
|
|
|
|
# Handle cases where the name might be in brackets
|
|
name = name if name.startswith('[') else Path(name).name
|
|
processes[name] = pid
|
|
except Exception as e:
|
|
pass
|
|
|
|
return processes
|
|
|
|
def install_application(self, path: Path = None, url: str = None) -> bool:
|
|
"""
|
|
Installs an application from a local file path or a URL.
|
|
|
|
Args:
|
|
path (Path, optional): The local file path of the APK to install. Defaults to None.
|
|
url (str, optional): The URL to download the APK from. Defaults to None.
|
|
|
|
Returns:
|
|
bool: True if the installation was successful, False otherwise.
|
|
"""
|
|
prompt = [*self.prompt[:-1], 'install']
|
|
|
|
# Install from a local file path if provided
|
|
if path and path.is_file():
|
|
sp = shell([*prompt, path])
|
|
if sp.returncode == 0:
|
|
return True
|
|
self.logger.error('Installation failed for local path: %s (Error: %s)', path, sp.stdout.decode('utf-8').strip())
|
|
|
|
# Install from a URL if provided
|
|
status = False
|
|
if url:
|
|
file = Path('tmp.apk')
|
|
try:
|
|
r = requests.get(url, headers={'Accept': '*/*', 'User-Agent': 'KeyDive/ADB'})
|
|
r.raise_for_status()
|
|
|
|
# Write downloaded content to temporary APK file
|
|
file.write_bytes(r.content)
|
|
|
|
# Attempt installation from the downloaded file
|
|
status = self.install_application(path=file)
|
|
except Exception as e:
|
|
self.logger.error('Failed to download application from URL: %s (Error: %s)', url, e)
|
|
file.unlink(missing_ok=True)
|
|
|
|
return status
|
|
|
|
def open_url(self, url: str) -> bool:
|
|
"""
|
|
Opens a specified URL on the connected device.
|
|
|
|
Args:
|
|
url (str): The URL to open on the device.
|
|
|
|
Returns:
|
|
bool: True if the URL was opened successfully, False otherwise.
|
|
"""
|
|
# Execute the shell command to open the URL
|
|
sp = shell([*self.prompt, 'am', 'start', '-a', 'android.intent.action.VIEW', '-d', url])
|
|
|
|
# Check the result and log accordingly
|
|
if sp.returncode != 0:
|
|
self.logger.error('URL open failed for: %s (Return: %s)', url, sp.stdout.decode('utf-8').strip())
|
|
return False
|
|
return True
|
|
|
|
|
|
__all__ = ('ADB',)
|