def initUI(self):
self.tab = JPanel()
# UI for Output
self.outputLabel = JLabel("AutoRecon Log:")
self.outputLabel.setFont(Font("Tahoma", Font.BOLD, 14))
self.outputLabel.setForeground(Color(255, 102, 52))
self.logPane = JScrollPane()
self.outputTxtArea = JTextArea()
self.outputTxtArea.setFont(Font("Consolas", Font.PLAIN, 12))
self.outputTxtArea.setLineWrap(True)
self.logPane.setViewportView(self.outputTxtArea)
self.clearBtn = JButton("Clear Log", actionPerformed=self.clearLog)
self.exportBtn = JButton("Export Log", actionPerformed=self.exportLog)
self.parentFrm = JFileChooser()
# Layout
layout = GroupLayout(self.tab)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
self.tab.setLayout(layout)
layout.setHorizontalGroup(layout.createParallelGroup().addGroup(
layout.createSequentialGroup().addGroup(
layout.createParallelGroup().addComponent(
self.outputLabel).addComponent(self.logPane).addComponent(
self.clearBtn).addComponent(self.exportBtn))))
layout.setVerticalGroup(layout.createParallelGroup().addGroup(
layout.createParallelGroup().addGroup(
layout.createSequentialGroup().addComponent(
self.outputLabel).addComponent(self.logPane).addComponent(
self.clearBtn).addComponent(self.exportBtn))))
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
self.helpers = callbacks.getHelpers()
callbacks.setExtensionName("WebTech")
self.out = callbacks.getStdout()
self.callbacks.printOutput("Sucessfully loaded WebTech {}".format(VERSION))
try:
self.webtech = WebTech(options={'json': True})
except UpdateInBurpException as e:
#self.callbacks.printOutput(e)
for db_file in databases:
db = self.callbacks.makeHttpRequest(
'raw.githubusercontent.com', # we are hardcoding this since there isn't a nice api for that
443,
True,
self.helpers.buildHttpRequest(URL(db_file[0]))
);
db = db.tostring()
save_database_file(db[db.index("\r\n\r\n") + len("\r\n\r\n"):], db_file[1])
self.webtech = WebTech(options={'json': True})
# define all checkboxes
self.cbPassiveChecks = self.defineCheckBox("Enable Passive Scanner Checks")
self.cbActiveChecks = self.defineCheckBox("Enable Active Scanner Checks", True, False)
self.btnSave = JButton("Set as default", actionPerformed=self.saveConfig)
self.btnRestore = JButton("Restore", actionPerformed=self.restoreConfig)
self.grpConfig = JPanel()
self.grpConfig.add(self.btnSave)
self.grpConfig.add(self.btnRestore)
self.restoreConfig()
# definition of config tab
self.tab = JPanel()
layout = GroupLayout(self.tab)
self.tab.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
layout.setHorizontalGroup(
layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(self.cbPassiveChecks)
.addComponent(self.cbActiveChecks)
)
.addGroup(layout.createParallelGroup()
.addComponent(self.grpConfig, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
)
layout.setVerticalGroup(
layout.createSequentialGroup()
.addComponent(self.cbPassiveChecks)
.addComponent(self.cbActiveChecks)
.addComponent(self.grpConfig, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
callbacks.registerScannerCheck(self)
callbacks.addSuiteTab(self)
def __init__(self):
TrafficMod.__init__(self)
self.setLayout(BorderLayout())
settings_panel = JPanel()
# settings_panel.setBorder(LineBorder(Color.RED))
layout = GroupLayout(settings_panel)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
settings_panel.setLayout(layout)
self.add(settings_panel)
# Horizontal group
horizontal = layout.createParallelGroup()
layout.setHorizontalGroup(
layout.createSequentialGroup().addGroup(horizontal))
self.horizontal_layout = horizontal
# Vertical group
vertical = layout.createSequentialGroup()
layout.setVerticalGroup(vertical)
self.vertical_layout = vertical
# Main title, label only
main_title = JLabel(self.title)
main_title.setFont(get_title_font(main_title))
horizontal.addGroup(
layout.createSequentialGroup().addComponent(main_title))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(main_title))
# Profile name
name_lbl = JLabel("Profile Name")
name_lbl.setToolTipText("The name of this profile")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(name_lbl).addComponent(
self.name_textfield))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
name_lbl).addComponent(self.name_textfield))
# Profile description
descr_lbl = JLabel("Description")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
descr_lbl).addComponent(self.descr_textarea))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
descr_lbl).addComponent(self.descr_textarea))
layout.linkSize(SwingConstants.HORIZONTAL, name_lbl, descr_lbl)
def registerExtenderCallbacks(self, this_callbacks):
global callbacks, helpers, checkbox_perHost, checkbox_common, checkbox_ssh, checkbox_key, checkbox_php, checkbox_sql
callbacks = this_callbacks
ui_label = JLabel('Scans to perform:')
checkbox_perHost = self.defineCheckBox('Scan once per domain (web root only, not every subdirectory)')
checkbox_common = self.defineCheckBox('Interesting files')
checkbox_ssh = self.defineCheckBox('SSH private keys')
checkbox_key = self.defineCheckBox('.key files')
checkbox_php = self.defineCheckBox('PHP file scans')
checkbox_sql = self.defineCheckBox('SQL file scans')
self.tab = JPanel()
layout = GroupLayout(self.tab)
self.tab.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
layout.setHorizontalGroup(
layout.createSequentialGroup()
.addComponent(ui_label)
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.LEADING)
.addComponent(checkbox_common)
.addComponent(checkbox_ssh)
.addComponent(checkbox_key)
.addComponent(checkbox_php)
.addComponent(checkbox_sql))
.addComponent(checkbox_perHost)
)
layout.setVerticalGroup(
layout.createSequentialGroup()
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(ui_label)
.addComponent(checkbox_common)
.addComponent(checkbox_perHost))
.addComponent(checkbox_ssh)
.addComponent(checkbox_key)
.addComponent(checkbox_php)
.addComponent(checkbox_sql)
)
helpers = callbacks.getHelpers()
callbacks.setExtensionName("Interesting Files Scanner")
callbacks.registerScannerCheck(FileScanner())
callbacks.addSuiteTab(self)
print 'Interesting Files Scanner'
print 'Created by Sven Fassbender (@mezdanak, @modzero) v' + VERSION
print 'https://github.com/modzero/interestingFileScanner'
return
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
self.helpers = callbacks.helpers
self.checkboxEnable = JCheckBox('Enabled')
self.checkboxEnable.setSelected(False)
self.checkboxEnable.setEnabled(True)
self.scriptpane = JTextPane()
self.scriptpane.setFont(Font('Monospaced', Font.PLAIN, 11))
self.scrollpane = JScrollPane()
self.scrollpane.setViewportView(self.scriptpane)
self.tab = JPanel()
layout = GroupLayout(self.tab)
self.tab.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
layout.setHorizontalGroup(layout.createParallelGroup().addComponent(
self.checkboxEnable).addComponent(self.scrollpane))
layout.setVerticalGroup(layout.createSequentialGroup().addComponent(
self.checkboxEnable).addComponent(self.scrollpane))
self._code = compile('', '<string>', 'exec')
self._script = ''
script = callbacks.loadExtensionSetting('script')
if script:
script = base64.b64decode(script)
self.scriptpane.document.insertString(
self.scriptpane.document.length, script, SimpleAttributeSet())
self._script = script
try:
self._code = compile(script, '<string>', 'exec')
except Exception as e:
traceback.print_exc(file=self.callbacks.getStderr())
callbacks.setExtensionName("Python Scripter (modified)")
callbacks.registerSessionHandlingAction(self)
callbacks.registerExtensionStateListener(self)
callbacks.registerHttpListener(self)
callbacks.customizeUiComponent(self.getUiComponent())
callbacks.addSuiteTab(self)
self.scriptpane.requestFocus()
return
def __init__(self):
TrafficMod.__init__(self)
self.setLayout(FlowLayout(FlowLayout.LEADING))
settings_panel = JPanel()
layout = GroupLayout(settings_panel)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
settings_panel.setLayout(layout)
self.add(settings_panel)
# Horizontal group
horizontal = layout.createParallelGroup()
layout.setHorizontalGroup(
layout.createSequentialGroup().addGroup(horizontal))
# Vertical group
vertical = layout.createSequentialGroup()
layout.setVerticalGroup(vertical)
# Main title, label only
main_title = JLabel(self.title)
main_title.setFont(Font(main_title.getFont().getName(), Font.BOLD, 14))
horizontal.addGroup(
layout.createSequentialGroup().addComponent(main_title))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(main_title))
# Duplicate percentage
percent_lbl = JLabel("Duplicate %")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
percent_lbl).addComponent(self.duplicate_percent))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
percent_lbl).addComponent(self.duplicate_percent))
# Duplicate percentage correlation
corr_lbl = JLabel("Correlation %")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(corr_lbl).addComponent(
self.duplicate_percent_correlation))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
corr_lbl).addComponent(self.duplicate_percent_correlation))
layout.linkSize(SwingConstants.HORIZONTAL, corr_lbl, percent_lbl)
def registerExtenderCallbacks(self, callbacks):
self._callbacks = callbacks
self._helpers = callbacks.getHelpers()
self._stdout = PrintWriter(callbacks.getStdout(), True)
callbacks.setExtensionName("Argument Injection Hammer")
self._create_brute_payloads()
self._stdout.println('==================================')
self._stdout.println(" ,")
self._stdout.println(" /( ___________")
self._stdout.println(" | >:===========`")
self._stdout.println(" )(")
self._stdout.println('== AIH "" Hammer Smash Party =====')
self._stdout.println('== Neil Bergman - NCC Group =====')
self._stdout.println('==================================')
self._checkbox_brute = self._define_check_box("Brute-force Short Argument Flags", False)
self._button_save = JButton("Save Configuration", actionPerformed=self._save_config)
self.tab = JPanel()
layout = GroupLayout(self.tab)
self.tab.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
layout.setHorizontalGroup(
layout.createSequentialGroup().addGroup(
layout.createParallelGroup()
.addComponent(self._checkbox_brute)
).addGroup(
layout.createParallelGroup()
.addComponent(self._button_save)
)
)
layout.setVerticalGroup(
layout.createSequentialGroup().addGroup(
layout.createParallelGroup()
.addComponent(self._checkbox_brute)
).addGroup(
layout.createParallelGroup()
.addComponent(self._button_save)
)
)
callbacks.addSuiteTab(self)
self._restore_config()
callbacks.registerScannerCheck(self)
return
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
self.helpers = callbacks.getHelpers()
callbacks.setExtensionName("WebTech")
self.out = callbacks.getStdout()
self.callbacks.printOutput("Sucessfully loaded WebTech {}".format(VERSION))
self.webtech = WebTech(options={'json': True})
# define all checkboxes
self.cbPassiveChecks = self.defineCheckBox("Enable Passive Scanner Checks")
self.cbActiveChecks = self.defineCheckBox("Enable Active Scanner Checks", True, False)
self.btnSave = JButton("Set as default", actionPerformed=self.saveConfig)
self.btnRestore = JButton("Restore", actionPerformed=self.restoreConfig)
self.grpConfig = JPanel()
self.grpConfig.add(self.btnSave)
self.grpConfig.add(self.btnRestore)
self.restoreConfig()
# definition of config tab
self.tab = JPanel()
layout = GroupLayout(self.tab)
self.tab.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
layout.setHorizontalGroup(
layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(self.cbPassiveChecks)
.addComponent(self.cbActiveChecks)
)
.addGroup(layout.createParallelGroup()
.addComponent(self.grpConfig, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
)
layout.setVerticalGroup(
layout.createSequentialGroup()
.addComponent(self.cbPassiveChecks)
.addComponent(self.cbActiveChecks)
.addComponent(self.grpConfig, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
callbacks.registerScannerCheck(self)
callbacks.addSuiteTab(self)
class BurpExtender(IBurpExtender, ITab, IContextMenuFactory, DocumentListener, ChangeListener):
#
# implement IBurpExtender
#
def registerExtenderCallbacks(self, callbacks):
print "PhantomJS RIA Crawler extension"
print "Nikolay Matyunin @autorak <*****@*****.**>"
# keep a reference to our callbacks object and helpers object
self._callbacks = callbacks
self._helpers = callbacks.getHelpers()
# extension name
callbacks.setExtensionName("Phantom RIA Crawler")
# Create Tab UI components
self._jPanel = JPanel()
self._jPanel.setBorder(BorderFactory.createEmptyBorder(5,5,5,5));
_titleLabel = JLabel("Phantom RIA Crawler", SwingConstants.LEFT)
_titleLabelFont = _titleLabel.font
_titleLabelFont = _titleLabelFont.deriveFont(Font.BOLD, 12);
_titleLabel.setFont(_titleLabelFont);
_titleLabel.setForeground(Color(230, 142, 11))
self._addressTextField = JTextField('')
self._addressTextField.setColumns(50)
_addressTextLabel = JLabel("Target URL:", SwingConstants.RIGHT)
self._addressTextField.getDocument().addDocumentListener(self)
self._phantomJsPathField = JTextField('phantomjs') # TODO: set permanent config value
self._phantomJsPathField.setColumns(50)
_phantomJsPathLabel = JLabel("PhantomJS path:", SwingConstants.RIGHT)
self._startButton = JToggleButton('Start', actionPerformed=self.startToggled)
self._startButton.setEnabled(False)
_requestsMadeLabel = JLabel("DEPs found:", SwingConstants.RIGHT)
self._requestsMadeInfo = JLabel("", SwingConstants.LEFT)
_statesFoundLabel = JLabel("States found:", SwingConstants.RIGHT)
self._statesFoundInfo = JLabel("", SwingConstants.LEFT)
_separator = JSeparator(SwingConstants.HORIZONTAL)
_configLabel = JLabel("Crawling configuration:")
self._configButton = JButton("Load config", actionPerformed=self.loadConfigClicked)
self._configFile = ""
_listenersLabel= JLabel("Burp proxy listener:", SwingConstants.RIGHT)
self._listenersCombo = JComboBox()
self._configTimer = Timer(5000, None)
self._configTimer.actionPerformed = self._configUpdated
self._configTimer.stop()
self._configUpdated(None)
self._commandClient = CommandClient(self)
# Layout management
self._groupLayout = GroupLayout(self._jPanel)
self._jPanel.setLayout(self._groupLayout)
self._groupLayout.setAutoCreateGaps(True)
self._groupLayout.setAutoCreateContainerGaps(True)
self._groupLayout.setHorizontalGroup(self._groupLayout.createParallelGroup()
.addComponent(_titleLabel)
.addGroup(self._groupLayout.createSequentialGroup()
.addComponent(_addressTextLabel)
.addGroup(self._groupLayout.createParallelGroup()
.addComponent(self._addressTextField, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
.addGroup(self._groupLayout.createSequentialGroup()
.addComponent(_requestsMadeLabel)
.addComponent(self._requestsMadeInfo))
.addGroup(self._groupLayout.createSequentialGroup()
.addComponent(_statesFoundLabel)
.addComponent(self._statesFoundInfo)))
.addComponent(self._startButton))
.addComponent(_separator)
.addGroup(self._groupLayout.createSequentialGroup()
.addComponent(_configLabel)
.addComponent(self._configButton))
.addGroup(self._groupLayout.createSequentialGroup()
.addComponent(_phantomJsPathLabel)
.addComponent(self._phantomJsPathField, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE))
.addGroup(self._groupLayout.createSequentialGroup()
.addComponent(_listenersLabel)
.addComponent(self._listenersCombo, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE))
)
self._groupLayout.setVerticalGroup(self._groupLayout.createSequentialGroup()
.addComponent(_titleLabel)
.addGroup(self._groupLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(_addressTextLabel)
.addComponent(self._addressTextField)
.addComponent(self._startButton))
.addGroup(self._groupLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(_requestsMadeLabel)
.addComponent(self._requestsMadeInfo))
.addGroup(self._groupLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(_statesFoundLabel)
.addComponent(self._statesFoundInfo))
.addComponent(_separator, GroupLayout.PREFERRED_SIZE, GroupLayout.DEFAULT_SIZE, GroupLayout.PREFERRED_SIZE)
.addGroup(self._groupLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(_configLabel)
.addComponent(self._configButton))
.addGroup(self._groupLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(_phantomJsPathLabel)
.addComponent(self._phantomJsPathField))
.addGroup(self._groupLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(_listenersLabel)
.addComponent(self._listenersCombo))
)
self._groupLayout.linkSize(SwingConstants.HORIZONTAL, _configLabel, _phantomJsPathLabel);
self._groupLayout.linkSize(SwingConstants.HORIZONTAL, _configLabel, _listenersLabel);
self._groupLayout.linkSize(SwingConstants.HORIZONTAL, _statesFoundLabel, _requestsMadeLabel);
# context menu data
self._contextMenuData = None;
self._running = False;
# register callbacks
callbacks.customizeUiComponent(self._jPanel)
callbacks.registerContextMenuFactory(self)
callbacks.addSuiteTab(self)
return
#
# implement ITab and Tab ChangeListener
#
def getTabCaption(self):
return "Phantom RIA Crawler"
def getUiComponent(self):
return self._jPanel
def stateChanged(self, ev):
self._configUpdated()
def _configUpdated(self, ev):
config = self._callbacks.saveConfig()
# update proxy listeners
index = 0
listeners = DefaultComboBoxModel()
while (("proxy.listener" + str(index)) in config):
listenerItem = config["proxy.listener" + str(index)]
listenerItems = listenerItem.split(".")
if (listenerItems[0] == "1"):
address = ".".join(listenerItems[2][1:].split("|"))
if (len(address) == 0):
address = "127.0.0.1"
listeners.addElement(address + " : " + listenerItems[1])
index = index + 1
self._listenersCombo.setModel(listeners)
return;
#
# implement button actions
#
def startToggled(self, ev):
if (self._startButton.getModel().isSelected()):
try:
os.chdir(sys.path[0] + os.sep + "riacrawler" + os.sep + "scripts")
except Exception as e:
print >> sys.stderr, "RIA crawler scripts loading error", "I/O error({0}): {1}".format(e.errno, e.strerror)
self._startButton.setSelected(False)
return
phantomJsPath = self._phantomJsPathField.text
target = self._addressTextField.text
config = "crawler.config"
if (self._configFile):
config = self._configFile
listenerAddress = self._listenersCombo.getSelectedItem().replace(" ", "")
p = Popen("{0} --proxy={3} main.js --target={1} --config={2}".format(phantomJsPath, target, config, listenerAddress), shell=True)
self._running = True
self._requestsMadeInfo.setText("")
self._statesFoundInfo.setText("")
self._commandClient.startCrawling()
else:
if (self._running):
self._commandClient.stopCrawling()
self._running = False
def syncCrawlingState(self, result):
print "RIA crawling state: ", result
self._requestsMadeInfo.setText(str(result["requests_made"]))
self._statesFoundInfo.setText(str(result["states_detected"]))
if (result["running"] == False):
self._commandClient.stopCrawling()
self._running = False
self._startButton.setSelected(False)
def loadConfigClicked(self, ev):
openFile = JFileChooser();
openFile.showOpenDialog(None);
self._configFile = openFile.getSelectedFile()
#
# implement DocumentListener for _addressTextField
#
def removeUpdate(self, ev):
self.updateStartButton()
def insertUpdate(self, ev):
self.updateStartButton()
def updateStartButton(self):
self._startButton.setEnabled(len(self._addressTextField.text) > 0)
#
# implement IContextMenuFactory
#
def createMenuItems(self, contextMenuInvocation):
menuItemList = ArrayList()
context = contextMenuInvocation.getInvocationContext()
if (context == IContextMenuInvocation.CONTEXT_MESSAGE_VIEWER_REQUEST or context == IContextMenuInvocation.CONTEXT_MESSAGE_EDITOR_REQUEST or
context == IContextMenuInvocation.CONTEXT_PROXY_HISTORY or context == IContextMenuInvocation.CONTEXT_TARGET_SITE_MAP_TABLE):
self._contextMenuData = contextMenuInvocation.getSelectedMessages()
menuItemList.add(JMenuItem("Send to Phantom RIA Crawler", actionPerformed = self.menuItemClicked))
return menuItemList
def menuItemClicked(self, event):
if (self._running == True):
self._callbacks.issueAlert("Can't set data to Phantom RIA Crawler: crawling is running already.")
return;
dataIsSet = False;
for message in self._contextMenuData:
request = self._helpers.analyzeRequest(message)
url = request.getUrl().toString()
print url
if (url):
dataisSet = True;
self._addressTextField.setText(url)
def __init__(self):
self.listModel = DefaultListModel()
self.cbTitle = JLabel("Out-of-band Payloads")
self.cbTitle.setFont(self.cbTitle.getFont().deriveFont(14.0))
self.cbTitle.setFont(self.cbTitle.getFont().deriveFont(Font.BOLD))
self.cbSubTitle = JLabel(
"Add payloads to active scanner that interact "
"with out-of-band services (e.g., XSSHunter)")
self.cbSubTitle.setFont(self.cbSubTitle.getFont().deriveFont(12.0))
self.cbList = JList(self.listModel)
self.cbList.setCellRenderer(ListRenderer())
self.cbList.setVisibleRowCount(10)
self.listScrollPane = JScrollPane(self.cbList)
self.cbText = JTextField(actionPerformed=self.add_element)
self.cbRemoveButton = JButton("Remove",
actionPerformed=self.remove_element)
self.cbLoadButton = JButton("Load...", actionPerformed=self.add_file)
self.cbPasteButton = JButton("Paste",
actionPerformed=self.paste_elements)
self.cbClearButton = JButton("Clear",
actionPerformed=self.clear_elements)
self.cbAddButton = JButton("Add", actionPerformed=self.add_element)
self.cbDropDownLabel = JLabel("Payload Encoding: ")
self.cbDropDown = JComboBox()
self.cbDropDown.addItem("Non URL Encoded")
self.cbDropDown.addItem("URL Encoded")
self.cbDropDown.addItem("Both (two requests per payload)")
self.grpOOB = JPanel()
grpLayout = GroupLayout(self.grpOOB)
self.grpOOB.setLayout(grpLayout)
grpLayout.linkSize(SwingConstants.HORIZONTAL, self.cbRemoveButton,
self.cbLoadButton, self.cbPasteButton,
self.cbClearButton, self.cbAddButton)
grpLayout.setAutoCreateGaps(True)
grpLayout.setAutoCreateContainerGaps(True)
grpLayout.setHorizontalGroup(
grpLayout.createSequentialGroup().addGroup(
grpLayout.createParallelGroup().addComponent(
self.cbTitle).addGroup(
grpLayout.createParallelGroup().addComponent(
self.cbRemoveButton).addComponent(
self.cbLoadButton).addComponent(
self.cbPasteButton).addComponent(
self.cbClearButton)).addComponent(
self.cbAddButton).
addComponent(self.cbDropDownLabel)).addGroup(
grpLayout.createParallelGroup().addComponent(
self.cbSubTitle).addComponent(
self.listScrollPane).addComponent(
self.cbText).addComponent(self.cbDropDown)))
grpLayout.setVerticalGroup(grpLayout.createSequentialGroup().addGroup(
grpLayout.createParallelGroup().addComponent(self.cbTitle)
).addGroup(grpLayout.createParallelGroup().addComponent(
self.cbSubTitle)).addGroup(
grpLayout.createParallelGroup().addGroup(
grpLayout.createSequentialGroup().addComponent(
self.cbPasteButton).addComponent(
self.cbLoadButton).addComponent(
self.cbRemoveButton).addComponent(
self.cbClearButton)).addComponent(
self.listScrollPane)
).addGroup(grpLayout.createParallelGroup().addComponent(
self.cbAddButton).addComponent(self.cbText)).addGroup(
grpLayout.createParallelGroup().addComponent(
self.cbDropDownLabel).addComponent(self.cbDropDown)))
# Tab Layout
self.tab = JPanel()
tabLayout = GroupLayout(self.tab)
self.tab.setLayout(tabLayout)
tabLayout.setAutoCreateGaps(True)
tabLayout.setAutoCreateContainerGaps(True)
tabLayout.setHorizontalGroup(
tabLayout.createSequentialGroup().addGroup(
tabLayout.createParallelGroup().addComponent(
self.grpOOB, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)))
tabLayout.setVerticalGroup(tabLayout.createSequentialGroup().addGroup(
tabLayout.createParallelGroup().addComponent(
self.grpOOB, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)))
def __initLayout__(self):
self._levelComboBox = JComboBox()
levelComboBoxSize = Dimension(300, 30)
self._levelComboBox.setPreferredSize(levelComboBoxSize)
self._levelComboBox.setMaximumSize(levelComboBoxSize)
for level in range(0, 6):
self._levelComboBox.addItem(str(level))
self._techRenderedCheckBox = JCheckBox('Rendered', True)
self._techTimebasedCheckBox = JCheckBox('Time-based', True)
self._plugin_groups = {}
for plugin in plugins:
parent = plugin.__base__.__name__
if not self._plugin_groups.has_key(parent):
self._plugin_groups[parent] = []
self._plugin_groups[parent].append(plugin)
self._pluginCheckBoxes = []
for pluginGroup in self._plugin_groups.values():
for plugin in pluginGroup:
self._pluginCheckBoxes.append(PluginCheckBox(plugin))
self._positionReplaceCheckBox = JCheckBox('Replace', True)
self._positionAppendCheckBox = JCheckBox('Append', False)
displayItems = ({
'label':
'Level',
'components': (self._levelComboBox, ),
'description':
'Level of code context escape to perform (1-5, Default:0).'
}, {
'label':
'Techniques',
'components': (
self._techRenderedCheckBox,
self._techTimebasedCheckBox,
),
'description':
'Techniques R(endered) T(ime-based blind). Default: RT.'
}, {
'label':
'Template Engines',
'components':
self._pluginCheckBoxes,
'description':
'Force back-end template engine to this value(s).'
}, {
'label':
'Payload position',
'components': (
self._positionReplaceCheckBox,
self._positionAppendCheckBox,
),
'description':
'Scan payload position. This feature only appears in BurpExtension.'
})
layout = GroupLayout(self)
self.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
labelWidth = 200
hgroup = layout.createParallelGroup(GroupLayout.Alignment.LEADING)
vgroup = layout.createSequentialGroup()
for displayItem in displayItems:
label = JLabel(displayItem.get('label'))
label.setToolTipText(displayItem.get('description'))
_hgroup = layout.createSequentialGroup().addComponent(
label, labelWidth, labelWidth, labelWidth)
_vgroup = layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(label)
for component in displayItem.get('components'):
_hgroup.addComponent(component)
_vgroup.addComponent(component)
hgroup.addGroup(_hgroup)
vgroup.addGroup(_vgroup)
layout.setHorizontalGroup(hgroup)
layout.setVerticalGroup(vgroup)
def initUI(self):
layout = GroupLayout(self.pane)
self.pane.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
bc = settings.getBool("boot.use", False)
self.useBs = JCheckBox("Use bootstrap", bc)
binLabel = JLabel("Bootstrap file: ")
self.binTxt = JTextField()
self.binTxt.text = settings.getString("boot.path", "")
self.binBtn = JButton("Load", actionPerformed=mplab_chooseBootstrap)
entryLabel = JLabel("Bootstrap entry address (if bin): ")
self.entryTxt = JTextField()
self.entryTxt.text = settings.getString("boot.entry_adr", "")
prjLabel = JLabel("Bootstrap project: ")
self.prjTxt = JTextField()
self.prjTxt.text = settings.getString("boot.prj", "")
loadLabel = JLabel("Bootstrap load address (if bin): ")
self.loadTxt = JTextField()
self.loadTxt.text = settings.getString("boot.load_adr", "")
layout.setHorizontalGroup(layout.createSequentialGroup().addGroup(
layout.createParallelGroup().addComponent(
self.useBs).addComponent(prjLabel).addComponent(binLabel).
addComponent(entryLabel).addComponent(loadLabel)).addGroup(
layout.createParallelGroup()
# .addComponent(self.useBs)
.addComponent(self.prjTxt).addComponent(
self.binTxt).addComponent(self.entryTxt).addComponent(
self.loadTxt)).addGroup(
layout.createParallelGroup().addComponent(
self.binBtn)))
layout.setVerticalGroup(layout.createSequentialGroup().addGroup(
layout.createParallelGroup().addComponent(self.useBs)
).addGroup(
layout.createParallelGroup().addComponent(prjLabel).addComponent(
self.prjTxt, GroupLayout.PREFERRED_SIZE,
GroupLayout.DEFAULT_SIZE, GroupLayout.PREFERRED_SIZE)
).addGroup(
layout.createParallelGroup().addComponent(binLabel).addComponent(
self.binTxt, GroupLayout.PREFERRED_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.PREFERRED_SIZE).addComponent(self.binBtn)
).addGroup(
layout.createParallelGroup().addComponent(entryLabel).addComponent(
self.entryTxt, GroupLayout.PREFERRED_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.PREFERRED_SIZE)).addGroup(
layout.createParallelGroup().addComponent(
loadLabel).addComponent(self.loadTxt,
GroupLayout.PREFERRED_SIZE,
GroupLayout.DEFAULT_SIZE,
GroupLayout.PREFERRED_SIZE)))
self.panel.add(self.pane, BorderLayout.CENTER)
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
self.helpers = callbacks.getHelpers()
callbacks.setExtensionName("Missing Scanner Checks")
self.out = callbacks.getStdout()
# define all checkboxes
self.cbPassiveChecks = self.defineCheckBox("Passive Scanner Checks")
self.cbDOMXSS = self.defineCheckBox("DOM XSS", False)
self.cbDOMXSSSources = self.defineCheckBox("Sources", False)
self.cbDOMXSSSinks = self.defineCheckBox("Sinks")
self.cbDOMXSSjQuerySinks = self.defineCheckBox("jQuery Sinks", False)
self.grpDOMXSSSettings = JPanel()
self.grpDOMXSSSettings.add(self.cbDOMXSSSources)
self.grpDOMXSSSettings.add(self.cbDOMXSSSinks)
self.grpDOMXSSSettings.add(self.cbDOMXSSjQuerySinks)
self.cbSTS = self.defineCheckBox("Strict Transport Security")
self.lblSTSMin = JLabel("Minimum acceptable max-age")
self.inSTSMin = JTextField(str(STSMinimum), 9, actionPerformed=self.setSTSMinimum) # TODO: actionPerformed only fires on enter key - focus lost would be better
self.inSTSMin.setToolTipText("Enter the minimum max-age value which is considered as acceptable. Press return to change setting!")
self.grpSTSSettings = JPanel()
self.grpSTSSettings.add(self.lblSTSMin)
self.grpSTSSettings.add(self.inSTSMin)
self.cbXCTO = self.defineCheckBox("Content Sniffing")
self.cbXXP = self.defineCheckBox("Client-side XSS Filter Configuration")
self.cbRedirToHTTPS = self.defineCheckBox("Redirection from HTTP to HTTPS")
self.btnSave = JButton("Set as default", actionPerformed=self.saveConfig)
self.btnRestore = JButton("Restore", actionPerformed=self.restoreConfig)
self.grpConfig = JPanel()
self.grpConfig.add(self.btnSave)
self.grpConfig.add(self.btnRestore)
self.restoreConfig()
# definition of config tab
self.tab = JPanel()
layout = GroupLayout(self.tab)
self.tab.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
layout.setHorizontalGroup(
layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(self.cbPassiveChecks)
)
.addGroup(layout.createParallelGroup()
.addComponent(self.cbDOMXSS)
.addComponent(self.cbSTS)
.addComponent(self.cbXCTO)
.addComponent(self.cbXXP)
.addComponent(self.cbRedirToHTTPS)
)
.addGroup(layout.createParallelGroup()
.addComponent(self.grpDOMXSSSettings, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
.addComponent(self.grpSTSSettings, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
.addComponent(self.grpConfig, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
)
layout.setVerticalGroup(
layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(self.cbPassiveChecks)
.addComponent(self.cbDOMXSS)
.addComponent(self.grpDOMXSSSettings, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
.addGroup(layout.createParallelGroup()
.addComponent(self.cbSTS)
.addComponent(self.grpSTSSettings, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
.addComponent(self.cbXCTO)
.addComponent(self.cbXXP)
.addComponent(self.cbRedirToHTTPS)
.addComponent(self.grpConfig, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE, GroupLayout.PREFERRED_SIZE)
)
self.domXSSSourcesRE = re.compile("(location\s*[\[.])|([.\[]\s*[\"']?\s*(arguments|dialogArguments|innerHTML|write(ln)?|open(Dialog)?|showModalDialog|cookie|URL|documentURI|baseURI|referrer|name|opener|parent|top|content|self|frames)\W)|(localStorage|sessionStorage|Database)")
# NOTE: done some optimizations here, original RE caused too much noise
# - added leading dot in the first part - original recognized "<a href=..." etc.
# - removed "value" in first part
self.domXSSSinksRE = re.compile("(\.(src|href|data|location|code|action)\s*[\"'\]]*\s*\+?\s*=)|((replace|assign|navigate|getResponseHeader|open(Dialog)?|showModalDialog|eval|evaluate|execCommand|execScript|setTimeout|setInterval)\s*[\"'\]]*\s*\()")
self.domXSSjQuerySinksRE = re.compile("after\(|\.append\(|\.before\(|\.html\(|\.prepend\(|\.replaceWith\(|\.wrap\(|\.wrapAll\(|\$\(|\.globalEval\(|\.add\(|jQUery\(|\$\(|\.parseHTML\(")
self.headerSTSRE = re.compile("^Strict-Transport-Security:.*?max-age=\"?(\d+)\"?", re.I) # TODO: multiple max-age directives cause confusion!
self.headerXCTORE = re.compile("^X-Content-Type-Options:\s*nosniff\s*$", re.I)
self.headerXXP = re.compile("^X-XSS-Protection:\s*(\d)(?:\s*;\s*mode\s*=\s*\"?(block)\"?)?", re.I)
self.headerLocationHTTPS = re.compile("^(?:Content-)?Location:\s*(https://.*)$", re.I)
callbacks.registerScannerCheck(self)
callbacks.addSuiteTab(self)
def __initLayout__( self ):
self._levelComboBox = JComboBox()
levelComboBoxSize = Dimension( 300, 30 )
self._levelComboBox.setPreferredSize( levelComboBoxSize )
self._levelComboBox.setMaximumSize( levelComboBoxSize )
for level in range( 0, 6 ):
self._levelComboBox.addItem( str( level ) )
self._techRenderedCheckBox = JCheckBox( 'Rendered', True )
self._techTimebasedCheckBox = JCheckBox( 'Time-based', True )
self._plugin_groups = {}
for plugin in plugins:
parent = plugin.__base__.__name__
if not self._plugin_groups.has_key( parent ):
self._plugin_groups[ parent ] = []
self._plugin_groups[ parent ].append( plugin )
self._pluginCheckBoxes = []
for pluginGroup in self._plugin_groups.values():
for plugin in pluginGroup:
self._pluginCheckBoxes.append( PluginCheckBox( plugin ) )
self._positionReplaceCheckBox = JCheckBox( 'Replace', True )
self._positionAppendCheckBox = JCheckBox( 'Append', False )
displayItems = (
{
'label': 'Level',
'components': ( self._levelComboBox, ),
'description': 'Level of code context escape to perform (1-5, Default:0).'
},
{
'label': 'Techniques',
'components': ( self._techRenderedCheckBox, self._techTimebasedCheckBox, ),
'description': 'Techniques R(endered) T(ime-based blind). Default: RT.'
},
{
'label': 'Template Engines',
'components': self._pluginCheckBoxes,
'description': 'Force back-end template engine to this value(s).'
},
{
'label': 'Payload position',
'components': ( self._positionReplaceCheckBox, self._positionAppendCheckBox, ),
'description': 'Scan payload position. This feature only appears in BurpExtension.'
}
)
layout = GroupLayout( self )
self.setLayout( layout )
layout.setAutoCreateGaps( True )
layout.setAutoCreateContainerGaps( True )
labelWidth = 200
hgroup = layout.createParallelGroup( GroupLayout.Alignment.LEADING )
vgroup = layout.createSequentialGroup()
for displayItem in displayItems:
label = JLabel( displayItem.get( 'label' ) )
label.setToolTipText( displayItem.get( 'description' ) )
_hgroup = layout.createSequentialGroup().addComponent( label, labelWidth, labelWidth, labelWidth )
_vgroup = layout.createParallelGroup( GroupLayout.Alignment.BASELINE ).addComponent( label )
for component in displayItem.get( 'components' ):
_hgroup.addComponent( component )
_vgroup.addComponent( component )
hgroup.addGroup( _hgroup )
vgroup.addGroup( _vgroup )
layout.setHorizontalGroup( hgroup )
layout.setVerticalGroup( vgroup )
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
self.helpers = callbacks.getHelpers()
callbacks.setExtensionName("Missing Scanner Checks")
self.out = callbacks.getStdout()
# define all checkboxes
self.cbPassiveChecks = self.defineCheckBox("Passive Scanner Checks")
self.cbDOMXSS = self.defineCheckBox("DOM XSS", False)
self.cbDOMXSSSources = self.defineCheckBox("Sources", False)
self.cbDOMXSSSinks = self.defineCheckBox("Sinks")
self.cbDOMXSSjQuerySinks = self.defineCheckBox("jQuery Sinks", False)
self.grpDOMXSSSettings = JPanel()
self.grpDOMXSSSettings.add(self.cbDOMXSSSources)
self.grpDOMXSSSettings.add(self.cbDOMXSSSinks)
self.grpDOMXSSSettings.add(self.cbDOMXSSjQuerySinks)
self.cbSTS = self.defineCheckBox("Strict Transport Security")
self.lblSTSMin = JLabel("Minimum acceptable max-age")
self.inSTSMin = JTextField(
str(STSMinimum), 9, actionPerformed=self.setSTSMinimum
) # TODO: actionPerformed only fires on enter key - focus lost would be better
self.inSTSMin.setToolTipText(
"Enter the minimum max-age value which is considered as acceptable. Press return to change setting!"
)
self.grpSTSSettings = JPanel()
self.grpSTSSettings.add(self.lblSTSMin)
self.grpSTSSettings.add(self.inSTSMin)
self.cbXCTO = self.defineCheckBox("Content Sniffing")
self.cbXXP = self.defineCheckBox(
"Client-side XSS Filter Configuration")
self.cbRedirToHTTPS = self.defineCheckBox(
"Redirection from HTTP to HTTPS")
self.btnSave = JButton("Set as default",
actionPerformed=self.saveConfig)
self.btnRestore = JButton("Restore",
actionPerformed=self.restoreConfig)
self.grpConfig = JPanel()
self.grpConfig.add(self.btnSave)
self.grpConfig.add(self.btnRestore)
self.restoreConfig()
# definition of config tab
self.tab = JPanel()
layout = GroupLayout(self.tab)
self.tab.setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
layout.setHorizontalGroup(layout.createSequentialGroup().addGroup(
layout.createParallelGroup().addComponent(self.cbPassiveChecks)
).addGroup(layout.createParallelGroup().addComponent(
self.cbDOMXSS).addComponent(self.cbSTS).addComponent(
self.cbXCTO).addComponent(self.cbXXP).addComponent(
self.cbRedirToHTTPS)).addGroup(
layout.createParallelGroup().addComponent(
self.grpDOMXSSSettings, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE).addComponent(
self.grpSTSSettings,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE).addComponent(
self.grpConfig, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)))
layout.setVerticalGroup(layout.createSequentialGroup().addGroup(
layout.createParallelGroup().addComponent(
self.cbPassiveChecks).addComponent(self.cbDOMXSS).addComponent(
self.grpDOMXSSSettings, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)).addGroup(
layout.createParallelGroup().addComponent(
self.cbSTS).addComponent(
self.grpSTSSettings,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)).addComponent(
self.cbXCTO).addComponent(
self.cbXXP).addComponent(
self.cbRedirToHTTPS).addComponent(
self.grpConfig,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE))
self.domXSSSourcesRE = re.compile(
"(location\s*[\[.])|([.\[]\s*[\"']?\s*(arguments|dialogArguments|innerHTML|write(ln)?|open(Dialog)?|showModalDialog|cookie|URL|documentURI|baseURI|referrer|name|opener|parent|top|content|self|frames)\W)|(localStorage|sessionStorage|Database)"
)
# NOTE: done some optimizations here, original RE caused too much noise
# - added leading dot in the first part - original recognized "<a href=..." etc.
# - removed "value" in first part
self.domXSSSinksRE = re.compile(
"(\.(src|href|data|location|code|action)\s*[\"'\]]*\s*\+?\s*=)|((replace|assign|navigate|getResponseHeader|open(Dialog)?|showModalDialog|eval|evaluate|execCommand|execScript|setTimeout|setInterval)\s*[\"'\]]*\s*\()"
)
self.domXSSjQuerySinksRE = re.compile(
"after\(|\.append\(|\.before\(|\.html\(|\.prepend\(|\.replaceWith\(|\.wrap\(|\.wrapAll\(|\$\(|\.globalEval\(|\.add\(|jQUery\(|\$\(|\.parseHTML\("
)
self.headerSTSRE = re.compile(
"^Strict-Transport-Security:.*?max-age=\"?(\d+)\"?",
re.I) # TODO: multiple max-age directives cause confusion!
self.headerXCTORE = re.compile(
"^X-Content-Type-Options:\s*nosniff\s*$", re.I)
self.headerXXP = re.compile(
"^X-XSS-Protection:\s*(\d)(?:\s*;\s*mode\s*=\s*\"?(block)\"?)?",
re.I)
self.headerLocationHTTPS = re.compile(
"^(?:Content-)?Location:\s*(https://.*)$", re.I)
callbacks.registerScannerCheck(self)
callbacks.addSuiteTab(self)
def __init__(self):
super(WorkHelper, self).__init__()
self.clipboard = Toolkit.getDefaultToolkit().getSystemClipboard()
#self.initUI()
#def initUI(self):
#panel = JPanel()
#self.getContentPane().add(panel)
#############################################################
# Layout
layout = GroupLayout(self.getContentPane())
self.getContentPane().setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
#############################################################
#############################################################
# Scroll Area Input + Output
Larea1 = JLabel("InputArea:")
Larea2 = JLabel("OutputArea:")
Sarea1 = JScrollPane()
Sarea2 = JScrollPane()
self.area1 = JTextArea()
self.area1.setToolTipText("Input Area")
self.area1.setEditable(True)
self.area1.setBorder(BorderFactory.createLineBorder(Color.gray))
Sarea1.setPreferredSize(Dimension(300,100))
Sarea1.getViewport().setView((self.area1))
self.area2 = JTextArea()
self.area2.setToolTipText("Output Area")
self.area2.setEditable(False)
self.area2.setBorder(BorderFactory.createLineBorder(Color.gray))
Sarea2.setPreferredSize(Dimension(300,100))
Sarea2.getViewport().setView((self.area2))
#############################################################
#############################################################
# Buttons
self.cCurly = JCheckBox("Curly");
self.cCurly.setToolTipText("When 'Checked' Curly Brackets will surround the Categories")
self.cCurly.setSelected(1)
self.cCtClipB = JCheckBox("Auto-Copy");
self.cCtClipB.setToolTipText("When 'Checked' after the Categories are created they will added to the clipboard")
self.cCtClipB.setSelected(1)
self.cSemiC = JCheckBox("SemiColumn");
self.cSemiC.setToolTipText("When 'Checked' after the Categories are created at the end will be a semicolomn")
self.cSemiC.setSelected(1)
bRemoveNBSP_L = JButton("Clean LText", actionPerformed=self.bRemoveNBSP_L)
bRemoveNBSP_L.setToolTipText("Removes Spaces, Tabs from the start of every text line from the input Area")
bRemoveNBSP_R = JButton("Clean RText", actionPerformed=self.bRemoveNBSP_R)
bRemoveNBSP_R.setToolTipText("Removes Spaces, Tabs from the end of every text line from the input Area")
bCopyToInput = JButton("Copy to Input", actionPerformed=self.bCopyToInput)
bCopyToInput.setToolTipText("Copy the text from the Output Area to the Input Area for further Operations")
bClear = JButton("Clear", actionPerformed=self.bClear)
bClear.setToolTipText("Clears the text form both Input and Output text Areas")
self.iStart = JTextField(maximumSize=Dimension(40,25))
self.iStart.setToolTipText("The Start Index for the Making of the Categories")
self.RThis = JTextField()
self.RThis = JTextField(maximumSize=Dimension(120,25))
self.RThis.setToolTipText("Text to be replaced or The Starting C_Index")
self.RThat = JTextField()
self.RThat = JTextField(maximumSize=Dimension(120,25))
self.RThat.setToolTipText("Text to be placed or The Finish C_Index")
bSandReplace = JButton("Replace Text", actionPerformed=self.bSandReplace)
bSandReplace.setToolTipText("Replace the text from This with Thext from That in the Text from the Input Area and displays it in the Output Area")
bcCat = JButton("CreatCateg", actionPerformed=self.bcCat)
bcCat.setToolTipText("Create a categorical form starting C_Index to finish C_Index; Use the above text boxes to define the indexes")
bC_S = JButton("Create _Series", actionPerformed=self.bC_S)
bC_S.setToolTipText("Create a series form starting C_Index to finish C_Index; Use the above text boxes to define the indexes; It will create a series for every row in the Input Area")
bM_Categories = JButton("Categories", actionPerformed=self.mCategories)
bM_Categories.setToolTipText("Make Categories using the lines from the Input Area")
#bM_Categories = JButton(maximumSize=Dimension(40,25))
# de incercat daca merge cu ; sa grupezi in [dsa] elementele
#############################################################
#############################################################
# Aplication Layout 2 groups one Horizontal and one Vertical
layout.setHorizontalGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(Larea1)
.addComponent(Sarea1)
.addComponent(Sarea2)
.addComponent(bCopyToInput)
.addComponent(Larea2))
.addGroup(layout.createParallelGroup()
.addGroup(layout.createSequentialGroup()
.addComponent(bM_Categories)
.addComponent(self.iStart))
.addGroup(layout.createSequentialGroup()
.addComponent(self.cCurly)
.addComponent(self.cSemiC)
.addComponent(self.cCtClipB))
.addGroup(layout.createSequentialGroup()
.addComponent(bRemoveNBSP_L)
.addComponent(bRemoveNBSP_R))
.addGroup(layout.createSequentialGroup()
.addComponent(self.RThis)
.addComponent(self.RThat))
.addGroup(layout.createSequentialGroup()
.addComponent(bSandReplace)
.addComponent(bcCat))
.addGroup(layout.createSequentialGroup()
.addComponent(bC_S))
.addComponent(bClear))
)
layout.setVerticalGroup(layout.createSequentialGroup()
.addComponent(Larea1)
.addGroup(layout.createParallelGroup()
.addComponent(Sarea1)
.addGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(bM_Categories)
.addComponent(self.iStart))
.addGroup(layout.createParallelGroup()
.addComponent(self.cCurly)
.addComponent(self.cSemiC)
.addComponent(self.cCtClipB))
.addGroup(layout.createParallelGroup()
.addComponent(bRemoveNBSP_L)
.addComponent(bRemoveNBSP_R))
.addGroup(layout.createParallelGroup()
.addComponent(self.RThis)
.addComponent(self.RThat))
.addGroup(layout.createParallelGroup()
.addComponent(bSandReplace)
.addComponent(bcCat))
.addGroup(layout.createParallelGroup()
.addComponent(bC_S))
)
)
.addGroup(layout.createParallelGroup()
.addComponent(bCopyToInput)
.addComponent(bClear))
.addComponent(Larea2)
.addGroup(layout.createParallelGroup()
.addComponent(Sarea2))
)
#layout.linkSize(SwingConstants.HORIZONTAL, [ok, bCopyToInput, close, bM_Categories])
layout.linkSize(SwingConstants.HORIZONTAL, [self.RThis,self.RThat,bRemoveNBSP_L,bRemoveNBSP_R,bCopyToInput,bM_Categories,bSandReplace,bcCat,bC_S])
#layout.linkSize(SwingConstants.HORIZONTAL, [self.cCurly,bM_Categories])
#############################################################
#############################################################
# Aplication Settings
self.pack()
#self.setPreferredSize(Dimension(1000, 1000))
self.setTitle("Workhelper")
self.setSize(800, 500)
self.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
self.setLocationRelativeTo(None)
self.setVisible(True)
def __init__(self):
TrafficMod.__init__(self)
self.setLayout(FlowLayout(FlowLayout.LEADING))
settings_panel = JPanel()
# settings_panel.setBorder(LineBorder(Color.RED))
layout = GroupLayout(settings_panel)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
settings_panel.setLayout(layout)
self.add(settings_panel)
# Horizontal group
horizontal = layout.createParallelGroup()
layout.setHorizontalGroup(
layout.createSequentialGroup().addGroup(horizontal))
self.horizontal_layout = horizontal
# Vertical group
vertical = layout.createSequentialGroup()
layout.setVerticalGroup(vertical)
self.vertical_layout = vertical
# Main title, label only
main_title = JLabel(self.title)
main_title.setFont(get_title_font(main_title))
horizontal.addGroup(
layout.createSequentialGroup().addComponent(main_title))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(main_title))
rates_combo = ui_components.get_rate_combo()
# Base rate
rate_lbl = JLabel("Rate")
rate_lbl.setToolTipText(
"Simulates a network connector with a limited throughput")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(rate_lbl).addComponent(
self.rate_textfield).addComponent(rates_combo))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
rate_lbl).addComponent(
self.rate_textfield).addComponent(rates_combo))
# Packet overhead
packet_overhead_lbl = JLabel("Packet Overhead")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
packet_overhead_lbl).addComponent(self.packet_textfield))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
packet_overhead_lbl).addComponent(self.packet_textfield))
layout.linkSize(SwingConstants.HORIZONTAL, packet_overhead_lbl,
rate_lbl)
# Cell size
cell_size_lbl = JLabel("Cell Size")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
cell_size_lbl).addComponent(self.cellsize_textfield))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
cell_size_lbl).addComponent(self.cellsize_textfield))
layout.linkSize(SwingConstants.HORIZONTAL, cell_size_lbl, rate_lbl)
# Cell overhead
cell_overhead_lbl = JLabel("Cell Overhead")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
cell_overhead_lbl).addComponent(self.celloverhead_textfield))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
cell_overhead_lbl).addComponent(
self.celloverhead_textfield))
layout.linkSize(SwingConstants.HORIZONTAL, cell_overhead_lbl, rate_lbl)
class ChatClient(JFrame):
## Constructor method, receives the variables from the ChatApp class as parameters
def __init__(self, name, greeting, tn):
'''Constructor, initialises base class & assigns variables
'''
# Call to the super method to take care of the base class(es)
super(ChatClient, self).__init__()
# Assign the relevent variable names
self.username=name
self.greeting=greeting
self.tn = tn
self.no_users=[]
# Initiate the Threaded function for receiving messages
t1=Thread(target=self.recvFunction)
# Set to daemon
t1.daemon=True
t1.start()
#Call the main UI
uI=self.clientUI()
## Main GUI building function
def clientUI(self):
'''ClientUI and Widget creation
'''
# Colours
foreground_colour = Color(30,57,68)
background_colour = Color(247,246,242)
window_background = Color(145,190,210)
# Borders
self.border2=BorderFactory.createLineBorder(foreground_colour,1, True)
# Fonts
self.font= Font("Ubuntu Light", Font.BOLD, 20)
self.label_font= Font("Ubuntu Light", Font.BOLD, 17)
self.label_2_font= Font( "Ubuntu Light",Font.BOLD, 12)
self.btn_font=Font("Ubuntu Light", Font.BOLD, 15)
# Set the layout parameters
self.client_layout=GroupLayout(self.getContentPane())
self.getContentPane().setLayout(self.client_layout)
self.getContentPane().setBackground(window_background)
self.client_layout.setAutoCreateGaps(True)
self.client_layout.setAutoCreateContainerGaps(True)
self.setPreferredSize(Dimension(400, 450))
# Create widgets and assemble the GUI
# Main display area
self.main_content=JTextPane()
self.main_content.setBackground(background_colour)
#self.main_content.setForeground(foreground_colour)
self.main_content.setEditable(False)
# Message entry area
self.message=JTextArea( 2,2, border=self.border2, font=self.label_font, keyPressed=self.returnKeyPress)
self.message.requestFocusInWindow()
self.message.setBackground(background_colour)
self.message.setForeground(foreground_colour)
self.message.setLineWrap(True)
self.message.setWrapStyleWord(True)
self.message.setBorder(BorderFactory.createEmptyBorder(3,3,3,3))
self.message.getInputMap().put(KeyStroke.getKeyStroke(KeyEvent.VK_ENTER,0), self.returnKeyPress)
# BUttons
quit_btn=JButton("Quit!", actionPerformed=ChatApp().closeEvent, border=self.border2, font=self.btn_font)
go_btn=JButton("Send", actionPerformed=self.grabText, border=self.border2, font=self.btn_font)
quit_btn.setBackground(background_colour)
go_btn.setBackground(background_colour)
quit_btn.setForeground(foreground_colour)
go_btn.setForeground(foreground_colour)
# Make scrollable
self.scroll_content=JScrollPane(self.main_content)
self.scroll_content.setPreferredSize(Dimension(150,275))
self.scroll_content.setHorizontalScrollBarPolicy(ScrollPaneConstants.HORIZONTAL_SCROLLBAR_NEVER)
self.scroll_content.setViewportView(self.main_content)
self.scroll_content.setBackground(Color.WHITE)
self.scroll_message=JScrollPane(self.message)
self.scroll_message.setPreferredSize(Dimension(150,20))
self.scroll_message.setVerticalScrollBarPolicy(ScrollPaneConstants.VERTICAL_SCROLLBAR_ALWAYS)
# Test user label, still not updating after first round of messages
self.user_label=JLabel(" Users online : %s "%(str(len(self.no_users))),JLabel.RIGHT, font=self.label_2_font)
# Assemble the components
# Horizontal layout
self.client_layout.setHorizontalGroup(self.client_layout.createParallelGroup()
.addComponent(self.scroll_content)
.addGroup(self.client_layout.createParallelGroup(GroupLayout.Alignment.CENTER)
.addComponent(self.scroll_message))
.addGroup(self.client_layout.createSequentialGroup()
.addComponent(quit_btn)
.addComponent(go_btn).addGap(20))
.addGroup(self.client_layout.createParallelGroup()
.addComponent(self.user_label))
)
# Vertical layout
self.client_layout.setVerticalGroup(self.client_layout.createSequentialGroup()
.addGroup(self.client_layout.createParallelGroup()
.addComponent(self.scroll_content))
.addComponent(self.scroll_message)
.addGroup(self.client_layout.createParallelGroup()
.addComponent(quit_btn)
.addComponent(go_btn))
.addGroup(self.client_layout.createParallelGroup()
.addComponent(self.user_label))
)
# Finalise the GUI
self.client_layout.linkSize(SwingConstants.HORIZONTAL, [quit_btn,go_btn, self.user_label])
self.pack()
self.message.requestFocusInWindow()
self.setTitle(">>> Client %s <<<"%self.username)
self.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
self.setLocationRelativeTo(None)
self.setVisible(True)
# Display the server greeting
self.appendText('\n'+self.greeting+'\n')
## Function responsible for receiving and processing new messages
def recvFunction(self):
'''A function to control the receiving of data from the connection
'''
# While the connection is available
while self.tn:
# Try to receive data using "<<<" as the delimiter
try:
message = self.tn.read_until('<<<')
# If a message is received
if message:
garb, message=message.split('>>>')
message, garb = message.split('<<<')
message = ('\n'+message+'\n')
# Call the append text function
self.appendText(message)
# Except if there is no data available
except:
#print('No message')
pass
## Event driven function to retrieve and send data to the server
def grabText(self, event):
'''Function to repeatedly grab new messages entered into the text
area and display them in the main text area. Resets the entry area
'''
# Grab the text from the text area
text=self.message.getText()
# Don't allow an empty string through
if text=='':
return
text=text.strip()
# Call the append text function
self.appendText('\nYou : '+text+'\n', self.username)
# Reset the text to be empty and grab focus so that it is ready for new text input
self.message.requestFocusInWindow()
self.message.setText('')
# Send the message to the server
data=text.encode()
self.tn.write(data+'\r\n')
## Function to handle appending of messages
def appendText(self, message, user=None):
'''This function takes care of appending any new messages to the content area
'''
message_label=JTextArea(message,2,3, font=self.label_2_font)
# If this is a message from the grab text function, create a new label, assign it's colours
if user!=None:
message_label.setBackground(Color(240,240,240))
message_label.setForeground(Color(129,129,129))
# Otherwise set the format for receive function (no user passed in)
else:
message_label.setBackground(Color(215,215,215))
message_label.setForeground(Color(40,153,153))
# Format and style options for the new message labels
message_label.setEditable(False)
message_label.setLineWrap(True)
message_label.setWrapStyleWord(True)
message_label.setBorder(BorderFactory.createLineBorder( Color(247,246,242),4))
# Sets the positioning of messages
self.main_content.setCaretPosition(self.main_content.getDocument().getLength())
doc = self.main_content.getStyledDocument()
attr=SimpleAttributeSet()
self.main_content.insertComponent(message_label)
# Essential for jtextarea to be able to stack message
doc.insertString( self.main_content.getDocument().getLength(),'\n ', attr)
# Not sure if needed
self.main_content.repaint()
### This is a late edit so it isn't included in the documentation. Basically trying to dynamically update the number
### of users label at runtime. Works for incrementing the value but not decrementing it.
print(message)
# Only split the message if there are enough values to split (greeting messages differ in format to chat messages)
try:
user, text=message.split(' : ')
except:
return
#print('Split values are %s %s'%(user, text))
user=str(user.strip())
#print(self.no_users)
#print(user+' : '+text)
# If the user already in the list, pass
if user in self.no_users:
if text == ('User %s amach sa teach !'%user):
self.no_users.remove(user)
print('User % removed'%user)
else:
#print('User %s not in list'%user)
if str(user) == 'You':
#print('User is equal to "You"')
return
self.no_users.append(user)
print('User appended')
self.number_users=len(self.no_users)
#print('Length of user list is '+str(self.number_users))
self.user_label2=JLabel(" Users online : %s "%str(len(self.no_users)),JLabel.RIGHT, font=self.label_2_font)
#print('Label created')
#print('Attempt to replace label')
self.client_layout.replace(self.user_label, self.user_label2)
self.user_label = self.user_label2
self.user_label.repaint()
self.user_label.revalidate()
print('Label updated')
## Function to control return button press in message field
def returnKeyPress(self,event):
'''This function creates an object for return key press when inside the message entry area,
creates an object of KeyAdapter and tests keycode for a match, responds with grab text callback
'''
key_object=Key()
key_value=key_object.keyPressed(event)
if key_value == 10:
self.grabText(event)
def __init__(self):
super(WorkHelper, self).__init__()
self.clipboard = Toolkit.getDefaultToolkit().getSystemClipboard()
#############################################################
# Layout:
layout = GroupLayout(self.getContentPane())
self.getContentPane().setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
#############################################################
#############################################################
# Frame Area:
Larea1 = JLabel("InputArea:")
Sarea1 = JScrollPane()
self.area1 = JTextArea()
self.area1.setToolTipText("Input Area")
self.area1.setEditable(True)
self.area1.setBorder(BorderFactory.createLineBorder(Color.gray))
Sarea1.setPreferredSize(Dimension(300,100))
Sarea1.getViewport().setView((self.area1))
bClear = JButton("Clear", actionPerformed=self.bClear)
bClear.setToolTipText("Clears the text form both Input and Output text Areas")
bCopyToInput = JButton("Copy to Input", actionPerformed=self.bCopyToInput)
bCopyToInput.setToolTipText("Copy the text from the Output Area to the Input Area for further Operations")
self.cCtClipB = JCheckBox("Auto-Copy");
self.cCtClipB.setToolTipText("When 'Checked' after the Categories are created they will added to the clipboard")
self.cCtClipB.setSelected(1)
Larea2 = JLabel("OutputArea:")
Sarea2 = JScrollPane()
self.area2 = JTextArea()
self.area2.setToolTipText("Output Area")
self.area2.setEditable(False)
self.area2.setBorder(BorderFactory.createLineBorder(Color.gray))
Sarea2.setPreferredSize(Dimension(300,100))
Sarea2.getViewport().setView((self.area2))
#############################################################
# Tabbed Area:
tabPane = JTabbedPane(JTabbedPane.TOP)
self.getContentPane().add(tabPane)
#####################################################
# Text Edit pane
panel_TEdit = JPanel()
layout2 = GroupLayout(panel_TEdit)
layout2.setAutoCreateGaps(True)
layout2.setAutoCreateContainerGaps(True)
panel_TEdit.setLayout(layout2)
bRemoveNBSP_L = JButton("Clean LText", actionPerformed=self.bRemoveNBSP_L)
bRemoveNBSP_L.setToolTipText("Removes Spaces, Tabs from the start of every text line from the input Area")
bRemoveNBSP_R = JButton("Clean RText", actionPerformed=self.bRemoveNBSP_R)
bRemoveNBSP_R.setToolTipText("Removes Spaces, Tabs from the end of every text line from the input Area")
self.ReplaceThis = JTextField()
self.ReplaceThis = JTextField(maximumSize=Dimension(120,25))
self.ReplaceThis.setToolTipText("Text to be replaced")
self.ReplaceThat = JTextField()
self.ReplaceThat = JTextField(maximumSize=Dimension(120,25))
self.ReplaceThat.setToolTipText("Text to be placed")
bSandReplace = JButton("Replace Text", actionPerformed=self.bSandReplace)
bSandReplace.setToolTipText("Replace the text from This with Text from That in the Text from the Input Area and displays it in the Output Area")
bRemNumbers = JButton("Rem Numbers", actionPerformed=self.RemNumbers)
bRemNumbers.setToolTipText("Removes numbers from the start of every line")
#####################################################
# Dimension pane
panel_Dimensions = JPanel()
layout3 = GroupLayout(panel_Dimensions)
layout3.setAutoCreateGaps(True)
layout3.setAutoCreateContainerGaps(True)
panel_Dimensions.setLayout(layout3)
self.cCurly = JCheckBox("Curly");
self.cCurly.setToolTipText("When 'Checked' Curly Brackets will surround the Categories")
self.cCurly.setSelected(1)
self.cSemiC = JCheckBox("SemiColumn");
self.cSemiC.setToolTipText("When 'Checked' after the Categories are created at the end will be a semicolomn")
self.cSemiC.setSelected(1)
self.iStart = JTextField(maximumSize=Dimension(40,25))
self.iStart.setToolTipText("The Start Index for the Making of the Categories")
self.RThis = JTextField()
self.RThis = JTextField(maximumSize=Dimension(120,25))
self.RThis.setToolTipText("The Starting Index used in creating the Categorical")
self.RThat = JTextField()
self.RThat = JTextField(maximumSize=Dimension(120,25))
self.RThat.setToolTipText("The Finish Index used in creating the Categorical")
optioncCategories = JLabel("Options:")
bcCat = JButton("CreatCateg", actionPerformed=self.bcCat)
bcCat.setToolTipText("Create a categorical form starting C_Index to finish C_Index; Use the text boxes to define the indexes")
bM_Categories = JButton("Categories", actionPerformed=self.mCategories)
bM_Categories.setToolTipText("Make Categories using the lines from the Input Area. Use it to define Categorical questions.")
#####################################################
# ConfirmIt pane
panel_ConfirmIt = JPanel()
layout4 = GroupLayout(panel_ConfirmIt)
layout4.setAutoCreateGaps(True)
layout4.setAutoCreateContainerGaps(True)
panel_ConfirmIt.setLayout(layout4)
self.PID = JTextField()
self.PID = JTextField(maximumSize=Dimension(120,25))
self.PID.setToolTipText("The PID number used for creating links with PID and ids from every line of the Input Area")
bClinks = JButton("Create Links", actionPerformed=self.bClinks)
bClinks.setToolTipText("Create links for a project using PID and ID, ID`s are read from every line of the Input Area")
bClinksNA = JButton("Create Links NA ", actionPerformed=self.bClinksNA)
bClinksNA.setToolTipText("Create links for a project using PID and ID`s from the standard sample test for US")
bClinksCA = JButton("Create Links CA", actionPerformed=self.bClinksCA)
bClinksCA.setToolTipText("Create links for a project using PID and ID`s from the standard sample test for CA")
self.Width = JTextField()
self.Width = JTextField(maximumSize=Dimension(120,25))
self.Width.setToolTipText("The Width used in creating the DIV html tag, note the dimension used is in px")
baddDIVt = JButton("Add DIV tag", actionPerformed=self.baddDIVt)
baddDIVt.setToolTipText("Create a DIV tag for every line in the Input Area")
#####################################################
# Statistics pane
panel_Statistics = JPanel()
layout5 = GroupLayout(panel_Statistics)
layout5.setAutoCreateGaps(True)
layout5.setAutoCreateContainerGaps(True)
panel_Statistics.setLayout(layout5)
#####################################################
# TimeTraking pane
panel_TimeTraking = JPanel()
layout6 = GroupLayout(panel_TimeTraking)
layout6.setAutoCreateGaps(True)
layout6.setAutoCreateContainerGaps(True)
panel_TimeTraking.setLayout(layout6)
#####################################################
# Tabbed Area Tabs
tabPane.addTab("Text Edit", panel_TEdit)
tabPane.addTab("Dimensions", panel_Dimensions)
tabPane.addTab("ConfirmIt", panel_ConfirmIt)
tabPane.addTab("Statistics", panel_Statistics)
tabPane.addTab("TimeTraking", panel_TimeTraking)
#############################################################
#############################################################
# Aplication Layouts: 2 groups one Horizontal and one Vertical
#############################################################
# Frame Layout: 2 groups one Horizontal and one Vertical
layout.setHorizontalGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(Larea1)
.addComponent(Sarea1)
.addComponent(Sarea2)
.addGroup(layout.createSequentialGroup()
.addComponent(bCopyToInput)
.addComponent(bClear)
.addComponent(self.cCtClipB))
.addComponent(Larea2))
.addGroup(layout.createParallelGroup()
.addComponent(tabPane))
)
layout.setVerticalGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addGroup(layout.createSequentialGroup()
.addComponent(Larea1)
.addComponent(Sarea1)
.addGroup(layout.createParallelGroup()
.addComponent(bCopyToInput)
.addComponent(bClear)
.addComponent(self.cCtClipB) )
.addComponent(Larea2)
.addComponent(Sarea2))
.addGroup(layout.createSequentialGroup()
.addComponent(tabPane))
)
)
#############################################################
# TEdit Layout: 2 groups one Horizontal and one Vertical
layout2.setHorizontalGroup(layout2.createSequentialGroup()
.addGroup(layout2.createParallelGroup()
.addGroup(layout2.createSequentialGroup()
.addComponent(bRemNumbers)
.addComponent(bRemoveNBSP_L)
.addComponent(bRemoveNBSP_R))
.addGroup(layout2.createSequentialGroup()
.addComponent(bSandReplace)
.addComponent(self.ReplaceThis)
.addComponent(self.ReplaceThat))
))
layout2.setVerticalGroup(layout2.createSequentialGroup()
.addGroup(layout2.createParallelGroup()
.addComponent(bRemNumbers)
.addComponent(bRemoveNBSP_L)
.addComponent(bRemoveNBSP_R))
.addGroup(layout2.createParallelGroup()
.addComponent(bSandReplace)
.addComponent(self.ReplaceThis)
.addComponent(self.ReplaceThat))
)
#############################################################
# Dimensions Layout: 2 groups one Horizontal and one Vertical
layout3.setHorizontalGroup(layout3.createSequentialGroup()
.addGroup(layout3.createParallelGroup()
.addGroup(layout3.createSequentialGroup()
.addComponent(bM_Categories)
.addComponent(self.iStart))
.addGroup(layout3.createSequentialGroup()
.addComponent(optioncCategories)
.addComponent(self.cCurly)
.addComponent(self.cSemiC)
)
.addGroup(layout3.createSequentialGroup()
.addComponent(bcCat)
.addComponent(self.RThis)
.addComponent(self.RThat))
.addGroup(layout3.createSequentialGroup()
)
)
)
layout3.setVerticalGroup(layout3.createSequentialGroup()
.addGroup(layout3.createSequentialGroup()
.addGroup(layout3.createParallelGroup()
.addComponent(bM_Categories)
.addComponent(self.iStart))
.addGroup(layout3.createParallelGroup()
.addComponent(bcCat)
.addComponent(self.RThis)
.addComponent(self.RThat))
.addGroup(layout3.createParallelGroup()
.addGroup(layout3.createParallelGroup()
.addComponent(optioncCategories)
.addComponent(self.cCurly)
.addComponent(self.cSemiC)
)
)
)
)
#############################################################
# ConfimIT Layout: 2 groups one Horizontal and one Vertical
layout4.setHorizontalGroup(layout4.createSequentialGroup()
.addGroup(layout4.createParallelGroup()
.addGroup(layout4.createSequentialGroup()
.addComponent(bClinks)
.addComponent(self.PID)
)
.addGroup(layout4.createSequentialGroup()
.addComponent(bClinksNA)
.addComponent(bClinksCA)
)
.addGroup(layout4.createSequentialGroup()
.addComponent(baddDIVt)
.addComponent(self.Width)
)
))
layout4.setVerticalGroup(layout4.createSequentialGroup()
.addGroup(layout4.createSequentialGroup()
.addGroup(layout4.createParallelGroup()
.addComponent(bClinks)
.addComponent(self.PID))
.addGroup(layout4.createParallelGroup()
.addComponent(bClinksNA)
.addComponent(bClinksCA)
)
.addGroup(layout4.createParallelGroup()
.addComponent(baddDIVt)
.addComponent(self.Width)
)
))
#layout2.linkSize(SwingConstants.HORIZONTAL, [self.cCurly,bM_Categories])
#layout.linkSize(SwingConstants.HORIZONTAL, [ok, bCopyToInput, close, bM_Categories])
#layout3.linkSize(SwingConstants.HORIZONTAL, [self.RThis,self.RThat,bRemoveNBSP_L,bRemoveNBSP_R,bM_Categories,bSandReplace,bcCat])
#############################################################
#############################################################
# Aplication Settings
self.pack()
#self.setPreferredSize(Dimension(1000, 1000))
self.setTitle("Workhelper")
self.setSize(800, 500)
self.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
self.setLocationRelativeTo(None)
self.setVisible(True)
def __init__(self):
TrafficMod.__init__(self)
self.setLayout(FlowLayout(FlowLayout.LEADING))
settings_panel = JPanel()
layout = GroupLayout(settings_panel)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
settings_panel.setLayout(layout)
self.add(settings_panel)
# Horizontal group
horizontal = layout.createParallelGroup()
layout.setHorizontalGroup(
layout.createSequentialGroup().addGroup(horizontal))
# Vertical group
vertical = layout.createSequentialGroup()
layout.setVerticalGroup(vertical)
# Main title, label only
main_title = JLabel(self.title)
main_title.setFont(get_title_font(main_title))
horizontal.addGroup(
layout.createSequentialGroup().addComponent(main_title))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(main_title))
# loss type (random, state, or gemodel)
self.loss_type_combo = self.get_loss_type_combo()
loss_type_lbl = JLabel("Loss type")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_type_lbl).addComponent(self.loss_type_combo))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_type_lbl).addComponent(self.loss_type_combo))
# Random loss fields
# Loss percentage
loss_percent_lbl = JLabel("Loss %")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_percent_lbl).addComponent(self.loss_random_percent))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_percent_lbl).addComponent(self.loss_random_percent))
layout.linkSize(SwingConstants.HORIZONTAL, loss_percent_lbl,
loss_type_lbl)
self.loss_random_items.append(loss_percent_lbl)
self.loss_random_items.append(self.loss_random_percent)
# Loss percent correlation
loss_corr_lbl = JLabel("Correlation %")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_corr_lbl).addComponent(self.loss_random_percent_correlation))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_corr_lbl).addComponent(
self.loss_random_percent_correlation))
layout.linkSize(SwingConstants.HORIZONTAL, loss_corr_lbl,
loss_type_lbl)
self.loss_random_items.append(loss_corr_lbl)
self.loss_random_items.append(self.loss_random_percent_correlation)
# State loss fields
# P13
loss_p13_lbl = JLabel("P13")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_p13_lbl).addComponent(self.loss_state_p13))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_p13_lbl).addComponent(self.loss_state_p13))
self.loss_state_items.append(loss_p13_lbl)
self.loss_state_items.append(self.loss_state_p13)
# P31
loss_p31_lbl = JLabel("P31")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_p31_lbl).addComponent(self.loss_state_p31))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_p31_lbl).addComponent(self.loss_state_p31))
layout.linkSize(SwingConstants.HORIZONTAL, loss_p31_lbl, loss_p13_lbl)
self.loss_state_items.append(loss_p31_lbl)
self.loss_state_items.append(self.loss_state_p31)
# P32
loss_p32_lbl = JLabel("P32")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_p32_lbl).addComponent(self.loss_state_p32))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_p32_lbl).addComponent(self.loss_state_p32))
layout.linkSize(SwingConstants.HORIZONTAL, loss_p32_lbl, loss_p13_lbl)
self.loss_state_items.append(loss_p32_lbl)
self.loss_state_items.append(self.loss_state_p32)
# P23
loss_p23_lbl = JLabel("P23")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_p23_lbl).addComponent(self.loss_state_p23))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_p23_lbl).addComponent(self.loss_state_p23))
layout.linkSize(SwingConstants.HORIZONTAL, loss_p23_lbl, loss_p13_lbl)
self.loss_state_items.append(loss_p23_lbl)
self.loss_state_items.append(self.loss_state_p23)
# P14
loss_p14_lbl = JLabel("P14")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
loss_p14_lbl).addComponent(self.loss_state_p14))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
loss_p14_lbl).addComponent(self.loss_state_p14))
layout.linkSize(SwingConstants.HORIZONTAL, loss_p14_lbl, loss_p13_lbl)
self.loss_state_items.append(loss_p14_lbl)
self.loss_state_items.append(self.loss_state_p14)
# Gemodel loss fields
# P
ge_p_lbl = JLabel("p")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(ge_p_lbl).addComponent(
self.loss_gemodel_p))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
ge_p_lbl).addComponent(self.loss_gemodel_p))
self.loss_gemodel_items.append(ge_p_lbl)
self.loss_gemodel_items.append(self.loss_gemodel_p)
# R
ge_r_lbl = JLabel("r")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(ge_r_lbl).addComponent(
self.loss_gemodel_r))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
ge_r_lbl).addComponent(self.loss_gemodel_r))
layout.linkSize(SwingConstants.HORIZONTAL, ge_r_lbl, ge_p_lbl)
self.loss_gemodel_items.append(ge_r_lbl)
self.loss_gemodel_items.append(self.loss_gemodel_r)
# H
ge_h_lbl = JLabel("1-h")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(ge_h_lbl).addComponent(
self.loss_gemodel_h))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
ge_h_lbl).addComponent(self.loss_gemodel_h))
layout.linkSize(SwingConstants.HORIZONTAL, ge_h_lbl, ge_p_lbl)
self.loss_gemodel_items.append(ge_h_lbl)
self.loss_gemodel_items.append(self.loss_gemodel_h)
# K
ge_k_lbl = JLabel("1-k")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(ge_k_lbl).addComponent(
self.loss_gemodel_k))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
ge_k_lbl).addComponent(self.loss_gemodel_k))
layout.linkSize(SwingConstants.HORIZONTAL, ge_k_lbl, ge_p_lbl)
self.loss_gemodel_items.append(ge_k_lbl)
self.loss_gemodel_items.append(self.loss_gemodel_k)
# Initially show the loss_random textboxes
for item in self.loss_random_items:
item.setVisible(True)
for item in self.loss_state_items:
item.setVisible(False)
for item in self.loss_gemodel_items:
item.setVisible(False)
def __init__(self):
TrafficMod.__init__(self)
self.setLayout(FlowLayout(FlowLayout.LEADING))
settings_panel = JPanel()
layout = GroupLayout(settings_panel)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
settings_panel.setLayout(layout)
self.add(settings_panel)
# Horizontal group
horizontal = layout.createParallelGroup()
layout.setHorizontalGroup(
layout.createSequentialGroup().addGroup(horizontal))
# Vertical group
vertical = layout.createSequentialGroup()
layout.setVerticalGroup(vertical)
# Main title, label only
main_title = JLabel(self.title)
main_title.setFont(get_title_font(main_title))
horizontal.addGroup(
layout.createSequentialGroup().addComponent(main_title))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(main_title))
# Base rate
delay_lbl = JLabel("Delay")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
delay_lbl).addComponent(self.delay_time).addComponent(
self.times_combo))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
delay_lbl).addComponent(self.delay_time).addComponent(
self.times_combo))
# Deviation (jitter)
deviation_lbl = JLabel("Deviation")
horizontal.addGroup(layout.createSequentialGroup().addComponent(
deviation_lbl).addComponent(self.delay_deviation).addComponent(
self.deviation_combo))
vertical.addGroup(
layout.createParallelGroup(GroupLayout.Alignment.BASELINE).
addComponent(deviation_lbl).addComponent(
self.delay_deviation).addComponent(self.deviation_combo))
layout.linkSize(SwingConstants.HORIZONTAL, deviation_lbl, delay_lbl)
# Correlation
corr_lbl = JLabel("Correlation %")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(corr_lbl).addComponent(
self.delay_correlation))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
corr_lbl).addComponent(self.delay_correlation))
layout.linkSize(SwingConstants.HORIZONTAL, corr_lbl, delay_lbl)
# Distribution
dist_lbl = JLabel("Distribution")
horizontal.addGroup(
layout.createSequentialGroup().addComponent(dist_lbl).addComponent(
self.distribution_combo))
vertical.addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.BASELINE).addComponent(
dist_lbl).addComponent(self.distribution_combo))
layout.linkSize(SwingConstants.HORIZONTAL, dist_lbl, delay_lbl)
def __init__(self, pP):
'''
Constructor
'''
self.pP = pP
self.annotationType = self.pP.getAnnotationType()
self.setTitle("Random Picture Picker")
#annotation Panel
annoPanel = JPanel()
annoPanel.setBorder(BorderFactory.createTitledBorder("Annotations"))
annoPLayout = GroupLayout(annoPanel)
annoPanel.setLayout(annoPLayout)
annoPLayout.setAutoCreateContainerGaps(True)
annoPLayout.setAutoCreateGaps(True)
# dynamic creation of annotation panel
# yesNoIgnore, int, number, list
if len(self.pP.getAnnotationType()) == 1:
self.annoField = JTextField("", 16)
annoPLayout.setHorizontalGroup(annoPLayout.createParallelGroup().addComponent(self.annoField))
annoPLayout.setVerticalGroup(annoPLayout.createSequentialGroup().addComponent(self.annoField))
elif len(self.pP.getAnnotationType()) > 1:
choices = pP.getAnnotationType()
print "choices", choices
choiceBtns = []
self.annoField = ButtonGroup()
for c in choices:
Btn = JRadioButton(c, actionCommand=c)
self.annoField.add(Btn)
choiceBtns.append(Btn)
h = annoPLayout.createParallelGroup()
for b in choiceBtns:
h.addComponent(b)
annoPLayout.setHorizontalGroup(h)
v = annoPLayout.createSequentialGroup()
for b in choiceBtns:
v.addComponent(b)
annoPLayout.setVerticalGroup(v)
# Control Panel
ctrlPanel = JPanel()
ctrlPLayout = GroupLayout(ctrlPanel, autoCreateContainerGaps=True, autoCreateGaps=True)
ctrlPanel.setLayout(ctrlPLayout)
nextImgButton = JButton("Next >", actionPerformed=self.nextPicture)
prevImgButton = JButton("< Prev", actionPerformed=self.pP.prevPicture)
quitButton = JButton("Quit", actionPerformed=self.exit)
ctrlPLayout.setHorizontalGroup(ctrlPLayout.createParallelGroup(GroupLayout.Alignment.CENTER)
.addGroup(ctrlPLayout.createSequentialGroup()
.addComponent(prevImgButton)
.addComponent(nextImgButton))
.addComponent(quitButton))
ctrlPLayout.setVerticalGroup(ctrlPLayout.createSequentialGroup()
.addGroup(ctrlPLayout.createParallelGroup()
.addComponent(prevImgButton)
.addComponent(nextImgButton))
.addComponent(quitButton))
ctrlPLayout.linkSize(SwingConstants.HORIZONTAL, quitButton)
statusPanel = JPanel() # contains status information: progress bar
self.progressBar = JProgressBar()
self.progressBar.setStringPainted(True)
self.progressBar.setValue(0)
statusPanel.add(self.progressBar)
#MainLayout
mainLayout = GroupLayout(self.getContentPane())
self.getContentPane().setLayout(mainLayout)
mainLayout.setHorizontalGroup(mainLayout.createParallelGroup(GroupLayout.Alignment.CENTER)
.addComponent(annoPanel)
.addComponent(ctrlPanel)
.addComponent(statusPanel)
)
mainLayout.setVerticalGroup(mainLayout.createSequentialGroup()
.addComponent(annoPanel)
.addComponent(ctrlPanel)
.addComponent(statusPanel)
)
mainLayout.linkSize(SwingConstants.HORIZONTAL, annoPanel, ctrlPanel, statusPanel)
self.pack()
self.setVisible(True)
self.pP.nextPicture()
def initUI(self):
#panel =
layout = GroupLayout(self.getContentPane())
self.getContentPane().setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
#self.setPreferredSize(Dimension(300, 300))
self.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
# definition of elements
# labels
dateLabel = JLabel("Date:")
strainLabel = JLabel("Strain:")
tempLabel = JLabel("Temperature:")
dCelLabel = JLabel("C")
ODLabel = JLabel("OD:")
condLabel = JLabel("Condition:")
#textfields
dateField = JTextField( "today", 1)#str(date.today()), 8 )
strainField = JTextField( "2926",1 )
tempField = JTextField( "34",2 )
ODField = JTextField( "0.5",3 )
condField = JTextField( "0.5",3 )
#buttons
OKButton = JButton("OK",actionPerformed=None)
CancelButton = JButton("Cancel", actionPerformed=cancel)
'''horizontal layout = parallel group (sequential group (c1, c2), sequential group(c3,c4))
vertical layout = parallel group ( sequential group(c1,c3), sequential group(c2,c4))'''
layout.setHorizontalGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup()
.addComponent(dateLabel)
.addComponent(strainLabel)
.addComponent(tempLabel)
.addComponent(ODLabel)
.addComponent(condLabel))
.addGroup(layout.createParallelGroup()
.addComponent(dateField)
.addComponent(strainField)
.addGroup(layout.createSequentialGroup()
.addComponent(tempField)
.addComponent(dCelLabel))
.addComponent(ODField)
.addComponent(condField)
.addGroup(layout.createSequentialGroup()
.addComponent(CancelButton)
.addComponent(OKButton)
.addPreferredGap(LayoutStyle.ComponentPlacement.RELATED)
)
)
)
layout.setVerticalGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(dateLabel)
.addComponent(dateField)
)
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(strainLabel)
.addComponent(strainField)
)
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(tempLabel)
.addComponent(tempField)
.addComponent(dCelLabel)
)
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(ODLabel)
.addComponent(ODField)
)
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(condLabel)
.addComponent(condField)
)
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(CancelButton)
.addComponent(OKButton))
)
self.setTitle("CellSelector")
#self.setSize(200, 150)
self.pack()
self.setLocationRelativeTo(None)
self.setVisible(True)
def initUI(self):
'''Initial UI and Widget creation takes place here!
'''
self.setContentPane = JPanel()
#self.setDefaultLookAndFeelDecorated(True)
# Borders
foreground_colour = Color(30,57,68)
background_colour = Color(247,246,242)
window_background = Color(145,190,210)
self.border = BorderFactory.createLoweredBevelBorder()
self.border2 = BorderFactory.createLineBorder(foreground_colour, 1, True)
# Fonts
self.entry_font= Font("Ubuntu Light", Font.BOLD, 20)
self.label_font= Font("Ubuntu Light", Font.BOLD, 17)
self.btn_font=Font("Ubuntu Light", Font.BOLD, 15)
# Layout start
layout=GroupLayout(self.getContentPane())
self.getContentPane().setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
self.setPreferredSize(Dimension(300, 150))
# Create the labels
user_label= JLabel(" Username : "******" Server : ", JLabel.LEFT, font=self.label_font)
# Colours
user_label.setForeground(foreground_colour)
server_label.setForeground(foreground_colour)
# Create the text entries
self.username=JTextField(actionPerformed=self.continueEvent, border=self.border2, font = self.entry_font)
self.server=JTextField(actionPerformed=self.continueEvent, border=self.border2, font = self.entry_font)
# Colours
self.username.setBackground(background_colour)
self.server.setBackground(background_colour)
self.username.setForeground(foreground_colour)
self.server.setForeground(foreground_colour)
# Allow editable
self.username.setEditable(True)
self.server.setEditable(True)
# Create the buttons
quit_btn=JButton(" Quit! ", actionPerformed=self.closeEvent, border=self.border2, font=self.btn_font)
go_btn=JButton(" Go! ", actionPerformed=self.continueEvent, border=self.border2, font=self.btn_font)
# Colours
quit_btn.setBackground(background_colour)
go_btn.setBackground(background_colour)
quit_btn.setForeground(foreground_colour)
go_btn.setForeground(foreground_colour)
# Setting up the horizontal groups parameters
layout.setHorizontalGroup(layout.createSequentialGroup()
# Left side
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.TRAILING)
.addComponent(user_label)
.addComponent(server_label))
# Right side
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.CENTER)
.addComponent(self.username)
.addComponent(self.server)
.addGroup(layout.createSequentialGroup()
.addComponent(quit_btn)
.addComponent(go_btn)))
)
# Setting up Vertical Groups
layout.setVerticalGroup(layout.createSequentialGroup()
# Top group
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.CENTER)
.addComponent(user_label)
.addComponent(self.username))
# Middle group
.addGroup(layout.createParallelGroup(GroupLayout.Alignment.CENTER)
.addComponent(server_label)
.addComponent(self.server))
# Bottom group
.addGroup(layout.createParallelGroup()
.addComponent(quit_btn)
.addComponent(go_btn))
)
# Finalise the GUI
layout.linkSize(SwingConstants.HORIZONTAL, [quit_btn,go_btn])
self.getContentPane().setBackground(window_background)
self.pack()
self.setTitle('Chat Login')
self.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
self.setLocationRelativeTo(None)
self.setVisible(True)
def initUI(self, okayEvent):
pathPanel = JPanel()
pPLayout = GroupLayout(pathPanel)
pathPanel.setLayout(pPLayout)
pPLayout.setAutoCreateContainerGaps(True)
pPLayout.setAutoCreateGaps(True)
btnPanel = JPanel()
btnLayout = GroupLayout(btnPanel)
btnPanel.setLayout(btnLayout)
btnLayout.setAutoCreateContainerGaps(True)
btnLayout.setAutoCreateGaps(True)
paramPanel = JPanel()
paramLayout = GroupLayout(paramPanel)
paramPanel.setLayout(paramLayout)
paramLayout.setAutoCreateContainerGaps(True)
paramLayout.setAutoCreateGaps(True)
layout = GroupLayout(self.getContentPane())
self.getContentPane().setLayout(layout)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
self.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
# definition of elements
# labels
pathLabel = JLabel("Path:")
dateLabel = JLabel("Date:")
strainLabel = JLabel("Strain:")
tempLabel = JLabel("Temperature:")
dCelLabel = JLabel("C")
ODLabel = JLabel("OD:")
condLabel = JLabel("Condition:")
#textfields
self.pathField = JTextField("",1)
self.dateField = JTextField( str(date.today()), 8 )
self.strainField = JTextField( "2926",1 )
self.tempField = JTextField( "34",2 )
self.ODField = JTextField( "0.5",3 )
self.condField = JTextField( "",3 )
#buttons
OpenPathButton = JButton("Browse...", actionPerformed=self.browse)
OKButton = JButton("OK",actionPerformed=okayEvent)
CancelButton = JButton("Cancel", actionPerformed=self.cancel)
'''ContentPane Layout'''
layout.setHorizontalGroup(layout.createParallelGroup()
.addComponent(pathPanel)
.addComponent(paramPanel)
.addComponent(btnPanel)
)
layout.setVerticalGroup(layout.createSequentialGroup()
.addComponent(pathPanel)
.addComponent(paramPanel)
.addComponent(btnPanel)
)
''' PathChooser Panel Layout '''
pPLayout.setHorizontalGroup(pPLayout.createSequentialGroup()
.addComponent(pathLabel)
.addComponent(self.pathField)
.addComponent(OpenPathButton)
)
pPLayout.setVerticalGroup(pPLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(pathLabel)
.addComponent(self.pathField)
.addComponent(OpenPathButton)
)
'''Param Panel Layout'''
paramLayout.setHorizontalGroup(paramLayout.createSequentialGroup()
.addGroup(paramLayout.createParallelGroup()
.addComponent(dateLabel)
.addComponent(strainLabel)
.addComponent(tempLabel)
.addComponent(ODLabel)
.addComponent(condLabel))
.addGroup(paramLayout.createParallelGroup()
.addComponent(self.dateField)
.addComponent(self.strainField)
.addGroup(paramLayout.createSequentialGroup()
.addComponent(self.tempField)
.addComponent(dCelLabel))
.addComponent(self.ODField)
.addComponent(self.condField)
)
)
paramLayout.setVerticalGroup(paramLayout.createSequentialGroup()
.addGroup(paramLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(dateLabel)
.addComponent(self.dateField)
)
.addGroup(paramLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(strainLabel)
.addComponent(self.strainField)
)
.addGroup(paramLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(tempLabel)
.addComponent(self.tempField)
.addComponent(dCelLabel)
)
.addGroup(paramLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(ODLabel)
.addComponent(self.ODField)
)
.addGroup(paramLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(condLabel)
.addComponent(self.condField)
)
)
'''Buttons Panel Layout'''
btnLayout.setHorizontalGroup(btnLayout.createSequentialGroup()
.addComponent(CancelButton)
.addComponent(OKButton)
)
btnLayout.setVerticalGroup(btnLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(CancelButton)
.addComponent(OKButton)
)
self.setTitle("CellSelector")
#self.setSize(200, 150)
self.pack()
self.setLocationRelativeTo(None)
self.setVisible(True)
self.browse(self)
def registerExtenderCallbacks(self, callbacks):
self._panel = JPanel()
self._panel.setLayout(BorderLayout())
#self._panel.setSize(400,400)
# sourrounding try\except because Burp is not giving enough info
try:
# creating all the UI elements
# create the split pane
self._split_pane_horizontal = JSplitPane(
JSplitPane.HORIZONTAL_SPLIT)
self._split_panel_vertical = JSplitPane(JSplitPane.VERTICAL_SPLIT)
# create panels
self._panel_top = JPanel()
self._panel_top.setLayout(BorderLayout())
self._panel_bottom = JPanel()
self._panel_bottom.setLayout(BorderLayout())
self._panel_right = JPanel()
self._panel_right.setLayout(BorderLayout())
self._panel_request = JPanel()
self._panel_request.setLayout(BorderLayout())
self._panel_response = JPanel()
self._panel_response.setLayout(BorderLayout())
# create the tabbed pane used to show request\response
self._tabbed_pane = JTabbedPane(JTabbedPane.TOP)
# create the tabbed pane used to show aslan++\concretization file
self._tabbed_pane_editor = JTabbedPane(JTabbedPane.TOP)
# create the bottom command for selecting the SQL file and
# generating the model
self._button_generate = JButton(
'Generate!', actionPerformed=self._generate_model)
self._button_save = JButton('Save',
actionPerformed=self._save_model)
self._button_select_sql = JButton(
'Select SQL', actionPerformed=self._select_sql_file)
self._text_field_sql_file = JTextField(20)
self._panel_bottom_commands = JPanel()
layout = GroupLayout(self._panel_bottom_commands)
layout.setAutoCreateGaps(True)
layout.setAutoCreateContainerGaps(True)
seq_layout = layout.createSequentialGroup()
seq_layout.addComponent(self._text_field_sql_file)
seq_layout.addComponent(self._button_select_sql)
seq_layout.addComponent(self._button_generate)
seq_layout.addComponent(self._button_save)
layout.setHorizontalGroup(seq_layout)
# create the message editors that will be used to show request and response
self._message_editor_request = callbacks.createMessageEditor(
None, True)
self._message_editor_response = callbacks.createMessageEditor(
None, True)
# create the table that will be used to show the messages selected for
# the translation
self._columns_names = ('Host', 'Method', 'URL')
dataModel = NonEditableModel(self._table_data, self._columns_names)
self._table = JTable(dataModel)
self._scrollPane = JScrollPane()
self._scrollPane.getViewport().setView((self._table))
popmenu = JPopupMenu()
delete_item = JMenuItem("Delete")
delete_item.addActionListener(self)
popmenu.add(delete_item)
self._table.setComponentPopupMenu(popmenu)
self._table.addMouseListener(self)
# add all the elements
self._panel_request.add(
self._message_editor_request.getComponent())
self._panel_response.add(
self._message_editor_response.getComponent())
self._tabbed_pane.addTab("Request", self._panel_request)
self._tabbed_pane.addTab("Response", self._panel_response)
self._panel_top.add(self._scrollPane, BorderLayout.CENTER)
self._panel_bottom.add(self._tabbed_pane, BorderLayout.CENTER)
scroll = JScrollPane(self._panel_bottom)
self._panel_right.add(self._tabbed_pane_editor,
BorderLayout.CENTER)
self._panel_right.add(self._panel_bottom_commands,
BorderLayout.PAGE_END)
self._split_panel_vertical.setTopComponent(self._panel_top)
self._split_panel_vertical.setBottomComponent(scroll)
self._split_pane_horizontal.setLeftComponent(
self._split_panel_vertical)
self._split_pane_horizontal.setRightComponent(self._panel_right)
self._panel.addComponentListener(self)
self._panel.add(self._split_pane_horizontal)
self._callbacks = callbacks
callbacks.setExtensionName("WAFEx")
callbacks.addSuiteTab(self)
callbacks.registerContextMenuFactory(self)
except Exception as e:
exc_type, exc_obj, exc_tb = sys.exc_info()
fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
print(exc_type, fname, exc_tb.tb_lineno)
def drawPluginUI(self):
# Create the plugin user interface
self.pluginTab = JPanel()
self.uiTitle = JLabel('UPnP BHunter Load, Aim and Fire Console')
self.uiTitle.setFont(Font('Tahoma', Font.BOLD, 14))
self.uiTitle.setForeground(Color(250, 100, 0))
self.uiPanelA = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self.uiPanelA.setMaximumSize(Dimension(2500, 1000))
self.uiPanelA.setDividerSize(2)
self.uiPanelB = JSplitPane(JSplitPane.VERTICAL_SPLIT)
self.uiPanelB.setDividerSize(2)
self.uiPanelA.setBottomComponent(self.uiPanelB)
self.uiPanelA.setBorder(BorderFactory.createLineBorder(Color.gray))
# Create and configure labels and text fields
self.labeltitle_step1 = JLabel("[1st STEP] Discover UPnP Locations")
self.labeltitle_step1.setFont(Font('Tahoma', Font.BOLD, 14))
self.labeltitle_step2 = JLabel("[2nd STEP] Select an UPnP Service")
self.labeltitle_step2.setFont(Font('Tahoma', Font.BOLD, 14))
self.labeltitle_step3 = JLabel("[3rd STEP] Time to Attack it")
self.labeltitle_step3.setFont(Font('Tahoma', Font.BOLD, 14))
self.labelsubtitle_step1 = JLabel(
"Specify the IP version address in scope and start UPnP discovery")
self.labelsubtitle_step2 = JLabel(
"Select which of the found UPnP services will be probed")
self.labelsubtitle_step3 = JLabel(
"Select how to test the extracted UPnP SOAP requests having input arguments"
)
self.label_step1 = JLabel("Target IP")
self.label_step2 = JLabel("Found UPnp Services")
self.label_step3 = JLabel("Send all the extracted SOAP requests ")
self.labelstatus = JLabel(" Status")
self.labelempty_step1 = JLabel(" ")
self.labelempty_step2 = JLabel(" ")
self.labelupnp = JLabel("UPnP list")
self.labelip = JLabel("IP list")
self.labelLANHOST = JLabel(
"Send the interesting LANHostConfigManagement SOAP requests ")
self.labelWANCONNECTION = JLabel(
"Send the interesting WANIP/PPPConnection SOAP requests ")
self.labelSOAPnum = JLabel("0")
self.labelLANHOSTnum = JLabel("0")
self.labelWANCONNECTIONnum = JLabel("0")
self.labelNoneServiceFound = JLabel(" ")
self.labelNoneServiceFound.setFont(Font('Tahoma', Font.BOLD, 12))
self.labelNoneServiceFound.setForeground(Color.red)
# Create combobox for IP version selection
self.ip_versions = ["IPv4", "IPv6"]
self.combo_ipversion = JComboBox(self.ip_versions)
self.combo_ipversion.setSelectedIndex(0)
self.combo_ipversion.setEnabled(True)
# Create and configure progress bar
self.progressbar = JProgressBar(0, 100)
self.progressbar.setString("Ready")
self.progressbar.setStringPainted(True)
# Create and configure buttons
self.startbutton = JButton("Start Discovery",
actionPerformed=self.startHunting)
self.clearbutton = JButton("Clear All", actionPerformed=self.clearAll)
self.confirmbutton = JButton("Confirm Selection",
actionPerformed=self.selectUPnPService)
self.intruderbutton = JButton("to Intruder",
actionPerformed=self.sendUPnPToIntruder)
self.LANrepeaterbutton = JButton(
"to Repeater", actionPerformed=self.sendLANUPnPToRepeater)
self.WANrepeaterbutton = JButton(
"to Repeater", actionPerformed=self.sendWANUPnPToRepeater)
self.confirmbutton.setEnabled(False)
self.intruderbutton.setEnabled(False)
self.LANrepeaterbutton.setEnabled(False)
self.WANrepeaterbutton.setEnabled(False)
# Create the combo box, select item at index 0 (first item in list)
self.upnpservices = [" "]
self.upnpcombo_services = JComboBox(self.upnpservices)
self.upnpcombo_services.setSelectedIndex(0)
self.upnpcombo_services.setEnabled(False)
# Class neeeded to handle the combobox in second step panel
class ComboboxListener(ActionListener):
def __init__(self, upnpcombo_targets, upnpcombo_services,
scope_dict):
self.upnpcombo_targets = upnpcombo_targets
self.upnpcombo_services = upnpcombo_services
self.scope_dict = scope_dict
def actionPerformed(self, event):
# Update the location url combobox depending on the IP combobox
selected_target = self.upnpcombo_targets.getSelectedItem()
if self.scope_dict and selected_target:
self.upnpcombo_services.removeAllItems()
for scope_url in self.scope_dict[selected_target]:
self.upnpcombo_services.addItem(scope_url)
self.upnpcombo_services.setSelectedIndex(0)
# Create the combo box, select item at index 0 (first item in list)
self.upnptargets = [" "]
self.upnpcombo_targets = JComboBox(self.upnptargets)
self.upnpcombo_targets.setSelectedIndex(0)
self.upnpcombo_targets.setEnabled(False)
self.upnpcombo_targets.addActionListener(
ComboboxListener(self.upnpcombo_targets, self.upnpcombo_services,
self.scope_dict))
# Configuring first step panel
self.panel_step1 = JPanel()
self.panel_step1.setPreferredSize(Dimension(2250, 100))
self.panel_step1.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step1.setLayout(BorderLayout(15, 15))
self.titlepanel_step1 = JPanel()
self.titlepanel_step1.setLayout(BorderLayout())
self.titlepanel_step1.add(self.labeltitle_step1, BorderLayout.NORTH)
self.titlepanel_step1.add(self.labelsubtitle_step1)
self.targetpanel_step1 = JPanel()
self.targetpanel_step1.add(self.label_step1)
self.targetpanel_step1.add(self.combo_ipversion)
self.targetpanel_step1.add(self.startbutton)
self.targetpanel_step1.add(self.clearbutton)
self.targetpanel_step1.add(self.labelstatus)
self.targetpanel_step1.add(self.progressbar)
self.emptypanel_step1 = JPanel()
self.emptypanel_step1.setLayout(BorderLayout())
self.emptypanel_step1.add(self.labelempty_step1, BorderLayout.WEST)
# Assembling first step panel components
self.panel_step1.add(self.titlepanel_step1, BorderLayout.NORTH)
self.panel_step1.add(self.targetpanel_step1, BorderLayout.WEST)
self.panel_step1.add(self.emptypanel_step1, BorderLayout.SOUTH)
self.uiPanelA.setTopComponent(self.panel_step1)
# Configure second step panel
self.panel_step2 = JPanel()
self.panel_step2.setPreferredSize(Dimension(2250, 100))
self.panel_step2.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step2.setLayout(BorderLayout(15, 15))
self.titlepanel_step2 = JPanel()
self.titlepanel_step2.setLayout(BorderLayout())
self.titlepanel_step2.add(self.labeltitle_step2, BorderLayout.NORTH)
self.titlepanel_step2.add(self.labelsubtitle_step2)
self.selectpanel_step2 = JPanel()
self.selectpanel_step2.add(self.labelip)
self.selectpanel_step2.add(self.upnpcombo_targets)
self.selectpanel_step2.add(self.labelupnp)
self.selectpanel_step2.add(self.upnpcombo_services)
self.selectpanel_step2.add(self.confirmbutton)
self.emptypanel_step2 = JPanel()
self.emptypanel_step2.setLayout(BorderLayout())
self.emptypanel_step2.add(self.labelempty_step2, BorderLayout.WEST)
self.emptypanel_step2.add(self.labelNoneServiceFound)
# Assembling second step panel components
self.panel_step2.add(self.titlepanel_step2, BorderLayout.NORTH)
self.panel_step2.add(self.selectpanel_step2, BorderLayout.WEST)
self.panel_step2.add(self.emptypanel_step2, BorderLayout.SOUTH)
self.uiPanelB.setTopComponent(self.panel_step2)
# Configuring third step panel
self.panel_step3 = JPanel()
self.panel_step3.setPreferredSize(Dimension(2250, 100))
self.panel_step3.setBorder(EmptyBorder(10, 10, 10, 10))
self.panel_step3.setLayout(BorderLayout(15, 15))
self.titlepanel_step3 = JPanel()
self.titlepanel_step3.setLayout(BorderLayout())
self.titlepanel_step3.add(self.labeltitle_step3, BorderLayout.NORTH)
self.titlepanel_step3.add(self.labelsubtitle_step3)
self.underpanel_step3 = JPanel()
underlayout = GroupLayout(self.underpanel_step3)
self.underpanel_step3.setLayout(underlayout)
underlayout.setAutoCreateGaps(True)
underlayout.setAutoCreateContainerGaps(True)
left2right = underlayout.createSequentialGroup()
firstcolumn = underlayout.createParallelGroup()
firstcolumn.addComponent(self.label_step3, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
firstcolumn.addComponent(self.labelLANHOST, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
firstcolumn.addComponent(self.labelWANCONNECTION,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
secondcolumn = underlayout.createParallelGroup()
secondcolumn.addComponent(self.labelSOAPnum)
secondcolumn.addComponent(self.labelLANHOSTnum)
secondcolumn.addComponent(self.labelWANCONNECTIONnum)
thirdcolumn = underlayout.createParallelGroup()
thirdcolumn.addComponent(self.intruderbutton)
thirdcolumn.addComponent(self.LANrepeaterbutton)
thirdcolumn.addComponent(self.WANrepeaterbutton)
left2right.addGroup(firstcolumn)
left2right.addGroup(secondcolumn)
left2right.addGroup(thirdcolumn)
top2bottom = underlayout.createSequentialGroup()
firstrow = underlayout.createParallelGroup()
firstrow.addComponent(self.label_step3, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
firstrow.addComponent(self.labelSOAPnum)
firstrow.addComponent(self.intruderbutton)
secondrow = underlayout.createParallelGroup()
secondrow.addComponent(self.labelLANHOST, GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
secondrow.addComponent(self.labelLANHOSTnum)
secondrow.addComponent(self.LANrepeaterbutton)
thirdrow = underlayout.createParallelGroup()
thirdrow.addComponent(self.labelWANCONNECTION,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE,
GroupLayout.PREFERRED_SIZE)
thirdrow.addComponent(self.labelWANCONNECTIONnum)
thirdrow.addComponent(self.WANrepeaterbutton)
top2bottom.addGroup(firstrow)
top2bottom.addGroup(secondrow)
top2bottom.addGroup(thirdrow)
underlayout.setHorizontalGroup(left2right)
underlayout.setVerticalGroup(top2bottom)
# Assembling thirdd step panel components
self.panel_step3.add(self.titlepanel_step3, BorderLayout.NORTH)
self.panel_step3.add(self.underpanel_step3, BorderLayout.WEST)
self.uiPanelB.setBottomComponent(self.panel_step3)
# Assembling the group of all panels
layout = GroupLayout(self.pluginTab)
self.pluginTab.setLayout(layout)
layout.setHorizontalGroup(
layout.createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
layout.createSequentialGroup().addGap(10, 10, 10).addGroup(
layout.createParallelGroup(
GroupLayout.Alignment.LEADING).addComponent(
self.uiTitle).addGap(15, 15, 15).addComponent(
self.uiPanelA)).addContainerGap(
26, Short.MAX_VALUE)))
layout.setVerticalGroup(
layout.createParallelGroup(GroupLayout.Alignment.LEADING).addGroup(
layout.createSequentialGroup().addGap(15, 15, 15).addComponent(
self.uiTitle).addGap(15, 15, 15).addComponent(
self.uiPanelA).addGap(20, 20, 20).addGap(20, 20, 20)))
