def gui(self): xnode = self.xdoc.getDocumentElement() tnode = self.createTree(xnode) # create tree and display jt = JTree(tnode) jsp = JScrollPane(jt) tree_box = Box(BoxLayout.Y_AXIS) tree_box.add(jsp) tree_box.add(Box.createHorizontalStrut(10)) headerSorter = TableSorter(DefaultTableModel()) jtb = JTable(headerSorter) headerSorter.addMouseListenerToHeaderInTable(jtb) jtb.setAutoResizeMode(JTable.AUTO_RESIZE_OFF) jsp2 = JScrollPane(jtb) table_box = Box(BoxLayout.Y_AXIS) table_box.add(jsp2) table_box.add(Box.createHorizontalStrut(500)) mp = JPanel() mp.setLayout(BoxLayout(mp,BoxLayout.X_AXIS)) mp.add(tree_box) mp.add(table_box) # add listeners nsl = NodeSelectionListener(jtb,xnode) jt.addTreeSelectionListener(nsl) # return mp
def run(self):
frame = JFrame('horizontalBox',
locationRelativeTo=None,
defaultCloseOperation=JFrame.EXIT_ON_CLOSE)
vert = Box.createVerticalBox()
vert.add(Box.createGlue())
vert.add(JButton('<>'))
vert.add(Box.createVerticalStrut(5))
vert.add(JButton('<>'))
vert.add(Box.createVerticalStrut(5))
vert.add(JButton('<>'))
vert.add(Box.createGlue())
hor = Box.createHorizontalBox()
hor.add(Box.createGlue())
hor.add(JButton('<>'))
hor.add(Box.createHorizontalStrut(5))
hor.add(vert)
hor.add(Box.createHorizontalStrut(5))
hor.add(JButton('<>'))
hor.add(Box.createGlue())
frame.add(hor)
frame.pack()
frame.setVisible(1)
def __init__(self):
super(BeautifierPanel, self).__init__()
self.setLayout(BorderLayout())
self.beautifyTextArea = JTextArea(5, 10)
self.beautifyTextArea.setLineWrap(True)
self.beautifyTextArea.setDocument(self.CustomUndoPlainDocument())
# The undo doesn't work well before replace text. Below is rough fix, so not need to know how undo work for now
self.beautifyTextArea.setText(" ")
self.beautifyTextArea.setText("")
self.undoManager = UndoManager()
self.beautifyTextArea.getDocument().addUndoableEditListener(
self.undoManager)
self.beautifyTextArea.getDocument().addDocumentListener(
self.BeautifyDocumentListener(self))
beautifyTextWrapper = JPanel(BorderLayout())
beautifyScrollPane = JScrollPane(self.beautifyTextArea)
beautifyTextWrapper.add(beautifyScrollPane, BorderLayout.CENTER)
self.add(beautifyTextWrapper, BorderLayout.CENTER)
self.beautifyButton = JButton("Beautify")
self.beautifyButton.addActionListener(self.beautifyListener)
self.undoButton = JButton("Undo")
self.undoButton.addActionListener(self.undoListener)
formatLabel = JLabel("Format:")
self.formatsComboBox = JComboBox()
for f in supportedFormats:
self.formatsComboBox.addItem(f)
self.statusLabel = JLabel("Status: Ready")
preferredDimension = self.statusLabel.getPreferredSize()
self.statusLabel.setPreferredSize(
Dimension(preferredDimension.width + 20,
preferredDimension.height))
self.sizeLabel = JLabel("0 B")
preferredDimension = self.sizeLabel.getPreferredSize()
self.sizeLabel.setPreferredSize(
Dimension(preferredDimension.width + 64,
preferredDimension.height))
self.sizeLabel.setHorizontalAlignment(SwingConstants.RIGHT)
buttonsPanel = JPanel(FlowLayout())
buttonsPanel.add(formatLabel)
buttonsPanel.add(self.formatsComboBox)
buttonsPanel.add(Box.createHorizontalStrut(10))
buttonsPanel.add(self.beautifyButton)
buttonsPanel.add(self.undoButton)
bottomPanel = JPanel(BorderLayout())
bottomPanel.add(self.statusLabel, BorderLayout.WEST)
bottomPanel.add(buttonsPanel, BorderLayout.CENTER)
bottomPanel.add(self.sizeLabel, BorderLayout.EAST)
self.add(bottomPanel, BorderLayout.SOUTH)
self.currentBeautifyThread = None
def initLayout(self):
# Create a horizontal layout and a 10 pixel border
self.layout = BoxLayout(self.contentPane, BoxLayout.X_AXIS)
self.contentPane.border = BorderFactory.createEmptyBorder(
10, 10, 10, 10)
# Add the table
self.add(JScrollPane(self.peopleTable))
# Add a 10 pixel gap
self.add(Box.createHorizontalStrut(10))
# Create a vertical container
vbox = Box(BoxLayout.Y_AXIS)
self.add(vbox)
# Add the form fields
grid = JPanel(GridLayout(5, 2, 10, 10))
grid.add(JLabel(u"First name"))
grid.add(self.firstNameField)
grid.add(JLabel(u"Last name"))
grid.add(self.lastNameField)
grid.add(JLabel(u"Birth year"))
grid.add(self.birthYearField)
grid.add(JLabel(u"Summary"))
grid.add(self.summaryField)
vbox.add(grid)
# Add the buttons
buttonRow = Box(BoxLayout.X_AXIS)
buttonRow.add(self.addButton)
buttonRow.add(Box.createHorizontalStrut(10))
buttonRow.add(self.removeButton)
vbox.add(buttonRow)
vbox.add(Box.createVerticalGlue())
def registerExtenderCallbacks(self, callbacks):
print "Loading..."
self._callbacks = callbacks
self._callbacks.setExtensionName('Burp SPA Explorer')
# self._callbacks.registerScannerCheck(self)
# self._callbacks.registerExtensionStateListener(self)
self._helpers = callbacks.getHelpers()
self.crawlingEvent = Event()
self.crawlerThread = None
# main split pane
self._splitpane = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self._splitpane.setBorder(EmptyBorder(20, 20, 20, 20))
# sub split pane (top)
self._topPanel = JPanel(BorderLayout(10, 10))
self._topPanel.setBorder(EmptyBorder(0, 0, 10, 0))
# Setup Panel : [Target: ] [______________________] [START BUTTON]
self.setupPanel = JPanel(FlowLayout(FlowLayout.LEADING, 10, 10))
self.setupPanel.add(JLabel("Target:", SwingConstants.LEFT),
BorderLayout.LINE_START)
self.hostField = JTextField('', 50)
self.setupPanel.add(self.hostField)
self.toggleButton = JButton('Start crawling',
actionPerformed=self.toggleCrawl)
self.setupPanel.add(self.toggleButton)
self._topPanel.add(self.setupPanel, BorderLayout.PAGE_START)
# Options Panel : [Buttons] [ RegEx ]
self.optionsPanel = JPanel()
self.optionsPanel.setLayout(
BoxLayout(self.optionsPanel, BoxLayout.LINE_AXIS))
# Button options panel : [Add][Edit][Up][Down][Remove]
self.buttonOptionsPanel = JPanel()
self.buttonOptionsPanel.setLayout(
BoxLayout(self.buttonOptionsPanel, BoxLayout.PAGE_AXIS))
self.addRegexButton = JButton('Add', actionPerformed=self.addRegex)
self.buttonOptionsPanel.add(self.addRegexButton)
self.editRegexButton = JButton('Edit', actionPerformed=self.editRegex)
self.buttonOptionsPanel.add(self.editRegexButton)
self.moveRegexUpButton = JButton('Move up',
actionPerformed=self.moveRegexUp)
self.buttonOptionsPanel.add(self.moveRegexUpButton)
self.moveRegexDownButton = JButton('Move down',
actionPerformed=self.moveRegexDown)
self.buttonOptionsPanel.add(self.moveRegexDownButton)
self.removeRegexButton = JButton('Remove',
actionPerformed=self.removeRegex)
self.buttonOptionsPanel.add(self.removeRegexButton)
self.buttonOptionsPanel.add(Box.createVerticalGlue())
self.optionsPanel.add(self.buttonOptionsPanel)
self.optionsPanel.add(Box.createHorizontalStrut(20))
self.regexTableModel = RegexTableModel([x for x in regex])
self.regexTable = Table(self.regexTableModel)
self.regexScrollPane = JScrollPane(self.regexTable)
self.optionsPanel.add(self.regexScrollPane)
self._topPanel.add(self.optionsPanel, BorderLayout.CENTER)
self._splitpane.setTopComponent(self._topPanel)
# Bottom Panel
self._bottomPanel = JPanel(BorderLayout(10, 10))
#self._bottomPanel.setLayout(BoxLayout(self._bottomPanel,BoxLayout.PAGE_AXIS))
# Status bar
self.crawlStatusPanel = JPanel(FlowLayout(FlowLayout.LEADING, 10, 10))
self.crawlStatusPanel.add(JLabel("Status: ", SwingConstants.LEFT))
self.crawlStatusLabel = JLabel("Ready to crawl", SwingConstants.LEFT)
self.crawlStatusPanel.add(self.crawlStatusLabel)
# Result Table
self.resultTableModel = Result([])
self.resultTable = Table(self.resultTableModel)
self.resultTable.setAutoCreateRowSorter(True)
self.resultScrollPane = JScrollPane(self.resultTable)
# Result Table popup menu
def selectWhenRightClickEvent(event):
def select(e):
rowAtPoint = self.resultTable.rowAtPoint(
SwingUtilities.convertPoint(self.resultTablePopupMenu,
Point(0, 0), self.resultTable))
if rowAtPoint > -1:
self.resultTable.setRowSelectionInterval(
rowAtPoint, rowAtPoint)
SwingUtilities.invokeLater(CrawlerRunnable(select, (event, )))
self.resultTablePopupMenu = JPopupMenu(
popupMenuWillBecomeVisible=selectWhenRightClickEvent)
self.resultTablePopupMenu.add(
JMenuItem("Send to scanner", actionPerformed=self.sendToScanner))
self.resultTablePopupMenu.add(
JMenuItem("Send to repeater", actionPerformed=self.sendToRepeater))
self.resultTablePopupMenu.add(
JMenuItem("Send to intruder", actionPerformed=self.sendToIntruder))
self.resultTablePopupMenu.add(
JMenuItem("Send to spider", actionPerformed=self.sendToSpider))
self.resultTable.setComponentPopupMenu(self.resultTablePopupMenu)
self._bottomPanel.add(self.resultScrollPane, BorderLayout.CENTER)
self._bottomPanel.add(self.crawlStatusPanel, BorderLayout.SOUTH)
self._splitpane.setBottomComponent(self._bottomPanel)
self._splitpane.setDividerLocation(300 +
self._splitpane.getInsets().left)
callbacks.customizeUiComponent(self._splitpane)
callbacks.addSuiteTab(self)
explorerMenu = ExplorerMenu(self)
callbacks.registerContextMenuFactory(explorerMenu)
print "SPA Explorer custom menu loaded"
#print "Loading chrome driver"
#a = Test(os.path.dirname(os.path.realpath('selenium-client.jar')) + '/chromedriver.exe')
#print "Chrome driver started"
print "Burp SPA Explorer loaded"
def __init__(self, extender):
super(BeautifierOptionsPanel, self).__init__()
self._extender = extender
self.contentWrapper = JPanel(GridBagLayout())
self.setViewportView(self.contentWrapper)
self.getVerticalScrollBar().setUnitIncrement(16)
self.addMouseListener(self.RequestFocusListener(
self)) # Let textArea lose focus when click empty area
innerContainer = JPanel(GridBagLayout())
innerContainer.setFocusable(
True
) # make sure the maxSizeText TextField is not focused when BeautifierOptionsPanel display
# generalOptionPanel and it's inner component
maxSizeLabel = JLabel("Max Size: ")
self.maxSizeText = JTextField(5)
self.maxSizeText.setHorizontalAlignment(SwingConstants.RIGHT)
self.maxSizeText.addFocusListener(self.MaxSizeTextListener(self))
sizeUnitLabel = JLabel("KB")
generalOptionPanel = JPanel(GridBagLayout())
generalOptionPanel.setBorder(
BorderFactory.createTitledBorder("General Options"))
gbc = GridBagConstraints()
gbc.anchor = GridBagConstraints.WEST
gbc.gridx = 0
gbc.gridy = 0
generalOptionPanel.add(maxSizeLabel, gbc)
gbc.fill = GridBagConstraints.HORIZONTAL
gbc.gridx = 1
gbc.gridy = 0
generalOptionPanel.add(self.maxSizeText, gbc)
gbc.fill = GridBagConstraints.NONE
gbc.gridx = 2
gbc.gridy = 0
gbc.weightx = 1.0
generalOptionPanel.add(sizeUnitLabel, gbc)
gbc = GridBagConstraints()
gbc.fill = GridBagConstraints.BOTH
gbc.gridx = 1
gbc.gridy = 0
gbc.gridheight = 2
innerContainer.add(generalOptionPanel, gbc)
# messageTabOptionPanel and it's inner component
self.messageTabFormatCheckBoxs = []
for f in supportedFormats:
ckb = JCheckBox(f)
ckb.addItemListener(self.messageTabFormatListener)
self.messageTabFormatCheckBoxs.append(ckb)
messageTabOptionPanel = JPanel()
messageTabOptionPanel.setLayout(
BoxLayout(messageTabOptionPanel, BoxLayout.Y_AXIS))
messageTabOptionPanel.setBorder(
BorderFactory.createTitledBorder("Enable in MessageEditorTab"))
for b in self.messageTabFormatCheckBoxs:
messageTabOptionPanel.add(b)
gbc.gridx = 1
gbc.gridy = 2
gbc.gridheight = 9
innerContainer.add(messageTabOptionPanel, gbc)
# replaceResponsePanel and it's inner component
self.chkEnableReplace = JCheckBox("Enable")
self.chkEnableReplace.addItemListener(self.repalceResponseBoxListener)
replaceResponseFormatLabel = JLabel("Format")
self.replaceResponseFormatCheckBoxs = []
for f in supportedFormats:
ckb = JCheckBox(f)
ckb.addItemListener(self.replaceResponseFormatListener)
self.replaceResponseFormatCheckBoxs.append(ckb)
replaceResponseIncludeLabel = JLabel(
"Include URL that matches below(one item one line)")
self.URLIncludeTextArea = JTextArea(6, 32)
self.URLIncludeTextArea.addFocusListener(
self.URLIncludeFocusListener(self))
URLIncludeScrollPane = JScrollPane(self.URLIncludeTextArea)
URLIncludeScrollPane.setAlignmentX(Component.LEFT_ALIGNMENT)
replaceResponseExcludeLabel = JLabel(
"Exclude URL that matches below(one item one line)")
self.URLExcludeTextArea = JTextArea(5, 32)
self.URLExcludeTextArea.addFocusListener(
self.URLExcludeFocusListener(self))
URLExcludeScrollPane = JScrollPane(self.URLExcludeTextArea)
URLExcludeScrollPane.setAlignmentX(Component.LEFT_ALIGNMENT)
replaceResponsePanel = JPanel()
replaceResponsePanel.setLayout(
BoxLayout(replaceResponsePanel, BoxLayout.Y_AXIS))
replaceResponsePanel.setBorder(
BorderFactory.createTitledBorder("Replace PROXY Response"))
replaceResponsePanel.add(self.chkEnableReplace)
replaceResponsePanel.add(Box.createVerticalStrut(10))
replaceResponsePanel.add(replaceResponseFormatLabel)
for b in self.replaceResponseFormatCheckBoxs:
replaceResponsePanel.add(b)
replaceResponsePanel.add(Box.createVerticalStrut(10))
replaceResponsePanel.add(replaceResponseIncludeLabel)
replaceResponsePanel.add(URLIncludeScrollPane)
replaceResponsePanel.add(Box.createVerticalStrut(10))
replaceResponsePanel.add(replaceResponseExcludeLabel)
replaceResponsePanel.add(URLExcludeScrollPane)
gbc.gridy = 11
innerContainer.add(replaceResponsePanel, gbc)
# let innerContainer keep away from left and up
gbc = GridBagConstraints()
gbc.gridx = 1
gbc.gridy = 1
self.contentWrapper.add(Box.createHorizontalStrut(15), gbc)
# gbc.ipadx = gbc.ipady = 25
gbc.gridx = 2
self.contentWrapper.add(innerContainer, gbc)
# let innerContainer stay left side
gbc = GridBagConstraints()
gbc.gridx = 3
gbc.gridy = 2
gbc.gridwidth = 1
gbc.weightx = gbc.weighty = 1
paddingPanel = JPanel()
self.contentWrapper.add(paddingPanel, gbc)
self.setDefaultOptionDisplay()
