From 94c364dd4d1e742e5c1aacef99a33ced13554048 Mon Sep 17 00:00:00 2001
From: EvilOS <gh0stkey>
Date: Mon, 30 Mar 2020 11:39:06 +0800
Subject: [PATCH] Fixed bug && Update

---
 .DS_Store  | Bin 0 -> 8196 bytes
 HaE/HaE.py |  24 ++++++++++++++++--------
 2 files changed, 16 insertions(+), 8 deletions(-)
 create mode 100644 .DS_Store

diff --git a/.DS_Store b/.DS_Store
new file mode 100644
index 0000000000000000000000000000000000000000..eb46c114a585ff2a059d02ea5233207ad69c89bf
GIT binary patch
literal 8196
zcmeHM&2JM&6n~QsSUU~bgcw50VO0x(L=G_ptR5;hPC{d~Qle}W93^0O?TNGMddKXp
z6GD>8syO#rksjN^AAq>^Qrb%|y|%X=`VaJyGrZXue<tCC5Nc<nncwWZ_h#n(X5Ox6
z3;-loFJ}NI06@dcQ6IwMn8f8gA4z?(Wf+m5J%lyKV-7z3?$nNGs|Lk@Vn8vV7*Gr-
z2Hpn-@SDw(cEWdGNL8sAPz?Mp8Q|{+12accVnbm3>cApC0w4}xbDXfQ^9rIN(UjN_
zh!RXhm?8>OB99nEm}9#k^-PHkfx;Y!JbVy2GLa_~5|57k8?ibNQ=lpp1B!v84Dhpi
zKg@#-bIx|>`Q3BFXlm+D($_yQc<QvKCAH*m@~*YXwynsH>S4|f*7(z+9c?<YzRZ}{
zs@u+rOSg;q_zd$S+x2M>8@pwSD(iP#zs$CCtnQb?xF1UhG%cYeiu%OvZYG<VG_GCy
zbbr#=&1R=2jq9J>+}}@V=PzYG%ipi=Jlfsc|N4pWh{(4BjP!VT(00!)ZOQ8i^4zHj
z_x{%Prh|#(&<De3M)c9Kv!mmqsZ{FR`P9VNhaVO7vyNT*suJ+2&M=P!Ij>HO`bdb*
z=S$3A4Cod*T8QYw$Z{W{XspLziI(rRo0fNKs59Pfh&%l}pwwTh1a5g@V<V(dfsarM
znCC6B(2ZQ?uXadR*W=M6TFv_#3`gWMSg3KeaD@gT;t~78svQKjU!l33yPoY(Z^6&8
zC}P!ui(c@V#{*b3yX5+n^sRpdm5H=@MbvaiH`&#6vh3Scio!+FmNvhTLJh9MG+3|>
z51;|x!4L2Pet|#WA2@_pWSE>IACs$OlH4F!Vv=bxL*_}IED7tr=0aX<4keJRA<BK~
zRWeqIRcjp@Qc`8Jv+@Q?8_dr=PNrT?x9zH|{03=l^Q2z4tibn}U%{UE>`s9TRg9Mk
zPzXC>=CHY`{vtAFa7!hI&I~2Gqo$}|lzaV-TaGrDsXyn@E$S}>U)m8BLG=~_sWUjW
zv{^K9>Hy8)x}Lqk11j4aT-t@px0mkKzxnpy`yYRL`t0Seu@$u9Yg9b9%HOcoMaVIW
z(<Sd*qW+)9_T&=v=(HE=B%NcHtE-QmMQcr1RTKk?0l`4OfC+y7pF8>fU(ux)Pz?Ns
z86X3*)@%+PZ2tJ_^_8Es4b1m3^ThQU0%H>_x-J6nx%G!3)(uqUHm1aeKy1OJE&^g2
LRG}DnrwqIXX2eaJ

literal 0
HcmV?d00001

diff --git a/HaE/HaE.py b/HaE/HaE.py
index f276723..be818f9 100644
--- a/HaE/HaE.py
+++ b/HaE/HaE.py
@@ -111,14 +111,14 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
         if colorText in colors:
             with open(configFile, 'r+') as content:
                 dicts = json.load(content)
-                # 解决r+写入问题
-                content.seek(0,0)
-                content.truncate()
                 if nameText in dicts:
                     self.tipString.setText("Name is existed!")
                 elif not(isHighlight or isExtract):
                     self.tipString.setText("Highlight or Extract?")
                 else:
+                    # 解决r+写入问题
+                    content.seek(0,0)
+                    content.truncate()
                     dicts[nameText] = {"regex": regexText, "highlight": isHighlight, "extract": isExtract, "color": colorText}
                     content.write(jsbeautifier.beautify(json.dumps(dicts)))
                     #print(dicts)
@@ -127,9 +127,18 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
             self.tipString.setText("Not in colors list.")
 
     def reloadConfig(self, event):
+        # 重新载入配置文件
         with open(configFile, 'r') as content:
             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):
         self.HaEPanel = JPanel()
         self.HaEPanel.setBorder(None)
@@ -172,11 +181,8 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
         self.configPanel = JPanel()
         self.tabbedPane.addTab("Config", None, self.configPanel, None)
         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.setEnabled(False)
+        # self.configTextArea.setEnabled(False)
         self.configTextArea.setTabSize(4)
         self.configTextArea.setLineWrap(True)
         self.configTextArea.setRows(20)
@@ -184,7 +190,9 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
         self.scrollPane = JScrollPane(self.configTextArea)
         self.configPanel.add(self.scrollPane, BorderLayout.SOUTH)
         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
         
 class MarkINFOTab(IMessageEditorTab):