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
|
/*
Copyright (C) 2006 - 2015 Evan Teran
evan.teran@gmail.com
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 2 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 <http://www.gnu.org/licenses/>.
*/
#include "DialogFunctions.h"
#include "DialogResults.h"
#include "IAnalyzer.h"
#include "MemoryRegions.h"
#include "edb.h"
#include <QDialog>
#include <QHeaderView>
#include <QMenu>
#include <QMessageBox>
#include <QPushButton>
#include <QSortFilterProxyModel>
namespace FunctionFinderPlugin {
/**
* @brief DialogFunctions::DialogFunctions
* @param parent
* @param f
*/
DialogFunctions::DialogFunctions(QWidget *parent, Qt::WindowFlags f)
: QDialog(parent, f) {
ui.setupUi(this);
ui.tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents);
filterModel_ = new QSortFilterProxyModel(this);
connect(ui.txtSearch, &QLineEdit::textChanged, filterModel_, &QSortFilterProxyModel::setFilterFixedString);
buttonFind_ = new QPushButton(QIcon::fromTheme("edit-find"), tr("Find"));
connect(buttonFind_, &QPushButton::clicked, this, [this]() {
buttonFind_->setEnabled(false);
ui.progressBar->setValue(0);
doFind();
ui.progressBar->setValue(100);
buttonFind_->setEnabled(true);
});
ui.buttonBox->addButton(buttonFind_, QDialogButtonBox::ActionRole);
}
/**
* @brief DialogFunctions::showEvent
*/
void DialogFunctions::showEvent(QShowEvent *) {
filterModel_->setFilterKeyColumn(3);
filterModel_->setSourceModel(&edb::v1::memory_regions());
ui.tableView->setModel(filterModel_);
ui.progressBar->setValue(0);
}
/**
* @brief DialogFunctions::doFind
*/
void DialogFunctions::doFind() {
if (IAnalyzer *const analyzer = edb::v1::analyzer()) {
const QItemSelectionModel *const selModel = ui.tableView->selectionModel();
const QModelIndexList sel = selModel->selectedRows();
if (sel.size() == 0) {
QMessageBox::critical(this, tr("No Region Selected"), tr("You must select a region which is to be scanned for functions."));
return;
}
auto analyzer_object = dynamic_cast<QObject *>(analyzer);
if (analyzer_object) {
connect(analyzer_object, SIGNAL(updateProgress(int)), ui.progressBar, SLOT(setValue(int)));
}
auto resultsDialog = new DialogResults(this);
for (const QModelIndex &selected_item : sel) {
const QModelIndex index = filterModel_->mapToSource(selected_item);
// do the search for this region!
if (auto region = *reinterpret_cast<const std::shared_ptr<IRegion> *>(index.internalPointer())) {
analyzer->analyze(region);
const IAnalyzer::FunctionMap &results = analyzer->functions(region);
for (const Function &function : results) {
resultsDialog->addResult(function);
}
}
}
if (resultsDialog->resultCount() == 0) {
QMessageBox::information(this, tr("No Results"), tr("No Functions Found!"));
delete resultsDialog;
} else {
resultsDialog->show();
}
if (analyzer_object) {
disconnect(analyzer_object, SIGNAL(updateProgress(int)), ui.progressBar, SLOT(setValue(int)));
}
}
}
}
|
