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

1 files changed, 151 insertions, 0 deletions

diff --git a/examples/triage/sections.cpp b/examples/triage/sections.cpp
new file mode 100644
index 00000000..b6acfd86
--- /dev/null
+++ b/examples/triage/sections.cpp
@@ -0,0 +1,151 @@
+#include <QtWidgets/QGridLayout>
+#include <QtWidgets/QHBoxLayout>
+#include <algorithm>
+#include <vector>
+#include "sections.h"
+#include "headers.h"
+#include "fontsettings.h"
+
+
+SegmentsWidget::SegmentsWidget(QWidget* parent, BinaryViewRef data): QWidget(parent)
+{
+	QGridLayout* layout = new QGridLayout();
+	layout->setContentsMargins(0, 0, 0, 0);
+	layout->setVerticalSpacing(1);
+	layout->setHorizontalSpacing(UIContext::getScaledWindowSize(16, 16).width());
+
+	for (auto& segment: data->GetSegments())
+		if ((segment->GetFlags() & (SegmentReadable | SegmentWritable | SegmentExecutable)) != 0)
+			m_segments.push_back(segment);
+	sort(m_segments.begin(), m_segments.end(), [&](SegmentRef a, SegmentRef b) {
+		return a->GetStart() < b->GetStart();
+	});
+
+	int row = 0;
+	for (auto& segment: m_segments)
+	{
+		QString begin = QString("0x") + QString::number(segment->GetStart(), 16);
+		QString end = QString("0x") + QString::number(segment->GetEnd(), 16);
+
+		QString permissions;
+		if (segment->GetFlags() & SegmentReadable)
+			permissions += "r";
+		else
+			permissions += "-";
+		if (segment->GetFlags() & SegmentWritable)
+			permissions += "w";
+		else
+			permissions += "-";
+		if (segment->GetFlags() & SegmentExecutable)
+			permissions += "x";
+		else
+			permissions += "-";
+
+		QHBoxLayout* rangeLayout = new QHBoxLayout();
+		rangeLayout->setContentsMargins(0, 0, 0, 0);
+		NavigationAddressLabel* beginLabel = new NavigationAddressLabel(begin);
+		QLabel* dashLabel = new QLabel("-");
+		dashLabel->setFont(getMonospaceFont(this));
+		NavigationAddressLabel* endLabel = new NavigationAddressLabel(end);
+		rangeLayout->addWidget(beginLabel);
+		rangeLayout->addWidget(dashLabel);
+		rangeLayout->addWidget(endLabel);
+		layout->addLayout(rangeLayout, row, 0);
+
+		QLabel* permissionsLabel = new QLabel(permissions);
+		permissionsLabel->setFont(getMonospaceFont(this));
+		layout->addWidget(permissionsLabel, row, 1);
+
+		row++;
+	}
+
+	layout->setColumnStretch(2, 1);
+	setLayout(layout);
+}
+
+
+SectionsWidget::SectionsWidget(QWidget* parent, BinaryViewRef data): QWidget(parent)
+{
+	QGridLayout* layout = new QGridLayout();
+	layout->setContentsMargins(0, 0, 0, 0);
+	layout->setVerticalSpacing(1);
+	layout->setHorizontalSpacing(UIContext::getScaledWindowSize(16, 16).width());
+
+	size_t maxNameLen = 0;
+	for (auto& section: data->GetSections())
+		if (section->GetName().size() > maxNameLen)
+			maxNameLen = section->GetName().size();
+	if (maxNameLen > 32)
+		maxNameLen = 32;
+
+	for (auto& section: data->GetSections())
+		if (section->GetSemantics() != ExternalSectionSemantics)
+			m_sections.push_back(section);
+	sort(m_sections.begin(), m_sections.end(), [&](SectionRef a, SectionRef b) {
+		return a->GetStart() < b->GetStart();
+	});
+
+	int row = 0;
+	for (auto& section: m_sections)
+	{
+		std::string name = section->GetName();
+		if (name.size() > maxNameLen)
+			name = name.substr(0, maxNameLen - 1) + std::string("…");
+
+		QString begin = QString("0x") + QString::number(section->GetStart(), 16);
+		QString end = QString("0x") + QString::number(section->GetStart() + section->GetLength(), 16);
+		QString typeName = QString::fromStdString(section->GetType());
+
+		QString permissions;
+		if (data->IsOffsetReadable(section->GetStart()))
+			permissions += "r";
+		else
+			permissions += "-";
+		if (data->IsOffsetWritable(section->GetStart()))
+			permissions += "w";
+		else
+			permissions += "-";
+		if (data->IsOffsetExecutable(section->GetStart()))
+			permissions += "x";
+		else
+			permissions += "-";
+
+		QString semantics;
+		if (section->GetSemantics() == ReadOnlyCodeSectionSemantics)
+			semantics = "Code";
+		else if (section->GetSemantics() == ReadOnlyDataSectionSemantics)
+			semantics = "Ready-only Data";
+		else if (section->GetSemantics() == ReadWriteDataSectionSemantics)
+			semantics = "Writable Data";
+
+		QLabel* nameLabel = new QLabel(QString::fromStdString(name));
+		nameLabel->setFont(getMonospaceFont(this));
+		layout->addWidget(nameLabel, row, 0);
+
+		QHBoxLayout* rangeLayout = new QHBoxLayout();
+		rangeLayout->setContentsMargins(0, 0, 0, 0);
+		NavigationAddressLabel* beginLabel = new NavigationAddressLabel(begin);
+		QLabel* dashLabel = new QLabel("-");
+		dashLabel->setFont(getMonospaceFont(this));
+		NavigationAddressLabel* endLabel = new NavigationAddressLabel(end);
+		rangeLayout->addWidget(beginLabel);
+		rangeLayout->addWidget(dashLabel);
+		rangeLayout->addWidget(endLabel);
+		layout->addLayout(rangeLayout, row, 1);
+
+		QLabel* permissionsLabel = new QLabel(permissions);
+		permissionsLabel->setFont(getMonospaceFont(this));
+		layout->addWidget(permissionsLabel, row, 2);
+		QLabel* typeLabel = new QLabel(typeName);
+		typeLabel->setFont(getMonospaceFont(this));
+		layout->addWidget(typeLabel, row, 3);
+		QLabel* semanticsLabel = new QLabel(semantics);
+		semanticsLabel->setFont(getMonospaceFont(this));
+		layout->addWidget(semanticsLabel, row, 4);
+
+		row++;
+	}
+
+	layout->setColumnStretch(5, 1);
+	setLayout(layout);
+}