admin/HaE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed bug && Update

Browse Source
This commit is contained in:
EvilOS
2020-03-30 11:39:06 +08:00
parent a2950d63a5
commit 94c364dd4d
2 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
.DS_Store vendored Normal file
View File

Binary file not shown.

24
HaE/HaE.py
View File

@@ -111,14 +111,14 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
if colorText in colors: if colorText in colors:
with open(configFile, 'r+') as content: with open(configFile, 'r+') as content:
dicts = json.load(content) dicts = json.load(content)
# 解决r+写入问题
content.seek(0,0)
content.truncate()
if nameText in dicts: if nameText in dicts:
self.tipString.setText("Name is existed!") self.tipString.setText("Name is existed!")
elif not(isHighlight or isExtract): elif not(isHighlight or isExtract):
self.tipString.setText("Highlight or Extract?") self.tipString.setText("Highlight or Extract?")
else: else:
# 解决r+写入问题
content.seek(0,0)
content.truncate()
dicts[nameText] = {"regex": regexText, "highlight": isHighlight, "extract": isExtract, "color": colorText} dicts[nameText] = {"regex": regexText, "highlight": isHighlight, "extract": isExtract, "color": colorText}
content.write(jsbeautifier.beautify(json.dumps(dicts))) content.write(jsbeautifier.beautify(json.dumps(dicts)))
#print(dicts) #print(dicts)
@@ -127,9 +127,18 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
self.tipString.setText("Not in colors list.") self.tipString.setText("Not in colors list.")
def reloadConfig(self, event): def reloadConfig(self, event):
# 重新载入配置文件
with open(configFile, 'r') as content: with open(configFile, 'r') as content:
self.configTextArea.setText(content.read()) self.configTextArea.setText(content.read())
def saveConfig(self, event):
# 保存配置文件
text = self.configTextArea.getText()
if text != "":
with open(configFile, 'w') as content:
content.write(text)
self.reloadConfig()
def getUiComponent(self): def getUiComponent(self):
self.HaEPanel = JPanel() self.HaEPanel = JPanel()
self.HaEPanel.setBorder(None) self.HaEPanel.setBorder(None)
@@ -172,11 +181,8 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
self.configPanel = JPanel() self.configPanel = JPanel()
self.tabbedPane.addTab("Config", None, self.configPanel, None) self.tabbedPane.addTab("Config", None, self.configPanel, None)
self.configPanel.setLayout(BorderLayout(0, 0)) self.configPanel.setLayout(BorderLayout(0, 0))
self.configString = JLabel("This is config file content.")
self.configString.setHorizontalAlignment(SwingConstants.CENTER)
self.configPanel.add(self.configString, BorderLayout.NORTH)
self.configTextArea = JTextArea() self.configTextArea = JTextArea()
self.configTextArea.setEnabled(False) # self.configTextArea.setEnabled(False)
self.configTextArea.setTabSize(4) self.configTextArea.setTabSize(4)
self.configTextArea.setLineWrap(True) self.configTextArea.setLineWrap(True)
self.configTextArea.setRows(20) self.configTextArea.setRows(20)
@@ -184,7 +190,9 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
self.scrollPane = JScrollPane(self.configTextArea) self.scrollPane = JScrollPane(self.configTextArea)
self.configPanel.add(self.scrollPane, BorderLayout.SOUTH) self.configPanel.add(self.scrollPane, BorderLayout.SOUTH)
self.reloadButton = JButton("Reload", actionPerformed=self.reloadConfig) self.reloadButton = JButton("Reload", actionPerformed=self.reloadConfig)
self.configPanel.add(self.reloadButton, BorderLayout.CENTER) self.configPanel.add(self.reloadButton, BorderLayout.NORTH)
self.saveButton = JButton("Save", actionPerformed=self.saveConfig)
self.configPanel.add(self.saveButton, BorderLayout.CENTER)
return self.HaEPanel return self.HaEPanel
class MarkINFOTab(IMessageEditorTab): class MarkINFOTab(IMessageEditorTab):