From bdaf3e65c7ea19da175656a02e8d16af5f13becc Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Fri, 15 Mar 2019 15:48:41 -0400
Subject: Updated triage example

---
 python/examples/triage/sections.py | 147 +++++++++++++++++++++++++++++++++++++
 1 file changed, 147 insertions(+)
 create mode 100644 python/examples/triage/sections.py

(limited to 'python/examples/triage/sections.py')

diff --git a/python/examples/triage/sections.py b/python/examples/triage/sections.py
new file mode 100644
index 00000000..4a8d0f65
--- /dev/null
+++ b/python/examples/triage/sections.py
@@ -0,0 +1,147 @@
+# coding: utf8
+
+from PySide2.QtWidgets import QWidget, QLabel, QGridLayout, QHBoxLayout
+from binaryninja.enums import SectionSemantics
+import binaryninjaui
+from binaryninjaui import ThemeColor, ViewFrame
+from . import headers
+
+
+class SegmentsWidget(QWidget):
+	def __init__(self, parent, data):
+		super(SegmentsWidget, self).__init__(parent)
+
+		layout = QGridLayout()
+		layout.setContentsMargins(0, 0, 0, 0)
+		layout.setVerticalSpacing(1)
+		layout.setHorizontalSpacing(16)
+
+		self.segments = []
+		for segment in data.segments:
+			if segment.readable or segment.writable or segment.executable:
+				self.segments.append(segment)
+		self.segments.sort(key = lambda segment: segment.start)
+
+		row = 0
+		for segment in self.segments:
+			begin = "0x%x" % segment.start
+			end = "0x%x" % segment.end
+
+			permissions = ""
+			if segment.readable:
+				permissions += "r"
+			else:
+				permissions += "-"
+			if segment.writable:
+				permissions += "w"
+			else:
+				permissions += "-"
+			if segment.executable:
+				permissions += "x"
+			else:
+				permissions += "-"
+
+			rangeLayout = QHBoxLayout()
+			rangeLayout.setContentsMargins(0, 0, 0, 0)
+			beginLabel = headers.ClickableAddressLabel(begin)
+			dashLabel = QLabel("-")
+			dashLabel.setFont(binaryninjaui.getMonospaceFont(self))
+			endLabel = headers.ClickableAddressLabel(end)
+			rangeLayout.addWidget(beginLabel)
+			rangeLayout.addWidget(dashLabel)
+			rangeLayout.addWidget(endLabel)
+			layout.addLayout(rangeLayout, row, 0)
+
+			permissionsLabel = QLabel(permissions)
+			permissionsLabel.setFont(binaryninjaui.getMonospaceFont(self))
+			layout.addWidget(permissionsLabel, row, 1)
+
+			row += 1
+
+		layout.setColumnStretch(2, 1)
+		self.setLayout(layout)
+
+
+class SectionsWidget(QWidget):
+	def __init__(self, parent, data):
+		super(SectionsWidget, self).__init__(parent)
+
+		layout = QGridLayout()
+		layout.setContentsMargins(0, 0, 0, 0)
+		layout.setVerticalSpacing(1)
+		layout.setHorizontalSpacing(16)
+
+		maxNameLen = 0
+		for section in data.sections.values():
+			if len(section.name) > maxNameLen:
+				maxNameLen = len(section.name)
+		if maxNameLen > 32:
+			maxNameLen = 32
+
+		self.sections = []
+		for section in data.sections.values():
+			if section.semantics != SectionSemantics.ExternalSectionSemantics:
+				self.sections.append(section)
+		self.sections.sort(key = lambda section: section.start)
+
+		row = 0
+		for section in self.sections:
+			name = section.name
+			if len(name) > maxNameLen:
+				name = name[:maxNameLen - 1] + "…"
+
+			begin = "0x%x" % section.start
+			end = "0x%x" % section.end
+			typeName = section.type
+
+			permissions = ""
+			if data.is_offset_readable(section.start):
+				permissions += "r"
+			else:
+				permissions += "-"
+			if data.is_offset_writable(section.start):
+				permissions += "w"
+			else:
+				permissions += "-"
+			if data.is_offset_executable(section.start):
+				permissions += "x"
+			else:
+				permissions += "-"
+
+			semantics = ""
+			if section.semantics == SectionSemantics.ReadOnlyCodeSectionSemantics:
+				semantics = "Code"
+			elif section.semantics == SectionSemantics.ReadOnlyDataSectionSemantics:
+				semantics = "Ready-only Data"
+			elif section.semantics == SectionSemantics.ReadWriteDataSectionSemantics:
+				semantics = "Writable Data"
+
+			nameLabel = QLabel(name)
+			nameLabel.setFont(binaryninjaui.getMonospaceFont(self))
+			layout.addWidget(nameLabel, row, 0)
+
+			rangeLayout = QHBoxLayout()
+			rangeLayout.setContentsMargins(0, 0, 0, 0)
+			beginLabel = headers.ClickableAddressLabel(begin)
+			dashLabel = QLabel("-")
+			dashLabel.setFont(binaryninjaui.getMonospaceFont(self))
+			endLabel = headers.ClickableAddressLabel(end)
+			rangeLayout.addWidget(beginLabel)
+			rangeLayout.addWidget(dashLabel)
+			rangeLayout.addWidget(endLabel)
+			layout.addLayout(rangeLayout, row, 1)
+
+			permissionsLabel = QLabel(permissions)
+			permissionsLabel.setFont(binaryninjaui.getMonospaceFont(self))
+			layout.addWidget(permissionsLabel, row, 2)
+			typeLabel = QLabel(typeName)
+			typeLabel.setFont(binaryninjaui.getMonospaceFont(self))
+			layout.addWidget(typeLabel, row, 3)
+			semanticsLabel = QLabel(semantics)
+			semanticsLabel.setFont(binaryninjaui.getMonospaceFont(self))
+			layout.addWidget(semanticsLabel, row, 4)
+
+			row += 1
+
+		layout.setColumnStretch(5, 1)
+		self.setLayout(layout)
-- 
cgit v1.3.1