Files
Mobile-Security-Framework-M…/mobsf/StaticAnalyzer/views/common/shared_func.py
T
Ajin Abraham e08524fe81 [HOTFIX][EFR-08] Dylib + Symbols + Other Features (#2239)
* Dylib analysis support + PDF for iOS Binary
* Dylib string extraction
* Improved iOS Plist secret extraction
* iOS/Android Form Validation QA
* Independent Dylib scan
* Symbols view for dylib and so
* Trackers support for so
2023-08-18 17:53:45 -07:00

278 lines
9.0 KiB
Python
Executable File

# -*- coding: utf_8 -*-
"""
Shared Functions.
Module providing the shared functions for iOS and Android
"""
import io
import hashlib
import logging
import platform
import re
import shutil
import subprocess
import zipfile
from urllib.parse import urlparse
from pathlib import Path
import requests
from django.utils.html import escape
from mobsf.MobSF import settings
from mobsf.MobSF.utils import (
is_md5,
print_n_send_error_response,
upstream_proxy,
)
from mobsf.StaticAnalyzer.views.comparer import (
generic_compare,
)
from mobsf.StaticAnalyzer.views.common.entropy import (
get_entropies,
)
logger = logging.getLogger(__name__)
# Regex to capture strings between quotes or <string> tag
STRINGS_REGEX = re.compile(r'(?<=\")(.+?)(?=\")|(?<=\<string>)(.+?)(?=\<)')
# MobSF Custom regex to catch maximum URI like strings
URL_REGEX = re.compile(
(
r'((?:https?://|s?ftps?://|'
r'file://|javascript:|data:|www\d{0,3}[.])'
r'[\w().=/;,#:@?&~*+!$%\'{}-]+)'
),
re.UNICODE)
EMAIL_REGEX = re.compile(r'[\w.-]{1,20}@[\w-]{1,20}\.[\w]{2,10}')
def hash_gen(app_path) -> tuple:
"""Generate and return sha1 and sha256 as a tuple."""
try:
logger.info('Generating Hashes')
sha1 = hashlib.sha1()
sha256 = hashlib.sha256()
block_size = 65536
with io.open(app_path, mode='rb') as afile:
buf = afile.read(block_size)
while buf:
sha1.update(buf)
sha256.update(buf)
buf = afile.read(block_size)
sha1val = sha1.hexdigest()
sha256val = sha256.hexdigest()
return sha1val, sha256val
except Exception:
logger.exception('Generating Hashes')
def unzip(app_path, ext_path):
logger.info('Unzipping')
try:
files = []
with zipfile.ZipFile(app_path, 'r') as zipptr:
for fileinfo in zipptr.infolist():
filename = fileinfo.filename
if not isinstance(filename, str):
filename = str(
filename, encoding='utf-8', errors='replace')
files.append(filename)
zipptr.extract(filename, ext_path)
return files
except Exception:
logger.exception('Unzipping Error')
if platform.system() == 'Windows':
logger.info('Not yet Implemented.')
else:
logger.info('Using the Default OS Unzip Utility.')
try:
unzip_b = shutil.which('unzip')
subprocess.call(
[unzip_b, '-o', '-q', app_path, '-d', ext_path])
dat = subprocess.check_output([unzip_b, '-qq', '-l', app_path])
dat = dat.decode('utf-8').split('\n')
files_det = ['Length Date Time Name']
files_det = files_det + dat
return files_det
except Exception:
logger.exception('Unzipping Error')
def url_n_email_extract(dat, relative_path):
"""Extract URLs and Emails from Source Code."""
urls = []
emails = []
urllist = []
url_n_file = []
email_n_file = []
# URL Extraction
urllist = URL_REGEX.findall(dat.lower())
uflag = 0
for url in urllist:
if url not in urls:
urls.append(url)
uflag = 1
if uflag == 1:
url_n_file.append(
{'urls': urls, 'path': escape(relative_path)})
# Email Extraction
eflag = 0
for email in EMAIL_REGEX.findall(dat.lower()):
if (email not in emails) and (not email.startswith('//')):
emails.append(email)
eflag = 1
if eflag == 1:
email_n_file.append(
{'emails': emails, 'path': escape(relative_path)})
return urllist, url_n_file, email_n_file
# This is just the first sanity check that triggers generic_compare
def compare_apps(request, hash1: str, hash2: str, api=False):
if hash1 == hash2:
error_msg = 'Results with same hash cannot be compared'
return print_n_send_error_response(request, error_msg, api)
# Second Validation for REST API
if not (is_md5(hash1) and is_md5(hash2)):
error_msg = 'Invalid hashes'
return print_n_send_error_response(request, error_msg, api)
logger.info(
'Starting App compare for %s and %s', hash1, hash2)
return generic_compare(request, hash1, hash2, api)
def get_avg_cvss(findings):
# Average CVSS Score
cvss_scores = []
avg_cvss = 0
for finding in findings.values():
find = finding.get('metadata')
if not find:
# Hack to support iOS Binary Scan Results
find = finding
if find.get('cvss'):
if find['cvss'] != 0:
cvss_scores.append(find['cvss'])
if cvss_scores:
avg_cvss = round(sum(cvss_scores) / len(cvss_scores), 1)
if not getattr(settings, 'CVSS_SCORE_ENABLED', False):
avg_cvss = None
return avg_cvss
def open_firebase(url):
# Detect Open Firebase Database
try:
purl = urlparse(url)
base_url = '{}://{}/.json'.format(purl.scheme, purl.netloc)
proxies, verify = upstream_proxy('https')
headers = {
'User-Agent': ('Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1)'
' AppleWebKit/537.36 (KHTML, like Gecko) '
'Chrome/39.0.2171.95 Safari/537.36')}
resp = requests.get(base_url, headers=headers,
proxies=proxies, verify=verify)
if resp.status_code == 200:
return base_url, True
except Exception:
logger.warning('Open Firebase DB detection failed.')
return url, False
def firebase_analysis(urls):
# Detect Firebase URL
firebase_db = []
logger.info('Detecting Firebase URL(s)')
for url in urls:
if 'firebaseio.com' in url:
returl, is_open = open_firebase(url)
fbdic = {'url': returl, 'open': is_open}
if fbdic not in firebase_db:
firebase_db.append(fbdic)
return firebase_db
def find_java_source_folder(base_folder: Path):
# Find the correct java/kotlin source folder for APK/source zip
# Returns a Tuple of - (SRC_PATH, SRC_TYPE, SRC_SYNTAX)
return next(p for p in [(base_folder / 'java_source',
'java', '*.java'),
(base_folder / 'app' / 'src' / 'main' / 'java',
'java', '*.java'),
(base_folder / 'app' / 'src' / 'main' / 'kotlin',
'kotlin', '*.kt'),
(base_folder / 'src',
'java', '*.java')]
if p[0].exists())
def is_secret_key(inp):
"""Check if the key in the key/value pair is interesting."""
inp = inp.lower()
iden = (
'api"', 'key"', 'api_', 'key_', 'secret"',
'password"', 'aws', 'gcp', 's3_', '_s3', 'secret_',
'token"', 'username"', 'user_name"', 'user"',
'bearer', 'jwt', 'certificate"', 'credential',
'azure', 'webhook', 'twilio_', 'bitcoin',
'_auth', 'firebase', 'oauth', 'authorization',
'private', 'pwd', 'session', 'token_',
)
not_string = (
'label_', 'text', 'hint', 'msg_', 'create_',
'message', 'new', 'confirm', 'activity_',
'forgot', 'dashboard_', 'current_', 'signup',
'sign_in', 'signin', 'title_', 'welcome_',
'change_', 'this_', 'the_', 'placeholder',
'invalid_', 'btn_', 'action_', 'prompt_',
'lable', 'hide_', 'old', 'update', 'error',
'empty', 'txt_', 'lbl_',
)
not_str = any(i in inp for i in not_string)
return any(i in inp for i in iden) and not not_str
def strings_and_entropies(src, exts):
"""Get Strings and Entropies."""
logger.info('Extracting Data from Source Code')
data = {
'strings': set(),
'secrets': set(),
}
try:
if not src.exists():
return data
excludes = ('\\u0', 'com.google.')
eslash = ('Ljava', 'Lkotlin', 'kotlin', 'android')
for p in src.rglob('*'):
if p.suffix not in exts or not p.exists():
continue
matches = STRINGS_REGEX.finditer(
p.read_text(encoding='utf-8', errors='ignore'),
re.MULTILINE)
for match in matches:
string = match.group()
if len(string) < 4:
continue
if any(i in string for i in excludes):
continue
if any(i in string and '/' in string for i in eslash):
continue
if not string[0].isalnum():
continue
data['strings'].add(string)
if data['strings']:
data['secrets'] = get_entropies(data['strings'])
except Exception:
logger.exception('Extracting Data from Code')
return data
def get_symbols(symbols):
for i in symbols:
for _, val in i.items():
return val
return []