"""
Dwarf - Copyright (C) 2018-2020 Giovanni Rocca (iGio90)
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>
"""
from PyQt5.Qt import Qt
from PyQt5.QtGui import QFontMetrics
from PyQt5.QtWidgets import *
from dwarf_debugger.ui.widgets.code_editor import JsCodeEditor
from dwarf_debugger.ui.dialogs.dwarf_dialog import DwarfDialog
class InputDialogTextEdit(JsCodeEditor):
def __init__(self, dialog, *__args):
super().__init__(*__args)
self.dialog = dialog
self.setStyleSheet('padding: 0; padding: 0 5px;')
bar = QScrollBar()
bar.setFixedHeight(0)
bar.setFixedWidth(0)
font_metric = QFontMetrics(self.font())
row_height = font_metric.lineSpacing()
self.setFixedHeight(row_height + 10) # 10 == 2*5px padding
self.setMinimumWidth(400)
def keyPressEvent(self, event):
# when code completion popup dont respond to enter
if self.completer and self.completer.popup() and self.completer.popup().isVisible():
event.ignore()
super(InputDialogTextEdit, self).keyPressEvent(event)
else:
if event.key() == Qt.Key_Return:
self.dialog.accept()
else:
super(InputDialogTextEdit, self).keyPressEvent(event)
class InputDialog(DwarfDialog):
def __init__(self, parent=None, hint=None, input_content='', placeholder='', options_callback=None):
super(InputDialog, self).__init__(parent)
box = QVBoxLayout(self)
if hint:
label = QLabel(hint)
label.setAttribute(Qt.WA_TranslucentBackground, True) # keep this
box.addWidget(label)
# wtf this hack to prevent segfault on adding breakpoint with shortcuts from breakpoints panel.
# use qtextedit instead of qlineedit won't cause issues
self.input_widget = InputDialogTextEdit(self)
self.input_widget.setPlaceholderText(placeholder)
if len(input_content) > 0:
self.input_widget.setPlainText(input_content)
tc = self.input_widget.textCursor()
tc.setPosition(len(input_content))
self.input_widget.setTextCursor(tc)
box.addWidget(self.input_widget)
buttons = QHBoxLayout()
ok = QPushButton('ok')
buttons.addWidget(ok)
if options_callback:
options = QPushButton('options')
options.clicked.connect(options_callback)
buttons.addWidget(options)
ok.clicked.connect(self.accept)
cancel = QPushButton('cancel')
cancel.clicked.connect(self.close)
buttons.addWidget(cancel)
box.addLayout(buttons)
self.setLayout(box)
def keyPressEvent(self, event):
if event.key() == Qt.Key_Return:
self.accept()
else:
super(InputDialog, self).keyPressEvent(event)
@staticmethod
def input(parent=None, hint=None, input_content='', placeholder='', options_callback=None):
dialog = InputDialog(parent=parent, hint=hint, input_content=input_content,
placeholder=placeholder, options_callback=options_callback)
dialog.title = hint
result = dialog.exec_()
text = dialog.input_widget.toPlainText()
return result == QDialog.Accepted, text
@staticmethod
def input_pointer(parent=None, input_content='', hint='insert pointer'):
accept, inp = InputDialog.input(
parent=parent,
hint=hint,
input_content=input_content,
placeholder='Module.findExportByName(\'target\', \'export\')')
if not accept:
return 0, ''
try:
return int(parent.dwarf.dwarf_api('evaluatePtr', inp), 16), inp
except:
return 0, ''
