Skip to content

Commit 2c26b1b

Browse files
author
thort
committed
Changing behaviour of reset message
Not sure why it was set to only reset in raw mode. Also, the status message should be cleared after reset.
1 parent c792f6a commit 2c26b1b

File tree

3 files changed

+21
-39
lines changed

3 files changed

+21
-39
lines changed

src/main/java/application/SamlTabController.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -337,9 +337,7 @@ public void removeSignature() {
337337
}
338338

339339
public void resetMessage() {
340-
if (isRawMode) {
341-
samlMessage = orgSAMLMessage;
342-
}
340+
samlMessage = orgSAMLMessage;
343341
textArea.setContents(ByteArray.byteArray(samlMessage));
344342
isEdited = false;
345343
}

src/main/java/gui/SamlPanelAction.java

Lines changed: 20 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -21,26 +21,26 @@ public class SamlPanelAction extends JPanel {
2121

2222
private SamlTabController controller;
2323

24-
private JButton btnMessageReset;
25-
private JCheckBox chkRawMode;
24+
private final JButton btnMessageReset = new JButton("Reset Message");
25+
private final JCheckBox chkRawMode = new JCheckBox("Raw Mode (don't parse XML before sending)");
2626

27-
private JButton btnXSWHelp;
28-
private JComboBox<String> cmbboxXSW;
29-
private JButton btnXSWPreview;
30-
private JButton btnMatchAndReplace;
31-
private JButton btnXSWApply;
27+
private final JButton btnXSWHelp = new JButton("Help");
28+
private final JComboBox<String> cmbboxXSW = new JComboBox<>();
29+
private final JButton btnXSWPreview = new JButton("Preview in Browser...");
30+
private final JButton btnMatchAndReplace = new JButton("Match and Replace");
31+
private final JButton btnXSWApply = new JButton("Apply XSW");
3232

33-
private JButton btnTestXXE;
34-
private JButton btnTestXSLT;
33+
private final JButton btnTestXXE = new JButton("Test XXE");
34+
private final JButton btnTestXSLT = new JButton("Test XSLT");
3535

36-
private JButton btnSignatureHelp;
37-
private JComboBox<BurpCertificate> cmbboxCertificate;
38-
private JButton btnSignatureRemove;
39-
private JButton btnResignAssertion;
40-
private JButton btnSendCertificate;
41-
private JButton btnResignMessage;
36+
private final JButton btnSignatureHelp = new JButton("Help");
37+
private final JComboBox<BurpCertificate> cmbboxCertificate = new JComboBox<>();
38+
private final JButton btnSignatureRemove = new JButton("Remove Signatures");
39+
private final JButton btnResignAssertion = new JButton("(Re-)Sign Assertion");
40+
private final JButton btnSendCertificate = new JButton("Send Certificate to SAML Raider Certificates");
41+
private final JButton btnResignMessage = new JButton("(Re-)Sign Message");
4242

43-
private JLabel lblStatusMessage;
43+
private final JLabel lblStatusMessage = new JLabel("");
4444

4545
public SamlPanelAction() {
4646
initialize();
@@ -52,10 +52,11 @@ public SamlPanelAction(SamlTabController controller) {
5252
}
5353

5454
private void initialize() {
55-
btnMessageReset = new JButton("Reset Message");
56-
btnMessageReset.addActionListener(event -> controller.resetMessage());
55+
btnMessageReset.addActionListener(event -> {
56+
controller.resetMessage();
57+
lblStatusMessage.setText("");
58+
});
5759

58-
chkRawMode = new JCheckBox("Raw Mode (don't parse XML before sending)");
5960
chkRawMode.addActionListener(event -> controller.setRawMode(chkRawMode.isSelected()));
6061

6162
var samlMessagePanel = new JPanel();
@@ -64,18 +65,12 @@ private void initialize() {
6465
samlMessagePanel.add(btnMessageReset, "wrap");
6566
samlMessagePanel.add(chkRawMode, "wrap");
6667

67-
btnXSWHelp = new JButton("Help");
6868
btnXSWHelp.addActionListener(event -> controller.showXSWHelp());
6969

70-
cmbboxXSW = new JComboBox<>();
71-
72-
btnXSWPreview = new JButton("Preview in Browser...");
7370
btnXSWPreview.addActionListener(event -> controller.showXSWPreview());
7471

75-
btnMatchAndReplace = new JButton("Match and Replace");
7672
btnMatchAndReplace.addActionListener(event -> showMatchAndReplaceDialog());
7773

78-
btnXSWApply = new JButton("Apply XSW");
7974
btnXSWApply.addActionListener(event -> controller.applyXSW());
8075

8176
var xswAttacksPanel = new JPanel();
@@ -87,12 +82,10 @@ private void initialize() {
8782
xswAttacksPanel.add(btnXSWPreview);
8883
xswAttacksPanel.add(btnXSWApply, "wrap");
8984

90-
btnTestXXE = new JButton("Test XXE");
9185
btnTestXXE.addActionListener(event ->
9286
Optional.ofNullable(JOptionPane.showInputDialog(btnXSWApply, "Enter Burp Collaborator URL (e.g. https://xyz.burpcollaborator.net)"))
9387
.ifPresent(controller::applyXXE));
9488

95-
btnTestXSLT = new JButton("Test XSLT");
9689
btnTestXSLT.addActionListener(event ->
9790
Optional.ofNullable(JOptionPane.showInputDialog(btnXSWApply, "Enter Burp Collaborator URL (e.g. https://xyz.burpcollaborator.net)"))
9891
.ifPresent(controller::applyXSLT));
@@ -103,21 +96,14 @@ private void initialize() {
10396
xmlAttacksPanel.add(btnTestXXE, "split 2");
10497
xmlAttacksPanel.add(btnTestXSLT, "wrap");
10598

106-
btnSignatureHelp = new JButton("Help");
10799
btnSignatureHelp.addActionListener(event -> controller.showSignatureHelp());
108100

109-
cmbboxCertificate = new JComboBox<>();
110-
111-
btnSignatureRemove = new JButton("Remove Signatures");
112101
btnSignatureRemove.addActionListener(event -> controller.removeSignature());
113102

114-
btnResignAssertion = new JButton("(Re-)Sign Assertion");
115103
btnResignAssertion.addActionListener(event -> controller.resignAssertion());
116104

117-
btnSendCertificate = new JButton("Send Certificate to SAML Raider Certificates");
118105
btnSendCertificate.addActionListener(event -> controller.sendToCertificatesTab());
119106

120-
btnResignMessage = new JButton("(Re-)Sign Message");
121107
btnResignMessage.addActionListener(event -> controller.resignMessage());
122108

123109
var signatureAttacksPanel = new JPanel();
@@ -130,7 +116,6 @@ private void initialize() {
130116
signatureAttacksPanel.add(btnResignAssertion);
131117
signatureAttacksPanel.add(btnResignMessage, "wrap");
132118

133-
lblStatusMessage = new JLabel("");
134119
lblStatusMessage.setForeground(new Color(255, 140, 0));
135120

136121
var statusMessagePanel = new JPanel();

src/main/java/gui/SignatureHelpWindow.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
package gui;
22

33
import javax.swing.*;
4-
import java.awt.*;
54
import java.io.Serial;
65

76
public class SignatureHelpWindow extends JFrame {

0 commit comments

Comments
 (0)