def helpMenu(self,event):
self._helpPopup = JFrame('JWT Fuzzer', size=(550, 450) );
self._helpPopup.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE)
helpPanel = JPanel()
helpPanel.setPreferredSize(Dimension(550, 450))
helpPanel.setBorder(EmptyBorder(10, 10, 10, 10))
helpPanel.setLayout(BoxLayout(helpPanel, BoxLayout.Y_AXIS))
self._helpPopup.setContentPane(helpPanel)
helpHeadingText = JLabel("<html><h2>JWT Fuzzer</h2></html>")
authorText = JLabel("<html><p>@author: <pinnace></p></html>")
aboutText = JLabel("<html><br /> <p>This extension adds an Intruder payload processor for JWTs.</p><br /></html>")
repositoryText = JLabel("<html>Documentation and source code:</html>")
repositoryLink = JLabel("<html>- <a href=\"https://github.com/pinnace/burp-jwt-fuzzhelper-extension\">https://github.com/pinnace/burp-jwt-fuzzhelper-extension</a></html>")
licenseText = JLabel("<html><br/><p>JWT Fuzzer uses a GPL 3 license. This license does not apply to the dependency below:<p></html>")
dependencyLink = JLabel("<html>- <a href=\"https://github.com/jpadilla/pyjwt/blob/master/LICENSE\">pyjwt</a></html>")
dependencyLink.addMouseListener(ClickListener())
dependencyLink.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR))
repositoryLink.addMouseListener(ClickListener())
repositoryLink.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR))
helpPanel.add(helpHeadingText)
helpPanel.add(authorText)
helpPanel.add(aboutText)
helpPanel.add(repositoryText)
helpPanel.add(repositoryLink)
helpPanel.add(licenseText)
helpPanel.add(dependencyLink)
self._helpPopup.setSize(Dimension(550, 450))
self._helpPopup.pack()
self._helpPopup.setLocationRelativeTo(None)
self._helpPopup.setVisible(True)
return
def createButtonPane(self):
"""Create a new button pane for the message editor tab"""
self._button_listener = EditorButtonListener(self)
panel = JPanel()
panel.setLayout(BoxLayout(panel, BoxLayout.Y_AXIS))
panel.setBorder(EmptyBorder(5, 5, 5, 5))
panel.add(Box.createRigidArea(Dimension(0, 5)))
type_scroll_pane = JScrollPane(self._type_list_component)
type_scroll_pane.setMaximumSize(Dimension(200, 100))
type_scroll_pane.setMinimumSize(Dimension(150, 100))
panel.add(type_scroll_pane)
panel.add(Box.createRigidArea(Dimension(0, 3)))
new_type_panel = JPanel()
new_type_panel.setLayout(BoxLayout(new_type_panel, BoxLayout.X_AXIS))
new_type_panel.add(self._new_type_field)
new_type_panel.add(Box.createRigidArea(Dimension(3, 0)))
new_type_panel.add(
self.createButton(
"New", "new-type", "Save this message's type under a new name"
)
)
new_type_panel.setMaximumSize(Dimension(200, 20))
new_type_panel.setMinimumSize(Dimension(150, 20))
panel.add(new_type_panel)
button_panel = JPanel()
button_panel.setLayout(FlowLayout())
if self._editable:
button_panel.add(
self.createButton(
"Validate", "validate", "Validate the message can be encoded."
)
)
button_panel.add(
self.createButton("Edit Type", "edit-type", "Edit the message type")
)
button_panel.add(
self.createButton(
"Reset Message", "reset", "Reset the message and undo changes"
)
)
button_panel.add(
self.createButton(
"Clear Type", "clear-type", "Reparse the message with an empty type"
)
)
button_panel.setMinimumSize(Dimension(100, 200))
button_panel.setPreferredSize(Dimension(200, 1000))
panel.add(button_panel)
return panel
def getMainComponent(self):
self._mainPanel = JPanel(BorderLayout())
#Left panel
self._leftPanel = JPanel(BorderLayout())
self._leftPanel.setBorder(EmptyBorder(10, 10, 10, 10))
#Left subpanel - Positions editor
self._positionsEditor = Utils.callbacks.createTextEditor()
#TODO Remove a normal editor? self._positionsEditor = Utils.callbacks.createMessageEditor(self, True)
self._positionsEditor.getComponent().setBorder(BorderFactory.createLineBorder(Color.BLACK))
#Left subpanel - Title pane
self._leftTitlePanel = JPanel(GridLayout(0, 1))
self._leftTitlePanel.setBorder(EmptyBorder(0, 10, 10, 10))
self._titleText = JLabel("Commands Position")
self._titleText.setForeground(COLOR_BURP_TITLE_ORANGE)
self._titleText.setFont(self._titleText.getFont().deriveFont(16.0))
self._titleSubtitleText = JTextArea("Configure the position where commands will be inserted into the base request. Select the requests that were send Shell in the dropdown, then select the part of the request where commands need to be inserted and click the 'Add $' button.")
self._titleSubtitleText.setEditable(False)
self._titleSubtitleText.setLineWrap(True)
self._titleSubtitleText.setWrapStyleWord(True)
self._titleSubtitleText.setHighlighter(None)
self._titleSubtitleText.setBorder(None)
self._leftTitlePanel.add(self._titleText)
self._leftTitlePanel.add(self._titleSubtitleText)
#Left subpanel - Add positions editor and title
self._leftPanel.add(self._leftTitlePanel, BorderLayout.NORTH)
self._leftPanel.add(self._positionsEditor.getComponent(), BorderLayout.CENTER)
#Right panel
#self._rightPanel = JPanel(GridLayout(20, 1))
self._rightPanel = JPanel()
self._rightPanel.setLayout(BoxLayout(self._rightPanel, BoxLayout.Y_AXIS))
#self._rightPanel.setPreferredSize(Dimension(150, 30))
self._rightPanel.setBorder(EmptyBorder(10, 10, 10, 10))
#Right panel - buttons
self._buttonAdd = JButton(" Add $ ", actionPerformed=self.buttonAddClick)
self._buttonClear = JButton(" Clear $ ", actionPerformed=self.buttonClearClick) #, actionPerformed=None
# Right panel - add components
self._rightPanel.add(self._buttonAdd)
self._rightPanel.add(self._buttonClear)
self._mainPanel.add(self._rightPanel, BorderLayout.EAST)
self._mainPanel.add(self._leftPanel, BorderLayout.CENTER)
return self._mainPanel
def __init__(self, parent, name):
JTextField.__init__(self, name)
self.setBorder(EmptyBorder(0, 0, 0, 0)) #works good, looks bad
self.setBackground(Color(0, 0, 0, 0)) #transparent background
self._parent = parent
self.setEditable(False)
self.addMouseListener(self)
self.addFocusListener(self)
def makeUI(model):
# Components:
table = JTable(model)
jsp = JScrollPane(table)
regex_label = JLabel("Search: ")
regex_field = JTextField(20)
base_path_label = JLabel("Base path:")
base_path_field = JTextField(50)
if base_path is not None:
base_path_field.setText(base_path)
# Panel for all components
all = JPanel()
all.setBorder(EmptyBorder(20, 20, 20, 20))
layout, c = GridBagLayout(), GC()
all.setLayout(layout)
# First row: label and regex text field
add(all, regex_label, gridx=0, gridy=0) # with default constraints
add(all, regex_field, gridx=1, gridy=0, fill=GC.HORIZONTAL, weightx=1.0)
# Second row: the table
add(all,
jsp,
gridx=0,
gridy=1,
fill=GC.BOTH,
gridwidth=2,
weightx=1.0,
weighty=1.0) # full weights so it stretches when resizing
# Third row: the base path
add(all, base_path_label, gridx=0, gridy=2)
add(all,
base_path_field,
gridx=1,
gridy=2,
fill=GC.HORIZONTAL,
weightx=1.0)
# Window frame
frame = JFrame("File paths")
frame.getContentPane().add(all)
#frame.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE)
frame.addWindowListener(Closing())
frame.pack()
frame.setVisible(True)
# Listeners
regex_field.addKeyListener(EnterListener(table))
table.addMouseListener(RowClickListener(base_path_field))
al = ArrowListener(table, regex_field)
table.addKeyListener(al)
regex_field.addKeyListener(al)
# Instead of a KeyListener, use the input vs action map
table.getInputMap().put(KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, 0),
"enter")
table.getActionMap().put("enter", OpenImageFromTableCell())
#
return model, table, regex_field, frame
def __init__(self, extender):
# Initialize self
super(HeadPanel, self).__init__()
self.extender = extender
self.setLayout(BorderLayout())
self.setBorder(EmptyBorder(10, 10, 10, 10))
# Create children
self.title = TitlePanel(extender)
self.actions = ActionsPanel(extender)
# Add children
self.add(self.title, BorderLayout.WEST)
self.add(self.actions, BorderLayout.EAST)
def createPanel(scroll=False, ptop=0, pleft=0, pbottom=0, pright=0):
panel = JPanel()
panel.setLayout(BoxLayout(panel, BoxLayout.PAGE_AXIS))
panel.setAlignmentX(Component.LEFT_ALIGNMENT)
panel.setBorder(EmptyBorder(ptop, pleft, pbottom, pright))
# if scroll:
# scrollpane = JScrollPane(panel)
# scrollpane.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED)
# scrollpane.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_NEVER)
# return JPanel().add(scrollpane)
return panel
def editRegex(self, event):
selectedRowIdx = self.regexTable.getSelectedRow()
if selectedRowIdx == -1: return False
selectedRow = self.regexTableModel.data[selectedRowIdx]
optionPane = JOptionPane()
dialog = optionPane.createDialog(self._splitpane, "Edit RegEx")
panel = JPanel(GridLayout(0, 2))
panel.setBorder(EmptyBorder(10, 10, 10, 10))
nameField = JTextField('', 15)
nameField.text = selectedRow[0]
panel.add(JLabel("Name:", SwingConstants.LEFT))
panel.add(nameField)
regexField = JTextField('', 15)
regexField.text = selectedRow[1]
panel.add(JLabel("RegEx:", SwingConstants.LEFT))
panel.add(regexField)
crawlField = JCheckBox()
crawlField.setSelected(selectedRow[2])
panel.add(JLabel("Crawl:", SwingConstants.LEFT))
panel.add(crawlField)
def closeDialog(event):
if len(nameField.text) == 0 or len(regexField.text) == 0:
JOptionPane.showMessageDialog(self._splitpane,
"Name or RegEx can't be empty",
"Error",
JOptionPane.ERROR_MESSAGE)
return
self.regexTableModel.editRow(
selectedRowIdx,
[nameField.text, regexField.text,
crawlField.isSelected()])
dialog.hide()
editButton = JButton('OK', actionPerformed=closeDialog)
panel.add(editButton)
dialog.setSize(600, 200)
dialog.setContentPane(panel)
self._callbacks.customizeUiComponent(dialog)
dialog.show()
return True
def build_gui(self):
"""Construct GUI elements."""
panel = JPanel(BorderLayout(3, 3))
panel.setBorder(EmptyBorder(160, 160, 160, 160))
self.aws_access_key_inpt = JTextField(10)
self.aws_secret_key_inpt = JTextField(10)
self.aws_session_token_inpt = JTextField(10)
self.gs_access_key_inpt = JTextField(10)
self.gs_secret_key_inpt = JTextField(10)
self.wordlist_path_inpt = JTextField(10)
self.passive_mode = JCheckBox('Enabled')
self.ssl_verification = JCheckBox('Enabled')
save_btn = JButton('Save', actionPerformed=self.save_config)
labels = JPanel(GridLayout(0, 1))
inputs = JPanel(GridLayout(0, 1))
panel.add(labels, BorderLayout.WEST)
panel.add(inputs, BorderLayout.CENTER)
top_label = JLabel('<html><b>Settings</b><br><br></html>')
top_label.setHorizontalAlignment(JLabel.CENTER)
panel.add(top_label, BorderLayout.NORTH)
labels.add(JLabel('AWS Access Key:'))
inputs.add(self.aws_access_key_inpt)
labels.add(JLabel('AWS Secret Key:'))
inputs.add(self.aws_secret_key_inpt)
labels.add(JLabel('AWS Session Key (optional):'))
inputs.add(self.aws_session_token_inpt)
labels.add(JLabel('GS Access Key:'))
inputs.add(self.gs_access_key_inpt)
labels.add(JLabel('GS Secret Key:'))
inputs.add(self.gs_secret_key_inpt)
labels.add(JLabel('Wordlist Filepath (optional):'))
inputs.add(self.wordlist_path_inpt)
labels.add(JLabel('Passive Mode:'))
inputs.add(self.passive_mode)
labels.add(JLabel('SSL Verification:'))
inputs.add(self.ssl_verification)
panel.add(save_btn, BorderLayout.SOUTH)
return panel
def __init__(self, extension, burp_callbacks):
self._burp_callbacks = burp_callbacks
self._extension = extension
self._type_list_component = JList(extension.known_message_model)
self._type_list_component.setSelectionMode(
ListSelectionModel.MULTIPLE_INTERVAL_SELECTION)
self._component = JPanel()
self._component.setLayout(BoxLayout(self._component, BoxLayout.Y_AXIS))
splitPane = JSplitPane(JSplitPane.HORIZONTAL_SPLIT)
splitPane.setRightComponent(JScrollPane(self._type_list_component))
splitPane.setLeftComponent(self.createButtonPane())
splitPane.setResizeWeight(0.03)
splitPane.setMaximumSize(Dimension(1000, 1000))
self._component.add(splitPane)
self._component.add(Box.createVerticalGlue())
self._component.setBorder(EmptyBorder(10, 10, 10, 10))
def __init__(self, view, doc):
EnhancedDialog.__init__(self, view,
jEdit.getProperty("jython.export.title"), 1)
self.view = view
self.doc = doc
panel = Box.createVerticalBox()
panel.border = EmptyBorder(10, 10, 10, 10)
self.header = JCheckBox(jEdit.getProperty("jython.export.header"), 0)
panel.add(self.header)
self.prompts = JCheckBox(jEdit.getProperty("jython.export.prompts"), 0)
panel.add(self.prompts)
self.errors = JCheckBox(jEdit.getProperty("jython.export.errors"), 0)
panel.add(self.errors)
self.contentPane.layout = BorderLayout()
self.contentPane.add(panel, BorderLayout.CENTER)
lowerPanel = JPanel(FlowLayout.LEFT)
lowerPanel.add(JButton(jEdit.getProperty("common.ok"), actionPerformed = \
self._ok))
lowerPanel.add(JButton(jEdit.getProperty("common.cancel"), actionPerformed = \
self._cancel))
self.contentPane.add(lowerPanel, BorderLayout.SOUTH)
def drawUI(self):
self.tab = swing.JPanel()
self.uiLabel = swing.JLabel('Site Map Extractor Options')
self.uiLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiLabel.setForeground(Color(235, 136, 0))
self.uiScopeOnly = swing.JRadioButton('In-scope only', True)
self.uiScopeAll = swing.JRadioButton('Full site map', False)
self.uiScopeButtonGroup = swing.ButtonGroup()
self.uiScopeButtonGroup.add(self.uiScopeOnly)
self.uiScopeButtonGroup.add(self.uiScopeAll)
self.uipaneA = swing.JSplitPane(swing.JSplitPane.HORIZONTAL_SPLIT)
self.uipaneA.setMaximumSize(Dimension(900, 125))
self.uipaneA.setDividerSize(2)
self.uipaneB = swing.JSplitPane(swing.JSplitPane.HORIZONTAL_SPLIT)
self.uipaneB.setDividerSize(2)
self.uipaneA.setRightComponent(self.uipaneB)
self.uipaneA.setBorder(BorderFactory.createLineBorder(Color.black))
# UI for Export <a href Links
self.uiLinksPanel = swing.JPanel()
self.uiLinksPanel.setPreferredSize(Dimension(200, 75))
self.uiLinksPanel.setBorder(EmptyBorder(10, 10, 10, 10))
self.uiLinksPanel.setLayout(BorderLayout())
self.uiLinksLabel = swing.JLabel("Extract '<a href=' Links")
self.uiLinksLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiLinksAbs = swing.JCheckBox('Absolute ', True)
self.uiLinksRel = swing.JCheckBox('Relative ', True)
# create a subpanel so Run button will be centred
self.uiLinksRun = swing.JButton('Run',
actionPerformed=self.extractLinks)
self.uiLinksSave = swing.JButton('Save Log to CSV File',
actionPerformed=self.savetoCsvFile)
self.uiLinksClear = swing.JButton('Clear Log',
actionPerformed=self.clearLog)
self.uiLinksButtonPanel = swing.JPanel()
self.uiLinksButtonPanel.add(self.uiLinksRun)
self.uiLinksButtonPanel.add(self.uiLinksSave)
self.uiLinksButtonPanel.add(self.uiLinksClear)
# add all elements to main Export Links panel
self.uiLinksPanel.add(self.uiLinksLabel, BorderLayout.NORTH)
self.uiLinksPanel.add(self.uiLinksAbs, BorderLayout.WEST)
self.uiLinksPanel.add(self.uiLinksRel, BorderLayout.CENTER)
self.uiLinksPanel.add(self.uiLinksButtonPanel, BorderLayout.SOUTH)
self.uipaneA.setLeftComponent(
self.uiLinksPanel) # add Export Links panel to splitpane
# UI for Export Response Codes
self.uiCodesPanel = swing.JPanel()
self.uiCodesPanel.setPreferredSize(Dimension(200, 75))
self.uiCodesPanel.setBorder(EmptyBorder(10, 10, 10, 10))
self.uiCodesPanel.setLayout(BorderLayout())
self.uiCodesLabel = swing.JLabel('Extract Response Codes')
self.uiCodesLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiRcodePanel = swing.JPanel()
self.uiRcodePanel.setLayout(GridLayout(1, 1))
self.uiRcode1xx = swing.JCheckBox('1XX ', False)
self.uiRcode2xx = swing.JCheckBox('2XX ', True)
self.uiRcode3xx = swing.JCheckBox('3XX ', True)
self.uiRcode4xx = swing.JCheckBox('4XX ', True)
self.uiRcode5xx = swing.JCheckBox('5XX ', True)
self.uiCodesRun = swing.JButton('Run',
actionPerformed=self.exportCodes)
self.uiCodesSave = swing.JButton('Save Log to CSV File',
actionPerformed=self.savetoCsvFile)
self.uiCodesClear = swing.JButton('Clear Log',
actionPerformed=self.clearLog)
self.uiCodesButtonPanel = swing.JPanel()
self.uiCodesButtonPanel.add(self.uiCodesRun)
self.uiCodesButtonPanel.add(self.uiCodesSave)
self.uiCodesButtonPanel.add(self.uiCodesClear)
self.uiRcodePanel.add(self.uiRcode1xx)
self.uiRcodePanel.add(self.uiRcode2xx)
self.uiRcodePanel.add(self.uiRcode3xx)
self.uiRcodePanel.add(self.uiRcode4xx)
self.uiRcodePanel.add(self.uiRcode5xx)
self.uiCodesPanel.add(self.uiCodesLabel, BorderLayout.NORTH)
self.uiCodesPanel.add(self.uiRcodePanel, BorderLayout.WEST)
self.uiCodesPanel.add(self.uiCodesButtonPanel, BorderLayout.SOUTH)
self.uipaneB.setLeftComponent(self.uiCodesPanel)
# Option 3 UI for Export Sitemap
self.uiExportPanel = swing.JPanel()
self.uiExportPanel.setPreferredSize(Dimension(200, 75))
self.uiExportPanel.setBorder(EmptyBorder(10, 10, 10, 10))
self.uiExportPanel.setLayout(BorderLayout())
self.uiExportLabel = swing.JLabel('Export Site Map to File')
self.uiExportLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiMustHaveResponse = swing.JRadioButton(
'Must have a response ', True)
self.uiAllRequests = swing.JRadioButton('All ', False)
self.uiResponseButtonGroup = swing.ButtonGroup()
self.uiResponseButtonGroup.add(self.uiMustHaveResponse)
self.uiResponseButtonGroup.add(self.uiAllRequests)
self.uiExportRun = swing.JButton('Run',
actionPerformed=self.exportSiteMap)
self.uiExportClear = swing.JButton('Clear Log',
actionPerformed=self.clearLog)
self.uiExportButtonPanel = swing.JPanel()
self.uiExportButtonPanel.add(self.uiExportRun)
self.uiExportButtonPanel.add(self.uiExportClear)
self.uiExportPanel.add(self.uiExportLabel, BorderLayout.NORTH)
self.uiExportPanel.add(self.uiMustHaveResponse, BorderLayout.WEST)
self.uiExportPanel.add(self.uiAllRequests, BorderLayout.CENTER)
self.uiExportPanel.add(self.uiExportButtonPanel, BorderLayout.SOUTH)
self.uipaneB.setRightComponent(self.uiExportPanel)
# UI Common Elements
self.uiLogLabel = swing.JLabel('Log:')
self.uiLogLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiLogPane = swing.JScrollPane()
layout = swing.GroupLayout(self.tab)
self.tab.setLayout(layout)
# Thank you to Smeege (https://github.com/SmeegeSec/Burp-Importer/) for helping me figure out how this works.
# He in turn gave credit to Antonio Sanchez (https://github.com/Dionach/HeadersAnalyzer/)
layout.setHorizontalGroup(
layout.createParallelGroup(
swing.GroupLayout.Alignment.LEADING).addGroup(
layout.createSequentialGroup().addGap(10, 10, 10).addGroup(
layout.createParallelGroup(
swing.GroupLayout.Alignment.LEADING).addComponent(
self.uiLabel).addGroup(
layout.createSequentialGroup().addGap(
10, 10, 10).addComponent(
self.uiScopeOnly).addGap(
10, 10, 10).addComponent(
self.uiScopeAll)).
addGap(15, 15,
15).addComponent(self.uipaneA).addComponent(
self.uiLogLabel).addComponent(
self.uiLogPane)).addContainerGap(
26, lang.Short.MAX_VALUE)))
layout.setVerticalGroup(
layout.createParallelGroup(swing.GroupLayout.Alignment.LEADING).
addGroup(layout.createSequentialGroup().addGap(
15, 15,
15).addComponent(self.uiLabel).addGap(15, 15, 15).addGroup(
layout.createParallelGroup().addComponent(
self.uiScopeOnly).addComponent(
self.uiScopeAll)).addGap(
20, 20, 20).addComponent(self.uipaneA).addGap(
20, 20,
20).addComponent(self.uiLogLabel).addGap(
5, 5,
5).addComponent(self.uiLogPane).addGap(
20, 20, 20)))
class UI:
global dssFile
mainWindow = ListSelection.getMainWindow()
dssFileName = mainWindow.getDSSFilename()
dssFile = HecDss.open(dssFileName)
global frame, lbl_close, list_locations, chckbxShowLocationPlot, eventsTable, dm_events, dm_meltRate
global swePaths, precipPaths, tempPaths, bList, meltRateTable, startDateField, endDateField
frame = JFrame("Snow PAC - Parameter Aggregator & Calculator")
# frame.setUndecorated(True)
frame.setBackground(Color.WHITE)
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
frame.setBounds(100, 100, 1110, 775)
contentPane = JPanel()
contentPane.setBackground(Color.WHITE)
contentPane.setBorder(EmptyBorder(5, 5, 5, 5))
frame.setContentPane(contentPane)
contentPane.setLayout(None)
class MouseListener(MouseAdapter):
# @Override
def mousePressed(self, e):
global xx
global xy
xx = e.getX()
xy = e.getY()
class MouseMotionListener(MouseMotionAdapter):
# @Override
def mouseDragged(self, arg0):
x = arg0.getXOnScreen()
y = arg0.getYOnScreen()
frame.setLocation(x - xx, y - xy)
mL = MouseListener()
mML = MouseMotionListener()
contentPane.addMouseListener(mL)
contentPane.addMouseMotionListener(mML)
if os.path.exists(img_dir + "/button.jpg"):
btnIcon = ImageIcon(img_dir + "/button.jpg")
else:
btnIcon = ImageIcon(img_dir2 + "/button.jpg")
scrollPane_events = JScrollPane()
scrollPane_events.setBounds(270, 372, 403, 263)
contentPane.add(scrollPane_events)
scrollPane_locations = JScrollPane()
scrollPane_locations.setBounds(270, 49, 403, 203)
contentPane.add(scrollPane_locations)
class deleteAction(AbstractAction):
def actionPerformed(self, deleteEvent):
# Get selected Rows and reverse list. Removes each row in list one at a time.
# List is Reversed using [::-1], so it doesn't mess up the ordering as it deletes through the loop.
for row in meltRateTable.getSelectedRows()[::-1]:
dm_meltRate.removeRow(row)
dm_meltRate.insertRow(row, [None, None])
scrollPane_table = JScrollPane()
scrollPane_table.setBounds(708, 49, 338, 586)
contentPane.add(scrollPane_table)
meltRateTable = JTable()
scrollPane_table.setViewportView(meltRateTable)
scrollPane_table.setBorder(LineBorder(Color(1, 1, 1), 2, True))
meltRateTable.setFont(Font("Tahoma", Font.PLAIN, 11))
columns = ("ATI (Deg F-Day)", "Meltrate (Inches/Deg F-Day)")
data = []
datarows = 100
data.append([0, None])
for i in range(datarows):
data.append([None, None])
dm_meltRate = DefaultTableModel(data, columns)
meltRateTable.setModel(dm_meltRate)
meltRateTable.getColumnModel().getColumn(0).setPreferredWidth(154)
meltRateTable.getColumnModel().getColumn(1).setResizable(False)
meltRateTable.getColumnModel().getColumn(1).setPreferredWidth(154)
meltRateTable.setCellSelectionEnabled(True)
# Delete data from the table using the Delete Key.
# meltRateTable.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
inputMap = meltRateTable.getInputMap(JComponent.WHEN_FOCUSED)
actionMap = meltRateTable.getActionMap()
deleteActionStr = "delete"
inputMap.put(KeyStroke.getKeyStroke(KeyEvent.VK_DELETE, 0),
deleteActionStr)
actionMap.put(deleteActionStr, deleteAction())
# jLabelStartDate = JLabel()
# jLabelStartDate.setText("Optional Start Date:")
# jLabelStartDate.setToolTipText("Optional User Specified Date Range for Paired Data. If Specified, Will be Calculated for each Water Year.")
# jLabelStartDate.setBounds(420, 263, 120, 20);
# jLabelStartDate.setFont( Font("Tahoma", Font.PLAIN, 12))
# contentPane.add(jLabelStartDate)
#
# startDateField = CalendarField();
# jLabelStartDate.setLabelFor(startDateField);
# startDateField.setMargin(Insets(0, 4, 0, 0));
# startDateField.setBounds(540, 263, 118, 22);
# startDateField.setFont(Font("Tahoma", Font.PLAIN, 12));
# startDateField.setToolTipText("Optional User Specified Date Range for Paired Data. If Specified, Will be Calculated for each Water Year.")
# contentPane.add(startDateField);
#
# jLabelEndDate = JLabel()
# jLabelEndDate.setText("Optional End Date:")
# jLabelEndDate.setToolTipText("Optional User Specified Date Range for Paired Data. If Specified, Will be Calculated for each Water Year.")
# jLabelEndDate.setBounds(420, 293, 120, 20);
# jLabelEndDate.setFont( Font("Tahoma", Font.PLAIN, 12))
# contentPane.add(jLabelEndDate)
#
# endDateField = CalendarField();
# jLabelEndDate.setLabelFor(endDateField);
# endDateField.setMargin(Insets(0, 4, 0, 0));
# endDateField.setBounds(540, 293, 118, 22);
# endDateField.setFont(Font("Tahoma", Font.PLAIN, 12));
# endDateField.setToolTipText("Optional User Specified Date Range for Paired Data. If Specified, Will be Calculated for each Water Year.")
# contentPane.add(endDateField);
def recalcBtnSelect(event):
global swePaths, precipPaths, tempPaths
selectedLocations = list_locations.getSelectedValuesList()
swePaths, precipPaths, tempPaths = Locations.getSelectedLocations(
selectedLocations)
pathsNoDPart, sList, pList, tList = CalcAtiMelt.processPathsLists(
swePaths, precipPaths, tempPaths)
dList = CalcAtiMelt.processPathsDatesList(pathsNoDPart)
aList, mList = CalcAtiMelt.calcATIMelt(selectedLocations, sList, pList,
tList, dList)
# Write Melt-Cum and ATI Locations to DSS.
CalcAtiMelt.writeAtiMelt(selectedLocations, dList, aList, mList)
# Plot Locations if checkbox.selected = True.
if chckbxShowLocationPlot.selected is True:
# print '\nPLOT TEAM ACTIVATE'
PlotAtiMelt.plotAtiMelt(selectedLocations)
# Use optional specified dates if fields are not blank.
# if startDateField.getText() and endDateField.getText() is not None:
# pdStart, pdEnd = ProcessPD.getSpecifiedDates(startDateField, endDateField)
# else:
pdStart = None
pdEnd = None
# Create Paired Data for Selected Locations.
ProcessPD.processPairedData(selectedLocations, dList, mList, aList,
pdStart, pdEnd)
# Populate the UI Paired Data Table.
CalcPD.updatePDTable(dssFile, eventsTable, dm_events)
# Close the DSS File.
dssFile.close()
def plotPDBtnSelect(event):
selected_Events = eventsTable.getSelectedRows()
# Print 'selected_Events: ', selected_Events
# Sorting of the table by selecting the headers is doen by using: eventsTable.setAutoCreateRowSorter(True)
# This sorts the table but does not update the table model.
# To ensure sorting and selecting of resulting paths works properly,
# we must convert our selection using: eventsTable.convertRowIndexToModel(event)
selectedEvents = []
for event in selected_Events:
selectedEvents.append(eventsTable.convertRowIndexToModel(event))
# print 'selectedEvents: ', selectedEvents
PlotPD.plotPD(eventsTable, selectedEvents, dssFile)
dssFile.close()
def calcMeltRateBtnSelect(event):
selected_Events = eventsTable.getSelectedRows()
selectedEvents = []
for event in selected_Events:
selectedEvents.append(eventsTable.convertRowIndexToModel(event))
meltRateList = CalcMeltRate.calcMeltRate(selectedEvents, eventsTable,
meltRateTable, dssFile)
CalcMeltRate.updateTable(meltRateTable, meltRateList, dm_meltRate)
dssFile.close()
locDict, bList = Locations.getPaths(dssFile)
locList = Locations.getList(locDict)
list_locations = JList(locList)
scrollPane_locations.setViewportView(list_locations)
list_locations.setBorder(LineBorder(Color(0, 0, 0), 2, True))
eventsTable = JTable()
scrollPane_events.setViewportView(eventsTable)
scrollPane_events.setBorder(LineBorder(Color(1, 1, 1), 2, True))
eventsTable.setFont(Font("Tahoma", Font.PLAIN, 11))
locationsList, eventsList = Locations.getPairedData(dssFile)
columns = ("Location", "Event")
data = []
for l, e in zip(locationsList, eventsList):
data.append([l, e])
dm_events = DefaultTableModel(data, columns)
eventsTable.setModel(dm_events)
eventsTable.setAutoCreateRowSorter(True)
eventsTable.getColumnModel().getColumn(0).setPreferredWidth(154)
eventsTable.getColumnModel().getColumn(1).setResizable(False)
eventsTable.getColumnModel().getColumn(1).setPreferredWidth(154)
eventsTable.setRowSelectionAllowed(True)
inputPanel = JPanel()
inputPanel.setBorder(EmptyBorder(0, 0, 0, 0))
inputPanel.setBackground(Color(255, 255, 255))
inputPanel.setBounds(270, 11, 410, 27)
contentPane.add(inputPanel)
inputPanel.setLayout(None)
inputPanel.setVisible(True)
lbl_locations = JLabel(
"DSS Locations that have PRECIP-INC, TEMPERATURE-AIR-AVG, and SWE. ")
lbl_locations.setFont(Font("Tahoma", Font.PLAIN, 12))
lbl_locations.setBounds(0, 11, 410, 15)
inputPanel.add(lbl_locations)
btnRecalc = JButton(btnIcon, actionPerformed=recalcBtnSelect)
btnRecalc.setText("Calculate Paired Data")
btnRecalc.setFont(Font("Tahoma", Font.BOLD, 12))
btnRecalc.setForeground(Color.WHITE)
btnRecalc.setBackground(Color.WHITE)
btnRecalc.setBorderPainted(False)
btnRecalc.setContentAreaFilled(False)
btnRecalc.setFocusPainted(False)
btnRecalc.setOpaque(True)
btnRecalc.setVerticalTextPosition(SwingConstants.CENTER)
btnRecalc.setHorizontalTextPosition(SwingConstants.CENTER)
btnRecalc.setBounds(382, 293, 165, 54)
contentPane.add(btnRecalc)
leftPanel = JPanel()
leftPanel.setBackground(Color.DARK_GRAY)
leftPanel.setBounds(0, 0, 250, 780)
contentPane.add(leftPanel)
leftPanel.setLayout(None)
lbl_castle = JLabel("")
lbl_castle.setBounds(110, 678, 40, 25)
leftPanel.add(lbl_castle)
try:
i_corps = ImageIO.read(File(img_dir + "/CorpsCastle.png"))
except:
i_corps = ImageIO.read(File(img_dir2 + "/CorpsCastle.png"))
corpsCastle = i_corps.getScaledInstance(lbl_castle.getWidth(),
lbl_castle.getHeight(),
Image.SCALE_SMOOTH)
lbl_castle.setVerticalAlignment(SwingConstants.TOP)
lbl_castle.setIcon(ImageIcon(corpsCastle))
lbl_logo = JLabel("")
lbl_logo.setBounds(18, 294, 218, 148)
leftPanel.add(lbl_logo)
try:
snowLogo = ImageIO.read(File(img_dir + "/melted-snowman.png"))
except:
snowLogo = ImageIO.read(File(img_dir2 + "/melted-snowman.png"))
dssLogo = snowLogo.getScaledInstance(lbl_logo.getWidth(),
lbl_logo.getHeight(),
Image.SCALE_SMOOTH)
lbl_logo.setVerticalAlignment(SwingConstants.TOP)
lbl_logo.setIcon(ImageIcon(dssLogo))
lbl_logo2 = JLabel("")
lbl_logo2.setBounds(18, 11, 218, 148)
leftPanel.add(lbl_logo2)
try:
snowPacLogo = ImageIO.read(File(img_dir + "/SnowPac.png"))
except:
snowPacLogo = ImageIO.read(File(img_dir2 + "/SnowPac.png"))
snowPac = snowPacLogo.getScaledInstance(lbl_logo2.getWidth(),
lbl_logo2.getHeight(),
Image.SCALE_SMOOTH)
lbl_logo2.setVerticalAlignment(SwingConstants.TOP)
lbl_logo2.setIcon(ImageIcon(snowPac))
# lbl_close = JLabel("X")
#
# class CloseClickListener(MouseAdapter):
# # @Override
# def mouseEntered(self):
# lbl_close.setBorder(LineBorder.createGrayLineBorder())
# # @Override
# def mouseExited(self):
# lbl_close.setBorder(None)
# # @Override
# def mouseClicked(self):
# lbl_close.setBorder(BorderFactory.createLineBorder(Color.red));
# sys.exit();
# cL = CloseClickListener()
# lbl_close.addMouseListener(cL)
#
# lbl_close.setHorizontalAlignment(SwingConstants.CENTER);
# lbl_close.setForeground(Color(241, 57, 83));
# lbl_close.setFont(Font("Tahoma", Font.PLAIN, 18));
# lbl_close.setBounds(1071, 0, 37, 27);
# contentPane.add(lbl_close);
lblPxf = JLabel("Base Temperature (F):")
lblPxf.setToolTipText("The temperature at which melt will occur.")
lblPxf.setFont(Font("Tahoma", Font.PLAIN, 12))
lblPxf.setBounds(400, 263, 132, 15)
contentPane.add(lblPxf)
textField_8 = JTextField()
textField_8.setFont(Font("Tahoma", Font.PLAIN, 12))
textField_8.setToolTipText("The temperature at which melt will occur.")
textField_8.setText("32.0")
textField_8.setBounds(548, 263, 40, 20)
contentPane.add(textField_8)
textField_8.setColumns(10)
chckbxShowLocationPlot = JCheckBox("Plot Locations")
chckbxShowLocationPlot.setToolTipText(
"Will plot the Temp, Precip, SWE, ATI, and Melt for each selected location."
)
chckbxShowLocationPlot.setSelected(True)
chckbxShowLocationPlot.setBackground(Color.WHITE)
chckbxShowLocationPlot.setFont(Font("Tahoma", Font.PLAIN, 12))
chckbxShowLocationPlot.setBounds(547, 310, 120, 23)
contentPane.add(chckbxShowLocationPlot)
lblEvents = JLabel("Paired Data")
lblEvents.setBounds(270, 346, 72, 15)
contentPane.add(lblEvents)
lblEvents.setFont(Font("Tahoma", Font.PLAIN, 12))
# lblAtiThreshold = JLabel("ATI Threshold:");
# lblAtiThreshold.setToolTipText("Some Melt Events are small & can be ignored. The ATI Threshold is a value that must be reached for the event to be listed.");
# lblAtiThreshold.setFont(Font("Tahoma", Font.PLAIN, 12));
# lblAtiThreshold.setBounds(500, 610, 82, 15);
# contentPane.add(lblAtiThreshold);
# textField_9 = JTextField();
# textField_9.setFont(Font("Tahoma", Font.PLAIN, 12));
# textField_9.setText("0.0");
# textField_9.setToolTipText("Some Melt Events are small & can be ignored. The ATI Threshold is a value that must be reached for the event to be listed.");
# textField_9.setColumns(10);
# textField_9.setBounds(600, 608, 60, 20);
# contentPane.add(textField_9);
btnPlot = JButton(btnIcon, actionPerformed=plotPDBtnSelect)
btnPlot.setText("Plot Paired Data")
btnPlot.setFont(Font("Tahoma", Font.BOLD, 12))
btnPlot.setForeground(Color.WHITE)
btnPlot.setBackground(Color.WHITE)
btnPlot.setBorderPainted(False)
btnPlot.setContentAreaFilled(False)
btnPlot.setFocusPainted(False)
btnPlot.setOpaque(False)
btnPlot.setVerticalTextPosition(SwingConstants.CENTER)
btnPlot.setHorizontalTextPosition(SwingConstants.CENTER)
btnPlot.setBounds(385, 657, 163, 54)
contentPane.add(btnPlot)
lblAtimeltrateTable = JLabel("ATI-Meltrate Table")
lblAtimeltrateTable.setFont(Font("Tahoma", Font.PLAIN, 12))
lblAtimeltrateTable.setBounds(708, 10, 410, 15)
contentPane.add(lblAtimeltrateTable)
lblAtimeltrateTable2 = JLabel(
"The first ATI value should be 0. ATI values must be ascending.")
lblAtimeltrateTable2.setFont(Font("Tahoma", Font.PLAIN, 11))
lblAtimeltrateTable2.setBounds(708, 30, 410, 15)
contentPane.add(lblAtimeltrateTable2)
btnCalculateMeltrate = JButton(btnIcon,
actionPerformed=calcMeltRateBtnSelect)
btnCalculateMeltrate.setText("Calculate Meltrate")
btnCalculateMeltrate.setFont(Font("Tahoma", Font.BOLD, 12))
btnCalculateMeltrate.setToolTipText(
"Calculate Meltrate for ATI values in the ATI-Meltrate Table. Calculation will performed on the Paired Data Records Selected in the Paired Data Table."
)
btnCalculateMeltrate.setForeground(Color.WHITE)
btnCalculateMeltrate.setBackground(Color.WHITE)
btnCalculateMeltrate.setBorderPainted(False)
btnCalculateMeltrate.setContentAreaFilled(False)
btnCalculateMeltrate.setFocusPainted(False)
btnCalculateMeltrate.setOpaque(False)
btnCalculateMeltrate.setVerticalTextPosition(SwingConstants.CENTER)
btnCalculateMeltrate.setHorizontalTextPosition(SwingConstants.CENTER)
btnCalculateMeltrate.setBounds(792, 657, 163, 54)
contentPane.add(btnCalculateMeltrate)
frame.setVisible(True)
dssFile.close()
def initComponents(self):
self.apps_checkboxes_list = []
self.devices_checkboxes_list = []
self.selected_apps = []
self.selected_devices = []
self.setLayout(BoxLayout(self, BoxLayout.PAGE_AXIS))
self.setPreferredSize(
Dimension(543, 172)
) #Max 544x173 https://www.sleuthkit.org/autopsy/docs/api-docs/3.1/interfaceorg_1_1sleuthkit_1_1autopsy_1_1corecomponentinterfaces_1_1_data_source_processor.html#a068919818c017ee953180cc79cc68c80
# info menu
self.p_info = SettingsUtils.createPanel()
self.p_info.setPreferredSize(Dimension(543, 172))
self.d_method = SettingsUtils.createPanel(pbottom=15)
self.label = JLabel(
'Press "Find Devices" to search for devices to extract information.'
)
self.label.setBorder(EmptyBorder(0, 0, 5, 0))
self.d_method.add(self.label)
self.label = JLabel('It will generate a file set per device.')
self.label.setBorder(EmptyBorder(0, 0, 10, 0))
self.d_method.add(self.label)
self.label = JLabel(
'This extract method requires ADB enabled on the device and may require root privilege for some paths.'
)
self.label.setFont(self.label.getFont().deriveFont(Font.BOLD, 11))
self.label.setBorder(EmptyBorder(0, 0, 10, 0))
self.d_method.add(self.label)
self.search_devices = JButton('Find Devices',
actionPerformed=self.findDevices)
self.d_method.add(self.search_devices)
self.p_method = SettingsUtils.createPanel(ptop=15)
self.sp2 = SettingsUtils.createSeparators(0)
self.p_info.add(self.sp2, BorderLayout.SOUTH)
self.p_method.add(JLabel("Extract user data from:"))
self.p_apps = SettingsUtils.createPanel(True, pbottom=10)
self.p_devices = SettingsUtils.createPanel(True)
self.choose_device = JLabel("Choose device:")
self.choose_device.setVisible(False)
self.appsBlock()
self.add(self.d_method)
self.add(JSeparator())
self.add(self.p_method)
self.add(self.p_apps)
self.add(self.choose_device)
self.add(self.p_devices)
self.add(self.p_info)
self.findDevices("")
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, scannerInstance):
self.scannerInstance = scannerInstance
JSplitPane.__init__(self, JSplitPane.VERTICAL_SPLIT)
# super(MyPanel, self).__init__(self, JSplitPane.VERTICAL_SPLIT)
# self.setSize(640, 460)
self.setBorder(EmptyBorder(20, 20, 20, 20))
self.topPanel = JPanel(BorderLayout(10, 10))
self.topPanel.setBorder(EmptyBorder(0, 0, 10, 0))
# self.topPanel.setBackground(Color.blue)
self.bottomPanel = JPanel()
self.bottomPanel.setBorder(EmptyBorder(10, 0, 0, 0))
# self.bottomPanel.setBackground(Color.yellow)
self.bottomPanel.setPreferredSize(Dimension(580, 40))
# plain
self.plainPanel = JPanel(BorderLayout(10, 10))
self.plainTextPane = JTextArea()
self.plainTextPane.setLineWrap(True)
self.plainTextPane.setEditable(True)
self.plainScrollPane = JScrollPane(self.plainTextPane)
self.plainScrollPane.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_NEVER)
self.plainPanel.add(JLabel("PLAIN:", SwingConstants.CENTER), BorderLayout.PAGE_START)
self.plainPanel.add(self.plainScrollPane, BorderLayout.CENTER)
self.topPanel.add(self.plainPanel, BorderLayout.LINE_START)
# button
self.btnsPanel = JPanel(GridBagLayout())
gbc = GridBagConstraints()
gbc.fill = GridBagConstraints.HORIZONTAL
gbc.gridx = 0
gbc.gridy = 0
self.btnsPanel.add(JButton("=Encrypt=>", actionPerformed=self.encrypt), gbc)
gbc.fill = GridBagConstraints.HORIZONTAL
gbc.gridx = 0
gbc.gridy = 1
self.btnsPanel.add(JButton("<=Decrypt=", actionPerformed=self.decrypt), gbc)
gbc.fill = GridBagConstraints.HORIZONTAL
gbc.gridx = 0
gbc.gridy = 2
gbc.gridheight = 2
gbc.ipadx = 10
self.btnsPanel.add(JButton("SIGN", actionPerformed=self.sign), gbc)
# b_enc.setPreferredSize(Dimension(30, 20))
# b_dec.setPreferredSize(Dimension(30, 20))
self.topPanel.add(self.btnsPanel, BorderLayout.CENTER)
# cipher
self.cipherPanel = JPanel(BorderLayout(10, 10))
self.cipherTextPane = JTextArea()
self.cipherTextPane.setLineWrap(True)
self.cipherTextPane.setEditable(True)
self.cipherScrollPane = JScrollPane(self.cipherTextPane)
self.cipherScrollPane.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_NEVER)
self.cipherPanel.add(JLabel("CIPHER:", SwingConstants.CENTER), BorderLayout.PAGE_START)
self.cipherPanel.add(self.cipherScrollPane, BorderLayout.CENTER)
self.topPanel.add(self.cipherPanel, BorderLayout.LINE_END)
self.signPanel = JPanel(FlowLayout(FlowLayout.LEADING, 10, 10))
self.signPanel.add(JLabel("SIGN:", SwingConstants.LEFT), BorderLayout.LINE_START)
self.signField = JTextField(50)
self.signPanel.add(self.signField)
self.bottomPanel.add(self.signPanel)
self.plainPanel.setPreferredSize(Dimension(260, 400))
self.btnsPanel.setPreferredSize(Dimension(80, 400))
self.cipherPanel.setPreferredSize(Dimension(260, 400))
self.setTopComponent(self.topPanel)
self.setBottomComponent(self.bottomPanel)
def getMainComponent(self):
self._mainPanel = JPanel()
#self._mainPanel.setLayout(BoxLayout(self._mainPanel, BoxLayout.Y_AXIS))
self._mainPanel.setLayout(BorderLayout())
self._titlePanel = JPanel(GridLayout(0, 1))
self._titlePanel.setBorder(EmptyBorder(10, 20, 10, 10))
self._titleText = JLabel("Mode configuration")
self._titleText.setForeground(COLOR_BURP_TITLE_ORANGE)
self._titleText.setFont(self._titleText.getFont().deriveFont(16.0))
self._titleSubtitleText = JTextArea("Select a mode that works best for the type of exploit you are dealing with.")
self._titleSubtitleText.setEditable(False)
self._titleSubtitleText.setLineWrap(True)
self._titleSubtitleText.setWrapStyleWord(True)
self._titleSubtitleText.setHighlighter(None)
self._titleSubtitleText.setBorder(None)
#Local File Inclusion (LFI): Explore files
#Command injection (visual): Shell
#Command injection (blind): Shell
modes = ['Select mode...', 'Local File Inclusion (LFI) - Discover files', 'OS Command injection (visual) - Shell']
self._cboModes = JComboBox(modes)
self._cboModes.addActionListener(self.cboModesChanged())
self._titlePanel.add(self._titleText)
self._titlePanel.add(self._titleSubtitleText)
self._titlePanel.add(self._cboModes)
self._mainPanel.add(self._titlePanel, BorderLayout.NORTH)
self._switchPanel = JPanel()
self._switchPanel.setLayout(BoxLayout(self._switchPanel, BoxLayout.Y_AXIS))
self._switchPanel.setBorder(EmptyBorder(5, 30, 10, 10))
self._outputCompleteResponseSwitch = JCheckBox("Use full response (not only the response body)")
self._outputCompleteResponseSwitch.setSelected(True)
self._outputCompleteResponseSwitch.addItemListener(self.OutputCompleteResponseSwitchListener())
self._switchPanel.add(self._outputCompleteResponseSwitch)
self._urlEncodeSwitch = JCheckBox("Use url encode")
self._urlEncodeSwitch.setSelected(True)
self._urlEncodeSwitch.addItemListener(self.UrlEncodeSwitchListener())
self._switchPanel.add(self._urlEncodeSwitch)
self._outputIsolatorSwitch = JCheckBox("Use output isolator")
self._outputIsolatorSwitch.setSelected(False)
self._outputIsolatorSwitch.addItemListener(self.OutputIsolatorSwitchListener())
self._switchPanel.add(self._outputIsolatorSwitch)
self._removeSpacesSwitch = JCheckBox("Remove leading and trailing spaces")
self._removeSpacesSwitch.setSelected(False)
self._removeSpacesSwitch.addItemListener(self.SpacesSwitch())
self._switchPanel.add(self._removeSpacesSwitch)
self._removeHtmlTagsSwitch = JCheckBox("Remove html-tags (regex '<.*?>')")
self._removeHtmlTagsSwitch.setSelected(False)
self._removeHtmlTagsSwitch.addItemListener(self.HtmlTagsSwitch())
self._switchPanel.add(self._removeHtmlTagsSwitch)
self._tabCompletionSwitch = JCheckBox("Use tab-completion")
self._tabCompletionSwitch.setSelected(False)
self._tabCompletionSwitch.addItemListener(self.TabCompletionSwitchListener())
self._switchPanel.add(self._tabCompletionSwitch)
self._virtualPersistenceSwitch = JCheckBox("Use virtual persistence")
self._virtualPersistenceSwitch.setSelected(False)
self._virtualPersistenceSwitch.addItemListener(self.VirtualPersistenceSwitchListener())
self._switchPanel.add(self._virtualPersistenceSwitch)
self._wafSwitch = JCheckBox("WAF: Prepend each non-whitespace character (regex '\\w') with '\\'")
self._wafSwitch.setSelected(False)
self._wafSwitch.addItemListener(self.WafSwitch())
self._switchPanel.add(self._wafSwitch)
self._mainPanel.add(self._switchPanel, BorderLayout.CENTER)
return self._mainPanel
def registerExtenderCallbacks(self, callbacks):
# for error handling
sys.stdout = callbacks.getStdout()
sys.stderr = callbacks.getStderr()
self._callbacks = callbacks
self._helpers = callbacks.getHelpers()
jsonFilter = FileNameExtensionFilter("JSON files", ['json'])
basePath = self._callbacks.loadExtensionSetting(SETTING_LAST_PATH)
# print("got last path {}".format(basePath))
self._collectionChooser = swing.JFileChooser(basePath)
self._collectionChooser.setFileFilter(jsonFilter)
self._environmentChooser = swing.JFileChooser(basePath)
self._environmentChooser.setFileFilter(jsonFilter)
# ### Top "buttons" pane
controlPane = swing.JPanel()
controlPane.setBorder(EmptyBorder(10, 20, 0, 10))
controlPane.setLayout(
swing.BoxLayout(controlPane, swing.BoxLayout.PAGE_AXIS))
controlPane.setAlignmentX(swing.Box.LEFT_ALIGNMENT)
box1 = swing.Box.createHorizontalBox()
box1.setAlignmentX(swing.Box.LEFT_ALIGNMENT)
box1.add(
swing.JButton('Load Collection',
actionPerformed=self.loadCollection))
self._collectionLabel = swing.JLabel("Choose a collection file")
box1.add(self._collectionLabel)
controlPane.add(box1)
box2 = swing.Box.createHorizontalBox()
box2.setAlignmentX(swing.Box.LEFT_ALIGNMENT)
box2.add(
swing.JButton('Load Environment',
actionPerformed=self.loadEnvironment))
self._environmentLabel = swing.JLabel("Choose an environment file")
box2.add(self._environmentLabel)
controlPane.add(box2)
# ### end Top "controls" pane
# ### instructions
instructionsPane = swing.JPanel(BorderLayout())
instructions = swing.JLabel()
instructions.setText("""<html><body>
<h3>Usage:</h3>
<ol>
<li>Select the Collection Postman JSON file. This should extract all discovered environment variables.</li>
<li>(Optional) Select an Environment Postman JSON file. This can be the same as the Collection file.</li>
<li>Set environment variables below.</li>
<li>Choose 'Create Requests' to create Repeater tabs.</li>
</ol>
</body></html>""")
instructionsPane.add(instructions, BorderLayout.NORTH)
# ### end instructions
# ### environment variables
envTablePane = swing.JPanel(BorderLayout())
envLabel = swing.JLabel("Environment Variables")
envLabel.setBorder(EmptyBorder(5, 5, 5, 5))
envLabel.setFont(
Font(envLabel.getFont().getName(), Font.BOLD,
envLabel.getFont().getSize() + 2))
envTablePane.add(envLabel, BorderLayout.NORTH)
self._envTable = swing.JTable(DefaultTableModel([], self._envCols))
self._envTable.setAutoCreateRowSorter(True)
self._envTable.getTableHeader().setReorderingAllowed(False)
tableMenu = swing.JPopupMenu()
tableMenu.add(swing.JMenuItem("Add New", actionPerformed=self._addEnv))
tableMenu.add(
swing.JMenuItem("Clear All", actionPerformed=self._clearEnv))
deleteMenuItem = swing.JMenuItem("Delete Row",
actionPerformed=self._deleteEnv)
deleteMenuItem.setEnabled(False)
tableMenu.add(deleteMenuItem)
self._envTable.setComponentPopupMenu(tableMenu)
listener = self._envTableListener(self)
self._envTable.addMouseListener(listener)
renderer = self._envTableRenderer()
self._envTable.setDefaultRenderer(Class.forName('java.lang.Object'),
renderer)
envTablePaneMenu = swing.JPopupMenu()
envTablePaneMenu.add(
swing.JMenuItem("Add New", actionPerformed=self._addEnv))
envTablePaneMenu.add(
swing.JMenuItem("Clear All", actionPerformed=self._clearEnv))
scrl = swing.JScrollPane(self._envTable)
scrl.setComponentPopupMenu(envTablePaneMenu)
envTablePane.add(scrl)
# ### end environment variables
# ### requests
reqTablePane = swing.JPanel(BorderLayout())
reqLabel = swing.JLabel("Requests")
reqLabel.setBorder(EmptyBorder(5, 5, 5, 5))
reqLabel.setFont(envLabel.getFont())
reqTablePane.add(reqLabel, BorderLayout.NORTH)
self._reqTable = self._reqTableClass(
DefaultTableModel([], self._reqCols))
self._reqTable.setAutoCreateRowSorter(True)
self._reqTable.getTableHeader().setReorderingAllowed(False)
self._reqTable.setAutoResizeMode(swing.JTable.AUTO_RESIZE_LAST_COLUMN)
self._reqTable.getTableHeader().setReorderingAllowed(False)
self._reqTable.getColumnModel().getColumn(0).setMaxWidth(150)
self._reqTable.getColumnModel().getColumn(0).setMinWidth(150)
self._reqTable.getColumnModel().getColumn(2).setMaxWidth(150)
self._reqTable.getColumnModel().getColumn(2).setMinWidth(150)
scrl2 = swing.JScrollPane(self._reqTable)
reqTablePane.add(scrl2)
# ### end requests
# ### Logs
logPane = swing.JPanel(BorderLayout())
buttonBox = swing.JPanel(FlowLayout(FlowLayout.LEFT, 20, 0))
requestButtonBox = swing.Box.createHorizontalBox()
self._selectButtons = [
swing.JButton('Select All', actionPerformed=self.selectAll),
swing.JButton('Select None', actionPerformed=self.selectNone),
swing.JButton('Invert Selection',
actionPerformed=self.selectInvert)
]
for btn in self._selectButtons:
requestButtonBox.add(btn)
btn.setEnabled(False)
buttonBox.add(requestButtonBox)
self._createRequestsButton = swing.JButton(
'Create Requests', actionPerformed=self.createRequests)
self._createRequestsButton.setEnabled(False)
requestButtonBox.add(self._createRequestsButton)
buttonBox.add(self._createRequestsButton)
self._logButton = swing.JButton('Clear Log',
actionPerformed=self.clearLog)
self._logButton.setEnabled(False)
buttonBox.add(self._logButton)
logPane.add(buttonBox, BorderLayout.NORTH)
self._log = swing.JTextPane()
self._log.setEditable(False)
self._log.setFont(Font("monospaced", Font.PLAIN, 12))
logPane.add(swing.JScrollPane(self._log))
# ### end Logs
# ### add panels
self._topControlsPane = swing.JSplitPane(
swing.JSplitPane.HORIZONTAL_SPLIT, controlPane, instructionsPane)
p1 = swing.JSplitPane(swing.JSplitPane.VERTICAL_SPLIT,
self._topControlsPane, envTablePane)
p2 = swing.JSplitPane(swing.JSplitPane.VERTICAL_SPLIT, p1,
reqTablePane)
p2.setResizeWeight(0.5)
self._panel = swing.JSplitPane(swing.JSplitPane.VERTICAL_SPLIT, p2,
logPane)
self._panel.setResizeWeight(0.6)
# ### end add panels
callbacks.setExtensionName(NAME)
callbacks.addSuiteTab(self)
def registerExtenderCallbacks(self, callbacks):
print "Loading..."
self._callbacks = callbacks
self._callbacks.setExtensionName('Burp SSL Scanner')
# self._callbacks.registerScannerCheck(self)
# self._callbacks.registerExtensionStateListener(self)
self._helpers = callbacks.getHelpers()
# initialize the main scanning event and thread
self.scanningEvent = Event()
self.scannerThread = None
self.targetURL = 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 scanning', actionPerformed=self.startScan)
self.setupPanel.add(self.toggleButton)
if 'Professional' in callbacks.getBurpVersion()[0] :
self.addToSitemapCheckbox = JCheckBox('Add to sitemap', True)
else :
self.addToSitemapCheckbox = JCheckBox('Add to sitemap (requires Professional version)', False)
self.addToSitemapCheckbox.setEnabled(False)
self.setupPanel.add(self.addToSitemapCheckbox)
self.scanSiteMapHostCheckbox = JCheckBox('Scan sitemap hosts', True)
self.setupPanel.add(self.scanSiteMapHostCheckbox)
self._topPanel.add(self.setupPanel, BorderLayout.PAGE_START)
# Status bar
self.scanStatusPanel = JPanel(FlowLayout(FlowLayout.LEADING, 10, 10))
self.scanStatusPanel.add(JLabel("Status: ", SwingConstants.LEFT))
self.scanStatusLabel = JLabel("Ready to scan", SwingConstants.LEFT)
self.scanStatusPanel.add(self.scanStatusLabel)
self._topPanel.add(self.scanStatusPanel, BorderLayout.LINE_START)
self._splitpane.setTopComponent(self._topPanel)
# bottom panel
self._bottomPanel = JPanel(BorderLayout(10, 10))
self._bottomPanel.setBorder(EmptyBorder(10, 0, 0, 0))
self.initialText = ('<h1 style="color: red;">Burp SSL Scanner<br />'
'Please note that TLS1.3 is still not supported by this extension.</h1>')
self.currentText = self.initialText
self.textPane = JTextPane()
self.textScrollPane = JScrollPane(self.textPane)
self.textPane.setContentType("text/html")
self.textPane.setText(self.currentText)
self.textPane.setEditable(False)
self._bottomPanel.add(self.textScrollPane, BorderLayout.CENTER)
self.savePanel = JPanel(FlowLayout(FlowLayout.LEADING, 10, 10))
self.saveButton = JButton('Save to file', actionPerformed=self.saveToFile)
self.saveButton.setEnabled(False)
self.savePanel.add(self.saveButton)
self.clearScannedHostButton = JButton('Clear scanned host', actionPerformed=self.clearScannedHost)
self.savePanel.add(self.clearScannedHostButton)
self.savePanel.add(JLabel("Clear hosts that were scanned by active scan to enable rescanning", SwingConstants.LEFT))
self._bottomPanel.add(self.savePanel, BorderLayout.PAGE_END)
self._splitpane.setBottomComponent(self._bottomPanel)
callbacks.customizeUiComponent(self._splitpane)
callbacks.addSuiteTab(self)
print "SSL Scanner tab loaded"
self.scannerMenu = ScannerMenu(self)
callbacks.registerContextMenuFactory(self.scannerMenu)
print "SSL Scanner custom menu loaded"
self.scannerCheck = ScannerCheck(self, self.scanSiteMapHostCheckbox.isSelected)
callbacks.registerScannerCheck(self.scannerCheck)
print "SSL Scanner check registered"
projectConfig = json.loads(self._callbacks.saveConfigAsJson())
scanAccuracy = projectConfig['scanner']['active_scanning_optimization']['scan_accuracy']
scanSpeed = projectConfig['scanner']['active_scanning_optimization']['scan_speed']
print(scanAccuracy, scanSpeed)
self.scannedHost = []
print 'SSL Scanner loaded'
def drawPluginUI(self):
# Create the plugin user interface
self.pluginTab = JPanel()
self.uiTitle = JLabel('UPnP BHunter Load, Aim and Fire Console')
self.uiTitle.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiTitle.setForeground(Color(250, 100, 0))
self.uiPanelA = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self.uiPanelA.setMaximumSize(Dimension(2500, 1000))
self.uiPanelA.setDividerSize(2)
self.uiPanelB = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self.uiPanelB.setDividerSize(2)
self.uiPanelA.setBottomComponent(self.uiPanelB)
self.uiPanelA.setBorder(BorderFactory.createLineBorder(Color.gray))
# Create and configure labels and text fields
self.labeltitle_step1 = JLabel("[1st STEP] Discover UPnP Locations")
self.labeltitle_step1.setFont(Font('Tahoma', Font.BOLD, 14))
self.labeltitle_step2 = JLabel("[2nd STEP] Select an UPnP Service")
self.labeltitle_step2.setFont(Font('Tahoma', Font.BOLD, 14))
self.labeltitle_step3 = JLabel("[3rd STEP] Time to Attack it")
self.labeltitle_step3.setFont(Font('Tahoma', Font.BOLD, 14))
self.labelsubtitle_step1 = JLabel(
"Specify the IP version address in scope and start UPnP discovery")
self.labelsubtitle_step2 = JLabel(
"Select which of the found UPnP services will be probed")
self.labelsubtitle_step3 = JLabel(
"Select how to test the extracted UPnP SOAP requests having input arguments"
)
self.label_step1 = JLabel("Target IP")
self.label_step2 = JLabel("Found UPnp Services")
self.label_step3 = JLabel("Send all the extracted SOAP requests ")
self.labelstatus = JLabel(" Status")
self.labelempty_step1 = JLabel(" ")
self.labelempty_step2 = JLabel(" ")
self.labelupnp = JLabel("UPnP list")
self.labelip = JLabel("IP list")
self.labelLANHOST = JLabel(
"Send the interesting LANHostConfigManagement SOAP requests ")
self.labelWANCONNECTION = JLabel(
"Send the interesting WANIP/PPPConnection SOAP requests ")
self.labelSOAPnum = JLabel("0")
self.labelLANHOSTnum = JLabel("0")
self.labelWANCONNECTIONnum = JLabel("0")
self.labelNoneServiceFound = JLabel(" ")
self.labelNoneServiceFound.setFont(Font('Tahoma', Font.BOLD, 12))
self.labelNoneServiceFound.setForeground(Color.red)
# Create combobox for IP version selection
self.ip_versions = ["IPv4", "IPv6"]
self.combo_ipversion = JComboBox(self.ip_versions)
self.combo_ipversion.setSelectedIndex(0)
self.combo_ipversion.setEnabled(True)
# Create and configure progress bar
self.progressbar = JProgressBar(0, 100)
self.progressbar.setString("Ready")
self.progressbar.setStringPainted(True)
# Create and configure buttons
self.startbutton = JButton("Start Discovery",
actionPerformed=self.startHunting)
self.clearbutton = JButton("Clear All", actionPerformed=self.clearAll)
self.confirmbutton = JButton("Confirm Selection",
actionPerformed=self.selectUPnPService)
self.intruderbutton = JButton("to Intruder",
actionPerformed=self.sendUPnPToIntruder)
self.LANrepeaterbutton = JButton(
"to Repeater", actionPerformed=self.sendLANUPnPToRepeater)
self.WANrepeaterbutton = JButton(
"to Repeater", actionPerformed=self.sendWANUPnPToRepeater)
self.confirmbutton.setEnabled(False)
self.intruderbutton.setEnabled(False)
self.LANrepeaterbutton.setEnabled(False)
self.WANrepeaterbutton.setEnabled(False)
# Create the combo box, select item at index 0 (first item in list)
self.upnpservices = [" "]
self.upnpcombo_services = JComboBox(self.upnpservices)
self.upnpcombo_services.setSelectedIndex(0)
self.upnpcombo_services.setEnabled(False)
# Class neeeded to handle the combobox in second step panel
class ComboboxListener(ActionListener):
def __init__(self, upnpcombo_targets, upnpcombo_services,
scope_dict):
self.upnpcombo_targets = upnpcombo_targets
self.upnpcombo_services = upnpcombo_services
self.scope_dict = scope_dict
def actionPerformed(self, event):
# Update the location url combobox depending on the IP combobox
selected_target = self.upnpcombo_targets.getSelectedItem()
if self.scope_dict and selected_target:
self.upnpcombo_services.removeAllItems()
for scope_url in self.scope_dict[selected_target]:
self.upnpcombo_services.addItem(scope_url)
self.upnpcombo_services.setSelectedIndex(0)
# Create the combo box, select item at index 0 (first item in list)
self.upnptargets = [" "]
self.upnpcombo_targets = JComboBox(self.upnptargets)
self.upnpcombo_targets.setSelectedIndex(0)
self.upnpcombo_targets.setEnabled(False)
self.upnpcombo_targets.addActionListener(
ComboboxListener(self.upnpcombo_targets, self.upnpcombo_services,
self.scope_dict))
# Configuring first step panel
self.panel_step1 = JPanel()
self.panel_step1.setPreferredSize(Dimension(2250, 100))
self.panel_step1.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step1.setLayout(BorderLayout(15, 15))
self.titlepanel_step1 = JPanel()
self.titlepanel_step1.setLayout(BorderLayout())
self.titlepanel_step1.add(self.labeltitle_step1, BorderLayout.NORTH)
self.titlepanel_step1.add(self.labelsubtitle_step1)
self.targetpanel_step1 = JPanel()
self.targetpanel_step1.add(self.label_step1)
self.targetpanel_step1.add(self.combo_ipversion)
self.targetpanel_step1.add(self.startbutton)
self.targetpanel_step1.add(self.clearbutton)
self.targetpanel_step1.add(self.labelstatus)
self.targetpanel_step1.add(self.progressbar)
self.emptypanel_step1 = JPanel()
self.emptypanel_step1.setLayout(BorderLayout())
self.emptypanel_step1.add(self.labelempty_step1, BorderLayout.WEST)
# Assembling first step panel components
self.panel_step1.add(self.titlepanel_step1, BorderLayout.NORTH)
self.panel_step1.add(self.targetpanel_step1, BorderLayout.WEST)
self.panel_step1.add(self.emptypanel_step1, BorderLayout.SOUTH)
self.uiPanelA.setTopComponent(self.panel_step1)
# Configure second step panel
self.panel_step2 = JPanel()
self.panel_step2.setPreferredSize(Dimension(2250, 100))
self.panel_step2.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step2.setLayout(BorderLayout(15, 15))
self.titlepanel_step2 = JPanel()
self.titlepanel_step2.setLayout(BorderLayout())
self.titlepanel_step2.add(self.labeltitle_step2, BorderLayout.NORTH)
self.titlepanel_step2.add(self.labelsubtitle_step2)
self.selectpanel_step2 = JPanel()
self.selectpanel_step2.add(self.labelip)
self.selectpanel_step2.add(self.upnpcombo_targets)
self.selectpanel_step2.add(self.labelupnp)
self.selectpanel_step2.add(self.upnpcombo_services)
self.selectpanel_step2.add(self.confirmbutton)
self.emptypanel_step2 = JPanel()
self.emptypanel_step2.setLayout(BorderLayout())
self.emptypanel_step2.add(self.labelempty_step2, BorderLayout.WEST)
self.emptypanel_step2.add(self.labelNoneServiceFound)
# Assembling second step panel components
self.panel_step2.add(self.titlepanel_step2, BorderLayout.NORTH)
self.panel_step2.add(self.selectpanel_step2, BorderLayout.WEST)
self.panel_step2.add(self.emptypanel_step2, BorderLayout.SOUTH)
self.uiPanelB.setTopComponent(self.panel_step2)
# Configuring third step panel
self.panel_step3 = JPanel()
self.panel_step3.setPreferredSize(Dimension(2250, 100))
self.panel_step3.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step3.setLayout(BorderLayout(15, 15))
self.titlepanel_step3 = JPanel()
self.titlepanel_step3.setLayout(BorderLayout())
self.titlepanel_step3.add(self.labeltitle_step3, BorderLayout.NORTH)
self.titlepanel_step3.add(self.labelsubtitle_step3)
self.underpanel_step3 = JPanel()
underlayout = GroupLayout(self.underpanel_step3)
self.underpanel_step3.setLayout(underlayout)
underlayout.setAutoCreateGaps(True)
underlayout.setAutoCreateContainerGaps(True)
left2right = underlayout.createSequentialGroup()
firstcolumn = underlayout.createParallelGroup()
firstcolumn.addComponent(self.label_step3, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
firstcolumn.addComponent(self.labelLANHOST, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
firstcolumn.addComponent(self.labelWANCONNECTION,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
secondcolumn = underlayout.createParallelGroup()
secondcolumn.addComponent(self.labelSOAPnum)
secondcolumn.addComponent(self.labelLANHOSTnum)
secondcolumn.addComponent(self.labelWANCONNECTIONnum)
thirdcolumn = underlayout.createParallelGroup()
thirdcolumn.addComponent(self.intruderbutton)
thirdcolumn.addComponent(self.LANrepeaterbutton)
thirdcolumn.addComponent(self.WANrepeaterbutton)
left2right.addGroup(firstcolumn)
left2right.addGroup(secondcolumn)
left2right.addGroup(thirdcolumn)
top2bottom = underlayout.createSequentialGroup()
firstrow = underlayout.createParallelGroup()
firstrow.addComponent(self.label_step3, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
firstrow.addComponent(self.labelSOAPnum)
firstrow.addComponent(self.intruderbutton)
secondrow = underlayout.createParallelGroup()
secondrow.addComponent(self.labelLANHOST, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
secondrow.addComponent(self.labelLANHOSTnum)
secondrow.addComponent(self.LANrepeaterbutton)
thirdrow = underlayout.createParallelGroup()
thirdrow.addComponent(self.labelWANCONNECTION,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
thirdrow.addComponent(self.labelWANCONNECTIONnum)
thirdrow.addComponent(self.WANrepeaterbutton)
top2bottom.addGroup(firstrow)
top2bottom.addGroup(secondrow)
top2bottom.addGroup(thirdrow)
underlayout.setHorizontalGroup(left2right)
underlayout.setVerticalGroup(top2bottom)
# Assembling thirdd step panel components
self.panel_step3.add(self.titlepanel_step3, BorderLayout.NORTH)
self.panel_step3.add(self.underpanel_step3, BorderLayout.WEST)
self.uiPanelB.setBottomComponent(self.panel_step3)
# Assembling the group of all panels
layout = GroupLayout(self.pluginTab)
self.pluginTab.setLayout(layout)
layout.setHorizontalGroup(
layout.createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
layout.createSequentialGroup().addGap(10, 10, 10).addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.LEADING).addComponent(
self.uiTitle).addGap(15, 15, 15).addComponent(
self.uiPanelA)).addContainerGap(
26, Short.MAX_VALUE)))
layout.setVerticalGroup(
layout.createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
layout.createSequentialGroup().addGap(15, 15, 15).addComponent(
self.uiTitle).addGap(15, 15, 15).addComponent(
self.uiPanelA).addGap(20, 20, 20).addGap(20, 20, 20)))
def drawPluginUI(self):
# Create the plugin user interface
self.pluginTab = JPanel()
self.uiTitle = JLabel('UPnP BHunter Load, Aim and Fire Console')
self.uiTitle.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiTitle.setForeground(Color(250, 100, 0))
self.uiPanelA = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self.uiPanelA.setMaximumSize(Dimension(2500, 1000))
self.uiPanelA.setDividerSize(2)
self.uiPanelB = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self.uiPanelB.setDividerSize(2)
self.uiPanelA.setBottomComponent(self.uiPanelB)
self.uiPanelA.setBorder(BorderFactory.createLineBorder(Color.gray))
# Create and configure labels and text fields
self.labeltitle_step1 = JLabel("[1st STEP] Discover UPnP Locations")
self.labeltitle_step1.setFont(Font('Tahoma', Font.BOLD, 14))
self.labeltitle_step2 = JLabel(
"[2nd STEP] Select a UPnP Service and Action")
self.labeltitle_step2.setFont(Font('Tahoma', Font.BOLD, 14))
self.labeltitle_step3 = JLabel("[3rd STEP] Time to Attack it")
self.labeltitle_step3.setFont(Font('Tahoma', Font.BOLD, 14))
self.labelsubtitle_step1 = JLabel(
"Specify the IP version address in scope and start UPnP discovery")
self.labelsubtitle_step2 = JLabel(
"Select which of the found UPnP services will be probed")
self.labelsubtitle_step3 = JLabel(
"Review and modify the request, then send it to one of the attack tools"
)
self.label_step1 = JLabel("Target IP")
self.label_step2 = JLabel("Found UPnp Services")
self.labelstatus = JLabel(" Status")
self.labelempty_step1 = JLabel(" ")
self.labelempty_step2 = JLabel(" ")
self.labelupnp = JLabel("UPnP list")
self.labelip = JLabel("IP list")
self.labelactions = JLabel("Actions")
self.labelNoneServiceFound = JLabel(" ")
self.labelNoneServiceFound.setFont(Font('Tahoma', Font.BOLD, 12))
self.labelNoneServiceFound.setForeground(Color.red)
# Create combobox for IP version selection
self.ip_versions = ["IPv4", "IPv6"]
self.combo_ipversion = JComboBox(self.ip_versions)
self.combo_ipversion.setSelectedIndex(0)
self.combo_ipversion.setEnabled(True)
# Create and configure progress bar
self.progressbar = JProgressBar(0, 100)
self.progressbar.setString("Ready")
self.progressbar.setStringPainted(True)
# Create and configure buttons
self.startbutton = JButton("Start Discovery",
actionPerformed=self.startHunting)
self.clearbutton = JButton("Clear All", actionPerformed=self.clearAll)
self.intruderbutton = JButton("Send to Intruder",
actionPerformed=self.sendToIntruder)
self.repeaterbutton = JButton("Send to Repeater",
actionPerformed=self.sendToRepeater)
#self.WANrepeaterbutton = JButton("to Repeater", actionPerformed=self.sendWANUPnPToRepeater)
self.textarea_request = JTextArea(18, 90)
self.intruderbutton.setEnabled(False)
self.repeaterbutton.setEnabled(False)
# Class neeeded to handle the target combobox in second step panel
class TargetComboboxListener(ActionListener):
def __init__(self, upnpcombo_targets, upnpcombo_services,
ip_service_dict):
self.upnpcombo_targets = upnpcombo_targets
self.upnpcombo_services = upnpcombo_services
self.ip_service_dict = ip_service_dict
def actionPerformed(self, event):
try:
# Update the location url combobox depending on the IP combobox
selected_target = self.upnpcombo_targets.getSelectedItem()
if self.ip_service_dict and selected_target:
self.upnpcombo_services.removeAllItems()
for service_url in self.ip_service_dict[
selected_target]:
self.upnpcombo_services.addItem(service_url)
self.upnpcombo_services.setSelectedIndex(0)
except BaseException as e:
print("[!] Exception selecting service: \"%s\" ") % e
# Class neeeded to handle the service combobox in second step panel
class ServiceComboboxListener(ActionListener):
def __init__(self, upnpcombo_services, upnpcombo_actions,
services_dict):
self.upnpcombo_services = upnpcombo_services
self.upnpcombo_actions = upnpcombo_actions
self.services = services_dict
def actionPerformed(self, event):
try:
# Update the location url combobox depending on the IP combobox
selected_service = self.upnpcombo_services.getSelectedItem(
)
if self.services and selected_service:
self.upnpcombo_actions.removeAllItems()
actions = self.services[selected_service]
for action in actions:
self.upnpcombo_actions.addItem(action)
self.upnpcombo_actions.setSelectedIndex(0)
except BaseException as e:
print("[!] Exception selecting service: \"%s\" ") % e
# Class neeeded to handle the action combobox in second step panel
class ActionComboboxListener(ActionListener):
def __init__(self, upnpcombo_services, upnpcombo_actions,
textarea_request, services_dict):
self.upnpcombo_services = upnpcombo_services
self.upnpcombo_actions = upnpcombo_actions
self.textarea_request = textarea_request
self.services = services_dict
def actionPerformed(self, event):
try:
# Update the location url combobox depending on the IP combobox
selected_action = self.upnpcombo_actions.getSelectedItem()
selected_service = self.upnpcombo_services.getSelectedItem(
)
if self.services and selected_action:
self.textarea_request.setText(
self.services[selected_service][selected_action])
except BaseException as e:
print("[!] Exception selecting action: \"%s\" ") % e
self.upnpactions = [" "]
self.upnpcombo_actions = JComboBox(self.upnpactions)
self.upnpcombo_actions.setSelectedIndex(0)
self.upnpcombo_actions.setEnabled(False)
# Create the combo box, select item at index 0 (first item in list)
self.upnpservices = [" "]
self.upnpcombo_services = JComboBox(self.upnpservices)
self.upnpcombo_services.setSelectedIndex(0)
self.upnpcombo_services.setEnabled(False)
# Create the combo box, select item at index 0 (first item in list)
self.upnptargets = [" "]
self.upnpcombo_targets = JComboBox(self.upnptargets)
self.upnpcombo_targets.setSelectedIndex(0)
self.upnpcombo_targets.setEnabled(False)
# Set the action listeners for all the comboboxes
self.upnpcombo_targets.addActionListener(
TargetComboboxListener(self.upnpcombo_targets,
self.upnpcombo_services,
self.ip_service_dict))
self.upnpcombo_services.addActionListener(
ServiceComboboxListener(self.upnpcombo_services,
self.upnpcombo_actions,
self.services_dict))
self.upnpcombo_actions.addActionListener(
ActionComboboxListener(self.upnpcombo_services,
self.upnpcombo_actions,
self.textarea_request, self.services_dict))
# Configuring first step panel
self.panel_step1 = JPanel()
self.panel_step1.setPreferredSize(Dimension(2250, 100))
self.panel_step1.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step1.setLayout(BorderLayout(15, 15))
self.titlepanel_step1 = JPanel()
self.titlepanel_step1.setLayout(BorderLayout())
self.titlepanel_step1.add(self.labeltitle_step1, BorderLayout.NORTH)
self.titlepanel_step1.add(self.labelsubtitle_step1)
self.targetpanel_step1 = JPanel()
self.targetpanel_step1.add(self.label_step1)
self.targetpanel_step1.add(self.combo_ipversion)
self.targetpanel_step1.add(self.startbutton)
self.targetpanel_step1.add(self.clearbutton)
self.targetpanel_step1.add(self.labelstatus)
self.targetpanel_step1.add(self.progressbar)
self.emptypanel_step1 = JPanel()
self.emptypanel_step1.setLayout(BorderLayout())
self.emptypanel_step1.add(self.labelempty_step1, BorderLayout.WEST)
# Assembling first step panel components
self.panel_step1.add(self.titlepanel_step1, BorderLayout.NORTH)
self.panel_step1.add(self.targetpanel_step1, BorderLayout.WEST)
self.panel_step1.add(self.emptypanel_step1, BorderLayout.SOUTH)
self.uiPanelA.setTopComponent(self.panel_step1)
# Configure second step panel
self.panel_step2 = JPanel()
self.panel_step2.setPreferredSize(Dimension(2250, 100))
self.panel_step2.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step2.setLayout(BorderLayout(15, 15))
self.titlepanel_step2 = JPanel()
self.titlepanel_step2.setLayout(BorderLayout())
self.titlepanel_step2.add(self.labeltitle_step2, BorderLayout.NORTH)
self.titlepanel_step2.add(self.labelsubtitle_step2)
self.selectpanel_step2 = JPanel()
self.selectpanel_step2.add(self.labelip)
self.selectpanel_step2.add(self.upnpcombo_targets)
self.selectpanel_step2.add(self.labelupnp)
self.selectpanel_step2.add(self.upnpcombo_services)
self.selectpanel_step2.add(self.labelactions)
self.selectpanel_step2.add(self.upnpcombo_actions)
self.emptypanel_step2 = JPanel()
self.emptypanel_step2.setLayout(BorderLayout())
self.emptypanel_step2.add(self.labelempty_step2, BorderLayout.WEST)
self.emptypanel_step2.add(self.labelNoneServiceFound)
# Assembling second step panel components
self.panel_step2.add(self.titlepanel_step2, BorderLayout.NORTH)
self.panel_step2.add(self.selectpanel_step2, BorderLayout.WEST)
self.panel_step2.add(self.emptypanel_step2, BorderLayout.SOUTH)
self.uiPanelB.setTopComponent(self.panel_step2)
# Configuring third step panel
self.panel_step3 = JPanel()
self.panel_step3.setPreferredSize(Dimension(2250, 100))
self.panel_step3.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step3.setLayout(BorderLayout(15, 15))
self.titlepanel_step3 = JPanel()
self.titlepanel_step3.setLayout(BorderLayout())
self.titlepanel_step3.add(self.labeltitle_step3, BorderLayout.NORTH)
self.titlepanel_step3.add(self.labelsubtitle_step3)
self.underpanel_step3 = JPanel()
self.underpanel_step3.setLayout(BorderLayout())
self.underpanel_step3.add((JScrollPane(self.textarea_request)),
BorderLayout.NORTH)
self.actionpanel_step3 = JPanel()
self.actionpanel_step3.add(self.intruderbutton)
self.actionpanel_step3.add(self.repeaterbutton)
self.extrapanel_step3 = JPanel()
self.extrapanel_step3.setLayout(BorderLayout())
self.extrapanel_step3.add(self.actionpanel_step3, BorderLayout.WEST)
# Assembling thirdd step panel components
self.panel_step3.add(self.titlepanel_step3, BorderLayout.NORTH)
self.panel_step3.add(self.underpanel_step3, BorderLayout.WEST)
self.panel_step3.add(self.extrapanel_step3, BorderLayout.SOUTH)
self.uiPanelB.setBottomComponent(self.panel_step3)
# Assembling the group of all panels
layout = GroupLayout(self.pluginTab)
self.pluginTab.setLayout(layout)
layout.setHorizontalGroup(
layout.createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
layout.createSequentialGroup().addGap(10, 10, 10).addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.LEADING).addComponent(
self.uiTitle).addGap(15, 15, 15).addComponent(
self.uiPanelA)).addContainerGap(
26, Short.MAX_VALUE)))
layout.setVerticalGroup(
layout.createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
layout.createSequentialGroup().addGap(15, 15, 15).addComponent(
self.uiTitle).addGap(15, 15, 15).addComponent(
self.uiPanelA).addGap(20, 20, 20).addGap(20, 20, 20)))
def createAndShowGUI(self):
# print "creatAndShowGUI"
from javax.swing import JScrollPane as JScrollPane
import javax.swing.border.CompoundBorder as CompoundBorder
import javax.swing.border.MatteBorder as MatteBorder
import javax.swing.border.EmptyBorder as EmptyBorder
import java.awt.Color as Color
import javax.swing.JFrame as JFrame
from javax.swing import WindowConstants as WindowConstants
from java.awt import BorderLayout
import javax.swing.JTable as JTable
from java.awt import Dimension as Dimension
root = moneydance.getRootAccount()
self.book = root.getBook()
# StockGlance75.tableModel = self.getTableModel(self.book)
# StockGlance75.table = self.SGTable(StockGlance75.tableModel)
self.tableModel = self.getTableModel(self.book)
self.table = self.SGTable(self.tableModel)
self.footerModel = self.getFooterModel()
self.footerTable = self.SGFooterTable(self.footerModel)
# StockGlance75.footerModel = self.getFooterModel()
# StockGlance75.footerTable = self.SGFooterTable(StockGlance75.footerModel)
self.scrollPane = JScrollPane(self.table, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED)
# self.scrollPane = JScrollPane(StockGlance75.table, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED)
self.scrollPane.setBorder(CompoundBorder(MatteBorder(0, 0, 1, 0, Color.gray), EmptyBorder(0, 0, 0, 0)))
self.footerScrollPane = JScrollPane(self.footerTable, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED)
# self.footerScrollPane = JScrollPane(StockGlance75.footerTable, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED)
self.footerScrollPane.setBorder(CompoundBorder(MatteBorder(0, 0, 1, 0, Color.gray), EmptyBorder(0, 0, 0, 0)))
frame_ = JFrame("StockGlance")
frame_.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE)
frame_.add(self.scrollPane, BorderLayout.CENTER)
self.table.setAutoResizeMode(JTable.AUTO_RESIZE_OFF )
# StockGlance75.table.setAutoResizeMode(JTable.AUTO_RESIZE_OFF )
self.footerTable.setColumnSelectionAllowed(False)
self.footerTable.setRowSelectionAllowed(False)
self.footerTable.setAutoResizeMode(JTable.AUTO_RESIZE_OFF )
# StockGlance75.footerTable.setColumnSelectionAllowed(False)
# StockGlance75.footerTable.setRowSelectionAllowed(False)
# StockGlance75.footerTable.setAutoResizeMode(JTable.AUTO_RESIZE_OFF )
frame_.add(self.footerScrollPane, BorderLayout.SOUTH)
frame_.setSize(1170,1000)
tcm = self.table.getColumnModel()
# tcm = StockGlance75.table.getColumnModel()
tcm.getColumn(0).setPreferredWidth(80)
tcm.getColumn(1).setPreferredWidth(250)
tcm.getColumn(2).setPreferredWidth(80)
tcm.getColumn(3).setPreferredWidth(80)
tcm.getColumn(4).setPreferredWidth(80)
tcm.getColumn(5).setPreferredWidth(80)
tcm.getColumn(6).setPreferredWidth(80)
tcm.getColumn(7).setPreferredWidth(80)
tcm.getColumn(8).setPreferredWidth(80)
tcm.getColumn(9).setPreferredWidth(260)
tcm = self.footerTable.getColumnModel()
# tcm = StockGlance75.footerTable.getColumnModel()
tcm.getColumn(0).setPreferredWidth(80)
tcm.getColumn(1).setPreferredWidth(250)
tcm.getColumn(2).setPreferredWidth(80)
tcm.getColumn(3).setPreferredWidth(80)
tcm.getColumn(4).setPreferredWidth(80)
tcm.getColumn(5).setPreferredWidth(80)
tcm.getColumn(6).setPreferredWidth(80)
tcm.getColumn(7).setPreferredWidth(80)
tcm.getColumn(8).setPreferredWidth(80)
tcm.getColumn(9).setPreferredWidth(260)
self.footerScrollPane.setPreferredSize(Dimension(10,30))# (width,height) width doesn't matter
self.footerTableHeader = self.footerTable.getTableHeader()
# self.footerTableHeader = StockGlance75.footerTable.getTableHeader()
self.footerTableHeader.setEnabled(False) # may have worked
self.footerTableHeader.setPreferredSize(Dimension(0,0)) # this worked no more footer Table header
self.tableHeader = self.table.getTableHeader()
# self.tableHeader = StockGlance75.table.getTableHeader()
self.tableHeader.setReorderingAllowed(False) # no more drag and drop columns, it didn't work
# frame_.pack()
frame_.setVisible(True)
def __init__(self, ui):
JSplitPane.__init__(self, JSplitPane.HORIZONTAL_SPLIT)
self._ui = ui
# create the executor object
self._executor = Executor(self, ui.callbacks)
####
# start Left Top split layout
jLeftTopPanel = JPanel()
jMenuPanel = JPanel()
#Load button
self.jLoad = JButton(Strings.jLoad_text)
self.jLoad.addActionListener(self)
#File name text field
self.jFileName = JTextField(Strings.jFileName_default, 30)
self.jFileName.setHorizontalAlignment(JTextField.CENTER)
self.jFileName.setEditable(False)
#Save button
self.jSave = JButton(Strings.jSave_text)
self.jSave.addActionListener(self)
#Exit button
self.jExit = JButton(Strings.jExit_text)
self.jExit.addActionListener(self)
#Wiki button (URL)
self.jWiki = JButton(Strings.jWiki_title)
self.jWiki.setToolTipText(Strings.jWiki_tooltip)
self.jWiki.addActionListener(self)
# make it borderless
self.jWiki.setBorder(EmptyBorder(0, 0, 0, 0))
self.jWiki.setBorderPainted(False)
self.jWiki.setContentAreaFilled(False)
#Console text area
jConsoleText = JTextArea()
jConsoleText.setEditable(0)
jConsoleText.setWrapStyleWord(1)
jConsoleText.setRows(10)
#set initial text
jConsoleText.setText(Strings.jConsoleText_help)
#make scrollable
jScrollConsolePane = JScrollPane()
jScrollConsolePane.setViewportView(jConsoleText)
jMenuPanelLayout = GroupLayout(jMenuPanel)
jMenuPanel.setLayout(jMenuPanelLayout)
jMenuPanelLayout.setHorizontalGroup(
jMenuPanelLayout.createParallelGroup(
GroupLayout.Alignment.LEADING).
addGroup(jMenuPanelLayout.createSequentialGroup().addContainerGap(
).addComponent(self.jLoad).addComponent(
self.jFileName).addPreferredGap(
LayoutStyle.ComponentPlacement.RELATED).addComponent(
self.jSave).addPreferredGap(
LayoutStyle.ComponentPlacement.RELATED).
addComponent(self.jWiki).addPreferredGap(
LayoutStyle.ComponentPlacement.RELATED).addComponent(
self.jExit).addContainerGap()))
jMenuPanelLayout.setVerticalGroup(
jMenuPanelLayout.createParallelGroup(
GroupLayout.Alignment.LEADING).addGroup(
jMenuPanelLayout.createSequentialGroup().addGroup(
jMenuPanelLayout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
self.jLoad).addComponent(
self.jFileName, GroupLayout.PREFERRED_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.PREFERRED_SIZE).addComponent(
self.jSave).addComponent(
self.jWiki).addComponent(
self.jExit))))
jLeftTopPanelLayout = GroupLayout(jLeftTopPanel)
jLeftTopPanel.setLayout(jLeftTopPanelLayout)
jLeftTopPanelLayout.setHorizontalGroup(
jLeftTopPanelLayout.createParallelGroup(
GroupLayout.Alignment.LEADING).addComponent(
jMenuPanel, GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE).addComponent(
jScrollConsolePane, GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE, 32767))
jLeftTopPanelLayout.setVerticalGroup(
jLeftTopPanelLayout.
createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
GroupLayout.Alignment.TRAILING,
jLeftTopPanelLayout.createSequentialGroup().addComponent(
jMenuPanel, GroupLayout.PREFERRED_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.PREFERRED_SIZE).addPreferredGap(
LayoutStyle.ComponentPlacement.RELATED).addComponent(
jScrollConsolePane, GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE, 32767)))
# end Left Top split layout
####
####
# start Left Down split layout
jLeftDownPanel = JPanel()
jMenu2Panel = JPanel()
#Clear button
self.jClear = JButton(Strings.jClear_text)
self.jClear.setToolTipText(Strings.jClear_tooltip)
self.jClear.addActionListener(self)
#Run button
self.jRun = JButton(Strings.jRun_text)
self.jRun.setToolTipText(Strings.jRun_tooltip)
self.jRun.addActionListener(self)
#Variables text area
jVarsPane = JTextPane()
jVarsPane.setFont(Font('Monospaced', Font.PLAIN, 11))
jVarsPane.addFocusListener(self)
# set initial value
jVarsPane.setText(Strings.jVarsPane_header)
# make scrollable
jScrollpaneLeftDown = JScrollPane()
jScrollpaneLeftDown.setViewportView(jVarsPane)
jMenu2PanelLayout = GroupLayout(jMenu2Panel)
jMenu2Panel.setLayout(jMenu2PanelLayout)
jMenu2PanelLayout.setHorizontalGroup(
jMenu2PanelLayout.createParallelGroup(
GroupLayout.Alignment.LEADING).addGroup(
jMenu2PanelLayout.createSequentialGroup().addContainerGap(
).addComponent(self.jClear).addPreferredGap(
LayoutStyle.ComponentPlacement.RELATED, 100,
32767).addComponent(self.jRun).addContainerGap()))
jMenu2PanelLayout.setVerticalGroup(
jMenu2PanelLayout.createParallelGroup(
GroupLayout.Alignment.LEADING).addGroup(
jMenu2PanelLayout.createSequentialGroup().addGroup(
jMenu2PanelLayout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
self.jClear).addComponent(self.jRun))))
jLeftDownPanelLayout = GroupLayout(jLeftDownPanel)
jLeftDownPanel.setLayout(jLeftDownPanelLayout)
jLeftDownPanelLayout.setHorizontalGroup(
jLeftDownPanelLayout.createParallelGroup(
GroupLayout.Alignment.LEADING).addComponent(
jMenu2Panel, GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE).addComponent(
jScrollpaneLeftDown, GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE, 32767))
jLeftDownPanelLayout.setVerticalGroup(
jLeftDownPanelLayout.
createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
GroupLayout.Alignment.TRAILING,
jLeftDownPanelLayout.createSequentialGroup().addComponent(
jMenu2Panel, GroupLayout.PREFERRED_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.PREFERRED_SIZE).addPreferredGap(
LayoutStyle.ComponentPlacement.RELATED).addComponent(
jScrollpaneLeftDown, GroupLayout.DEFAULT_SIZE,
GroupLayout.DEFAULT_SIZE, 32767)))
# end Left Down split layout
####
####
# start Left layout
jSplitPaneLeft = JSplitPane(JSplitPane.VERTICAL_SPLIT, jLeftTopPanel,
jLeftDownPanel)
jSplitPaneLeft.setDividerLocation(300)
# end Left layout
####
####
# start Right layout
jScriptPane = JTextPane()
jScriptPane.setFont(Font('Monospaced', Font.PLAIN, 11))
# set initial value
jScriptPane.setText(Strings.jScriptPane_header)
#jScriptPane.addMouseListener(self)
jScrollPaneRight = JScrollPane()
jScrollPaneRight.setViewportView(jScriptPane)
# end Right layout
####
self.setLeftComponent(jSplitPaneLeft)
self.setRightComponent(jScrollPaneRight)
self.setDividerLocation(450)
#Exported variables
self.jConsoleText = jConsoleText
self.jScrollConsolePane = jScrollConsolePane
self.jScriptPane = jScriptPane
self.jVarsPane = jVarsPane
def drawUI(self):
# Make a whole Burp Suite tab just for this plugin
self.tab = swing.JPanel()
# Draw title area
self.uiLabel = swing.JLabel('Site Map to CSV Options')
self.uiLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiLabel.setForeground(Color(235,136,0))
# UI for high-level options
self.uiScopeOnly = swing.JRadioButton('In-scope only', True)
self.uiScopeAll = swing.JRadioButton('All (disregard scope)', False)
self.uiScopeButtonGroup = swing.ButtonGroup()
self.uiScopeButtonGroup.add(self.uiScopeOnly)
self.uiScopeButtonGroup.add(self.uiScopeAll)
# Draw areas in the tab to keep different UI commands separate
self.uipaneA = swing.JSplitPane(swing.JSplitPane.HORIZONTAL_SPLIT)
self.uipaneA.setMaximumSize(Dimension(900,125))
self.uipaneA.setDividerSize(2)
self.uipaneB = swing.JSplitPane(swing.JSplitPane.HORIZONTAL_SPLIT)
self.uipaneB.setDividerSize(2)
self.uipaneA.setRightComponent(self.uipaneB)
self.uipaneA.setBorder(BorderFactory.createLineBorder(Color.black))
# Fill in UI area for response code filters
self.uiCodesPanel = swing.JPanel()
self.uiCodesPanel.setPreferredSize(Dimension(200, 75))
self.uiCodesPanel.setBorder(EmptyBorder(10,10,10,10))
self.uiCodesPanel.setLayout(BorderLayout())
self.uiCodesLabel = swing.JLabel('Response code filters')
self.uiCodesLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiRcodePanel = swing.JPanel()
self.uiRcodePanel.setLayout(GridLayout(1,1))
self.uiRcode1xx = swing.JCheckBox('1XX ', False)
self.uiRcode2xx = swing.JCheckBox('2XX ', True)
self.uiRcode3xx = swing.JCheckBox('3XX ', True)
self.uiRcode4xx = swing.JCheckBox('4XX ', True)
self.uiRcode5xx = swing.JCheckBox('5XX ', True)
self.uiRcodePanel.add(self.uiRcode1xx)
self.uiRcodePanel.add(self.uiRcode2xx)
self.uiRcodePanel.add(self.uiRcode3xx)
self.uiRcodePanel.add(self.uiRcode4xx)
self.uiRcodePanel.add(self.uiRcode5xx)
self.uiCodesPanel.add(self.uiCodesLabel,BorderLayout.NORTH)
self.uiCodesPanel.add(self.uiRcodePanel,BorderLayout.WEST)
self.uipaneA.setLeftComponent(self.uiCodesPanel)
# Fill in UI area for initiating export to CSV
self.uiExportPanel = swing.JPanel()
self.uiExportPanel.setPreferredSize(Dimension(200, 75))
self.uiExportPanel.setBorder(EmptyBorder(10,10,10,10))
self.uiExportPanel.setLayout(BorderLayout())
self.uiExportLabel = swing.JLabel('Export')
self.uiExportLabel.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiMustHaveResponse = swing.JRadioButton('Must have a response ', True)
self.uiAllRequests = swing.JRadioButton('All (overrides response code filters) ', False)
self.uiResponseButtonGroup = swing.ButtonGroup()
self.uiResponseButtonGroup.add(self.uiMustHaveResponse)
self.uiResponseButtonGroup.add(self.uiAllRequests)
self.uiExportRun = swing.JButton('Export',actionPerformed=self.exportAndSaveCsv)
self.uiExportButtonPanel = swing.JPanel()
self.uiExportButtonPanel.add(self.uiExportRun)
self.uiExportPanel.add(self.uiExportLabel,BorderLayout.NORTH)
self.uiExportPanel.add(self.uiMustHaveResponse,BorderLayout.WEST)
self.uiExportPanel.add(self.uiAllRequests,BorderLayout.CENTER)
self.uiExportPanel.add(self.uiExportButtonPanel,BorderLayout.SOUTH)
self.uipaneB.setLeftComponent(self.uiExportPanel)
# Common UI stuff
layout = swing.GroupLayout(self.tab)
self.tab.setLayout(layout)
# Thank you to Smeege (https://github.com/SmeegeSec/Burp-Importer/) for helping me figure out how this works.
# He in turn gave credit to Antonio Sanchez (https://github.com/Dionach/HeadersAnalyzer/)
layout.setHorizontalGroup(
layout.createParallelGroup(swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addGap(10, 10, 10)
.addGroup(layout.createParallelGroup(swing.GroupLayout.Alignment.LEADING)
.addComponent(self.uiLabel)
.addGroup(layout.createSequentialGroup()
.addGap(10,10,10)
.addComponent(self.uiScopeOnly)
.addGap(10,10,10)
.addComponent(self.uiScopeAll))
.addGap(15,15,15)
.addComponent(self.uipaneA))
.addContainerGap(26, lang.Short.MAX_VALUE)))
layout.setVerticalGroup(
layout.createParallelGroup(swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addGap(15,15,15)
.addComponent(self.uiLabel)
.addGap(15,15,15)
.addGroup(layout.createParallelGroup()
.addComponent(self.uiScopeOnly)
.addComponent(self.uiScopeAll))
.addGap(20,20,20)
.addComponent(self.uipaneA)
.addGap(20,20,20)
.addGap(5,5,5)
.addGap(20,20,20)))
def __init__(self):
TrafficMod.__init__(self)
self.setLayout(BorderLayout())
self.setBorder(EmptyBorder(18, 18, 18, 18))
main_title = JLabel(self.title)
main_title.setFont(get_title_font(main_title))
self.add(main_title, BorderLayout.PAGE_START)
self.filter_list.setCellRenderer(ui_components.FilterRenderer())
self.filter_list.setModel(self.filter_list_model)
self.add(self.filter_list, BorderLayout.CENTER)
# Buttons to add or remove a filter
add_remove_panel = JPanel()
add_remove_panel.setLayout(FlowLayout(FlowLayout.LEADING))
# add_remove_panel.setBorder(EmptyBorder(10, 10, 10, 10))
# add_remove_panel.setBorder(LineBorder(Color.RED))
# add_remove_panel.setLayout(BoxLayout(add_remove_panel, BoxLayout.X_AXIS))
# add_remove_panel.add(Box.createRigidArea(Dimension(0, 10)))
# Edit Filter button
edit_filter_panel = self.edit_filter_panel()
def edit_filter(event):
# TODO Set the properties on the edit filter based on the passed event
result = JOptionPane.showConfirmDialog(
self, edit_filter_panel, "Edit Filter",
JOptionPane.OK_CANCEL_OPTION, JOptionPane.PLAIN_MESSAGE)
if result == 0:
print "Will save filter"
edit_btn = JButton("Edit", actionPerformed=edit_filter)
edit_btn.setEnabled(False)
add_remove_panel.add(edit_btn)
# Add Filter button
def add_filter(event):
result = JOptionPane.showConfirmDialog(
self, edit_filter_panel, "Add Filter",
JOptionPane.OK_CANCEL_OPTION, JOptionPane.PLAIN_MESSAGE)
if result == 0:
print "Will save new filter"
add_btn = JButton("Add", actionPerformed=add_filter)
# add_btn.setMinimumSize(Dimension(200, 30))
# add_btn.setPreferredSize(Dimension(100, 30))
# add_btn.setAlignmentX(Component.CENTER_ALIGNMENT)
add_remove_panel.add(add_btn)
# add_remove_panel.add(Box.createRigidArea(Dimension(0, 10)))
# minSize = Dimension(100, 10)
# prefSize = Dimension(100, 10)
# maxSize = Dimension(100, 10)
# add_remove_panel.add(Box.Filler(minSize, prefSize, maxSize))
# todo add an action listener
remove_btn = JButton("Remove")
remove_btn.setEnabled(False)
# remove_btn.setAlignmentX(Component.CENTER_ALIGNMENT)
add_remove_panel.add(remove_btn)
self.add(add_remove_panel, BorderLayout.PAGE_END)
