| 8 | import pandas as pd |
| 9 | |
| 10 | class MyForm(QDialog): |
| 11 | def __init__(self): |
| 12 | super().__init__() |
| 13 | self.ui = Ui_Dialog() |
| 14 | self.ui.setupUi(self) |
| 15 | self.ui.pushButton.clicked.connect(self.select_file) |
| 16 | self.ui.lineEdit.setAcceptDrops(True) # 设置lineEdit接受拖拽 |
| 17 | self.ui.lineEdit.dragEnterEvent = self.dragEnterEvent # 设置拖拽进入事件 |
| 18 | self.ui.lineEdit.dropEvent = self.dropEvent # 设置拖拽释放事件 |
| 19 | self.profile_to_comboBox() # 添加这一行以自动填充comboBox |
| 20 | self.setWindowTitle("VolatilityPro GUI By: Tokeii") |
| 21 | self.setWindowIcon(QIcon("res/1.png")) |
| 22 | self.setFixedSize(self.width(), self.height()) |
| 23 | #鼠标点击事件 |
| 24 | self.ui.pushButton_2.clicked.connect(self.run) |
| 25 | self.ui.pushButton_3.clicked.connect(self.run2) |
| 26 | self.ui.listWidget.itemDoubleClicked.connect(self.add_to_lineEdit_2) # 添加这一行以响应双击事件 |
| 27 | self.ui.reloadingButton.clicked.connect(self.reloadingButton) |
| 28 | self.load_filescan_combobox() |
| 29 | self.ui.pushButton_find.clicked.connect(lambda:self.find_regex_in_large_file(self.ui.lineEdit.text(),self.ui.lineEdit_findstr.text())) |
| 30 | #dumpfilename |
| 31 | |
| 32 | #选择后直接执行load_filescan |
| 33 | self.ui.comboBox_2.currentIndexChanged.connect(self.load_filescan) |
| 34 | self.ui.pushButton_4.clicked.connect(lambda:self.load_csv_to_table('pslist')) |
| 35 | self.ui.pushButton_5.clicked.connect(lambda:self.load_clipboard_to_table()) |
| 36 | self.ui.pushButton_6.clicked.connect(lambda:self.load_csv_to_table('atomscan')) |
| 37 | self.ui.pushButton_7.clicked.connect(lambda:self.load_csv_to_table('drivermodule')) |
| 38 | self.ui.pushButton_8.clicked.connect(lambda:self.load_csv_to_table('driverscan')) |
| 39 | self.ui.pushButton_9.clicked.connect(lambda:self.load_csv_to_table('envars')) |
| 40 | self.ui.pushButton_10.clicked.connect(lambda:self.load_csv_to_table('gditimers')) |
| 41 | self.ui.pushButton_11.clicked.connect(lambda:self.load_csv_to_table('hivelist')) |
| 42 | self.ui.pushButton_12.clicked.connect(lambda:self.load_csv_to_table('joblinks')) |
| 43 | self.ui.pushButton_13.clicked.connect(lambda:self.load_csv_to_table('ldrmodules')) |
| 44 | self.ui.pushButton_14.clicked.connect(lambda:self.load_csv_to_table('modscan')) |
| 45 | self.ui.pushButton_15.clicked.connect(lambda:self.load_csv_to_table('modules')) |
| 46 | self.ui.pushButton_16.clicked.connect(lambda:self.load_csv_to_table('netscan')) |
| 47 | self.ui.pushButton_17.clicked.connect(lambda:self.load_csv_to_table('objtypescan')) |
| 48 | self.ui.pushButton_18.clicked.connect(lambda:self.load_csv_to_table('psscan')) |
| 49 | self.ui.pushButton_19.clicked.connect(lambda:self.load_csv_to_table('psxview')) |
| 50 | self.ui.pushButton_20.clicked.connect(lambda:self.load_csv_to_table('shimcache')) |
| 51 | self.ui.pushButton_21.clicked.connect(lambda:self.load_csv_to_table('callbacks')) |
| 52 | self.ui.pushButton_22.clicked.connect(self.load_cmdline_to_table) |
| 53 | self.show() |
| 54 | |
| 55 | |
| 56 | def select_file(self): |
| 57 | #加一个*.*格式 |
| 58 | file_path, _ = QFileDialog.getOpenFileName(self, "Select File", filter="Memory Dump Files (*.dmp *.iso *.bin *.img *.vmem *.hprof *.core *.elf *.raw)") |
| 59 | self.ui.lineEdit.setText(file_path.split('/')[-1]) |
| 60 | #label 设置 |
| 61 | self.setWindowTitle('/'.join(file_path.split('/')[0:-1])) |
| 62 | |
| 63 | def profile_to_comboBox(self): |
| 64 | with open("profilelist.cfg", "r") as f: |
| 65 | profile_list = f.readlines() |
| 66 | for profile in profile_list: |
| 67 | self.ui.comboBox.addItem(profile.strip()) |