168 lines
9.0 KiB
Python
168 lines
9.0 KiB
Python
import images_ico
|
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
|
|
|
class Ui_MainWindow(object):
|
|
def setupUi(self, MainWindow):
|
|
setini = """:::::::::导出的人员信息:::::::::
|
|
检查单位=宝宝
|
|
检查人员=宝宝
|
|
|
|
:::::::::自定义导出数据 - 可删除、添加、调换位置:::::::::
|
|
1:6:检查单位|2:28:系统名称|3:8:主机名|4:16:IP地址|5:8:端口|6:8:协议|7:8:服务|8:45:漏洞名称|9:8:风险分类|10:8:风险等级|11:45:整改建议|12:45:漏洞描述|13:13:漏洞CVE编号|14:11:扫描起始时间|15:11:扫描结束时间|16:11:漏洞发现月份
|
|
|
|
:::::::::自定义标题颜色:::::::::
|
|
A-S:ff0000
|
|
|
|
:::::::::自定义漏洞等级 - 可删除，添加:::::::::
|
|
中|检测到目标NTP服务支持monlist命令
|
|
中|目标主机rpcinfo -p信息泄露
|
|
中|可通过finger服务猜测用户名列表
|
|
中|可通过finger服务获取用户信息
|
|
中|可通过rusers服务获取用户信息
|
|
高|目标主机showmount -e信息泄露"""
|
|
try:
|
|
with open('set.ini') as info_ini:
|
|
ini = info_ini.readlines()[1:3]
|
|
self.name_ini = ini[1].split('=')[1].strip()
|
|
self.company_ini = ini[0].split('=')[1].strip()
|
|
except Exception as e:
|
|
with open('set.ini','a',encoding='gb18030') as set_ini:
|
|
set_ini.write(setini)
|
|
QtWidgets.QMessageBox.information(MainWindow, "提示！", "请修改同目录下的配置文件，重新打开软件！", QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No , QtWidgets.QMessageBox.Yes )
|
|
exit()
|
|
|
|
title = 'RSAS漏洞数据导出工具1.8'
|
|
font = QtGui.QFont()
|
|
font.setFamily("宋体")
|
|
font.setPointSize(10)
|
|
MainWindow.setFont(font)
|
|
#定义程序的标题
|
|
MainWindow.setWindowTitle(title)
|
|
#设定程序的最大分辨率，禁止最大化、拖动窗口
|
|
MainWindow.setFixedSize(560, 310)
|
|
#设置图标
|
|
MainWindow.setWindowIcon(QtGui.QIcon(':/favicon.ico'))
|
|
#获取显示器的分辨率
|
|
screen = QtWidgets.QDesktopWidget().screenGeometry()
|
|
#获取程序的宽和高
|
|
size = MainWindow.geometry()
|
|
#实现在屏幕中间显示程序
|
|
MainWindow.move((screen.width() - size.width())/2, (screen.height() - size.height())/2)
|
|
|
|
#这是底部的状态栏
|
|
MainWindow.status = MainWindow.statusBar()
|
|
MainWindow.status.showMessage("检查单位：%s 检查人员：%s" % (self.company_ini,self.name_ini))
|
|
|
|
#这是一个框架，用来固定按钮用的
|
|
self.formLayoutWidget = QtWidgets.QWidget(MainWindow)
|
|
self.formLayoutWidget.setGeometry(QtCore.QRect(10, 10, 451, 54))
|
|
self.formLayoutWidget.setObjectName("formLayoutWidget")
|
|
self.formLayout = QtWidgets.QFormLayout(self.formLayoutWidget)
|
|
self.formLayout.setContentsMargins(0, 0, 0, 0)
|
|
self.formLayout.setObjectName("formLayout")
|
|
#文字：原始报告路径
|
|
self.input_label = QtWidgets.QLabel(self.formLayoutWidget)
|
|
self.input_label.setObjectName("input_label")
|
|
self.formLayout.setWidget(0, QtWidgets.QFormLayout.LabelRole, self.input_label)
|
|
#文字：输出报告路径
|
|
self.output_label = QtWidgets.QLabel(self.formLayoutWidget)
|
|
self.output_label.setObjectName("output_label")
|
|
self.formLayout.setWidget(2, QtWidgets.QFormLayout.LabelRole, self.output_label)
|
|
|
|
#原始报告路径后边的文本框
|
|
self.input_lineEdit = QtWidgets.QLineEdit(self.formLayoutWidget)
|
|
self.input_lineEdit.setObjectName("input_lineEdit")
|
|
self.formLayout.addWidget(self.input_lineEdit)
|
|
self.formLayout.setWidget(0, QtWidgets.QFormLayout.FieldRole, self.input_lineEdit)
|
|
|
|
#输出报告路径后边的文本框
|
|
self.output_lineEdit = QtWidgets.QLineEdit(self.formLayoutWidget)
|
|
self.output_lineEdit.setObjectName("output_lineEdit")
|
|
self.formLayout.addWidget(self.output_lineEdit)
|
|
self.formLayout.setWidget(2, QtWidgets.QFormLayout.FieldRole, self.output_lineEdit)
|
|
|
|
#框架的结束部分
|
|
spacerItem = QtWidgets.QSpacerItem(20, 40, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)
|
|
self.formLayout.setItem(1, QtWidgets.QFormLayout.LabelRole, spacerItem)
|
|
#这玩意就是打开路径按钮的框架
|
|
self.start_verticalLayoutWidget = QtWidgets.QWidget(MainWindow)
|
|
self.start_verticalLayoutWidget.setGeometry(QtCore.QRect(463, 2, 91, 71))
|
|
self.start_verticalLayoutWidget.setObjectName("start_verticalLayoutWidget")
|
|
self.start_verticalLayout = QtWidgets.QVBoxLayout(self.start_verticalLayoutWidget)
|
|
self.start_verticalLayout.setContentsMargins(0, 0, 0, 0)
|
|
self.start_verticalLayout.setObjectName("start_verticalLayout")
|
|
#这是原始报告路径后边的文本框后边的打开路径按钮
|
|
self.input_Button = QtWidgets.QPushButton(self.start_verticalLayoutWidget)
|
|
self.input_Button.setObjectName("input_Button")
|
|
self.start_verticalLayout.addWidget(self.input_Button)
|
|
|
|
#这是输出报告路径后边的文本框后边的打开路径按钮
|
|
self.output_Button = QtWidgets.QPushButton(self.start_verticalLayoutWidget)
|
|
self.output_Button.setObjectName("output_Button")
|
|
self.start_verticalLayout.addWidget(self.output_Button)
|
|
|
|
#这又是一个框架，固定用的
|
|
self.horizontalLayoutWidget = QtWidgets.QWidget(MainWindow)
|
|
self.horizontalLayoutWidget.setGeometry(QtCore.QRect(10, 75, 451, 21))
|
|
self.horizontalLayoutWidget.setObjectName("horizontalLayoutWidget")
|
|
self.horizontalLayout = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget)
|
|
self.horizontalLayout.setContentsMargins(0, 0, 0, 0)
|
|
self.horizontalLayout.setObjectName("horizontalLayout")
|
|
#文字：选择导出数据
|
|
self.data_label = QtWidgets.QLabel(self.horizontalLayoutWidget)
|
|
self.data_label.setObjectName("data_label")
|
|
self.horizontalLayout.addWidget(self.data_label)
|
|
#复选框：高危
|
|
self.hight_checkBox = QtWidgets.QCheckBox(self.horizontalLayoutWidget)
|
|
self.hight_checkBox.setObjectName("hight_checkBox")
|
|
self.horizontalLayout.addWidget(self.hight_checkBox)
|
|
#复选框：中危
|
|
self.middle_checkBox = QtWidgets.QCheckBox(self.horizontalLayoutWidget)
|
|
self.middle_checkBox.setObjectName("middle_checkBox")
|
|
self.horizontalLayout.addWidget(self.middle_checkBox)
|
|
#复选框：低危
|
|
self.low_checkBox = QtWidgets.QCheckBox(self.horizontalLayoutWidget)
|
|
self.low_checkBox.setObjectName("low_checkBox")
|
|
self.horizontalLayout.addWidget(self.low_checkBox)
|
|
#复选框：端口
|
|
self.port_checkBox = QtWidgets.QCheckBox(self.horizontalLayoutWidget)
|
|
self.port_checkBox.setObjectName("port_checkBox")
|
|
self.horizontalLayout.addWidget(self.port_checkBox)
|
|
#复选框：网站
|
|
self.web_checkBox = QtWidgets.QCheckBox(self.horizontalLayoutWidget)
|
|
self.web_checkBox.setObjectName("web_checkBox")
|
|
self.horizontalLayout.addWidget(self.web_checkBox)
|
|
#这又是一个框架
|
|
self.end_verticalLayoutWidget = QtWidgets.QWidget(MainWindow)
|
|
self.end_verticalLayoutWidget.setGeometry(QtCore.QRect(463, 69, 91, 31))
|
|
self.end_verticalLayoutWidget.setObjectName("end_verticalLayoutWidget")
|
|
self.end_verticalLayout = QtWidgets.QVBoxLayout(self.end_verticalLayoutWidget)
|
|
self.end_verticalLayout.setContentsMargins(0, 0, 0, 0)
|
|
self.end_verticalLayout.setObjectName("end_verticalLayout")
|
|
#按钮：开始导出
|
|
self.start_Button = QtWidgets.QPushButton(self.end_verticalLayoutWidget)
|
|
self.start_Button.setObjectName("start_Button")
|
|
#框架结尾
|
|
self.end_verticalLayout.addWidget(self.start_Button)
|
|
#详细输出日志的文本框
|
|
self.log_textEdit = QtWidgets.QTextEdit(MainWindow)
|
|
self.log_textEdit.setGeometry(QtCore.QRect(5, 120, 550, 171))
|
|
self.log_textEdit.setObjectName("log_textEdit")
|
|
|
|
self.retranslateUi(MainWindow)
|
|
QtCore.QMetaObject.connectSlotsByName(MainWindow)
|
|
|
|
def retranslateUi(self, MainWindow):
|
|
_translate = QtCore.QCoreApplication.translate
|
|
self.input_label.setText(_translate("MainWindow", "原始报告路径："))
|
|
self.output_label.setText(_translate("MainWindow", "输出报告路径："))
|
|
self.input_Button.setText(_translate("MainWindow", "打开路径"))
|
|
self.output_Button.setText(_translate("MainWindow", "打开路径"))
|
|
self.data_label.setText(_translate("MainWindow", "选择导出数据："))
|
|
self.hight_checkBox.setText(_translate("MainWindow", "高危"))
|
|
self.middle_checkBox.setText(_translate("MainWindow", "中危"))
|
|
self.low_checkBox.setText(_translate("MainWindow", "低危"))
|
|
self.port_checkBox.setText(_translate("MainWindow", "端口"))
|
|
self.web_checkBox.setText(_translate("MainWindow", "网站"))
|
|
self.start_Button.setText(_translate("MainWindow", "开始导出"))
|