From 73a9399f5d1d86c7e291498b7fe836550f2f76ab Mon Sep 17 00:00:00 2001 From: Rusty Wagner Date: Tue, 2 Apr 2019 15:30:45 -0400 Subject: Add a C++ version of the triage plugin (no Makefile yet) --- examples/triage/imports.h | 75 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 75 insertions(+) create mode 100644 examples/triage/imports.h (limited to 'examples/triage/imports.h') diff --git a/examples/triage/imports.h b/examples/triage/imports.h new file mode 100644 index 00000000..cacda087 --- /dev/null +++ b/examples/triage/imports.h @@ -0,0 +1,75 @@ +#pragma once + +#include +#include +#include "filter.h" + + +class GenericImportsModel: public QAbstractItemModel +{ + std::vector m_allEntries, m_entries; + bool m_hasModules; + int m_nameCol, m_moduleCol, m_ordinalCol; + int m_totalCols, m_sortCol; + Qt::SortOrder m_sortOrder; + + QString getNamespace(SymbolRef sym) const; + void performSort(int col, Qt::SortOrder order); + +public: + GenericImportsModel(BinaryViewRef data); + + 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 std::string& filterText); + + SymbolRef getSymbol(const QModelIndex& index); + + bool HasOrdinalCol() const { return m_ordinalCol != -1; } + int GetOrdinalCol() const { return m_ordinalCol; } +}; + + +class TriageView; +class ImportsWidget; + +class ImportsTreeView: public QTreeView, public FilterTarget +{ + BinaryViewRef m_data; + ImportsWidget* m_parent; + TriageView* m_view; + UIActionHandler m_actionHandler; + GenericImportsModel* m_model; + +public: + ImportsTreeView(ImportsWidget* parent, TriageView* view, BinaryViewRef data); + + virtual void setFilter(const std::string& filterText) override; + virtual void scrollToFirstItem() override; + virtual void scrollToCurrentItem() override; + virtual void selectFirstItem() override; + virtual void activateFirstItem() override; + virtual void closeFilter() override; + +protected: + virtual void keyPressEvent(QKeyEvent* event) override; + +private Q_SLOTS: + void importSelected(const QModelIndex& cur, const QModelIndex& prev); + void importDoubleClicked(const QModelIndex& cur); +}; + + +class ImportsWidget: public QWidget +{ + FilteredView* m_filter; + +public: + ImportsWidget(QWidget* parent, TriageView* view, BinaryViewRef data); + void showFilter(const QString& filter); +}; -- cgit v1.3.1