From 00e02eec13f37a052052e887b75b28504783a7b3 Mon Sep 17 00:00:00 2001
From: Regala <miguel@regala.im>
Date: Thu, 17 Dec 2020 09:05:39 +0000
Subject: [PATCH] Tentative guess all option

---
 src/burp/OfferParamGuess.java | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/burp/OfferParamGuess.java b/src/burp/OfferParamGuess.java
index 772170f..fabcd31 100644
--- a/src/burp/OfferParamGuess.java
+++ b/src/burp/OfferParamGuess.java
@@ -26,16 +26,22 @@ public List<JMenuItem> createMenuItems(IContextMenuInvocation invocation) {
             return options;
         }
 
+        JMenuItem allButton = new JMenuItem("Guess everything!");
+        allButton.addActionListener(new TriggerParamGuesser(reqs, false, IParameter.PARAM_URL, paramGrabber, taskEngine));
+        
         JMenuItem probeButton = new JMenuItem("Guess GET parameters");
         probeButton.addActionListener(new TriggerParamGuesser(reqs, false, IParameter.PARAM_URL, paramGrabber, taskEngine));
+        allButton.addActionListener(new TriggerParamGuesser(reqs, false, IParameter.PARAM_URL, paramGrabber, taskEngine));
         options.add(probeButton);
 
         JMenuItem cookieProbeButton = new JMenuItem("Guess cookie parameters");
         cookieProbeButton.addActionListener(new TriggerParamGuesser(reqs, false, IParameter.PARAM_COOKIE, paramGrabber, taskEngine));
+        allButton.addActionListener(new TriggerParamGuesser(reqs, false, IParameter.PARAM_COOKIE, paramGrabber, taskEngine));
         options.add(cookieProbeButton);
 
         JMenuItem headerProbeButton = new JMenuItem("Guess headers");
         headerProbeButton.addActionListener(new TriggerParamGuesser(reqs, false, Utilities.PARAM_HEADER, paramGrabber, taskEngine));
+        allButton.addActionListener(new TriggerParamGuesser(reqs, false, Utilities.PARAM_HEADER, paramGrabber, taskEngine));
         options.add(headerProbeButton);
 
 //        if (invocation.getSelectionBounds() != null && reqs.length == 1) {
@@ -51,6 +57,7 @@ public List<JMenuItem> createMenuItems(IContextMenuInvocation invocation) {
             if (Utilities.countMatches(resp, Utilities.helpers.stringToBytes("%253c%2561%2560%2527%2522%2524%257b%257b%255c")) > 0) {
                 JMenuItem backendProbeButton = new JMenuItem("*Identify backend parameters*");
                 backendProbeButton.addActionListener(new TriggerParamGuesser(reqs, true, IParameter.PARAM_URL, paramGrabber, taskEngine));
+                allButton.addActionListener(new TriggerParamGuesser(reqs, true, IParameter.PARAM_URL, paramGrabber, taskEngine));
                 options.add(backendProbeButton);
             }
 
@@ -93,11 +100,13 @@ public List<JMenuItem> createMenuItems(IContextMenuInvocation invocation) {
 
                     JMenuItem postProbeButton = new JMenuItem("Guess " + humanType + " parameter");
                     postProbeButton.addActionListener(new TriggerParamGuesser(reqs, false, type, paramGrabber, taskEngine));
+                    allButton.addActionListener(new TriggerParamGuesser(reqs, false, type, paramGrabber, taskEngine));
                     options.add(postProbeButton);
                 }
             }
         }
 
+        options.add(allButton);
         return options;
     }
-}
\ No newline at end of file
+}