mirror of https://github.com/mvt-project/mvt.git
Sorting imports and removing unused ones
This commit is contained in:
parent
4a568835d2
commit
7222bc82e1
|
@ -5,8 +5,9 @@
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import sys
|
import sys
|
||||||
from pymobiledevice3.lockdown import LockdownClient
|
|
||||||
from pymobiledevice3.exceptions import ConnectionFailedError
|
from pymobiledevice3.exceptions import ConnectionFailedError
|
||||||
|
from pymobiledevice3.lockdown import LockdownClient
|
||||||
|
|
||||||
from mvt.common.command import Command
|
from mvt.common.command import Command
|
||||||
|
|
||||||
|
|
|
@ -4,7 +4,9 @@
|
||||||
# https://license.mvt.re/1.1/
|
# https://license.mvt.re/1.1/
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
from pymobiledevice3.services.installation_proxy import InstallationProxyService
|
|
||||||
|
from pymobiledevice3.services.installation_proxy import \
|
||||||
|
InstallationProxyService
|
||||||
|
|
||||||
from .base import IOSUSBExtraction
|
from .base import IOSUSBExtraction
|
||||||
|
|
||||||
|
|
|
@ -3,18 +3,9 @@
|
||||||
# Use of this software is governed by the MVT License 1.1 that can be found at
|
# Use of this software is governed by the MVT License 1.1 that can be found at
|
||||||
# https://license.mvt.re/1.1/
|
# https://license.mvt.re/1.1/
|
||||||
|
|
||||||
import base64
|
|
||||||
import getpass
|
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import random
|
|
||||||
import string
|
|
||||||
import sys
|
|
||||||
import tempfile
|
|
||||||
import time
|
|
||||||
from typing import Callable
|
|
||||||
|
|
||||||
from mvt.common.module import InsufficientPrivileges, MVTModule
|
from mvt.common.module import MVTModule
|
||||||
|
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
|
@ -3,8 +3,8 @@
|
||||||
# Use of this software is governed by the MVT License 1.1 that can be found at
|
# Use of this software is governed by the MVT License 1.1 that can be found at
|
||||||
# https://license.mvt.re/1.1/
|
# https://license.mvt.re/1.1/
|
||||||
|
|
||||||
import logging
|
|
||||||
import base64
|
import base64
|
||||||
|
import logging
|
||||||
|
|
||||||
from mvt.ios.versions import latest_ios_version
|
from mvt.ios.versions import latest_ios_version
|
||||||
|
|
||||||
|
|
|
@ -4,6 +4,7 @@
|
||||||
# https://license.mvt.re/1.1/
|
# https://license.mvt.re/1.1/
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from pymobiledevice3.services.os_trace import OsTraceService
|
from pymobiledevice3.services.os_trace import OsTraceService
|
||||||
|
|
||||||
from .base import IOSUSBExtraction
|
from .base import IOSUSBExtraction
|
||||||
|
|
|
@ -5,14 +5,14 @@
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
from pymobiledevice3.lockdown import LockdownClient
|
||||||
|
|
||||||
from mvt.common.module import run_module
|
from mvt.common.module import run_module
|
||||||
from mvt.ios.modules.usb.applications import Applications
|
from mvt.ios.modules.usb.applications import Applications
|
||||||
from pymobiledevice3.lockdown import LockdownClient
|
|
||||||
|
|
||||||
|
|
||||||
class TestUSBApplication:
|
class TestUSBApplication:
|
||||||
def test_run(self, mocker):
|
def test_run(self, mocker):
|
||||||
# Mock
|
|
||||||
mocker.patch("pymobiledevice3.lockdown.LockdownClient.start_service")
|
mocker.patch("pymobiledevice3.lockdown.LockdownClient.start_service")
|
||||||
mocker.patch("pymobiledevice3.usbmux.select_device")
|
mocker.patch("pymobiledevice3.usbmux.select_device")
|
||||||
mocker.patch("pymobiledevice3.service_connection.ServiceConnection.create")
|
mocker.patch("pymobiledevice3.service_connection.ServiceConnection.create")
|
||||||
|
|
|
@ -5,14 +5,14 @@
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
from pymobiledevice3.lockdown import LockdownClient
|
||||||
|
|
||||||
from mvt.common.module import run_module
|
from mvt.common.module import run_module
|
||||||
from mvt.ios.modules.usb.device_info import DeviceInfo
|
from mvt.ios.modules.usb.device_info import DeviceInfo
|
||||||
from pymobiledevice3.lockdown import LockdownClient
|
|
||||||
|
|
||||||
|
|
||||||
class TestUSBDeviceInfo:
|
class TestUSBDeviceInfo:
|
||||||
def test_run(self, mocker):
|
def test_run(self, mocker):
|
||||||
# Mock
|
|
||||||
mocker.patch("pymobiledevice3.usbmux.select_device")
|
mocker.patch("pymobiledevice3.usbmux.select_device")
|
||||||
mocker.patch("pymobiledevice3.service_connection.ServiceConnection.create")
|
mocker.patch("pymobiledevice3.service_connection.ServiceConnection.create")
|
||||||
mocker.patch(
|
mocker.patch(
|
||||||
|
|
|
@ -5,21 +5,19 @@
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
from mvt.common.indicators import Indicators
|
||||||
from mvt.common.module import run_module
|
from mvt.common.module import run_module
|
||||||
from mvt.ios.modules.usb.processes import Processes
|
from mvt.ios.modules.usb.processes import Processes
|
||||||
from mvt.common.indicators import Indicators
|
|
||||||
|
|
||||||
|
|
||||||
class TestUSBProcesses:
|
class TestUSBProcesses:
|
||||||
def test_run(self, mocker, indicator_file):
|
def test_run(self, mocker, indicator_file):
|
||||||
# Mock
|
|
||||||
mocker.patch("pymobiledevice3.services.base_service.BaseService.__init__")
|
mocker.patch("pymobiledevice3.services.base_service.BaseService.__init__")
|
||||||
mocker.patch(
|
mocker.patch(
|
||||||
"pymobiledevice3.services.os_trace.OsTraceService.get_pid_list",
|
"pymobiledevice3.services.os_trace.OsTraceService.get_pid_list",
|
||||||
return_value={"Payload": {"1": {"ProcessName": "storebookkeeperd"}, "1854": {"ProcessName": "cfprefssd"}}}
|
return_value={"Payload": {"1": {"ProcessName": "storebookkeeperd"}, "1854": {"ProcessName": "cfprefssd"}}}
|
||||||
)
|
)
|
||||||
|
|
||||||
# indicators
|
|
||||||
ind = Indicators(log=logging)
|
ind = Indicators(log=logging)
|
||||||
ind.parse_stix2(indicator_file)
|
ind.parse_stix2(indicator_file)
|
||||||
ind.ioc_collections[0]["processes"].append("cfprefssd")
|
ind.ioc_collections[0]["processes"].append("cfprefssd")
|
||||||
|
|
Loading…
Reference in New Issue