class WatchpointsWidget(QWidget):
""" WatchpointsWidget
Signals:
onItemSelected(addr_str) - item dblclicked
onItemAddClick
Constants:
MEMORY_ACCESS_READ = 1
MEMORY_ACCESS_WRITE = 2
MEMORY_ACCESS_EXECUTE = 4
MEMORY_WATCH_SINGLESHOT = 8
"""
MEMORY_ACCESS_READ = 1
MEMORY_ACCESS_WRITE = 2
MEMORY_ACCESS_EXECUTE = 4
MEMORY_WATCH_SINGLESHOT = 8
onItemDoubleClicked = pyqtSignal(int, name='onItemDoubleClicked')
onItemAdded = pyqtSignal(int, name='onItemAdded')
onItemRemoved = pyqtSignal(int, name='onItemRemoved')
def __init__(self, parent=None): # pylint: disable=too-many-statements
super(WatchpointsWidget, self).__init__(parent=parent)
self._app_window = parent
if self._app_window.dwarf is None:
print('WatchpointsWidget created before Dwarf exists')
return
self._uppercase_hex = True
self.setAutoFillBackground(True)
# connect to dwarf
self._app_window.dwarf.onWatchpointAdded.connect(
self._on_watchpoint_added)
self._app_window.dwarf.onWatchpointRemoved.connect(
self._on_watchpoint_removed)
# setup our model
self._watchpoints_model = QStandardItemModel(0, 5)
self._watchpoints_model.setHeaderData(0, Qt.Horizontal, 'Address')
self._watchpoints_model.setHeaderData(1, Qt.Horizontal, 'R')
self._watchpoints_model.setHeaderData(1, Qt.Horizontal, Qt.AlignCenter,
Qt.TextAlignmentRole)
self._watchpoints_model.setHeaderData(2, Qt.Horizontal, 'W')
self._watchpoints_model.setHeaderData(2, Qt.Horizontal, Qt.AlignCenter,
Qt.TextAlignmentRole)
self._watchpoints_model.setHeaderData(3, Qt.Horizontal, 'X')
self._watchpoints_model.setHeaderData(3, Qt.Horizontal, Qt.AlignCenter,
Qt.TextAlignmentRole)
self._watchpoints_model.setHeaderData(4, Qt.Horizontal, 'S')
self._watchpoints_model.setHeaderData(4, Qt.Horizontal, Qt.AlignCenter,
Qt.TextAlignmentRole)
# setup ui
v_box = QVBoxLayout(self)
v_box.setContentsMargins(0, 0, 0, 0)
self.list_view = DwarfListView()
self.list_view.setModel(self._watchpoints_model)
self.list_view.header().setSectionResizeMode(0, QHeaderView.Stretch)
self.list_view.header().setSectionResizeMode(
1, QHeaderView.ResizeToContents | QHeaderView.Fixed)
self.list_view.header().setSectionResizeMode(
2, QHeaderView.ResizeToContents | QHeaderView.Fixed)
self.list_view.header().setSectionResizeMode(
3, QHeaderView.ResizeToContents | QHeaderView.Fixed)
self.list_view.header().setSectionResizeMode(
4, QHeaderView.ResizeToContents | QHeaderView.Fixed)
self.list_view.header().setStretchLastSection(False)
self.list_view.doubleClicked.connect(self._on_item_dblclick)
self.list_view.setContextMenuPolicy(Qt.CustomContextMenu)
self.list_view.customContextMenuRequested.connect(self._on_contextmenu)
v_box.addWidget(self.list_view)
#header = QHeaderView(Qt.Horizontal, self)
h_box = QHBoxLayout()
h_box.setContentsMargins(5, 2, 5, 5)
btn1 = QPushButton(QIcon(utils.resource_path('assets/icons/plus.svg')),
'')
btn1.setFixedSize(20, 20)
btn1.clicked.connect(self._on_additem_clicked)
btn2 = QPushButton(QIcon(utils.resource_path('assets/icons/dash.svg')),
'')
btn2.setFixedSize(20, 20)
btn2.clicked.connect(self.delete_items)
btn3 = QPushButton(
QIcon(utils.resource_path('assets/icons/trashcan.svg')), '')
btn3.setFixedSize(20, 20)
btn3.clicked.connect(self.clear_list)
h_box.addWidget(btn1)
h_box.addWidget(btn2)
h_box.addSpacerItem(
QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Preferred))
h_box.addWidget(btn3)
# header.setLayout(h_box)
# header.setFixedHeight(25)
# v_box.addWidget(header)
v_box.addLayout(h_box)
# create a centered dot icon
_section_width = self.list_view.header().sectionSize(2)
self._new_pixmap = QPixmap(_section_width, 20)
self._new_pixmap.fill(Qt.transparent)
painter = QPainter(self._new_pixmap)
rect = QRect(int(_section_width * 0.5), 0, 20, 20)
painter.setBrush(QColor('#666'))
painter.setPen(QColor('#666'))
painter.drawEllipse(rect)
self._dot_icon = QIcon(self._new_pixmap)
# shortcuts
shortcut_add = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_W),
self._app_window, self._on_additem_clicked)
shortcut_add.setAutoRepeat(False)
self.setLayout(v_box)
# ************************************************************************
# **************************** Properties ********************************
# ************************************************************************
@property
def uppercase_hex(self):
""" Addresses displayed lower/upper-case
"""
return self._uppercase_hex
@uppercase_hex.setter
def uppercase_hex(self, value):
""" Addresses displayed lower/upper-case
value - bool or str
'upper', 'lower'
"""
if isinstance(value, bool):
self._uppercase_hex = value
elif isinstance(value, str):
self._uppercase_hex = (value == 'upper')
# ************************************************************************
# **************************** Functions *********************************
# ************************************************************************
def do_addwatchpoint_dlg(self, ptr=None): # pylint: disable=too-many-branches
""" Shows AddWatchpointDialog
"""
watchpoint_dlg = AddWatchpointDialog(self, ptr)
if watchpoint_dlg.exec_() == QDialog.Accepted:
mem_r = watchpoint_dlg.acc_read.isChecked()
mem_w = watchpoint_dlg.acc_write.isChecked()
mem_x = watchpoint_dlg.acc_execute.isChecked()
mem_s = watchpoint_dlg.singleshot.isChecked()
ptr = watchpoint_dlg.text_field.toPlainText()
if ptr:
if isinstance(ptr, str):
if ptr.startswith('0x') or ptr.startswith('#'):
ptr = utils.parse_ptr(ptr)
else:
try:
ptr = int(ptr, 10)
except ValueError:
pass
# int now?
if not isinstance(ptr, int):
try:
ptr = int(
self._app_window.dwarf.dwarf_api(
'evaluatePtr', ptr), 16)
except ValueError:
ptr = 0
if ptr == 0:
return
if not self._app_window.dwarf.dwarf_api(
'isValidPointer', ptr):
return
else:
return
mem_val = 0
if mem_r:
mem_val |= self.MEMORY_ACCESS_READ
if mem_w:
mem_val |= self.MEMORY_ACCESS_WRITE
if mem_x:
mem_val |= self.MEMORY_ACCESS_EXECUTE
if mem_s:
mem_val |= self.MEMORY_WATCH_SINGLESHOT
self.add_address(ptr, mem_val, from_api=False)
# return [ptr, mem_val]
def add_address(self, ptr, flags, from_api=False):
""" Adds Address to display
ptr - str or int
flags - int
"""
if isinstance(ptr, str):
ptr = utils.parse_ptr(ptr)
if not isinstance(flags, int):
try:
flags = int(flags, 10)
except ValueError:
flags = 3
if not from_api:
# function was called directly so add it to dwarf
if not self._app_window.dwarf.is_address_watched(ptr):
self._app_window.dwarf.dwarf_api('putWatchpoint', [ptr, flags])
return
# show header
self.list_view.setHeaderHidden(False)
# create items to add
if self._uppercase_hex:
str_frmt = '0x{0:X}'
else:
str_frmt = '0x{0:x}'
addr = QStandardItem()
addr.setText(str_frmt.format(ptr))
read = QStandardItem()
write = QStandardItem()
execute = QStandardItem()
singleshot = QStandardItem()
if flags & self.MEMORY_ACCESS_READ:
read.setIcon(self._dot_icon)
if flags & self.MEMORY_ACCESS_WRITE:
write.setIcon(self._dot_icon)
if flags & self.MEMORY_ACCESS_EXECUTE:
execute.setIcon(self._dot_icon)
if flags & self.MEMORY_WATCH_SINGLESHOT:
singleshot.setIcon(self._dot_icon)
# add items as new row on top
self._watchpoints_model.insertRow(
0, [addr, read, write, execute, singleshot])
def remove_address(self, ptr, from_api=False):
""" Remove Address from List
"""
if isinstance(ptr, str):
ptr = utils.parse_ptr(ptr)
if not from_api:
# called somewhere so remove watchpoint in dwarf too
self._app_window.dwarf.dwarf_api('removeWatchpoint', ptr)
return
str_frmt = ''
if self._uppercase_hex:
str_frmt = '0x{0:X}'.format(ptr)
else:
str_frmt = '0x{0:x}'.format(ptr)
model = self.list_view.model()
for item in range(model.rowCount()):
if str_frmt == model.item(item).text():
model.removeRow(item)
def delete_items(self):
""" Delete selected Items
"""
model = self.list_view.model()
index = self.list_view.selectionModel().currentIndex().row()
if index != -1:
ptr = model.item(index, 0).text()
self.remove_address(ptr)
def clear_list(self):
""" Clear the List
"""
model = self.list_view.model()
# go through all items and tell it gets removed
for item in range(model.rowCount()):
ptr = model.item(item, 0).text()
self.remove_address(ptr)
if model.rowCount() > 0:
# something was wrong it should be empty
model.removeRows(0, model.rowCount())
# ************************************************************************
# **************************** Handlers **********************************
# ************************************************************************
def _on_contextmenu(self, pos):
index = self.list_view.indexAt(pos).row()
glbl_pt = self.list_view.mapToGlobal(pos)
context_menu = QMenu(self)
context_menu.addAction('Add watchpoint', self._on_additem_clicked)
if index != -1:
context_menu.addSeparator()
context_menu.addAction(
'Copy address', lambda: utils.copy_hex_to_clipboard(
self._watchpoints_model.item(index, 0).text()))
context_menu.addAction(
'Jump to address', lambda: self._app_window.jump_to_address(
self._watchpoints_model.item(index, 0).text()))
context_menu.addAction(
'Delete watchpoint', lambda: self.remove_address(
self._watchpoints_model.item(index, 0).text()))
if self.list_view.search_enabled:
context_menu.addSeparator()
context_menu.addAction('Search', self.list_view._on_cm_search)
context_menu.exec_(glbl_pt)
def _on_item_dblclick(self, model_index):
row = self._watchpoints_model.itemFromIndex(model_index).row()
if row != -1:
ptr = self._watchpoints_model.item(row, 0).text()
self.onItemDoubleClicked.emit(ptr)
def _on_additem_clicked(self):
if self._app_window.dwarf.pid == 0:
return
self.do_addwatchpoint_dlg()
def _on_watchpoint_added(self, watchpoint):
""" Callback from Dwarf after Watchpoint is added
"""
# add to watchpointslist
self.add_address(watchpoint.address, watchpoint.flags, from_api=True)
self.onItemAdded.emit(watchpoint.address)
def _on_watchpoint_removed(self, ptr):
""" Callback from Dwarf after watchpoint is removed
"""
ptr = utils.parse_ptr(ptr)
# remove from list
self.remove_address(ptr, from_api=True)
self.onItemRemoved.emit(ptr)
class ProcessList(QWidget):
""" ProcessListWidget wich shows running Processes on Device
Includes a Refresh Button to manually start refreshthread
args:
device needed
Signals:
onProcessSelected([pid, name]) - pid(int) name(str)
onRefreshError(str)
"""
onProcessSelected = pyqtSignal(list, name='onProcessSelected')
onRefreshError = pyqtSignal(str, name='onRefreshError')
def __init__(self, device, parent=None):
super(ProcessList, self).__init__(parent=parent)
# if not isinstance(device, frida.core.Device):
# print('No FridaDevice')
# return
self._device = device
self.process_list = DwarfListView()
model = QStandardItemModel(0, 2, parent)
model.setHeaderData(0, Qt.Horizontal, "PID")
model.setHeaderData(0, Qt.Horizontal, Qt.AlignCenter,
Qt.TextAlignmentRole)
model.setHeaderData(1, Qt.Horizontal, "Name")
self.process_list.doubleClicked.connect(self._on_item_clicked)
v_box = QVBoxLayout()
v_box.setContentsMargins(0, 0, 0, 0)
v_box.addWidget(self.process_list)
self.refresh_button = QPushButton('Refresh')
self.refresh_button.clicked.connect(self._on_refresh_procs)
self.refresh_button.setEnabled(False)
v_box.addWidget(self.refresh_button)
self.setLayout(v_box)
self.process_list.setModel(model)
self.process_list.header().setSectionResizeMode(
0, QHeaderView.ResizeToContents)
self.procs_update_thread = ProcsThread(self, self._device)
self.procs_update_thread.add_proc.connect(self._on_add_proc)
self.procs_update_thread.is_error.connect(self._on_error)
self.procs_update_thread.is_finished.connect(self._on_refresh_finished)
self.procs_update_thread.device = self._device
self.procs_update_thread.start()
# ************************************************************************
# **************************** Properties ********************************
# ************************************************************************
@property
def device(self):
""" Sets Device needs frida.core.device
"""
return self._device
@device.setter
def device(self, value):
self.set_device(value)
# ************************************************************************
# **************************** Functions *********************************
# ************************************************************************
def clear(self):
""" Clears the List
"""
self.process_list.clear()
def set_device(self, device):
""" Set frida Device
"""
if isinstance(device, frida.core.Device):
self._device = device
self.procs_update_thread.device = device
self._on_refresh_procs()
# ************************************************************************
# **************************** Handlers **********************************
# ************************************************************************
def _on_item_clicked(self, model_index):
model = self.process_list.model()
index = model.itemFromIndex(model_index).row()
if index != -1:
sel_pid = self.process_list.get_item_text(index, 0)
sel_name = self.process_list.get_item_text(index, 1)
self.onProcessSelected.emit([int(sel_pid), sel_name])
def _on_add_proc(self, item):
model = self.process_list.model()
pid = QStandardItem()
pid.setText(str(item['pid']))
pid.setTextAlignment(Qt.AlignCenter)
name = QStandardItem()
name.setText(item['name'])
model.appendRow([pid, name])
def _on_error(self, error_str):
self.onRefreshError.emit(error_str)
def _on_refresh_procs(self):
if not self._device:
return
if self.procs_update_thread.isRunning():
self.procs_update_thread.terminate()
if not self.procs_update_thread.isRunning():
self.clear()
self.refresh_button.setEnabled(False)
self.procs_update_thread.device = self._device
self.procs_update_thread.start()
def _on_refresh_finished(self):
self.refresh_button.setEnabled(True)
class SpawnsList(QWidget):
""" ProcessListWidget wich shows running Processes on Device
Includes a Refresh Button to manually start refreshthread
args:
device needed
Signals:
onProcessSelected([pid, name]) - pid(int) name(str)
onRefreshError(str)
"""
onProcessSelected = pyqtSignal(list, name='onProcessSelected')
onRefreshError = pyqtSignal(str, name='onRefreshError')
def __init__(self, device, parent=None):
super(SpawnsList, self).__init__(parent=parent)
self.break_at_start = False
self._device = device
self.spawn_list = DwarfListView()
model = QStandardItemModel(0, 2, parent)
model.setHeaderData(0, Qt.Horizontal, "Name")
model.setHeaderData(1, Qt.Horizontal, "Package")
self.spawn_list.doubleClicked.connect(self._on_item_clicked)
v_box = QVBoxLayout()
v_box.setContentsMargins(0, 0, 0, 0)
v_box.addWidget(self.spawn_list)
break_spawn_start = QCheckBox('Break at spawn')
break_spawn_start.stateChanged.connect(self._on_toggle_break_spawn)
v_box.addWidget(break_spawn_start)
self.refresh_button = QPushButton('Refresh')
self.refresh_button.clicked.connect(self._on_refresh_procs)
self.refresh_button.setEnabled(False)
v_box.addWidget(self.refresh_button)
self.setLayout(v_box)
self.spawn_list.setModel(model)
self.spawn_list.header().setSectionResizeMode(
0, QHeaderView.ResizeToContents)
self.spaw_update_thread = SpawnsThread(self, self._device)
self.spaw_update_thread.add_spawn.connect(self._on_add_proc)
self.spaw_update_thread.is_error.connect(self._on_error)
self.spaw_update_thread.is_finished.connect(self._on_refresh_finished)
self.spaw_update_thread.device = self._device
self.spaw_update_thread.start()
# ************************************************************************
# **************************** Properties ********************************
# ************************************************************************
@property
def device(self):
""" Sets Device needs frida.core.device
"""
return self._device
@device.setter
def device(self, value):
self.set_device(value)
# ************************************************************************
# **************************** Functions *********************************
# ************************************************************************
def clear(self):
""" Clears the List
"""
self.spawn_list.clear()
def set_device(self, device):
""" Set frida Device
"""
if isinstance(device, frida.core.Device):
self._device = device
self.spaw_update_thread.device = device
self._on_refresh_procs()
# ************************************************************************
# **************************** Handlers **********************************
# ************************************************************************
def _on_item_clicked(self, model_index):
model = self.spawn_list.model()
index = model.itemFromIndex(model_index).row()
if index != -1:
sel_pid = self.spawn_list.get_item_text(index, 0)
if model_index.column() == 0:
sel_name = model.data(model_index, Qt.UserRole + 2)
else:
sel_name = self.spawn_list.get_item_text(index, 1)
self.onProcessSelected.emit([sel_pid, sel_name])
def _on_add_proc(self, item):
model = self.spawn_list.model()
name = QStandardItem()
name.setText(item[0])
name.setData(item[1], Qt.UserRole + 2)
package = QStandardItem()
package.setText(item[1])
model.appendRow([name, package])
def _on_error(self, error_str):
self.onRefreshError.emit(error_str)
def _on_refresh_procs(self):
if not self._device:
return
if self.spaw_update_thread.isRunning():
self.spaw_update_thread.terminate()
if not self.spaw_update_thread.isRunning():
self.clear()
self.refresh_button.setEnabled(False)
self.spaw_update_thread.device = self._device
self.spaw_update_thread.start()
def _on_refresh_finished(self):
self.spawn_list.resizeColumnToContents(0)
self.refresh_button.setEnabled(True)
def _on_toggle_break_spawn(self, state):
self.break_at_start = state == Qt.Checked