#!/usr/bin/env python3 # MTK Flash Client (c) G.Kreileman, B.Kerler 2021. # Licensed under GPLv3 License import sys import time import mock import traceback import math import logging import ctypes from PySide2.QtCore import Qt, QVariantAnimation, Signal, QObject, QSize, QTranslator, QLocale, QLibraryInfo from PySide2.QtGui import QTextOption, QPixmap, QTransform, QIcon from PySide2.QtWidgets import QMainWindow, QApplication, QAction, QWidget from mtkclient.Library.mtk import Mtk from mtkclient.Library.mtk_da_cmd import DA_handler from mtkclient.Library.gpt import gpt_settings from mtkclient.Library.mtk_main import Main, Mtk_Config from mtkclient.gui.readFlashPartitions import ReadFlashWindow from mtkclient.gui.readFull import ReadFullFlashWindow from mtkclient.gui.toolsMenu import generateKeysMenu from mtkclient.gui.toolkit import asyncThread, trap_exc_during_debug from mtkclient.config.payloads import pathconfig from mtkclient.gui.main_gui import Ui_MainWindow import os os.environ['QT_MAC_WANTS_LAYER'] = '1' #This fixes a bug in pyside2 on MacOS Big Sur # TO do Move all GUI modifications to signals! # install exception hook: without this, uncaught exception would cause application to exit sys.excepthook = trap_exc_during_debug # Initiate MTK classes variables = mock.Mock() variables.cmd = "stage" variables.debugmode = True path=pathconfig() # if sys.platform.startswith('darwin'): # config.ptype = "kamakiri" #Temp for Mac testing MtkTool = Main(variables) guiState = "welcome" phoneInfo = {"chipset": "", "bootMode": "", "daInit": False, "cdcInit": False} class DeviceHandler(QObject): sendToLogSignal = Signal(str) mtkClass = None da_handler = None def __init__(self, parent, preloader: str = None, loglevel=logging.INFO, *args, **kwargs): super().__init__(parent, *args, **kwargs) config = Mtk_Config(loglevel=logging.INFO, gui=self.sendToLogSignal) config.gpt_settings = gpt_settings(gpt_num_part_entries='0', gpt_part_entry_size='0', gpt_part_entry_start_lba='0') # This actually sets the right GPT settings.. self.mtkClass = Mtk(config=config, loglevel=logging.INFO) self.loglevel = logging.DEBUG self.da_handler = DA_handler(self.mtkClass, loglevel) def getDevInfo(self, parameters): loglevel = parameters[0] phoneInfo = parameters[1] devhandler = parameters[2] mtkClass = devhandler.mtkClass da_handler = devhandler.da_handler try: if not mtkClass.port.cdc.connect(): mtkClass.preloader.init() else: phoneInfo['cdcInit'] = True except: phoneInfo['cantConnect'] = True phoneInfo['chipset'] = str(mtkClass.config.chipconfig.name) + \ " (" + str(mtkClass.config.chipconfig.description) + ")" self.sendUpdateSignal.emit() mtkClass = da_handler.configure_da(mtkClass, preloader=None) if mtkClass: phoneInfo['daInit'] = True phoneInfo['chipset'] = str(mtkClass.config.chipconfig.name) + \ " (" + str(mtkClass.config.chipconfig.description) + ")" if mtkClass.config.is_brom: phoneInfo['bootMode'] = "Bootrom mode" elif mtkClass.config.chipconfig.damode: phoneInfo['bootMode'] = "DA mode" else: phoneInfo['bootMode'] = "Preloader mode" self.sendUpdateSignal.emit() else: phoneInfo['cantConnect'] = True self.sendUpdateSignal.emit() # try: # print(mtkClass.daloader.get_partition_data(parttype="user")) # except Exception: # print(traceback.format_exc()) # MtkTool.cmd_stage(mtkClass, None, None, None, False) def load_translations(application): """ Load application translations and the QT base translations for the current locale :param application: QApplication object """ locale = QLocale.system() translator = QTranslator(application) directory = os.path.dirname(__file__) lang = 'mtkclient/gui/i18n/' + locale.name() #lang = 'mtkclient/gui/i18n/fr_FR' #lang = 'mtkclient/gui/i18n/de_DE' #lang = 'mtkclient/gui/i18n/en_EN' if translator.load(lang, directory): application.installTranslator(translator) translations_path = QLibraryInfo.location(QLibraryInfo.TranslationsPath) base_translator = QTranslator(application) if base_translator.load(locale, "qtbase", "_", translations_path): application.installTranslator(base_translator) class MainWindow(QMainWindow): def __init__(self): super(MainWindow, self).__init__() w = QWidget(self) self.ui=Ui_MainWindow() self.ui.setupUi(self) # Menubar self.ui.readFlashMenu.triggered[QAction].connect(self.openReadflashWindow) self.ui.toolsFlashMenu.triggered[QAction].connect(self.openToolsMenu) self.initpixmap() self.ui.debugBtn.clicked.connect(self.showDebugInfo) self.ui.logBox.setWordWrapMode(QTextOption.NoWrap) self.ui.menubar.setEnabled(False) def sendToLog(self, info): t = time.localtime() self.ui.logBox.appendPlainText(time.strftime("[%H:%M:%S", t) + "]: " + info) self.ui.logBox.verticalScrollBar().setValue(self.ui.logBox.verticalScrollBar().maximum()) def updateGui(self): global phoneInfo if phoneInfo['cdcInit'] and phoneInfo['bootMode'] == "": self.ui.phoneInfoTextbox.setText("Phone detected:\nReading model info...") else: self.ui.phoneInfoTextbox.setText( "Phone detected:\n" + phoneInfo['chipset'] + "\n" + phoneInfo['bootMode']) self.ui.status.setText("Device detected, please wait.\nThis can take a while...") if phoneInfo['daInit']: self.ui.status.setText("Device connected :)") self.ui.menubar.setEnabled(True) self.pixmap = QPixmap(path.get_images_path("phone_connected.png")) self.ui.pic.setPixmap(self.pixmap) self.spinnerAnim.stop() self.ui.spinner_pic.setHidden(True) else: if 'cantConnect' in phoneInfo: self.ui.status.setText("Error initialising. Did you install the drivers?") self.spinnerAnim.start() self.ui.spinner_pic.setHidden(False) def spinnerAnimRot(self, angle): trans = QTransform() dimension = self.pixmap.width() / math.sqrt(2) newPixmap = self.pixmap.transformed(QTransform().rotate(angle), Qt.SmoothTransformation) xoffset = (newPixmap.width() - self.pixmap.width())/2 yoffset = (newPixmap.height() - self.pixmap.height())/2 rotated = newPixmap.copy(xoffset, yoffset, self.pixmap.width(), self.pixmap.height()) self.ui.spinner_pic.setPixmap(rotated) def initpixmap(self): # phone spinner self.pixmap = QPixmap(path.get_images_path("phone_loading.png")).scaled(96, 96, Qt.KeepAspectRatio, Qt.SmoothTransformation) self.pixmap.setDevicePixelRatio(2) self.ui.spinner_pic.setPixmap(self.pixmap) self.ui.spinner_pic.show() nfpixmap = QPixmap(path.get_images_path("phone_notfound.png")) self.ui.pic.setPixmap(nfpixmap) logo = QPixmap(path.get_images_path("logo_256.png")) self.ui.logoPic.setPixmap(logo) self.ui.logBox.setHidden(True) self.spinnerAnim = QVariantAnimation() self.spinnerAnim.setDuration(3000) self.spinnerAnim.setStartValue(0) self.spinnerAnim.setEndValue(360) self.spinnerAnim.setLoopCount(-1) self.spinnerAnim.valueChanged.connect(self.spinnerAnimRot) self.ui.spinner_pic.setHidden(True) def openReadflashWindow(self, q): # status.setText("OH YEAH"+str(q.text())) curmenutext = q.text() info=self.tr("Read partition(s)") if curmenutext == self.tr("Read partition(s)"): readFlashWindowVal = ReadFlashWindow(self, devhandler.mtkClass, devhandler.da_handler, self.sendToLog) elif curmenutext == self.tr("Read full flash"): readFlashWindowVal = ReadFullFlashWindow(self, devhandler.mtkClass, devhandler.da_handler, self.sendToLog) # w.close() def openToolsMenu(self, q): # status.setText("OH YEAH"+str(q.text())) curmenutext=q.text() if curmenutext == self.tr("Generate RPMB keys"): readFlashWindowVal = generateKeysMenu(self, devhandler.mtkClass, devhandler.da_handler, self.sendToLog) # w.close() def showDebugInfo(self): if self.ui.logBox.isHidden(): self.ui.logBox.setHidden(False) #win.setFixedSize(746, 600 + addTopMargin) self.ui.debugBtn.setText(u"Hide debug info") else: self.ui.logBox.setHidden(True) #win.setFixedSize(746, 400 + addTopMargin) self.ui.debugBtn.setText(u"Show debug info") if __name__ == '__main__': # Enable nice 4K Scaling QApplication.setAttribute(Qt.AA_EnableHighDpiScaling, True) os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1" # Init the app window app = QApplication(sys.argv) load_translations(app) win = MainWindow() icon = QIcon() icon.addFile(path.get_images_path('logo_32.png'), QSize(32, 32)) icon.addFile(path.get_images_path('logo_64.png'), QSize(64, 64)) icon.addFile(path.get_images_path('logo_256.png'), QSize(256, 256)) icon.addFile(path.get_images_path('logo_512.png'), QSize(512, 512)) app.setWindowIcon(icon) win.setWindowIcon(icon) if sys.platform.startswith('win'): ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID('MTKTools.Gui') dpiMultiplier = win.logicalDpiX() if dpiMultiplier == 72: dpiMultiplier = 2 else: dpiMultiplier = 1 addTopMargin = 20 if sys.platform.startswith('darwin'): # MacOS has the toolbar in the top bar insted of in the app... addTopMargin = 0 win.setWindowTitle("MTKClient - Version 2.0 beta") # lay = QVBoxLayout(self) win.show() #win.setFixedSize(746, 400 + addTopMargin) # Device setup loglevel = logging.INFO devhandler = DeviceHandler(parent=app,preloader=None,loglevel=loglevel) devhandler.sendToLogSignal.connect(win.sendToLog) # Get the device info thread = asyncThread(parent=app, n=0, function=getDevInfo, parameters=[loglevel,phoneInfo,devhandler]) thread.sendToLogSignal.connect(win.sendToLog) thread.sendUpdateSignal.connect(win.updateGui) thread.start() # Run loop the app app.exec_() # Prevent thread from not being closed and call error end codes thread.terminate() thread.wait()