From fd787868288521ae655ddd1ac8000895662eaa77 Mon Sep 17 00:00:00 2001 From: Tony Wasserka Date: Wed, 2 Jul 2014 21:16:36 +0200 Subject: [PATCH] citra-qt: Rewrite disassembly view. --- src/citra_qt/debugger/disassembler.cpp | 205 ++++++++++++++++++------- src/citra_qt/debugger/disassembler.hxx | 41 ++++- 2 files changed, 186 insertions(+), 60 deletions(-) diff --git a/src/citra_qt/debugger/disassembler.cpp b/src/citra_qt/debugger/disassembler.cpp index 4e8f841f4a..4fd6a6e629 100644 --- a/src/citra_qt/debugger/disassembler.cpp +++ b/src/citra_qt/debugger/disassembler.cpp @@ -1,5 +1,3 @@ -#include - #include "disassembler.hxx" #include "../bootmanager.hxx" @@ -14,14 +12,143 @@ #include "core/arm/interpreter/armdefs.h" #include "core/arm/disassembler/arm_disasm.h" +DisassemblerModel::DisassemblerModel(QObject* parent) : QAbstractItemModel(parent), base_address(0), code_size(0), program_counter(0), selection(QModelIndex()) { + +} + +QModelIndex DisassemblerModel::index(int row, int column, const QModelIndex& parent) const { + return createIndex(row, column); +} + +QModelIndex DisassemblerModel::parent(const QModelIndex& child) const { + return QModelIndex(); +} + +int DisassemblerModel::columnCount(const QModelIndex& parent) const { + return 3; +} + +int DisassemblerModel::rowCount(const QModelIndex& parent) const { + return code_size; +} + +QVariant DisassemblerModel::data(const QModelIndex& index, int role) const { + switch (role) { + case Qt::DisplayRole: + { + static char result[255]; + + u32 address = base_address + index.row() * 4; + ARM_Disasm::disasm(address, Memory::Read32(address), result); + + if (index.column() == 0) { + return QString("0x%1").arg((uint)(address), 8, 16, QLatin1Char('0')); + } else if (index.column() == 1) { + return QString::fromLatin1(result); + } else if (index.column() == 2 && Symbols::HasSymbol(address)) { + TSymbol symbol = Symbols::GetSymbol(address); + return QString("%1 - Size:%2").arg(QString::fromStdString(symbol.name)) + .arg(symbol.size / 4); // divide by 4 to get instruction count + } + + break; + } + + case Qt::BackgroundRole: + { + unsigned int address = base_address + 4 * index.row(); + + if (breakpoints.IsAddressBreakPoint(address)) + return QBrush(QColor(0xFF, 0xC0, 0xC0)); + else if (address == program_counter) + return QBrush(QColor(0xC0, 0xC0, 0xFF)); + + break; + } + + default: + break; + } + + return QVariant(); +} + +QModelIndex DisassemblerModel::IndexFromAbsoluteAddress(unsigned int address) const { + return index((address - base_address) / 4, 0); +} + +const BreakPoints& DisassemblerModel::GetBreakPoints() const { + return breakpoints; +} + +void DisassemblerModel::ParseFromAddress(unsigned int address) { + const unsigned int chunk_size = 1000*1000; // 10*1000*1000 is critical + + // If we haven't loaded anything yet, initialize base address to the parameter address + if (code_size == 0) + base_address = address; + + // If the new area is already loaded, just continue + if (base_address + code_size > address + chunk_size && base_address <= address) + return; + + // Insert rows before currently loaded data + if (base_address > address) { + unsigned int num_rows = (address - base_address) / 4; + + beginInsertRows(QModelIndex(), 0, num_rows); + code_size += num_rows; + base_address = address; + + endInsertRows(); + } + + // Insert rows after currently loaded data + if (base_address + code_size < address + chunk_size) { + unsigned int num_rows = (base_address + chunk_size - code_size - address) / 4; + + beginInsertRows(QModelIndex(), 0, num_rows); + code_size += num_rows; + endInsertRows(); + } + + SetNextInstruction(address); +} + +void DisassemblerModel::OnSelectionChanged(const QModelIndex& new_selection) { + selection = new_selection; +} + +void DisassemblerModel::OnSetOrUnsetBreakpoint() { + if (!selection.isValid()) + return; + + unsigned int address = base_address + selection.row() * 4; + + if (breakpoints.IsAddressBreakPoint(address)) { + breakpoints.Remove(address); + } else { + breakpoints.Add(address); + } + + emit dataChanged(selection, selection); +} + +void DisassemblerModel::SetNextInstruction(unsigned int address) { + QModelIndex cur_index = IndexFromAbsoluteAddress(program_counter); + QModelIndex prev_index = IndexFromAbsoluteAddress(address); + + program_counter = address; + + emit dataChanged(cur_index, cur_index); + emit dataChanged(prev_index, prev_index); +} + DisassemblerWidget::DisassemblerWidget(QWidget* parent, EmuThread& emu_thread) : QDockWidget(parent), base_addr(0), emu_thread(emu_thread) { disasm_ui.setupUi(this); - breakpoints = new BreakPoints(); - - model = new QStandardItemModel(this); - model->setColumnCount(3); + model = new DisassemblerModel(this); disasm_ui.treeView->setModel(model); RegisterHotkey("Disassembler", "Start/Stop", QKeySequence(Qt::Key_F5), Qt::ApplicationShortcut); @@ -29,70 +156,33 @@ DisassemblerWidget::DisassemblerWidget(QWidget* parent, EmuThread& emu_thread) : RegisterHotkey("Disassembler", "Step into", QKeySequence(Qt::Key_F11), Qt::ApplicationShortcut); RegisterHotkey("Disassembler", "Set Breakpoint", QKeySequence(Qt::Key_F9), Qt::ApplicationShortcut); - connect(disasm_ui.button_breakpoint, SIGNAL(clicked()), this, SLOT(OnSetBreakpoint())); + connect(disasm_ui.button_breakpoint, SIGNAL(clicked()), model, SLOT(OnSetOrUnsetBreakpoint())); connect(disasm_ui.button_step, SIGNAL(clicked()), this, SLOT(OnStep())); connect(disasm_ui.button_pause, SIGNAL(clicked()), this, SLOT(OnPause())); connect(disasm_ui.button_continue, SIGNAL(clicked()), this, SLOT(OnContinue())); + connect(disasm_ui.treeView->selectionModel(), SIGNAL(currentChanged(const QModelIndex&, const QModelIndex&)), + model, SLOT(OnSelectionChanged(const QModelIndex&))); + connect(GetHotkey("Disassembler", "Start/Stop", this), SIGNAL(activated()), this, SLOT(OnToggleStartStop())); connect(GetHotkey("Disassembler", "Step", this), SIGNAL(activated()), this, SLOT(OnStep())); connect(GetHotkey("Disassembler", "Step into", this), SIGNAL(activated()), this, SLOT(OnStepInto())); - connect(GetHotkey("Disassembler", "Set Breakpoint", this), SIGNAL(activated()), this, SLOT(OnSetBreakpoint())); + connect(GetHotkey("Disassembler", "Set Breakpoint", this), SIGNAL(activated()), model, SLOT(OnSetOrUnsetBreakpoint())); } void DisassemblerWidget::Init() { - ARM_Disasm* disasm = new ARM_Disasm(); + model->ParseFromAddress(Core::g_app_core->GetPC()); - base_addr = Core::g_app_core->GetPC(); - unsigned int curInstAddr = base_addr; - char result[255]; - - for (int i = 0; i < 20000; i++) // fixed for now - { - disasm->disasm(curInstAddr, Memory::Read32(curInstAddr), result); - model->setItem(i, 0, new QStandardItem(QString("0x%1").arg((uint)(curInstAddr), 8, 16, QLatin1Char('0')))); - model->setItem(i, 1, new QStandardItem(QString(result))); - if (Symbols::HasSymbol(curInstAddr)) - { - TSymbol symbol = Symbols::GetSymbol(curInstAddr); - model->setItem(i, 2, new QStandardItem(QString("%1 - Size:%2").arg(QString::fromStdString(symbol.name)) - .arg(symbol.size / 4))); // divide by 4 to get instruction count - - } - curInstAddr += 4; - } disasm_ui.treeView->resizeColumnToContents(0); disasm_ui.treeView->resizeColumnToContents(1); disasm_ui.treeView->resizeColumnToContents(2); - QModelIndex model_index = model->index(0, 0); + QModelIndex model_index = model->IndexFromAbsoluteAddress(Core::g_app_core->GetPC()); disasm_ui.treeView->scrollTo(model_index); disasm_ui.treeView->selectionModel()->setCurrentIndex(model_index, QItemSelectionModel::SelectCurrent | QItemSelectionModel::Rows); } -void DisassemblerWidget::OnSetBreakpoint() -{ - int selected_row = SelectedRow(); - - if (selected_row == -1) - return; - - u32 address = base_addr + (selected_row * 4); - if (breakpoints->IsAddressBreakPoint(address)) - { - breakpoints->Remove(address); - model->item(selected_row, 0)->setBackground(QBrush()); - model->item(selected_row, 1)->setBackground(QBrush()); - } - else - { - breakpoints->Add(address); - model->item(selected_row, 0)->setBackground(QBrush(QColor(0xFF, 0x99, 0x99))); - model->item(selected_row, 1)->setBackground(QBrush(QColor(0xFF, 0x99, 0x99))); - } -} - void DisassemblerWidget::OnContinue() { emu_thread.SetCpuRunning(true); @@ -112,6 +202,9 @@ void DisassemblerWidget::OnStepInto() void DisassemblerWidget::OnPause() { emu_thread.SetCpuRunning(false); + + // TODO: By now, the CPU might not have actually stopped... + model->SetNextInstruction(Core::g_app_core->GetPC()); } void DisassemblerWidget::OnToggleStartStop() @@ -123,13 +216,15 @@ void DisassemblerWidget::OnCPUStepped() { ARMword next_instr = Core::g_app_core->GetPC(); - if (breakpoints->IsAddressBreakPoint(next_instr)) + // TODO: Make BreakPoints less crappy (i.e. const-correct) so that this doesn't need a const_cast. + if (const_cast(model->GetBreakPoints()).IsAddressBreakPoint(next_instr)) { emu_thread.SetCpuRunning(false); } - unsigned int index = (next_instr - base_addr) / 4; - QModelIndex model_index = model->index(index, 0); + model->SetNextInstruction(next_instr); + + QModelIndex model_index = model->IndexFromAbsoluteAddress(next_instr); disasm_ui.treeView->scrollTo(model_index); disasm_ui.treeView->selectionModel()->setCurrentIndex(model_index, QItemSelectionModel::SelectCurrent | QItemSelectionModel::Rows); } @@ -140,5 +235,5 @@ int DisassemblerWidget::SelectedRow() if (!index.isValid()) return -1; - return model->itemFromIndex(disasm_ui.treeView->selectionModel()->currentIndex())->row(); -} \ No newline at end of file + return disasm_ui.treeView->selectionModel()->currentIndex().row(); +} diff --git a/src/citra_qt/debugger/disassembler.hxx b/src/citra_qt/debugger/disassembler.hxx index e668bbbeb9..a842da9568 100644 --- a/src/citra_qt/debugger/disassembler.hxx +++ b/src/citra_qt/debugger/disassembler.hxx @@ -1,3 +1,4 @@ +#include #include #include "ui_disassembler.h" @@ -5,9 +6,41 @@ #include "common/break_points.h" class QAction; -class QStandardItemModel; class EmuThread; +class DisassemblerModel : public QAbstractItemModel +{ + Q_OBJECT + +public: + DisassemblerModel(QObject* parent); + + QModelIndex index(int row, int column, const QModelIndex& parent = QModelIndex()) const override; + QModelIndex parent(const QModelIndex& child) const override; + int columnCount(const QModelIndex& parent = QModelIndex()) const override; + int rowCount(const QModelIndex& parent = QModelIndex()) const override; + QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const override; + + QModelIndex IndexFromAbsoluteAddress(unsigned int address) const; + const BreakPoints& GetBreakPoints() const; + +public slots: + void ParseFromAddress(unsigned int address); + void OnSelectionChanged(const QModelIndex&); + void OnSetOrUnsetBreakpoint(); + void SetNextInstruction(unsigned int address); + +private: + unsigned int base_address; + unsigned int code_size; + unsigned int program_counter; + + QModelIndex selection; + + // TODO: Make BreakPoints less crappy (i.e. const-correct) so that this needn't be mutable. + mutable BreakPoints breakpoints; +}; + class DisassemblerWidget : public QDockWidget { Q_OBJECT @@ -18,7 +51,6 @@ public: void Init(); public slots: - void OnSetBreakpoint(); void OnContinue(); void OnStep(); void OnStepInto(); @@ -32,11 +64,10 @@ private: int SelectedRow(); Ui::DockWidget disasm_ui; - QStandardItemModel* model; + + DisassemblerModel* model; u32 base_addr; - BreakPoints* breakpoints; - EmuThread& emu_thread; };