1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
#include <progresstask.h>
#include "symboltable.h"
#include <QHeaderView>
#include "binaryninjaapi.h"
using namespace BinaryNinja;
using namespace SharedCacheAPI;
SymbolTableModel::SymbolTableModel(SymbolTableView* parent)
: QAbstractTableModel(parent), m_parent(parent) {
}
int SymbolTableModel::rowCount(const QModelIndex& parent) const {
Q_UNUSED(parent);
return static_cast<int>(m_modelSymbols.size());
}
int SymbolTableModel::columnCount(const QModelIndex& parent) const {
Q_UNUSED(parent);
// We have 3 columns: Address, Type, Name
return 3;
}
QVariant SymbolTableModel::data(const QModelIndex& index, int role) const {
if (!index.isValid() || role != Qt::DisplayRole) {
return QVariant();
}
auto symbol = symbolAt(index.row());
auto symbolType = GetSymbolTypeAsString(symbol.type);
switch (index.column()) {
case 0: // Address column
return QString("0x%1").arg(symbol.address, 0, 16); // Display address as hexadecimal
case 1: // Type column
return QString::fromUtf8(symbolType.c_str(), symbolType.size());
case 2: // Name column
return QString::fromUtf8(symbol.name.c_str(), symbol.name.size());
default:
return QVariant();
}
}
QVariant SymbolTableModel::headerData(int section, Qt::Orientation orientation, int role) const {
if (role != Qt::DisplayRole || orientation != Qt::Horizontal) {
return QVariant();
}
switch (section) {
case 0:
return QString("Address");
case 1:
return QString("Type");
case 2:
return QString("Name");
default:
return QVariant();
}
}
void SymbolTableModel::updateSymbols(std::vector<CacheSymbol>&& symbols)
{
m_preparedSymbols = symbols;
setFilter(m_filter);
}
const CacheSymbol& SymbolTableModel::symbolAt(int row) const
{
return m_modelSymbols.at(row);
}
void SymbolTableModel::setFilter(std::string text)
{
beginResetModel();
m_filter = text;
m_modelSymbols = {};
// Skip filtering if no filter applied.
if (!m_filter.empty())
{
m_modelSymbols.reserve(m_preparedSymbols.size());
for (const auto& symbol : m_preparedSymbols)
if (((std::string_view)symbol.name).find(m_filter) != std::string::npos)
m_modelSymbols.push_back(symbol);
m_modelSymbols.shrink_to_fit();
} else
{
m_modelSymbols = m_preparedSymbols;
}
endResetModel();
}
SymbolTableView::SymbolTableView(QWidget* parent)
: m_model(new SymbolTableModel(this)) {
// Set up the filter model
setModel(m_model);
// Configure view settings
horizontalHeader()->setSectionResizeMode(0, QHeaderView::ResizeToContents);
horizontalHeader()->setSectionResizeMode(1, QHeaderView::ResizeToContents);
horizontalHeader()->setSectionResizeMode(2, QHeaderView::Stretch);
setSelectionBehavior(QAbstractItemView::SelectRows);
setSelectionMode(QAbstractItemView::SingleSelection);
setSortingEnabled(true);
}
SymbolTableView::~SymbolTableView() {
delete m_model;
}
void SymbolTableView::populateSymbols(BinaryView &view)
{
if (auto controller = SharedCacheController::GetController(view)) {
typedef std::vector<CacheSymbol> SymbolList;
// Retrieve the symbols from the controller in a future than pass that to the model.
QPointer<QFutureWatcher<SymbolList>> watcher = new QFutureWatcher<SymbolList>(this);
connect(watcher, &QFutureWatcher<SymbolList>::finished, this, [watcher, this]() {
if (watcher)
{
auto symbols = watcher->result();
m_model->updateSymbols(std::move(symbols));
}
});
QFuture<SymbolList> future = QtConcurrent::run([controller]() {
return controller->GetSymbols();
});
watcher->setFuture(future);
connect(this, &QObject::destroyed, this, [watcher]() {
if (watcher && watcher->isRunning()) {
watcher->cancel();
watcher->waitForFinished();
}
});
}
}
void SymbolTableView::setFilter(const std::string& filter) {
m_model->setFilter(filter);
}
|
