Add a C++ version of the triage plugin (no Makefile yet)

1 files changed, 99 insertions, 0 deletions

diff --git a/examples/triage/headers.h b/examples/triage/headers.h
new file mode 100644
index 00000000..0a9af5ff
--- /dev/null
+++ b/examples/triage/headers.h
@@ -0,0 +1,99 @@
+#pragma once
+
+#include <QtWidgets/QLabel>
+#include <QtWidgets/QWidget>
+#include <functional>
+#include "uitypes.h"
+
+
+class NavigationLabel: public QLabel
+{
+	std::function<void()> m_func;
+
+public:
+	NavigationLabel(const QString& text, QColor color, const std::function<void()>& func);
+
+protected:
+	virtual void mousePressEvent(QMouseEvent* event) override;
+};
+
+
+class NavigationAddressLabel: public NavigationLabel
+{
+	uint64_t m_address;
+
+	void clickEvent();
+
+public:
+	NavigationAddressLabel(const QString& text);
+};
+
+
+class NavigationCodeLabel: public NavigationLabel
+{
+	uint64_t m_address;
+
+	void clickEvent();
+
+public:
+	NavigationCodeLabel(const QString& text);
+};
+
+
+enum HeaderFieldType
+{
+	TextHeaderField,
+	AddressHeaderField,
+	CodeHeaderField
+};
+
+
+struct HeaderField
+{
+	QString title;
+	std::vector<QString> values;
+	HeaderFieldType type;
+};
+
+
+class Headers
+{
+	std::vector<HeaderField> m_fields;
+	size_t m_columns, m_rowsPerColumn;
+
+public:
+	Headers();
+	void AddField(const QString& title, const QString& value, HeaderFieldType type = TextHeaderField);
+	void AddField(const QString& title, const std::vector<QString>& values, HeaderFieldType type = TextHeaderField);
+	const std::vector<HeaderField>& GetFields() const { return m_fields; }
+	void SetColumns(size_t cols) { m_columns = cols; }
+	void SetRowsPerColumn(size_t rows) { m_rowsPerColumn = rows; }
+	size_t GetColumns() const { return m_columns; }
+	size_t GetRowsPerColumn() const { return m_rowsPerColumn; }
+};
+
+
+class GenericHeaders: public Headers
+{
+public:
+	GenericHeaders(BinaryViewRef data);
+};
+
+
+class PEHeaders: public Headers
+{
+	uint64_t GetValueOfStructMember(BinaryViewRef data, const std::string& structName, uint64_t structStart,
+		const std::string& fieldName);
+	uint64_t GetAddressAfterStruct(BinaryViewRef data, const std::string& structName, uint64_t structStart);
+	QString GetNameOfEnumerationMember(BinaryViewRef data, const std::string& enumName, uint64_t value);
+
+public:
+	PEHeaders(BinaryViewRef data);
+};
+
+
+class HeaderWidget: public QWidget
+{
+public:
+	HeaderWidget(QWidget* parent, const Headers& headers);
+};