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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
|
#pragma once
#include <QtCore/QTimer>
#include <QtWidgets/QLabel>
#include <QtWidgets/QLineEdit>
#include <QtWidgets/QPushButton>
#include <QtWidgets/QTextEdit>
#include <QtWidgets/QPlainTextEdit>
#include <QtWidgets/QWidget>
#include <QtWidgets/QDialog>
#include <QtWidgets/QListView>
#include <QtCore/QAbstractListModel>
#include <QtCore/QMimeData>
#include <mutex>
#include <string>
#include <utility>
#include <vector>
#include <functional>
#include "binaryninjaapi.h"
#include "action.h"
#include "globalarea.h"
#include "uitypes.h"
#define SCRIPT_OUTPUT_UPDATE_INTERVAL 100
class ScriptingConsole;
class BINARYNINJAUIAPI ScriptingCompletionModel : public QAbstractListModel
{
Q_OBJECT
std::vector<std::string> m_completions;
bool m_searching;
public:
ScriptingCompletionModel(QWidget* parent);
virtual QModelIndex index(int row, int col, const QModelIndex& parent = QModelIndex()) const override;
virtual QModelIndex parent(const QModelIndex& i) const override;
virtual bool hasChildren(const QModelIndex& parent) const override;
virtual int rowCount(const QModelIndex& parent = QModelIndex()) const override;
virtual int columnCount(const QModelIndex& parent) const override;
virtual QVariant data(const QModelIndex& i, int role) const override;
void setModelData(const std::vector<std::string>& completions, bool searching);
};
class BINARYNINJAUIAPI ScriptingCompletionPopup : public QDialog
{
Q_OBJECT
QListView* m_list;
ScriptingCompletionModel* m_model;
protected:
virtual bool eventFilter(QObject* obj, QEvent* event) override;
public:
ScriptingCompletionPopup(QWidget* parent);
void showWithData(QPoint pt, int cursorSize, const std::vector<std::string>& completions, bool searching = false);
bool handleKeyEvent(QKeyEvent* event);
virtual ~ScriptingCompletionPopup();
private Q_SLOTS:
void clickRow(const QModelIndex& index);
Q_SIGNALS:
void complete(QString text);
};
class BINARYNINJAUIAPI ScriptingConsoleEdit : public QTextEdit
{
Q_OBJECT
public:
typedef std::function<std::vector<std::string>(const std::string&)> CompletionCallback;
private:
ScriptingConsole* m_console;
int m_charHeight;
bool m_continuation;
bool m_searching;
bool m_searchError;
CompletionCallback m_completionCallback;
ScriptingCompletionPopup* m_popup;
uint64_t m_completionRegionStart;
uint64_t m_completionRegionInitialStop;
uint64_t m_completionRegionStop;
public:
ScriptingConsoleEdit(ScriptingConsole* parent);
void setCharHeight(int height);
void setContinuation(bool cont);
void setCompletionCallback(CompletionCallback callback) { m_completionCallback = callback; }
void insertFromMimeData(const QMimeData* source) override;
private Q_SLOTS:
void complete(QString text);
protected:
virtual void keyPressEvent(QKeyEvent* event) override;
};
class BINARYNINJAUIAPI ScriptingConsoleOutput : public QPlainTextEdit
{
Q_OBJECT
ScriptingConsole* m_console;
UIActionHandler* m_handler;
UIActionHandler m_actionHandler;
ContextMenuManager* m_contextMenuManager;
BinaryViewRef m_data;
Menu* m_menu;
public:
ScriptingConsoleOutput(ScriptingConsole* parent, Menu* menu);
bool IsNavigable(const QString& str, const std::pair<int, int>& offsetLen, uint64_t& value, bool highlight) const;
protected:
void contextMenuEvent(QContextMenuEvent* event) override;
public Q_SLOTS:
virtual void mouseReleaseEvent(QMouseEvent* event) override;
void viewChanged(QWidget* frame);
};
class ScriptingConsoleWidget;
class BINARYNINJAUIAPI ScriptingConsole : public GlobalAreaWidget, BinaryNinja::ScriptingOutputListener
{
Q_OBJECT
struct ScriptOutput
{
std::string text;
bool isError;
};
QString m_providerName;
QString m_instanceName;
ScriptingInstanceRef m_instance;
ScriptingConsoleOutput* m_output;
ScriptingConsoleEdit* m_input;
QLabel* m_prompt;
QPushButton* m_button;
QTimer* m_runTimer;
bool m_scriptActive;
BinaryNinja::Ref<BinaryNinja::Logger> m_logger;
std::mutex m_mutex;
std::vector<ScriptOutput> m_pendingOutput;
QTimer* m_updateTimer;
BNScriptingProviderInputReadyState m_currentState;
QStringList m_history;
QString m_inputOutsideHistory;
int m_currentHistoryEntry;
static int m_stateUpdatedEventType;
private Q_SLOTS:
void updateTimerEvent();
void consoleTextChanged();
void cancel();
void showCancelButton();
Q_SIGNALS:
void viewChanged(QWidget* frame);
void onScriptExecution();
void onScriptCompletion();
protected:
void customEvent(QEvent* event) override;
public:
ScriptingConsole(
QWidget* parent, const QString& providerName, const QString& instanceName, ScriptingInstanceRef instance);
virtual ~ScriptingConsole();
QString getProviderName() const { return m_providerName; }
QString getInstanceName() const { return m_instanceName; }
ScriptingInstanceRef getInstance() { return m_instance; }
bool getScriptIsActive() const { return m_scriptActive; }
void clearConsole();
void hideConsole();
void addInput(const std::string& text);
virtual void NotifyOutput(const std::string& text) override;
virtual void NotifyError(const std::string& text) override;
virtual void NotifyInputReadyStateChanged(BNScriptingProviderInputReadyState state) override;
virtual void notifyViewChanged(ViewFrame* frame) override;
virtual void notifyFontChanged() override;
virtual void focus() override;
void moveUpInHistory();
void moveDownInHistory();
std::vector<std::string> reverseSearch(const QString& text);
void closing();
void runScriptFromFile(const std::string& filename);
};
|
