diff --git a/README.md b/README.md index 373130f..6781693 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,7 @@ 一个高级web目录扫描工具，功能将会强于DirBuster、Dirsearch、cansina、御剑 -![dirmap](doc/dirmap.png) +![dirmap](dirmap.png) # 需求分析 diff --git a/dirmap.ico b/dirmap.ico new file mode 100644 index 0000000..7d7a38e Binary files /dev/null and b/dirmap.ico differ diff --git a/dirmap.png b/dirmap.png new file mode 100644 index 0000000..7a1c6e6 Binary files /dev/null and b/dirmap.png differ diff --git a/dirmap_qtgui.py b/dirmap_qtgui.py new file mode 100644 index 0000000..4d8bd59 --- /dev/null +++ b/dirmap_qtgui.py @@ -0,0 +1,149 @@ +import sys +import os + +from gevent import monkey +monkey.patch_all() +from PyQt5.QtWidgets import QMainWindow,QGridLayout,QLabel,QLineEdit,QPushButton,QTableWidget,QWidget,QDesktopWidget,QApplication,QTableWidgetItem +from PyQt5.QtGui import QIcon,QColor +from PyQt5.QtCore import QThread,Qt,pyqtSignal +from gevent import monkey +monkey.patch_all() +from lib.controller.engine import run +from lib.core.common import setPaths +from lib.core.data import cmdLineOptions, conf, paths, tasks +from lib.core.option import initOptions +from QCandyUi import CandyWindow + +class WorkThread(QThread): + stop = pyqtSignal() + update = pyqtSignal(str,str,str,str) + update_status = pyqtSignal() + + def __int__(self): + super(WorkThread, self).__init__() + + def run(self): + run() + self.stop.emit() # 循环完毕后发出信号 + +class DirmapGUI(QMainWindow): + def __init__(self): + super().__init__() + self.is_stop = True + self.initUI() + + def initUI(self): + self.resize(1000, 600) + self.center() + self.setWindowTitle('Dirmap - PyQT') + self.setWindowIcon(QIcon('dirmap.ico')) + + grid = QGridLayout() + urlLable = QLabel('URL') + urlLable.setAlignment((Qt.AlignRight | Qt.AlignVCenter)) + self.urlText = QLineEdit() + startButton = QPushButton("开始") + stopButton = QPushButton("结束") + startButton.clicked.connect(self.start_button_clicked) + stopButton.clicked.connect(self.stop_button_clicked) + self.status = self.statusBar() + self.status.showMessage("准备就绪,相关参数请自行修改配置文件,无需重启") + + grid.addWidget(urlLable, 0, 0) + grid.addWidget(self.urlText, 0, 1, 1, 12) + grid.addWidget(startButton, 0, 13) + grid.addWidget(stopButton, 0, 14) + + self.tableWidget = QTableWidget() # 创建一个表格 + # self.tableWidget.setRowCount(1) + self.tableWidget.setColumnCount(4) + self.tableWidget.setHorizontalHeaderLabels(['地址', '类型', '大小', '响应码']) + + self.tableWidget.setColumnWidth(0, 570) + self.tableWidget.setColumnWidth(1, 200) + self.tableWidget.setEditTriggers(QTableWidget.NoEditTriggers) + self.tableWidget.horizontalHeader().setFixedHeight(30) + grid.addWidget(self.tableWidget, 2, 0, 1, 15) # 把表格加入布局 + + wid_get = QWidget() + wid_get.setLayout(grid) + self.setCentralWidget(wid_get) + + # 控制窗口显示在屏幕中心的方法 + def center(self): + # 获得窗口 + qr = self.frameGeometry() + # 获得屏幕中心点 + cp = QDesktopWidget().availableGeometry().center() + # 显示到屏幕中心 + qr.moveCenter(cp) + self.move(qr.topLeft()) + + def start_button_clicked(self): + if not conf.is_stop: + return + text = self.urlText.text() + if text == "": return + conf.is_stop = False + row_index = self.tableWidget.rowCount() + for i in range(row_index): + self.tableWidget.removeRow(0) + paths.ROOT_PATH = os.getcwd() + setPaths() + dicts = {'thread_num': 8, 'target_input': text, 'target_file': '', 'load_config_file': True, + 'debug': False} + cmdLineOptions.update(dicts) + initOptions(cmdLineOptions) + self.workThread = WorkThread() + conf["thread"] = self.workThread + self.workThread.start() + self.workThread.stop.connect(self.stop_button_clicked) + self.workThread.update.connect(self.update_table) + self.workThread.update_status.connect(self.update_status_bar) + + def stop_button_clicked(self): + conf.is_stop = True + self.status.showMessage("扫描结束") + tasks['task_count'] = 0 + while not tasks.all_task.empty(): + tasks.all_task.get(block=True, timeout=3) + + def update_table(self, url, types, size, code): + row_index = self.tableWidget.rowCount() + self.tableWidget.setRowCount(row_index + 1) + url_item = QLabel() + url_item.setOpenExternalLinks(True) + url_item.setText(f"{url}") + self.tableWidget.setCellWidget(row_index, 0, url_item) + new_item = QTableWidgetItem(types) + new_item.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter) + self.tableWidget.setItem(row_index, 1, new_item) + new_item = QTableWidgetItem(size) + new_item.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter) + self.tableWidget.setItem(row_index, 2, new_item) + new_item = QTableWidgetItem(code) + new_item.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter) + if code[0] == '2': + new_item.setForeground(QColor("green")) + elif code[0] == '3': + new_item.setForeground(QColor("blue")) + elif code[0] == '4': + new_item.setForeground(QColor("red")) + self.tableWidget.setItem(row_index, 3, new_item) + + def update_status_bar(self): + if tasks['task_length']: + message = f"{tasks['task_count']}/{tasks['task_length']}" + else: + message = str(tasks['task_count']) + self.status.showMessage(message) + + +if __name__ == '__main__': + app = QApplication(sys.argv) + conf["is_stop"] = True + conf["ex"] = DirmapGUI() + conf["ex"] = CandyWindow.createWindow(conf["ex"], 'blueGreen', title="DirMap - PyQT", ico_path="dirmap.ico") + conf["ex"].show() + app.setWindowIcon(QIcon('dirmap.ico')) + sys.exit(app.exec_()) \ No newline at end of file diff --git a/lib/controller/bruter.py b/lib/controller/bruter.py index 6c227f9..5c512fb 100644 --- a/lib/controller/bruter.py +++ b/lib/controller/bruter.py @@ -453,6 +453,7 @@ def responseHandler(response): msg += '[{}] '.format(str(size)) msg += response.url outputscreen.info('\r'+msg+' '*(th.console_width-len(msg)+1)) + conf['thread'].update.emit(response.url,response.headers.get('content-type'),str(size),str(response.status_code)) #已去重复，结果保存。NOTE:此处使用response.url进行文件名构造，解决使用-iL参数时，不能按照域名来命名文件名的问题 #使用replace()，替换`:`，修复window下不能创建有`:`的文件问题 saveResults(urllib.parse.urlparse(response.url).netloc.replace(':','_'),msg) @@ -513,6 +514,7 @@ def worker(): #更新进度条 tasks.task_count += 1 bar.log.update(tasks.task_count) + conf['thread'].update_status.emit() def boss(): ''' @@ -520,7 +522,7 @@ def boss(): @param {type} @return: ''' - while not tasks.all_task.empty(): + while not tasks.all_task.empty() and not conf.is_stop: worker() def bruter(url): diff --git a/requirement.txt b/requirement.txt index 3990adc..0889167 100644 --- a/requirement.txt +++ b/requirement.txt @@ -1,4 +1,6 @@ gevent requests progressbar2 -lxml \ No newline at end of file +lxml +QCandyUi +PyQt5 diff --git a/start_dirmap_no_console.vbs b/start_dirmap_no_console.vbs new file mode 100644 index 0000000..c0ac39c --- /dev/null +++ b/start_dirmap_no_console.vbs @@ -0,0 +1,2 @@ +Set ws = CreateObject("Wscript.Shell") +ws.run "cmd /c python dirmap_qtgui.py" ,vbhide \ No newline at end of file diff --git a/start_dirmap_with_console.vbs b/start_dirmap_with_console.vbs new file mode 100644 index 0000000..4c0ddff --- /dev/null +++ b/start_dirmap_with_console.vbs @@ -0,0 +1,2 @@ +Set ws = CreateObject("Wscript.Shell") +ws.run "cmd /k python dirmap_qtgui.py" \ No newline at end of file