2018-03-08 12:24:15 +00:00
|
|
|
#include <QDockWidget>
|
|
|
|
#include <QTreeWidget>
|
|
|
|
#include <QComboBox>
|
|
|
|
#include "SearchWidget.h"
|
|
|
|
#include "ui_SearchWidget.h"
|
|
|
|
#include "MainWindow.h"
|
|
|
|
#include "utils/Helpers.h"
|
|
|
|
|
|
|
|
SearchModel::SearchModel(QList<SearchDescription> *search, QObject *parent)
|
|
|
|
: QAbstractListModel(parent),
|
|
|
|
search(search)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
int SearchModel::rowCount(const QModelIndex &) const
|
|
|
|
{
|
|
|
|
return search->count();
|
|
|
|
}
|
|
|
|
|
|
|
|
int SearchModel::columnCount(const QModelIndex &) const
|
|
|
|
{
|
|
|
|
return Columns::COUNT;
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariant SearchModel::data(const QModelIndex &index, int role) const
|
|
|
|
{
|
|
|
|
if (index.row() >= search->count())
|
|
|
|
return QVariant();
|
|
|
|
|
|
|
|
const SearchDescription &exp = search->at(index.row());
|
|
|
|
|
2018-03-21 20:32:32 +00:00
|
|
|
switch (role) {
|
2018-03-08 12:24:15 +00:00
|
|
|
case Qt::DisplayRole:
|
2018-03-21 20:32:32 +00:00
|
|
|
switch (index.column()) {
|
2018-03-08 12:24:15 +00:00
|
|
|
case OFFSET:
|
|
|
|
return RAddressString(exp.offset);
|
|
|
|
case SIZE:
|
|
|
|
return RSizeString(exp.size);
|
|
|
|
case CODE:
|
|
|
|
return exp.code;
|
|
|
|
case DATA:
|
|
|
|
return exp.data;
|
|
|
|
default:
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
case SearchDescriptionRole:
|
|
|
|
return QVariant::fromValue(exp);
|
|
|
|
default:
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariant SearchModel::headerData(int section, Qt::Orientation, int role) const
|
|
|
|
{
|
2018-03-21 20:32:32 +00:00
|
|
|
switch (role) {
|
2018-03-08 12:24:15 +00:00
|
|
|
case Qt::DisplayRole:
|
2018-03-21 20:32:32 +00:00
|
|
|
switch (section) {
|
2018-03-08 12:24:15 +00:00
|
|
|
case SIZE:
|
|
|
|
return tr("Size");
|
|
|
|
case OFFSET:
|
|
|
|
return tr("Offset");
|
|
|
|
case CODE:
|
|
|
|
return tr("Code");
|
|
|
|
case DATA:
|
|
|
|
return tr("Data");
|
|
|
|
default:
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void SearchModel::beginReloadSearch()
|
|
|
|
{
|
|
|
|
beginResetModel();
|
|
|
|
}
|
|
|
|
|
|
|
|
void SearchModel::endReloadSearch()
|
|
|
|
{
|
|
|
|
endResetModel();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
SearchSortFilterProxyModel::SearchSortFilterProxyModel(SearchModel *source_model, QObject *parent)
|
|
|
|
: QSortFilterProxyModel(parent)
|
|
|
|
{
|
|
|
|
setSourceModel(source_model);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool SearchSortFilterProxyModel::filterAcceptsRow(int row, const QModelIndex &parent) const
|
|
|
|
{
|
|
|
|
QModelIndex index = sourceModel()->index(row, 0, parent);
|
2018-03-21 20:32:32 +00:00
|
|
|
SearchDescription search = index.data(
|
|
|
|
SearchModel::SearchDescriptionRole).value<SearchDescription>();
|
2018-03-08 12:24:15 +00:00
|
|
|
return search.code.contains(filterRegExp());
|
|
|
|
}
|
|
|
|
|
|
|
|
bool SearchSortFilterProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const
|
|
|
|
{
|
2018-03-21 20:32:32 +00:00
|
|
|
SearchDescription left_search = left.data(
|
|
|
|
SearchModel::SearchDescriptionRole).value<SearchDescription>();
|
|
|
|
SearchDescription right_search = right.data(
|
|
|
|
SearchModel::SearchDescriptionRole).value<SearchDescription>();
|
2018-03-08 12:24:15 +00:00
|
|
|
|
2018-03-21 20:32:32 +00:00
|
|
|
switch (left.column()) {
|
2018-03-08 12:24:15 +00:00
|
|
|
case SearchModel::SIZE:
|
|
|
|
return left_search.size < right_search.size;
|
|
|
|
case SearchModel::OFFSET:
|
|
|
|
return left_search.offset < right_search.offset;
|
|
|
|
case SearchModel::CODE:
|
|
|
|
return left_search.code < right_search.code;
|
|
|
|
case SearchModel::DATA:
|
|
|
|
return left_search.data < right_search.data;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return left_search.offset < right_search.offset;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2018-03-16 21:46:57 +00:00
|
|
|
SearchWidget::SearchWidget(MainWindow *main, QAction *action) :
|
|
|
|
CutterDockWidget(main, action),
|
|
|
|
ui(new Ui::SearchWidget)
|
2018-03-08 12:24:15 +00:00
|
|
|
{
|
|
|
|
ui->setupUi(this);
|
|
|
|
|
|
|
|
search_model = new SearchModel(&search, this);
|
|
|
|
search_proxy_model = new SearchSortFilterProxyModel(search_model, this);
|
|
|
|
ui->searchTreeView->setModel(search_proxy_model);
|
|
|
|
ui->searchTreeView->sortByColumn(SearchModel::OFFSET, Qt::AscendingOrder);
|
|
|
|
|
|
|
|
setScrollMode();
|
|
|
|
|
|
|
|
connect(Core(), SIGNAL(refreshAll()), this, SLOT(refreshSearchspaces()));
|
|
|
|
|
|
|
|
QShortcut *enter_press = new QShortcut(QKeySequence(Qt::Key_Return), this);
|
|
|
|
connect(enter_press, &QShortcut::activated, this, [this]() {
|
|
|
|
refreshSearch();
|
|
|
|
});
|
|
|
|
enter_press->setContext(Qt::WidgetWithChildrenShortcut);
|
|
|
|
|
|
|
|
connect(ui->searchButton, &QAbstractButton::clicked, this, [this]() {
|
|
|
|
refreshSearch();
|
|
|
|
});
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
SearchWidget::~SearchWidget() {}
|
|
|
|
|
|
|
|
void SearchWidget::on_searchTreeView_doubleClicked(const QModelIndex &index)
|
|
|
|
{
|
2018-03-21 20:32:32 +00:00
|
|
|
SearchDescription search = index.data(
|
|
|
|
SearchModel::SearchDescriptionRole).value<SearchDescription>();
|
2018-03-08 12:24:15 +00:00
|
|
|
Core()->seek(search.offset);
|
|
|
|
}
|
|
|
|
|
|
|
|
void SearchWidget::searchChanged()
|
|
|
|
{
|
|
|
|
refreshSearchspaces();
|
|
|
|
}
|
|
|
|
|
|
|
|
void SearchWidget::refreshSearchspaces()
|
|
|
|
{
|
|
|
|
int cur_idx = ui->searchspaceCombo->currentIndex();
|
|
|
|
if (cur_idx < 0)
|
|
|
|
cur_idx = 0;
|
|
|
|
|
|
|
|
ui->searchspaceCombo->clear();
|
|
|
|
ui->searchspaceCombo->addItem(tr("asm code"), QVariant("/cj"));
|
|
|
|
ui->searchspaceCombo->addItem(tr("string"), QVariant("/j"));
|
|
|
|
ui->searchspaceCombo->addItem(tr("hex string"), QVariant("/xj"));
|
2018-03-16 09:07:41 +00:00
|
|
|
ui->searchspaceCombo->addItem(tr("ROP gadgets"), QVariant("/Rj"));
|
2018-03-21 20:32:32 +00:00
|
|
|
|
2018-03-08 12:24:15 +00:00
|
|
|
if (cur_idx > 0)
|
|
|
|
ui->searchspaceCombo->setCurrentIndex(cur_idx);
|
|
|
|
|
|
|
|
refreshSearch();
|
|
|
|
}
|
|
|
|
|
|
|
|
void SearchWidget::refreshSearch()
|
|
|
|
{
|
|
|
|
QString search_for = ui->filterLineEdit->text();
|
|
|
|
QVariant searchspace_data = ui->searchspaceCombo->currentData();
|
|
|
|
QString searchspace = searchspace_data.toString();
|
|
|
|
|
|
|
|
search_model->beginReloadSearch();
|
|
|
|
search = Core()->getAllSearch(search_for, searchspace);
|
|
|
|
search_model->endReloadSearch();
|
|
|
|
|
2018-04-01 08:25:31 +00:00
|
|
|
qhelpers::adjustColumns(ui->searchTreeView, 3, 0);
|
2018-03-08 12:24:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void SearchWidget::setScrollMode()
|
|
|
|
{
|
|
|
|
qhelpers::setVerticalScrollMode(ui->searchTreeView);
|
|
|
|
}
|