def setupTree(self): try: self.ui.tree_area.layout().deleteLater() except AttributeError: pass self.tree = TreeWindow(self, self, session=self.session) self.tree.setWindowTitle("Tree model") layout = QtGui.QVBoxLayout() layout.addWidget(self.tree) self.ui.tree_area.setLayout(layout) self.tree.fill() self.setupCentral()
def setupTree(self): try: self.ui.tree_area.layout().deleteLater() except AttributeError: pass self.tree = TreeWindow(self,self,session=self.session) self.tree.setWindowTitle("Tree model") layout = QtGui.QVBoxLayout() layout.addWidget(self.tree) self.ui.tree_area.setLayout(layout) self.tree.fill() self.setupCentral()
class MainWindow(QtWidgets.QMainWindow): """Main window: self.central: QTabWidget in center area self.dock: QDockWidget in left area self.tree: TreeWindow(QTreeWidget) in self.dock """ def __init__(self, parent=None, session=session_module.Session(), input_file=None, input_plugin=None): super(MainWindow, self).__init__(parent) self.session = session self.bin_windows = {} self.setupFileMenu() self.setupViewMenu() self.setupPluginsMenu() self.setupHelpMenu() self.setupCentral() self.setupEmptyTree() self.setupDock() self.setupSession() self.setWindowTitle("Androguard GUI") self.showStatus("Androguard GUI") self.installEventFilter(self) self.input_plugin = input_plugin if input_file: self._openFile(input_file) def eventFilter(self, watched, event): for bin_window in self.bin_windows.values(): bin_window.eventFilter(watched, event) return False def showStatus(self, msg): """Helper function called by any window to display a message in status bar. """ androconf.debug(msg) self.statusBar().showMessage(msg) def about(self): """User clicked About menu. Display a Message box.""" QtWidgets.QMessageBox.about( self, "About Androguard GUI", "<p><b>Androguard GUI</b> is basically a GUI for Androguard :)." "<br>Have fun !</p>", ) def setupSession(self): androconf.debug("Setup Session") self.fileLoadingThread = FileLoadingThread(self) self.fileLoadingThread.file_loaded.connect(self.loadedFile) def loadedFile(self, success): if not success: self.showStatus("Analysis of %s failed :(" % str(self.fileLoadingThread.file_path)) return self.updateDockWithTree() self.cleanCentral() self.showStatus("Analysis of %s done!" % str(self.fileLoadingThread.file_path)) if self.input_plugin: self._runPlugin(self.input_plugin) def openFile(self): self.session.reset() filepath, _ = QtWidgets.QFileDialog.getOpenFileName( self, "Open File", ".", "Android Files (*.apk *.jar *.dex *.odex *.dey);;Androguard Session (*.ag)" ) self._openFile(filepath) def _openFile(self, filepath=None): if filepath: self.setupTree() self.showStatus("Analyzing %s..." % str(filepath)) self.fileLoadingThread.load(filepath) def addFile(self): if not self.session.isOpen(): return filepath, _ = QtWidgets.QFileDialog.getOpenFileName( self, "Add File", "", "Android Files (*.apk *.jar *.dex *.odex *.dey)" ) if filepath: self.showStatus("Analyzing %s..." % str(filepath)) self.fileLoadingThread.load(filepath) def saveFile(self): """User clicked Save menu. Display a Dialog to ask whwre to save.""" filepath, _ = QtWidgets.QFileDialog.getSaveFileName(self, "Save File", "", "Androguard Session (*.ag)") if filepath: self.showStatus("Saving %s..." % str(filepath)) self.saveSession(filepath) def saveSession(self, filepath): """Save androguard session.""" try: session_module.Save(self.session, filepath) except RuntimeError as e: androconf.error(str(e)) os.remove(filepath) androconf.warning("Session not saved") def _runPlugin(self, filepath): androconf.debug("RUN plugin from %s" % filepath) module_name = os.path.splitext(os.path.basename(filepath))[0] f, filename, description = imp.find_module(module_name, [os.path.dirname(filepath)]) print f, filename, description mod = imp.load_module(module_name, f, filename, description) mod.PluginEntry(self.session) def openRunPluginWindow(self): filepath, _ = QtWidgets.QFileDialog.getOpenFileName(self, "Open File", "", "Python Files (*.py);;") if filepath: self._runPlugin(filepath) def closeEvent(self, event): """Clicked [x] to close main window""" event.accept() def setupEmptyTree(self): """Setup empty Tree at startup. """ if hasattr(self, "tree"): del self.tree self.tree = QtWidgets.QTreeWidget(self) self.tree.header().close() def setupDock(self): """Setup empty Dock at startup. """ self.dock = QtWidgets.QDockWidget("Classes", self) self.dock.setWidget(self.tree) self.dock.setFeatures(QtWidgets.QDockWidget.NoDockWidgetFeatures) self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.dock) def setupTree(self): androconf.debug("Setup Tree") self.tree = TreeWindow(win=self, session=self.session) self.tree.setWindowTitle("Tree model") self.dock.setWidget(self.tree) def setupCentral(self): """Setup empty window supporting tabs at startup. """ self.central = TabsWindow(self.bin_windows, self) self.setCentralWidget(self.central) def cleanCentral(self): self.central.actioncloseAllTabs() def setupFileMenu(self): androconf.debug("Setup File Menu") self.fileMenu = self.menuBar().addMenu("&File") self.fileMenu.addAction("&Open...", self.openFile, "Ctrl+O") self.fileMenu.addAction("&Add...", self.addFile, "Ctrl+A") self.fileMenu.addAction("&Save...", self.saveFile, "Ctrl+S") self.fileMenu.addAction("E&xit", self.close, "Ctrl+Q") def setupViewMenu(self): androconf.debug("Setup View Menu") self.viewMenu = self.menuBar().addMenu("&View") self.viewMenu.addAction("&Strings...", self.openStringsWindow) self.viewMenu.addAction("&Methods...", self.openMethodsWindow) self.viewMenu.addAction("&API...", self.openAPIWindow) self.viewMenu.addAction("&APK...", self.openApkWindow) self.viewMenu.addAction("&Resources...", self.openResourcesWindow) def setupPluginsMenu(self): androconf.debug("Setup Plugins Menu") self.pluginsMenu = self.menuBar().addMenu("&Plugins") self.pluginsMenu.addAction("&Run...", self.openRunPluginWindow) def setupHelpMenu(self): androconf.debug("Setup Help Menu") self.helpMenu = self.menuBar().addMenu("&Help") self.helpMenu.addAction("&About", self.about) self.helpMenu.addAction("About &Qt", QtWidgets.qApp.aboutQt) def updateDockWithTree(self, empty=False): """Update the classes tree. Called when - a new APK has been imported - a classe has been renamed (displayed in the tree) """ self.setupTree() self.tree.fill() def openStringsWindow(self): stringswin = StringsWindow(win=self, session=self.session) self.central.addTab(stringswin, stringswin.title) self.central.setTabToolTip(self.central.indexOf(stringswin), stringswin.title) self.central.setCurrentWidget(stringswin) def openMethodsWindow(self): methodswin = MethodsWindow(win=self, session=self.session) self.central.addTab(methodswin, methodswin.title) self.central.setTabToolTip(self.central.indexOf(methodswin), methodswin.title) self.central.setCurrentWidget(methodswin) def openResourcesWindow(self): resourceswin = ResourcesWindow(win=self, session=self.session) self.central.addTab(resourceswin, resourceswin.title) self.central.setTabToolTip(self.central.indexOf(resourceswin), resourceswin.title) self.central.setCurrentWidget(resourceswin) def openAPIWindow(self): apiwin = APIWindow(win=self, session=self.session) self.central.addTab(apiwin, apiwin.title) self.central.setTabToolTip(self.central.indexOf(apiwin), apiwin.title) self.central.setCurrentWidget(apiwin) def openApkWindow(self): androconf.debug("openApkWindow for %s" % self.session.analyzed_apk) bin_window = binWidget(self, ApkModel(self.session.get_objects_apk(self.fileLoadingThread.file_path)[0]), "APK") bin_window.activateWindow() self.central.addTab(bin_window, bin_window.title) self.central.setCurrentWidget(bin_window) self.bin_windows[bin_window.title] = bin_window def openBinWindow(self, current_class): androconf.debug("openBinWindow for %s" % current_class) bin_window = self.getMeOpenedWindowIfExists(current_class.current_title) if not bin_window: bin_window = binWidget(self, DexClassModel(current_class), current_class.get_name()) bin_window.activateWindow() self.central.addTab(bin_window, current_class.current_title) self.central.setTabToolTip(self.central.indexOf(bin_window), current_class.current_title) self.bin_windows[current_class.current_title] = bin_window bin_window.enable() self.central.setCurrentWidget(bin_window) def openSourceWindow(self, current_class, method=None): """Main function to open a decompile source window It checks if it already opened and open that tab, otherwise, initialize a new window. """ androconf.debug("openSourceWindow for %s" % current_class) sourcewin = self.getMeOpenedWindowIfExists(current_class.current_title + "(S)") if not sourcewin: current_filename = self.session.get_filename_by_class(current_class) current_digest = self.session.get_digest_by_class(current_class) sourcewin = SourceWindow( win=self, current_class=current_class, current_title=current_class.current_title + "(S)", current_filename=current_filename, current_digest=current_digest, session=self.session, ) sourcewin.reload_java_sources() self.central.addTab(sourcewin, sourcewin.title) self.central.setTabToolTip(self.central.indexOf(sourcewin), sourcewin.title) if method: sourcewin.browse_to_method(method) self.central.setCurrentWidget(sourcewin) def getMeOpenedWindowIfExists(self, name): for idx in range(self.central.count()): if name == self.central.tabToolTip(idx): androconf.debug("Tab %s already opened at: %d" % (name, idx)) return self.central.widget(idx) return None def doesClassExist(self, path): arg = class2func(path) try: getattr(self.d, arg) except AttributeError: return False return True
def setupTree(self): androconf.debug("Setup Tree") self.tree = TreeWindow(win=self, session=self.session) self.tree.setWindowTitle("Tree model") self.dock.setWidget(self.tree)
def setupTree(self): log.debug("Setup Tree") self.tree = TreeWindow(win=self, session=self.session) self.tree.setWindowTitle("Tree model") self.dock.setWidget(self.tree)
class MainWindow(QtWidgets.QMainWindow): """Main window: self.central: QTabWidget in center area self.dock: QDockWidget in left area self.tree: TreeWindow(QTreeWidget) in self.dock """ def __init__(self, parent=None, session=session_module.Session(), input_file=None, input_plugin=None): super(MainWindow, self).__init__(parent) self.session = session self.bin_windows = {} self.setupFileMenu() self.setupViewMenu() self.setupPluginsMenu() self.setupHelpMenu() self.setupCentral() self.setupEmptyTree() self.setupDock() self.setupSession() self.setWindowTitle("Androguard GUI") self.showStatus("Androguard GUI") self.installEventFilter(self) self.input_plugin = input_plugin if input_file: self._openFile(input_file) root = os.path.dirname(os.path.realpath(__file__)) self.setWindowIcon(QtGui.QIcon(os.path.join(root, "androguard.ico"))) def eventFilter(self, watched, event): for bin_window in list(self.bin_windows.values()): bin_window.eventFilter(watched, event) return False def showStatus(self, msg): """Helper function called by any window to display a message in status bar. """ log.debug(msg) self.statusBar().showMessage(msg) def about(self): """User clicked About menu. Display a Message box.""" QtWidgets.QMessageBox.about(self, "About Androguard GUI", "<p><b>Androguard GUI</b> is basically a GUI for Androguard :)." \ "<br>Have fun !</p>") def _no_apk_loaded(self): """Show a message if no APK was loaded yet...""" QtWidgets.QMessageBox.information( self, "No APK loaded yet!", "<p>There was no APK loaded yet. Please load one using File->Open.</p>" ) def setupSession(self): log.debug("Setup Session") self.fileLoadingThread = FileLoadingThread(self) self.fileLoadingThread.file_loaded.connect(self.loadedFile) def loadedFile(self, success): if not success: self.showStatus("Analysis of %s failed :(" % str(self.fileLoadingThread.file_path)) return self.updateDockWithTree() self.cleanCentral() self.showStatus("Analysis of %s done!" % str(self.fileLoadingThread.file_path)) if self.input_plugin: self._runPlugin(self.input_plugin) def openFile(self): self.session.reset() filepath, _ = QtWidgets.QFileDialog.getOpenFileName( self, "Open File", '.', "Android Files (*.apk *.jar *.dex *.odex *.dey);;Androguard Session (*.ag)" ) self._openFile(filepath) def _openFile(self, filepath=None): if filepath: self.setupTree() self.showStatus("Analyzing %s..." % str(filepath)) self.fileLoadingThread.load(filepath) def addFile(self): if not self.session.isOpen(): log.debug(self.session.analyzed_digest) self._no_apk_loaded() return filepath, _ = QtWidgets.QFileDialog.getOpenFileName( self, "Add File", '', "Android Files (*.apk *.jar *.dex *.odex *.dey)") if filepath: self.showStatus("Analyzing %s..." % str(filepath)) self.fileLoadingThread.load(filepath) def saveFile(self): """User clicked Save menu. Display a Dialog to ask whwre to save.""" filepath, _ = QtWidgets.QFileDialog.getSaveFileName( self, "Save File", '', "Androguard Session (*.ag)") if filepath: # Ensure .ag as file ending if not filepath.endswith(".ag"): filepath = "{}.ag".format(filepath) self.showStatus("Saving %s..." % str(filepath)) self.saveSession(filepath) self.showStatus("Saved Session to %s!" % str(filepath)) def saveSession(self, filepath): """Save androguard session.""" try: session_module.Save(self.session, filepath) except RuntimeError as e: log.exception(e) os.remove(filepath) log.warning("Session not saved") def _runPlugin(self, filepath): module_name = os.path.splitext(os.path.basename(filepath))[0] log.debug("RUN plugin '{}' from {}".format(module_name, filepath)) mod = load_module(module_name, filepath) log.debug("Loaded %s", mod) if not mod or not hasattr(mod, 'PluginEntry'): QtWidgets.QMessageBox.warning( self, "Not a valid Plugin", "<p>This python file does not look like a valid plugin.</p>") return mod.PluginEntry(self.session) def openRunPluginWindow(self): filepath, _ = QtWidgets.QFileDialog.getOpenFileName( self, "Open File", '', "Python Files (*.py);;") if filepath: self._runPlugin(filepath) def closeEvent(self, event): """Clicked [x] to close main window""" event.accept() def setupEmptyTree(self): """Setup empty Tree at startup. """ if hasattr(self, "tree"): del self.tree self.tree = QtWidgets.QTreeWidget(self) self.tree.header().close() def setupDock(self): """Setup empty Dock at startup. """ self.dock = QtWidgets.QDockWidget("Classes", self) self.dock.setWidget(self.tree) self.dock.setFeatures(QtWidgets.QDockWidget.NoDockWidgetFeatures) self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.dock) def setupTree(self): log.debug("Setup Tree") self.tree = TreeWindow(win=self, session=self.session) self.tree.setWindowTitle("Tree model") self.dock.setWidget(self.tree) def setupCentral(self): """Setup empty window supporting tabs at startup. """ self.central = TabsWindow(self.bin_windows, self) self.setCentralWidget(self.central) def cleanCentral(self): self.central.actioncloseAllTabs() def setupFileMenu(self): log.debug("Setup File Menu") self.fileMenu = self.menuBar().addMenu("&File") self.fileMenu.addAction("&Open...", self.openFile, "Ctrl+O") self.fileMenu.addAction("&Add...", self.addFile, "Ctrl+A") self.fileMenu.addAction("&Save...", self.saveFile, "Ctrl+S") self.fileMenu.addAction("E&xit", self.close, "Ctrl+Q") def setupViewMenu(self): log.debug("Setup View Menu") self.viewMenu = self.menuBar().addMenu("&View") self.viewMenu.addAction("&Strings...", self.openStringsWindow) self.viewMenu.addAction("&Methods...", self.openMethodsWindow) self.viewMenu.addAction("&API...", self.openAPIWindow) self.viewMenu.addAction("&APK...", self.openApkWindow) self.viewMenu.addAction("&Resources...", self.openResourcesWindow) def setupPluginsMenu(self): log.debug("Setup Plugins Menu") self.pluginsMenu = self.menuBar().addMenu("&Plugins") self.pluginsMenu.addAction("&Run...", self.openRunPluginWindow) def setupHelpMenu(self): log.debug("Setup Help Menu") self.helpMenu = self.menuBar().addMenu("&Help") self.helpMenu.addAction("&About", self.about) self.helpMenu.addAction("About &Qt", QtWidgets.qApp.aboutQt) def updateDockWithTree(self, empty=False): """Update the classes tree. Called when - a new APK has been imported - a classe has been renamed (displayed in the tree) """ self.setupTree() self.tree.fill() def openStringsWindow(self): stringswin = StringsWindow(win=self, session=self.session) self.central.addTab(stringswin, stringswin.title) self.central.setTabToolTip(self.central.indexOf(stringswin), stringswin.title) self.central.setCurrentWidget(stringswin) def openMethodsWindow(self): methodswin = MethodsWindow(win=self, session=self.session) self.central.addTab(methodswin, methodswin.title) self.central.setTabToolTip(self.central.indexOf(methodswin), methodswin.title) self.central.setCurrentWidget(methodswin) def openResourcesWindow(self): resourceswin = ResourcesWindow(win=self, session=self.session) self.central.addTab(resourceswin, resourceswin.title) self.central.setTabToolTip(self.central.indexOf(resourceswin), resourceswin.title) self.central.setCurrentWidget(resourceswin) def openAPIWindow(self): apiwin = APIWindow(win=self, session=self.session) self.central.addTab(apiwin, apiwin.title) self.central.setTabToolTip(self.central.indexOf(apiwin), apiwin.title) self.central.setCurrentWidget(apiwin) def openApkWindow(self): log.debug("openApkWindow for %s" % self.session.analyzed_apk) if not self.fileLoadingThread.file_path: self._no_apk_loaded() return bin_window = binWidget( self, ApkModel( self.session.get_objects_apk( self.fileLoadingThread.file_path)[0]), "APK") bin_window.activateWindow() self.central.addTab(bin_window, bin_window.title) self.central.setCurrentWidget(bin_window) self.bin_windows[bin_window.title] = bin_window def openBinWindow(self, current_class): log.debug("openBinWindow for %s" % current_class) dx = self.session.get_analysis(current_class) bin_window = self.getMeOpenedWindowIfExists( current_class.current_title) if not bin_window: bin_window = binWidget(self, DexClassModel(current_class, dx), current_class.get_name()) bin_window.activateWindow() self.central.addTab(bin_window, current_class.current_title) self.central.setTabToolTip(self.central.indexOf(bin_window), current_class.current_title) self.bin_windows[current_class.current_title] = bin_window bin_window.enable() self.central.setCurrentWidget(bin_window) def openSourceWindow(self, current_class, method=None): """Main function to open a decompile source window It checks if it already opened and open that tab, otherwise, initialize a new window. """ log.debug("openSourceWindow for %s" % current_class) sourcewin = self.getMeOpenedWindowIfExists( current_class.current_title + "(S)") if not sourcewin: current_filename = self.session.get_filename_by_class( current_class) current_digest = self.session.get_digest_by_class(current_class) sourcewin = SourceWindow( win=self, current_class=current_class, current_title=current_class.current_title + "(S)", current_filename=current_filename, current_digest=current_digest, session=self.session) sourcewin.reload_java_sources() self.central.addTab(sourcewin, sourcewin.title) self.central.setTabToolTip(self.central.indexOf(sourcewin), sourcewin.title) if method: sourcewin.browse_to_method(method) self.central.setCurrentWidget(sourcewin) def getMeOpenedWindowIfExists(self, name): for idx in range(self.central.count()): if name == self.central.tabToolTip(idx): log.debug("Tab %s already opened at: %d" % (name, idx)) return self.central.widget(idx) return None def doesClassExist(self, path): arg = class2func(path) try: getattr(self.d, arg) except AttributeError: return False return True
class MainView(QtGui.QMainWindow): def __init__(self, parent=None): super(MainView, self).__init__(parent) self.ui = Ui_MainWindow() self.ui.setupUi(self) self.__logger = Logger(self.ui.logArea) self.__logger.log(Logger.INFO, '') self.init_actions() self.setup_fileloading() self.apk_path = '' self.apk = None self.x = self.d = self.a = None self.manifest = None self.show_sample_question() def setup_fileloading(self): self.session = Session() self.fileLoadingThread = FileLoadingThread(self.session) self.connect(self.fileLoadingThread, QtCore.SIGNAL("loadedFile(bool)"), self.loadedApk) def show_sample_question(self): self.sample_dialog = SampleDialog(self) self.sample_dialog.exec_() def get_logger(self): return self.__logger def loadedApk(self, success): self.sample_dialog.close() if not success: self.__logger.log( Logger.ERROR, "Analysis of %s failed :(" % str(self.fileLoadingThread.file_path)) self.set_loading_progressbar_disabled() return self.d = self.session.get_DalvikForm() self.d.create_python_export() self.x = uVMAnalysis(self.d) self.setupTree() self.setupResourceController() self.load_app_info_table() self.load_permissions() try: self.__logger.log( Logger.INFO, "Analysis of %s done!" % str(self.apk.get_app_name())) self.ui.loadedAPK_label.setText("Loaded: " + str(self.apk.get_app_name())) except UnicodeEncodeError: self.ui.loadedAPK_label.setText( "APK Loaded! (non-ascii chars detected)") self.__logger.log( Logger.WARNING, "Non ascii code characters detected, some strings are possibly not displayed properly." ) self.set_loading_progressbar_disabled() def setupResourceController(self): self.resourcecontroller = ResourceFileController( self.apk, self.ui.resourceFileTable, self.ui.graphicview, self.ui.fileResourceInfo) def get_android_manifest_xml(self): self.set_loading_progressbar_text("Decompiling AndroidManifest.xml") buff = self.apk.get_android_manifest_xml().toprettyxml( encoding="utf-8") doc = QtGui.QTextEdit() doc.setWindowTitle("AndroidManifest.xml") hl = XMLHighlighter(doc.document()) doc.setPlainText(str(buff).strip()) return doc def setupTree(self): try: self.ui.tree_area.layout().deleteLater() except AttributeError: pass self.tree = TreeWindow(self, self, session=self.session) self.tree.setWindowTitle("Tree model") layout = QtGui.QVBoxLayout() layout.addWidget(self.tree) self.ui.tree_area.setLayout(layout) self.tree.fill() self.setupCentral() def setupCentral(self): self.central = QtGui.QTabWidget() self.central.setTabBar(CustomTabBar()) self.central.setTabsClosable(True) self.central.tabCloseRequested.connect(self.tabCloseRequestedHandler) self.central.currentChanged.connect(self.currentTabChanged) layout = QtGui.QVBoxLayout() layout.addWidget(self.central) self.ui.sourceTextWidget.setLayout(layout) def tabCloseRequestedHandler(self, index): self.central.removeTab(index) def currentTabChanged(self, index): if index == -1: return # all tab closed def openSourceWindow(self, current_class, method=None): sourcewin = self.getMeSourceWindowIfExists(current_class) if not sourcewin: current_filename = self.session.get_filename_by_class( current_class) current_digest = self.session.get_digest_by_class(current_class) sourcewin = SourceWindow(win=self, current_class=current_class, current_title=current_class.current_title, current_filename=current_filename, current_digest=current_digest, session=self.session) sourcewin.reload_java_sources() self.central.addTab(sourcewin, sourcewin.title) self.central.setTabToolTip(self.central.indexOf(sourcewin), current_class.get_name()) if method: sourcewin.browse_to_method(method) self.central.setCurrentWidget(sourcewin) def openManifestWindow(self): manifest_tab = self.get_android_manifest_xml() self.central.addTab(manifest_tab, "AndroidManifest.xml") self.central.setCurrentWidget(manifest_tab) def openStringsWindow(self): stringswin = StringsWindow(win=self, session=self.session) self.central.addTab(stringswin, stringswin.title) self.central.setTabToolTip(self.central.indexOf(stringswin), stringswin.title) self.central.setCurrentWidget(stringswin) def openBytecodeWindow(self, current_class, method=None): byte_code_str = '' for clazz in self.d.get_classes(): if clazz.get_name() == current_class.get_name(): for method in clazz.get_methods(): byte_code_str += "# " + method.get_name( ) + " " + method.get_descriptor() + "\n" byte_code = method.get_code() if byte_code != None: byte_code = byte_code.get_bc() idx = 0 for i in byte_code.get_instructions(): byte_code_str += "\t, %x " % (idx) + i.get_name( ) + " " + i.get_output() + "\n" idx += i.get_length() bytecode_tab = self.get_bytecode_window(byte_code_str) self.central.addTab( bytecode_tab, "Bytecode: " + current_class.get_name()) self.central.setCurrentWidget(bytecode_tab) def showStatus(self, text): QtGui.QMessageBox.information(self, "Info", text) def get_bytecode_window(self, byte_code): return BytecodeWindow(byte_code, self) def getMeSourceWindowIfExists(self, path): '''Helper for openSourceWindow''' for idx in range(self.central.count()): if path == self.central.tabToolTip(idx): return self.central.widget(idx) return None def load_app_info_table(self): self.info = {} self.info["Application Name"] = self.apk.get_app_name() self.info["Application Size"] = util.sizeof_fmt( os.path.getsize(self.apk_path)) self.info["Android Version Name"] = self.apk.get_androidversion_name() self.info["Android Version Code"] = self.apk.get_androidversion_code() self.info["Android Package Name"] = self.apk.get_package() self.info["Signature Name"] = self.apk.get_signature_name() self.info["Uses Dynamic Code Loading"] = str( analysis.is_dyn_code(self.x)) self.info["Uses Reflection"] = str(analysis.is_reflection_code(self.x)) self.info["Uses Crypto"] = str(analysis.is_crypto_code(self.x)) self.info["Privacy Leaks"] = str(len(self.get_privacy_leaks())) self.info["Number of Providers"] = str(len(self.apk.get_providers())) self.info["Number of Activities"] = str(len(self.apk.get_activities())) self.info["Number of Services"] = str(len(self.apk.get_services())) self.info["Number of Libraries"] = str(len(self.apk.get_libraries())) self.info["Number of Permissions"] = str( len(self.get_uses_permissions())) self.info_actions = {} self.info_actions["Application Name"] = None self.info_actions["Application Size"] = None self.info_actions["Android Version Name"] = None self.info_actions["Android Version Code"] = None self.info_actions["Android Package Name"] = None self.info_actions["Signature Name"] = self.show_signature self.info_actions["Uses Dynamic Code Loading"] = self.show_dyncode self.info_actions["Uses Reflection"] = self.show_reflection self.info_actions["Uses Crypto"] = self.show_cryptocode self.info_actions["Privacy Leaks"] = self.show_privacy_leaks self.info_actions["Number of Providers"] = self.show_providers self.info_actions["Number of Activities"] = self.show_activities self.info_actions["Number of Services"] = self.show_services self.info_actions["Number of Libraries"] = self.show_libraries self.info_actions["Number of Permissions"] = self.show_permissions info_table = self.ui.appInfoTable info_table.setRowCount(len(self.info)) info_table.setColumnWidth(1, 200) info_table.horizontalHeader().setResizeMode(0, QtGui.QHeaderView.Stretch) row = 0 for key in sorted(self.info): action = self.info_actions[key] action_button = None if action is not None: action_button = QtGui.QPushButton() action_button.setText("Show") action_button.clicked.connect(action) key_item = QtGui.QTableWidgetItem(key) value_item = QtGui.QTableWidgetItem(self.info[key]) info_table.setItem(row, 0, key_item) info_table.setItem(row, 1, value_item) if action_button is not None: info_table.setCellWidget(row, 2, action_button) row += 1 def show_providers(self): self.__logger.log_with_title("Providers", self.apk.get_providers()) def show_signature(self): certname = "" signature_expr = re.compile("^(META-INF/)(.*)(\.RSA|\.DSA)$") for i in self.apk.get_files(): if signature_expr.search(i): f = tempfile.NamedTemporaryFile(delete=False) f.write(self.apk.get_file(i)) f.close() certname = f.name cmd = ["keytool", "-printcert", "-file", certname] proc = subprocess.Popen(cmd, stdout=subprocess.PIPE) (out, err) = proc.communicate() self.__logger.log_with_title("Signature", out) def show_services(self): self.__logger.log_with_title( "Services", '\n\t -' + '\n\t -'.join(self.apk.get_services())) def show_permissions(self): self.__logger.log( Logger.INFO, "Searching for permission usage, this can take a while depending on the size of the app." ) p = self.x.get_permissions([]) self.__logger.log_with_title("Permissions Usage", "") for i in p: self.__logger.log_with_color(Logger.WARNING, "\n\t=======" + i + "=======\n") for j in p[i]: self.__logger.log_with_color( Logger.INFO, "\t\t -" + self.show_Path(self.x.get_vm(), j) + "\n") def show_cryptocode(self): if analysis.is_crypto_code(self.x) is False: self.__logger.log( Logger.WARNING, "No crypto code was found! Maybe it uses adapted or obfuscated encryption code." ) return paths = [] paths.extend(self.x.get_tainted_packages().search_methods( "Ljavax/crypto/.", ".", ".")) paths.extend(self.x.get_tainted_packages().search_methods( "Ljava/security/spec/.", ".", ".")) str_info_dyn = "\t" for path in paths: str_info_dyn += (self.show_Path(self.x.get_vm(), path) + "\n\n\t") self.__logger.log_with_title("Usage of Crypto Code", str_info_dyn) def show_dyncode(self): if analysis.is_dyn_code(self.x) is False: self.__logger.log(Logger.WARNING, "No dynamic code was found!") return paths = [] paths.extend(self.x.get_tainted_packages().search_methods( "Ldalvik/system/BaseDexClassLoader;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods( "Ldalvik/system/PathClassLoader;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods( "Ldalvik/system/DexClassLoader;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods( "Ldalvik/system/DexFile;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods( "Ldalvik/system/DexFile;", "loadDex", ".")) str_info_dyn = "\t" for path in paths: str_info_dyn += (self.show_Path(self.x.get_vm(), path) + "\n\n\t") self.__logger.log_with_title("Usage of Dynamic Code", str_info_dyn) def get_privacy_leaks(self): paths = [] paths.extend(self.x.get_tainted_packages().search_methods( ".", "getDeviceId", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getSubscriberId", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getSimSerialNumber", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getLine1Number", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getAddress", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getMacAddress", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getSSID", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getCid", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getAccounts", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getTimeZone", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getAllBookmarks", ".")) paths.extend(self.x.get_tainted_packages().search_methods( ".", "getAllVisitedUrls", ".")) return paths def show_privacy_leaks(self): paths = self.get_privacy_leaks() if len(paths) == 0: self.__logger.log(Logger.WARNING, "No privacy leaks were found!") return str_info_privacy = "\t" for path in paths: str_info_privacy += (self.show_Path(self.x.get_vm(), path) + "\n\n\t") self.__logger.log_with_title("Possible Privacy Leaks", str_info_privacy) def show_reflection(self): if analysis.is_reflection_code(self.x) is False: self.__logger.log(Logger.WARNING, "No reflection code was found!") return paths = self.x.get_tainted_packages().search_methods( "Ljava/lang/reflect/Method;", ".", ".") str_info_reflection = "" for path in paths: str_info_reflection += (self.show_Path(self.x.get_vm(), path) + "\n\n\t") self.__logger.log_with_title("Usage of Reflection", str_info_reflection) def show_activities(self): self.__logger.log_with_title( "Activities", '\n\t -' + '\n\t -'.join(self.apk.get_activities())) def show_libraries(self): if len(self.apk.get_libraries()) == 0: self.__logger.log(Logger.WARNING, "No libraries were found.") return self.__logger.log(Logger.INFO, "Libraries Used: " + str(self.apk.get_libraries())) def init_actions(self): self.ui.saveLogBtn.clicked.connect(self.__logger.saveLog) self.ui.clearLogBtn.clicked.connect(self.__logger.clearLog) self.ui.showStringsBtn.clicked.connect(self.openStringsWindow) self.ui.showManifestBtn.clicked.connect(self.openManifestWindow) def load_apk_from_device(self): table = DeviceTable(self, parent=self) self.sample_dialog.close() table.show_data() table.exec_() def load_apk(self, path=None): if not path: path = QtGui.QFileDialog.getOpenFileName( self, "Open File", '', "APK Files (*.apk);;Androguard Session (*.ag)") self.apk_path = path self.set_loading_progressbar( "Analyzing APK", "Please wait while the APK is being dissected") self.__logger.log(Logger.INFO, "Analyzing %s..." % str(path)) self.apk = apk.APK(self.apk_path) self.manifest = self.apk.get_AndroidManifest().getElementsByTagName( "manifest")[0] self.fileLoadingThread.load(str(path)) def load_permissions(self): perms = self.get_uses_permissions() self.ui.permsTable.setRowCount(len(perms)) self.ui.permsTable.horizontalHeader().setStretchLastSection(True) for i, uses_permission in enumerate(perms): newitem = QtGui.QTableWidgetItem(uses_permission) self.ui.permsTable.setItem(0, i, newitem) def get_uses_permissions(self): permissions = [] for uses_permission in self.manifest.getElementsByTagName( "uses-permission"): permissions.append( uses_permission.attributes["android:name"].value) return permissions def set_loading_progressbar(self, title, text): self.dialog = QtGui.QProgressDialog(self) self.dialog.setMinimum(0) self.dialog.setLabelText(text) self.dialog.setMaximum(0) self.dialog.setWindowTitle(title) self.dialog.setCancelButton(None) self.dialog.setModal(True) self.dialog.show() def set_loading_progressbar_text(self, text): if self.dialog is not None: self.dialog.setLabelText(text) def set_loading_progressbar_disabled(self): self.dialog.hide() def show_Path(self, vm, path): cm = vm.get_class_manager() if isinstance(path, analysis.PathVar): dst_class_name, dst_method_name, dst_descriptor = path.get_dst(cm) info_var = path.get_var_info() return "%s %s (0x%x) ---> %s->%s%s " % ( path.get_access_flag(), info_var, path.get_idx(), dst_class_name, dst_method_name, dst_descriptor) else: if path.get_access_flag() == analysis.TAINTED_PACKAGE_CALL: src_class_name, src_method_name, src_descriptor = path.get_src( cm) dst_class_name, dst_method_name, dst_descriptor = path.get_dst( cm) return "%d %s->%s%s (0x%x) ---> %s->%s%s" % ( path.get_access_flag(), src_class_name, src_method_name, src_descriptor, path.get_idx(), dst_class_name, dst_method_name, dst_descriptor) else: src_class_name, src_method_name, src_descriptor = path.get_src( cm) return "%d %s->%s%s (0x%x)" % (path.get_access_flag(), src_class_name, src_method_name, src_descriptor, path.get_idx())
class MainView(QtGui.QMainWindow): def __init__(self, parent=None): super(MainView, self).__init__(parent) self.ui = Ui_MainWindow() self.ui.setupUi(self) self.__logger = Logger(self.ui.logArea) self.__logger.log(Logger.INFO, '') self.init_actions() self.setup_fileloading() self.apk_path = '' self.apk = None self.x = self.d = self.a = None self.manifest = None self.show_sample_question() def setup_fileloading(self): self.session = Session() self.fileLoadingThread = FileLoadingThread(self.session) self.connect(self.fileLoadingThread,QtCore.SIGNAL("loadedFile(bool)"),self.loadedApk) def show_sample_question(self): self.sample_dialog = SampleDialog(self) self.sample_dialog.exec_() def get_logger(self): return self.__logger; def loadedApk(self, success): self.sample_dialog.close() if not success: self.__logger.log(Logger.ERROR,"Analysis of %s failed :(" % str(self.fileLoadingThread.file_path)) self.set_loading_progressbar_disabled() return self.d = self.session.get_DalvikForm() self.d.create_python_export() self.x = uVMAnalysis(self.d) self.setupTree() self.setupResourceController() self.load_app_info_table() self.load_permissions() try: self.__logger.log(Logger.INFO,"Analysis of %s done!" % str(self.apk.get_app_name())) self.ui.loadedAPK_label.setText("Loaded: "+str(self.apk.get_app_name())) except UnicodeEncodeError: self.ui.loadedAPK_label.setText("APK Loaded! (non-ascii chars detected)") self.__logger.log(Logger.WARNING,"Non ascii code characters detected, some strings are possibly not displayed properly.") self.set_loading_progressbar_disabled() def setupResourceController(self): self.resourcecontroller = ResourceFileController(self.apk, self.ui.resourceFileTable, self.ui.graphicview, self.ui.fileResourceInfo) def get_android_manifest_xml(self): self.set_loading_progressbar_text("Decompiling AndroidManifest.xml") buff = self.apk.get_android_manifest_xml().toprettyxml(encoding="utf-8") doc = QtGui.QTextEdit() doc.setWindowTitle("AndroidManifest.xml") hl = XMLHighlighter(doc.document()) doc.setPlainText(str(buff).strip()) return doc def setupTree(self): try: self.ui.tree_area.layout().deleteLater() except AttributeError: pass self.tree = TreeWindow(self,self,session=self.session) self.tree.setWindowTitle("Tree model") layout = QtGui.QVBoxLayout() layout.addWidget(self.tree) self.ui.tree_area.setLayout(layout) self.tree.fill() self.setupCentral() def setupCentral(self): self.central = QtGui.QTabWidget() self.central.setTabBar(CustomTabBar()) self.central.setTabsClosable(True) self.central.tabCloseRequested.connect(self.tabCloseRequestedHandler) self.central.currentChanged.connect(self.currentTabChanged) layout = QtGui.QVBoxLayout() layout.addWidget(self.central) self.ui.sourceTextWidget.setLayout(layout) def tabCloseRequestedHandler(self, index): self.central.removeTab(index) def currentTabChanged(self, index): if index == -1: return # all tab closed def openSourceWindow(self, current_class, method=None): sourcewin = self.getMeSourceWindowIfExists(current_class) if not sourcewin: current_filename = self.session.get_filename_by_class(current_class) current_digest = self.session.get_digest_by_class(current_class) sourcewin = SourceWindow(win=self, current_class=current_class, current_title=current_class.current_title, current_filename=current_filename, current_digest=current_digest, session=self.session) sourcewin.reload_java_sources() self.central.addTab(sourcewin, sourcewin.title) self.central.setTabToolTip(self.central.indexOf(sourcewin), current_class.get_name()) if method: sourcewin.browse_to_method(method) self.central.setCurrentWidget(sourcewin) def openManifestWindow(self): manifest_tab = self.get_android_manifest_xml() self.central.addTab(manifest_tab,"AndroidManifest.xml") self.central.setCurrentWidget(manifest_tab) def openStringsWindow(self): stringswin = StringsWindow(win=self, session=self.session) self.central.addTab(stringswin, stringswin.title) self.central.setTabToolTip(self.central.indexOf(stringswin), stringswin.title) self.central.setCurrentWidget(stringswin) def openBytecodeWindow(self, current_class, method=None): byte_code_str = '' for clazz in self.d.get_classes(): if clazz.get_name() == current_class.get_name(): for method in clazz.get_methods(): byte_code_str += "# "+method.get_name()+" "+method.get_descriptor()+"\n" byte_code = method.get_code() if byte_code != None: byte_code = byte_code.get_bc() idx = 0 for i in byte_code.get_instructions(): byte_code_str += "\t, %x " % (idx)+i.get_name()+" "+i.get_output()+"\n" idx += i.get_length() bytecode_tab = self.get_bytecode_window(byte_code_str) self.central.addTab(bytecode_tab,"Bytecode: "+current_class.get_name()) self.central.setCurrentWidget(bytecode_tab) def showStatus(self,text): QtGui.QMessageBox.information(self, "Info", text) def get_bytecode_window(self,byte_code): return BytecodeWindow(byte_code,self) def getMeSourceWindowIfExists(self, path): '''Helper for openSourceWindow''' for idx in range(self.central.count()): if path == self.central.tabToolTip(idx): return self.central.widget(idx) return None def load_app_info_table(self): self.info = {} self.info["Application Name"] = self.apk.get_app_name() self.info["Application Size"] = util.sizeof_fmt(os.path.getsize(self.apk_path)) self.info["Android Version Name"] = self.apk.get_androidversion_name() self.info["Android Version Code"] = self.apk.get_androidversion_code() self.info["Android Package Name"] = self.apk.get_package() self.info["Signature Name"] = self.apk.get_signature_name() self.info["Uses Dynamic Code Loading"] = str(analysis.is_dyn_code(self.x)) self.info["Uses Reflection"] = str(analysis.is_reflection_code(self.x)) self.info["Uses Crypto"] = str(analysis.is_crypto_code(self.x)) self.info["Privacy Leaks"] = str(len(self.get_privacy_leaks())) self.info["Number of Providers"] = str(len(self.apk.get_providers())) self.info["Number of Activities"] = str(len(self.apk.get_activities())) self.info["Number of Services"] = str(len(self.apk.get_services())) self.info["Number of Libraries"] = str(len(self.apk.get_libraries())) self.info["Number of Permissions"] = str(len(self.get_uses_permissions())) self.info_actions = {} self.info_actions["Application Name"] = None self.info_actions["Application Size"] = None self.info_actions["Android Version Name"] = None self.info_actions["Android Version Code"] = None self.info_actions["Android Package Name"] = None self.info_actions["Signature Name"] = self.show_signature self.info_actions["Uses Dynamic Code Loading"] = self.show_dyncode self.info_actions["Uses Reflection"] = self.show_reflection self.info_actions["Uses Crypto"] = self.show_cryptocode self.info_actions["Privacy Leaks"] = self.show_privacy_leaks self.info_actions["Number of Providers"] = self.show_providers self.info_actions["Number of Activities"] = self.show_activities self.info_actions["Number of Services"] = self.show_services self.info_actions["Number of Libraries"] = self.show_libraries self.info_actions["Number of Permissions"] = self.show_permissions info_table = self.ui.appInfoTable info_table.setRowCount(len(self.info)) info_table.setColumnWidth(1, 200) info_table.horizontalHeader().setResizeMode(0, QtGui.QHeaderView.Stretch) row = 0 for key in sorted(self.info): action = self.info_actions[key] action_button = None if action is not None: action_button = QtGui.QPushButton() action_button.setText("Show") action_button.clicked.connect(action) key_item = QtGui.QTableWidgetItem(key) value_item = QtGui.QTableWidgetItem(self.info[key]) info_table.setItem(row,0,key_item) info_table.setItem(row,1,value_item) if action_button is not None: info_table.setCellWidget(row,2,action_button) row += 1 def show_providers(self): self.__logger.log_with_title("Providers",self.apk.get_providers()) def show_signature(self): certname = "" signature_expr = re.compile("^(META-INF/)(.*)(\.RSA|\.DSA)$") for i in self.apk.get_files(): if signature_expr.search(i): f = tempfile.NamedTemporaryFile(delete=False) f.write(self.apk.get_file(i)) f.close() certname = f.name cmd = ["keytool","-printcert","-file",certname] proc = subprocess.Popen(cmd,stdout=subprocess.PIPE) (out,err) = proc.communicate() self.__logger.log_with_title("Signature",out) def show_services(self): self.__logger.log_with_title("Services",'\n\t -'+'\n\t -'.join(self.apk.get_services())) def show_permissions(self): self.__logger.log(Logger.INFO,"Searching for permission usage, this can take a while depending on the size of the app.") p = self.x.get_permissions( [] ) self.__logger.log_with_title("Permissions Usage","") for i in p: self.__logger.log_with_color(Logger.WARNING,"\n\t======="+i+"=======\n") for j in p[i]: self.__logger.log_with_color(Logger.INFO,"\t\t -"+self.show_Path(self.x.get_vm(), j )+"\n") def show_cryptocode(self): if analysis.is_crypto_code(self.x) is False: self.__logger.log(Logger.WARNING,"No crypto code was found! Maybe it uses adapted or obfuscated encryption code.") return paths = [] paths.extend(self.x.get_tainted_packages().search_methods("Ljavax/crypto/.", ".", ".")) paths.extend(self.x.get_tainted_packages().search_methods("Ljava/security/spec/.", ".", ".")) str_info_dyn="\t" for path in paths: str_info_dyn += (self.show_Path(self.x.get_vm(), path )+"\n\n\t") self.__logger.log_with_title("Usage of Crypto Code",str_info_dyn) def show_dyncode(self): if analysis.is_dyn_code(self.x) is False: self.__logger.log(Logger.WARNING,"No dynamic code was found!") return paths = [] paths.extend(self.x.get_tainted_packages().search_methods("Ldalvik/system/BaseDexClassLoader;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods("Ldalvik/system/PathClassLoader;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods("Ldalvik/system/DexClassLoader;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods("Ldalvik/system/DexFile;", "<init>", ".")) paths.extend(self.x.get_tainted_packages().search_methods("Ldalvik/system/DexFile;", "loadDex", ".")) str_info_dyn="\t" for path in paths: str_info_dyn += (self.show_Path(self.x.get_vm(), path )+"\n\n\t") self.__logger.log_with_title("Usage of Dynamic Code",str_info_dyn) def get_privacy_leaks(self): paths = [] paths.extend(self.x.get_tainted_packages().search_methods(".", "getDeviceId", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getSubscriberId", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getSimSerialNumber", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getLine1Number", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getAddress", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getMacAddress", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getSSID", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getCid", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getAccounts", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getTimeZone", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getAllBookmarks", ".")) paths.extend(self.x.get_tainted_packages().search_methods(".", "getAllVisitedUrls", ".")) return paths def show_privacy_leaks(self): paths = self.get_privacy_leaks() if len(paths) == 0: self.__logger.log(Logger.WARNING,"No privacy leaks were found!") return str_info_privacy="\t" for path in paths: str_info_privacy += (self.show_Path(self.x.get_vm(), path )+"\n\n\t") self.__logger.log_with_title("Possible Privacy Leaks",str_info_privacy) def show_reflection(self): if analysis.is_reflection_code(self.x) is False: self.__logger.log(Logger.WARNING,"No reflection code was found!") return paths = self.x.get_tainted_packages().search_methods("Ljava/lang/reflect/Method;", ".", ".") str_info_reflection="" for path in paths: str_info_reflection += (self.show_Path(self.x.get_vm(), path )+"\n\n\t") self.__logger.log_with_title("Usage of Reflection",str_info_reflection) def show_activities(self): self.__logger.log_with_title("Activities",'\n\t -'+'\n\t -'.join(self.apk.get_activities())) def show_libraries(self): if len(self.apk.get_libraries()) == 0: self.__logger.log(Logger.WARNING,"No libraries were found.") return self.__logger.log(Logger.INFO,"Libraries Used: "+str(self.apk.get_libraries())) def init_actions(self): self.ui.saveLogBtn.clicked.connect(self.__logger.saveLog) self.ui.clearLogBtn.clicked.connect(self.__logger.clearLog) self.ui.showStringsBtn.clicked.connect(self.openStringsWindow) self.ui.showManifestBtn.clicked.connect(self.openManifestWindow) def load_apk_from_device(self): table = DeviceTable(self,parent=self) self.sample_dialog.close() table.show_data() table.exec_() def load_apk(self,path=None): if not path: path = QtGui.QFileDialog.getOpenFileName(self, "Open File",'', "APK Files (*.apk);;Androguard Session (*.ag)") path = str(path[0]) self.apk_path = path if path: self.set_loading_progressbar("Analyzing APK","Please wait while the APK is being dissected") self.__logger.log(Logger.INFO,"Analyzing %s..." % str(path)) self.apk = apk.APK(path) self.manifest = self.apk.get_AndroidManifest().getElementsByTagName("manifest")[0] self.fileLoadingThread.load(path) def load_permissions(self): perms = self.get_uses_permissions() self.ui.permsTable.setRowCount(len(perms)) self.ui.permsTable.horizontalHeader().setStretchLastSection(True) for i, uses_permission in enumerate(perms): newitem = QtGui.QTableWidgetItem(uses_permission) self.ui.permsTable.setItem(0, i, newitem) def get_uses_permissions(self): permissions = [] for uses_permission in self.manifest.getElementsByTagName("uses-permission"): permissions.append(uses_permission.attributes["android:name"].value) return permissions def set_loading_progressbar(self,title,text): self.dialog = QtGui.QProgressDialog(self) self.dialog.setMinimum(0) self.dialog.setLabelText(text) self.dialog.setMaximum(0) self.dialog.setWindowTitle(title) self.dialog.setCancelButton(None) self.dialog.setModal(True) self.dialog.show() def set_loading_progressbar_text(self,text): if self.dialog is not None: self.dialog.setLabelText(text); def set_loading_progressbar_disabled(self): self.dialog.hide() def show_Path(self,vm, path): cm = vm.get_class_manager() if isinstance(path, analysis.PathVar): dst_class_name, dst_method_name, dst_descriptor = path.get_dst( cm ) info_var = path.get_var_info() return "%s %s (0x%x) ---> %s->%s%s " % (path.get_access_flag(), info_var, path.get_idx(), dst_class_name, dst_method_name, dst_descriptor) else: if path.get_access_flag() == analysis.TAINTED_PACKAGE_CALL: src_class_name, src_method_name, src_descriptor = path.get_src( cm ) dst_class_name, dst_method_name, dst_descriptor = path.get_dst( cm ) return "%d %s->%s%s (0x%x) ---> %s->%s%s" % (path.get_access_flag(), src_class_name, src_method_name, src_descriptor, path.get_idx(), dst_class_name, dst_method_name, dst_descriptor) else: src_class_name, src_method_name, src_descriptor = path.get_src( cm ) return "%d %s->%s%s (0x%x)" % (path.get_access_flag(), src_class_name, src_method_name, src_descriptor, path.get_idx())
class MainWindow(QtGui.QMainWindow): '''Main window: self.central: QTabWidget in center area self.dock: QDockWidget in left area self.tree: TreeWindow(QTreeWidget) in self.dock ''' def __init__(self, parent=None, input_file=None): super(MainWindow, self).__init__(parent) self.session = None self.setupSession() self.setupFileMenu() self.setupViewMenu() self.setupHelpMenu() self.setupCentral() self.setupEmptyTree() self.setupDock() self.setWindowTitle("Androguard GUI") self.showStatus("Androguard GUI") if input_file != None: self.openFile(input_file) def showStatus(self, msg): '''Helper function called by any window to display a message in status bar. ''' androconf.debug(msg) self.statusBar().showMessage(msg) def about(self): '''User clicked About menu. Display a Message box.''' QtGui.QMessageBox.about(self, "About Androguard GUI", "<p><b>Androguard GUI</b> is basically a GUI for Androguard :)." "<br>Have fun !</p>") def setupSession(self): self.session = Session() self.fileLoadingThread = FileLoadingThread(self.session) self.connect(self.fileLoadingThread, QtCore.SIGNAL("loadedFile(bool)"), self.loadedFile) def loadedFile(self, success): if not success: self.showStatus("Analysis of %s failed :(" % str(self.fileLoadingThread.file_path)) return self.updateDockWithTree() self.cleanCentral() self.showStatus("Analysis of %s done!" % str(self.fileLoadingThread.file_path)) def openFile(self, path=None): '''User clicked Open menu. Display a Dialog to ask which file to open.''' self.session.reset() if not path: path = QtGui.QFileDialog.getOpenFileName(self, "Open File", '', "Android Files (*.apk *.jar *.dex *.odex *.dey);;Androguard Session (*.ag)") path = str(path[0]) if path: self.setupTree() self.showStatus("Analyzing %s..." % str(path)) self.fileLoadingThread.load(path) def addFile(self, path=None): '''User clicked Open menu. Display a Dialog to ask which APK to open.''' if not self.session.isOpen(): return if not path: path = QtGui.QFileDialog.getOpenFileName(self, "Add File", '', "Android Files (*.apk *.jar *.dex *.odex *.dey)") path = str(path[0]) if path: self.showStatus("Analyzing %s..." % str(path)) self.fileLoadingThread.load(path) def saveFile(self, path=None): '''User clicked Save menu. Display a Dialog to ask whwre to save.''' if not path: path = QtGui.QFileDialog.getSaveFileName(self, "Save File", '', "Androguard Session (*.ag)") path = str(path[0]) if path: self.showStatus("Saving %s..." % str(path)) self.saveSession(path) def saveSession(self, path): '''Save androguard session.''' try: self.session.save(path) except RuntimeError as e: androconf.error(str(e)) # http://stackoverflow.com/questions/2134706/hitting-maximum-recursion-depth-using-pythons-pickle-cpickle androconf.error("Try increasing sys.recursionlimit") os.remove(path) androconf.warning("Session not saved") def quit(self): '''Clicked in File menu to exit or CTRL+Q to close main window''' QtGui.qApp.quit() def closeEvent(self, event): '''Clicked [x] to close main window''' event.accept() def setupEmptyTree(self): '''Setup empty Tree at startup. ''' if hasattr(self, "tree"): del self.tree self.tree = QtGui.QTreeWidget(self) self.tree.header().close() def setupDock(self): '''Setup empty Dock at startup. ''' self.dock = QtGui.QDockWidget("Classes", self) self.dock.setWidget(self.tree) self.dock.setFeatures(QtGui.QDockWidget.NoDockWidgetFeatures) self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.dock) def setupTree(self): androconf.debug("Setup Tree") self.tree = TreeWindow(win=self, session=self.session) self.tree.setWindowTitle("Tree model") self.dock.setWidget(self.tree) def setupCentral(self): '''Setup empty window supporting tabs at startup. ''' self.central = QtGui.QTabWidget() self.central.setTabsClosable(True) self.central.tabCloseRequested.connect(self.tabCloseRequestedHandler) self.central.currentChanged.connect(self.currentTabChanged) self.setCentralWidget(self.central) def tabCloseRequestedHandler(self, index): self.central.removeTab(index) def currentTabChanged(self, index): androconf.debug("curentTabChanged -> %d" % index) if index == -1: return # all tab closed def cleanCentral(self): # TOFIX: Removes all the pages, but does not delete them. self.central.clear() def setupFileMenu(self): fileMenu = QtGui.QMenu("&File", self) self.menuBar().addMenu(fileMenu) fileMenu.addAction("&Open...", self.openFile, "Ctrl+O") fileMenu.addAction("&Add...", self.addFile, "Ctrl+A") fileMenu.addAction("&Save...", self.saveFile, "Ctrl+S") fileMenu.addAction("E&xit", self.quit, "Ctrl+Q") def setupViewMenu(self): viewMenu = QtGui.QMenu("&View", self) self.menuBar().addMenu(viewMenu) viewMenu.addAction("&Strings...", self.openStringsWindow) def setupHelpMenu(self): helpMenu = QtGui.QMenu("&Help", self) self.menuBar().addMenu(helpMenu) helpMenu.addAction("&About", self.about) helpMenu.addAction("About &Qt", QtGui.qApp.aboutQt) def updateDockWithTree(self, empty=False): '''Update the classes tree. Called when - a new APK has been imported - a classe has been renamed (displayed in the tree) ''' self.setupTree() self.tree.fill() def openStringsWindow(self): stringswin = StringsWindow(win=self, session=self.session) self.central.addTab(stringswin, stringswin.title) self.central.setTabToolTip( self.central.indexOf(stringswin), stringswin.title) self.central.setCurrentWidget(stringswin) def openBytecodeWindow(self, current_class, method=None): pass # self.central.setCurrentWidget(sourcewin) def openSourceWindow(self, current_class, method=None): '''Main function to open a .java source window It checks if it already opened and open that tab, otherwise, initialize a new window. ''' androconf.debug("openSourceWindow for %s" % current_class) sourcewin = self.getMeSourceWindowIfExists(current_class) if not sourcewin: current_filename = self.session.get_filename_by_class( current_class) current_digest = self.session.get_digest_by_class(current_class) sourcewin = SourceWindow(win=self, current_class=current_class, current_title=current_class.current_title, current_filename=current_filename, current_digest=current_digest, session=self.session) sourcewin.reload_java_sources() self.central.addTab(sourcewin, sourcewin.title) self.central.setTabToolTip(self.central.indexOf( sourcewin), current_class.get_name()) if method: sourcewin.browse_to_method(method) self.central.setCurrentWidget(sourcewin) def getMeSourceWindowIfExists(self, current_class): '''Helper for openSourceWindow''' for idx in range(self.central.count()): if current_class.get_name() == self.central.tabToolTip(idx): androconf.debug("Tab %s already opened at: %d" % (current_class.get_name(), idx)) return self.central.widget(idx) return None def doesClassExist(self, path): arg = class2func(path) try: getattr(self.d, arg) except AttributeError: return False return True
class MainWindow(QtGui.QMainWindow): '''Main window: self.central: QTabWidget in center area self.dock: QDockWidget in left area self.tree: TreeWindow(QTreeWidget) in self.dock ''' def __init__(self, parent=None, session=Session(), input_file=None): super(MainWindow, self).__init__(parent) self.session = session self.setupSession() self.setupFileMenu() self.setupViewMenu() self.setupHelpMenu() self.setupCentral() self.setupEmptyTree() self.setupDock() self.setWindowTitle("Androguard GUI") self.showStatus("Androguard GUI") if input_file != None: self.openFile(input_file) def showStatus(self, msg): '''Helper function called by any window to display a message in status bar. ''' androconf.debug(msg) self.statusBar().showMessage(msg) def about(self): '''User clicked About menu. Display a Message box.''' QtGui.QMessageBox.about(self, "About Androguard GUI", "<p><b>Androguard GUI</b> is basically a GUI for Androguard :)." \ "<br>Have fun !</p>") def setupSession(self): self.fileLoadingThread = FileLoadingThread(self.session) self.connect(self.fileLoadingThread, QtCore.SIGNAL("loadedFile(bool)"), self.loadedFile) def loadedFile(self, success): if not success: self.showStatus("Analysis of %s failed :(" % str(self.fileLoadingThread.file_path)) return self.updateDockWithTree() self.cleanCentral() self.showStatus("Analysis of %s done!" % str(self.fileLoadingThread.file_path)) def openFile(self, path=None): '''User clicked Open menu. Display a Dialog to ask which file to open.''' self.session.reset() if not path: path = QtGui.QFileDialog.getOpenFileName( self, "Open File", '', "Android Files (*.apk *.jar *.dex *.odex *.dey);;Androguard Session (*.ag)") path = str(path[0]) if path: self.setupTree() self.showStatus("Analyzing %s..." % str(path)) self.fileLoadingThread.load(path) def addFile(self, path=None): '''User clicked Open menu. Display a Dialog to ask which APK to open.''' if not self.session.isOpen(): return if not path: path = QtGui.QFileDialog.getOpenFileName( self, "Add File", '', "Android Files (*.apk *.jar *.dex *.odex *.dey)") path = str(path[0]) if path: self.showStatus("Analyzing %s..." % str(path)) self.fileLoadingThread.load(path) def saveFile(self, path=None): '''User clicked Save menu. Display a Dialog to ask whwre to save.''' if not path: path = QtGui.QFileDialog.getSaveFileName( self, "Save File", '', "Androguard Session (*.ag)") path = str(path[0]) if path: self.showStatus("Saving %s..." % str(path)) self.saveSession(path) def saveSession(self, path): '''Save androguard session.''' try: self.session.save(path) except RuntimeError as e: androconf.error(str(e)) # http://stackoverflow.com/questions/2134706/hitting-maximum-recursion-depth-using-pythons-pickle-cpickle androconf.error("Try increasing sys.recursionlimit") os.remove(path) androconf.warning("Session not saved") def quit(self): '''Clicked in File menu to exit or CTRL+Q to close main window''' QtGui.qApp.quit() def closeEvent(self, event): '''Clicked [x] to close main window''' event.accept() def setupEmptyTree(self): '''Setup empty Tree at startup. ''' if hasattr(self, "tree"): del self.tree self.tree = QtGui.QTreeWidget(self) self.tree.header().close() def setupDock(self): '''Setup empty Dock at startup. ''' self.dock = QtGui.QDockWidget("Classes", self) self.dock.setWidget(self.tree) self.dock.setFeatures(QtGui.QDockWidget.NoDockWidgetFeatures) self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.dock) def setupTree(self): androconf.debug("Setup Tree") self.tree = TreeWindow(win=self, session=self.session) self.tree.setWindowTitle("Tree model") self.dock.setWidget(self.tree) def setupCentral(self): '''Setup empty window supporting tabs at startup. ''' self.central = QtGui.QTabWidget() self.central.setTabsClosable(True) self.central.tabCloseRequested.connect(self.tabCloseRequestedHandler) self.central.currentChanged.connect(self.currentTabChanged) self.setCentralWidget(self.central) def tabCloseRequestedHandler(self, index): self.central.removeTab(index) def currentTabChanged(self, index): androconf.debug("curentTabChanged -> %d" % index) if index == -1: return # all tab closed def cleanCentral(self): #TOFIX: Removes all the pages, but does not delete them. self.central.clear() def setupFileMenu(self): fileMenu = QtGui.QMenu("&File", self) self.menuBar().addMenu(fileMenu) fileMenu.addAction("&Open...", self.openFile, "Ctrl+O") fileMenu.addAction("&Add...", self.addFile, "Ctrl+A") fileMenu.addAction("&Save...", self.saveFile, "Ctrl+S") fileMenu.addAction("E&xit", self.quit, "Ctrl+Q") def setupViewMenu(self): viewMenu = QtGui.QMenu("&View", self) self.menuBar().addMenu(viewMenu) viewMenu.addAction("&Strings...", self.openStringsWindow) def setupHelpMenu(self): helpMenu = QtGui.QMenu("&Help", self) self.menuBar().addMenu(helpMenu) helpMenu.addAction("&About", self.about) helpMenu.addAction("About &Qt", QtGui.qApp.aboutQt) def updateDockWithTree(self, empty=False): '''Update the classes tree. Called when - a new APK has been imported - a classe has been renamed (displayed in the tree) ''' self.setupTree() self.tree.fill() def openStringsWindow(self): stringswin = StringsWindow(win=self, session=self.session) self.central.addTab(stringswin, stringswin.title) self.central.setTabToolTip(self.central.indexOf(stringswin), stringswin.title) self.central.setCurrentWidget(stringswin) def openBytecodeWindow(self, current_class, method=None): pass #self.central.setCurrentWidget(sourcewin) def openSourceWindow(self, current_class, method=None): '''Main function to open a .java source window It checks if it already opened and open that tab, otherwise, initialize a new window. ''' androconf.debug("openSourceWindow for %s" % current_class) sourcewin = self.getMeSourceWindowIfExists(current_class) if not sourcewin: current_filename = self.session.get_filename_by_class(current_class) current_digest = self.session.get_digest_by_class(current_class) sourcewin = SourceWindow(win=self, current_class=current_class, current_title=current_class.current_title, current_filename=current_filename, current_digest=current_digest, session=self.session) sourcewin.reload_java_sources() self.central.addTab(sourcewin, sourcewin.title) self.central.setTabToolTip(self.central.indexOf(sourcewin), current_class.get_name()) if method: sourcewin.browse_to_method(method) self.central.setCurrentWidget(sourcewin) def getMeSourceWindowIfExists(self, current_class): '''Helper for openSourceWindow''' for idx in range(self.central.count()): if current_class.get_name() == self.central.tabToolTip(idx): androconf.debug("Tab %s already opened at: %d" % (current_class.get_name(), idx)) return self.central.widget(idx) return None def doesClassExist(self, path): arg = class2func(path) try: getattr(self.d, arg) except AttributeError: return False return True