blob: 37b46f310f90e7ccd0a873e4032478dcdb1fdf9f (
plain)
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
|
#pragma once
#include <QtWidgets/QDialog>
#include <QtWidgets/QLineEdit>
#include <QtWidgets/QPushButton>
#include <QtWidgets/QTextEdit>
#include <QtWidgets/QSplitter>
#include <QtWidgets/QTextBrowser>
#include <QtWidgets/QTreeWidget>
#include <QtWidgets/QCheckBox>
#include <QtGui/QKeyEvent>
#include "uitypes.h"
#include "binaryninjaapi.h"
constexpr int IndexRole = Qt::UserRole;
constexpr int ItemRole = Qt::UserRole + 1;
constexpr int LocationRole = Qt::UserRole + 2;
constexpr int NameColumn = 1;
constexpr int LocationColumn = 2;
class BackgroundTreeWidget : public QTreeWidget
{
Q_OBJECT
public:
explicit BackgroundTreeWidget(QWidget *parent);
Q_SIGNALS:
void addressDoubleClicked(uint64_t address);
protected:
virtual void keyPressEvent(QKeyEvent* event) override;
};
/*!
\ingroup uiapi
*/
class BINARYNINJAUIAPI BndbImportDialog : public QDialog
{
Q_OBJECT
QLineEdit* m_fileEdit;
QPushButton* m_browseButton;
QWidget* m_resultsWidget;
QTreeWidget* m_typesTree;
QPushButton* m_previewButton;
QPushButton* m_importButton;
BinaryViewRef m_data;
std::string m_filePath;
struct SymbolAndType
{
SymbolAndType(SymbolRef name, TypeRef type): name(name), type(type) {}
SymbolRef name;
TypeRef type;
};
std::vector<BinaryNinja::QualifiedNameAndType> m_types;
std::vector<SymbolAndType> m_functions;
std::vector<SymbolAndType> m_functionsToImports;
std::vector<SymbolAndType> m_dataVariables;
enum ItemType {
TypeItem,
FunctionItem,
FunctionToImportItem,
DataVariableItem
};
BinaryViewRef m_incomingView;
LoggerRef m_logger;
protected Q_SLOTS:
void browseFile();
void updateButtons();
void previewTypes();
void importTypes();
protected:
virtual void keyPressEvent(QKeyEvent* event) override;
private:
bool loadTypes();
bool isExistingType(const BinaryNinja::QualifiedName& name, bool function) const;
bool isBuiltinType(const BinaryNinja::QualifiedName& name) const;
void ApplyFunctionTypes(const std::vector<SymbolAndType>& functions);
void ApplyFunctionTypesToImports(const std::vector<SymbolAndType>& functions);
void ApplyDataVariables(const std::vector<SymbolAndType>& dataVariables);
std::vector<SymbolRef> matchingSymbol(const SymbolRef sym, std::vector<BNSymbolType> allowed, bool allowPrefix);
void navigateToItem(QTreeWidgetItem *item, int column);
bool inSymbolBlackList(const SymbolRef sym);
public:
BndbImportDialog(QWidget* parent, BinaryViewRef view);
~BndbImportDialog() = default;
};
|
