def __init__(self):
# set up the layout
self.__component = JPanel(GridBagLayout())
self.__image = JLabel()
self.__album = JLabel()
self.__artist = JLabel()
self.__application = None
self.__image.setVerticalAlignment(SwingConstants.TOP)
self.__album.setVerticalAlignment(SwingConstants.TOP)
self.__artist.setVerticalAlignment(SwingConstants.TOP)
gbc = GridBagConstraints()
gbc.fill = GridBagConstraints.VERTICAL
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridx = 0
gbc.gridy = 0
gbc.weighty = 2
gbc.gridheight = 2
self.__component.add(self.__image, gbc)
gbc.fill = GridBagConstraints.HORIZONTAL
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridx = 1
gbc.gridy = 0
gbc.gridheight = 1
gbc.weighty = 0
gbc.insets = Insets(0, 10, 0, 10)
self.__component.add(self.__album, gbc)
gbc.fill = GridBagConstraints.BOTH
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridx = 1
gbc.gridy = 1
gbc.weightx = 2
gbc.weighty = 2
gbc.gridheight = 1
self.__component.add(self.__artist, gbc)
def initComponents(self):
self.setLayout(GridBagLayout())
gc = GridBagConstraints()
gc.weightx = 1
gc.weighty = 1
# First Row
gc.gridy = 0
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.fill = GridBagConstraints.NONE
gc.anchor = GridBagConstraints.LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.usernameLabel, gc)
gc.gridx = 1
gc.anchor = GridBagConstraints.LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.usernameField, gc)
# Second Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.superPassLabel, gc)
gc.gridx = 1
gc.anchor = GridBagConstraints.LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.superPassField, gc)
#Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 2.0
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
self.add(self.deleteAdminBtn, gc)
innerBorder = BorderFactory.createTitledBorder('Delete Admin')
outerBorder = BorderFactory.createEmptyBorder(5, 5, 5, 5)
self.setBorder(
BorderFactory.createCompoundBorder(outerBorder, innerBorder))
def addStatusLabel(self, msg):
gbc = GridBagConstraints()
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridx = 0
gbc.gridy = 7
lab = JLabel(msg)
self.add(lab, gbc)
def p_build_ui(self, event):
"""
Adds a list of checkboxes, one for each loaded plugin
to the Selct plugins window
"""
if not self.loaded_p_list:
self.update_scroll("[!!] No plugins loaded!")
return
scroll_pane = JScrollPane()
scroll_pane.setPreferredSize(Dimension(200, 250))
check_frame = JPanel(GridBagLayout())
constraints = GridBagConstraints()
constraints.fill = GridBagConstraints.HORIZONTAL
constraints.gridy = 0
constraints.anchor = GridBagConstraints.FIRST_LINE_START
for plug in self.loaded_p_list:
check_frame.add(JCheckBox(plug.get_name(), plug.enabled,
actionPerformed=self.update_box),
constraints)
constraints.gridy += 1
vport = JViewport()
vport.setView(check_frame)
scroll_pane.setViewport(vport)
self.window.contentPane.add(scroll_pane)
self.window.pack()
self.window.setVisible(True)
def initComponents(self):
self.setLayout(GridBagLayout())
gbc = GridBagConstraints()
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridx = 0
gbc.gridy = 0
descriptionLabel = JLabel("FEA - Credit Card module")
self.add(descriptionLabel, gbc)
self.cbGenerateExcel = JCheckBox("Generate Excel format report (more detailed)", actionPerformed=self.cbGenerateExcelActionPerformed)
self.cbGenerateExcel.setSelected(True)
gbc.gridy = 2
self.add(self.cbGenerateExcel, gbc)
self.cbGenerateCSV = JCheckBox("Generate CSV format report (plaintext)", actionPerformed=self.cbGenerateCSVActionPerformed)
self.cbGenerateCSV.setSelected(True)
gbc.gridy = 3
self.add(self.cbGenerateCSV, gbc)
self.cbRemoveFalsePositives = JCheckBox("Remove False Positives from Autopsy", actionPerformed=self.cbRemoveFalsePositivesActionPerformed)
self.cbRemoveFalsePositives.setSelected(True)
gbc.gridy = 4
self.cbRemoveFalsePositives.setEnabled(False)
self.add(self.cbRemoveFalsePositives, gbc)
def add(parent,
child,
gridx=0,
gridy=0,
anchor=GC.NORTHWEST,
fill=GC.NONE,
weightx=0.0,
weighty=0.0,
gridwidth=1):
c = GC()
c.gridx = gridx
c.gridy = gridy
c.anchor = anchor
c.fill = fill
c.weightx = weightx
c.weighty = weighty
c.gridwidth = gridwidth
"""
# Same, more flexible, less verbose: BUT FAILS at parent, child args
kv = locals() # dict of local variables including the function arguments
c = GC()
for key, value in kv.iteritems():
setattr(c, key, value)
"""
#
parent.getLayout().setConstraints(child, c)
parent.add(child)
def p_build_ui(self, event):
"""
Adds a list of checkboxes, one for each loaded plugin
to the Selct plugins window
"""
if not self.loaded_p_list:
self.update_scroll("[!!] No plugins loaded!")
return
scroll_pane = JScrollPane()
scroll_pane.setPreferredSize(Dimension(200, 250))
check_frame = JPanel(GridBagLayout())
constraints = GridBagConstraints()
constraints.fill = GridBagConstraints.HORIZONTAL
constraints.gridy = 0
constraints.anchor = GridBagConstraints.FIRST_LINE_START
for plug in self.loaded_p_list:
check_frame.add(JCheckBox(plug.get_name(), plug.enabled,
actionPerformed=self.update_box),
constraints)
constraints.gridy += 1
vport = JViewport()
vport.setView(check_frame)
scroll_pane.setViewport(vport)
self.window.contentPane.add(scroll_pane)
self.window.pack()
self.window.setVisible(True)
def cachedBPROM(genome, fileName, frame):
"""
genome: Genome as a string.
fileName: File to save the BPROM results in.
swing: A JFrame or None. If this is None then messages will be printed, if it isn't then
they will also be put in a dialog box.
return: Results of the BPROM prediction stored in a list of Promoter objects.
If the file Specified by fileName already exists then this function simply parses the file
already there. Also, if a request is made to BPROM and nothing is returned, no file is
created, the user is warned, and an empty list is returned.
"""
offset = 25 if ".forward.bprom" in fileName else 50
direction = "forward" if offset == 50 else "reverse"
def getPromoters():
input = open(fileName, "r")
results = parseBPROM(input.read())
input.close()
return results
if not os.path.isfile(fileName):
results = urllib.urlopen("http://linux1.softberry.com/cgi-bin/programs/gfindb/bprom.pl",
urllib.urlencode({"DATA" : genome}))
resultString = results.read()
results.close()
resultString = resultString[resultString.find("<pre>"):resultString.find("</pre>")]
resultString = re.sub("<+.+>+", "", resultString).strip()
if resultString:
output = open(fileName, "w")
output.write(resultString)
output.close()
return getPromoters()
else:
if frame:
messageFrame = JFrame("BPROM Error",
defaultCloseOperation = WindowConstants.DISPOSE_ON_CLOSE)
messageFrame.setLocation(frame.location().x + offset, frame.location().y + offset)
messageFrame.contentPane.layout = GridBagLayout()
constraints = GridBagConstraints()
constraints.gridx, constraints.gridy = 0, 0
constraints.gridwidth, constraints.gridheight = 1, 1
constraints.fill = GridBagConstraints.BOTH
constraints.weightx, constraints.weighty = 1, 1
messageFrame.contentPane.add(JLabel("<html>The pipeline will continue to run but BPROM<br/>did not process the request for promoters on the<br/>" + direction + " strand. Try again tomorrow.</html>"), constraints)
constraints.gridx, constraints.gridy = 0, 1
constraints.fill = GridBagConstraints.NONE
constraints.weightx, constraints.weighty = 1, 1
constraints.anchor = GridBagConstraints.LINE_END
messageFrame.contentPane.add(JButton("Ok", actionPerformed = lambda e: messageFrame.dispose()), constraints)
messageFrame.pack()
messageFrame.visible = True
print "BPROM Error:", "The pipeline will continue to run but BPROM did not process the request for promoters on the " + direction + " strand. Try again tomorrow"
return []
else:
return getPromoters()
def __init__(self, callbacks):
self._callbacks = callbacks
self.tab = JPanel(GridBagLayout())
about_constraints = GridBagConstraints()
about_author = (("<html><center><h2>SpyDir</h2><b>Version:</b> "
"%s<br/>Created by: <em>Ryan Reid</em>"
" (@_aur3lius)<br/>https://github.com/aur3lius-dev/"
"SpyDir</center><br/>")
% VERSION)
about_spydir = """<em><center>
SpyDir is an extension that assists in the enumeration of
application<br/>
endpoints via an input directory containing the application's<br/>
source code. It provides an option to process files as endpoints,<br/>
think: ASP, PHP, HTML, or parse files to attempt to enumerate<br/>
endpoints via plugins, think: MVC. Users may opt to send the<br/>
discovered endpoints directly to the Burp Spider.
</center></em><br/>
This tool is in <b>Alpha</b>! <b>Please</b> provide feedback on the
GitHub page!<br/><br/>"""
getting_started = """<b>Getting started:</b><br/>
<ul>
<li>Add a local source repository</li>
<li>Add the target URL</li>
<li>Use the String delimiter to construct the appropriate
directory path (if necessary)</li>
<li>Alternatively, parse each file by selecting plugins
and checking the checkbox</li>
<li>Explicitly define the file extensions to process</li>
<li>Parse the directory</li>
<li>Verify output is correct <b>before</b> sending to spider</li>
<li>Send requests to the Burp Spider</li>
</ul>
"""
advanced_info = r"""<html><b>String Delimiter</b><br/>
String Delimiter allows us to append the necessary section
of the folder structure.<br/>
Suppose the target application is hosted at the following URL:
https://localhost:8080. <br/>The target code base is stored in:
'C:\Source\TestApp'. <br/>Within the TestApp folder there is a
subfolder, 'views', with static .html files.<br/>
In this case the String Delimiter will need to equal 'TestApp'.
<br/>With the expectation that the tool will produce an example URL
will such as:<br/>https://localhost:8080/views/view1.html.<br/>
<b>Note:</b> String Delimiter is ignored if parsing files using
plugins!
</html>"""
about_constraints.anchor = GridBagConstraints.FIRST_LINE_START
about_constraints.weightx = 1.0
about_constraints.weighty = 1.0
self.tab.add(JLabel("%s\n%s\n%s\n%s" % (about_author, about_spydir,
getting_started,
advanced_info)),
about_constraints)
def addSigningKeyTextArea(self):
self._signingKeyTextArea = JTextArea()
self._signingKeyTextArea.setColumns(50)
self._signingKeyTextArea.setRows(10)
self._signingKeyScrollPane = JScrollPane(self._signingKeyTextArea)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 6
c.anchor = GridBagConstraints.LINE_START
self._configurationPanel.add(self._signingKeyScrollPane,c)
def initComponents(self):
self.setLayout(GridBagLayout())
gc = GridBagConstraints()
gc.weightx = 1
gc.weighty = 1
# First Row
gc.gridy = 0
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.fill = GridBagConstraints.NONE
gc.anchor = GridBagConstraints.LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.vehicleLabel, gc)
gc.gridx = 1
gc.anchor = GridBagConstraints.LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.vehicleField, gc)
#Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 2.0
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
self.add(self.bypassBtn, gc)
innerBorder = BorderFactory.createTitledBorder('Emergency Bypass')
outerBorder = BorderFactory.createEmptyBorder(5, 5, 5, 5)
self.setBorder(
BorderFactory.createCompoundBorder(outerBorder, innerBorder))
def _constructAboutPanel(self, insets):
aboutPanel = JPanel(GridBagLayout())
with open("about.html") as file:
aboutBody = file.read()
aboutLabel = JLabel(aboutBody.format(extension_name=EXTENSION_NAME))
aboutLabelConstraints = GridBagConstraints()
aboutLabelConstraints.weightx = 1
aboutLabelConstraints.weighty = 1
aboutLabelConstraints.insets = insets
aboutLabelConstraints.fill = GridBagConstraints.HORIZONTAL
aboutLabelConstraints.anchor = GridBagConstraints.PAGE_START
aboutPanel.add(aboutLabel, aboutLabelConstraints)
return aboutPanel
def initializeDisplay(self, queries, swing):
"""
queries: A list of the fasts files to be processed.
swing: If true then updates about progress will be displayed in a swing window, otherwise they will be written to stdout.
Initializes the interface for telling the user about progress in the pipeline. Queries is used to count the
number of queries the pipeline will process and to size the swing display(if it is used) so that text
isn't cutoff at the edge of the window. The swing display is setup if swing is true.
"""
self.numJobs = len(queries)
if swing:
self.frame = JFrame("Neofelis")
self.frame.addWindowListener(PipelineWindowAdapter(self))
contentPane = JPanel(GridBagLayout())
self.frame.setContentPane(contentPane)
self.globalLabel = JLabel(max(queries, key = len))
self.globalProgress = JProgressBar(0, self.numJobs)
self.currentLabel = JLabel(max(self.messages, key = len))
self.currentProgress = JProgressBar(0, len(self.messages))
self.doneButton = JButton(DoneAction(self.frame))
self.doneButton.setEnabled(False)
constraints = GridBagConstraints()
constraints.gridx, constraints.gridy = 0, 0
constraints.gridwidth, constraints.gridheight = 1, 1
constraints.weightx = 1
constraints.fill = GridBagConstraints.HORIZONTAL
contentPane.add(self.globalLabel, constraints)
constraints.gridy = 1
contentPane.add(self.globalProgress, constraints)
constraints.gridy = 2
contentPane.add(self.currentLabel, constraints)
constraints.gridy = 3
contentPane.add(self.currentProgress, constraints)
constraints.gridy = 4
constraints.weightx = 0
constraints.fill = GridBagConstraints.NONE
constraints.anchor = GridBagConstraints.LINE_END
contentPane.add(self.doneButton, constraints)
self.frame.pack()
self.frame.setResizable(False)
self.globalLabel.setText(" ")
self.currentLabel.setText(" ")
self.frame.setLocationRelativeTo(None)
self.frame.setVisible(True)
def initializeDisplay(self, queries, swing):
"""
queries: A list of the fasts files to be processed.
swing: If true then updates about progress will be displayed in a swing window, otherwise they will be written to stdout.
Initializes the interface for telling the user about progress in the pipeline. Queries is used to count the
number of queries the pipeline will process and to size the swing display(if it is used) so that text
isn't cutoff at the edge of the window. The swing display is setup if swing is true.
"""
self.numJobs = len(queries)
if swing:
self.frame = JFrame("Neofelis")
self.frame.addWindowListener(PipelineWindowAdapter(self))
contentPane = JPanel(GridBagLayout())
self.frame.setContentPane(contentPane)
self.globalLabel = JLabel(max(queries, key=len))
self.globalProgress = JProgressBar(0, self.numJobs)
self.currentLabel = JLabel(max(self.messages, key=len))
self.currentProgress = JProgressBar(0, len(self.messages))
self.doneButton = JButton(DoneAction(self.frame))
self.doneButton.setEnabled(False)
constraints = GridBagConstraints()
constraints.gridx, constraints.gridy = 0, 0
constraints.gridwidth, constraints.gridheight = 1, 1
constraints.weightx = 1
constraints.fill = GridBagConstraints.HORIZONTAL
contentPane.add(self.globalLabel, constraints)
constraints.gridy = 1
contentPane.add(self.globalProgress, constraints)
constraints.gridy = 2
contentPane.add(self.currentLabel, constraints)
constraints.gridy = 3
contentPane.add(self.currentProgress, constraints)
constraints.gridy = 4
constraints.weightx = 0
constraints.fill = GridBagConstraints.NONE
constraints.anchor = GridBagConstraints.LINE_END
contentPane.add(self.doneButton, constraints)
self.frame.pack()
self.frame.setResizable(False)
self.globalLabel.setText(" ")
self.currentLabel.setText(" ")
self.frame.setLocationRelativeTo(None)
self.frame.setVisible(True)
def __fillQuestions(self):
panel = JPanel()
panel.setLayout(GridBagLayout())
panel.setBorder(None)
c = GridBagConstraints()
c.gridx = 0
c.gridy = 0
c.weightx = 1.0
c.fill = GridBagConstraints.HORIZONTAL
c.anchor = GridBagConstraints.PAGE_START
line = 0
for question in self.test.getQuestions():
c.gridy = line
panel.add(question.getPanel().asJComponent(),c)
line += 1
scrollPanel = JScrollPane(panel)
scrollPanel.setBorder(None)
self.questionsContainer.setLayout(BorderLayout())
self.questionsContainer.add(scrollPanel, BorderLayout.CENTER)
def initComponents(self):
self.setLayout(GridBagLayout())
gbc = GridBagConstraints()
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridx = 0
gbc.gridy = 0
descriptionLabel = JLabel("FEA - BitCoin Validation module")
self.add(descriptionLabel, gbc)
tlHitlist = JLabel("Base list of hashes to analyze: ")
gbc.gridy = 1
self.add(tlHitlist, gbc)
self.tbHitlist = JTextField("testlist", 20)
self.tbHitlist.addActionListener(self.tbHitlistActionPerformed)
gbc.gridx = 1
self.add(self.tbHitlist, gbc)
gbc.gridx = 0
self.cbBlockchainCheck = JCheckBox(
"Query Blockchain.info",
actionPerformed=self.cbBlockchainCheckActionPerformed)
self.cbBlockchainCheck.setSelected(True)
gbc.gridy = 2
self.add(self.cbBlockchainCheck, gbc)
tlBCMaxHits = JLabel(
"Timeout (in seconds) between calls to Blockchain.info: ")
gbc.gridy = 3
self.add(tlBCMaxHits, gbc)
self.tbMaxBCHits = JTextField("5", 5)
self.tbMaxBCHits.addActionListener(self.tbMaxBCHitsActionPerformed)
gbc.gridx = 1
self.add(self.tbMaxBCHits, gbc)
def __init__(self, callbacks, parent):
# Initialze self stuff
self._callbacks = callbacks
self.config = {}
self.ext_stats = {}
self.url_reqs = []
self.parse_files = False
self.tab = JPanel(GridBagLayout())
self.view_port_text = JTextArea("===SpyDir===")
self.delim = JTextField(30)
self.ext_white_list = JTextField(30)
# I'm not sure if these fields are necessary still
# why not just use Burp func to handle this?
# leaving them in case I need it for the HTTP handler later
# self.cookies = JTextField(30)
# self.headers = JTextField(30)
self.url = JTextField(30)
self.parent_window = parent
self.plugins = {}
self.loaded_p_list = set()
self.loaded_plugins = False
self.config['Plugin Folder'] = None
self.double_click = False
self.source_input = ""
self.print_stats = True
self.curr_conf = JLabel()
self.window = JFrame("Select plugins",
preferredSize=(200, 250),
windowClosing=self.p_close)
self.window.setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE)
self.window.setVisible(False)
self.path_vars = JTextField(30)
# Initialize local stuff
tab_constraints = GridBagConstraints()
status_field = JScrollPane(self.view_port_text)
# Configure view port
self.view_port_text.setEditable(False)
labels = self.build_ui()
# Add things to rows
tab_constraints.anchor = GridBagConstraints.FIRST_LINE_END
tab_constraints.gridx = 1
tab_constraints.gridy = 0
tab_constraints.fill = GridBagConstraints.HORIZONTAL
self.tab.add(JButton(
"Resize screen", actionPerformed=self.resize),
tab_constraints)
tab_constraints.gridx = 0
tab_constraints.gridy = 1
tab_constraints.anchor = GridBagConstraints.FIRST_LINE_START
self.tab.add(labels, tab_constraints)
tab_constraints.gridx = 1
tab_constraints.gridy = 1
tab_constraints.fill = GridBagConstraints.BOTH
tab_constraints.weightx = 1.0
tab_constraints.weighty = 1.0
tab_constraints.anchor = GridBagConstraints.FIRST_LINE_END
self.tab.add(status_field, tab_constraints)
try:
self._callbacks.customizeUiComponent(self.tab)
except Exception:
pass
except:
print "Malformed regex pattern"
def typingInSearchField(event):
if KeyEvent.VK_ENTER == event.getKeyCode():
filterTable()
elif KeyEvent.VK_ESCAPE == event.getKeyCode():
search_field.setText("")
filterTable() # to restore the full list of rows
# Top-left element: a text field for filtering rows by regular expression match
c.gridx = 0
c.gridy = 0
c.anchor = GridBagConstraints.CENTER
c.fill = GridBagConstraints.HORIZONTAL
search_field = JTextField("", keyPressed=typingInSearchField)
gb.setConstraints(search_field, c)
all.add(search_field)
# Bottom left element: the table, wrapped in a scrollable component
table = JTable(TableModel())
table.setSelectionMode(ListSelectionModel.SINGLE_SELECTION)
#table.setCellSelectionEnabled(True)
table.setAutoCreateRowSorter(
True) # to sort the view only, not the data in the underlying TableModel
c.gridx = 0
c.gridy = 1
c.anchor = GridBagConstraints.NORTHWEST
c.fill = GridBagConstraints.BOTH # resize with the frame
def __init__(self, callbacks):
self._callbacks = callbacks
self.tab = JPanel(GridBagLayout())
about_constraints = GridBagConstraints()
about_author = (("<html><center><h2>SpyDir</h2><b>Version:</b> "
"%s<br/>Created by: <em>Ryan Reid</em>"
" (@_aur3lius)<br/>https://github.com/aur3lius-dev/"
"SpyDir</center><br/>")
% VERSION)
about_spydir = """<em><center>
SpyDir is an extension that assists in the enumeration of
application<br/>
endpoints via an input directory containing the application's<br/>
source code. It provides an option to process files as endpoints,<br/>
think: ASP, PHP, HTML, or parse files to attempt to enumerate<br/>
endpoints via plugins, think: MVC. Users may opt to send the<br/>
discovered endpoints directly to the Burp Spider.
</center></em><br/>
This tool is in <b>Alpha</b>! <b>Please</b> provide feedback on the
GitHub page!<br/><br/>"""
getting_started = """<b>Getting started:</b><br/>
<ul>
<li>Add a local source repository</li>
<li>Add the target URL</li>
<li>Use the String delimiter to construct the appropriate
directory path (if necessary)</li>
<li>Alternatively, parse each file by selecting plugins
and checking the checkbox</li>
<li>Explicitly define the file extensions to process</li>
<li>Parse the directory</li>
<li>Verify output is correct <b>before</b> sending to spider</li>
<li>Send requests to the Burp Spider</li>
</ul>
"""
advanced_info = r"""<html><b>String Delimiter</b><br/>
String Delimiter allows us to append the necessary section
of the folder structure.<br/>
Suppose the target application is hosted at the following URL:
https://localhost:8080. <br/>The target code base is stored in:
'C:\Source\TestApp'. <br/>Within the TestApp folder there is a
subfolder, 'views', with static .html files.<br/>
In this case the String Delimiter will need to equal 'TestApp'.
<br/>With the expectation that the tool will produce an example URL
will such as:<br/>https://localhost:8080/views/view1.html.<br/><br/>
<b>Path Vars</b><br/>Use this option to swap values for dynamically
enumerated query string parameters. This needs to be a JSON object.
<br/><b>Example:</b>{"{userID}": "aur3lius", "admin_status=":
"admin_status=True"}<br/><br/>
<b>Note:</b> String Delimiter is ignored if parsing files using
plugins!
</html>"""
about_constraints.anchor = GridBagConstraints.FIRST_LINE_START
about_constraints.weightx = 1.0
about_constraints.weighty = 1.0
self.tab.add(JLabel("%s\n%s\n%s\n%s" % (about_author, about_spydir,
getting_started,
advanced_info)),
about_constraints)
def show_detectable_objects_dialog(self, e):
parentComponent = SwingUtilities.windowForComponent(self.panel0)
self.detectable_obejcts_dialog = JDialog(
parentComponent, "List of Objects to Detect",
ModalityType.APPLICATION_MODAL)
panel = JPanel()
self.detectable_obejcts_dialog.add(panel)
gbPanel = GridBagLayout()
gbcPanel = GridBagConstraints()
panel.setLayout(gbPanel)
y = 0
x = 0
for line in self.local_settings.getClassesOfInterest():
if y > 15:
y = 0
x = x + 1
class_check_box = JCheckBox(line['name'])
self.classes_of_interest_checkboxes.append(class_check_box)
class_check_box.setEnabled(True)
class_check_box.setSelected(line['enabled'])
class_check_box.addItemListener(self.on_class_checkbox_clicked)
gbcPanel.gridx = x
gbcPanel.gridy = y
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(class_check_box, gbcPanel)
panel.add(class_check_box)
y = y + 1
blank_1_L = JLabel(" ")
blank_1_L.setEnabled(True)
gbcPanel.gridx = 0
gbcPanel.gridy = y + 1
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 0
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(blank_1_L, gbcPanel)
panel.add(blank_1_L)
deselect_all_button = JButton("Deselect all")
deselect_all_button.setEnabled(True)
deselect_all_button.addActionListener(self.on_deselect_all_clicked)
gbcPanel.gridx = 1
gbcPanel.gridy = y + 2
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(deselect_all_button, gbcPanel)
panel.add(deselect_all_button)
select_all_button = JButton("Select all")
select_all_button.setEnabled(True)
select_all_button.addActionListener(self.on_select_all_clicked)
gbcPanel.gridx = 3
gbcPanel.gridy = y + 2
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(select_all_button, gbcPanel)
panel.add(select_all_button)
blank_2_L = JLabel(" ")
blank_2_L.setEnabled(True)
gbcPanel.gridx = 0
gbcPanel.gridy = y + 3
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 0
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(blank_2_L, gbcPanel)
panel.add(blank_2_L)
cancel_button = JButton("Cancel")
cancel_button.setEnabled(True)
cancel_button.addActionListener(
self.on_cancel_classes_of_interest_click)
gbcPanel.gridx = 1
gbcPanel.gridy = y + 4
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(cancel_button, gbcPanel)
panel.add(cancel_button)
save_button = JButton("Save")
save_button.setEnabled(True)
save_button.addActionListener(self.on_save_classes_of_interest_click)
gbcPanel.gridx = 3
gbcPanel.gridy = y + 4
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(save_button, gbcPanel)
panel.add(save_button)
blank_3_L = JLabel(" ")
blank_3_L.setEnabled(True)
gbcPanel.gridx = 0
gbcPanel.gridy = y + 5
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 0
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(blank_3_L, gbcPanel)
panel.add(blank_3_L)
self.detectable_obejcts_dialog.pack()
screenSize = Toolkit.getDefaultToolkit().getScreenSize()
self.detectable_obejcts_dialog.setLocation(
int((screenSize.getWidth() / 2) -
(self.detectable_obejcts_dialog.getWidth() / 2)),
int((screenSize.getHeight() / 2) -
(self.detectable_obejcts_dialog.getHeight() / 2)))
self.detectable_obejcts_dialog.setVisible(True)
def __init__(self, extender):
super(BeautifierOptionsPanel, self).__init__()
self._extender = extender
self.contentWrapper = JPanel(GridBagLayout())
self.setViewportView(self.contentWrapper)
self.getVerticalScrollBar().setUnitIncrement(16)
self.addMouseListener(self.RequestFocusListener(
self)) # Let textArea lose focus when click empty area
innerContainer = JPanel(GridBagLayout())
innerContainer.setFocusable(
True
) # make sure the maxSizeText TextField is not focused when BeautifierOptionsPanel display
# generalOptionPanel and it's inner component
maxSizeLabel = JLabel("Max Size: ")
self.maxSizeText = JTextField(5)
self.maxSizeText.setHorizontalAlignment(SwingConstants.RIGHT)
self.maxSizeText.addFocusListener(self.MaxSizeTextListener(self))
sizeUnitLabel = JLabel("KB")
generalOptionPanel = JPanel(GridBagLayout())
generalOptionPanel.setBorder(
BorderFactory.createTitledBorder("General Options"))
gbc = GridBagConstraints()
gbc.anchor = GridBagConstraints.WEST
gbc.gridx = 0
gbc.gridy = 0
generalOptionPanel.add(maxSizeLabel, gbc)
gbc.fill = GridBagConstraints.HORIZONTAL
gbc.gridx = 1
gbc.gridy = 0
generalOptionPanel.add(self.maxSizeText, gbc)
gbc.fill = GridBagConstraints.NONE
gbc.gridx = 2
gbc.gridy = 0
gbc.weightx = 1.0
generalOptionPanel.add(sizeUnitLabel, gbc)
gbc = GridBagConstraints()
gbc.fill = GridBagConstraints.BOTH
gbc.gridx = 1
gbc.gridy = 0
gbc.gridheight = 2
innerContainer.add(generalOptionPanel, gbc)
# messageTabOptionPanel and it's inner component
self.messageTabFormatCheckBoxs = []
for f in supportedFormats:
ckb = JCheckBox(f)
ckb.addItemListener(self.messageTabFormatListener)
self.messageTabFormatCheckBoxs.append(ckb)
messageTabOptionPanel = JPanel()
messageTabOptionPanel.setLayout(
BoxLayout(messageTabOptionPanel, BoxLayout.Y_AXIS))
messageTabOptionPanel.setBorder(
BorderFactory.createTitledBorder("Enable in MessageEditorTab"))
for b in self.messageTabFormatCheckBoxs:
messageTabOptionPanel.add(b)
gbc.gridx = 1
gbc.gridy = 2
gbc.gridheight = 9
innerContainer.add(messageTabOptionPanel, gbc)
# replaceResponsePanel and it's inner component
self.chkEnableReplace = JCheckBox("Enable")
self.chkEnableReplace.addItemListener(self.repalceResponseBoxListener)
replaceResponseFormatLabel = JLabel("Format")
self.replaceResponseFormatCheckBoxs = []
for f in supportedFormats:
ckb = JCheckBox(f)
ckb.addItemListener(self.replaceResponseFormatListener)
self.replaceResponseFormatCheckBoxs.append(ckb)
replaceResponseIncludeLabel = JLabel(
"Include URL that matches below(one item one line)")
self.URLIncludeTextArea = JTextArea(6, 32)
self.URLIncludeTextArea.addFocusListener(
self.URLIncludeFocusListener(self))
URLIncludeScrollPane = JScrollPane(self.URLIncludeTextArea)
URLIncludeScrollPane.setAlignmentX(Component.LEFT_ALIGNMENT)
replaceResponseExcludeLabel = JLabel(
"Exclude URL that matches below(one item one line)")
self.URLExcludeTextArea = JTextArea(5, 32)
self.URLExcludeTextArea.addFocusListener(
self.URLExcludeFocusListener(self))
URLExcludeScrollPane = JScrollPane(self.URLExcludeTextArea)
URLExcludeScrollPane.setAlignmentX(Component.LEFT_ALIGNMENT)
replaceResponsePanel = JPanel()
replaceResponsePanel.setLayout(
BoxLayout(replaceResponsePanel, BoxLayout.Y_AXIS))
replaceResponsePanel.setBorder(
BorderFactory.createTitledBorder("Replace PROXY Response"))
replaceResponsePanel.add(self.chkEnableReplace)
replaceResponsePanel.add(Box.createVerticalStrut(10))
replaceResponsePanel.add(replaceResponseFormatLabel)
for b in self.replaceResponseFormatCheckBoxs:
replaceResponsePanel.add(b)
replaceResponsePanel.add(Box.createVerticalStrut(10))
replaceResponsePanel.add(replaceResponseIncludeLabel)
replaceResponsePanel.add(URLIncludeScrollPane)
replaceResponsePanel.add(Box.createVerticalStrut(10))
replaceResponsePanel.add(replaceResponseExcludeLabel)
replaceResponsePanel.add(URLExcludeScrollPane)
gbc.gridy = 11
innerContainer.add(replaceResponsePanel, gbc)
# let innerContainer keep away from left and up
gbc = GridBagConstraints()
gbc.gridx = 1
gbc.gridy = 1
self.contentWrapper.add(Box.createHorizontalStrut(15), gbc)
# gbc.ipadx = gbc.ipady = 25
gbc.gridx = 2
self.contentWrapper.add(innerContainer, gbc)
# let innerContainer stay left side
gbc = GridBagConstraints()
gbc.gridx = 3
gbc.gridy = 2
gbc.gridwidth = 1
gbc.weightx = gbc.weighty = 1
paddingPanel = JPanel()
self.contentWrapper.add(paddingPanel, gbc)
self.setDefaultOptionDisplay()
def _constructAttackPanel(self, insets, messageEditorComponent):
attackPanel = JPanel(GridBagLayout())
targetHeadingLabel = JLabel("<html><b>Target</b></html>")
targetHeadingLabelConstraints = GridBagConstraints()
targetHeadingLabelConstraints.gridx = 0
targetHeadingLabelConstraints.gridy = 0
targetHeadingLabelConstraints.gridwidth = 4
targetHeadingLabelConstraints.anchor = GridBagConstraints.LINE_START
targetHeadingLabelConstraints.insets = insets
attackPanel.add(targetHeadingLabel, targetHeadingLabelConstraints)
startAttackButton = JButton("<html><b>Start Attack</b></html>",
actionPerformed=self._startAttack)
startAttackButtonConstraints = GridBagConstraints()
startAttackButtonConstraints.gridx = 4
startAttackButtonConstraints.gridy = 0
startAttackButtonConstraints.insets = insets
attackPanel.add(startAttackButton, startAttackButtonConstraints)
hostLabel = JLabel("Host:")
hostLabelConstraints = GridBagConstraints()
hostLabelConstraints.gridx = 0
hostLabelConstraints.gridy = 1
hostLabelConstraints.anchor = GridBagConstraints.LINE_START
hostLabelConstraints.insets = insets
attackPanel.add(hostLabel, hostLabelConstraints)
self._hostTextField = JTextField(25)
self._hostTextField.setMinimumSize(
self._hostTextField.getPreferredSize())
hostTextFieldConstraints = GridBagConstraints()
hostTextFieldConstraints.gridx = 1
hostTextFieldConstraints.gridy = 1
hostTextFieldConstraints.weightx = 1
hostTextFieldConstraints.gridwidth = 2
hostTextFieldConstraints.anchor = GridBagConstraints.LINE_START
hostTextFieldConstraints.insets = insets
attackPanel.add(self._hostTextField, hostTextFieldConstraints)
portLabel = JLabel("Port:")
portLabelConstraints = GridBagConstraints()
portLabelConstraints.gridx = 0
portLabelConstraints.gridy = 2
portLabelConstraints.anchor = GridBagConstraints.LINE_START
portLabelConstraints.insets = insets
attackPanel.add(portLabel, portLabelConstraints)
self._portTextField = JTextField(5)
self._portTextField.setMinimumSize(
self._portTextField.getPreferredSize())
portTextFieldConstraints = GridBagConstraints()
portTextFieldConstraints.gridx = 1
portTextFieldConstraints.gridy = 2
portTextFieldConstraints.gridwidth = 2
portTextFieldConstraints.anchor = GridBagConstraints.LINE_START
portTextFieldConstraints.insets = insets
attackPanel.add(self._portTextField, portTextFieldConstraints)
self._protocolCheckBox = JCheckBox("Use HTTPS")
protocolCheckBoxConstraints = GridBagConstraints()
protocolCheckBoxConstraints.gridx = 0
protocolCheckBoxConstraints.gridy = 3
protocolCheckBoxConstraints.gridwidth = 3
protocolCheckBoxConstraints.anchor = GridBagConstraints.LINE_START
protocolCheckBoxConstraints.insets = insets
attackPanel.add(self._protocolCheckBox, protocolCheckBoxConstraints)
requestHeadingLabel = JLabel("<html><b>Request</b></html>")
requestHeadingLabelConstraints = GridBagConstraints()
requestHeadingLabelConstraints.gridx = 0
requestHeadingLabelConstraints.gridy = 4
requestHeadingLabelConstraints.gridwidth = 4
requestHeadingLabelConstraints.anchor = GridBagConstraints.LINE_START
requestHeadingLabelConstraints.insets = insets
attackPanel.add(requestHeadingLabel, requestHeadingLabelConstraints)
messageEditorComponentConstraints = GridBagConstraints()
messageEditorComponentConstraints.gridx = 0
messageEditorComponentConstraints.gridy = 5
messageEditorComponentConstraints.weightx = 1
messageEditorComponentConstraints.weighty = .75
messageEditorComponentConstraints.gridwidth = 4
messageEditorComponentConstraints.gridheight = 2
messageEditorComponentConstraints.fill = GridBagConstraints.BOTH
messageEditorComponentConstraints.insets = insets
attackPanel.add(
messageEditorComponent, messageEditorComponentConstraints)
addPayloadButton = JButton(
"Add \xa7", actionPerformed=self._addPayload)
addPayloadButtonConstraints = GridBagConstraints()
addPayloadButtonConstraints.gridx = 4
addPayloadButtonConstraints.gridy = 5
addPayloadButtonConstraints.fill = GridBagConstraints.HORIZONTAL
addPayloadButtonConstraints.insets = insets
attackPanel.add(addPayloadButton, addPayloadButtonConstraints)
clearPayloadButton = JButton(
"Clear \xa7", actionPerformed=self._clearPayloads)
clearPayloadButtonConstraints = GridBagConstraints()
clearPayloadButtonConstraints.gridx = 4
clearPayloadButtonConstraints.gridy = 6
clearPayloadButtonConstraints.anchor = GridBagConstraints.PAGE_START
clearPayloadButtonConstraints.fill = GridBagConstraints.HORIZONTAL
clearPayloadButtonConstraints.insets = insets
attackPanel.add(clearPayloadButton, clearPayloadButtonConstraints)
payloadHeadingLabel = JLabel("<html><b>Payloads<b></html>")
payloadHeadingLabelConstraints = GridBagConstraints()
payloadHeadingLabelConstraints.gridx = 0
payloadHeadingLabelConstraints.gridy = 7
payloadHeadingLabelConstraints.gridwidth = 4
payloadHeadingLabelConstraints.anchor = GridBagConstraints.LINE_START
payloadHeadingLabelConstraints.insets = insets
attackPanel.add(payloadHeadingLabel, payloadHeadingLabelConstraints)
self._payloadTextArea = JTextArea()
payloadScrollPane = JScrollPane(self._payloadTextArea)
payloadScrollPaneConstraints = GridBagConstraints()
payloadScrollPaneConstraints.gridx = 0
payloadScrollPaneConstraints.gridy = 8
payloadScrollPaneConstraints.weighty = .25
payloadScrollPaneConstraints.gridwidth = 3
payloadScrollPaneConstraints.fill = GridBagConstraints.BOTH
payloadScrollPaneConstraints.insets = insets
attackPanel.add(payloadScrollPane, payloadScrollPaneConstraints)
requestsNumLabel = JLabel("Number of requests for each payload:")
requestsNumLabelConstraints = GridBagConstraints()
requestsNumLabelConstraints.gridx = 0
requestsNumLabelConstraints.gridy = 9
requestsNumLabelConstraints.gridwidth = 2
requestsNumLabelConstraints.anchor = GridBagConstraints.LINE_START
requestsNumLabelConstraints.insets = insets
attackPanel.add(requestsNumLabel, requestsNumLabelConstraints)
self._requestsNumTextField = JTextField("100", 4)
self._requestsNumTextField.setMinimumSize(
self._requestsNumTextField.getPreferredSize())
requestsNumTextFieldConstraints = GridBagConstraints()
requestsNumTextFieldConstraints.gridx = 2
requestsNumTextFieldConstraints.gridy = 9
requestsNumTextFieldConstraints.anchor = GridBagConstraints.LINE_START
requestsNumTextFieldConstraints.insets = insets
attackPanel.add(
self._requestsNumTextField, requestsNumTextFieldConstraints)
return attackPanel
def initComponents(self):
self.setLayout(GridBagLayout())
gc = GridBagConstraints()
gc.weightx = 1
gc.weighty = 1
# First Row
gc.gridy = 0
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.fill = GridBagConstraints.NONE
gc.anchor = GridBagConstraints.LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.nameLabel, gc)
gc.gridx = 1
gc.anchor = GridBagConstraints.LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.nameField, gc)
# Second Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.mobLabel, gc)
gc.gridx = 1
gc.anchor = GridBagConstraints.LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.mobField, gc)
#Third Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.FIRST_LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.emailLabel, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.emailField, gc)
#Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.FIRST_LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.vehicleLabel, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.vehicleField, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 2
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.automateRecogBtn, gc)
#Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 2.0
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
self.add(self.regBtn, gc)
innerBorder = BorderFactory.createTitledBorder('Register Vehicle')
outerBorder = BorderFactory.createEmptyBorder(5, 5, 5, 5)
self.setBorder(
BorderFactory.createCompoundBorder(outerBorder, innerBorder))
def registerExtenderCallbacks(self, this_callbacks): ### IBurpExtender
global callbacks, helpers
global extension_enable, in_scope_only
global change_method_to_post
callbacks = this_callbacks
helpers = callbacks.getHelpers()
callbacks.setExtensionName(NAME)
self.settings = JPanel(GridBagLayout())
c = GridBagConstraints()
self.extension_enable_box = JCheckBox('Enable extension',
extension_enable)
self.setFontBold(self.extension_enable_box)
self.extension_enable_box.setForeground(Color(0, 0, 153))
c.insets = Insets(5, 5, 5, 5)
c.gridx = 0
c.gridy = 0
c.gridwidth = 1
c.weightx = 1
c.fill = GridBagConstraints.NONE
c.anchor = GridBagConstraints.WEST
self.settings.add(self.extension_enable_box, c)
self.in_scope_only_box = JCheckBox('Modify only in-scope requests',
in_scope_only)
self.setFontBold(self.in_scope_only_box)
self.in_scope_only_box.setForeground(Color(0, 0, 153))
c.insets = Insets(40, 5, 5, 5)
c.gridx = 0
c.gridy = 1
self.settings.add(self.in_scope_only_box, c)
self.change_method_to_post_box = JCheckBox(
'Change HTTP method to POST', change_method_to_post)
self.setFontBold(self.change_method_to_post_box)
self.change_method_to_post_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 6
self.settings.add(self.change_method_to_post_box, c)
change_method_to_post_lbl = JLabel(
'Check to convert PUT/DELETE/PATCH method to POST in all requests.'
)
self.setFontItalic(change_method_to_post_lbl)
c.gridx = 0
c.gridy = 7
self.settings.add(change_method_to_post_lbl, c)
self.change_to_get_box = JCheckBox('Change to GET', change_to_get)
self.setFontBold(self.change_to_get_box)
self.change_to_get_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 12
self.settings.add(self.change_to_get_box, c)
change_to_get_lbl = JLabel(
'Check to convert POST/PUT/DELETE/PATCH url-encoded requests to GET.'
)
self.setFontItalic(change_to_get_lbl)
c.gridx = 0
c.gridy = 13
self.settings.add(change_to_get_lbl, c)
self.tabs = JTabbedPane()
self.tabs.addTab('Settings', self.settings)
callbacks.customizeUiComponent(self.tabs)
callbacks.addSuiteTab(self)
callbacks.registerProxyListener(self)
print "Successfully loaded %s v%s by Mohammed alsaggaf " % (NAME,
VERSION)
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
self.helpers = callbacks.getHelpers()
callbacks.setExtensionName("Session Authentication Tool")
self.out = callbacks.getStdout()
# definition of suite tab
self.tab = JPanel(GridBagLayout())
self.tabledata = MappingTableModel(callbacks)
self.table = JTable(self.tabledata)
#self.table.getColumnModel().getColumn(0).setPreferredWidth(50);
#self.table.getColumnModel().getColumn(1).setPreferredWidth(100);
self.tablecont = JScrollPane(self.table)
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.anchor = GridBagConstraints.FIRST_LINE_START
c.gridx = 0
c.gridy = 0
c.gridheight = 6
c.weightx = 0.3
c.weighty = 0.5
self.tab.add(self.tablecont, c)
c = GridBagConstraints()
c.weightx = 0.1
c.anchor = GridBagConstraints.FIRST_LINE_START
c.gridx = 1
c.gridy = 0
label_id = JLabel("Identifier:")
self.tab.add(label_id, c)
self.input_id = JTextField(20)
self.input_id.setToolTipText(
"Enter the identifier which is used by the application to identifiy a particular test user account, e.g. a numerical user id or a user name."
)
c.gridy = 1
self.tab.add(self.input_id, c)
c.gridy = 2
label_content = JLabel("Content:")
self.tab.add(label_content, c)
self.input_content = JTextField(20, actionPerformed=self.btn_add_id)
self.input_content.setToolTipText(
"Enter some content which is displayed in responses of the application and shows that the current session belongs to a particular user, e.g. the full name of the user."
)
c.gridy = 3
self.tab.add(self.input_content, c)
self.btn_add = JButton("Add/Edit Identity",
actionPerformed=self.btn_add_id)
c.gridy = 4
self.tab.add(self.btn_add, c)
self.btn_del = JButton("Delete Identity",
actionPerformed=self.btn_del_id)
c.gridy = 5
self.tab.add(self.btn_del, c)
callbacks.customizeUiComponent(self.tab)
callbacks.customizeUiComponent(self.table)
callbacks.customizeUiComponent(self.tablecont)
callbacks.customizeUiComponent(self.btn_add)
callbacks.customizeUiComponent(self.btn_del)
callbacks.customizeUiComponent(label_id)
callbacks.customizeUiComponent(self.input_id)
callbacks.addSuiteTab(self)
callbacks.registerScannerCheck(self)
callbacks.registerIntruderPayloadGeneratorFactory(self)
callbacks.registerContextMenuFactory(self)
def registerExtenderCallbacks(self, this_callbacks): ### IBurpExtender
global callbacks, helpers
global extension_enable
global remove_csrf_headers, remove_csrf_params, change_method_to_post
global change_ct_to_json, change_ct_to_plain, change_to_get
callbacks = this_callbacks
helpers = callbacks.getHelpers()
callbacks.setExtensionName(NAME)
self.settings = JPanel(GridBagLayout())
c = GridBagConstraints()
self.extension_enable_box = JCheckBox('Enable extension',
extension_enable)
self.extension_enable_box.setFont(Font("Serif", Font.BOLD, 20))
self.extension_enable_box.setForeground(Color(0, 0, 153))
c.insets = Insets(5, 5, 5, 5)
c.gridx = 0
c.gridy = 0
c.gridwidth = 1
c.weightx = 1
c.fill = GridBagConstraints.NONE
c.anchor = GridBagConstraints.WEST
self.settings.add(self.extension_enable_box, c)
self.remove_csrf_headers_box = JCheckBox('Remove CSRF headers',
remove_csrf_params)
self.remove_csrf_headers_box.setFont(Font("Serif", Font.BOLD, 20))
self.remove_csrf_headers_box.setForeground(Color(0, 0, 153))
c.insets = Insets(40, 5, 5, 5)
c.gridx = 0
c.gridy = 1
self.settings.add(self.remove_csrf_headers_box, c)
remove_csrf_headers_box_lbl = JLabel(
'Check to remove headers with CSRF tokens from all requests.')
c.insets = Insets(5, 5, 5, 5)
c.gridx = 0
c.gridy = 2
self.settings.add(remove_csrf_headers_box_lbl, c)
self.remove_csrf_params_box = JCheckBox('Remove CSRF parameters',
remove_csrf_params)
self.remove_csrf_params_box.setFont(Font("Serif", Font.BOLD, 20))
self.remove_csrf_params_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 3
self.settings.add(self.remove_csrf_params_box, c)
remove_csrf_params_box_lbl = JLabel(
'Check to remove URL/body parameters with CSRF tokens from all requests. URL-encoded, multipart, JSON parameters are supported.'
)
c.gridx = 0
c.gridy = 4
self.settings.add(remove_csrf_params_box_lbl, c)
self.change_method_to_post_box = JCheckBox(
'Change HTTP method to POST', change_method_to_post)
self.change_method_to_post_box.setFont(Font("Serif", Font.BOLD, 20))
self.change_method_to_post_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 5
self.settings.add(self.change_method_to_post_box, c)
change_method_to_post_lbl = JLabel(
'Check to convert PUT/DELETE/PATCH method to POST in all requests.'
)
c.gridx = 0
c.gridy = 6
self.settings.add(change_method_to_post_lbl, c)
self.change_ct_to_json_box = JCheckBox('Change media type to json',
change_ct_to_json)
self.change_ct_to_json_box.setFont(Font("Serif", Font.BOLD, 20))
self.change_ct_to_json_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 7
self.settings.add(self.change_ct_to_json_box, c)
change_ct_to_json_lbl = JLabel(
'Check to convert body to json and set Content-Type to application/json in url-encoded requests.'
)
c.gridx = 0
c.gridy = 8
self.settings.add(change_ct_to_json_lbl, c)
self.change_ct_to_plain_box = JCheckBox(
'Change Content-Type to text/plain', change_ct_to_plain)
self.change_ct_to_plain_box.setFont(Font("Serif", Font.BOLD, 20))
self.change_ct_to_plain_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 9
self.settings.add(self.change_ct_to_plain_box, c)
change_ct_to_plain_lbl = JLabel(
'Check to set Content-Type to text/plain in request with non-simple media type. Simple media types - application/application/x-www-form-urlencoded, text/plain, multipart/form-data.'
)
c.gridx = 0
c.gridy = 10
self.settings.add(change_ct_to_plain_lbl, c)
self.change_to_get_box = JCheckBox('Change to GET', change_to_get)
self.change_to_get_box.setFont(Font("Serif", Font.BOLD, 20))
self.change_to_get_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 11
self.settings.add(self.change_to_get_box, c)
change_to_get_lbl = JLabel(
'Check to convert POST/PUT/DELETE/PATCH url-encoded requests to GET.'
)
c.gridx = 0
c.gridy = 12
self.settings.add(change_to_get_lbl, c)
callbacks.customizeUiComponent(self.settings)
callbacks.addSuiteTab(self)
callbacks.registerProxyListener(self)
print "Successfully loaded %s v%s" % (NAME, VERSION)
def registerExtenderCallbacks(self, callbacks):
Cb(callbacks)
Cb.callbacks.setExtensionName(self.ext_name)
try:
global compuracer_communication_lock
# option picker item objects (for Java compatibility)
item1 = {'key': 'item1', 'name': '2'}
item2 = {'key': 'item2', 'name': '3'}
item3 = {'key': 'item3', 'name': '4'}
item4 = {'key': 'item4', 'name': '5'}
item5 = {'key': 'item5', 'name': '10'}
item6 = {'key': 'item6', 'name': '15'}
item7 = {'key': 'item7', 'name': '20'}
item8 = {'key': 'item8', 'name': '25'}
item9 = {'key': 'item9', 'name': '50'}
item10 = {'key': 'item10', 'name': '100'}
# main splitted pane + top pane
self._main_splitpane = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self._outer_settings_pane = JPanel(BorderLayout())
self._settings_pane = JPanel(GridBagLayout())
c = GridBagConstraints()
self.label_1 = JLabel("Number of parallel requests:")
c.fill = GridBagConstraints.NONE
c.gridx = 0
c.gridy = 0
c.insets = Insets(0, 5, 0, 10)
c.anchor = GridBagConstraints.LINE_START
self._settings_pane.add(self.label_1, c)
self.input_parallel_requests = JComboBox([
Item(item1),
Item(item2),
Item(item3),
Item(item4),
Item(item5),
Item(item6),
Item(item7),
Item(item8),
Item(item9),
Item(item10)
])
self.input_parallel_requests.setSelectedIndex(4)
self.input_parallel_requests.setToolTipText(
"Select the number of parallel requests that will be sent")
self.input_parallel_requests.addActionListener(
self.change_parallel_requests)
c.gridx = 1
c.gridy = 0
c.insets = Insets(0, 5, 0, 10)
self._settings_pane.add(self.input_parallel_requests, c)
self.option_allow_redirects = JCheckBox(
"Allow redirects", actionPerformed=self.check_allow_redirects)
self.option_allow_redirects.setToolTipText(
"Select whether redirect responses are followed")
c.gridx = 2
c.gridy = 0
c.insets = Insets(0, 20, 0, 10)
self._settings_pane.add(self.option_allow_redirects, c)
self.option_sync_last_byte = JCheckBox(
"Sync last byte", actionPerformed=self.check_sync_last_byte)
self.option_sync_last_byte.setToolTipText(
"Select whether last byte synchronisation is enabled")
c.gridx = 2
c.gridy = 1
c.insets = Insets(0, 20, 0, 0)
self._settings_pane.add(self.option_sync_last_byte, c)
self.label_2 = JLabel("Send timeout in seconds:")
c.gridx = 0
c.gridy = 1
c.insets = Insets(0, 5, 0, 0)
self._settings_pane.add(self.label_2, c)
self.input_send_timeout = JComboBox([
Item(item2),
Item(item4),
Item(item5),
Item(item7),
Item(item9),
Item(item10)
])
self.input_send_timeout.setSelectedIndex(3)
self.input_send_timeout.setToolTipText(
"Select the wait-for-response timeout after sending the request(s)"
)
self.input_send_timeout.addActionListener(self.change_send_timeout)
c.gridx = 1
c.gridy = 1
c.insets = Insets(0, 5, 0, 0)
self._settings_pane.add(self.input_send_timeout, c)
self.button_resend_batch = JButton("Resend requests")
self.button_resend_batch.setToolTipText(
"Resend all requests with the current configuration")
self.button_resend_batch.setEnabled(False)
self.button_resend_batch.addActionListener(
MenuFactory.start_request_transmitter_button)
c.gridx = 3
c.gridy = 0
c.insets = Insets(0, 20, 0, 10)
self._settings_pane.add(self.button_resend_batch, c)
immediate_data_ui_elements[
"parallel_requests"] = self.input_parallel_requests
immediate_data_ui_elements[
"allow_redirects"] = self.option_allow_redirects
immediate_data_ui_elements[
"sync_last_byte"] = self.option_sync_last_byte
immediate_data_ui_elements[
"send_timeout"] = self.input_send_timeout
immediate_data_ui_elements[
"resend_batch"] = self.button_resend_batch
c = GridBagConstraints()
c.anchor = GridBagConstraints.WEST
self._outer_settings_pane.add(self._settings_pane,
BorderLayout.WEST)
self._main_splitpane.setTopComponent(self._outer_settings_pane)
self._results_splitpane = JSplitPane(JSplitPane.HORIZONTAL_SPLIT)
self._main_splitpane.setBottomComponent(self._results_splitpane)
# table of log entries
self.tabs_right = JTabbedPane()
global _textEditors, DEFAULT_RESULTS
for i in range(3):
_textEditors.append(Cb.callbacks.createTextEditor())
_textEditors[-1].setText(str.encode("\n" + DEFAULT_RESULTS))
self.tabs_right.add("Summary", _textEditors[0].getComponent())
self.tabs_right.add("Full result", _textEditors[1].getComponent())
self.tabs_right.add("Config", _textEditors[2].getComponent())
self._results_splitpane.setRightComponent(self.tabs_right)
# tabs with request/response viewers
global _requestViewers, _requestPane
_requestPane = JTabbedPane()
_requestViewers.append(
Cb.callbacks.createMessageEditor(None, False))
_requestPane.addTab("Request", _requestViewers[-1].getComponent())
self._results_splitpane.setLeftComponent(_requestPane)
# customize our UI components
Cb.callbacks.customizeUiComponent(self._settings_pane)
Cb.callbacks.customizeUiComponent(self.tabs_right)
Cb.callbacks.customizeUiComponent(_requestPane)
# add the custom tab to Burp's UI
Cb.callbacks.addSuiteTab(self)
except RuntimeException as e:
callbacks.printError(traceback.format_exc())
e = PyException(e)
print("10")
print(str(self))
print("{}\t{}\n{}\n".format(e.type, e.value, e.traceback))
Cb.callbacks.registerContextMenuFactory(MenuFactory())
callbacks.registerExtensionStateListener(self)
self.start_alive_checker()
Cb.callbacks.printOutput('%s v%s extension loaded\n' %
(self.ext_name, self.ext_version))
def registerExtenderCallbacks(self, this_callbacks): ### IBurpExtender
global callbacks, helpers
global extension_enable, in_scope_only
global remove_csrf_headers, remove_csrf_params, change_method_to_post
global change_ct_to_json, change_ct_to_plain, change_to_get
callbacks = this_callbacks
helpers = callbacks.getHelpers()
callbacks.setExtensionName(NAME)
self.settings = JPanel(GridBagLayout())
c = GridBagConstraints()
self.extension_enable_box = JCheckBox('Enable extension',
extension_enable)
self.setFontBold(self.extension_enable_box)
self.extension_enable_box.setForeground(Color(0, 0, 153))
c.insets = Insets(5, 5, 5, 5)
c.gridx = 0
c.gridy = 0
c.gridwidth = 1
c.weightx = 1
c.fill = GridBagConstraints.NONE
c.anchor = GridBagConstraints.WEST
self.settings.add(self.extension_enable_box, c)
self.in_scope_only_box = JCheckBox('Modify only in-scope requests',
in_scope_only)
self.setFontBold(self.in_scope_only_box)
self.in_scope_only_box.setForeground(Color(0, 0, 153))
c.insets = Insets(40, 5, 5, 5)
c.gridx = 0
c.gridy = 1
self.settings.add(self.in_scope_only_box, c)
self.remove_csrf_headers_box = JCheckBox('Remove CSRF headers',
remove_csrf_params)
self.setFontBold(self.remove_csrf_headers_box)
self.remove_csrf_headers_box.setForeground(Color(0, 0, 153))
c.insets = Insets(40, 5, 5, 5)
c.gridx = 0
c.gridy = 2
self.settings.add(self.remove_csrf_headers_box, c)
remove_csrf_headers_box_lbl = JLabel(
'Check to remove headers with CSRF tokens from all requests.')
self.setFontItalic(remove_csrf_headers_box_lbl)
c.insets = Insets(5, 5, 5, 5)
c.gridx = 0
c.gridy = 3
self.settings.add(remove_csrf_headers_box_lbl, c)
self.remove_csrf_params_box = JCheckBox('Remove CSRF parameters',
remove_csrf_params)
self.setFontBold(self.remove_csrf_params_box)
self.remove_csrf_params_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 4
self.settings.add(self.remove_csrf_params_box, c)
remove_csrf_params_box_lbl = JLabel(
'Check to remove URL/body parameters with CSRF tokens from all requests. URL-encoded, multipart, JSON parameters are supported.'
)
self.setFontItalic(remove_csrf_params_box_lbl)
c.gridx = 0
c.gridy = 5
self.settings.add(remove_csrf_params_box_lbl, c)
self.change_method_to_post_box = JCheckBox(
'Change HTTP method to POST', change_method_to_post)
self.setFontBold(self.change_method_to_post_box)
self.change_method_to_post_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 6
self.settings.add(self.change_method_to_post_box, c)
change_method_to_post_lbl = JLabel(
'Check to convert PUT/DELETE/PATCH method to POST in all requests.'
)
self.setFontItalic(change_method_to_post_lbl)
c.gridx = 0
c.gridy = 7
self.settings.add(change_method_to_post_lbl, c)
self.change_ct_to_json_box = JCheckBox('Change media type to json',
change_ct_to_json)
self.setFontBold(self.change_ct_to_json_box)
self.change_ct_to_json_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 8
self.settings.add(self.change_ct_to_json_box, c)
change_ct_to_json_lbl = JLabel(
'Check to convert body to json and set Content-Type to application/json in url-encoded requests.'
)
self.setFontItalic(change_ct_to_json_lbl)
c.gridx = 0
c.gridy = 9
self.settings.add(change_ct_to_json_lbl, c)
self.change_ct_to_plain_box = JCheckBox(
'Change Content-Type to text/plain', change_ct_to_plain)
self.setFontBold(self.change_ct_to_plain_box)
self.change_ct_to_plain_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 10
self.settings.add(self.change_ct_to_plain_box, c)
change_ct_to_plain_lbl = JLabel(
'Check to set Content-Type to text/plain in request with non-simple media type. Simple media types - application/application/x-www-form-urlencoded, text/plain, multipart/form-data.'
)
self.setFontItalic(change_ct_to_plain_lbl)
c.gridx = 0
c.gridy = 11
self.settings.add(change_ct_to_plain_lbl, c)
self.change_to_get_box = JCheckBox('Change to GET', change_to_get)
self.setFontBold(self.change_to_get_box)
self.change_to_get_box.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 12
self.settings.add(self.change_to_get_box, c)
change_to_get_lbl = JLabel(
'Check to convert POST/PUT/DELETE/PATCH url-encoded requests to GET.'
)
self.setFontItalic(change_to_get_lbl)
c.gridx = 0
c.gridy = 13
self.settings.add(change_to_get_lbl, c)
self.csrf_headers_params = JPanel(GridBagLayout())
c = GridBagConstraints()
lblParams = JLabel("CSRF parameters:")
self.setFontBold(lblParams)
lblParams.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 0
c.insets = Insets(5, 5, 5, 5)
c.fill = GridBagConstraints.NONE
c.anchor = GridBagConstraints.FIRST_LINE_END
self.csrf_headers_params.add(lblParams, c)
self.csrf_param_text_field = JTextField()
c.fill = GridBagConstraints.BOTH
c.gridx = 1
c.gridy = 0
self.csrf_headers_params.add(self.csrf_param_text_field, c)
lblParamsNote = JLabel(
"Remove parameter from request if name contains")
self.setFontItalic(lblParamsNote)
c.fill = GridBagConstraints.NONE
c.gridx = 0
c.gridy = 1
self.csrf_headers_params.add(lblParamsNote, c)
self.csrf_params_text_area = JTextArea()
self.csrf_params_text_area.setColumns(20)
self.csrf_params_text_area.setRows(10)
self.csrf_params_text_area.setEditable(False)
c.fill = GridBagConstraints.BOTH
self.csrf_params_mouse_listener = TextAreaMouseListener(
self.csrf_params_text_area)
self.csrf_params_text_area.addMouseListener(
self.csrf_params_mouse_listener)
for name in csrf_params_names:
self.csrf_params_text_area.append(name + os.linesep)
c.gridx = 1
c.gridy = 1
sp = JScrollPane(self.csrf_params_text_area)
self.csrf_headers_params.add(sp, c)
buttonsPanel = JPanel(GridBagLayout())
_c = GridBagConstraints()
_c.insets = Insets(3, 3, 3, 3)
_c.gridx = 0
_c.gridy = 0
_c.fill = GridBagConstraints.BOTH
_c.weightx = 1
_c.gridwidth = 1
handlers = ButtonHandlers(self.csrf_param_text_field,
self.csrf_params_text_area,
self.csrf_params_mouse_listener,
csrf_params_names)
self.csrf_param_add_button = JButton(
'Add', actionPerformed=handlers.handler_add)
self.csrf_param_rm_button = JButton(
'Remove', actionPerformed=handlers.handler_rm)
self.csrf_param_restore_button = JButton(
'Restore', actionPerformed=handlers.handler_restore)
buttonsPanel.add(self.csrf_param_add_button, _c)
_c.gridy = 1
buttonsPanel.add(self.csrf_param_rm_button, _c)
_c.gridy = 2
buttonsPanel.add(self.csrf_param_restore_button, _c)
_c.gridy = 3
c.gridx = 2
c.gridy = 1
c.fill = GridBagConstraints.NONE
self.csrf_headers_params.add(buttonsPanel, c)
lblHeaders = JLabel("CSRF headers:")
self.setFontBold(lblHeaders)
lblHeaders.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 2
c.insets = Insets(40, 5, 5, 5)
c.fill = GridBagConstraints.NONE
c.anchor = GridBagConstraints.FIRST_LINE_END
self.csrf_headers_params.add(lblHeaders, c)
self.csrf_header_text_field = JTextField()
c.fill = GridBagConstraints.BOTH
c.gridx = 1
c.gridy = 2
self.csrf_headers_params.add(self.csrf_header_text_field, c)
lblHeadersNote = JLabel("Remove header from request if name equals to")
self.setFontItalic(lblHeadersNote)
c.fill = GridBagConstraints.NONE
c.insets = Insets(5, 5, 5, 5)
c.gridx = 0
c.gridy = 3
self.csrf_headers_params.add(lblHeadersNote, c)
self.csrf_headers_text_area = JTextArea()
self.csrf_headers_text_area.setColumns(20)
self.csrf_headers_text_area.setRows(10)
self.csrf_headers_text_area.setEditable(False)
c.fill = GridBagConstraints.BOTH
self.csrf_headers_mouse_listener = TextAreaMouseListener(
self.csrf_headers_text_area)
self.csrf_headers_text_area.addMouseListener(
self.csrf_headers_mouse_listener)
for name in csrf_headers_names:
self.csrf_headers_text_area.append(name + os.linesep)
c.gridx = 1
c.gridy = 3
sp = JScrollPane(self.csrf_headers_text_area)
self.csrf_headers_params.add(sp, c)
buttonsPanel = JPanel(GridBagLayout())
_c = GridBagConstraints()
_c.insets = Insets(3, 3, 3, 3)
_c.gridx = 0
_c.gridy = 0
_c.fill = GridBagConstraints.BOTH
_c.weightx = 1
_c.gridwidth = 1
handlers = ButtonHandlers(self.csrf_header_text_field,
self.csrf_headers_text_area,
self.csrf_headers_mouse_listener,
csrf_headers_names)
self.csrf_header_add_button = JButton(
'Add', actionPerformed=handlers.handler_add)
self.csrf_header_rm_button = JButton(
'Remove', actionPerformed=handlers.handler_rm)
self.csrf_header_restore_button = JButton(
'Restore', actionPerformed=handlers.handler_restore)
buttonsPanel.add(self.csrf_header_add_button, _c)
_c.gridy = 1
buttonsPanel.add(self.csrf_header_rm_button, _c)
_c.gridy = 2
buttonsPanel.add(self.csrf_header_restore_button, _c)
_c.gridy = 3
c.gridx = 2
c.gridy = 3
c.fill = GridBagConstraints.NONE
self.csrf_headers_params.add(buttonsPanel, c)
self.whitelist = JPanel(GridBagLayout())
c = GridBagConstraints()
lblWhitelist = JLabel("URLs whitelist:")
self.setFontBold(lblWhitelist)
lblWhitelist.setForeground(Color(0, 0, 153))
c.gridx = 0
c.gridy = 0
c.insets = Insets(5, 5, 5, 5)
c.fill = GridBagConstraints.NONE
c.anchor = GridBagConstraints.FIRST_LINE_END
self.whitelist.add(lblWhitelist, c)
self.whitelist_text_field = JTextField()
c.fill = GridBagConstraints.BOTH
c.gridx = 1
c.gridy = 0
self.whitelist.add(self.whitelist_text_field, c)
lblWhitelistNote = JLabel(
"Do not perform request modification if URL starts with")
self.setFontItalic(lblWhitelistNote)
c.fill = GridBagConstraints.NONE
c.gridx = 0
c.gridy = 1
self.whitelist.add(lblWhitelistNote, c)
self.whitelist_text_area = JTextArea()
self.whitelist_text_area.setColumns(30)
self.whitelist_text_area.setRows(10)
self.whitelist_text_area.setEditable(False)
c.fill = GridBagConstraints.BOTH
self.whitelist_mouse_listener = TextAreaMouseListener(
self.whitelist_text_area)
self.whitelist_text_area.addMouseListener(
self.whitelist_mouse_listener)
c.gridx = 1
c.gridy = 1
sp = JScrollPane(self.whitelist_text_area)
self.whitelist.add(sp, c)
buttonsPanel = JPanel(GridBagLayout())
_c = GridBagConstraints()
_c.insets = Insets(3, 3, 3, 3)
_c.gridx = 0
_c.gridy = 0
_c.fill = GridBagConstraints.BOTH
_c.weightx = 1
_c.gridwidth = 1
handlers = ButtonHandlers(self.whitelist_text_field,
self.whitelist_text_area,
self.whitelist_mouse_listener, [])
self.whitelist_add_button = JButton(
'Add', actionPerformed=handlers.handler_add)
self.whitelist_rm_button = JButton('Remove',
actionPerformed=handlers.handler_rm)
self.whitelist_clear_button = JButton(
'Clear', actionPerformed=handlers.handler_restore)
buttonsPanel.add(self.whitelist_add_button, _c)
_c.gridy = 1
buttonsPanel.add(self.whitelist_rm_button, _c)
_c.gridy = 2
buttonsPanel.add(self.whitelist_clear_button, _c)
_c.gridy = 3
c.gridx = 2
c.gridy = 1
c.fill = GridBagConstraints.NONE
self.whitelist.add(buttonsPanel, c)
self.tabs = JTabbedPane()
self.tabs.addTab('Settings', self.settings)
self.tabs.addTab('CSRF params/headers to remove',
self.csrf_headers_params)
self.tabs.addTab('Requests whitelist', self.whitelist)
callbacks.customizeUiComponent(self.tabs)
callbacks.addSuiteTab(self)
callbacks.registerProxyListener(self)
callbacks.registerContextMenuFactory(
SendToWhitelist(self.whitelist_text_area))
print "Successfully loaded %s v%s by Mikhail Egorov @0ang3el" % (
NAME, VERSION)
def getArguments(self):
"""
This function brings up a window to retreive any required arguments. It uses a window with fields for each argument, filled with any arguments already given.
While this window is visible the program will wait, once it is no longer visible all the arguments will be filled with the entries in the fields.
"""
class LocationAction(AbstractAction):
"""
Action to set the text of a text field to a folder or directory.
"""
def __init__(self, field):
AbstractAction.__init__(self, "...")
self.field = field
def actionPerformed(self, event):
fileChooser = JFileChooser()
fileChooser.fileSelectionMode = JFileChooser.FILES_AND_DIRECTORIES
if fileChooser.showOpenDialog(None) == JFileChooser.APPROVE_OPTION:
self.field.text = fileChooser.selectedFile.absolutePath
class HelpAction(AbstractAction):
"""
Displays a help page in a web browser.
"""
def __init__(self):
AbstractAction.__init__(self, "Help")
def actionPerformed(self, event):
browsers = ["google-chrome", "firefox", "opera", "epiphany", "konqueror", "conkeror", "midori", "kazehakase", "mozilla"]
osName = System.getProperty("os.name")
helpHTML = ClassLoader.getSystemResource("help.html").toString()
if osName.find("Mac OS") == 0:
Class.forName("com.apple.eio.FileManager").getDeclaredMethod( "openURL", [String().getClass()]).invoke(None, [helpHTML])
elif osName.find("Windows") == 0:
Runtime.getRuntime().exec( "rundll32 url.dll,FileProtocolHandler " + helpHTML)
else:
browser = None
for b in browsers:
if browser == None and Runtime.getRuntime().exec(["which", b]).getInputStream().read() != -1:
browser = b
Runtime.getRuntime().exec([browser, helpHTML])
class OKAction(AbstractAction):
"""
Action for starting the pipeline. This action will simply make the window invisible.
"""
def __init__(self):
AbstractAction.__init__(self, "Ok")
def actionPerformed(self, event):
frame.setVisible(False)
class CancelAction(AbstractAction):
"""
Action for canceling the pipeline. Exits the program.
"""
def __init__(self):
AbstractAction.__init__(self, "Cancel")
def actionPerformed(self, event):
sys.exit(0)
frame = JFrame("Neofelis")
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
constraints = GridBagConstraints()
contentPane = JPanel(GridBagLayout())
frame.setContentPane(contentPane)
blastField = JTextField(self.blastLocation)
genemarkField = JTextField(self.genemarkLocation)
transtermField = JTextField(self.transtermLocation)
tRNAscanField = JTextField(self.tRNAscanLocation)
databaseLocationField = JTextField(os.path.split(self.database)[0])
databaseField = JTextField(os.path.split(self.database)[1])
matrixField = JTextField(str(self.matrix))
eValueField = JTextField(str(self.eValue))
minLengthField = JTextField(str(self.minLength))
scaffoldingDistanceField = JTextField(str(self.scaffoldingDistance))
promoterScoreField = JTextField(str(self.promoterScoreCutoff))
queryField = JTextField(self.sources[0])
constraints.gridx = 0
constraints.gridy = 0
constraints.gridwidth = 1
constraints.gridheight = 1
constraints.fill = GridBagConstraints.HORIZONTAL
constraints.weightx = 0
constraints.weighty = 0
contentPane.add(JLabel("Blast Location"), constraints)
constraints.gridy = 1
contentPane.add(JLabel("Genemark Location"), constraints)
constraints.gridy = 2
contentPane.add(JLabel("Transterm Location"), constraints)
constraints.gridy = 3
contentPane.add(JLabel("tRNAscan Location"), constraints)
constraints.gridy = 4
contentPane.add(JLabel("Databases Location"), constraints)
constraints.gridy = 5
contentPane.add(JLabel("Database"), constraints)
constraints.gridy = 6
contentPane.add(JLabel("Matrix(Leave blank to use heuristic matrix)"), constraints)
constraints.gridy = 7
contentPane.add(JLabel("E Value"), constraints)
constraints.gridy = 8
contentPane.add(JLabel("Minimum Intergenic Length"), constraints)
constraints.gridy = 9
contentPane.add(JLabel("Scaffold Distance"), constraints)
constraints.gridy = 10
contentPane.add(JLabel("Promoter Score Cutoff"), constraints)
constraints.gridy = 11
contentPane.add(JLabel("Query"), constraints)
constraints.gridx = 1
constraints.gridy = 0
constraints.weightx = 1
contentPane.add(blastField, constraints)
constraints.gridy = 1
contentPane.add(genemarkField, constraints)
constraints.gridy = 2
contentPane.add(transtermField, constraints)
constraints.gridy = 3
contentPane.add(tRNAscanField, constraints)
constraints.gridy = 4
contentPane.add(databaseLocationField, constraints)
constraints.gridy = 5
contentPane.add(databaseField, constraints)
constraints.gridy = 6
contentPane.add(matrixField, constraints)
constraints.gridy = 7
contentPane.add(eValueField, constraints)
constraints.gridy = 8
contentPane.add(minLengthField, constraints)
constraints.gridy = 9
contentPane.add(scaffoldingDistanceField, constraints)
constraints.gridy = 10
contentPane.add(promoterScoreField, constraints)
constraints.gridy = 11
contentPane.add(queryField, constraints)
constraints.gridx = 2
constraints.gridy = 0
constraints.weightx = 0
constraints.fill = GridBagConstraints.NONE
constraints.anchor = GridBagConstraints.LINE_END
contentPane.add(JButton(LocationAction(blastField)), constraints)
constraints.gridy = 1
contentPane.add(JButton(LocationAction(genemarkField)), constraints)
constraints.gridy = 2
contentPane.add(JButton(LocationAction(transtermField)), constraints)
constraints.gridy = 3
contentPane.add(JButton(LocationAction(tRNAscanField)), constraints)
constraints.gridy = 4
contentPane.add(JButton(LocationAction(databaseLocationField)), constraints)
constraints.gridy = 11
contentPane.add(JButton(LocationAction(queryField)), constraints)
constraints.gridx = 0
constraints.gridy = 12
constraints.anchor = GridBagConstraints.LINE_START
contentPane.add(JButton(HelpAction()), constraints)
constraints.gridx = 1
constraints.anchor = GridBagConstraints.LINE_END
contentPane.add(JButton(OKAction()), constraints)
constraints.gridx = 2
contentPane.add(JButton(CancelAction()), constraints)
frame.pack()
frame.setLocationRelativeTo(None)
frame.setVisible(True)
while frame.isVisible():
pass
self.blastLocation = blastField.getText()
self.genemarkLocation = genemarkField.getText()
self.transtermLocation = transtermField.getText()
self.database = databaseLocationField.getText() + "/" + databaseField.getText()
self.matrix = matrixField.getText()
self.eValue = float(eValueField.getText())
self.minLength = int(minLengthField.getText())
self.scaffoldingDistance = int(scaffoldingDistanceField.getText())
self.promoterScoreCutoff = float(promoterScoreField.getText())
self.sources = [queryField.getText()]
def get_gbc():
gbc = GridBagConstraints()
gbc.weightx = 1.0
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.ipady = 2
return gbc
t.setPriority(Thread.NORM_PRIORITY)
t.start()
# Create the UI: a 3-column table and a text area next to it
# to show the Motivation column of any selected row:
all = JPanel()
gb = GridBagLayout()
all.setLayout(gb)
c = GridBagConstraints()
table = JTable(TableModel())
table.setSelectionMode(ListSelectionModel.SINGLE_SELECTION)
table.setAutoCreateRowSorter(True) # to sort the view only, not the data in the underlying TableModel
jsp = JScrollPane(table)
jsp.setPreferredSize(Dimension(200, 500))
c.anchor = GridBagConstraints.NORTHWEST
c.fill = GridBagConstraints.BOTH # resize with the frame
gb.setConstraints(jsp, c)
all.add(jsp)
c.gridx = 1
c.weightx = 1.0 # take any additionally available horizontal space
c.weighty = 1.0
textarea = JTextArea()
textarea.setLineWrap(True)
textarea.setWrapStyleWord(True) # wrap text by cutting at whitespace
textarea.setEditable(False) # avoid changing the text, as any changes wouldn't be persisted to disk
font = textarea.getFont().deriveFont(20.0)
textarea.setFont(font)
textarea.setPreferredSize(Dimension(500, 500))
gb.setConstraints(textarea, c)
all.add(textarea)
def makeTranslationUI(affines, imp, show=True, print_button_text="Print transforms"):
"""
A GUI to control the translation components of a list of AffineTransform3D instances.
When updated, the ImagePlus is refreshed.
affines: a list (will be read multiple times) of one affine transform per image.
imp: the ImagePlus that hosts the virtual stack with the transformed images.
show: defaults to True, whether to make the GUI visible.
print_button_text: whatever you want the print button to read like, defaults to "Print transforms".
Returns the JFrame, the main JPanel and the lower JButton panel.
"""
panel = JPanel()
panel.setBorder(BorderFactory.createEmptyBorder(10,10,10,10))
gb = GridBagLayout()
panel.setLayout(gb)
gc = GBC()
gc.anchor = GBC.WEST
gc.fill = GBC.NONE
# Column labels
gc.gridy = 0
for i, title in enumerate(["Camera", "translation X", "translation Y", "translation Z"]):
gc.gridx = i
gc.anchor = GBC.CENTER
label = JLabel(title)
gb.setConstraints(label, gc)
panel.add(label)
gc.anchor = GBC.WEST
listeners = []
# One row per affine to control: skip the first
for i, affine in enumerate(affines[1:]):
gc.gridx = 0
gc.gridy += 1
label = JLabel("CM0%i: " % (i + 1))
gb.setConstraints(label, gc)
panel.add(label)
# One JTextField per dimension
for dimension, translation in enumerate(affine.getTranslation()):
tf = JTextField(str(translation), 10)
listener = MatrixTextFieldListener(affine, dimension, tf, imp)
listeners.append(listener)
imp.addImageListener(listener) # to disable the JTextField when closed
tf.addKeyListener(listener)
tf.addMouseWheelListener(listener)
gc.gridx += 1
gb.setConstraints(tf, gc)
panel.add(tf)
# Documentation for the user
help_lines = ["Type a number and push enter,",
"or use the scroll wheel."]
gc.gridx = 0
gc.gridwidth = 4
for line in help_lines:
gc.gridy += 1
help = JLabel(line)
gb.setConstraints(help, gc)
panel.add(help)
# Buttons
printButton = JButton(print_button_text)
def printTransforms(event):
for i, aff in enumerate(affines): # print all, including the first
matrix = zeros(12, 'd')
aff.toArray(matrix)
msg = "# Coarse affine matrix " + str(i) + ": \n" + \
"affine" + str(i) + ".set(*[%f, %f, %f, %f,\n %f, %f, %f, %f,\n %f, %f, %f, %f])" % tuple(matrix.tolist())
# Print everywhere
print msg
IJ.log(msg)
System.out.println(msg)
printButton.addActionListener(printTransforms)
gc.gridx = 0
gc.gridy += 1
gc.gridwidth = 4
button_panel = JPanel()
button_panel.add(printButton)
gb.setConstraints(button_panel, gc)
panel.add(button_panel)
frame = JFrame("Translation control")
frame.setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE)
frame.addWindowListener(CloseControl(destroyables=listeners))
frame.getContentPane().add(panel)
frame.pack()
frame.setLocationRelativeTo(None) # center in the screen
frame.setVisible(show)
return frame, panel, button_panel
def registerExtenderCallbacks(self, callbacks):
endianVal = 0
def generateTextBox(defaultTxt, editable):
sample = JTextArea(5, 20)
scrollPane1 = JScrollPane(sample)
sample.setBounds(0, 0, 400, 400)
sample.setEditable(editable)
sample.setLineWrap(True)
sample.setWrapStyleWord(True)
sample.setBorder(BorderFactory.createLineBorder(Color.BLACK))
sample.setText(defaultTxt)
return sample
def decodeData(event):
#Get Base64 token
full_str = base64.b64decode(self._left_tb1.getText())
sha_mac = full_str[44:64].encode('hex')
inflate_data = full_str[76:]
data = zlib.decompress(inflate_data)
user_length = data[20]
loc = 21
tokenDetails = "User name :" + data[
loc:loc + int(user_length.encode('hex'), 16)].replace(
"\x00", "") + "\n"
loc = loc + int(user_length.encode('hex'), 16)
lang_length = data[loc]
loc = loc + 1
tokenDetails += "Lang Code :" + data[
loc:loc + int(lang_length.encode('hex'), 16)].replace(
"\x00", "") + "\n"
tmp1.setText(data[loc:loc +
int(lang_length.encode('hex'), 16)].replace(
"\x00", ""))
loc = loc + int(lang_length.encode('hex'), 16)
node_length = data[loc]
loc = loc + 1
tokenDetails += "Node name :" + data[
loc:loc + int(node_length.encode('hex'), 16)].replace(
"\x00", "") + "\n"
tmp2.setText(data[loc:loc +
int(node_length.encode('hex'), 16)].replace(
"\x00", ""))
loc = loc + int(node_length.encode('hex'), 16)
time_length = data[loc]
loc = loc + 1
tokenDetails += "Creation time :" + data[
loc:loc + int(time_length.encode('hex'), 16)].replace(
"\x00", "")
tmp = data[loc:loc + int(time_length.encode('hex'), 16)].replace(
"\x00", "")
datestamp = tmp[:len(tmp) - 7]
tmp3.setText(datestamp)
# Determine if it's little or big endian
if (data[4:8].encode('hex') == '04030201'):
endianVal = 0
else:
endianVal = 1
tmp4.setText(str(endianVal))
hashcatFormat = sha_mac + ":" + data.encode("hex")
left_tb2.setText(tokenDetails)
left_tb3.setText(hashcatFormat)
def make_field(part, size):
part = chr(len(part) + size) + part
return part
def generateToken(event):
newtoken = ""
endianVal = int(tmp4.getText())
if endianVal == 1:
username = right_tb2.getText().encode('utf_16_be')
nodepw = right_tb1.getText().encode('utf_16_le')
lang = tmp1.getText().encode('utf_16_be')
nodename = tmp2.getText().encode('utf_16_be')
datestamp = (tmp3.getText() + '.164266').encode('utf_16_be')
token_ver = "8.10".encode('utf_16_be')
unknown_field = "N".encode('utf_16_be')
uncompressed_data = '\x01\x02\x03\x04\x00\x01\x00\x00\x00\x00\x02\xbc\x00\x00\x00\x00' + make_field(
username, 0) + make_field(lang, 0) + make_field(
nodename, 0) + make_field(datestamp, 0) + '\x00'
uncompressed_field = '\x00\x00\x00' + make_field(
uncompressed_data, 4)
inflate_data = zlib.compress(uncompressed_field)
sha1_mac = hashlib.sha1(uncompressed_field + nodepw).digest()
uncompressed_length = chr(len(uncompressed_field))
static_headers1 = '\x01\x02\x03\x04\x00\x01\x00\x00\x00\x00\x02\xbc\x00\x00\x00\x00\x00\x00\x00\x2c\x00\x04\x53\x68\x64\x72\x02' + unknown_field + uncompressed_length + '\x08' + token_ver + '\x14'
static_headers2 = '\x00\x05\x53\x64\x61\x74\x61'
body = '\x00\x00\x00' + make_field(
static_headers2 + make_field(inflate_data, 0), 4)
token = '\x00\x00\x00' + make_field(
static_headers1 + sha1_mac + body, 4)
newtoken = base64.b64encode(token)
elif endianVal == 0:
username = right_tb2.getText().encode('utf_16_le')
nodepw = right_tb1.getText().encode('utf_16_le')
lang = tmp1.getText().encode('utf_16_le')
nodename = tmp2.getText().encode('utf_16_le')
datestamp = (tmp3.getText() + '.999543').encode('utf_16_le')
token_ver = "8.10".encode('utf_16_le')
unknown_field = "N".encode('utf_16_le')
uncompressed_data = '\x00\x00\x00\x04\x03\x02\x01\x01\x00\x00\x00\xbc\x02\x00\x00\x00\x00\x00\x00' + make_field(
username, 0) + make_field(lang, 0) + make_field(
nodename, 0) + make_field(datestamp, 0) + '\x00'
uncompressed_field = make_field(uncompressed_data, 1)
inflate_data = zlib.compress(uncompressed_field)
sha1_mac = hashlib.sha1(uncompressed_field + nodepw).digest()
uncompressed_length = chr(len(uncompressed_field))
static_headers1 = '\x00\x00\x00\x04\x03\x02\x01\x01\x00\x00\x00\xbc\x02\x00\x00\x00\x00\x00\x00\x2c\x00\x00\x00\x04\x00\x53\x68\x64\x72\x02' + unknown_field + uncompressed_length + '\x08' + token_ver + '\x14'
static_headers2 = '\x00\x00\x00\x05\x00\x53\x64\x61\x74\x61'
body = make_field(
static_headers2 + make_field(inflate_data, 0), 1)
token = make_field(static_headers1 + sha1_mac + body, 1)
newtoken = base64.b64encode(token)
right_tb3.setText("PS_TOKEN=" + newtoken + ";")
# keep a reference to our callbacks object
self._callbacks = callbacks
# obtain an extension helpers object
self._helpers = callbacks.getHelpers()
# set our extension name
callbacks.setExtensionName("PeopleSoft PSToken Processor")
# main split pane
self._splitpane = JSplitPane(JSplitPane.HORIZONTAL_SPLIT)
self._splitpane.setResizeWeight(0.5)
c = GridBagConstraints()
c.weightx = 1
c.weighty = 1
c.anchor = GridBagConstraints.NORTHWEST
#Temp variables
tmp1 = JLabel("tmp")
tmp2 = JLabel("tmp")
tmp3 = JLabel("tmp")
tmp4 = JLabel("tmp")
# add left panel
panel1 = JPanel(GridBagLayout())
header1 = JLabel("EXTRACTION")
header1.setFont(Font("Myriad Pro", Font.BOLD, 24))
left_t1 = JLabel("PS_TOKEN Cookie")
left_t2 = JLabel("Token Details")
left_t3 = JLabel("Token Hash + Salt (Hashcat Format)")
left_t4 = JLabel(
"Save this into a .hash file and run the following Hashcat command: hashcat -m 13500 <hashfile> <dictionary file>"
)
self._left_tb1 = generateTextBox("Your PS_TOKEN value here", True)
left_tb2 = generateTextBox("Token Details here", False)
left_tb3 = generateTextBox("Hashcat format here", False)
btnDecode = JButton("Decode", actionPerformed=decodeData)
btnGenerate = JButton("Generate", actionPerformed=generateToken)
#add right panel
panel2 = JPanel(GridBagLayout())
header2 = JLabel("GENERATION")
header2.setFont(Font("Myriad Pro", Font.BOLD, 24))
right_t1 = JLabel("Node password")
right_t2 = JLabel("New username")
right_t3 = JLabel("New Base64 PS_TOKEN")
right_t4 = JLabel(
"Match & Replace rule to modify PS_TOKEN (Type: Request Header, enable regex)"
)
right_t5 = JLabel("Match rule: PS_TOKEN=[A-Za-z0-9\/\+]*;")
right_t6 = JLabel("Replace: PS_TOKEN=<new generated PS_TOKEN>;")
right_tb1 = generateTextBox("Password here", True)
right_tb2 = generateTextBox("PSADMIN", True)
right_tb3 = generateTextBox("Your new token here", False)
panel1.add(header1, c)
panel2.add(header2, c)
c.gridx = 0
c.gridy = 1
panel1.add(left_t1, c)
panel2.add(right_t1, c)
c.gridy += 1
panel1.add(self._left_tb1, c)
panel2.add(right_tb1, c)
c.gridy += 1
panel1.add(left_t2, c)
panel2.add(right_t2, c)
c.gridy += 1
panel1.add(left_tb2, c)
panel2.add(right_tb2, c)
c.gridy += 1
panel1.add(left_t3, c)
panel2.add(right_t3, c)
c.gridy += 1
panel1.add(left_t4, c)
panel2.add(right_t4, c)
c.gridy += 1
panel1.add(left_tb3, c)
panel2.add(right_t5, c)
c.gridy += 1
panel2.add(right_t6, c)
c.gridy += 1
panel2.add(right_tb3, c)
c.gridy += 1
panel1.add(btnDecode, c)
panel2.add(btnGenerate, c)
self._splitpane.setLeftComponent(panel1)
self._splitpane.setRightComponent(panel2)
# customize our UI components
callbacks.customizeUiComponent(self._splitpane)
#callbacks.customizeUiComponent(panel1)
#callbacks.customizeUiComponent(scrollPane)
# add the custom tab to Burp's UI
callbacks.addSuiteTab(self)
#add option to do right-click
callbacks.registerContextMenuFactory(self)
return
def makeCropUI(imp, images, tgtDir, panel=None, cropContinuationFn=None):
""" imp: the ImagePlus to work on.
images: the list of ImgLib2 images, one per frame, not original but already isotropic.
(These are views that use a nearest neighbor interpolation using the calibration to scale to isotropy.)
tgtDir: the target directory where e.g. CSV files will be stored, for ROI, features, pointmatches.
panel: optional, a JPanel controlled by a GridBagLayout.
cropContinuationFn: optional, a function to execute after cropping,
which is given as arguments the original images,
minC, maxC (both define a ROI), and the cropped images. """
independent = None == panel
if not panel:
panel = JPanel()
panel.setBorder(BorderFactory.createEmptyBorder(10,10,10,10))
gb = GridBagLayout()
gc = GBC()
else:
gb = panel.getLayout()
# Constraints of the last component
gc = gb.getConstraints(panel.getComponent(panel.getComponentCount() - 1))
# Horizontal line to separate prior UI components from crop UI
gc.gridx = 0
gc.gridy += 1
gc.gridwidth = 4
gc.anchor = GBC.WEST
gc.fill = GBC.HORIZONTAL
sep = JSeparator()
sep.setMinimumSize(Dimension(200, 10))
gb.setConstraints(sep, gc)
panel.add(sep)
# ROI UI header
title = JLabel("ROI controls:")
gc.gridy +=1
gc.anchor = GBC.WEST
gc.gridwidth = 4
gb.setConstraints(title, gc)
panel.add(title)
# Column labels for the min and max coordinates
gc.gridy += 1
gc.gridwidth = 1
for i, title in enumerate(["", "X", "Y", "Z"]):
gc.gridx = i
gc.anchor = GBC.CENTER
label = JLabel(title)
gb.setConstraints(label, gc)
panel.add(label)
textfields = []
rms = []
# Load stored ROI if any
roi_path = path = os.path.join(tgtDir, "crop-roi.csv")
if os.path.exists(roi_path):
with open(roi_path, 'r') as csvfile:
reader = csv.reader(csvfile, delimiter=',', quotechar="\"")
reader.next() # header
minC = map(int, reader.next()[1:])
maxC = map(int, reader.next()[1:])
# Place the ROI over the ImagePlus
imp.setRoi(Roi(minC[0], minC[1], maxC[0] + 1 - minC[0], maxC[1] + 1 - minC[1]))
else:
# Use whole image dimensions
minC = [0, 0, 0]
maxC = [v -1 for v in Intervals.dimensionsAsLongArray(images[0])]
# Text fields for the min and max coordinates
for rowLabel, coords in izip(["min coords: ", "max coords: "],
[minC, maxC]):
gc.gridx = 0
gc.gridy += 1
label = JLabel(rowLabel)
gb.setConstraints(label, gc)
panel.add(label)
for i in xrange(3):
gc.gridx += 1
tf = JTextField(str(coords[i]), 10)
gb.setConstraints(tf, gc)
panel.add(tf)
textfields.append(tf)
listener = RoiMaker(imp, textfields, len(textfields) -1)
rms.append(listener)
tf.addKeyListener(listener)
tf.addMouseWheelListener(listener)
# Listen to changes in the ROI of imp
rfl = RoiFieldListener(imp, textfields)
Roi.addRoiListener(rfl)
# ... and enable cleanup
ImagePlus.addImageListener(FieldDisabler(rfl, rms))
# Functions for cropping images
cropped = None
cropped_imp = None
def storeRoi(minC, maxC):
if os.path.exists(roi_path):
# Load ROI
with open(path, 'r') as csvfile:
reader = csv.reader(csvfile, delimiter=',', quotechar="\"")
reader.next() # header
same = True
for a, b in izip(minC + maxC, map(int, reader.next()[1:] + reader.next()[1:])):
if a != b:
same = False
# Invalidate any CSV files for features and pointmatches: different cropping
for filename in os.listdir(tgtDir):
if filename.endswith("features.csv") or filename.endswith("pointmatches.csv"):
os.remove(os.path.join(tgtDir, filename))
break
if same:
return
# Store the ROI as crop-roi.csv
with open(roi_path, 'w') as csvfile:
w = csv.writer(csvfile, delimiter=',', quotechar="\"", quoting=csv.QUOTE_NONNUMERIC)
w.writerow(["coords", "x", "y", "z"])
w.writerow(["min"] + map(int, minC))
w.writerow(["max"] + map(int, maxC))
def crop(event):
global cropped, cropped_imp
coords = [int(float(tf.getText())) for tf in textfields]
minC = [max(0, c) for c in coords[0:3]]
maxC = [min(d -1, c) for d, c in izip(Intervals.dimensionsAsLongArray(images[0]), coords[3:6])]
storeRoi(minC, maxC)
print "ROI min and max coordinates"
print minC
print maxC
cropped = [Views.zeroMin(Views.interval(img, minC, maxC)) for img in images]
cropped_imp = showAsStack(cropped, title="cropped")
cropped_imp.setDisplayRange(imp.getDisplayRangeMin(), imp.getDisplayRangeMax())
if cropContinuationFn:
cropContinuationFn(images, minC, maxC, cropped, cropped_imp)
# Buttons to create a ROI and to crop to ROI,
# which when activated enables the fine registration buttons
crop_button = JButton("Crop to ROI")
crop_button.addActionListener(crop)
gc.gridx = 0
gc.gridy += 1
gc.gridwidth = 4
gc.anchor = GBC.WEST
buttons_panel = JPanel()
buttons_panel.add(crop_button)
gb.setConstraints(buttons_panel, gc)
panel.add(buttons_panel)
if independent:
frame = JFrame("Crop by ROI")
frame.getContentPane().add(panel)
frame.pack()
frame.setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE)
frame.addWindowListener(CloseControl(destroyables=rms + [rfl]))
frame.setVisible(True)
else:
# Re-pack the JFrame
parent = panel.getParent()
while not isinstance(parent, JFrame) and parent is not None:
parent = parent.getParent()
if parent:
frame = parent
frame.pack()
found = False
for wl in frame.getWindowListeners():
if isinstance(wl, CloseControl):
wl.addDestroyables(rms + [rfl])
found = True
break
if not found:
frame.addWindowListener(CloseControl(destroyables=rms + [rfl]))
frame.setVisible(True)
return panel
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
self.helpers = callbacks.getHelpers()
callbacks.setExtensionName("Session Authentication Tool")
self.out = callbacks.getStdout()
# definition of suite tab
self.tab = JPanel(GridBagLayout())
self.tabledata = MappingTableModel(callbacks)
self.table = JTable(self.tabledata)
#self.table.getColumnModel().getColumn(0).setPreferredWidth(50);
#self.table.getColumnModel().getColumn(1).setPreferredWidth(100);
self.tablecont = JScrollPane(self.table)
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.anchor = GridBagConstraints.FIRST_LINE_START
c.gridx = 0
c.gridy = 0
c.gridheight = 6
c.weightx = 0.3
c.weighty = 0.5
self.tab.add(self.tablecont, c)
c = GridBagConstraints()
c.weightx = 0.1
c.anchor = GridBagConstraints.FIRST_LINE_START
c.gridx = 1
c.gridy = 0
label_id = JLabel("Identifier:")
self.tab.add(label_id, c)
self.input_id = JTextField(20)
self.input_id.setToolTipText("Enter the identifier which is used by the application to identifiy a particular test user account, e.g. a numerical user id or a user name.")
c.gridy = 1
self.tab.add(self.input_id, c)
c.gridy = 2
label_content = JLabel("Content:")
self.tab.add(label_content, c)
self.input_content = JTextField(20, actionPerformed=self.btn_add_id)
self.input_content.setToolTipText("Enter some content which is displayed in responses of the application and shows that the current session belongs to a particular user, e.g. the full name of the user.")
c.gridy = 3
self.tab.add(self.input_content, c)
self.btn_add = JButton("Add/Edit Identity", actionPerformed=self.btn_add_id)
c.gridy = 4
self.tab.add(self.btn_add, c)
self.btn_del = JButton("Delete Identity", actionPerformed=self.btn_del_id)
c.gridy = 5
self.tab.add(self.btn_del, c)
callbacks.customizeUiComponent(self.tab)
callbacks.customizeUiComponent(self.table)
callbacks.customizeUiComponent(self.tablecont)
callbacks.customizeUiComponent(self.btn_add)
callbacks.customizeUiComponent(self.btn_del)
callbacks.customizeUiComponent(label_id)
callbacks.customizeUiComponent(self.input_id)
callbacks.addSuiteTab(self)
callbacks.registerScannerCheck(self)
callbacks.registerIntruderPayloadGeneratorFactory(self)
callbacks.registerContextMenuFactory(self)
def generateDeconvolutionScriptUI(srcDir,
tgtDir,
calibration,
preCropAffines,
ROI,
postCropAffines):
"""
Open an UI to automatically generate a script to:
1. Register the views of each time point TM folder, and deconvolve them.
2. Register deconvolved time points to each other, for a range of consecutive time points.
Will ask for the file path to the kernel file,
and also for the range of time points to process,
and for the deconvolution iterations for CM00-CM01, and CM02-CM03.
"""
template = """
# AUTOMATICALLY GENERATED - %s
import sys, os
sys.path.append("%s")
from lib.isoview import deconvolveTimePoints
from mpicbg.models import RigidModel3D, TranslationModel3D
from net.imglib2.img.display.imagej import ImageJFunctions as IL
# The folder with the sequence of TM\d+ folders, one per time point in the 4D series.
# Each folder should contain 4 KLB files, one per camera view of the IsoView microscope.
srcDir = "%s"
# A folder to save deconvolved images in, and CSV files describing features, point matches and transformations
targetDir = "%s"
# Path to the volume describing the point spread function (PSF)
kernelPath = "%s"
calibration = [%s] # An array with 3 floats (identity--all 1.0--because the coarse affines, that is,
# the camera transformations, already include the scaling to isotropy computed using the original calibration.
# The transformations of each timepoint onto the camera at index zero.
def cameraTransformations(dims0, dims1, dims2, dims3, calibration):
return {
0: [%s],
1: [%s],
2: [%s],
3: [%s]
}
# Deconvolution parameters
paramsDeconvolution = {
"blockSizes": None, # None means the image size + kernel size. Otherwise specify like e.g. [[128, 128, 128]] for img in images]
"CM_0_1_n_iterations": %i,
"CM_2_3_n_iterations": %i,
}
# Joint dictionary of parameters
params = {}
params.update(paramsDeconvolution)
# A region of interest for each camera view, for cropping after registration but prior to deconvolution
roi = ([%s], # array of 3 integers, top-left coordinates
[%s]) # array of 3 integers, bottom-right coordinates
# All 4 cameras relative to CM00
fineTransformsPostROICrop = \
[[1, 0, 0, 0,
0, 1, 0, 0,
0, 0, 1, 0],
[%s],
[%s],
[%s]]
deconvolveTimePoints(srcDir, targetDir, kernelPath, calibration,
cameraTransformations, fineTransformsPostROICrop,
params, roi, fine_fwd=True, subrange=range(%i, %i))
"""
od = OpenDialog("Choose kernel file", srcDir)
kernel_path = od.getPath()
if not kernel_path:
JOptionPane.showMessageDialog(None, "Can't proceed without a filepath to the kernel", "Alert", JOptionPane.ERROR_MESSAGE)
return
panel = JPanel()
panel.setBorder(BorderFactory.createEmptyBorder(10,10,10,10))
gb = GridBagLayout()
panel.setLayout(gb)
gc = GBC()
msg = ["Edit parameters, then push the button",
"to generate a script that, when run,",
"will execute the deconvolution for each time point",
"saving two 3D stacks per time point as ZIP files",
"in the target directory under subfolder 'deconvolved'.",
"Find the script in a new Script Editor window.",
" "]
gc.gridy = -1 # init
for line in msg:
label = JLabel(line)
gc.anchor = GBC.WEST
gc.gridx = 0
gc.gridy += 1
gc.gridwidth = 2
gc.gridheight = 1
gb.setConstraints(label, gc)
panel.add(label)
strings = [["Deconvolution iterations",
"CM_0_1_n_iterations", "CM_2_3_n_iterations"],
["Range",
"First time point", "Last time point"]]
params = {"CM_0_1_n_iterations": 5,
"CM_2_3_n_iterations": 7,
"First time point": 0,
"Last time point": -1} # -1 means last
insertFloatFields(panel, gb, gc, params, strings)
def asString(affine):
matrix = zeros(12, 'd')
affine.toArray(matrix)
return ",".join(imap(str, matrix))
def generateScript(event):
script = template % (str(datetime.now()),
filter(lambda path: path.endswith("IsoView-GCaMP"), sys.path)[-1],
srcDir,
tgtDir,
kernel_path,
", ".join(imap(str, calibration)),
asString(preCropAffines[0]),
asString(preCropAffines[1]),
asString(preCropAffines[2]),
asString(preCropAffines[3]),
params["CM_0_1_n_iterations"],
params["CM_2_3_n_iterations"],
", ".join(imap(str, ROI[0])),
", ".join(imap(str, ROI[1])),
asString(postCropAffines[1]),
asString(postCropAffines[2]),
asString(postCropAffines[3]),
params["First time point"],
params["Last time point"])
tab = None
for frame in JFrame.getFrames():
if str(frame).startswith("org.scijava.ui.swing.script.TextEditor["):
try:
tab = frame.newTab(script, "python")
break
except:
print sys.exc_info()
if not tab:
try:
now = datetime.now()
with open(os.path.join(System.getProperty("java.io.tmpdir"),
"script-%i-%i-%i_%i:%i.py" % (now.year, now.month, now.day,
now.hour, now.minute)), 'w') as f:
f.write(script)
except:
print sys.exc_info()
print script
gen = JButton("Generate script")
gen.addActionListener(generateScript)
gc.anchor = GBC.CENTER
gc.gridx = 0
gc.gridy += 1
gc.gridwidth = 2
gb.setConstraints(gen, gc)
panel.add(gen)
frame = JFrame("Generate deconvolution script")
frame.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE)
frame.getContentPane().add(panel)
frame.pack()
frame.setLocationRelativeTo(None) # center in the screen
frame.setVisible(True)
def build_welcome_panel(self):
'''
Construct the welcome panel here.
'''
panel = JPanel(GridBagLayout())
constraints = GridBagConstraints()
constraints.insets = Insets(10, 10, 10, 10)
message = ('<html><body>'
'<h1>Welcome to Nammu</h1>'
'<h2>An editor for the ORACC project<h2>'
'<p>'
'<a href=\'oracc\'>Click here</a> for help with ORACC.'
'</p>'
'<p>Learn more about Nammu <a href=\'nammu\'>here</a>.</p>'
'</body></html>')
# Configure a JEditorPane to display HTML for our welcome message
msg_pane = JEditorPane()
msg_pane.setEditable(False)
kit = HTMLEditorKit()
msg_pane.setEditorKit(kit)
scrollPane = JScrollPane(msg_pane)
# This handles the stylesheet applied to the welcome message
styleSheet = kit.getStyleSheet()
styleSheet.addRule('body {color:black; font-size: 16 pt; }')
styleSheet.addRule('h1 {text-align:center; }')
styleSheet.addRule('h2 {text-align:center; }')
# Set the JEditorPane background to match the rest of the window
msg_pane.border = BorderFactory.createEmptyBorder(4, 4, 4, 4)
msg_pane.background = Color(238, 238, 238)
# Add the message and the css and to the JEditorPane
doc = kit.createDefaultDocument()
msg_pane.setDocument(doc)
msg_pane.setText(message)
# Set up a hyperlink listener
listener = addEventListener(msg_pane, HyperlinkListener,
'hyperlinkUpdate', self.handleEvent)
# Configure the placement of the JEditorPane
constraints.gridx = 1
constraints.gridy = 1
constraints.fill = GridBagConstraints.BOTH
constraints.anchor = GridBagConstraints.CENTER
panel.add(msg_pane, constraints)
# Build and place the checkbox
self.checkbox = JCheckBox('Don\'t show this message again.',
selected=False)
constraints.gridx = 1
constraints.gridy = 2
panel.add(self.checkbox, constraints)
# Build and place the close button
close_button = JButton('Close', actionPerformed=self.close_action)
constraints.gridx = 2
constraints.gridy = 2
panel.add(close_button, constraints)
return panel
def specifyRoiUI(roi=Roi(0, 0, 0, 0)):
# A panel in which to place UI elements
panel = JPanel()
panel.setBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10))
gb = GridBagLayout()
panel.setLayout(gb)
gc = GBC()
bounds = roi.getBounds() if roi else Rectangle()
textfields = []
roimakers = []
# Basic properties of most UI elements, will edit when needed
gc.gridx = 0 # can be any natural number
gc.gridy = 0 # idem.
gc.gridwidth = 1 # when e.g. 2, the UI element will occupy two horizontally adjacent grid cells
gc.gridheight = 1 # same but vertically
gc.fill = GBC.NONE # can also be BOTH, VERTICAL and HORIZONTAL
for title in ["x", "y", "width", "height"]:
# label
gc.gridx = 0
gc.anchor = GBC.EAST
label = JLabel(title + ": ")
gb.setConstraints(label, gc) # copies the given constraints 'gc',
# so we can modify and reuse gc later.
panel.add(label)
# text field, below the title
gc.gridx = 1
gc.anchor = GBC.WEST
text = str(getattr(bounds,
title)) # same as e.g. bounds.x, bounds.width, ...
textfield = JTextField(text,
10) # 10 is the size of the field, in digits
gb.setConstraints(textfield, gc)
panel.add(textfield)
textfields.append(
textfield) # collect all 4 created textfields for the listeners
# setup ROI and text field listeners
listener = RoiMaker(textfields,
len(textfields) -
1) # second argument is the index of textfield
# in the list of textfields.
roimakers.append(listener)
textfield.addKeyListener(listener)
textfield.addMouseWheelListener(listener)
# Position next ROI property in a new row by increasing the Y coordinate of the layout grid
gc.gridy += 1
# User documentation (uses HTML to define line breaks)
doc = JLabel(
"<html><body><br />Click on a field to activate it, then:<br />" +
"Type in integer numbers<br />" +
"or use arrow keys to increase by 1<br />" +
"or use the scroll wheel on a field.</body></html>")
gc.gridx = 0 # start at the first column
gc.gridwidth = 2 # Spans both columns
gb.setConstraints(doc, gc)
panel.add(doc)
# Listen to changes in the ROI of imp
roilistener = TextFieldUpdater(textfields)
Roi.addRoiListener(roilistener)
# Show window
frame = JFrame("Specify rectangular ROI")
frame.getContentPane().add(panel)
frame.pack()
frame.setLocationRelativeTo(None) # center in the screen
frame.setDefaultCloseOperation(
JFrame.DO_NOTHING_ON_CLOSE) # prevent closing the window
frame.addWindowListener(
CloseControl(roilistener)) # handles closing the window
frame.setVisible(True)
def initComponents(self):
#set the layout and add form components
self.setLayout(GridBagLayout())
gc = GridBagConstraints()
gc.weightx = 1
gc.weighty = 1
# First Row
gc.gridy = 0
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.fill = GridBagConstraints.NONE
gc.anchor = GridBagConstraints.LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.nameLabel, gc)
gc.gridx = 1
gc.anchor = GridBagConstraints.LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.nameField, gc)
# Second Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.mobLabel, gc)
gc.gridx = 1
gc.anchor = GridBagConstraints.LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.mobField, gc)
#Third Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.FIRST_LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.emailLabel, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.emailField, gc)
#Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.FIRST_LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.userNameLabel, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.userNameField, gc)
# Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.FIRST_LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.passwordLabel, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.passwordField, gc)
# Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.FIRST_LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.confirmPasswordLabel, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.confirmPasswordField, gc)
# Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 0.1
gc.gridx = 0
gc.anchor = GridBagConstraints.FIRST_LINE_END
gc.insets = Insets(0, 0, 0, 5)
self.add(self.superPassLabel, gc)
gc.weightx = 1
gc.weighty = 0.2
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
gc.insets = Insets(0, 0, 0, 0)
self.add(self.superPassField, gc)
# Next Row
gc.gridy += 1
gc.weightx = 1
gc.weighty = 2.0
gc.gridx = 1
gc.anchor = GridBagConstraints.FIRST_LINE_START
self.add(self.addAdminBtn, gc)
innerBorder = BorderFactory.createTitledBorder('Add Admin')
outerBorder = BorderFactory.createEmptyBorder(5, 5, 5, 5)
self.setBorder(
BorderFactory.createCompoundBorder(outerBorder, innerBorder))
def initComponents(self):
self.setLayout(GridBagLayout())
gbc = GridBagConstraints()
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridx = 0
gbc.gridy = 0
descriptionLabel = JLabel("FEA - Forensics Email Analysis")
self.add(descriptionLabel, gbc)
gbc.gridy = 1
self.cbNSLookup = JCheckBox(
"Perform DNS Lookup on email domains",
actionPerformed=self.cbNSLookupActionPerformed)
self.cbNSLookup.setSelected(True)
self.add(self.cbNSLookup, gbc)
# TODO: include option to browse for text file with list of emails to exclude from analysis
numberThreadsLabel = JLabel(
"Maximum number of threads for DNS Lookup task: ")
gbc.gridy = 2
self.add(numberThreadsLabel, gbc)
self.numberThreadsSlider = JSlider(
JSlider.HORIZONTAL,
1,
16,
8,
stateChanged=self.sliderActionPerformed)
self.numberThreadsSlider.setMajorTickSpacing(1)
self.numberThreadsSlider.setPaintLabels(True)
self.numberThreadsSlider.setPaintTicks(True)
self.numberThreadsSlider.setSnapToTicks(True)
self.numberThreadsSlider.setToolTipText(
"set maximum number of concurrent threads when performing DNS lookup on email domains"
)
gbc.gridy = 5
gbc.gridwidth = 15
gbc.gridheight = 1
gbc.fill = GridBagConstraints.BOTH
gbc.weightx = 0
gbc.weighty = 0
gbc.anchor = GridBagConstraints.NORTHWEST
gbc.gridy = 3
self.add(self.numberThreadsSlider, gbc)
self.cbGenerateExcel = JCheckBox(
"Generate Excel format report (more detailed)",
actionPerformed=self.cbGenerateExcelActionPerformed)
self.cbGenerateExcel.setSelected(True)
gbc.gridy = 4
self.add(self.cbGenerateExcel, gbc)
self.cbGenerateCSV = JCheckBox(
"Generate CSV format report (plaintext)",
actionPerformed=self.cbGenerateCSVActionPerformed)
self.cbGenerateCSV.setSelected(True)
gbc.gridy = 5
self.add(self.cbGenerateCSV, gbc)
gbc.gridy = 6
self.cbWayback = JCheckBox(
"Perform Wayback Machine Lookup on email domains (WARNING: can be a slow process!)",
actionPerformed=self.cbWaybackActionPerformed)
self.cbWayback.setSelected(True)
self.add(self.cbWayback, gbc)
def registerExtenderCallbacks( self, callbacks):
self._helpers = callbacks.getHelpers()
callbacks.setExtensionName("JWT FuzzHelper")
callbacks.registerIntruderPayloadProcessor(self)
callbacks.registerExtensionStateListener(self)
self._stdout = PrintWriter(callbacks.getStdout(), True)
self._stderr = PrintWriter(callbacks.getStderr(), True)
# Holds values passed by user from Configuration panel
self._fuzzoptions = {
"target" : "Header",
"selector" : None,
"signature" : False,
"algorithm" : "HS256",
"key" : "",
"key_cmd" : ""
}
self._isNone = lambda val: isinstance(val, type(None))
# Configuration panel Layout
self._configurationPanel = JPanel()
gridBagLayout = GridBagLayout()
gridBagLayout.columnWidths = [ 0, 0, 0]
gridBagLayout.rowHeights = [ 10, 10, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ]
gridBagLayout.columnWeights = [ 0.0, 0.0, 0.0 ]
gridBagLayout.rowWeights = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0]
self._configurationPanel.setLayout(gridBagLayout)
# Setup tabs
self._tabs = JTabbedPane()
self._tabs.addTab('Configuration',self._configurationPanel)
#self._tabs.addTab('Help',self._helpPanel)
# Target Options
targetLabel = JLabel("Target Selection (Required): ")
targetLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 1
c.insets = Insets(0,10,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(targetLabel,c)
options = [ 'Header', 'Payload' ]
self._targetComboBox = JComboBox(options)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 1
c.anchor = GridBagConstraints.LINE_START
self._configurationPanel.add(self._targetComboBox,c)
# Help Button
self._helpButton = JButton("Help", actionPerformed=self.helpMenu)
c = GridBagConstraints()
c.gridx = 2
c.gridy = 1
c.anchor = GridBagConstraints.FIRST_LINE_START
self._configurationPanel.add(self._helpButton,c)
# Selector Options
self._selectorLabel = JLabel("JSON Selector [Object Identifier-Index Syntax] (Required): ")
self._selectorLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 2
c.insets = Insets(0,10,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(self._selectorLabel, c)
self._selectorTextField = JTextField('',50)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 2
self._configurationPanel.add(self._selectorTextField, c)
# Regex option
self._regexLabel = JLabel("Use regex as JSON Selector? (Optional): ")
self._regexLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 3
c.insets = Insets(0,0,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(self._regexLabel,c)
self._regexCheckBox = JCheckBox("", actionPerformed=self.regexSelector)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 3
c.anchor = GridBagConstraints.FIRST_LINE_START
self._configurationPanel.add(self._regexCheckBox,c)
# Signature Options
generateSignatureLabel = JLabel("Generate signature? (Required): ")
generateSignatureLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 4
c.insets = Insets(0,10,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(generateSignatureLabel,c)
options = ["False", "True"]
self._generateSignatureComboBox = JComboBox(options)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 4
c.anchor = GridBagConstraints.LINE_START
self._configurationPanel.add(self._generateSignatureComboBox,c)
signatureAlgorithmLabel = JLabel("Signature Algorithm (Optional): ")
signatureAlgorithmLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 5
c.insets = Insets(0,10,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(signatureAlgorithmLabel,c)
options = ["None", "HS256","HS384","HS512","ES256","ES384","ES512","RS256","RS384","RS512","PS256","PS256","PS384","PS512"]
self._algorithmSelectionComboBox = JComboBox(options)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 5
c.anchor = GridBagConstraints.LINE_START
self._configurationPanel.add(self._algorithmSelectionComboBox,c)
# Signing key options
self._signingKeyLabel = JLabel("Signing Key (Optional): ")
self._signingKeyLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 6
c.insets = Insets(0,10,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(self._signingKeyLabel,c)
self.addSigningKeyTextArea()
self._fromFileTextField = JTextField('',50)
fromFileLabel = JLabel("Signing key from file? (Optional): ")
fromFileLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 7
c.insets = Insets(0,0,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(fromFileLabel,c)
self._fromFileCheckBox = JCheckBox("", actionPerformed=self.fromFile)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 7
c.anchor = GridBagConstraints.FIRST_LINE_START
self._configurationPanel.add(self._fromFileCheckBox,c)
self._fromCmdTextField = JTextField('',50)
fromCmdLabel = JLabel("Signing key from command? (Optional): ")
fromCmdLabel.setFont(Font("Tahoma",Font.BOLD, 12))
c = GridBagConstraints()
c.gridx = 0
c.gridy = 8
c.insets = Insets(0,0,0,0)
c.anchor = GridBagConstraints.LINE_END
self._configurationPanel.add(fromCmdLabel,c)
self._fromCmdCheckBox = JCheckBox("", actionPerformed=self.fromCmd)
c = GridBagConstraints()
c.gridx = 1
c.gridy = 8
c.anchor = GridBagConstraints.FIRST_LINE_START
self._configurationPanel.add(self._fromCmdCheckBox,c)
self._saveButton = JButton("Save Configuration", actionPerformed=self.saveOptions)
self._saveButton.setText("Save Configuration")
c = GridBagConstraints()
c.gridx = 1
c.gridy = 9
c.anchor = GridBagConstraints.FIRST_LINE_START
self._configurationPanel.add(self._saveButton,c)
callbacks.customizeUiComponent(self._configurationPanel)
callbacks.customizeUiComponent(self._tabs)
callbacks.addSuiteTab(self)
self._stdout.println("[JWT FuzzHelper] Loaded successfully")
return
def __init__(self, extender=None, *rows):
self.extender = extender
self.table = table = JTable(ParameterProcessingRulesTableModel(*rows))
table.setPreferredScrollableViewportSize(Dimension(500, 70))
table.setRowSorter(TableRowSorter(table.getModel()))
table.setFillsViewportHeight(True)
gridBagLayout = GridBagLayout()
gridBagLayout.columnWidths = [0, 0, 25, 0 ]
gridBagLayout.rowHeights = [0, 0, 0, 0]
gridBagLayout.columnWeights = [0.0, 1.0, 1.0, Double.MIN_VALUE]
gridBagLayout.rowWeights = [0.0, 0.0, 0.0, Double.MIN_VALUE]
self.setLayout(gridBagLayout)
addButton = JButton("Add")
addButton.addActionListener(AddRemoveParameterListener(table))
addButtonConstraints = GridBagConstraints()
addButtonConstraints.fill = GridBagConstraints.HORIZONTAL
addButtonConstraints.insets = Insets(0, 0, 5, 5)
addButtonConstraints.gridx = 0
addButtonConstraints.gridy = 0
self.add(addButton, addButtonConstraints)
removeButton = JButton("Remove")
removeButton.addActionListener(AddRemoveParameterListener(table))
removeButtonConstraints = GridBagConstraints()
removeButtonConstraints.fill = GridBagConstraints.HORIZONTAL
removeButtonConstraints.insets = Insets(0, 0, 5, 5)
removeButtonConstraints.gridx = 0
removeButtonConstraints.gridy = 1
self.add(removeButton, removeButtonConstraints)
upButton = JButton("Up")
upButton.addActionListener(AddRemoveParameterListener(table))
upButtonConstraints = GridBagConstraints()
upButtonConstraints.fill = GridBagConstraints.HORIZONTAL
upButtonConstraints.insets = Insets(0, 0, 5, 5)
upButtonConstraints.gridx = 0
upButtonConstraints.gridy = 2
self.add(upButton, upButtonConstraints)
downButton = JButton("Down")
downButton.addActionListener(AddRemoveParameterListener(table))
downButtonConstraints = GridBagConstraints()
downButtonConstraints.fill = GridBagConstraints.HORIZONTAL
downButtonConstraints.anchor = GridBagConstraints.NORTH
downButtonConstraints.insets = Insets(0, 0, 5, 5)
downButtonConstraints.gridx = 0
downButtonConstraints.gridy = 3
self.add(downButton, downButtonConstraints)
scrollPane = JScrollPane(table)
scrollPaneConstraints = GridBagConstraints()
scrollPaneConstraints.gridwidth = 2
scrollPaneConstraints.gridheight = 5
scrollPaneConstraints.insets = Insets(0, 0, 5, 5)
scrollPaneConstraints.anchor = GridBagConstraints.NORTHWEST
scrollPaneConstraints.gridx = 1
scrollPaneConstraints.gridy = 0
self.add(scrollPane, scrollPaneConstraints)
self.initParameterColumn(table)
self.initColumnSizes(table)
