panel_modules.py

"""
Dwarf - Copyright (C) 2019 Giovanni Rocca (iGio90)

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>
"""
import json

from PyQt5.QtCore import Qt

from ui.widget_item_not_editable import NotEditableTableWidgetItem
from ui.widget_memory_address import MemoryAddressWidget
from ui.widget_table_base import TableBaseWidget


class ModulesPanel(TableBaseWidget):
    def __init__(self, app, *__args):
        super().__init__(app, 0, 4)
        self.setHorizontalHeaderLabels(['name', 'base', 'size', 'path'])
        self.setColumnWidth(1, 120)

    def set_menu_actions(self, item, menu):
        action_refresh = menu.addAction("Refresh")
        action_refresh.setData('refresh')
        if item is not None:
            menu.addSeparator()
            action_exports = menu.addAction('Exports')
            action_exports.setData('exports')
            action_imports = menu.addAction('Imports')
            action_imports.setData('imports')
            action_symbols = menu.addAction('Symbols')
            action_symbols.setData('symbols')

    def on_menu_action(self, action_data, item):
        if action_data == 'refresh':
            self.app.dwarf_api('updateModules')
            return False
        elif action_data == 'exports':
            module = self.item(item.row(), 0).text()
            self.add_extra_table_tab(module, 'exports %s' % module,
                                     'enumerateExports',
                                     ['name', 'address', 'type'],
                                     self.build_exports_row)
            return False
        elif action_data == 'imports':
            module = self.item(item.row(), 0).text()
            self.add_extra_table_tab(module, 'imports %s' % module,
                                     'enumerateImports',
                                     ['name', 'address', 'module', 'type'],
                                     self.build_imports_row)
            return False
        elif action_data == 'symbols':
            module = self.item(item.row(), 0).text()
            self.add_extra_table_tab(module, 'symbols %s' % module,
                                     'enumerateSymbols',
                                     ['name', 'address', 'type'],
                                     self.build_exports_row)
            return False
        return True

    @staticmethod
    def build_exports_row(table, item):
        row = table.rowCount()
        table.insertRow(row)

        q = NotEditableTableWidgetItem(item['name'])
        q.setForeground(Qt.gray)
        table.setItem(row, 0, q)

        q = MemoryAddressWidget(item['address'])
        table.setItem(row, 1, q)

        q = NotEditableTableWidgetItem(item['type'])
        table.setItem(row, 2, q)

    @staticmethod
    def build_imports_row(table, item):
        row = table.rowCount()
        table.insertRow(row)

        q = NotEditableTableWidgetItem(item['name'])
        q.setForeground(Qt.gray)
        table.setItem(row, 0, q)

        q = MemoryAddressWidget(item['address'])
        table.setItem(row, 1, q)

        q = NotEditableTableWidgetItem(item['module'])
        table.setItem(row, 2, q)

        q = NotEditableTableWidgetItem(item['type'])
        table.setItem(row, 3, q)

    def add_extra_table_tab(self, module, tab_name, dwarf_api, headers, item_builder):
        table = TableBaseWidget(self.app, 0, 3)
        self.app.get_session_ui().add_tab(table, tab_name)
        data = self.app.dwarf_api(dwarf_api, module)
        if data:
            data = json.loads(data)
            if len(data) > 0:
                table.setHorizontalHeaderLabels(headers)
                for item in data:
                    item_builder(table, item)

                table.resizeColumnsToContents()
                table.horizontalHeader().setStretchLastSection(True)

    def set_modules(self, modules):
        self.setRowCount(0)
        i = 0
        for module in modules:
            self.insertRow(i)
            q = NotEditableTableWidgetItem(module['name'])
            q.setFlags(Qt.NoItemFlags)
            q.setForeground(Qt.gray)
            self.setItem(i, 0, q)
            q = MemoryAddressWidget(module['base'])
            q.set_size(module['size'])
            q.set_base_address(module['base'])
            self.setItem(i, 1, q)
            q = NotEditableTableWidgetItem(str(module['size']))
            q.setFlags(Qt.NoItemFlags)
            self.setItem(i, 2, q)
            q = NotEditableTableWidgetItem(module['path'])
            q.setFlags(Qt.NoItemFlags)
            q.setForeground(Qt.lightGray)
            self.setItem(i, 3, q)
            i += 1
        self.resizeRowsToContents()
        self.horizontalHeader().setStretchLastSection(True)
        self.sortByColumn(0, 0)