mirror of
https://github.com/mvt-project/mvt.git
synced 2024-06-18 18:38:58 +00:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
# Mobile Verification Toolkit (MVT)
|
|
# Copyright (c) 2021-2022 Claudio Guarnieri.
|
|
# See the file 'LICENSE' for usage and copying permissions, or find a copy at
|
|
# https://github.com/mvt-project/mvt/blob/main/LICENSE
|
|
|
|
import fnmatch
|
|
import logging
|
|
import os
|
|
|
|
from mvt.common.module import MVTModule
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class BugReportModule(MVTModule):
|
|
"""This class provides a base for all Android Bug Report modules."""
|
|
|
|
zip_archive = None
|
|
|
|
def from_folder(self, extract_path, extract_files):
|
|
self.extract_path = extract_path
|
|
self.extract_files = extract_files
|
|
|
|
def from_zip(self, zip_archive, zip_files):
|
|
self.zip_archive = zip_archive
|
|
self.zip_files = zip_files
|
|
|
|
def _get_files_by_pattern(self, pattern):
|
|
file_names = []
|
|
if self.zip_archive:
|
|
for zip_file in self.zip_files:
|
|
file_names.append(zip_file)
|
|
else:
|
|
file_names = self.extract_files
|
|
|
|
return fnmatch.filter(file_names, pattern)
|
|
|
|
def _get_file_content(self, file_path):
|
|
if self.zip_archive:
|
|
handle = self.zip_archive.open(file_path)
|
|
else:
|
|
handle = open(os.path.join(self.extract_path, file_path), "rb")
|
|
|
|
data = handle.read()
|
|
handle.close()
|
|
|
|
return data
|