mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
cclauss
116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
from __future__ import print_function
|
|
# -----------------------------------------------------------------------
|
|
# Debugger command prompt with CustomViewers
|
|
# (c) Hex-Rays
|
|
#
|
|
import ida_idaapi
|
|
import ida_kernwin
|
|
import ida_lines
|
|
import ida_expr
|
|
import ida_dbg
|
|
|
|
# The viewer instance
|
|
dbgcmd = None
|
|
|
|
# -----------------------------------------------------------------------
|
|
class base_dbgcmd_ah_t(ida_kernwin.action_handler_t):
|
|
def __init__(self):
|
|
ida_kernwin.action_handler_t.__init__(self)
|
|
|
|
def update(self, ctx):
|
|
if dbgcmd and ctx.widget == dbgcmd.GetWidget():
|
|
return ida_kernwin.AST_ENABLE_FOR_WIDGET
|
|
else:
|
|
return ida_kernwin.AST_DISABLE_FOR_WIDGET
|
|
|
|
# -----------------------------------------------------------------------
|
|
class clear_dbgcmd_ah_t(base_dbgcmd_ah_t):
|
|
def activate(self, ctx):
|
|
dbgcmd.ResetOutput()
|
|
|
|
# -----------------------------------------------------------------------
|
|
class newcmd_dbgcmd_ah_t(base_dbgcmd_ah_t):
|
|
def activate(self, ctx):
|
|
dbgcmd.IssueCommand()
|
|
|
|
# -----------------------------------------------------------------------
|
|
class close_dbgcmd_ah_t(base_dbgcmd_ah_t):
|
|
def activate(self, ctx):
|
|
dbgcmd.Close()
|
|
|
|
# -----------------------------------------------------------------------
|
|
# Register actions (if needed)
|
|
ACTNAME_CLEAR = "dbgcmd:clear"
|
|
ACTNAME_NEWCMD = "dbgcmd:newcmd"
|
|
ACTNAME_CLOSE = "dbgcmd:close"
|
|
ida_kernwin.register_action(
|
|
ida_kernwin.action_desc_t(
|
|
ACTNAME_CLEAR, "Clear", clear_dbgcmd_ah_t(), "x"))
|
|
ida_kernwin.register_action(
|
|
ida_kernwin.action_desc_t(
|
|
ACTNAME_NEWCMD, "New command", newcmd_dbgcmd_ah_t(), "Insert"))
|
|
ida_kernwin.register_action(
|
|
ida_kernwin.action_desc_t(
|
|
ACTNAME_CLOSE, "Close", close_dbgcmd_ah_t(), "Escape"))
|
|
|
|
# -----------------------------------------------------------------------
|
|
class dbgcmd_t(ida_kernwin.simplecustviewer_t):
|
|
def Create(self):
|
|
# Form the title
|
|
title = "Debugger command window"
|
|
# Create the customview
|
|
if not ida_kernwin.simplecustviewer_t.Create(self, title):
|
|
return False
|
|
self.last_cmd = ""
|
|
self.ResetOutput()
|
|
return True
|
|
|
|
def IssueCommand(self):
|
|
s = ida_kernwin.ask_str(self.last_cmd, 0, "Please enter a debugger command")
|
|
if not s:
|
|
return
|
|
|
|
# Save last command
|
|
self.last_cmd = s
|
|
|
|
# Add it using a different color
|
|
self.AddLine("debugger>" + ida_lines.COLSTR(s, ida_lines.SCOLOR_VOIDOP))
|
|
|
|
ok, out = ida_dbg.send_dbg_command(s)
|
|
if ok:
|
|
for line in out.split("\n"):
|
|
self.AddLine(ida_lines.COLSTR(line, ida_lines.SCOLOR_LIBNAME))
|
|
else:
|
|
self.AddLine(
|
|
ida_lines.COLSTR(
|
|
"Debugger is not active or does not export ida_dbg.send_dbg_command() (%s)" % out,
|
|
ida_lines.SCOLOR_ERROR))
|
|
self.Refresh()
|
|
|
|
def ResetOutput(self):
|
|
self.ClearLines()
|
|
self.AddLine(ida_lines.COLSTR("Please press INS to enter command; X to clear output", ida_lines.SCOLOR_AUTOCMT))
|
|
self.Refresh()
|
|
|
|
|
|
# -----------------------------------------------------------------------
|
|
def show_win():
|
|
x = dbgcmd_t()
|
|
if not x.Create():
|
|
print("Failed to create debugger command line!")
|
|
return None
|
|
x.Show()
|
|
|
|
# Attach actions to this widget's popup menu
|
|
widget = x.GetWidget()
|
|
ida_kernwin.attach_action_to_popup(widget, None, ACTNAME_CLEAR)
|
|
ida_kernwin.attach_action_to_popup(widget, None, ACTNAME_NEWCMD)
|
|
ida_kernwin.attach_action_to_popup(widget, None, ACTNAME_CLOSE)
|
|
return x
|
|
|
|
if dbgcmd is not None:
|
|
dbgcmd.Close()
|
|
dbgcmd = None
|
|
|
|
dbgcmd = show_win()
|