def __init__(self, app, *__args):
super().__init__(app)
self.app = app
self.tracing = False
self.trace_classes = []
self.trace_depth = 0
layout = QVBoxLayout()
buttons = QHBoxLayout()
self.btn_start = QPushButton('start')
self.btn_start.clicked.connect(self.start_trace)
self.btn_pause = QPushButton('pause')
self.btn_pause.setEnabled(False)
self.btn_pause.clicked.connect(self.pause_trace)
self.btn_stop = QPushButton('stop')
self.btn_stop.setEnabled(False)
self.btn_stop.clicked.connect(self.stop_trace)
buttons.addWidget(self.btn_start)
buttons.addWidget(self.btn_pause)
buttons.addWidget(self.btn_stop)
layout.addLayout(buttons)
self.setup_splitter = QSplitter()
self.events_list = QListWidget()
self.events_list.setVisible(False)
self.trace_list = QListWidget()
self.class_list = QListWidget()
self.trace_list.itemDoubleClicked.connect(self.trace_list_double_click)
self.class_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.class_list.customContextMenuRequested.connect(
self.show_class_list_menu)
self.class_list.itemDoubleClicked.connect(self.class_list_double_click)
self.current_class_search = ''
bar = QScrollBar()
bar.setFixedWidth(0)
bar.setFixedHeight(0)
self.trace_list.setHorizontalScrollBar(bar)
bar = QScrollBar()
bar.setFixedWidth(0)
bar.setFixedHeight(0)
self.class_list.setHorizontalScrollBar(bar)
self.setup_splitter.addWidget(self.trace_list)
self.setup_splitter.addWidget(self.class_list)
self.setup_splitter.setHandleWidth(1)
layout.addWidget(self.setup_splitter)
layout.addWidget(self.events_list)
self.setLayout(layout)
def __init__(self, app, *__args):
super().__init__(0, 3)
self.app = app
self.setHorizontalHeaderLabels(['tid', 'pc', 'symbol'])
self.verticalHeader().hide()
scrollbar = QScrollBar()
scrollbar.setFixedWidth(0)
scrollbar.setFixedHeight(0)
self.setHorizontalScrollBar(scrollbar)
self.itemDoubleClicked.connect(self.on_context_item_double_click)
self.horizontalHeader().setStretchLastSection(True)
self.setShowGrid(False)
class ScrollBar(QWidget):
""" 定义一个可以变换样式的滚动条 """
def __init__(self, externalScrollBar=None, parent=None):
super().__init__(parent)
self.externalScrollBar = externalScrollBar
# 实例化两个滚动条
self.minScrollBar = QScrollBar(Qt.Vertical, self)
self.maxScrollBar = QScrollBar(Qt.Vertical, self)
# 实例化一个控制滚动条显示的计时器
self.timer = QTimer(self)
# 初始化
self.initWidget()
self.associateScrollBar()
self.setQss()
def initWidget(self):
""" 初始化小部件 """
self.setFixedWidth(20)
self.minScrollBar.move(15, 0)
self.maxScrollBar.hide()
self.timer.setInterval(2000)
self.timer.timeout.connect(self.showMinScrollBar)
self.setAttribute(Qt.WA_TranslucentBackground)
# 分配ID
self.setObjectName('father')
self.minScrollBar.setObjectName('minScrollBar')
self.maxScrollBar.setObjectName('maxScrollBar')
def adjustSrollBarHeight(self):
""" 根据父级窗口的高度调整滚动条高度 """
if self.parent():
self.minScrollBar.setFixedHeight(self.parent().height() - 153)
self.maxScrollBar.setFixedHeight(self.parent().height() - 153)
def enterEvent(self, e: QEnterEvent):
""" 鼠标进入界面时显示大滚动条并停止秒表 """
self.maxScrollBar.show()
self.minScrollBar.hide()
self.timer.stop()
def leaveEvent(self, e):
""" 鼠标离开打开秒表 """
self.timer.start()
def showMinScrollBar(self):
""" 定时溢出时隐藏大滚动条 """
self.timer.stop()
self.maxScrollBar.hide()
self.minScrollBar.show()
def setQss(self):
""" 设置层叠样式 """
with open(r'resource\css\my_scrollBar.qss', encoding='utf-8') as f:
self.setStyleSheet(f.read())
def associateScrollBar(self):
""" 关联滚动条 """
if self.externalScrollBar:
# 设置最大值
self.minScrollBar.setMaximum(self.externalScrollBar.maximum())
self.maxScrollBar.setMaximum(self.externalScrollBar.maximum())
# 关联滚动条
self.externalScrollBar.valueChanged.connect(
lambda: self.minScrollBar.setValue(self.externalScrollBar.
value()))
self.minScrollBar.valueChanged.connect(self.__minScrollBarChanged)
self.maxScrollBar.valueChanged.connect(
lambda: self.minScrollBar.setValue(self.maxScrollBar.value()))
def __minScrollBarChanged(self):
""" minScrollBar改变时同时改变另外两个滚动条的值 """
self.maxScrollBar.setValue(self.minScrollBar.value())
self.externalScrollBar.setValue(self.minScrollBar.value())
def __init__(self, app, *__args):
super().__init__(app)
self.app = app
self.app.dwarf.onJavaTraceEvent.connect(self.on_event)
self.app.dwarf.onEnumerateJavaClassesStart.connect(self.on_enumeration_start)
self.app.dwarf.onEnumerateJavaClassesMatch.connect(self.on_enumeration_match)
self.app.dwarf.onEnumerateJavaClassesComplete.connect(self.on_enumeration_complete)
self.tracing = False
self.trace_classes = []
self.trace_depth = 0
layout = QVBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
self._record_icon = QIcon(utils.resource_path('assets/icons/record.png'))
self._pause_icon = QIcon(utils.resource_path('assets/icons/pause.png'))
self._stop_icon = QIcon(utils.resource_path('assets/icons/stop.png'))
self._tool_bar = QToolBar()
self._tool_bar.addAction('Start', self.start_trace)
self._tool_bar.addAction('Pause', self.pause_trace)
self._tool_bar.addAction('Stop', self.stop_trace)
self._tool_bar.addSeparator()
self._entries_lbl = QLabel('Entries: 0')
self._entries_lbl.setStyleSheet('color: #ef5350;')
self._entries_lbl.setContentsMargins(10, 0, 10, 2)
self._entries_lbl.setAttribute(Qt.WA_TranslucentBackground, True) # keep this
self._entries_lbl.setAlignment(Qt.AlignRight| Qt.AlignVCenter)
self._entries_lbl.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Preferred)
self._tool_bar.addWidget(self._entries_lbl)
layout.addWidget(self._tool_bar)
self.setup_splitter = QSplitter()
self.events_list = JavaTraceView(self)
self.events_list.setVisible(False)
self.trace_list = QListWidget()
self.class_list = QListWidget()
self.trace_list.itemDoubleClicked.connect(self.trace_list_double_click)
self.class_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.class_list.customContextMenuRequested.connect(self.show_class_list_menu)
self.class_list.itemDoubleClicked.connect(self.class_list_double_click)
self.current_class_search = ''
bar = QScrollBar()
bar.setFixedWidth(0)
bar.setFixedHeight(0)
self.trace_list.setHorizontalScrollBar(bar)
bar = QScrollBar()
bar.setFixedWidth(0)
bar.setFixedHeight(0)
self.class_list.setHorizontalScrollBar(bar)
self.setup_splitter.addWidget(self.trace_list)
self.setup_splitter.addWidget(self.class_list)
layout.addWidget(self.setup_splitter)
layout.addWidget(self.events_list)
self.setLayout(layout)
class MainUi(QMainWindow):
obj_num = 0
container = {}
def __init__(self, info):
super(MainUi, self).__init__()
self.icon = [ico for ico in info]
self.column = len(self.icon)
self.num = 10 # 界面头部展示模块数量
self.init_ui()
def init_ui(self):
self.setWindowIcon(QIcon('index.ico'))
self.setWindowTitle("工具")
self.resize(960, 600)
self.statusBar().show()
self._center()
self.main_widget = QWidget() # 创建窗口主部件
self.main_layout = QGridLayout() # 创建主部件的网格布局
self.main_widget.setLayout(self.main_layout) # 设置窗口主部件布局为网格布局
# 创建上侧部件
self.up_widget = QWidget()
self.up_widget.setObjectName('up_widget')
self.up_widget.setFixedHeight(80)
self.up_layout = QGridLayout()
self.up_widget.setLayout(self.up_layout)
# 创建滚动条
self.slide = QScrollBar(True)
self.slide.setObjectName('slide_roll_bar')
self.slide.setFixedHeight(const.SLIDE_HEIGHT)
self.slide.setMaximum(self.column)
self.slide.setMinimum(self.num)
self.slide.valueChanged.connect(self._value_change)
# 创建日志记录
# ************************************************##
name = "Analysis"
self.notice_btn = QPushButton(name)
self.notice_btn.setObjectName(name)
self.container["selectedNoticeName"] = name
self.notice_btn.setFixedSize(const.NOTICE_WIDTH, const.BTN_HEIGHT)
self.notice_btn.clicked.connect(self._clicked)
self.notice_close_btn = QPushButton("✘")
self.notice_close_btn.setObjectName("close_btn")
self.notice_close_btn.setFixedSize(30, 30)
self.notice_close_btn.setStyleSheet(
"""#close_btn{
background:red;
border:none;
margin-bottom: 20px;
margin-left: 20px;
}
#close_btn:hover{
background:white;
}
"""
)
self.notice_btn.setStyleSheet("""
#Analysis:hover#close_btn {
background:red;
}
""")
self.close_layout = QGridLayout()
self.close_layout.addWidget(self.notice_close_btn)
self.close_layout.setAlignment(Qt.AlignAbsolute|Qt.AlignRight)
self.close_layout.setSpacing(0)
self.notice_btn.setLayout(self.close_layout)
# self.notice_btn.set(self.notice_close_btn)
# ************************************************##
# 创建中侧部件
self.mid_widget = QWidget()
self.mid_widget.setObjectName('mid_widget')
# self.mid_widget.setFixedHeight(400)
self.mid_layout = QGridLayout()
self.mid_widget.setLayout(self.mid_layout)
# 创建下侧部件
self.down_widget = QWidget()
self.down_widget.setObjectName('down_widget')
self.down_widget.setFixedHeight(60)
self.down_layout = QGridLayout()
self.down_widget.setLayout(self.down_layout)
self.main_layout.addWidget(self.up_widget, 0, 0, 1, self.column)
self.main_layout.addWidget(self.slide, 1, 0, 1, self.column)
self.main_layout.addWidget(self.notice_btn, 2, 0, 1, 1)
self.main_layout.addWidget(self.mid_widget, 3, 1, 1, self.column - 1)
self.main_layout.addWidget(self.down_widget, 4, self.column - 1, 1, 1)
self.setCentralWidget(self.main_widget)
# 生成analysis
analysis = Analysis(self, name)
self.container[name] = {"noticeName": name, "modelObj": analysis, "index": self.obj_num}
self.container["selectedModelName"] = name
# 创建头部（head）
self.head_widget = QWidget()
self.head_layout = QGridLayout()
self.head_widget.setLayout(self.head_layout)
self.load_model_ico(0, self.num) if self.num < self.column else self.load_model_ico(0, self.column)
self.up_layout.addWidget(self.head_widget, 0, 0, 0, 0)
# 创建中部
# 创建尾部 按钮
self.tail_btn = QPushButton("Execute")
self.tail_btn.setObjectName('execute')
self.tail_btn.setFixedSize(const.TAIL_HEIGHT, const.BTN_HEIGHT)
self.tail_btn.clicked.connect(self.run)
self.down_layout.addWidget(self.tail_btn, 0, 0, 0, 0)
self.down_layout.setAlignment(Qt.AlignRight)
# 展示进度条
self.progressBar = QProgressBar()
self.progressBar.setFixedHeight(10)
self.progressBar.setMaximum(const.HUNDRED)
label = QLabel()
label.setText("运行进度：")
self.statusBar().addPermanentWidget(label)
self.statusBar().addPermanentWidget(self.progressBar)
self.main_layout.setSpacing(const.ZERO)
set_style(self)
# self.setWindowFlag(Qt.FramelessWindowHint) # 隐藏边框
# self.setWindowOpacity(0.9) # 设置窗口透明度
# self.setAttribute(Qt.WA_TranslucentBackground) # 设置窗口背景透明
def run(self):
mode_name = self.container.get("selectedModelName", "")
mode_info = self.container.get(mode_name, {})
mode = mode_info.get("modelObj")
if not mode:
print("error %s" % mode)
return
try:
mode.run()
except:
self.set_progress()
traceback.print_exc()
def set_progress(self):
self.tail_btn.setEnabled(True)
def load_model_ico(self, start, end):
for i in range(start, end):
self.tool_btn = QToolButton()
self.tool_btn.setText(f"{self.icon[i]}") # 设置按钮文本
self.tool_btn.setObjectName(f"{self.icon[i]}") # 设置按钮文本
self.tool_btn.setFixedSize(85, 52)
# self.tool_btn.setFixedHeight(85)
self.tool_btn.setIcon(qtawesome.icon(f'fa.{self.icon[i]}', color='white')) # 设置按钮图标
self.tool_btn.setIconSize(QSize(const.ICO_SIZE, const.ICO_SIZE)) # 设置图标大小
self.tool_btn.setToolButtonStyle(Qt.ToolButtonTextUnderIcon)
self.tool_btn.clicked.connect(self._clicked)
self.head_layout.addWidget(self.tool_btn, 0, i, 1, 1)
def info_btn(self, name):
notice_btn = QPushButton(name)
notice_btn.setObjectName(f"{name}_{self.obj_num}")
notice_btn.setFixedSize(const.NOTICE_WIDTH, const.BTN_HEIGHT)
self.container[name]["noticeName"] = f"{name}_{self.obj_num}"
notice_btn.clicked.connect(self._clicked)
self.main_layout.addWidget(notice_btn, 2, self.obj_num, 1, 1)
def _clicked(self):
model_name = self.sender().text()
if model_name == self.container.get("selectedModelName"):
self.show_widget(model_name)
return # message.info()
self.statusBar().showMessage(f'Current Value is {model_name}')
model_info = self.container.get(model_name, {})
if not model_info:
mode = None
try:
mode = ObjFactory.create(self, model_name)
except:
traceback.print_exc()
# return # message.info()
if mode is None:
set_mid_widget_background_style(self.mid_widget, False)
self.show_widget(model_name)
print("failed")
return # message.info()
model_info["modelObj"] = mode
self.obj_num += 1
model_info["index"] = self.obj_num
self.container[model_name] = model_info
self.info_btn(model_name)
self.show_widget(model_name)
self.show()
@staticmethod
def del_layout(layout):
for i in range(layout.count()):
layout.itemAt(i).widget().deleteLater()
def show_widget(self, curr_mode_name):
previous_name = self.container.get("selectedModelName")
if previous_name is not None:
index = self.container.get(previous_name).get("index")
if index is not None:
self.mid_layout.itemAt(index).widget().close()
index = self.container.get(curr_mode_name, {}).get("index")
if index is None:
set_ico_style(self.head_widget, curr_mode_name)
self.show_notice_widget(curr_mode_name)
return
self.mid_layout.itemAt(index).widget().show()
self.show_notice_widget(curr_mode_name)
set_mid_widget_background_style(self.mid_widget)
set_ico_style(self.head_widget, curr_mode_name)
self.container["selectedModelName"] = curr_mode_name
print(f"当前Container：{self.container}")
print(f"当前选中模块：{curr_mode_name}")
def show_notice_widget(self, model_name):
curr_notice_name = self.container.get(model_name, {}).get("noticeName", "failed-model")
args = []
for key, item in self.container.items():
if not isinstance(item, dict):
continue
elif key == model_name:
continue
else:
args.append(item.get("noticeName"))
args.append(curr_notice_name)
set_notice_style(self.main_widget, tuple(args))
self.container["selectedNoticeName"] = curr_notice_name
def _value_change(self):
value = self.slide.value()
start = value - self.num
self.statusBar().showMessage(f'Current Value is {value}')
self.del_layout(self.head_layout)
self.load_model_ico(start, value)
def paintEvent(self, event):
# 设置背景颜色
painter = QPainter(self)
background_color = QColor()
background_color.setNamedColor('#006CAB')
painter.setBrush(background_color)
painter.drawRect(self.rect())
def _center(self):
# 窗口居中
screen = QDesktopWidget().screenGeometry()
size = self.geometry()
self.move((screen.width() - size.width()) / 2,
(screen.height() - size.height()) / 2)
def setup_options_view(self):
self.options.setOrientation(Qt.Horizontal)
filter_functions_layout = QVBoxLayout()
filter_functions_layout.addWidget(QLabel("Filter functions"))
bar = QScrollBar()
bar.setFixedHeight(0)
bar.setFixedWidth(0)
self.filter_functions.setHorizontalScrollBar(bar)
self.filter_functions.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.filter_functions.setPlainText(self.ftrace.get_current_filters())
self.filter_functions.setPlaceholderText("*SyS_open*")
filter_functions_layout.addWidget(self.filter_functions)
filter_events_layout = QVBoxLayout()
filter_events_layout.addWidget(QLabel("Filter events"))
bar = QScrollBar()
bar.setFixedHeight(0)
bar.setFixedWidth(0)
self.filter_events.setHorizontalScrollBar(bar)
self.filter_events.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.filter_events.setPlainText(self.ftrace.get_current_events())
self.filter_events.setPlaceholderText(
'raw_syscalls:sys_exit\nraw_syscalls:sys_enter')
filter_events_layout.addWidget(self.filter_events)
filter_functions_widget = QWidget()
filter_functions_widget.setLayout(filter_functions_layout)
filter_events_widget = QWidget()
filter_events_widget.setLayout(filter_events_layout)
options_list_layout = QVBoxLayout()
options_list_layout.addWidget(QLabel("Options"))
for option in self.ftrace.get_options():
if len(option) < 1:
continue
q = NotEditableListWidgetItem()
q.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsEnabled)
if not option.startswith('no'):
q.setCheckState(Qt.Checked)
q.setText(option)
else:
q.setCheckState(Qt.Unchecked)
q.setText(option[2:])
self.options_list.addItem(q)
options_list_layout.addWidget(self.options_list)
options_list_widget = QWidget()
options_list_widget.setLayout(options_list_layout)
self.options.addWidget(filter_functions_widget)
self.options.addWidget(filter_events_widget)
self.options.addWidget(options_list_widget)
