cutter/src/widgets/SearchWidget.cpp

203 lines
5.3 KiB
C++
Raw Normal View History

#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());
switch (role)
{
case Qt::DisplayRole:
switch (index.column())
{
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
{
switch (role)
{
case Qt::DisplayRole:
switch (section)
{
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);
SearchDescription search = index.data(SearchModel::SearchDescriptionRole).value<SearchDescription>();
return search.code.contains(filterRegExp());
}
bool SearchSortFilterProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const
{
SearchDescription left_search = left.data(SearchModel::SearchDescriptionRole).value<SearchDescription>();
SearchDescription right_search = right.data(SearchModel::SearchDescriptionRole).value<SearchDescription>();
switch (left.column())
{
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;
}
SearchWidget::SearchWidget(MainWindow *main, QWidget *parent) :
QDockWidget(parent),
ui(new Ui::SearchWidget),
main(main)
{
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)
{
SearchDescription search = index.data(SearchModel::SearchDescriptionRole).value<SearchDescription>();
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"));
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();
ui->searchTreeView->resizeColumnToContents(0);
ui->searchTreeView->resizeColumnToContents(1);
ui->searchTreeView->resizeColumnToContents(2);
}
void SearchWidget::setScrollMode()
{
qhelpers::setVerticalScrollMode(ui->searchTreeView);
}