Added Dynamic Path and binray script support #8

Anof-cyber · Jul 1, 2024 · 46b8b8c · 46b8b8c
1 parent cb0d04d
commit 46b8b8c
Show file tree

Hide file tree

Showing 5 changed files with 37 additions and 22 deletions.
diff --git a/pycript.py b/pycript.py
@@ -235,10 +235,21 @@ def registerExtenderCallbacks(self, callbacks):
         self.AdditionalSettinglabel.setFont(Font("Segoe UI", 1, 14))
 
         self.languagelabel = JLabel();
-        self.languagelabel.setText("Language");
+        self.languagelabel.setText("Language Binary(ie python.exe emptydirect executable for enc/dec)");
 
-        self.langdata = ("JavaScript", "Python", "Java Jar")
-        self.languagecombobox = JComboBox(self.langdata)
+
+        #self.langdata = ("JavaScript", "Python", "Java Jar")
+        self.languagejpanel =  JPanel() #JComboBox(self.langdata)
+        self.language_select_button = JButton("Select Language Binary Path")
+        self.language_select_button.addActionListener(self.select_language_file_path)
+
+
+
+        self.languagepath = JTextField(20)
+        self.languagepath.setText("C:/Program Files/nodejs/node.exe")
+        self.languagejpanel.add(self.languagepath,BorderLayout.NORTH)
+        self.languagejpanel.add(self.language_select_button,BorderLayout.NORTH)
+
 
 
         self.reqmethodlabel = JLabel();
@@ -675,7 +686,7 @@ def registerExtenderCallbacks(self, callbacks):
         self.additionallayerpane.setBorder(BorderFactory.createLineBorder(Color(0, 0, 0)));
         self.additionallayerpane.setLayer(self.AdditionalSettinglabel, JLayeredPane.DEFAULT_LAYER);
         self.additionallayerpane.setLayer(self.languagelabel, JLayeredPane.DEFAULT_LAYER);
-        self.additionallayerpane.setLayer(self.languagecombobox, JLayeredPane.DEFAULT_LAYER);
+        self.additionallayerpane.setLayer(self.languagejpanel, JLayeredPane.DEFAULT_LAYER);
         self.additionallayerpane.setLayer(self.reqmethodlabel, JLayeredPane.DEFAULT_LAYER);
         self.additionallayerpane.setLayer(self.reqmethodcombobox, JLayeredPane.DEFAULT_LAYER);
         self.additionallayerpane.setLayer(self.reqresponselabel, JLayeredPane.DEFAULT_LAYER);
