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
|
#pragma once
#include <QtCore/QAbstractItemModel>
#include <QtCore/QTimer>
#include <QtWidgets/QTreeView>
#include "filter.h"
class GenericStringsModel : public QAbstractItemModel, public BinaryNinja::BinaryDataNotification
{
Q_OBJECT
BinaryViewRef m_data;
std::vector<BNStringReference> m_allEntries, m_entries;
int m_totalCols, m_sortCol;
Qt::SortOrder m_sortOrder;
QTimer* m_updateTimer;
QString m_filter;
FilterOptions m_filterOptions;
QRegularExpression m_filterRegex;
// Read from arbitrary threads while processing notifications.
std::atomic<bool> m_updatesPaused = false;
// Read/written from arbitrary threads while processing notifications.
std::atomic<bool> m_needsUpdate = true;
// Tracks if notifications arrived while paused
std::atomic<bool> m_dirtyWhilePaused = false;
void performSort(int col, Qt::SortOrder order);
void updateModel();
void applyFilter();
void updateTimer(bool);
void setNeedsUpdate(bool);
void onBinaryViewNotification();
signals:
void updateTimerOnUIThread();
public:
GenericStringsModel(QWidget* parent, BinaryViewRef data);
virtual ~GenericStringsModel();
virtual int columnCount(const QModelIndex& parent) const override;
virtual int rowCount(const QModelIndex& parent) const override;
virtual QVariant data(const QModelIndex& index, int role) const override;
virtual QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
virtual QModelIndex index(int row, int col, const QModelIndex& parent) const override;
virtual QModelIndex parent(const QModelIndex& index) const override;
virtual void sort(int col, Qt::SortOrder order) override;
void setFilter(const QString& filterText, FilterOptions options);
void pauseUpdates();
void resumeUpdates();
BNStringReference getStringRefAt(const QModelIndex& index) const;
QString stringRefToQString(const BNStringReference& index) const;
virtual void OnStringFound(BinaryNinja::BinaryView* data, BNStringType type, uint64_t offset, size_t len) override;
virtual void OnStringRemoved(BinaryNinja::BinaryView* data, BNStringType type, uint64_t offset, size_t len) override;
virtual void OnDerivedStringFound(BinaryNinja::BinaryView* data, const BinaryNinja::DerivedString& str) override;
virtual void OnDerivedStringRemoved(BinaryNinja::BinaryView* data, const BinaryNinja::DerivedString& str) override;
};
class TriageView;
class StringsWidget;
class StringsTreeView : public QTreeView, public FilterTarget
{
Q_OBJECT
BinaryViewRef m_data;
StringsWidget* m_parent;
TriageView* m_view;
UIActionHandler m_actionHandler;
GenericStringsModel* m_model;
void updateColumnWidths();
public:
StringsTreeView(StringsWidget* parent, TriageView* view, BinaryViewRef data);
void copySelection();
bool canCopySelection() const;
virtual void setFilter(const std::string& filterText, FilterOptions options) override;
virtual void scrollToFirstItem() override;
virtual void scrollToCurrentItem() override;
virtual void ensureSelection() override;
virtual void activateSelection() override;
virtual void closeFilter() override;
protected:
virtual void keyPressEvent(QKeyEvent* event) override;
virtual bool event(QEvent* event) override;
virtual void showEvent(QShowEvent* event) override;
virtual void hideEvent(QHideEvent* event) override;
private Q_SLOTS:
void stringSelected(const QModelIndex& cur, const QModelIndex& prev);
void stringDoubleClicked(const QModelIndex& cur);
};
class StringsWidget : public QWidget
{
FilteredView* m_filter;
public:
StringsWidget(QWidget* parent, TriageView* view, BinaryViewRef data);
void showFilter(const QString& filter);
};
|
