def __init__(self, parent, data):
super(SegmentsWidget, self).__init__(parent)
layout = QGridLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setVerticalSpacing(1)
layout.setHorizontalSpacing(
UIContext.getScaledWindowSize(16, 16).width())
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)
def __init__(self,
DISPLAY_LINE_NUMBERS=True,
HIGHLIGHT_CURRENT_LINE=True,
SyntaxHighlighter=Pylighter,
lang="python",
font_size=11,
delimeter=" ",
*args):
super(QCodeEditor, self).__init__()
font = getMonospaceFont(self)
font.setPointSize(font_size)
self.setFont(font)
self.setLineWrapMode(QPlainTextEdit.NoWrap)
self.completionState = 0
self.completing = False
self.delimeter = delimeter
self.completer = bncompleter.Completer()
self.cursorPositionChanged.connect(self.resetCompletion)
self.DISPLAY_LINE_NUMBERS = DISPLAY_LINE_NUMBERS
if DISPLAY_LINE_NUMBERS:
self.number_bar = self.NumberBar(self)
if SyntaxHighlighter is not None: # add highlighter to textdocument
self.highlighter = SyntaxHighlighter(self.document(), lang)
def __init__(self, parent, name, data):
if not type(data) == binaryninja.binaryview.BinaryView:
raise Exception('expected widget data to be a BinaryView')
self.bv = data
QWidget.__init__(self, parent)
DockContextHandler.__init__(self, self, name)
self.actionHandler = UIActionHandler()
self.actionHandler.setupActionHandler(self)
layout = QVBoxLayout()
self.consoleText = QTextEdit(self)
self.consoleText.setReadOnly(True)
self.consoleText.setFont(getMonospaceFont(self))
layout.addWidget(self.consoleText, 1)
inputLayout = QHBoxLayout()
inputLayout.setContentsMargins(4, 4, 4, 4)
promptLayout = QVBoxLayout()
promptLayout.setContentsMargins(0, 5, 0, 5)
inputLayout.addLayout(promptLayout)
self.consoleEntry = QLineEdit(self)
inputLayout.addWidget(self.consoleEntry, 1)
label = QLabel("dbg>>> ", self)
label.setFont(getMonospaceFont(self))
promptLayout.addWidget(label)
promptLayout.addStretch(1)
self.consoleEntry.returnPressed.connect(lambda: self.sendLine())
layout.addLayout(inputLayout)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(0)
self.setLayout(layout)
def __init__(self, parent, name):
QWidget.__init__(self, parent)
DockContextHandler.__init__(self, self, name)
self.actionHandler = UIActionHandler()
self.actionHandler.setupActionHandler(self)
font = getMonospaceFont(self)
fm = QFontMetricsF(font)
table_layout = QVBoxLayout()
self.table = QTableView()
self.table.setFont(font)
self.table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.table.verticalHeader().hide()
self.table.verticalHeader().setSectionResizeMode(QHeaderView.Fixed)
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# sorting
self.table.setSortingEnabled(True)
self.table.horizontalHeader().setSortIndicator(0, QtCore.Qt.AscendingOrder)
data = []
self.model = TableModel(data)
self.table.setModel(self.model)
table_layout.addWidget(self.table)
layout = QVBoxLayout()
layout.addLayout(table_layout)
self.setLayout(layout)
# init double click action
self.table.doubleClicked.connect(self._ui_entry_double_click)
# init right click menu
self.ctx_menu = QMenu()
self._action_patch = QAction("Invert Branch", None)
self.ctx_menu.addAction(self._action_patch)
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self._ui_table_ctx_menu_handler)
self.bv = None
self.filename = None
self.do_sync = True
def __init__(self, bv, match_db, role, primary_be, secondary_be):
super(BindiffViewerDialog, self).__init__()
self.bv = bv
self.primary_be = primary_be
self.secondary_be = secondary_be
self.role = role
# UI
self.match_model = BindiffMatchModel(bv, match_db, role, primary_be, secondary_be)
self.match_view = QTreeView()
self.match_view.setModel(self.match_model)
self.match_view.setSelectionMode(QTreeView.ExtendedSelection)
self.match_view.setContextMenuPolicy(Qt.CustomContextMenu)
self.match_view.customContextMenuRequested.connect(self.match_view_context_menu_requested)
self.match_view.doubleClicked.connect(self.match_view_double_clicked)
self.match_view.setRootIsDecorated(False)
self.match_view.setFont(binaryninjaui.getMonospaceFont(self))
for i in range(len(self.match_model.column_infos)):
self.match_view.resizeColumnToContents(i)
self.match_view.setSortingEnabled(True)
self.match_view.sortByColumn(0, Qt.AscendingOrder)
layout = QVBoxLayout()
layout.addWidget(self.match_view)
self.setLayout(layout)
self.setWindowTitle("BinDiff Viewer")
self.resize(1000, 640)
flags = self.windowFlags()
flags |= Qt.WindowMaximizeButtonHint
flags &= ~Qt.WindowContextHelpButtonHint
self.setWindowFlags(flags)
def __init__(self, parent, name):
global panes
panes.append(self)
QWidget.__init__(self, parent)
DockContextHandler.__init__(self, self, name)
self.actionHandler = UIActionHandler()
self.actionHandler.setupActionHandler(self)
# Top: Headers with line info
header_layout = QFormLayout()
self.function_info = QLabel("")
self.line_info = QLabel("")
header_layout.addRow(self.tr("Function:"), self.function_info)
header_layout.addRow(self.tr("Line:"), self.line_info)
# Middle: main source display pane
textbox_layout = QVBoxLayout()
self.textbox = QPlainTextEdit()
self.textbox.setLineWrapMode(QPlainTextEdit.LineWrapMode.NoWrap)
self.textbox.setReadOnly(True)
font = getMonospaceFont(self)
self.textbox.setFont(font)
font = QFontMetrics(font)
self.textbox.setMinimumWidth(40 * font.averageCharWidth())
self.textbox.setMinimumHeight(30 * font.lineSpacing())
textbox_layout.addWidget(self.textbox)
# Bottom: buttons for stop/start, and substitution paths
footer_layout = QVBoxLayout()
sync_button_layout = QHBoxLayout()
self.sync_button = QPushButton("Turn Source Sync Off")
sync_button_layout.addWidget(self.sync_button)
path_layout = QFormLayout()
self.original_path = QLineEdit()
self.substitute_path = QLineEdit()
self.substitute_path_button = QPushButton("Do Path Substitution")
path_layout.addRow(self.tr("Original Path:"), self.original_path)
path_layout.addRow(self.substitute_path_button, self.substitute_path)
footer_layout.addLayout(sync_button_layout)
footer_layout.addLayout(path_layout)
# Putting all the child layouts together
layout = QVBoxLayout()
layout.addLayout(header_layout)
layout.addLayout(textbox_layout)
layout.addLayout(footer_layout)
self.setLayout(layout)
# Set up button signals
self.substitute_path_button.clicked.connect(self.do_path_substitution)
self.sync_button.clicked.connect(self.toggle_sync)
# Actual storage variables
self.bv = None
self.filename = None
self.do_sync = True
self.path_substitutions = {}
self.failed_substitutions = []
def __init__(self, context, parent=None):
super(Snippets, self).__init__(parent)
# Create widgets
self.setWindowFlags(self.windowFlags()
& ~Qt.WindowContextHelpButtonHint)
self.title = QLabel(self.tr("Snippet Editor"))
self.saveButton = QPushButton(self.tr("&Save"))
self.saveButton.setShortcut(QKeySequence(self.tr("Ctrl+S")))
self.runButton = QPushButton(self.tr("&Run"))
self.runButton.setShortcut(QKeySequence(self.tr("Ctrl+R")))
self.closeButton = QPushButton(self.tr("Close"))
self.clearHotkeyButton = QPushButton(self.tr("Clear Hotkey"))
self.setWindowTitle(self.title.text())
#self.newFolderButton = QPushButton("New Folder")
self.browseButton = QPushButton("Browse Snippets")
self.browseButton.setIcon(QIcon.fromTheme("edit-undo"))
self.deleteSnippetButton = QPushButton("Delete")
self.newSnippetButton = QPushButton("New Snippet")
indentation = Settings().get_string("snippets.indentation")
if Settings().get_bool("snippets.syntaxHighlight"):
self.edit = QCodeEditor(SyntaxHighlighter=Pylighter,
delimeter=indentation)
else:
self.edit = QCodeEditor(SyntaxHighlighter=None,
delimeter=indentation)
self.edit.setPlaceholderText("python code")
self.resetting = False
self.columns = 3
self.context = context
self.keySequenceEdit = QKeySequenceEdit(self)
self.currentHotkey = QKeySequence()
self.currentHotkeyLabel = QLabel("")
self.currentFileLabel = QLabel()
self.currentFile = ""
self.snippetDescription = QLineEdit()
self.snippetDescription.setPlaceholderText("optional description")
#Set Editbox Size
font = getMonospaceFont(self)
self.edit.setFont(font)
font = QFontMetrics(font)
self.edit.setTabStopDistance(
4 * font.horizontalAdvance(' ')) #TODO, replace with settings API
#Files
self.files = QFileSystemModel()
self.files.setRootPath(snippetPath)
self.files.setNameFilters(["*.py"])
#Tree
self.tree = QTreeView()
self.tree.setModel(self.files)
self.tree.setSortingEnabled(True)
self.tree.hideColumn(2)
self.tree.sortByColumn(0, Qt.AscendingOrder)
self.tree.setRootIndex(self.files.index(snippetPath))
for x in range(self.columns):
#self.tree.resizeColumnToContents(x)
self.tree.header().setSectionResizeMode(
x, QHeaderView.ResizeToContents)
treeLayout = QVBoxLayout()
treeLayout.addWidget(self.tree)
treeButtons = QHBoxLayout()
#treeButtons.addWidget(self.newFolderButton)
treeButtons.addWidget(self.browseButton)
treeButtons.addWidget(self.newSnippetButton)
treeButtons.addWidget(self.deleteSnippetButton)
treeLayout.addLayout(treeButtons)
treeWidget = QWidget()
treeWidget.setLayout(treeLayout)
# Create layout and add widgets
buttons = QHBoxLayout()
buttons.addWidget(self.clearHotkeyButton)
buttons.addWidget(self.keySequenceEdit)
buttons.addWidget(self.currentHotkeyLabel)
buttons.addWidget(self.closeButton)
buttons.addWidget(self.runButton)
buttons.addWidget(self.saveButton)
description = QHBoxLayout()
description.addWidget(QLabel(self.tr("Description: ")))
description.addWidget(self.snippetDescription)
vlayoutWidget = QWidget()
vlayout = QVBoxLayout()
vlayout.addLayout(description)
vlayout.addWidget(self.edit)
vlayout.addLayout(buttons)
vlayoutWidget.setLayout(vlayout)
hsplitter = QSplitter()
hsplitter.addWidget(treeWidget)
hsplitter.addWidget(vlayoutWidget)
hlayout = QHBoxLayout()
hlayout.addWidget(hsplitter)
self.showNormal() #Fixes bug that maximized windows are "stuck"
#Because you can't trust QT to do the right thing here
if (sys.platform == "darwin"):
self.settings = QSettings("Vector35", "Snippet Editor")
else:
self.settings = QSettings("Vector 35", "Snippet Editor")
if self.settings.contains("ui/snippeteditor/geometry"):
self.restoreGeometry(
self.settings.value("ui/snippeteditor/geometry"))
else:
self.edit.setMinimumWidth(80 * font.averageCharWidth())
self.edit.setMinimumHeight(30 * font.lineSpacing())
# Set dialog layout
self.setLayout(hlayout)
# Add signals
self.saveButton.clicked.connect(self.save)
self.closeButton.clicked.connect(self.close)
self.runButton.clicked.connect(self.run)
self.clearHotkeyButton.clicked.connect(self.clearHotkey)
self.tree.selectionModel().selectionChanged.connect(self.selectFile)
self.newSnippetButton.clicked.connect(self.newFileDialog)
self.deleteSnippetButton.clicked.connect(self.deleteSnippet)
#self.newFolderButton.clicked.connect(self.newFolder)
self.browseButton.clicked.connect(self.browseSnippets)
if self.settings.contains("ui/snippeteditor/selected"):
selectedName = self.settings.value("ui/snippeteditor/selected")
self.tree.selectionModel().select(
self.files.index(selectedName),
QItemSelectionModel.ClearAndSelect | QItemSelectionModel.Rows)
if self.tree.selectionModel().hasSelection():
self.selectFile(self.tree.selectionModel().selection(), None)
self.edit.setFocus()
cursor = self.edit.textCursor()
cursor.setPosition(self.edit.document().characterCount() - 1)
self.edit.setTextCursor(cursor)
else:
self.readOnly(True)
else:
self.readOnly(True)
def __init__(self, parent, data):
super(SectionsWidget, self).__init__(parent)
layout = QGridLayout()
layout.setContentsMargins(0, 0, 0, 0)
layout.setVerticalSpacing(1)
layout.setHorizontalSpacing(
UIContext.getScaledWindowSize(16, 16).width())
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)