@@ -695,7 +706,7 @@ def registerExtenderCallbacks(self, callbacks):
                             .addGroup(self.additionallayerpaneLayout.createSequentialGroup()
                                 .addComponent(self.languagelabel)
                                 .addGap(18, 18, 18)
-                                .addComponent(self.languagecombobox, GroupLayout.PREFERRED_SIZE, GroupLayout.DEFAULT_SIZE, GroupLayout.PREFERRED_SIZE)))
+                                .addComponent(self.languagejpanel, GroupLayout.PREFERRED_SIZE, GroupLayout.DEFAULT_SIZE, GroupLayout.PREFERRED_SIZE)))
                         .addPreferredGap(LayoutStyle.ComponentPlacement.RELATED)
                         .addComponent(self.reqmethodcombobox, GroupLayout.PREFERRED_SIZE, GroupLayout.DEFAULT_SIZE, GroupLayout.PREFERRED_SIZE))
                     .addGroup(self.additionallayerpaneLayout.createSequentialGroup()
@@ -712,7 +723,8 @@ def registerExtenderCallbacks(self, callbacks):
                 .addPreferredGap(LayoutStyle.ComponentPlacement.UNRELATED)
                 .addGroup(self.additionallayerpaneLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
                     .addComponent(self.languagelabel)
-                    .addComponent(self.languagecombobox, GroupLayout.PREFERRED_SIZE, GroupLayout.DEFAULT_SIZE, GroupLayout.PREFERRED_SIZE))
+                    .addComponent(self.languagejpanel, GroupLayout.PREFERRED_SIZE, GroupLayout.DEFAULT_SIZE, GroupLayout.PREFERRED_SIZE)
+                    .addGap(0, 40, 40))
                 .addPreferredGap(LayoutStyle.ComponentPlacement.RELATED)
                 .addGroup(self.additionallayerpaneLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
                     .addComponent(self.reqmethodlabel)
@@ -1095,8 +1107,15 @@ def importdecryptionjsfile(self,e):
             self.requestdecryptionpath.setText(self.decryptionfilepath)
             self.callbacks.saveExtensionSetting("requestdecryptionfilesave", self.decryptionfilepath)
 
-
-
+    # handle language binrary path 
+    def select_language_file_path(self,e):
+        chooseFile = JFileChooser()
+        ret = chooseFile.showDialog(self.tab, "Choose file")
+        if ret == JFileChooser.APPROVE_OPTION:
+            fileLoad = chooseFile.getSelectedFile()
+            self.languagefullpath = fileLoad.getAbsolutePath()
+            self.languagepath.setText(self.languagefullpath)
+
     # Returning the Extension Tab name to burp ITAB
     def getTabCaption(self):
         return "PyCript"
@@ -1256,7 +1275,7 @@ def encryptstring(self,invocation):
             output = StringCrypto(self,encpath,query,http_request_response)
             encryptedstring = output.encrypt_string_request()
         else:
-            encryptedstring = Parameterencrypt(self.languagecombobox.getSelectedItem(), encpath, query)
+            encryptedstring = Parameterencrypt(self.languagepath.getText(), encpath, query)
         #JOptionPane.showInputDialog(None, "Encrypted String", "Decryption", JOptionPane.PLAIN_MESSAGE, None, None, encryptedstring)
         #JOptionPane.showMessageDialog(None, encryptedstring, "String", JOptionPane.INFORMATION_MESSAGE)
         showEditableDialog(encryptedstring, "Encrypted String")
@@ -1301,7 +1320,7 @@ def decryptstring(self,invocation):
             output = StringCrypto(self,encpath,query,http_request_response)
             decryptedstring = output.decrypt_string_request()
         else:
-            decryptedstring = Parameterdecrypt(self.languagecombobox.getSelectedItem(), encpath, query)
+            decryptedstring = Parameterdecrypt(self.languagepath.getText(), encpath, query)
 
         showEditableDialog(decryptedstring, "Decryted String")
 

diff --git a/pycript/Reqcheck.py b/pycript/Reqcheck.py
@@ -9,7 +9,7 @@
 
 def EncryptRequest(extender, currentreq,req):
     encryptionpath = extender.encryptionfilepath
-    selectedlang = extender.languagecombobox.getSelectedItem()
+    selectedlang = extender.languagepath.getText()
     selected_method = extender.reqmethodcombobox.getSelectedItem()
     parameters = req.getParameters()
     header = req.getHeaders()  # Get Array/last format header from burp header api (used for Custom Request)
@@ -44,7 +44,7 @@ def EncryptRequest(extender, currentreq,req):
 ## Function to decrypt request when Burp Menu to decrypt request is triggered
 def DecryptRequest(extender, currentreq,req):
     decryptionpath = extender.decryptionfilepath
-    selectedlang = extender.languagecombobox.getSelectedItem()
+    selectedlang = extender.languagepath.getText()
     selected_method = extender.reqmethodcombobox.getSelectedItem()
     selected_request_inc_ex_ctype = extender.selected_request_inc_ex_ctype
     listofparam = extender.requestparamlist.getText().split(',')

diff --git a/pycript/execution.py b/pycript/execution.py
@@ -1,17 +1,13 @@
 import subprocess
 from .gui import logerrors
-
+import os 
 def execute_command(selectedlang, path, data, headervalue=None):
     try:
         command = []
-        if selectedlang == "JavaScript":
-            command.extend(["node", '"' + path + '"'])  # Surround path with double quotes
-        elif selectedlang == "Python":
-            command.extend(["python", '"' + path + '"'])  # Surround path with double quotes
-        elif selectedlang == "Java Jar":
-            command.extend(["java", "-jar", '"' + path + '"'])  # Surround path with double quotes
+        if selectedlang:
+            command.append('"' + selectedlang + '"')
 
-        command.extend(["-d", data])
+        command.extend(['"' + path + '"',"-d", data])
 
         if headervalue is not None:
             command.extend(["-h", headervalue])

diff --git a/pycript/response_handler.py b/pycript/response_handler.py
@@ -5,7 +5,7 @@
 
 
 def encrypt_decrypt_response(extender,currentresp,response,enc_dec,enc_dec_type):
-    selectedlang = extender.languagecombobox.getSelectedItem()
+    selectedlang = extender.languagepath.getText()
     if enc_dec_type== "Decrypt":
         enc_dec_file_path = extender.responsedecryptionfilepath
     else:

diff --git a/pycript/stringcrypto.py b/pycript/stringcrypto.py
@@ -6,7 +6,7 @@ def __init__(self, extender, encpath, query, http_request_response):
         self._extender = extender
         self._selectedmessage = query
         self.message = http_request_response
-        self.selectedlang = extender.languagecombobox.getSelectedItem()
+        self.selectedlang = extender.languagepath.getText()
         self.encpath = encpath
         self.header = self.get_headers()
         self.headers_str = self.get_headers_str()