mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
#
|
|
# This example illustrates how one can execute commands in the
|
|
# "Output window", from their own widgets.
|
|
#
|
|
# In order to do so, we have to be careful that:
|
|
# - the original, underlying 'cli:Execute' action, that has to be
|
|
# triggered for the code present in the input field to execute
|
|
# and be placed in the history, requires that the input field
|
|
# has focus (otherwise it simply won't do anything.)
|
|
# - this, in turn, forces us to do "delayed" execution of that action,
|
|
# hence the need for a QTimer
|
|
# - the IDA/SWiG 'TWidget' type that we retrieve through
|
|
# `ida_kernwin.find_widget`, is not the same type as a
|
|
# `QtWidgets.QWidget`. We therefore need to convert it using
|
|
# `ida_kernwin.PluginForm.TWidgetToPyQtWidget`
|
|
#
|
|
|
|
from PyQt5 import QtCore
|
|
from PyQt5 import QtGui
|
|
from PyQt5 import QtWidgets
|
|
|
|
import ida_kernwin
|
|
import ida_segment
|
|
|
|
import idc
|
|
|
|
delayed_exec_timer = QtCore.QTimer()
|
|
|
|
def show_dialog():
|
|
dialog = QtWidgets.QDialog()
|
|
dialog.setWindowTitle("Inject command")
|
|
dialog.setMinimumSize(600, 480)
|
|
|
|
run_text = "Run"
|
|
buttons_box = QtWidgets.QDialogButtonBox()
|
|
button = buttons_box.addButton(run_text, QtWidgets.QDialogButtonBox.AcceptRole)
|
|
button.setDefault(True)
|
|
button.clicked.connect(dialog.accept)
|
|
|
|
text_edit = QtWidgets.QPlainTextEdit()
|
|
text_edit.setPlaceholderText(
|
|
"Type an expression, and press '%s' to execute through the regular input" % run_text)
|
|
|
|
layout = QtWidgets.QVBoxLayout()
|
|
layout.addWidget(text_edit)
|
|
layout.addWidget(buttons_box)
|
|
|
|
dialog.setLayout(layout)
|
|
|
|
# disable script timeout, otherwise a "Please wait ..." dialog
|
|
# might briefly show after the dialog was accepted/rejected
|
|
with ida_kernwin.disabled_script_timeout_t():
|
|
if dialog.exec_() == QtWidgets.QDialog.Accepted:
|
|
|
|
# We'll now have to schedule a call to the standard
|
|
# 'execute' action. We can't call it right away, because
|
|
# the "Output window" doesn't have focus, and thus
|
|
# the action will fail to execute since it requires
|
|
# the "Output window" as context.
|
|
text = text_edit.toPlainText()
|
|
|
|
def delayed_exec(*args):
|
|
output_window_title = "Output window"
|
|
tw = ida_kernwin.find_widget(output_window_title)
|
|
if not tw:
|
|
raise Exception("Couldn't find widget '%s'" % output_window_title)
|
|
|
|
# convert from a SWiG 'TWidget*' facade,
|
|
# into an object that PyQt will understand
|
|
w = ida_kernwin.PluginForm.TWidgetToPyQtWidget(tw)
|
|
|
|
line_edit = w.findChild(QtWidgets.QLineEdit)
|
|
if not line_edit:
|
|
raise Exception("Couldn't find input")
|
|
line_edit.setFocus() # ensure it has focus
|
|
QtWidgets.QApplication.instance().processEvents() # and that it received the focus event
|
|
|
|
# inject text into widget
|
|
line_edit.setText(text)
|
|
|
|
# and execute the standard 'execute' action
|
|
ida_kernwin.process_ui_action("cli:Execute")
|
|
|
|
delayed_exec_timer.singleShot(0, delayed_exec)
|
|
|
|
|
|
show_dialog()
|