def createMenuBar(self):
menuBar = JMenuBar()
fileMenu = JMenu('File')
exitItem = fileMenu.add(JMenuItem('Exit', actionPerformed=self.exit))
menuBar.add(fileMenu)
codeMenu = JMenu('Encoding')
data = ['ANSI', 'UTF-8', 'UCS-2 Big Endian', 'UCS-2 Little Endian']
bGroup = ButtonGroup()
for suffix in data:
name = 'Encoding in ' + suffix
rb = JRadioButtonMenuItem(name, selected=(suffix == 'ANSI'))
bGroup.add(rb)
codeMenu.add(rb)
menuBar.add(codeMenu)
viewMenu = JMenu('View')
viewMenu.add(JCheckBoxMenuItem('Full screen'))
viewMenu.add(JSeparator()) # Using JSeparator()
# viewMenu.addSeparator() # Using addSeparator()
# viewMenu.insertSeparator( 1 ) #
viewMenu.add(JCheckBoxMenuItem('Word wrap'))
menuBar.add(viewMenu)
return menuBar
def run(self):
frame = JFrame('Table11',
size=(300, 170),
locationRelativeTo=None,
defaultCloseOperation=JFrame.EXIT_ON_CLOSE)
menuBar = JMenuBar()
resize = JMenu('AUTO_RESIZE')
bGroup = ButtonGroup()
for name, value in self.info:
rb = JRadioButtonMenuItem(name,
actionPerformed=self.handler,
selected=(name == 'Rest'))
bGroup.add(rb)
resize.add(rb)
menuBar.add(resize)
frame.setJMenuBar(menuBar)
headings = 'T/F,Date,Integer,Float,Double'.split(',')
model = myTM(self.data, headings)
self.table = table = JTable(
model, selectionMode=ListSelectionModel.SINGLE_SELECTION)
table.getColumnModel().getColumn(model.getColumnCount() -
1 # i.e., last column
).setCellRenderer(myRenderer())
setColumnWidths(table)
frame.add(JScrollPane(table))
frame.setVisible(1)
class SpeechToTextReport_ConfigPanel(JPanel):
def __init__(self):
self.initComponents()
def initComponents(self):
self.panel = JPanel(GridLayout(0, 1))
self.combo = makeLanguageSelectionComboBox(self.panel, "english")
#mode
self.panel.add(JLabel("Choose mode: "))
self.modeGroup = ButtonGroup()
rb = JRadioButton
self.radioButtonTranscribed = rb(
'Generate report of files with \'Transcribed\' tag.')
radioButtons = (
self.radioButtonTranscribed,
rb('Transcribe files with \'Transcribe\' tag and generate report'))
for a_radiobutton in radioButtons:
self.modeGroup.add(a_radiobutton)
self.panel.add(a_radiobutton)
self.radioButtonTranscribed.selected = 1
#type
self.panel.add(JLabel("Choose type: "))
self.typeGroup = ButtonGroup()
self.radioButtonHTML = rb('HTML')
radioButtons = (self.radioButtonHTML, rb('CSV'))
for a_radiobutton in radioButtons:
self.typeGroup.add(a_radiobutton)
self.panel.add(a_radiobutton)
self.radioButtonHTML.selected = 1
self.add(self.panel)
def createMenuBar(self):
menuBar = JMenuBar()
fileMenu = JMenu('File')
data = [['Spam', self.spam], ['Eggs', self.eggs],
['Bacon', self.bacon]]
bGroup = ButtonGroup()
for name, handler in data:
rb = JRadioButtonMenuItem(name,
actionPerformed=handler,
selected=(name == 'Spam'))
bGroup.add(rb)
fileMenu.add(rb)
fileMenu.add(JSeparator()) # Using JSeparator()
for name, handler in data:
fileMenu.add(JCheckBoxMenuItem(name, actionPerformed=handler))
fileMenu.addSeparator() # Using addSeparator()
exitItem = fileMenu.add(
JMenuItem('Exit',
KeyEvent.VK_X,
actionPerformed=self.exit,
accelerator=KeyStroke.getKeyStroke(
'x', InputEvent.ALT_DOWN_MASK)))
menuBar.add(fileMenu)
helpMenu = JMenu('Help')
aboutItem = helpMenu.add(
JMenuItem('About', KeyEvent.VK_A, actionPerformed=self.about))
menuBar.add(helpMenu)
return menuBar
def createMenuBar(self):
menuBar = JMenuBar()
fileMenu = JMenu('File')
data = [['Spam', self.spam], ['Eggs', self.eggs],
['Bacon', self.bacon]]
bGroup = ButtonGroup()
for name, handler in data:
rb = JRadioButtonMenuItem(name,
actionPerformed=handler,
selected=(name == 'Spam'))
bGroup.add(rb)
fileMenu.add(rb)
exitItem = fileMenu.add(JMenuItem('Exit', actionPerformed=self.exit))
menuBar.add(fileMenu)
helpMenu = JMenu('Help')
aboutItem = helpMenu.add(JMenuItem('About',
actionPerformed=self.about))
menuBar.add(helpMenu)
return menuBar
class TrueFalsePanel(QuestionPanel):
def __init__(self, truefalse):
super(TrueFalsePanel, self).__init__(os.path.join(os.path.dirname(__file__), "truefalse.xml"), truefalse)
self.rdbOption1.setText(truefalse.getOption1())
self.rdbOption2.setText(truefalse.getOption2())
self.btgOptions = ButtonGroup()
self.btgOptions.add(self.rdbOption1)
self.btgOptions.add(self.rdbOption2)
self.question = truefalse
def rdbOption1_click(self, *args):
self.feedback(1)
def rdbOption2_click(self, *args):
self.feedback(2)
def feedback(self, userAnswer):
self.question.setUserAnswer(userAnswer)
if self.question.getUserAnswer() == self.question.getAnswer():
commonsdialog.msgbox("Right answer!", "Feedback ", commonsdialog.IDEA)
self.question.setUserPoints(self.question.getPoints())
else:
commonsdialog.msgbox("Ups! ... wrong answer.", "Feedback ", commonsdialog.FORBIDEN)
self.question.setUserPoints(0)
self.rdbOption1.setEnabled(0)
self.rdbOption2.setEnabled(0)
def getQuestion(self):
return self.question
def setQuestion(self, question):
self.question = question
def display(self, values):
button_group = ButtonGroup()
for operator in self._OPERATORS:
button = JRadioButton(operator)
button.setSelected(operator == values['tags_operator'])
button.addItemListener(self)
button_group.add(button)
self._buttons.append(button)
self.add(button)
def addActionsRadio(self, actions, selected=0):
group = ButtonGroup()
for i, action in enumerate(actions):
item = JRadioButtonMenuItem(action.toSwingAction())
item.selected = (i == selected)
self.applyStyle(item)
group.add(item)
self.menu.add(item)
def __initRadioCaseCmd(self):
# dead
listener = RadioListener(self)
cases = self.__addRadioButton("cases", listener)
cmds = self.__addRadioButton("commands", listener)
group = ButtonGroup()
group.add(cases)
group.add(cmds)
cases.setSelected(1)
def display(self, values):
self.add(JLabel('<html><b>Capturing:</b></html'))
button_group = ButtonGroup()
for option in self._OPTIONS:
button = JRadioButton(option)
button.setSelected(option == values['capturing'])
button.addItemListener(self)
button_group.add(button)
self._buttons.append(button)
self.add(button)
class QuestionBooleanPanel(QuestionBasePanel):
def __init__(self, question):
QuestionBasePanel.__init__(self,os.path.join(os.path.dirname(__file__),"questionboolean.xml"), question)
self.rdbOption1.setText(self.question.option1)
self.rdbOption2.setText(self.question.option2)
self.btgAnswers = ButtonGroup()
self.btgAnswers.add(self.rdbOption1)
self.btgAnswers.add(self.rdbOption2)
def getAnswerCode(self):
if self.rdbOption1.isSelected():
return 1
elif self.rdbOption2.isSelected():
return 2
class GameEditorPanel(FormPanel):
def __init__(self):
FormPanel.__init__(self, os.path.join(os.path.dirname(__file__), "gameeditor.xml"))
self.setPreferredSize(180, 125)
self.btgEditors = ButtonGroup()
self.btgEditors.add(self.rdbInfo)
self.btgEditors.add(self.rdbStages)
self.btgEditors.add(self.rdbStudents)
self.rdbInfo.setSelected(1)
def btnEnter_click(self, *args):
info = 1
stages = 1
students = 1
if self.rdbInfo.isSelected():
info = commonsdialog.confirmDialog("Do you want to edit info?", "Enter", commonsdialog.YES_NO, commonsdialog.QUESTION)
elif self.rdbStages.isSelected():
stages = commonsdialog.confirmDialog("Do you want to edit stages?", "Enter", commonsdialog.YES_NO, commonsdialog.QUESTION)
elif self.rdbStudents.isSelected():
students = commonsdialog.confirmDialog("Do you want to edit students?", "Enter", commonsdialog.YES_NO, commonsdialog.QUESTION)
if info == 0:
infoEditorPanel = InfoEditorPanel()
infoEditorPanel.showWindow("Info")
elif stages == 0:
stagesEditorPanel = StagesEditorPanel()
stagesEditorPanel.showWindow("Stages")
elif students == 0:
studentsEditorPanel = StudentsEditorPanel()
studentsEditorPanel.showWindow("Students")
def btnExit_click(self, *args):
exit = commonsdialog.confirmDialog("Do you want to exit?", "Exit", commonsdialog.YES_NO, commonsdialog.QUESTION)
if exit == 0:
self.hide()
def fix_popup(self):
self.popup.add(JPopupMenu.Separator())
# Add submenus for x and y axes
x_menu = JMenu("X Axis")
y_menu = JMenu("Y Axis")
self.popup.add(x_menu)
self.popup.add(y_menu)
x_btngrp = ButtonGroup()
y_btngrp = ButtonGroup()
# Calculate number of submenu layers needed
max_ind = len(self.data.get_first())
num_sub = max(1, int(ceil(log(max_ind) / log(self.max_show_dim))))
max_sub = [self.max_show_dim ** (num_sub - i) for i in range(num_sub)]
x_subs = [x_menu] * num_sub
y_subs = [y_menu] * num_sub
for i in range(max_ind):
if(i % self.max_show_dim == 0):
for n in range(num_sub - 1):
if(i % max_sub[n + 1] == 0):
new_xmenu = JMenu("%s[%d:%d]" % ('v', i, min(max_ind, i + max_sub[n + 1]) - 1))
new_ymenu = JMenu("%s[%d:%d]" % ('v', i, min(max_ind, i + max_sub[n + 1]) - 1))
x_subs[n].add(new_xmenu)
x_subs[n + 1] = new_xmenu
y_subs[n].add(new_ymenu)
y_subs[n + 1] = new_ymenu
x_radio = JRadioButtonMenuItem('%s[%d]' % ('v', i), i == self.indices[0], actionPerformed=lambda x, index=i, self=self: self.indices.__setitem__(0, index))
y_radio = JRadioButtonMenuItem('%s[%d]' % ('v', i), i == self.indices[1], actionPerformed=lambda x, index=i, self=self: self.indices.__setitem__(1, index))
x_btngrp.add(x_radio)
y_btngrp.add(y_radio)
x_subs[num_sub - 1].add(x_radio)
y_subs[num_sub - 1].add(y_radio)
def createMenuBar(self):
menuBar = JMenuBar()
fileMenu = JMenu('File')
exitItem = fileMenu.add(
JMenuItem('Exit',
KeyEvent.VK_X,
actionPerformed=self.exit,
accelerator=KeyStroke.getKeyStroke(
KeyEvent.VK_X, ActionEvent.ALT_MASK)))
menuBar.add(fileMenu)
codeMenu = JMenu('Encoding')
data = [['ANSI', KeyEvent.VK_A], ['UTF-8', KeyEvent.VK_U],
['UCS-2 Big Endian', KeyEvent.VK_B],
['UCS-2 Little Endian', KeyEvent.VK_L]]
bGroup = ButtonGroup()
for suffix, mnemonic in data:
name = 'Encoding in ' + suffix
rb = JRadioButtonMenuItem(name,
mnemonic=mnemonic,
selected=(suffix == 'ANSI'))
bGroup.add(rb)
codeMenu.add(rb)
menuBar.add(codeMenu)
viewMenu = JMenu('View')
viewMenu.add(JCheckBoxMenuItem('Full screen'))
viewMenu.add(JSeparator()) # Using JSeparator()
# viewMenu.addSeparator() # Using addSeparator()
# viewMenu.insertSeparator( 1 ) #
viewMenu.add(JCheckBoxMenuItem('Word wrap'))
menuBar.add(viewMenu)
return menuBar
def initGui(self):
#~ if DEBUG:
#~ import pdb;
#~ pdb.set_trace()
tabPane = JTabbedPane(JTabbedPane.TOP)
CreditsText = "<html># Burp Custom Deserializer<br/># Copyright (c) 2016, Marco Tinari<br/>#<br/># This program is free software: you can redistribute it and/or modify<br/># it under the terms of the GNU General Public License as published by<br/># the Free Software Foundation, either version 3 of the License, or<br/># (at your option) any later version.<br/>#<br/># This program is distributed in the hope that it will be useful,<br/># but WITHOUT ANY WARRANTY; without even the implied warranty of<br/># MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the<br/># GNU General Public License for more details.<br/>#<br/># You should have received a copy of the GNU General Public License<br/># along with this program. If not, see <http://www.gnu.org/licenses/>.)<br/></html>"
label1 = JLabel("<html>Usage:<br>1 - Select the desired encoding functions<br>2 - Enter the name of the parameter in the input field below and press the Apply button!</html>")
label2 = JLabel(CreditsText)
panel1 = JPanel()
#set layout
panel1.setLayout(GridLayout(11,1))
panel2 = JPanel()
panel1.add(label1)
panel2.add(label2)
tabPane.addTab("Configuration", panel1)
tabPane.addTab("Credits", panel2)
applyButton = JButton('Apply',actionPerformed=self.reloadConf)
panel1.add(applyButton, BorderLayout.SOUTH)
#define GET/POST/COOKIE radio button
self.GETparameterTypeRadioButton = JRadioButton('GET parameter')
self.POSTparameterTypeRadioButton = JRadioButton('POST parameter')
self.COOKIEparameterTypeRadioButton = JRadioButton('COOKIE parameter')
self.POSTparameterTypeRadioButton.setSelected(True)
group = ButtonGroup()
group.add(self.GETparameterTypeRadioButton)
group.add(self.POSTparameterTypeRadioButton)
group.add(self.COOKIEparameterTypeRadioButton)
self.base64Enabled = JCheckBox("Base64 encode")
self.URLEnabled = JCheckBox("URL encode")
self.ASCII2HexEnabled = JCheckBox("ASCII to Hex")
self.ScannerEnabled = JCheckBox("<html>Enable serialization in Burp Scanner<br>Usage:<br>1.Place unencoded values inside intruder request and define the placeholder positions<br>2.rightclick->Actively scan defined insertion points)</html>")
self.IntruderEnabled = JCheckBox("<html>Enable serialization in Burp Intruder<br>Usage:<br>1.Place unencoded values inside intruder request and define the placeholder positions<br>2.Start the attack</html>")
self.parameterName = JTextField("Parameter name goes here...",60)
#set the tooltips
self.parameterName.setToolTipText("Fill in the parameter name and apply")
self.base64Enabled.setToolTipText("Enable base64 encoding/decoding")
self.ASCII2HexEnabled.setToolTipText("Enable ASCII 2 Hex encoding/decoding")
self.URLEnabled.setToolTipText("Enable URL encoding/decoding")
self.IntruderEnabled.setToolTipText("Check this if You want the extension to intercept and modify every request made by the Burp Intruder containing the selected paramter")
self.ScannerEnabled.setToolTipText("Check this if You want the extension to intercept and modify every request made by the Burp Scanner containing the selected paramter")
#add checkboxes to the panel
panel1.add(self.parameterName)
panel1.add(self.POSTparameterTypeRadioButton)
panel1.add(self.GETparameterTypeRadioButton)
panel1.add(self.COOKIEparameterTypeRadioButton)
panel1.add(self.base64Enabled)
panel1.add(self.URLEnabled)
panel1.add(self.ASCII2HexEnabled)
panel1.add(self.IntruderEnabled)
panel1.add(self.ScannerEnabled)
#assign tabPane
self.tab = tabPane
def __init__(self, parent, title, modal, app):
from java.awt import CardLayout
self.app = app
border = BorderFactory.createEmptyBorder(5, 7, 7, 7)
self.getContentPane().setBorder(border)
self.setLayout(BoxLayout(self.getContentPane(), BoxLayout.Y_AXIS))
self.FAVAREALAYERNAME = "Favourite zone editing"
info = JLabel(self.app.strings.getString("Create_a_new_favourite_zone"))
info.setAlignmentX(Component.LEFT_ALIGNMENT)
#Name
nameLbl = JLabel(self.app.strings.getString("fav_zone_name"))
self.nameTextField = JTextField(20)
self.nameTextField.setMaximumSize(self.nameTextField.getPreferredSize())
self.nameTextField.setToolTipText(self.app.strings.getString("fav_zone_name_tooltip"))
namePanel = JPanel()
namePanel.setLayout(BoxLayout(namePanel, BoxLayout.X_AXIS))
namePanel.add(nameLbl)
namePanel.add(Box.createHorizontalGlue())
namePanel.add(self.nameTextField)
#Country
countryLbl = JLabel(self.app.strings.getString("fav_zone_country"))
self.countryTextField = JTextField(20)
self.countryTextField.setMaximumSize(self.countryTextField.getPreferredSize())
self.countryTextField.setToolTipText(self.app.strings.getString("fav_zone_country_tooltip"))
countryPanel = JPanel()
countryPanel.setLayout(BoxLayout(countryPanel, BoxLayout.X_AXIS))
countryPanel.add(countryLbl)
countryPanel.add(Box.createHorizontalGlue())
countryPanel.add(self.countryTextField)
#Type
modeLbl = JLabel(self.app.strings.getString("fav_zone_type"))
RECTPANEL = "rectangle"
POLYGONPANEL = "polygon"
BOUNDARYPANEL = "boundary"
self.modesStrings = [RECTPANEL, POLYGONPANEL, BOUNDARYPANEL]
modesComboModel = DefaultComboBoxModel()
for i in (self.app.strings.getString("rectangle"),
self.app.strings.getString("delimited_by_a_closed_way"),
self.app.strings.getString("delimited_by_an_administrative_boundary")):
modesComboModel.addElement(i)
self.modesComboBox = JComboBox(modesComboModel,
actionListener=self,
editable=False)
#- Rectangle
self.rectPanel = JPanel()
self.rectPanel.setLayout(BoxLayout(self.rectPanel, BoxLayout.Y_AXIS))
capturePane = JPanel()
capturePane.setLayout(BoxLayout(capturePane, BoxLayout.X_AXIS))
capturePane.setAlignmentX(Component.LEFT_ALIGNMENT)
josmP = JPanel()
self.captureRBtn = JRadioButton(self.app.strings.getString("capture_area"))
self.captureRBtn.addActionListener(self)
self.captureRBtn.setSelected(True)
self.bboxFromJosmBtn = JButton(self.app.strings.getString("get_current_area"),
actionPerformed=self.on_bboxFromJosmBtn_clicked)
self.bboxFromJosmBtn.setToolTipText(self.app.strings.getString("get_capture_area_tooltip"))
josmP.add(self.bboxFromJosmBtn)
capturePane.add(self.captureRBtn)
capturePane.add(Box.createHorizontalGlue())
capturePane.add(self.bboxFromJosmBtn)
manualPane = JPanel()
manualPane.setLayout(BoxLayout(manualPane, BoxLayout.X_AXIS))
manualPane.setAlignmentX(Component.LEFT_ALIGNMENT)
self.manualRBtn = JRadioButton(self.app.strings.getString("use_this_bbox"))
self.manualRBtn.addActionListener(self)
self.bboxTextField = JTextField(20)
self.bboxTextField.setMaximumSize(self.bboxTextField.getPreferredSize())
self.bboxTextField.setToolTipText(self.app.strings.getString("fav_bbox_tooltip"))
self.bboxTextFieldDefaultBorder = self.bboxTextField.getBorder()
self.bboxTextField.getDocument().addDocumentListener(TextListener(self))
manualPane.add(self.manualRBtn)
manualPane.add(Box.createHorizontalGlue())
manualPane.add(self.bboxTextField)
group = ButtonGroup()
group.add(self.captureRBtn)
group.add(self.manualRBtn)
previewPane = JPanel()
previewPane.setLayout(BoxLayout(previewPane, BoxLayout.X_AXIS))
previewPane.setAlignmentX(Component.LEFT_ALIGNMENT)
bboxPreviewInfo = JTextField(self.app.strings.getString("coordinates"),
editable=0,
border=None)
bboxPreviewInfo.setMaximumSize(bboxPreviewInfo.getPreferredSize())
self.bboxPreviewTextField = JTextField(20,
editable=0,
border=None)
self.bboxPreviewTextField.setMaximumSize(self.bboxPreviewTextField.getPreferredSize())
previewPane.add(bboxPreviewInfo)
previewPane.add(Box.createHorizontalGlue())
previewPane.add(self.bboxPreviewTextField)
self.rectPanel.add(capturePane)
self.rectPanel.add(Box.createRigidArea(Dimension(0, 10)))
self.rectPanel.add(manualPane)
self.rectPanel.add(Box.createRigidArea(Dimension(0, 20)))
self.rectPanel.add(previewPane)
#- Polygon (closed way) drawn by hand
self.polygonPanel = JPanel(BorderLayout())
self.polygonPanel.setLayout(BoxLayout(self.polygonPanel, BoxLayout.Y_AXIS))
polyInfo = JLabel("<html>%s</html>" % self.app.strings.getString("polygon_info"))
polyInfo.setFont(polyInfo.getFont().deriveFont(Font.ITALIC))
polyInfo.setAlignmentX(Component.LEFT_ALIGNMENT)
editPolyPane = JPanel()
editPolyPane.setAlignmentX(Component.LEFT_ALIGNMENT)
editPolyBtn = JButton(self.app.strings.getString("create_fav_layer"),
actionPerformed=self.create_new_zone_editing_layer)
editPolyBtn.setToolTipText(self.app.strings.getString("create_fav_layer_tooltip"))
editPolyPane.add(editPolyBtn)
self.polygonPanel.add(polyInfo)
self.polygonPanel.add(Box.createRigidArea(Dimension(0, 15)))
self.polygonPanel.add(editPolyPane)
self.polygonPanel.add(Box.createRigidArea(Dimension(0, 15)))
#- Administrative Boundary
self.boundaryPanel = JPanel()
self.boundaryPanel.setLayout(BoxLayout(self.boundaryPanel, BoxLayout.Y_AXIS))
boundaryInfo = JLabel("<html>%s</html>" % app.strings.getString("boundary_info"))
boundaryInfo.setFont(boundaryInfo.getFont().deriveFont(Font.ITALIC))
boundaryInfo.setAlignmentX(Component.LEFT_ALIGNMENT)
boundaryTagsPanel = JPanel(GridLayout(3, 3, 5, 5))
boundaryTagsPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
boundaryTagsPanel.add(JLabel("name ="))
self.nameTagTextField = JTextField(20)
boundaryTagsPanel.add(self.nameTagTextField)
boundaryTagsPanel.add(JLabel("admin_level ="))
self.adminLevelTagTextField = JTextField(20)
self.adminLevelTagTextField.setToolTipText(self.app.strings.getString("adminLevel_tooltip"))
boundaryTagsPanel.add(self.adminLevelTagTextField)
boundaryTagsPanel.add(JLabel(self.app.strings.getString("other_tag")))
self.optionalTagTextField = JTextField(20)
self.optionalTagTextField.setToolTipText("key=value")
boundaryTagsPanel.add(self.optionalTagTextField)
downloadBoundariesPane = JPanel()
downloadBoundariesPane.setAlignmentX(Component.LEFT_ALIGNMENT)
downloadBoundariesBtn = JButton(self.app.strings.getString("download_boundary"),
actionPerformed=self.on_downloadBoundariesBtn_clicked)
downloadBoundariesBtn.setToolTipText(self.app.strings.getString("download_boundary_tooltip"))
downloadBoundariesPane.add(downloadBoundariesBtn)
self.boundaryPanel.add(boundaryInfo)
self.boundaryPanel.add(Box.createRigidArea(Dimension(0, 15)))
self.boundaryPanel.add(boundaryTagsPanel)
self.boundaryPanel.add(Box.createRigidArea(Dimension(0, 10)))
self.boundaryPanel.add(downloadBoundariesPane)
self.editingPanels = {"rectangle": self.rectPanel,
"polygon": self.polygonPanel,
"boundary": self.boundaryPanel}
#Main buttons
self.okBtn = JButton(self.app.strings.getString("OK"),
ImageProvider.get("ok"),
actionPerformed=self.on_okBtn_clicked)
self.cancelBtn = JButton(self.app.strings.getString("cancel"),
ImageProvider.get("cancel"),
actionPerformed=self.close_dialog)
self.previewBtn = JButton(self.app.strings.getString("Preview_zone"),
actionPerformed=self.on_previewBtn_clicked)
self.previewBtn.setToolTipText(self.app.strings.getString("preview_zone_tooltip"))
okBtnSize = self.okBtn.getPreferredSize()
viewBtnSize = self.previewBtn.getPreferredSize()
viewBtnSize.height = okBtnSize.height
self.previewBtn.setPreferredSize(viewBtnSize)
#layout
self.add(info)
self.add(Box.createRigidArea(Dimension(0, 15)))
namePanel.setAlignmentX(Component.LEFT_ALIGNMENT)
self.add(namePanel)
self.add(Box.createRigidArea(Dimension(0, 15)))
countryPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
self.add(countryPanel)
self.add(Box.createRigidArea(Dimension(0, 15)))
modeLbl.setAlignmentX(Component.LEFT_ALIGNMENT)
self.add(modeLbl)
self.add(Box.createRigidArea(Dimension(0, 5)))
self.add(self.modesComboBox)
self.modesComboBox.setAlignmentX(Component.LEFT_ALIGNMENT)
self.add(Box.createRigidArea(Dimension(0, 15)))
self.configPanel = JPanel(CardLayout())
self.configPanel.setBorder(BorderFactory.createEmptyBorder(5, 5, 5, 5))
self.configPanel.add(self.rectPanel, RECTPANEL)
self.configPanel.add(self.polygonPanel, POLYGONPANEL)
self.configPanel.add(self.boundaryPanel, BOUNDARYPANEL)
self.configPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
self.add(self.configPanel)
buttonsPanel = JPanel()
buttonsPanel.add(self.okBtn)
buttonsPanel.add(self.cancelBtn)
buttonsPanel.add(self.previewBtn)
buttonsPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
self.add(buttonsPanel)
self.addWindowListener(self)
self.pack()
class CuckooSettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.tag_list = []
self.initComponents()
self.customizeComponents()
self.path_to_cuckoo_exe = os.path.join(os.path.dirname(os.path.abspath(__file__)), "cuckoo_api.exe")
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Submit_File_CB.isSelected():
self.local_settings.setSubmit_File(True)
self.local_settings.setSubmit_URL(False)
else:
self.local_settings.setSubmit_File(False)
# if self.Submit_URL_CB.isSelected():
# self.local_settings.setSubmit_URL(True)
# self.local_settings.setSubmit_File(False)
# else:
# self.local_settings.setSubmit_URL(False)
def onchange_lb(self, event):
self.local_settings.cleartag_list()
list_selected = self.List_Box_LB.getSelectedValuesList()
self.local_settings.settag_list(list_selected)
def find_tags(self):
sql_statement = "SELECT distinct(display_name) u_tag_name FROM content_tags INNER JOIN tag_names ON " + \
" content_tags.tag_name_id = tag_names.tag_name_id;"
skCase = Case.getCurrentCase().getSleuthkitCase()
dbquery = skCase.executeQuery(sql_statement)
resultSet = dbquery.getResultSet()
while resultSet.next():
self.tag_list.append(resultSet.getString("u_tag_name"))
dbquery.close()
# Check to see if there are any entries that need to be populated from the database.
def check_Database_entries(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\gui_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Select Protocol, cuckoo_host, cuckoo_port from cuckoo_server'
resultSet = stmt.executeQuery(SQL_Statement)
while resultSet.next():
self.Protocol_TF.setText(resultSet.getString("Protocol"))
self.IP_Address_TF.setText(resultSet.getString("cuckoo_host"))
self.Port_Number_TF.setText(resultSet.getString("cuckoo_port"))
self.local_settings.setProtocol(resultSet.getString("Protocol"))
self.local_settings.setIP_Address(resultSet.getString("cuckoo_host"))
self.local_settings.setPort_Number(resultSet.getString("cuckoo_port"))
self.Error_Message.setText("Settings Read successfully!")
except SQLException as e:
self.Error_Message.setText("Error Reading Settings Database")
stmt.close()
dbConn.close()
# Save entries from the GUI to the database.
def SaveSettings(self, e):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings")
try:
stmt = dbConn.createStatement()
SQL_Statement = ""
SQL_Statement = 'Update cuckoo_server set Protocol = "' + self.Protocol_TF.getText() + '", ' + \
' Cuckoo_Host = "' + self.IP_Address_TF.getText() + '", ' + \
' Cuckoo_port = "' + self.Port_Number_TF.getText() + '";'
#self.Error_Message.setText(SQL_Statement)
stmt.execute(SQL_Statement)
self.Error_Message.setText("Cuckoo settings Saved")
#self.local_settings.setCuckoo_Directory(self.Program_Executable_TF.getText())
except SQLException as e:
self.Error_Message.setText(e.getMessage())
stmt.close()
dbConn.close()
# Check to see if the Cuckoo server is available and you can talk to it
def Check_Server(self, e):
pipe = Popen([self.path_to_cuckoo_exe, self.Protocol_TF.getText(),self.IP_Address_TF.getText(), self.Port_Number_TF.getText(), "cuckoo_status" ], stdout=PIPE, stderr=PIPE)
out_text = pipe.communicate()[0]
self.Error_Message.setText("Cuckoo Status is " + out_text)
#self.log(Level.INFO, "Cuckoo Status is ==> " + out_text)
# def onchange_cb(self, event):
# self.local_settings.setComboBox(event.item)
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.Label_1 = JLabel("Protocol:")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
self.panel0.add( self.Label_1 )
self.Protocol_TF = JTextField(20)
self.Protocol_TF.setEnabled(True)
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Protocol_TF, self.gbcPanel0 )
self.panel0.add( self.Protocol_TF )
self.Blank_1 = JLabel( " ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Label_2 = JLabel("Cuckoo IP Address")
self.Label_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_2, self.gbcPanel0 )
self.panel0.add( self.Label_2 )
self.IP_Address_TF = JTextField(20)
self.IP_Address_TF.setEnabled(True)
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.IP_Address_TF, self.gbcPanel0 )
self.panel0.add( self.IP_Address_TF )
self.Blank_2 = JLabel( " ")
self.Blank_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_2, self.gbcPanel0 )
self.panel0.add( self.Blank_2 )
self.Label_3 = JLabel("Port Number")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_3, self.gbcPanel0 )
self.panel0.add( self.Label_3 )
self.Port_Number_TF = JTextField(20)
self.Port_Number_TF.setEnabled(True)
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Port_Number_TF, self.gbcPanel0 )
self.panel0.add( self.Port_Number_TF )
self.Blank_3 = JLabel( " ")
self.Blank_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_3, self.gbcPanel0 )
self.panel0.add( self.Blank_3 )
self.Save_Settings_BTN = JButton( "Save Setup", actionPerformed=self.SaveSettings)
self.Save_Settings_BTN.setEnabled(True)
self.rbgPanel0.add( self.Save_Settings_BTN )
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Save_Settings_BTN, self.gbcPanel0 )
self.panel0.add( self.Save_Settings_BTN )
self.Blank_4 = JLabel( " ")
self.Blank_4.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_4, self.gbcPanel0 )
self.panel0.add( self.Blank_4 )
self.Blank_5 = JLabel( "Tag to Choose: ")
self.Blank_5.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 17
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_5, self.gbcPanel0 )
self.panel0.add( self.Blank_5 )
self.find_tags()
self.List_Box_LB = JList( self.tag_list, valueChanged=self.onchange_lb)
self.List_Box_LB.setVisibleRowCount( 3 )
self.scpList_Box_LB = JScrollPane( self.List_Box_LB )
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 17
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 1
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.scpList_Box_LB, self.gbcPanel0 )
self.panel0.add( self.scpList_Box_LB )
self.Blank_6 = JLabel( " ")
self.Blank_6.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_6, self.gbcPanel0 )
self.panel0.add( self.Blank_6 )
self.Submit_File_CB = JCheckBox("Submit a File", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 21
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Submit_File_CB, self.gbcPanel0 )
self.panel0.add( self.Submit_File_CB )
# self.Submit_URL_CB = JCheckBox("Submit a URL", actionPerformed=self.checkBoxEvent)
# self.gbcPanel0.gridx = 2
# self.gbcPanel0.gridy = 23
# self.gbcPanel0.gridwidth = 1
# self.gbcPanel0.gridheight = 1
# self.gbcPanel0.fill = GridBagConstraints.BOTH
# self.gbcPanel0.weightx = 1
# self.gbcPanel0.weighty = 0
# self.gbcPanel0.anchor = GridBagConstraints.NORTH
# self.gbPanel0.setConstraints( self.Submit_URL_CB, self.gbcPanel0 )
# self.panel0.add( self.Submit_URL_CB )
self.Check_Server_Status_BTN = JButton( "Check Server Status", actionPerformed=self.Check_Server)
self.Check_Server_Status_BTN.setEnabled(True)
self.rbgPanel0.add( self.Save_Settings_BTN )
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 25
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Check_Server_Status_BTN, self.gbcPanel0 )
self.panel0.add( self.Check_Server_Status_BTN )
self.Error_Message = JLabel( "")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 27
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
self.panel0.add( self.Error_Message )
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
#self.Exclude_File_Sources_CB.setSelected(self.local_settings.getExclude_File_Sources())
#self.Run_Cuckoo_CB.setSelected(self.local_settings.getRun_Cuckoo())
#self.Import_Cuckoo_CB.setSelected(self.local_settings.getImport_Cuckoo())
self.check_Database_entries()
# Return the settings used
def getSettings(self):
return self.local_settings
class GUI_Test_SQLSettingsWithUISettingsPanel(
IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Exec_Program_CB.isSelected():
self.local_settings.setSetting('Exec_Prog_Flag', 'true')
self.Program_Executable_TF.setEnabled(True)
self.Find_Program_Exec_BTN.setEnabled(True)
else:
self.local_settings.setSetting('Exec_Prog_Flag', 'false')
self.Program_Executable_TF.setText("")
self.Program_Executable_TF.setEnabled(False)
self.Find_Program_Exec_BTN.setEnabled(False)
# Check to see if there are any entries that need to be populated from the database.
def check_Database_entries(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = os.path.join(head, "GUI_Settings.db3")
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Select Setting_Name, Setting_Value from settings;'
resultSet = stmt.executeQuery(SQL_Statement)
while resultSet.next():
if resultSet.getString("Setting_Name") == "Program_Exec_Name":
self.Program_Executable_TF.setText(
resultSet.getString("Setting_Value"))
self.local_settings.setSetting(
'ExecFile', resultSet.getString("Setting_Value"))
self.local_settings.setSetting('Exec_Prog_Flag', 'true')
self.Exec_Program_CB.setSelected(True)
self.Program_Executable_TF.setEnabled(True)
self.Find_Program_Exec_BTN.setEnabled(True)
self.Error_Message.setText("Settings Read successfully!")
except SQLException as e:
self.Error_Message.setText("Error Reading Settings Database")
stmt.close()
dbConn.close()
# Save entries from the GUI to the database.
def SaveSettings(self, e):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = os.path.join(head, "GUI_Settings.db3")
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Insert into settings (Setting_Name, Setting_Value) values ("Program_Exec_Name", "' + \
self.Program_Executable_TF.getText() + '");'
resultSet = stmt.executeQuery(SQL_Statement)
self.Error_Message.setText("Settings Saved")
except SQLException as e:
self.Error_Message.setText("Error Inserting Settings " + str(e))
stmt.close()
dbConn.close()
# When button to find file is clicked then open dialog to find the file and return it.
def onClick(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("SQLite", ["sqlite"])
chooseFile.addChoosableFileFilter(filter)
ret = chooseFile.showDialog(self.panel0, "Select SQLite")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setSetting('ExecFile', Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout(self.gbPanel0)
self.Exec_Program_CB = JCheckBox("Execute Program",
actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Exec_Program_CB, self.gbcPanel0)
self.panel0.add(self.Exec_Program_CB)
self.Program_Executable_TF = JTextField(20)
self.Program_Executable_TF.setEnabled(False)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Program_Executable_TF,
self.gbcPanel0)
self.panel0.add(self.Program_Executable_TF)
self.Find_Program_Exec_BTN = JButton("Find Exec",
actionPerformed=self.onClick)
self.Find_Program_Exec_BTN.setEnabled(False)
self.rbgPanel0.add(self.Find_Program_Exec_BTN)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Find_Program_Exec_BTN,
self.gbcPanel0)
self.panel0.add(self.Find_Program_Exec_BTN)
self.Blank_1 = JLabel(" ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_1, self.gbcPanel0)
self.panel0.add(self.Blank_1)
self.Save_Settings_BTN = JButton("Save Settings",
actionPerformed=self.SaveSettings)
self.Save_Settings_BTN.setEnabled(True)
self.rbgPanel0.add(self.Save_Settings_BTN)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Save_Settings_BTN, self.gbcPanel0)
self.panel0.add(self.Save_Settings_BTN)
self.Blank_2 = JLabel(" ")
self.Blank_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_2, self.gbcPanel0)
self.panel0.add(self.Blank_2)
self.Label_1 = JLabel("Error Message:")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_1, self.gbcPanel0)
self.panel0.add(self.Label_1)
self.Error_Message = JLabel("")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Error_Message, self.gbcPanel0)
self.panel0.add(self.Error_Message)
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
self.Exec_Program_CB.setSelected(
self.local_settings.getSetting('Exec_Prog_Flag') == 'true')
self.check_Database_entries()
# Return the settings used
def getSettings(self):
return self.local_settings
class VolatilitySettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel
):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Exclude_File_Sources_CB.isSelected():
self.local_settings.setExclude_File_Sources(True)
else:
self.local_settings.setExclude_File_Sources(False)
# Check to see if there are any entries that need to be populated from the database.
def check_Database_entries(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Select Setting_Name, Setting_Value from settings;'
resultSet = stmt.executeQuery(SQL_Statement)
while resultSet.next():
if resultSet.getString(
"Setting_Name") == "Volatility_Executable_Directory":
self.Program_Executable_TF.setText(
resultSet.getString("Setting_Value"))
self.local_settings.setVolatility_Directory(
resultSet.getString("Setting_Value"))
self.local_settings.setVolatility_Dir_Found(True)
if resultSet.getString("Setting_Name") == "Volatility_Version":
self.Version_CB.setSelectedItem(
resultSet.getString("Setting_Value"))
self.Error_Message.setText("Settings Read successfully!")
except SQLException as e:
self.Error_Message.setText("Error Reading Settings Database")
stmt.close()
dbConn.close()
# Save entries from the GUI to the database.
def SaveSettings(self, e):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings")
try:
stmt = dbConn.createStatement()
SQL_Statement = ""
if (self.local_settings.getVolatility_Dir_Found()):
SQL_Statement = 'Update settings set Setting_Value = "' + self.Program_Executable_TF.getText() + '"' + \
' where setting_name = "Volatility_Executable_Directory";'
SQL_Statement2 = 'Update settings set Setting_Value = "' + self.Version_CB.getSelectedItem() + '"' + \
' where setting_name = "Volatility_Version";'
else:
SQL_Statement = 'Insert into settings (Setting_Name, Setting_Value) values ("Volatility_Executable_Directory", "' + \
self.Program_Executable_TF.getText() + '");'
SQL_Statement2 = 'Insert into settings (Setting_Name, Setting_Value) values ("Volatility_Version", "' + \
self.Version_CB.getSelectedItem() + '");'
stmt.execute(SQL_Statement)
stmt.execute(SQL_Statement2)
self.Error_Message.setText("Volatility Executable Directory Saved")
self.local_settings.setVolatility_Directory(
self.Program_Executable_TF.getText())
except SQLException as e:
self.Error_Message.setText(e.getMessage())
stmt.close()
dbConn.close()
def get_plugins(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
self.Error_Message.setText("Database opened")
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = "select plugin_name from plugins where volatility_version = '" + self.Version_CB.getSelectedItem() + "' and " + \
" plugin_name in ('dumpcerts', 'dumpfiles', 'dumpregistry', 'linux_librarydump', 'linux_procdump', 'mac_dump_file', 'mac_procdump', 'moddump', 'procdump', 'vaddump');"
resultSet = stmt.executeQuery(SQL_Statement)
plugin_list = []
while resultSet.next():
plugin_list.append(resultSet.getString("plugin_name"))
stmt.close()
dbConn.close()
return plugin_list
except SQLException as e:
self.Error_Message.setText("Error Reading plugins")
stmt.close()
dbConn.close()
return "Error"
def get_profiles(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
self.Error_Message.setText("Database opened")
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = "select profile_name from profiles where volatility_version = '" + self.Version_CB.getSelectedItem(
) + "';"
resultSet = stmt.executeQuery(SQL_Statement)
profile_list = []
while resultSet.next():
profile_list.append(resultSet.getString("profile_name"))
stmt.close()
dbConn.close()
return profile_list
except SQLException as e:
self.Error_Message.setText("Error Reading plugins")
stmt.close()
dbConn.close()
return "Error"
# When button to find file is clicked then open dialog to find the file and return it.
def Find_Dir(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("All", ["*.*"])
chooseFile.addChoosableFileFilter(filter)
#chooseFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY)
ret = chooseFile.showDialog(self.panel0, "Find Volatility Directory")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setVolatility_Directory(Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
def keyPressed(self, event):
self.local_settings.setProcessIDs(
self.Process_Ids_To_Dump_TF.getText())
#self.Error_Message.setText(self.Process_Ids_To_Dump_TF.getText())
def onchange_version(self, event):
self.local_settings.setVersion(event.item)
plugin_list = self.get_plugins()
profile_list = self.get_profiles()
self.Profile_CB.removeAllItems()
self.Plugin_LB.clearSelection()
self.Plugin_LB.setListData(plugin_list)
for profile in profile_list:
self.Profile_CB.addItem(profile)
#self.Profile_CB.addItems(profile)
self.panel0.repaint()
def onchange_plugins_lb(self, event):
self.local_settings.clearPluginListBox()
list_selected = self.Plugin_LB.getSelectedValuesList()
self.local_settings.setPluginListBox(list_selected)
def onchange_profile_cb(self, event):
self.local_settings.setProfile(event.item)
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout(self.gbPanel0)
self.Error_Message = JLabel("")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 31
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Error_Message, self.gbcPanel0)
self.panel0.add(self.Error_Message)
self.Label_1 = JLabel("Volatility Executable Directory")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_1, self.gbcPanel0)
self.panel0.add(self.Label_1)
self.Program_Executable_TF = JTextField(10)
self.Program_Executable_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Program_Executable_TF,
self.gbcPanel0)
self.panel0.add(self.Program_Executable_TF)
self.Find_Program_Exec_BTN = JButton("Find Dir",
actionPerformed=self.Find_Dir)
self.Find_Program_Exec_BTN.setEnabled(True)
self.rbgPanel0.add(self.Find_Program_Exec_BTN)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Find_Program_Exec_BTN,
self.gbcPanel0)
self.panel0.add(self.Find_Program_Exec_BTN)
self.Blank_1 = JLabel(" ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_1, self.gbcPanel0)
self.panel0.add(self.Blank_1)
self.Save_Settings_BTN = JButton("Save Volatility Exec Dir",
actionPerformed=self.SaveSettings)
self.Save_Settings_BTN.setEnabled(True)
self.rbgPanel0.add(self.Save_Settings_BTN)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Save_Settings_BTN, self.gbcPanel0)
self.panel0.add(self.Save_Settings_BTN)
self.Blank_2 = JLabel(" ")
self.Blank_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_2, self.gbcPanel0)
self.panel0.add(self.Blank_2)
self.Version_Label_1 = JLabel("Version:")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Version_Label_1, self.gbcPanel0)
self.panel0.add(self.Version_Label_1)
self.Version_List = ("2.5", "2.6")
self.Version_CB = JComboBox(self.Version_List)
self.Version_CB.itemStateChanged = self.onchange_version
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Version_CB, self.gbcPanel0)
self.panel0.add(self.Version_CB)
self.Blank_3 = JLabel(" ")
self.Blank_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_3, self.gbcPanel0)
self.panel0.add(self.Blank_3)
self.Plugin_Label_1 = JLabel("Plugins:")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Plugin_Label_1, self.gbcPanel0)
self.panel0.add(self.Plugin_Label_1)
self.Plugin_list = self.get_plugins()
self.Plugin_LB = JList(self.Plugin_list,
valueChanged=self.onchange_plugins_lb)
self.Plugin_LB.setVisibleRowCount(3)
self.scpPlugin_LB = JScrollPane(self.Plugin_LB)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 1
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.scpPlugin_LB, self.gbcPanel0)
self.panel0.add(self.scpPlugin_LB)
self.Blank_4 = JLabel(" ")
self.Blank_4.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 17
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_4, self.gbcPanel0)
self.panel0.add(self.Blank_4)
self.Profile_Label_1 = JLabel("Profile:")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Profile_Label_1, self.gbcPanel0)
self.panel0.add(self.Profile_Label_1)
self.Profile_List = self.get_profiles()
self.Profile_CB = JComboBox(self.Profile_List)
self.Profile_CB.itemStateChanged = self.onchange_profile_cb
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 1
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Profile_CB, self.gbcPanel0)
self.panel0.add(self.Profile_CB)
self.Blank_5 = JLabel(" ")
self.Blank_5.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 21
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_5, self.gbcPanel0)
self.panel0.add(self.Blank_5)
self.Label_2 = JLabel("Process ids to dump (comma seperated list):")
self.Label_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 23
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_2, self.gbcPanel0)
self.panel0.add(self.Label_2)
self.Process_Ids_To_Dump_TF = JTextField(10, focusLost=self.keyPressed)
#self.Process_Ids_To_Dump_TF.getDocument().addDocumentListener()
self.Process_Ids_To_Dump_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 25
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Process_Ids_To_Dump_TF,
self.gbcPanel0)
self.panel0.add(self.Process_Ids_To_Dump_TF)
self.Blank_6 = JLabel(" ")
self.Blank_6.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 27
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_6, self.gbcPanel0)
self.panel0.add(self.Blank_6)
self.Label_3 = JLabel("Message:")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 29
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_3, self.gbcPanel0)
self.panel0.add(self.Label_3)
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
#self.Exclude_File_Sources_CB.setSelected(self.local_settings.getExclude_File_Sources())
#self.Run_Plaso_CB.setSelected(self.local_settings.getRun_Plaso())
#self.Import_Plaso_CB.setSelected(self.local_settings.getImport_Plaso())
self.check_Database_entries()
#pass
# Return the settings used
def getSettings(self):
return self.local_settings
class GUI_TestWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# TODO: Update this for your UI
def checkBoxEvent(self, event):
if self.Exec_Program_CB.isSelected():
self.local_settings.setSetting('Exec_Prog_Flag', 'true')
self.Program_Executable_TF.setEnabled(True)
self.Find_Program_Exec_BTN.setEnabled(True)
else:
self.local_settings.setSetting('Exec_Prog_Flag', 'false')
self.Program_Executable_TF.setText("")
self.Program_Executable_TF.setEnabled(False)
self.Find_Program_Exec_BTN.setEnabled(False)
if self.Imp_File_CB.isSelected():
self.local_settings.setSetting('Imp_File_Flag', 'true')
self.File_Imp_TF.setEnabled(True)
self.Find_Imp_File_BTN.setEnabled(True)
else:
self.local_settings.setSetting('Imp_File_Flag', 'false')
self.File_Imp_TF.setText("")
self.local_settings.setSetting('File_Imp_TF', "")
self.File_Imp_TF.setEnabled(False)
self.Find_Imp_File_BTN.setEnabled(False)
def keyPressed(self, event):
self.local_settings.setSetting('Area', self.area.getText())
def onchange_cb(self, event):
self.local_settings.setSetting('ComboBox', event.item)
#self.Error_Message.setText(event.item)
def onchange_lb(self, event):
self.local_settings.setSetting('ListBox', "")
list_selected = self.List_Box_LB.getSelectedValuesList()
self.local_settings.setSetting('ListBox', str(list_selected))
# if (len(list_selected) > 0):
# self.Error_Message.setText(str(list_selected))
# else:
# self.Error_Message.setText("")
def onClick(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("SQLite", ["sqlite"])
chooseFile.addChoosableFileFilter(filter)
ret = chooseFile.showDialog(self.panel0, "Select SQLite")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
if self.File_Imp_TF.isEnabled():
self.File_Imp_TF.setText(Canonical_file)
self.local_settings.setSetting('File_Imp_TF', Canonical_file)
else:
self.local_settings.setSetting('ExecFile', Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
# TODO: Update this for your UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.Exec_Program_CB = JCheckBox("Execute Program", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Exec_Program_CB, self.gbcPanel0 )
self.panel0.add( self.Exec_Program_CB )
self.Program_Executable_TF = JTextField(20)
self.Program_Executable_TF.setEnabled(False)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Program_Executable_TF, self.gbcPanel0 )
self.panel0.add( self.Program_Executable_TF )
self.Find_Program_Exec_BTN = JButton( "Find Exec", actionPerformed=self.onClick)
self.Find_Program_Exec_BTN.setEnabled(False)
self.rbgPanel0.add( self.Find_Program_Exec_BTN )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Find_Program_Exec_BTN, self.gbcPanel0 )
self.panel0.add( self.Find_Program_Exec_BTN )
self.Imp_File_CB = JCheckBox( "Import File", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Imp_File_CB, self.gbcPanel0 )
self.panel0.add( self.Imp_File_CB )
self.File_Imp_TF = JTextField(20)
self.File_Imp_TF.setEnabled(False)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.File_Imp_TF, self.gbcPanel0 )
self.panel0.add( self.File_Imp_TF )
self.Find_Imp_File_BTN = JButton( "Find File", actionPerformed=self.onClick)
self.Find_Imp_File_BTN.setEnabled(False)
self.rbgPanel0.add( self.Find_Imp_File_BTN )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Find_Imp_File_BTN, self.gbcPanel0 )
self.panel0.add( self.Find_Imp_File_BTN )
self.Check_Box_CB = JCheckBox( "Check Box 1", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Check_Box_CB, self.gbcPanel0 )
self.panel0.add( self.Check_Box_CB )
self.dataComboBox_CB = ("Chocolate", "Ice Cream", "Apple Pie")
self.ComboBox_CB = JComboBox( self.dataComboBox_CB)
self.ComboBox_CB.itemStateChanged = self.onchange_cb
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.ComboBox_CB, self.gbcPanel0 )
self.panel0.add( self.ComboBox_CB )
self.dataList_Box_LB = ("Chocolate", "Ice Cream", "Apple Pie", "Pudding", "Candy" )
self.List_Box_LB = JList( self.dataList_Box_LB, valueChanged=self.onchange_lb)
#self.List_Box_LB.itemStateChanged = self.onchange_lb
self.List_Box_LB.setVisibleRowCount( 3 )
self.scpList_Box_LB = JScrollPane( self.List_Box_LB )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 1
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.scpList_Box_LB, self.gbcPanel0 )
self.panel0.add( self.scpList_Box_LB )
# self.Radio_Button_RB = JRadioButton( "Radio Button" )
# self.rbgPanel0.add( self.Radio_Button_RB )
# self.gbcPanel0.gridx = 7
# self.gbcPanel0.gridy = 17
# self.gbcPanel0.gridwidth = 1
# self.gbcPanel0.gridheight = 1
# self.gbcPanel0.fill = GridBagConstraints.BOTH
# self.gbcPanel0.weightx = 1
# self.gbcPanel0.weighty = 0
# self.gbcPanel0.anchor = GridBagConstraints.NORTH
# self.gbPanel0.setConstraints( self.Radio_Button_RB, self.gbcPanel0 )
# self.panel0.add( self.Radio_Button_RB )
# self.Label_1 = JLabel( "Error Message:")
# self.Label_1.setEnabled(True)
# self.gbcPanel0.gridx = 2
# self.gbcPanel0.gridy = 19
# self.gbcPanel0.gridwidth = 1
# self.gbcPanel0.gridheight = 1
# self.gbcPanel0.fill = GridBagConstraints.BOTH
# self.gbcPanel0.weightx = 1
# self.gbcPanel0.weighty = 0
# self.gbcPanel0.anchor = GridBagConstraints.NORTH
# self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
# self.panel0.add( self.Label_1 )
# self.Error_Message = JLabel( "")
# self.Error_Message.setEnabled(True)
# self.gbcPanel0.gridx = 6
# self.gbcPanel0.gridy = 19
# self.gbcPanel0.gridwidth = 1
# self.gbcPanel0.gridheight = 1
# self.gbcPanel0.fill = GridBagConstraints.BOTH
# self.gbcPanel0.weightx = 1
# self.gbcPanel0.weighty = 0
# self.gbcPanel0.anchor = GridBagConstraints.NORTH
# self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
# self.panel0.add( self.Error_Message )
self.add(self.panel0)
# TODO: Update this for your UI
def customizeComponents(self):
self.Exec_Program_CB.setSelected(self.local_settings.getSetting('Exec_Prog_Flag') == 'true')
self.Imp_File_CB.setSelected(self.local_settings.getSetting('Imp_File_Flag') == 'true')
self.Check_Box_CB.setSelected(self.local_settings.getSetting('Check_Box_1') == 'true')
# Return the settings used
def getSettings(self):
return self.local_settings
def initGui(self):
#~ if DEBUG:
#~ import pdb;
#~ pdb.set_trace()
tabPane = JTabbedPane(JTabbedPane.TOP)
CreditsText = "<html># Burp Custom Deserializer<br/># Copyright (c) 2016, Marco Tinari<br/>#<br/># This program is free software: you can redistribute it and/or modify<br/># it under the terms of the GNU General Public License as published by<br/># the Free Software Foundation, either version 3 of the License, or<br/># (at your option) any later version.<br/>#<br/># This program is distributed in the hope that it will be useful,<br/># but WITHOUT ANY WARRANTY; without even the implied warranty of<br/># MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the<br/># GNU General Public License for more details.<br/>#<br/># You should have received a copy of the GNU General Public License<br/># along with this program. If not, see <http://www.gnu.org/licenses/>.)<br/></html>"
label1 = JLabel(
"<html>Usage:<br>1 - Select the desired encoding functions<br>2 - Enter the name of the parameter in the input field below and press the Apply button!</html>"
)
label2 = JLabel(CreditsText)
panel1 = JPanel()
#set layout
panel1.setLayout(GridLayout(11, 1))
panel2 = JPanel()
panel1.add(label1)
panel2.add(label2)
tabPane.addTab("Configuration", panel1)
tabPane.addTab("Credits", panel2)
applyButton = JButton('Apply', actionPerformed=self.reloadConf)
panel1.add(applyButton, BorderLayout.SOUTH)
#define GET/POST/COOKIE radio button
self.GETparameterTypeRadioButton = JRadioButton('GET parameter')
self.POSTparameterTypeRadioButton = JRadioButton('POST parameter')
self.COOKIEparameterTypeRadioButton = JRadioButton('COOKIE parameter')
self.POSTparameterTypeRadioButton.setSelected(True)
group = ButtonGroup()
group.add(self.GETparameterTypeRadioButton)
group.add(self.POSTparameterTypeRadioButton)
group.add(self.COOKIEparameterTypeRadioButton)
self.base64Enabled = JCheckBox("Base64 encode")
self.URLEnabled = JCheckBox("URL encode")
self.ASCII2HexEnabled = JCheckBox("ASCII to Hex")
self.ScannerEnabled = JCheckBox(
"<html>Enable serialization in Burp Scanner<br>Usage:<br>1.Place unencoded values inside intruder request and define the placeholder positions<br>2.rightclick->Actively scan defined insertion points)</html>"
)
self.IntruderEnabled = JCheckBox(
"<html>Enable serialization in Burp Intruder<br>Usage:<br>1.Place unencoded values inside intruder request and define the placeholder positions<br>2.Start the attack</html>"
)
self.parameterName = JTextField("Parameter name goes here...", 60)
#set the tooltips
self.parameterName.setToolTipText(
"Fill in the parameter name and apply")
self.base64Enabled.setToolTipText("Enable base64 encoding/decoding")
self.ASCII2HexEnabled.setToolTipText(
"Enable ASCII 2 Hex encoding/decoding")
self.URLEnabled.setToolTipText("Enable URL encoding/decoding")
self.IntruderEnabled.setToolTipText(
"Check this if You want the extension to intercept and modify every request made by the Burp Intruder containing the selected paramter"
)
self.ScannerEnabled.setToolTipText(
"Check this if You want the extension to intercept and modify every request made by the Burp Scanner containing the selected paramter"
)
#add checkboxes to the panel
panel1.add(self.parameterName)
panel1.add(self.POSTparameterTypeRadioButton)
panel1.add(self.GETparameterTypeRadioButton)
panel1.add(self.COOKIEparameterTypeRadioButton)
panel1.add(self.base64Enabled)
panel1.add(self.URLEnabled)
panel1.add(self.ASCII2HexEnabled)
panel1.add(self.IntruderEnabled)
panel1.add(self.ScannerEnabled)
#assign tabPane
self.tab = tabPane
class GUI_Test_SQLSettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Exec_Program_CB.isSelected():
self.local_settings.setSetting('Exec_Prog_Flag', 'true')
self.Program_Executable_TF.setEnabled(True)
self.Find_Program_Exec_BTN.setEnabled(True)
else:
self.local_settings.setSetting('Exec_Prog_Flag', 'false')
self.Program_Executable_TF.setText("")
self.Program_Executable_TF.setEnabled(False)
self.Find_Program_Exec_BTN.setEnabled(False)
# Check to see if there are any entries that need to be populated from the database.
def check_Database_entries(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = os.path.join(head, "GUI_Settings.db3")
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Select Setting_Name, Setting_Value from settings;'
resultSet = stmt.executeQuery(SQL_Statement)
while resultSet.next():
if resultSet.getString("Setting_Name") == "Program_Exec_Name":
self.Program_Executable_TF.setText(resultSet.getString("Setting_Value"))
self.local_settings.setSetting('ExecFile', resultSet.getString("Setting_Value"))
self.local_settings.setSetting('Exec_Prog_Flag', 'true')
self.Exec_Program_CB.setSelected(True)
self.Program_Executable_TF.setEnabled(True)
self.Find_Program_Exec_BTN.setEnabled(True)
self.Error_Message.setText("Settings Read successfully!")
except SQLException as e:
self.Error_Message.setText("Error Reading Settings Database")
stmt.close()
dbConn.close()
# Save entries from the GUI to the database.
def SaveSettings(self, e):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = os.path.join(head, "GUI_Settings.db3")
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Insert into settings (Setting_Name, Setting_Value) values ("Program_Exec_Name", "' + \
self.Program_Executable_TF.getText() + '");'
resultSet = stmt.executeQuery(SQL_Statement)
self.Error_Message.setText("Settings Saved")
except SQLException as e:
self.Error_Message.setText("Error Inserting Settings " + str(e))
stmt.close()
dbConn.close()
# When button to find file is clicked then open dialog to find the file and return it.
def onClick(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("SQLite", ["sqlite"])
chooseFile.addChoosableFileFilter(filter)
ret = chooseFile.showDialog(self.panel0, "Select SQLite")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setSetting('ExecFile', Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.Exec_Program_CB = JCheckBox("Execute Program", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Exec_Program_CB, self.gbcPanel0 )
self.panel0.add( self.Exec_Program_CB )
self.Program_Executable_TF = JTextField(20)
self.Program_Executable_TF.setEnabled(False)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Program_Executable_TF, self.gbcPanel0 )
self.panel0.add( self.Program_Executable_TF )
self.Find_Program_Exec_BTN = JButton( "Find Exec", actionPerformed=self.onClick)
self.Find_Program_Exec_BTN.setEnabled(False)
self.rbgPanel0.add( self.Find_Program_Exec_BTN )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Find_Program_Exec_BTN, self.gbcPanel0 )
self.panel0.add( self.Find_Program_Exec_BTN )
self.Blank_1 = JLabel( " ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Save_Settings_BTN = JButton( "Save Settings", actionPerformed=self.SaveSettings)
self.Save_Settings_BTN.setEnabled(True)
self.rbgPanel0.add( self.Save_Settings_BTN )
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Save_Settings_BTN, self.gbcPanel0 )
self.panel0.add( self.Save_Settings_BTN )
self.Blank_2 = JLabel( " ")
self.Blank_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_2, self.gbcPanel0 )
self.panel0.add( self.Blank_2 )
self.Label_1 = JLabel( "Error Message:")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
self.panel0.add( self.Label_1 )
self.Error_Message = JLabel( "")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
self.panel0.add( self.Error_Message )
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
self.Exec_Program_CB.setSelected(self.local_settings.getSetting('Exec_Prog_Flag') == 'true')
self.check_Database_entries()
# Return the settings used
def getSettings(self):
return self.local_settings
def loginPage():
global heading
global rbAdmin
global rbTeacher
global rbStudent
global frame
global tfLoginId
global tfPassword
frame = JFrame("Login Form ")
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE)
frame.setSize(500, 500)
frame.setLocation(200, 200)
frame.setLayout(None)
frame.setVisible(True)
panel = JPanel()
panel.setSize(500, 500)
panel.setLocation(0, 0)
panel.setLayout(None)
panel.setBackground(Color.BLUE)
heading = JLabel("Admin Login")
heading.setBounds(200, 50, 150, 30)
rbAdmin = JRadioButton("Admin", actionPerformed=clickRadio)
rbTeacher = JRadioButton("Teacher", actionPerformed=clickRadio)
rbStudent = JRadioButton("Student", actionPerformed=clickRadio)
rbAdmin.setBounds(100, 150, 100, 20)
rbTeacher.setBounds(200, 150, 100, 20)
rbStudent.setBounds(300, 150, 100, 20)
btnGroup = ButtonGroup()
btnGroup.add(rbAdmin)
btnGroup.add(rbTeacher)
btnGroup.add(rbStudent)
lbLoginId = JLabel("LoginId")
lbPassword = JLabel("Password")
lbLoginId.setBounds(100, 230, 150, 30)
lbPassword.setBounds(100, 300, 150, 30)
tfLoginId = JTextField()
tfPassword = JTextField()
tfLoginId.setBounds(250, 230, 150, 30)
tfPassword.setBounds(250, 300, 150, 30)
btnLogin = JButton("Login", actionPerformed=clickLogin)
btnLogin.setBounds(350, 350, 100, 30)
btnReg = JButton("New Institute Registration", actionPerformed=clickReg)
btnReg.setBounds(350, 400, 100, 30)
panel.add(heading)
panel.add(rbAdmin)
panel.add(rbTeacher)
panel.add(rbStudent)
panel.add(lbLoginId)
panel.add(lbPassword)
panel.add(tfLoginId)
panel.add(tfPassword)
panel.add(btnLogin)
panel.add(btnReg)
panel.setVisible(True)
frame.add(panel)
def launchGui(self, caller):
self._stdout = PrintWriter(self._callbacks.getStdout(), True)
self._stdout.println('Launching gui')
callMessage = caller.getSelectedMessages()
self.msg1 = callMessage[0]
#setup frame
self.frame = JFrame('Create Issue', windowClosing=self.closeUI)
Border = BorderFactory.createLineBorder(Color.BLACK)
#create split panel to add issue panel and template panel
self.splitPane = JSplitPane(JSplitPane.HORIZONTAL_SPLIT)
self.frame.add(self.splitPane)
#panel setup and add to splitPane
self.issuePanel = JPanel(GridLayout(0,2))
self.splitPane.setLeftComponent(self.issuePanel)
#setup issue name text fields to add to panel
self.issueNameField = JTextField('',15)
self.issueNameLabel = JLabel("IssueName:", SwingConstants.CENTER)
self.issuePanel.add(self.issueNameLabel)
self.issuePanel.add(self.issueNameField)
#add issue detail text area
self.issueDetailField = JTextArea()
self.issueDetailField.editable = True
self.issueDetailField.wrapStyleWord = True
self.issueDetailField.lineWrap = True
self.issueDetailField.alignmentX = Component.LEFT_ALIGNMENT
self.issueDetailField.size = (200, 20)
self.issueDetailField.setBorder(Border)
self.idfSp = JScrollPane()
self.idfSp.getViewport().setView((self.issueDetailField))
self.issuePanel.add(JLabel("Issue Detail:", SwingConstants.CENTER))
self.issuePanel.add(self.idfSp)
self.issueBackgroundField= JTextArea()
self.issueBackgroundField.editable = True
self.issueBackgroundField.wrapStyleWord = True
self.issueBackgroundField.lineWrap = True
self.issueBackgroundField.alignmentX = Component.LEFT_ALIGNMENT
self.issueBackgroundField.size = (200, 20)
self.issueBackgroundField.setBorder(Border)
self.ibfSp = JScrollPane()
self.ibfSp.getViewport().setView((self.issueBackgroundField))
self.issuePanel.add(JLabel("Issue Background:", SwingConstants.CENTER))
self.issuePanel.add(self.ibfSp)
#add remediation detail text area
self.remediationDetailField = JTextArea()
self.remediationDetailField.editable = True
self.remediationDetailField.wrapStyleWord = True
self.remediationDetailField.lineWrap = True
self.remediationDetailField.alignmentX = Component.LEFT_ALIGNMENT
self.remediationDetailField.size = (200, 20)
self.remediationDetailField.setBorder(Border)
self.rdfSp = JScrollPane()
self.rdfSp.getViewport().setView((self.remediationDetailField))
self.issuePanel.add(JLabel("Remediation Detail:", SwingConstants.CENTER))
self.issuePanel.add(self.rdfSp)
self.remediationBackgroundField= JTextArea()
self.remediationBackgroundField.editable = True
self.remediationBackgroundField.wrapStyleWord = True
self.remediationBackgroundField.lineWrap = True
self.remediationBackgroundField.alignmentX = Component.LEFT_ALIGNMENT
self.remediationBackgroundField.size = (200, 20)
self.remediationBackgroundField.setBorder(Border)
self.rbfSp = JScrollPane()
self.rbfSp.getViewport().setView((self.remediationBackgroundField))
self.issuePanel.add(JLabel("Remediation Background:", SwingConstants.CENTER))
self.issuePanel.add(self.rbfSp)
#add radio buttons for severity
self.radioBtnSevHigh = JRadioButton('High', actionPerformed=None)
self.radioBtnSevMedium = JRadioButton('Medium', actionPerformed=None)
self.radioBtnSevLow = JRadioButton('Low', actionPerformed=None)
severityButtonGroup = ButtonGroup()
severityButtonGroup.add(self.radioBtnSevHigh)
severityButtonGroup.add(self.radioBtnSevMedium)
severityButtonGroup.add(self.radioBtnSevLow)
self.radioBtnSevHigh.setSelected(True)
self.issuePanel.add(JLabel("Severity:", SwingConstants.CENTER))
self.issuePanel.add(self.radioBtnSevHigh)
self.issuePanel.add(self.radioBtnSevMedium)
self.issuePanel.add(self.radioBtnSevLow)
self.reqPattern = JTextField('',15)
self.issuePanel.add(JLabel("Mark Pattern in Request:", SwingConstants.CENTER))
self.issuePanel.add(self.reqPattern)
self.resPattern = JTextField('',15)
self.issuePanel.add(JLabel("Mark Pattern in Response:", SwingConstants.CENTER))
self.issuePanel.add(self.resPattern)
#add a button
self.issueButton = JButton('Add!', actionPerformed=lambda x, m=self.msg1: self.logScanIssue(m))
self.issuePanel.add(self.issueButton)
#template panel setup
self.templatePanel = JPanel(GridLayout(1,2))
self.splitPane.setRightComponent(self.templatePanel)
#add a list of templates
self.templatePanel.add(JLabel("Select from Templates", SwingConstants.CENTER))
self.templateData = tuple(self.tmpl.keys())
self.templateList = JList(self.templateData)
self.templateScrollPane = JScrollPane()
#self.templateScrollPane.setPreferredSize(Dimension(100,125))
self.templateScrollPane.getViewport().setView((self.templateList))
self.templatePanel.add(self.templateScrollPane)
self.templateButton = JButton('Apply', actionPerformed=self.applyTemplate)
self.templatePanel.add(self.templateButton)
#pack up the frame and display it
self.frame.pack()
self.show()
class SpikeLineOverlay(core.DataViewComponent):
def __init__(self, view, name, infunc, outfunc, lfunc, inargs=(), outargs=(), largs=(), label=None):
core.DataViewComponent.__init__(self, label)
self.view = view
self.name = name
self.infunc = infunc
self.outfunc = outfunc
self.lfunc = lfunc
self.in_neuron = 0
self.out_neuron = 0
self.indata = self.view.watcher.watch(name, infunc, args=inargs)
self.outdata = self.view.watcher.watch(name, outfunc, args=outargs)
self.ldata = self.view.watcher.watch(name, lfunc, args=largs)
self.label_height = 10
self.border_top = 10
self.border_left = 30
self.border_right = 30
self.border_bottom = 20
self.setSize(300, 200)
self.last_maxy = None
self.neurons = len(self.outdata.get_first())
self.popup.add(JPopupMenu.Separator())
in_menu = JMenu("in neuron")
self.in_group = ButtonGroup()
for i in range(self.neurons):
checked = i == self.in_neuron
button = JRadioButtonMenuItem(
"%s[%d]" % ("in", i), checked, actionPerformed=lambda x, i=i: self.set_in_neuron(i)
)
in_menu.add(button)
self.in_group.add(button)
self.popup.add(in_menu)
out_menu = JMenu("out neuron")
self.in_group = ButtonGroup()
for i in range(self.neurons):
checked = i == self.out_neuron
button = JRadioButtonMenuItem(
"%s[%d]" % ("out", i), checked, actionPerformed=lambda x, i=i: self.set_out_neuron(i)
)
out_menu.add(button)
self.in_group.add(button)
self.popup.add(out_menu)
def set_in_neuron(self, x):
self.in_neuron = x
self.repaint()
def set_out_neuron(self, x):
self.out_neuron = x
self.repaint()
def save(self):
d = core.DataViewComponent.save(self)
return d
def restore(self, d):
core.DataViewComponent.restore(self, d)
def paintComponent(self, g):
core.DataViewComponent.paintComponent(self, g)
border_top = self.border_top + self.label_offset
g.color = Color(0.8, 0.8, 0.8)
g.drawLine(
self.border_left,
self.height - self.border_bottom,
self.size.width - self.border_right,
self.size.height - self.border_bottom,
)
pts = int(self.view.time_shown / self.view.dt)
start = self.view.current_tick - pts + 1
if start < 0:
start = 0
if start <= self.view.timelog.tick_count - self.view.timelog.tick_limit:
start = self.view.timelog.tick_count - self.view.timelog.tick_limit + 1
g.color = Color.black
txt = "%4g" % ((start + pts) * self.view.dt)
bounds = g.font.getStringBounds(txt, g.fontRenderContext)
g.drawString(
txt,
self.size.width - self.border_right - bounds.width / 2,
self.size.height - self.border_bottom + bounds.height,
)
txt = "%4g" % ((start) * self.view.dt)
bounds = g.font.getStringBounds(txt, g.fontRenderContext)
g.drawString(txt, self.border_left - bounds.width / 2, self.size.height - self.border_bottom + bounds.height)
indata = self.indata.get(start=start, count=pts)
outdata = self.outdata.get(start=start, count=pts)
ldata = self.ldata.get(start=start, count=pts)
now = self.view.current_tick - start
for i in range(now + 1, len(indata)):
indata[i] = None
outdata[i] = None
ldata[i] = None
maxy = None
miny = None
dx = float(self.size.width - self.border_left - self.border_right - 1) / (pts - 1)
index = (self.neurons * self.out_neuron) + self.in_neuron
fdata = [safe_get_index(x, index) for x in ldata]
trimmed = [x for x in fdata if x is not None]
if len(trimmed) > 0:
fmaxy = max(trimmed)
fminy = min(trimmed)
fmaxy = round(fmaxy)[1]
fminy = round(fminy)[0]
if maxy is None or fmaxy > maxy:
maxy = fmaxy
if miny is None or fminy < miny:
miny = fminy
if maxy is None:
maxy = 1.0
if miny is None:
miny = -1.0
if maxy > -miny:
miny = -maxy
if miny < -maxy:
maxy = -miny
self.last_maxy = maxy
if maxy == miny:
yscale = 0
else:
yscale = float(self.size.height - self.border_bottom - border_top) / (maxy - miny)
# draw zero line
g.color = Color(0.8, 0.8, 0.8)
y0 = int(self.size.height - (0 - miny) * yscale - self.border_bottom)
g.drawLine(self.border_left, y0, self.width - self.border_right, y0)
# draw ticks
tick_count = 10
if ("%f" % maxy)[0] == "2":
tick_count = 8
tick_span = maxy * 2.0 / tick_count
tick_size = 3
for i in range(1, tick_count):
yt = int(self.size.height - (i * tick_span) * yscale - self.border_bottom)
g.drawLine(self.border_left, yt, self.border_left + tick_size, yt)
g.drawLine(self.width - self.border_right, yt, self.width - self.border_right - tick_size, yt)
g.color = Color.black
if (self.size.width - self.border_left - self.border_right) > 0:
skip = (len(fdata) / (self.size.width - self.border_left - self.border_right)) - 1
if skip < 0:
skip = 0
offset = start % (skip + 1)
for i in range(len(fdata) - 1 - skip):
if skip == 0 or (i + offset) % (skip + 1) == 0:
if fdata[i] is not None and fdata[i + 1 + skip] is not None:
y1 = self.size.height - (fdata[i] - miny) * yscale - self.border_bottom
y2 = self.size.height - (fdata[i + 1 + skip] - miny) * yscale - self.border_bottom
g.drawLine(
int(i * dx + self.border_left),
int(y1),
int((i + 1 + skip) * dx + self.border_left),
int(y2),
)
if maxy is not None:
g.drawString("%6g" % maxy, 0, 20 + border_top)
if miny is not None:
g.drawString("%6g" % miny, 0, self.size.height - self.border_bottom - 5)
dy = (
float(self.size.height - self.border_bottom - border_top) / 2.0
) # div by 2 if you put in on top, out on bottom
# Mark in and out spike
txt = "in"
bounds = g.font.getStringBounds(txt, g.fontRenderContext)
g.drawString(txt, 5, 10 + border_top + dy * 0.5)
txt = "out"
bounds = g.font.getStringBounds(txt, g.fontRenderContext)
g.drawString(txt, 5, 10 + border_top + dy * 1.5)
g.color = Color.blue
for i, d in enumerate(indata):
if d is None:
continue
spike = d[self.in_neuron]
if spike:
x = int(i * dx + self.border_left)
y = int(border_top)
w = int(dx) - 1
h = int(dy) - 1
if w < 1:
w = 1
if h < 1:
h = 1
if w <= 1 and h <= 1:
g.drawLine(x, y, x, y)
else:
g.fillRect(x, y, w, h)
g.color = Color.red
for i, d in enumerate(outdata):
if d is None:
continue
spike = d[self.out_neuron]
if spike:
x = int(i * dx + self.border_left)
y = int(border_top + dy)
w = int(dx) - 1
h = int(dy) - 1
if w < 1:
w = 1
if h < 1:
h = 1
if w <= 1 and h <= 1:
g.drawLine(x, y, x, y)
else:
g.fillRect(x, y, w, h)
class VolatilitySettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel
):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# Check to see if there are any entries that need to be populated from the database.
def check_Database_entries(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Select Setting_Name, Setting_Value from settings;'
resultSet = stmt.executeQuery(SQL_Statement)
while resultSet.next():
if resultSet.getString(
"Setting_Name") == "Volatility_Executable_Directory":
self.Program_Executable_TF.setText(
resultSet.getString("Setting_Value"))
self.local_settings.setVolatility_Directory(
resultSet.getString("Setting_Value"))
self.local_settings.setVolatility_Dir_Found(True)
# if resultSet.getString("Setting_Name") == "Volatility_Version":
# self.Version_CB.setSelectedItem(resultSet.getString("Setting_Value"))
self.Error_Message.setText("Settings Read successfully!")
except SQLException as e:
self.Error_Message.setText("Error Reading Settings Database")
stmt.close()
dbConn.close()
# Save entries from the GUI to the database.
def SaveSettings(self, e):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" %
settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings")
try:
stmt = dbConn.createStatement()
SQL_Statement = ""
if (self.local_settings.getVolatility_Dir_Found()):
SQL_Statement = 'Update settings set Setting_Value = "' + self.Program_Executable_TF.getText() + '"' + \
' where setting_name = "Volatility_Executable_Directory";'
# SQL_Statement2 = 'Update settings set Setting_Value = "' + self.Version_CB.getSelectedItem() + '"' + \
# ' where setting_name = "Volatility_Version";'
else:
SQL_Statement = 'Insert into settings (Setting_Name, Setting_Value) values ("Volatility_Executable_Directory", "' + \
self.Program_Executable_TF.getText() + '");'
# SQL_Statement2 = 'Insert into settings (Setting_Name, Setting_Value) values ("Volatility_Version", "' + \
# self.Version_CB.getSelectedItem() + '");'
stmt.execute(SQL_Statement)
# stmt.execute(SQL_Statement2)
self.Error_Message.setText("Volatility Executable Directory Saved")
self.local_settings.setVolatility_Directory(
self.Program_Executable_TF.getText())
except SQLException as e:
self.Error_Message.setText(e.getMessage())
stmt.close()
dbConn.close()
# When button to find file is clicked then open dialog to find the file and return it.
def Find_Dir(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("All", ["*.*"])
chooseFile.addChoosableFileFilter(filter)
#chooseFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY)
ret = chooseFile.showDialog(self.panel0, "Find Volatility Directory")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setVolatility_Directory(Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
def keyPressed(self, event):
self.local_settings.setProcessIDs(
self.Process_Ids_To_Dump_TF.getText())
#self.Error_Message.setText(self.Process_Ids_To_Dump_TF.getText())
def checkBoxEvent(self, event):
if self.Check_Box.isSelected():
self.local_settings.setFlag(True)
else:
self.local_settings.setFlag(False)
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout(self.gbPanel0)
self.Error_Message = JLabel("")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 31
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Error_Message, self.gbcPanel0)
self.panel0.add(self.Error_Message)
self.Label_1 = JLabel("Volatility Executable Directory")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_1, self.gbcPanel0)
self.panel0.add(self.Label_1)
self.Program_Executable_TF = JTextField(10)
self.Program_Executable_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Program_Executable_TF,
self.gbcPanel0)
self.panel0.add(self.Program_Executable_TF)
self.Find_Program_Exec_BTN = JButton("Find Dir",
actionPerformed=self.Find_Dir)
self.Find_Program_Exec_BTN.setEnabled(True)
self.rbgPanel0.add(self.Find_Program_Exec_BTN)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Find_Program_Exec_BTN,
self.gbcPanel0)
self.panel0.add(self.Find_Program_Exec_BTN)
self.Blank_1 = JLabel(" ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_1, self.gbcPanel0)
self.panel0.add(self.Blank_1)
self.Save_Settings_BTN = JButton("Save Volatility Exec Dir",
actionPerformed=self.SaveSettings)
self.Save_Settings_BTN.setEnabled(True)
self.rbgPanel0.add(self.Save_Settings_BTN)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Save_Settings_BTN, self.gbcPanel0)
self.panel0.add(self.Save_Settings_BTN)
self.Blank_2 = JLabel(" ")
self.Blank_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_2, self.gbcPanel0)
self.panel0.add(self.Blank_2)
self.Blank_3 = JLabel(" ")
self.Blank_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_3, self.gbcPanel0)
self.panel0.add(self.Blank_3)
self.Check_Box = JCheckBox(
"Extract and Create Memory Image from Hiberfile",
actionPerformed=self.checkBoxEvent)
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Check_Box, self.gbcPanel0)
self.panel0.add(self.Check_Box)
self.Blank_4 = JLabel(" ")
self.Blank_4.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 17
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_4, self.gbcPanel0)
self.panel0.add(self.Blank_4)
self.Blank_5 = JLabel(" ")
self.Blank_5.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 21
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_5, self.gbcPanel0)
self.panel0.add(self.Blank_5)
self.Blank_6 = JLabel(" ")
self.Blank_6.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 27
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_6, self.gbcPanel0)
self.panel0.add(self.Blank_6)
self.Label_3 = JLabel("Message:")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 29
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_3, self.gbcPanel0)
self.panel0.add(self.Label_3)
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
self.Check_Box.setSelected(self.local_settings.getFlag())
self.check_Database_entries()
#pass
# Return the settings used
def getSettings(self):
return self.local_settings
def __init__(self):
#obtain prefixes from folder
self.dict1 = self.obtain_prefixes(
) #Run prefix selection function - sets source directory, requests prefix size, outputs prefix dictionary
lst = list(self.dict1.keys()) #pull prefixes only, as list
self.lang = lst
self.lst = JList(self.lang, valueChanged=self.listSelect
) # pass prefix list to GUI selection list
# general GUI layout parameters, no data processing here
self.frame = JFrame("Image Selection")
self.frame.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE)
self.frame.setLocation(100, 100)
self.frame.setSize(800, 350)
self.frame.setLayout(BorderLayout())
self.frame.add(self.lst, BorderLayout.NORTH)
self.lst.selectionMode = ListSelectionModel.MULTIPLE_INTERVAL_SELECTION
self.button1 = JButton('Select item(s)',
actionPerformed=self.clickhere)
#Save option radio buttons and file extension selection
#set main right panel (sub panels will fit within this)
rightpanel = JPanel()
rightpanel.setLayout(BoxLayout(rightpanel, BoxLayout.Y_AXIS))
#set up savestate panel
buttonpanel = JPanel()
self.radiobutton1 = JRadioButton(
"Open selected 3D stacks and max projections \n and save max projections",
True)
self.radiobutton2 = JRadioButton(
"Open selected 3D stacks and max projections \n and DO NOT save max projections"
)
infoLabel = JLabel(
"<html>Hold ctrl and click multiple prefixes to select multiple options. Will load stacks and MIPs separately <br><br> Type file extension in text field below:</html>",
SwingConstants.LEFT)
grp = ButtonGroup()
grp.add(self.radiobutton1)
grp.add(self.radiobutton2)
#buttonpanel.setLayout(BoxLayout(buttonpanel, BoxLayout.Y_AXIS))
buttonpanel.add(Box.createVerticalGlue())
buttonpanel.add(infoLabel)
buttonpanel.add(Box.createRigidArea(Dimension(0, 5)))
buttonpanel.add(self.radiobutton1)
buttonpanel.add(Box.createRigidArea(Dimension(0, 5)))
buttonpanel.add(self.radiobutton2)
#file extension instruction panel
infopanel = JPanel()
infopanel.setLayout(FlowLayout(FlowLayout.LEFT))
infopanel.setMaximumSize(
infopanel.setPreferredSize(Dimension(650, 100)))
infopanel.add(infoLabel)
#file extension input
inputPanel = JPanel()
inputPanel.setLayout(BoxLayout(inputPanel, BoxLayout.X_AXIS))
self.filetype = JTextField(".tif", 15)
self.filetype.setMaximumSize(self.filetype.getPreferredSize())
inputPanel.add(self.filetype)
########### WIP - integrate prefix selection with main pane, with dynamically updating prefix list
##infoLabel3 = JLabel("how long is the file prefix to group by?(integer value only)")
##self.prefix_init = JTextField()
##buttonpanel.add(infoLabel3)
##buttonpanel.add(self.prefix_init)
########### !WIP
#add file extension and savestate panels to main panel
rightpanel.add(infopanel)
rightpanel.add(inputPanel)
rightpanel.add(buttonpanel, BorderLayout.EAST)
#split list and radiobutton pane (construct overall window)
spl = JSplitPane(JSplitPane.HORIZONTAL_SPLIT)
spl.leftComponent = JScrollPane(self.lst)
spl.setDividerLocation(150)
spl.rightComponent = rightpanel
self.frame.add(spl)
self.frame.add(self.button1, BorderLayout.SOUTH)
# GUI layout done, initialise GUI to select prefixes, file extension and save option
self.frame.setVisible(True)
def __init__(self, parent, title, app):
from javax.swing import JCheckBox, JRadioButton, ButtonGroup
self.app = app
border = BorderFactory.createEmptyBorder(5, 7, 5, 7)
self.getContentPane().setBorder(border)
self.getContentPane().setLayout(BorderLayout(0, 5))
self.tabbedPane = JTabbedPane()
#1 Tab: general
panel1 = JPanel()
panel1.setBorder(BorderFactory.createEmptyBorder(7, 7, 7, 7))
panel1.setLayout(BoxLayout(panel1, BoxLayout.PAGE_AXIS))
#Checkbutton to enable/disable update check when script starts
self.updateCBtn = JCheckBox(self.app.strings.getString("updateCBtn"))
self.updateCBtn.setToolTipText(
self.app.strings.getString("updateCBtn_tooltip"))
#Download tools
downloadBtn = JButton(self.app.strings.getString("updatesBtn"),
ImageProvider.get("dialogs", "refresh"),
actionPerformed=self.on_downloadBtn_clicked)
downloadBtn.setToolTipText(
self.app.strings.getString("updatesBtn_tooltip"))
#Checkbuttons for enabling/disabling tools
toolsPanel = JPanel(BorderLayout(0, 5))
title = self.app.strings.getString("enable_disable_tools")
toolsPanel.setBorder(BorderFactory.createTitledBorder(title))
infoLbl = JLabel(self.app.strings.getString("JOSM_restart_warning"))
infoLbl.setFont(infoLbl.getFont().deriveFont(Font.ITALIC))
toolsPanel.add(infoLbl, BorderLayout.PAGE_START)
toolsStatusPane = JPanel(GridLayout(len(self.app.realTools), 0))
self.toolsCBtns = []
for tool in self.app.realTools:
toolCBtn = JCheckBox()
toolCBtn.addItemListener(self)
toolLbl = JLabel(tool.title, tool.bigIcon, JLabel.LEFT)
self.toolsCBtns.append(toolCBtn)
toolPane = JPanel()
toolPane.setLayout(BoxLayout(toolPane, BoxLayout.X_AXIS))
toolPane.add(toolCBtn)
toolPane.add(toolLbl)
toolsStatusPane.add(toolPane)
toolsPanel.add(toolsStatusPane, BorderLayout.CENTER)
#Radiobuttons for enabling/disabling layers when a new one
#is added
layersPanel = JPanel(GridLayout(0, 1))
title = self.app.strings.getString("errors_layers_manager")
layersPanel.setBorder(BorderFactory.createTitledBorder(title))
errorLayersLbl = JLabel(
self.app.strings.getString("errors_layers_info"))
errorLayersLbl.setFont(errorLayersLbl.getFont().deriveFont(
Font.ITALIC))
layersPanel.add(errorLayersLbl)
self.layersRBtns = {}
group = ButtonGroup()
for mode in self.app.layersModes:
layerRBtn = JRadioButton(self.app.strings.getString("%s" % mode))
group.add(layerRBtn)
layersPanel.add(layerRBtn)
self.layersRBtns[mode] = layerRBtn
#Max number of errors text field
self.maxErrorsNumberTextField = JTextField()
self.maxErrorsNumberTextField.setToolTipText(
self.app.strings.getString("maxErrorsNumberTextField_tooltip"))
self.maxErrorsNumberTFieldDefaultBorder = self.maxErrorsNumberTextField.getBorder(
)
self.maxErrorsNumberTextField.getDocument().addDocumentListener(
ErrNumTextListener(self))
#layout
self.updateCBtn.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(self.updateCBtn)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
downloadBtn.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(downloadBtn)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
toolsPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(toolsPanel)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
layersPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(layersPanel)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
maxErrP = JPanel(BorderLayout(5, 0))
maxErrP.add(JLabel(self.app.strings.getString("max_errors_number")),
BorderLayout.LINE_START)
maxErrP.add(self.maxErrorsNumberTextField, BorderLayout.CENTER)
p = JPanel(BorderLayout())
p.add(maxErrP, BorderLayout.PAGE_START)
p.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(p)
self.tabbedPane.addTab(self.app.strings.getString("tab_1_title"), None,
panel1, None)
#2 Tab: favourite zones
panel2 = JPanel(BorderLayout(5, 15))
panel2.setBorder(BorderFactory.createEmptyBorder(7, 7, 7, 7))
#status
topPanel = JPanel()
topPanel.setLayout(BoxLayout(topPanel, BoxLayout.Y_AXIS))
infoPanel = HtmlPanel(self.app.strings.getString("fav_zones_info"))
infoPanel.getEditorPane().addHyperlinkListener(self)
infoPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
self.favZoneStatusCBtn = JCheckBox(
self.app.strings.getString("activate_fav_area"),
actionListener=self)
self.favZoneStatusCBtn.setToolTipText(
self.app.strings.getString("activate_fav_area_tooltip"))
self.favZoneStatusCBtn.setAlignmentX(Component.LEFT_ALIGNMENT)
topPanel.add(infoPanel)
topPanel.add(Box.createRigidArea(Dimension(0, 10)))
topPanel.add(self.favZoneStatusCBtn)
#table
self.zonesTable = JTable()
tableSelectionModel = self.zonesTable.getSelectionModel()
tableSelectionModel.addListSelectionListener(ZonesTableListener(self))
columns = [
"",
self.app.strings.getString("Type"),
self.app.strings.getString("Name")
]
tableModel = ZonesTableModel([], columns)
self.zonesTable.setModel(tableModel)
self.scrollPane = JScrollPane(self.zonesTable)
#map
self.zonesMap = JMapViewer()
self.zonesMap.setZoomContolsVisible(False)
self.zonesMap.setMinimumSize(Dimension(100, 200))
#buttons
self.removeBtn = JButton(self.app.strings.getString("Remove"),
ImageProvider.get("dialogs", "delete"),
actionPerformed=self.on_removeBtn_clicked)
self.removeBtn.setToolTipText(
self.app.strings.getString("remove_tooltip"))
newBtn = JButton(self.app.strings.getString("New"),
ImageProvider.get("dialogs", "add"),
actionPerformed=self.on_newBtn_clicked)
newBtn.setToolTipText(self.app.strings.getString("new_tooltip"))
#layout
panel2.add(topPanel, BorderLayout.PAGE_START)
panel2.add(self.scrollPane, BorderLayout.LINE_START)
panel2.add(self.zonesMap, BorderLayout.CENTER)
self.buttonsPanel = JPanel()
self.buttonsPanel.add(self.removeBtn)
self.buttonsPanel.add(newBtn)
panel2.add(self.buttonsPanel, BorderLayout.PAGE_END)
self.tabbedPane.addTab(self.app.strings.getString("tab_2_title"), None,
panel2, None)
#3 Tab Tools options
panel3 = JPanel()
panel3.setLayout(BoxLayout(panel3, BoxLayout.Y_AXIS))
panel3.setBorder(BorderFactory.createEmptyBorder(7, 7, 7, 7))
for tool in self.app.realTools:
if hasattr(tool, 'prefs'):
p = JPanel(FlowLayout(FlowLayout.LEFT))
p.setBorder(BorderFactory.createTitledBorder(tool.title))
p.add(tool.prefsGui)
panel3.add(p)
self.tabbedPane.addTab(self.app.strings.getString("tab_3_title"), None,
panel3, None)
self.add(self.tabbedPane, BorderLayout.CENTER)
exitPanel = JPanel()
saveBtn = JButton(self.app.strings.getString("OK"),
ImageProvider.get("ok"),
actionPerformed=self.on_saveBtn_clicked)
cancelBtn = JButton(self.app.strings.getString("cancel"),
ImageProvider.get("cancel"),
actionPerformed=self.on_cancelBtn_clicked)
saveBtn.setToolTipText(self.app.strings.getString("save_preferences"))
saveBtn.setAlignmentX(0.5)
exitPanel.add(saveBtn)
exitPanel.add(cancelBtn)
self.add(exitPanel, BorderLayout.PAGE_END)
self.addWindowListener(self)
self.pack()
class VolatilitySettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# When button to find file is clicked then open dialog to find the file and return it.
def Find_Dir(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("All", ["*.*"])
chooseFile.addChoosableFileFilter(filter)
#chooseFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY)
ret = chooseFile.showDialog(self.panel0, "Find Volatility Directory")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setSetting('Volatility_Directory', Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
def keyPressed(self, event):
self.local_settings.setProcessIDs(self.Process_Ids_To_Dump_TF.getText())
#self.Error_Message.setText(self.Process_Ids_To_Dump_TF.getText())
def checkBoxEvent(self, event):
if self.Check_Box.isSelected():
self.local_settings.setSetting('Flag', 'true')
else:
self.local_settings.setSetting('Flag', 'False')
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.Error_Message = JLabel( "")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
self.panel0.add( self.Error_Message )
self.Label_1 = JLabel("Volatility Executable Directory")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
self.panel0.add( self.Label_1 )
self.Program_Executable_TF = JTextField(10)
self.Program_Executable_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Program_Executable_TF, self.gbcPanel0 )
self.panel0.add( self.Program_Executable_TF )
self.Find_Program_Exec_BTN = JButton( "Find Dir", actionPerformed=self.Find_Dir)
self.Find_Program_Exec_BTN.setEnabled(True)
self.rbgPanel0.add( self.Find_Program_Exec_BTN )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Find_Program_Exec_BTN, self.gbcPanel0 )
self.panel0.add( self.Find_Program_Exec_BTN )
self.Blank_1 = JLabel( " ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Blank_3 = JLabel( " ")
self.Blank_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_3, self.gbcPanel0 )
self.panel0.add( self.Blank_3 )
self.Check_Box = JCheckBox("Extract and Create Memory Image from Hiberfile", actionPerformed=self.checkBoxEvent)
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Check_Box, self.gbcPanel0 )
self.panel0.add( self.Check_Box )
self.Blank_4 = JLabel( " ")
self.Blank_4.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_4, self.gbcPanel0 )
self.panel0.add( self.Blank_4 )
self.Label_3 = JLabel( "Message:")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_3, self.gbcPanel0 )
self.panel0.add( self.Label_3 )
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
self.Check_Box.setSelected(self.local_settings.getSetting('Flag') == 'true')
self.Program_Executable_TF.setText(self.local_settings.getSetting('Volatility_Directory'))
#pass
# Return the settings used
def getSettings(self):
return self.local_settings
class Plaso_ImportSettingsWithUISettingsPanel(
IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Exclude_File_Sources_CB.isSelected():
self.local_settings.setSetting('Exclude_File_Sources', 'true')
else:
self.local_settings.setSetting('Exclude_File_Sources', 'false')
# When button to find file is clicked then open dialog to find the file and return it.
def Find_Plaso_Dir(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("All", ["*.*"])
chooseFile.addChoosableFileFilter(filter)
chooseFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY)
ret = chooseFile.showDialog(self.panel0, "Find Plaso Directory")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setSetting('Plaso_Directory', Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
def Find_Plaso_File(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("All", ["*.*"])
chooseFile.addChoosableFileFilter(filter)
ret = chooseFile.showDialog(self.panel0, "Find Plaso Storage File")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setSetting('Plaso_Storage_File',
Canonical_file)
self.Plaso_Storage_File_TF.setText(Canonical_file)
def setPlasoDirectory(self, event):
self.local_settings.setSetting('Plaso_Directory',
self.Program_Executable_TF.getText())
def setPlasoStorageFile(self, event):
self.local_settings.setSetting('Plaso_Storage_File',
self.Plaso_Storage_File_TF.getText())
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout(self.gbPanel0)
self.Label_1 = JLabel("Plaso Executable Directory")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_1, self.gbcPanel0)
self.panel0.add(self.Label_1)
self.Program_Executable_TF = JTextField(
20, focusLost=self.setPlasoDirectory)
self.Program_Executable_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Program_Executable_TF,
self.gbcPanel0)
self.panel0.add(self.Program_Executable_TF)
self.Find_Program_Exec_BTN = JButton(
"Find Dir", actionPerformed=self.Find_Plaso_Dir)
self.Find_Program_Exec_BTN.setEnabled(True)
self.rbgPanel0.add(self.Find_Program_Exec_BTN)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Find_Program_Exec_BTN,
self.gbcPanel0)
self.panel0.add(self.Find_Program_Exec_BTN)
self.Blank_1 = JLabel(" ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_1, self.gbcPanel0)
self.panel0.add(self.Blank_1)
self.Label_1 = JLabel("Plaso Storage File")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_1, self.gbcPanel0)
self.panel0.add(self.Label_1)
self.Plaso_Storage_File_TF = JTextField(
20, focusLost=self.setPlasoStorageFile)
self.Plaso_Storage_File_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Plaso_Storage_File_TF,
self.gbcPanel0)
self.panel0.add(self.Plaso_Storage_File_TF)
self.Find_Storage_BTN = JButton("Find Storage File",
actionPerformed=self.Find_Plaso_File)
self.Find_Storage_BTN.setEnabled(True)
self.rbgPanel0.add(self.Find_Storage_BTN)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Find_Storage_BTN, self.gbcPanel0)
self.panel0.add(self.Find_Storage_BTN)
self.Blank_3 = JLabel(" ")
self.Blank_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_3, self.gbcPanel0)
self.panel0.add(self.Blank_3)
self.Exclude_File_Sources_CB = JCheckBox(
"Exclude File Source", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Exclude_File_Sources_CB,
self.gbcPanel0)
self.panel0.add(self.Exclude_File_Sources_CB)
self.Blank_4 = JLabel(" ")
self.Blank_4.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 17
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Blank_4, self.gbcPanel0)
self.panel0.add(self.Blank_4)
self.Label_3 = JLabel("Message:")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Label_3, self.gbcPanel0)
self.panel0.add(self.Label_3)
self.Error_Message = JLabel("")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 23
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.Error_Message, self.gbcPanel0)
self.panel0.add(self.Error_Message)
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
self.Exclude_File_Sources_CB.setSelected(
self.local_settings.getSetting('Exclude_File_Sources') == 'true')
self.Program_Executable_TF.setText(
self.local_settings.getSetting('Plaso_Directory'))
self.Plaso_Storage_File_TF.setText(
self.local_settings.getSetting('Plaso_Storage_File'))
# Return the settings used
def getSettings(self):
return self.local_settings
class VolatilitySettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
_logger = Logger.getLogger(VolatilityIngestModuleFactory.moduleName)
def log(self, level, msg):
self._logger.logp(level, self.__class__.__name__, inspect.stack()[1][3], msg)
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Exclude_File_Sources_CB.isSelected():
self.local_settings.setSetting('Exclude_File_Sources', 'true')
else:
self.local_settings.setSetting('Exclude_File_Sources', 'false')
def get_plugins(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = os.path.join(head, "GUI_Settings.db3")
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
self.Error_Message.setText("Database opened")
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = "select plugin_name from plugins where volatility_version = '" + self.Version_CB.getSelectedItem() + "';"
resultSet = stmt.executeQuery(SQL_Statement)
plugin_list = []
while resultSet.next():
plugin_list.append(resultSet.getString("plugin_name"))
stmt.close()
dbConn.close()
return plugin_list
except SQLException as e:
self.Error_Message.setText("Error Reading plugins")
stmt.close()
dbConn.close()
return "Error"
def get_profiles(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = os.path.join(head, "GUI_Settings.db3")
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
self.Error_Message.setText("Database opened")
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = "select profile_name from profiles where volatility_version = '" + self.Version_CB.getSelectedItem() + "';"
resultSet = stmt.executeQuery(SQL_Statement)
profile_list = []
while resultSet.next():
profile_list.append(resultSet.getString("profile_name"))
stmt.close()
dbConn.close()
return profile_list
except SQLException as e:
self.Error_Message.setText("Error Reading plugins")
stmt.close()
dbConn.close()
return "Error"
# When button to find file is clicked then open dialog to find the file and return it.
def Find_Dir(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("All", ["*.*"])
chooseFile.addChoosableFileFilter(filter)
#chooseFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY)
ret = chooseFile.showDialog(self.panel0, "Find Volatility Directory")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setSetting('Volatility_Directory', Canonical_file)
self.Program_Executable_TF.setText(Canonical_file)
def keyPressed(self, event):
self.local_settings.setSetting('AdditionalParms', self.Additional_Parms_TF.getText())
#self.Error_Message.setText(self.Additional_Parms_TF.getText())
def onchange_version(self, event):
self.local_settings.setSetting('Version', event.item)
plugin_list = self.get_plugins()
profile_list = self.get_profiles()
self.Profile_CB.removeAllItems()
self.Plugin_LB.clearSelection()
self.Plugin_LB.setListData(plugin_list)
for profile in profile_list:
self.Profile_CB.addItem(profile)
#self.Profile_CB.addItems(profile)
self.panel0.repaint()
def onchange_plugins_lb(self, event):
self.local_settings.setSetting('PluginListBox' , '')
list_selected = self.Plugin_LB.getSelectedValuesList()
self.local_settings.setSetting('PluginListBox', str(list_selected))
def onchange_profile_cb(self, event):
self.local_settings.setSetting('Profile', event.item)
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.Error_Message = JLabel( "")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 31
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
self.panel0.add( self.Error_Message )
self.Label_1 = JLabel("Volatility Executable Directory")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
self.panel0.add( self.Label_1 )
self.Program_Executable_TF = JTextField(10)
self.Program_Executable_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Program_Executable_TF, self.gbcPanel0 )
self.panel0.add( self.Program_Executable_TF )
self.Find_Program_Exec_BTN = JButton( "Find Dir", actionPerformed=self.Find_Dir)
self.Find_Program_Exec_BTN.setEnabled(True)
self.rbgPanel0.add( self.Find_Program_Exec_BTN )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Find_Program_Exec_BTN, self.gbcPanel0 )
self.panel0.add( self.Find_Program_Exec_BTN )
self.Blank_1 = JLabel( " ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Version_Label_1 = JLabel( "Version:")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Version_Label_1, self.gbcPanel0 )
self.panel0.add( self.Version_Label_1 )
self.Version_List = ("2.5", "2.6")
self.Version_CB = JComboBox( self.Version_List)
self.Version_CB.itemStateChanged = self.onchange_version
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Version_CB, self.gbcPanel0 )
self.panel0.add( self.Version_CB )
self.Blank_3 = JLabel( " ")
self.Blank_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_3, self.gbcPanel0 )
self.panel0.add( self.Blank_3 )
self.Plugin_Label_1 = JLabel( "Plugins:")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Plugin_Label_1, self.gbcPanel0 )
self.panel0.add( self.Plugin_Label_1 )
self.Plugin_list = self.get_plugins()
self.Plugin_LB = JList( self.Plugin_list, valueChanged=self.onchange_plugins_lb)
self.Plugin_LB.setVisibleRowCount( 3 )
self.scpPlugin_LB = JScrollPane( self.Plugin_LB )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 1
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.scpPlugin_LB, self.gbcPanel0 )
self.panel0.add( self.scpPlugin_LB )
self.Blank_4 = JLabel( " ")
self.Blank_4.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 17
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_4, self.gbcPanel0 )
self.panel0.add( self.Blank_4 )
self.Profile_Label_1 = JLabel( "Profile:")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Profile_Label_1, self.gbcPanel0 )
self.panel0.add( self.Profile_Label_1 )
self.Profile_List = self.get_profiles()
self.Profile_CB = JComboBox( self.Profile_List)
self.Profile_CB.itemStateChanged = self.onchange_profile_cb
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 1
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Profile_CB, self.gbcPanel0 )
self.panel0.add( self.Profile_CB )
self.Blank_5 = JLabel( " ")
self.Blank_5.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 21
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_5, self.gbcPanel0 )
self.panel0.add( self.Blank_5 )
self.Label_2 = JLabel("Additional Parameters To Run With:")
self.Label_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 23
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_2, self.gbcPanel0 )
self.panel0.add( self.Label_2 )
self.Additional_Parms_TF = JTextField(10,focusLost=self.keyPressed)
#self.Additional_Parms_TF.getDocument().addDocumentListener()
self.Additional_Parms_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 25
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Additional_Parms_TF, self.gbcPanel0 )
self.panel0.add( self.Additional_Parms_TF )
self.Blank_6 = JLabel( " ")
self.Blank_6.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 27
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_6, self.gbcPanel0 )
self.panel0.add( self.Blank_6 )
self.Label_3 = JLabel( "Message:")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 29
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_3, self.gbcPanel0 )
self.panel0.add( self.Label_3 )
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
self.Program_Executable_TF.setText(self.local_settings.getSetting('Volatility_Directory'))
#pass
# Return the settings used
def getSettings(self):
return self.local_settings
class HashImageSettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
def FindFTKTxtFile(self, e):
chooseFile = JFileChooser()
filter = FileNameExtensionFilter("ALL", ["*.*"])
chooseFile.addChoosableFileFilter(filter)
ret = chooseFile.showDialog(self.panel0, "Find FTK Log File")
if ret == JFileChooser.APPROVE_OPTION:
file = chooseFile.getSelectedFile()
Canonical_file = file.getCanonicalPath()
#text = self.readPath(file)
self.local_settings.setSetting('FTKLogFile', Canonical_file)
setSetting('FTKLogFile', Canonical_file)
self.FTKLogFile_TF.setText(Canonical_file)
def keyPressedMD5(self, event):
self.local_settings.setSetting('MD5Hash', self.MD5HashValue_TF.getText())
def keyPressedSHA1(self, event):
self.local_settings.setSetting('SHA1Hash', self.SHA1HashValue_TF.getText())
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.Label_1 = JLabel("MD5 Hash Value To Verify")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
self.panel0.add( self.Label_1 )
self.MD5HashValue_TF = JTextField(20, focusLost=self.keyPressedMD5)
self.MD5HashValue_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.MD5HashValue_TF, self.gbcPanel0 )
self.panel0.add( self.MD5HashValue_TF )
self.Blank_1 = JLabel( " ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Label_2 = JLabel("SHA1 Hash Value To Verify")
self.Label_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_2, self.gbcPanel0 )
self.panel0.add( self.Label_2 )
self.SHA1HashValue_TF = JTextField(20, focusLost=self.keyPressedSHA1)
self.SHA1HashValue_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.SHA1HashValue_TF, self.gbcPanel0 )
self.panel0.add( self.SHA1HashValue_TF )
self.Blank_2 = JLabel( " ")
self.Blank_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Label_3 = JLabel("FTK Log File")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_3, self.gbcPanel0 )
self.panel0.add( self.Label_3 )
self.FTKLogFile_TF = JTextField(20)
self.FTKLogFile_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.FTKLogFile_TF, self.gbcPanel0 )
self.panel0.add( self.FTKLogFile_TF )
self.FTKLogFile_BTN = JButton( "Find File", actionPerformed=self.FindFTKTxtFile)
self.FTKLogFile_BTN.setEnabled(True)
self.rbgPanel0.add( self.FTKLogFile_BTN )
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.FTKLogFile_BTN, self.gbcPanel0 )
self.panel0.add( self.FTKLogFile_BTN )
self.Label_4 = JLabel( "Message:")
self.Label_4.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 29
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_4, self.gbcPanel0 )
self.panel0.add( self.Label_4 )
self.Error_Message = JLabel( "")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 31
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
self.panel0.add( self.Error_Message )
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
pass
# Return the settings used
def getSettings(self):
self.local_settings.setSetting('MD5Hash', self.MD5HashValue_TF.getText())
self.local_settings.setSetting('SHA1Hash', self.SHA1HashValue_TF.getText())
return self.local_settings
class initDialog(JDialog): # JFrame
def __init__(self, pP):
# fields
self.tPath = ""
self.cPath = ""
self.exPath = ""
self.picPicker = pP
self.annotationType = self.picPicker.getAnnotationType()
super(initDialog, self).__init__()
self.initUI()
def initUI(self):
inputPanel = JPanel()
inputPanel.setBorder(BorderFactory.createTitledBorder("Where are your control and treament images?"))
inputLayout = GroupLayout(inputPanel, autoCreateContainerGaps=True, autoCreateGaps=True)
inputPanel.setLayout(inputLayout)
annotatePanel = JPanel()
annotatePanel.setBorder(BorderFactory.createTitledBorder("How do you want to annotate your data?"))
anLayout = GroupLayout(annotatePanel, autoCreateContainerGaps=True, autoCreateGaps=True)
annotatePanel.setLayout(anLayout)
exportPanel = JPanel()
exportPanel.setBorder(BorderFactory.createTitledBorder("Where do you want to export your data"))
exportLayout = GroupLayout(exportPanel, autoCreateContainerGaps=True, autoCreateGaps=True)
exportPanel.setLayout(exportLayout)
btnPanel = JPanel()
btnLayout = GroupLayout(btnPanel, autoCreateContainerGaps=True, autoCreateGaps=True)
btnPanel.setLayout(btnLayout)
layout = GroupLayout(self.getContentPane(), autoCreateContainerGaps=True, autoCreateGaps=True)
self.getContentPane().setLayout(layout)
self.setModalityType(ModalityType.APPLICATION_MODAL)
self.setDefaultCloseOperation(JDialog.DISPOSE_ON_CLOSE)# JFrame.EXIT_ON_CLOSE)
# definition of elements
# labels
cPathLabel = JLabel("Control Path:")
tPathLabel = JLabel("Treatment Path:")
exPathLabel = JLabel("Save Results in:")
#textfields
self.cPathField = JTextField("/Users/schiklen/DotData/131118_dummy/131118_2926/cutout/", 16)
self.tPathField = JTextField("/Users/schiklen/DotData/131118_dummy/131118_2926/cutout/", 16)
self.exPathField = JTextField("/Users/schiklen/DotData/131118_dummy/131118_2926/cutout/", 16)
#Radiobuttons
yesNoRButton = JRadioButton("Yes / No / Ignore", selected=True, actionCommand="yesNoIgnore", actionPerformed=self.setAnnotationTypeDialog)
intRButton = JRadioButton("Integer", actionCommand="int", actionPerformed=self.setAnnotationTypeDialog)
nRButton = JRadioButton("Number", actionCommand="float", actionPerformed=self.setAnnotationTypeDialog)
listRButton = JRadioButton("From List...", actionCommand="list", actionPerformed=self.openListDialog)
self.rBGroup = ButtonGroup()
self.rBGroup.add(yesNoRButton)
self.rBGroup.add(intRButton)
self.rBGroup.add(nRButton)
self.rBGroup.add(listRButton)
#self.customListButton = JButton("Custom List...", actionPerformed=self.makeCustomList, enabled=0)
#buttons
cPathButton = JButton("Browse...", actionPerformed=self.browseC) # lambda on fieldvalue
tPathButton = JButton("Browse...", actionPerformed=self.browseT) # lambda on fieldvalue
exPathButton = JButton("Browse...", actionPerformed=self.browseE)
OKButton = JButton("OK", actionPerformed=self.okayEvent)
CancelButton = JButton("Cancel", actionPerformed=self.cancel)
'''General ContentPane Layout'''
layout.setHorizontalGroup(layout.createParallelGroup()
.addComponent(inputPanel)
.addComponent(annotatePanel)
.addComponent(exportPanel)
.addComponent(btnPanel)
)
layout.linkSize(SwingConstants.HORIZONTAL, inputPanel, annotatePanel, exportPanel, btnPanel)
layout.setVerticalGroup(layout.createSequentialGroup()
.addComponent(inputPanel)
.addComponent(annotatePanel)
.addComponent(exportPanel)
.addComponent(btnPanel)
)
''' Input panel Layout '''
inputLayout.setHorizontalGroup(inputLayout.createSequentialGroup()
.addGroup(inputLayout.createParallelGroup(GroupLayout.Alignment.TRAILING)
.addComponent(cPathLabel)
.addComponent(tPathLabel))
.addGroup(inputLayout.createParallelGroup(GroupLayout.Alignment.TRAILING)
.addComponent(self.cPathField)
.addComponent(self.tPathField))
.addGroup(inputLayout.createParallelGroup()
.addComponent(cPathButton)
.addComponent(tPathButton))
)
inputLayout.setVerticalGroup(inputLayout.createSequentialGroup()
.addGroup(inputLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(cPathLabel)
.addComponent(self.cPathField)
.addComponent(cPathButton))
.addGroup(inputLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(tPathLabel)
.addComponent(self.tPathField)
.addComponent(tPathButton))
)
'''Annotate panel layout'''
anLayout.setHorizontalGroup(anLayout.createParallelGroup()
.addComponent(yesNoRButton)
.addComponent(intRButton)
.addComponent(nRButton)
.addComponent(listRButton)
#.addComponent(self.customListButton)
)
anLayout.setVerticalGroup(anLayout.createSequentialGroup()
.addComponent(yesNoRButton)
.addComponent(intRButton)
.addComponent(nRButton)
.addComponent(listRButton)
#.addComponent(self.customListButton)
)
'''Export panel layout'''
exportLayout.setHorizontalGroup(exportLayout.createSequentialGroup()
.addComponent(exPathLabel)
.addComponent(self.exPathField)
.addComponent(exPathButton))
exportLayout.setVerticalGroup(exportLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(exPathLabel)
.addComponent(self.exPathField)
.addComponent(exPathButton))
'''Buttons Panel Layout'''
btnLayout.setHorizontalGroup(btnLayout.createSequentialGroup()
.addComponent(CancelButton)
.addComponent(OKButton)
)
btnLayout.setVerticalGroup(btnLayout.createParallelGroup(GroupLayout.Alignment.BASELINE)
.addComponent(CancelButton)
.addComponent(OKButton)
)
self.setTitle("Random Picture Picker")
self.pack()
self.setLocationRelativeTo(None)
self.setVisible(True)
def addDirToList(self, p):
self.dirPathList.append(p)
def browseT(self, event):
self.tPathField.text = DirectoryChooser("Select Treatment Folder").getDirectory()
def browseC(self, event):
self.cPathField.text = DirectoryChooser("Select Control Folder").getDirectory()
def browseE(self, event):
self.exPathField.text = DirectoryChooser("Select Export Folder").getDirectory()
self.picPicker.setOutputPath(self.exPathField.text)
def makeCustomList(self):
ld = listDialog(self)
ld.startUI()
def cancel(self, event):
exit(0)
def okayEvent(self, event):
inputDict = {self.tPathField.getText():"treatment", self.cPathField.getText():"control"} # this is for later extension
self.picPicker.setInputPathDict(inputDict)
self.picPicker.setOutputPath(self.exPathField.getText())
self.dispose()
def getPicPicker(self):
return self.picPicker
def setAnnotationTypeDialog(self, event):
annotype = self.rBGroup.getSelection().getActionCommand()
if annotype == "int" or "float":
self.picPicker.setAnnotationType(["text"])
if annotype == "yesNoIgnore":
self.picPicker.setAnnotationType(["Yes", "No", "Ignore"])
def openListDialog(self,event):
self.makeCustomList()
#self.customListButton.setEnabled(True)
class AmcacheScanWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Private_API_Key_CB.isSelected():
self.local_settings.setPrivate(True)
self.local_settings.setAPI_Key(self.API_Key_TF.getText())
else:
self.local_settings.setPrivate(False)
self.local_settings.setAPI_Key(self.API_Key_TF.getText())
# Check to see if there are any entries that need to be populated from the database.
def check_Database_entries(self):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings DB!")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'Select Setting_Name, Setting_Value from settings;'
resultSet = stmt.executeQuery(SQL_Statement)
while resultSet.next():
if resultSet.getString("Setting_Name") == "API_Key":
self.local_settings.setAPI_Key(resultSet.getString("Setting_Value"))
self.API_Key_TF.setText(resultSet.getString("Setting_Value"))
if resultSet.getString("Setting_Name") == "Private":
private = resultSet.getString("Setting_Value")
if private == "1":
self.local_settings.setPrivate(True)
else:
self.local_settings.setPrivate(False)
self.Error_Message.setText("Settings Read successfully!")
except SQLException as e:
self.Error_Message.setText("Error Reading Settings Database")
stmt.close()
dbConn.close()
# Save entries from the GUI to the database.
def SaveSettings(self, e):
head, tail = os.path.split(os.path.abspath(__file__))
settings_db = head + "\\GUI_Settings.db3"
try:
Class.forName("org.sqlite.JDBC").newInstance()
dbConn = DriverManager.getConnection("jdbc:sqlite:%s" % settings_db)
except SQLException as e:
self.Error_Message.setText("Error Opening Settings")
try:
stmt = dbConn.createStatement()
SQL_Statement = 'UPDATE settings SET Setting_Value = "' + self.API_Key_TF.getText() + '" WHERE Setting_Name = "API_Key";'
resultSet = stmt.executeQuery(SQL_Statement)
except:
pass
try:
if self.local_settings.getPrivate():
SQL_Statement = 'UPDATE settings SET Setting_Value = "1" WHERE Setting_Name = "Private";'
resultSet = stmt.executeQuery(SQL_Statement)
else:
SQL_Statement = 'UPDATE settings SET Setting_Value = "0" WHERE Setting_Name = "Private";'
resultSet = stmt.executeQuery(SQL_Statement)
except:
pass
self.Error_Message.setText("Settings Saved")
stmt.close()
dbConn.close()
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.LabelA = JLabel("VirusTotal API Key:")
self.LabelA.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.LabelA, self.gbcPanel0 )
self.panel0.add( self.LabelA )
self.API_Key_TF = JTextField(20)
self.API_Key_TF.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.API_Key_TF, self.gbcPanel0 )
self.panel0.add( self.API_Key_TF )
self.Blank_1 = JLabel( " ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 6
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Private_API_Key_CB = JCheckBox("Private API Key?", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Private_API_Key_CB, self.gbcPanel0 )
self.panel0.add( self.Private_API_Key_CB )
self.Save_Settings_BTN = JButton( "Save Settings", actionPerformed=self.SaveSettings)
self.Save_Settings_BTN.setEnabled(True)
self.rbgPanel0.add( self.Save_Settings_BTN )
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 8
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Save_Settings_BTN, self.gbcPanel0 )
self.panel0.add( self.Save_Settings_BTN )
self.Label_1 = JLabel( "Error Message:")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
self.panel0.add( self.Label_1 )
self.Error_Message = JLabel( "")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 6
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
self.panel0.add( self.Error_Message )
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
self.check_Database_entries()
self.Private_API_Key_CB.setSelected(self.local_settings.getPrivate())
# Return the settings used
def getSettings(self):
return self.local_settings
class CuckooSettingsWithUISettingsPanel(IngestModuleIngestJobSettingsPanel):
# Note, we can't use a self.settings instance variable.
# Rather, self.local_settings is used.
# https://wiki.python.org/jython/UserGuide#javabean-properties
# Jython Introspector generates a property - 'settings' on the basis
# of getSettings() defined in this class. Since only getter function
# is present, it creates a read-only 'settings' property. This auto-
# generated read-only property overshadows the instance-variable -
# 'settings'
# We get passed in a previous version of the settings so that we can
# prepopulate the UI
# TODO: Update this for your UI
def __init__(self, settings):
self.local_settings = settings
self.tag_list = []
self.initComponents()
self.customizeComponents()
self.path_to_cuckoo_exe = os.path.join(os.path.dirname(os.path.abspath(__file__)), "cuckoo_api.exe")
# Check the checkboxs to see what actions need to be taken
def checkBoxEvent(self, event):
if self.Submit_File_CB.isSelected():
self.local_settings.setSetting('Submit_File', 'true')
self.local_settings.setSetting('Submit_URL', 'false')
else:
self.local_settings.setSetting('Submit_File', 'false')
def onchange_lb(self, event):
self.local_settings.cleartag_list()
list_selected = self.List_Box_LB.getSelectedValuesList()
self.local_settings.setSetting('tag_list', str(list_selected))
def find_tags(self):
sql_statement = "SELECT distinct(display_name) u_tag_name FROM content_tags INNER JOIN tag_names ON " + \
" content_tags.tag_name_id = tag_names.tag_name_id;"
skCase = Case.getCurrentCase().getSleuthkitCase()
dbquery = skCase.executeQuery(sql_statement)
resultSet = dbquery.getResultSet()
while resultSet.next():
self.tag_list.append(resultSet.getString("u_tag_name"))
dbquery.close()
# Check to see if the Cuckoo server is available and you can talk to it
def Check_Server(self, e):
pipe = Popen([self.path_to_cuckoo_exe, self.Protocol_TF.getText(),self.IP_Address_TF.getText(), self.Port_Number_TF.getText(), "cuckoo_status" ], stdout=PIPE, stderr=PIPE)
out_text = pipe.communicate()[0]
self.Error_Message.setText("Cuckoo Status is " + out_text)
#self.log(Level.INFO, "Cuckoo Status is ==> " + out_text)
def setIPAddress(self, event):
self.local_settings.setSetting('IP_Address', self.IP_Address_TF.getText())
def setProtocol(self, event):
self.local_settings.setSetting('Protocol', self.Protocol_TF.getText())
def setPortNumber(self, event):
self.local_settings.setSetting('Port_Number', self.Port_Number_TF.getText())
# Create the initial data fields/layout in the UI
def initComponents(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout( self.gbPanel0 )
self.Label_1 = JLabel("Protocol:")
self.Label_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_1, self.gbcPanel0 )
self.panel0.add( self.Label_1 )
self.Protocol_TF = JTextField(20, focusLost=self.setProtocol)
self.Protocol_TF.setEnabled(True)
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Protocol_TF, self.gbcPanel0 )
self.panel0.add( self.Protocol_TF )
self.Blank_1 = JLabel( " ")
self.Blank_1.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_1, self.gbcPanel0 )
self.panel0.add( self.Blank_1 )
self.Label_2 = JLabel("Cuckoo IP Address")
self.Label_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_2, self.gbcPanel0 )
self.panel0.add( self.Label_2 )
self.IP_Address_TF = JTextField(20, focusLost=self.setIPAddress)
self.IP_Address_TF.setEnabled(True)
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.IP_Address_TF, self.gbcPanel0 )
self.panel0.add( self.IP_Address_TF )
self.Blank_2 = JLabel( " ")
self.Blank_2.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_2, self.gbcPanel0 )
self.panel0.add( self.Blank_2 )
self.Label_3 = JLabel("Port Number")
self.Label_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Label_3, self.gbcPanel0 )
self.panel0.add( self.Label_3 )
self.Port_Number_TF = JTextField(20, focusLost=self.setPortNumber)
self.Port_Number_TF.setEnabled(True)
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Port_Number_TF, self.gbcPanel0 )
self.panel0.add( self.Port_Number_TF )
self.Blank_3 = JLabel( " ")
self.Blank_3.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_3, self.gbcPanel0 )
self.panel0.add( self.Blank_3 )
self.Blank_5 = JLabel( "Tag to Choose: ")
self.Blank_5.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_5, self.gbcPanel0 )
self.panel0.add( self.Blank_5 )
self.find_tags()
self.List_Box_LB = JList( self.tag_list, valueChanged=self.onchange_lb)
self.List_Box_LB.setVisibleRowCount( 3 )
self.scpList_Box_LB = JScrollPane( self.List_Box_LB )
self.gbcPanel0.gridx = 4
self.gbcPanel0.gridy = 13
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 1
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.scpList_Box_LB, self.gbcPanel0 )
self.panel0.add( self.scpList_Box_LB )
self.Blank_6 = JLabel( " ")
self.Blank_6.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Blank_6, self.gbcPanel0 )
self.panel0.add( self.Blank_6 )
self.Submit_File_CB = JCheckBox("Submit a File", actionPerformed=self.checkBoxEvent)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 17
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Submit_File_CB, self.gbcPanel0 )
self.panel0.add( self.Submit_File_CB )
# self.Submit_URL_CB = JCheckBox("Submit a URL", actionPerformed=self.checkBoxEvent)
# self.gbcPanel0.gridx = 2
# self.gbcPanel0.gridy = 23
# self.gbcPanel0.gridwidth = 1
# self.gbcPanel0.gridheight = 1
# self.gbcPanel0.fill = GridBagConstraints.BOTH
# self.gbcPanel0.weightx = 1
# self.gbcPanel0.weighty = 0
# self.gbcPanel0.anchor = GridBagConstraints.NORTH
# self.gbPanel0.setConstraints( self.Submit_URL_CB, self.gbcPanel0 )
# self.panel0.add( self.Submit_URL_CB )
self.Check_Server_Status_BTN = JButton( "Check Server Status", actionPerformed=self.Check_Server)
self.Check_Server_Status_BTN.setEnabled(True)
self.rbgPanel0.add( self.Check_Server_Status_BTN )
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Check_Server_Status_BTN, self.gbcPanel0 )
self.panel0.add( self.Check_Server_Status_BTN )
self.Error_Message = JLabel( "")
self.Error_Message.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 21
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints( self.Error_Message, self.gbcPanel0 )
self.panel0.add( self.Error_Message )
self.add(self.panel0)
# Custom load any data field and initialize the values
def customizeComponents(self):
#self.Exclude_File_Sources_CB.setSelected(self.local_settings.getExclude_File_Sources())
#self.Run_Cuckoo_CB.setSelected(self.local_settings.getSetting('Submit_File') == 'true')
#self.Import_Cuckoo_CB.setSelected(self.local_settings.getImport_Cuckoo())
self.Port_Number_TF.setText(self.local_settings.getSetting('Port_Number'))
self.IP_Address_TF.setText(self.local_settings.getSetting('IP_Address'))
self.Protocol_TF.setText(self.local_settings.getSetting('Protocol'))
# Return the settings used
def getSettings(self):
return self.local_settings
class MenueFrame(object):
def __init__(self):
self.mainDir = ""
self.frame = JFrame("Dots Quality Check", size=(250,300))
self.frame.setLocation(20,120)
self.Panel = JPanel(GridLayout(0,1))
self.frame.add(self.Panel)
self.openNextButton = JButton('Open Next Random',actionPerformed=openRandom)
self.Panel.add(self.openNextButton)
self.saveButton = JButton('Save',actionPerformed=save)
self.saveButton.setEnabled(False)
self.Panel.add(self.saveButton)
self.cropButton = JButton('Crop values from here', actionPerformed=cropVals)
self.Panel.add(self.cropButton)
self.DiscardButton = JButton('Discard cell', actionPerformed=discardCell)
self.DiscardButton.setEnabled(True)
self.Panel.add(self.DiscardButton)
self.quitButton = JButton('Quit script',actionPerformed=quit)
self.Panel.add(self.quitButton)
annoPanel = JPanel()
#add gridlayout
wtRButton = JRadioButton("wt", actionCommand="wt")
defectRButton = JRadioButton("Defect", actionCommand="defect")
annoPanel.add(wtRButton)
annoPanel.add(defectRButton)
self.aButtonGroup = ButtonGroup()
self.aButtonGroup.add(wtRButton)
self.aButtonGroup.add(defectRButton)
self.Panel.add(annoPanel)
self.ProgBar = JProgressBar()
self.ProgBar.setStringPainted(True)
self.ProgBar.setValue(0)
self.Panel.add(self.ProgBar)
self.pathLabel = JLabel("-- No main directory chosen --")
self.pathLabel.setHorizontalAlignment( SwingConstants.CENTER )
self.Panel.add(self.pathLabel)
WindowManager.addWindow(self.frame)
self.show()
def show(self):
self.frame.visible = True
def getFrame(self):
return self.frame
def setSaveActive(self):
self.saveButton.setEnabled(True)
self.show()
def setSaveInactive(self):
self.saveButton.setEnabled(False)
self.show()
def setMainDir(self, path):
self.mainDir = path
self.pathLabel.setText("MainDir: " + os.path.basename(os.path.split(self.mainDir)[0]))
def getMainDir(self):
return self.mainDir
def setProgBarMax(self, maximum):
self.ProgBar.setMaximum(maximum)
def setProgBarVal(self, value):
self.ProgBar.setValue(value)
def close():
WindowManager.removeWindow(self.frame)
self.frame.dispose()
class MenueFrame(JFrame, ActionListener, WindowFocusListener): # should extend JFrame
def __init__(self):
self.mainDir = ""
self.setTitle("Dots Quality Check")
self.setSize(250, 300)
self.setLocation(20,120)
self.addWindowFocusListener(self)
self.Panel = JPanel(GridLayout(0,1))
self.add(self.Panel)
self.openNextButton = JButton("Open Next Random", actionPerformed=self.openRandom)
self.Panel.add(self.openNextButton)
self.saveButton = JButton("Save", actionPerformed=self.save, enabled=False)
self.Panel.add(self.saveButton)
self.cropButton = JButton("Crop values from here", actionPerformed=self.cropVals)
self.Panel.add(self.cropButton)
self.DiscardButton = JButton("Discard cell", actionPerformed=self.discardCell)
self.Panel.add(self.DiscardButton)
self.quitButton = JButton("Quit script",actionPerformed=self.quit)
self.Panel.add(self.quitButton)
annoPanel = JPanel()
#add gridlayout
self.wtRButton = JRadioButton("wt", actionCommand="wt")
self.wtRButton.addActionListener(self)
self.defectRButton = JRadioButton("Defect", actionCommand="defect")
self.defectRButton.addActionListener(self)
annoPanel.add(self.wtRButton)
annoPanel.add(self.defectRButton)
self.aButtonGroup = ButtonGroup()
self.aButtonGroup.add(self.wtRButton)
self.aButtonGroup.add(self.defectRButton)
self.Panel.add(annoPanel)
self.ProgBar = JProgressBar()
self.ProgBar.setStringPainted(True)
self.ProgBar.setValue(0)
self.Panel.add(self.ProgBar)
self.pathLabel = JLabel("-- No main directory chosen --")
self.pathLabel.setHorizontalAlignment( SwingConstants.CENTER )
self.Panel.add(self.pathLabel)
WindowManager.addWindow(self)
self.show()
# - - - - B U T T O N M E T H O D S - - - -
# - - - - - - - - - - - - - - - - - - - - - - -
def openRandom(self, event): # when click here: get random cell and meas.measure(csv, tif, savePath)
if self.mainDir == "":
self.mainDir = DirectoryChooser("Random QC - Please choose main directory containing ctrl and test folders").getDirectory()
self.pathLabel.setText("MainDir: " + os.path.basename(os.path.split(self.mainDir)[0]))
try:
# should be complete disposal!
self.cT.closeWindows()
finally:
inFiles = glob.glob(os.path.join(self.mainDir, "*", G_OPENSUBDIR, "val_*.csv")) # glob.glob returns list of paths
uncheckedCells = [cell(csvPath) for csvPath in inFiles if cell(csvPath).processed == False]
if len(uncheckedCells) > 0:
self.cell = random.choice(uncheckedCells)
#update progressbar
self.ProgBar.setMaximum(len(inFiles)-1)
self.ProgBar.setValue(len(inFiles)-len(uncheckedCells))
# open imp and resultstable
self.cT = correctionTable(self.cell, self) #self, openPath_csv, mF
self.RBActionListener.setCell(self.cell)
# delete previous Radiobutton annotation
self.wtRButton.setSelected(False)
self.defectRButton.setSelected(True)
else:
print "All cells measured!"
def save(self, event):
savepath = self.cell.getQcCsvPath()
anaphase = self.cell.getAnOn()
timeInterval = self.cT.getImp().getCalibration().frameInterval
annotation = self.getAnnotation()
position = str(self.cell.position)
cellIndex = str(self.cell.cellNo)
if not os.path.exists(os.path.split(savepath)[0]): # check if save folder present.
os.makedirs(os.path.split(savepath)[0]) # create save folder, if not present
f = open(savepath, "w")
# Position Cell Phenotype Frame Time AnOn Distance ch0x ch0y ch0z ch0vol ch1x ch1y ch1z ch1vol
f.write("Position,Cell,Phenotype,Frame,Time,Anaphase,Distance,ch0x,ch0y,ch0z,ch0vol,ch1x,ch1y,ch1z,ch1vol\n")
for i in range(self.cT.getLineCount()):
frame, distance, a = self.cT.getLine(i).split("\t")
corrFrame = str(int(frame)-int(anaphase))
time = "%.f" % (round(timeInterval) * int(corrFrame))
if distance == "NA":
ch0x, ch0y, ch0z, ch0vol, ch1x, ch1y, ch1z, ch1vol = ("NA," * 7 + "NA\n").split(",")
else:
ch0x, ch0y, ch0z, ch0vol, ch1x, ch1y, ch1z, ch1vol = self.cT.getXYZtable()[i]
f.write(position+","+cellIndex+","+annotation+","+corrFrame+","+time+","+anaphase+","+distance+","+ch0x+","+ch0y+","+ch0z+","+ch0vol+","+ch1x+","+ch1y+","+ch1z+","+ch1vol)
f.close()
print "Successfully saved!"
def cropVals(self, event): #"this function deletes all values with frame > current cursor"
for line in range(self.cT.getSelectionEnd(), self.cT.getLineCount(), 1):
frame, distance, AOCol = self.cT.getLine(line).split("\t")
self.cT.setLine(line, frame + "\tNA" + "\t" + AOCol)
def discardCell(self, event):
if not os.path.exists(os.path.split(self.cell.getQcCsvPath() )[0]): # check if save folder present.
os.makedirs(os.path.split(self.cell.getQcCsvPath() )[0]) # create save folder, if not present.
f = open(self.cell.getQcCsvPath() ,"w")
# Write dummy header. Position Cell Phenotype Frame Time AnOn Distance ch0x ch0y ch0z ch0vol ch1x ch1y ch1z ch1vol
f.write("Position,Cell,Phenotype,Frame,Time,AnOn,Distance,ch0x,ch0y,ch0z,ch0vol,ch1x,ch1y,ch1z,ch1vol\n")
f.close()
print "Discarded cell - saved dummy"
def quit(self, event):
try:
self.cT.closeWindows()
finally:
WindowManager.removeWindow(self)
self.dispose()
# Methods implementing ActionListener interfaces:
def actionPerformed(self, e):
# this function is called when RadioButtons are changed
self.cell.annotate( e.getSource().getActionCommand() )
self.setSaveActive()
def windowGainedFocus(self, e):
pass
def windowLostFocus(self, e):
pass
# - - - - - - - - - - - - -
# - get and set methods - -
# - - - - - - - - - - - - -
def getAnnotation(self):
return self.aButtonGroup.getSelection().getActionCommand()
def getMainDir(self):
return self.mainDir
def setSaveActive(self):
if (self.cell.getAnnotation() != None and self.cell.getAnOn() != None):
self.saveButton.setEnabled(True)
self.show()
def setSaveInactive(self):
self.saveButton.setEnabled(False)
self.show()
def setMainDir(self, path):
self.mainDir = path
self.pathLabel.setText("MainDir: " + os.path.basename(os.path.split(self.mainDir)[0]))
def __init__(self, parent, title, app):
from javax.swing import JCheckBox, JRadioButton, ButtonGroup
self.app = app
border = BorderFactory.createEmptyBorder(5, 7, 5, 7)
self.getContentPane().setBorder(border)
self.getContentPane().setLayout(BorderLayout(0, 5))
self.tabbedPane = JTabbedPane()
#1 Tab: general
panel1 = JPanel()
panel1.setBorder(BorderFactory.createEmptyBorder(7, 7, 7, 7))
panel1.setLayout(BoxLayout(panel1, BoxLayout.PAGE_AXIS))
#Checkbutton to enable/disable update check when script starts
self.updateCBtn = JCheckBox(self.app.strings.getString("updateCBtn"))
self.updateCBtn.setToolTipText(self.app.strings.getString("updateCBtn_tooltip"))
#Download tools
downloadBtn = JButton(self.app.strings.getString("updatesBtn"),
ImageProvider.get("dialogs", "refresh"),
actionPerformed=self.on_downloadBtn_clicked)
downloadBtn.setToolTipText(self.app.strings.getString("updatesBtn_tooltip"))
#Checkbuttons for enabling/disabling tools
toolsPanel = JPanel(BorderLayout(0, 5))
title = self.app.strings.getString("enable_disable_tools")
toolsPanel.setBorder(BorderFactory.createTitledBorder(title))
infoLbl = JLabel(self.app.strings.getString("JOSM_restart_warning"))
infoLbl.setFont(infoLbl.getFont().deriveFont(Font.ITALIC))
toolsPanel.add(infoLbl, BorderLayout.PAGE_START)
toolsStatusPane = JPanel(GridLayout(len(self.app.realTools), 0))
self.toolsCBtns = []
for tool in self.app.realTools:
toolCBtn = JCheckBox()
toolCBtn.addItemListener(self)
toolLbl = JLabel(tool.title, tool.bigIcon, JLabel.LEFT)
self.toolsCBtns.append(toolCBtn)
toolPane = JPanel()
toolPane.setLayout(BoxLayout(toolPane, BoxLayout.X_AXIS))
toolPane.add(toolCBtn)
toolPane.add(toolLbl)
toolsStatusPane.add(toolPane)
toolsPanel.add(toolsStatusPane, BorderLayout.CENTER)
#Radiobuttons for enabling/disabling layers when a new one
#is added
layersPanel = JPanel(GridLayout(0, 1))
title = self.app.strings.getString("errors_layers_manager")
layersPanel.setBorder(BorderFactory.createTitledBorder(title))
errorLayersLbl = JLabel(self.app.strings.getString("errors_layers_info"))
errorLayersLbl.setFont(errorLayersLbl.getFont().deriveFont(Font.ITALIC))
layersPanel.add(errorLayersLbl)
self.layersRBtns = {}
group = ButtonGroup()
for mode in self.app.layersModes:
layerRBtn = JRadioButton(self.app.strings.getString("%s" % mode))
group.add(layerRBtn)
layersPanel.add(layerRBtn)
self.layersRBtns[mode] = layerRBtn
#Max number of errors text field
self.maxErrorsNumberTextField = JTextField()
self.maxErrorsNumberTextField.setToolTipText(self.app.strings.getString("maxErrorsNumberTextField_tooltip"))
self.maxErrorsNumberTFieldDefaultBorder = self.maxErrorsNumberTextField.getBorder()
self.maxErrorsNumberTextField.getDocument().addDocumentListener(ErrNumTextListener(self))
#layout
self.updateCBtn.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(self.updateCBtn)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
downloadBtn.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(downloadBtn)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
toolsPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(toolsPanel)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
layersPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(layersPanel)
panel1.add(Box.createRigidArea(Dimension(0, 15)))
maxErrP = JPanel(BorderLayout(5, 0))
maxErrP.add(JLabel(self.app.strings.getString("max_errors_number")), BorderLayout.LINE_START)
maxErrP.add(self.maxErrorsNumberTextField, BorderLayout.CENTER)
p = JPanel(BorderLayout())
p.add(maxErrP, BorderLayout.PAGE_START)
p.setAlignmentX(Component.LEFT_ALIGNMENT)
panel1.add(p)
self.tabbedPane.addTab(self.app.strings.getString("tab_1_title"),
None,
panel1,
None)
#2 Tab: favourite zones
panel2 = JPanel(BorderLayout(5, 15))
panel2.setBorder(BorderFactory.createEmptyBorder(7, 7, 7, 7))
#status
topPanel = JPanel()
topPanel.setLayout(BoxLayout(topPanel, BoxLayout.Y_AXIS))
infoPanel = HtmlPanel(self.app.strings.getString("fav_zones_info"))
infoPanel.getEditorPane().addHyperlinkListener(self)
infoPanel.setAlignmentX(Component.LEFT_ALIGNMENT)
self.favZoneStatusCBtn = JCheckBox(self.app.strings.getString("activate_fav_area"),
actionListener=self)
self.favZoneStatusCBtn.setToolTipText(self.app.strings.getString("activate_fav_area_tooltip"))
self.favZoneStatusCBtn.setAlignmentX(Component.LEFT_ALIGNMENT)
topPanel.add(infoPanel)
topPanel.add(Box.createRigidArea(Dimension(0, 10)))
topPanel.add(self.favZoneStatusCBtn)
#table
self.zonesTable = JTable()
tableSelectionModel = self.zonesTable.getSelectionModel()
tableSelectionModel.addListSelectionListener(ZonesTableListener(self))
columns = ["",
self.app.strings.getString("Type"),
self.app.strings.getString("Name")]
tableModel = ZonesTableModel([], columns)
self.zonesTable.setModel(tableModel)
self.scrollPane = JScrollPane(self.zonesTable)
#map
self.zonesMap = JMapViewer()
self.zonesMap.setZoomContolsVisible(False)
self.zonesMap.setMinimumSize(Dimension(100, 200))
#buttons
self.removeBtn = JButton(self.app.strings.getString("Remove"),
ImageProvider.get("dialogs", "delete"),
actionPerformed=self.on_removeBtn_clicked)
self.removeBtn.setToolTipText(self.app.strings.getString("remove_tooltip"))
newBtn = JButton(self.app.strings.getString("New"),
ImageProvider.get("dialogs", "add"),
actionPerformed=self.on_newBtn_clicked)
newBtn.setToolTipText(self.app.strings.getString("new_tooltip"))
#layout
panel2.add(topPanel, BorderLayout.PAGE_START)
panel2.add(self.scrollPane, BorderLayout.LINE_START)
panel2.add(self.zonesMap, BorderLayout.CENTER)
self.buttonsPanel = JPanel()
self.buttonsPanel.add(self.removeBtn)
self.buttonsPanel.add(newBtn)
panel2.add(self.buttonsPanel, BorderLayout.PAGE_END)
self.tabbedPane.addTab(self.app.strings.getString("tab_2_title"),
None,
panel2,
None)
#3 Tab Tools options
panel3 = JPanel()
panel3.setLayout(BoxLayout(panel3, BoxLayout.Y_AXIS))
panel3.setBorder(BorderFactory.createEmptyBorder(7, 7, 7, 7))
for tool in self.app.realTools:
if hasattr(tool, 'prefs'):
p = JPanel(FlowLayout(FlowLayout.LEFT))
p.setBorder(BorderFactory.createTitledBorder(tool.title))
p.add(tool.prefsGui)
panel3.add(p)
self.tabbedPane.addTab(self.app.strings.getString("tab_3_title"),
None,
panel3,
None)
self.add(self.tabbedPane, BorderLayout.CENTER)
exitPanel = JPanel()
saveBtn = JButton(self.app.strings.getString("OK"),
ImageProvider.get("ok"),
actionPerformed=self.on_saveBtn_clicked)
cancelBtn = JButton(self.app.strings.getString("cancel"),
ImageProvider.get("cancel"),
actionPerformed=self.on_cancelBtn_clicked)
saveBtn.setToolTipText(self.app.strings.getString("save_preferences"))
saveBtn.setAlignmentX(0.5)
exitPanel.add(saveBtn)
exitPanel.add(cancelBtn)
self.add(exitPanel, BorderLayout.PAGE_END)
self.addWindowListener(self)
self.pack()
class Gui(JFrame):
'''
classdocs
'''
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 nextPicture(self, event):
percent = (float(len(self.pP.usedList))/len(self.pP.pictureList))*100
self.progressBar.setValue(int(percent))
self.setAnnotation()
self.pP.nextPicture()
#try:
# self.setAnnotation()
# self.pP.nextPicture()
#except AttributeError:
# print "Please choose something!"
def setAnnotationField(self, a):
if len(self.pP.getAnnotationType()) > 1:
[Rbutton.setSelected(True) for Rbutton in self.annoField.getElements() if Rbutton.getActionCommand()==a]
if len(self.pP.getAnnotationType()) == 1:
self.annoField.setText(a)
#rename this method to something clearer!
def setAnnotation(self):
if len(self.pP.getAnnotationType()) > 1:
annotation = self.annoField.getSelection().getActionCommand()
if len(self.pP.getAnnotationType()) == 1:
annotation = self.annoField.getText()
self.annoField.setText(None)
self.pP.getCurrentPicture().annotate(annotation)
def getAnnotation(self):
return self.annotation
def exit(self, event):
self.pP.exit()
self.dispose()
class RadioButtons( TextField ) :
#---------------------------------------------------------------------------
# Name: __init__()
# Role: constructor
#---------------------------------------------------------------------------
def __init__( self, outer ) :
InternalFrame.__init__(
self,
'RadioButtons',
outer,
size = ( 400, 85 ),
location = Point( 5, 225 )
)
self.add( JLabel( 'Timeout (minutes):' ) )
buttons = {}
self.bg = ButtonGroup()
for name in '0,15,30,60,Other'.split( ',' ) :
button = JRadioButton(
name,
itemStateChanged = self.stateChange
)
self.bg.add( button )
self.add( button )
buttons[ name ] = button
self.r00 = buttons[ '0' ]
self.r15 = buttons[ '15' ]
self.r30 = buttons[ '30' ]
self.r60 = buttons[ '60' ]
self.rot = buttons[ 'Other' ]
self.text = self.add(
JTextField(
'',
3,
actionPerformed = outer.update
)
)
self.message = self.add( JLabel() )
self.setting = 0 # see stateChange() and setValue()
self.setVisible( 1 )
#---------------------------------------------------------------------------
# Name: getValue()
# Role: getter
#---------------------------------------------------------------------------
def getValue( self ) :
if self.r00.isSelected() :
result = '0'
elif self.r15.isSelected() :
result = '15'
elif self.r30.isSelected() :
result = '30'
elif self.r60.isSelected() :
result = '60'
elif self.rot.isSelected() :
result = self.text.getText()
try :
int( result )
except :
messageText = badNumber % result
self.message.setText( messageText )
else :
result = None
return result
#---------------------------------------------------------------------------
# Name: setValue()
# Role: setter
#---------------------------------------------------------------------------
def setValue( self, value ) :
self.setting = 1
if value == '0' :
self.r00.setSelected( 1 )
self.r00.requestFocusInWindow()
self.text.setText( '' )
self.text.setEnabled( 0 )
elif value == '15' :
self.r15.setSelected( 1 )
self.r15.requestFocusInWindow()
self.text.setText( '' )
self.text.setEnabled( 0 )
elif value == '30' :
self.r30.setSelected( 1 )
self.r30.requestFocusInWindow()
self.text.setText( '' )
self.text.setEnabled( 0 )
elif value == '60' :
self.r60.setSelected( 1 )
self.r60.requestFocusInWindow()
self.text.setText( '' )
self.text.setEnabled( 0 )
else :
self.rot.setSelected( 1 )
self.text.setText( value )
self.text.setEnabled( 1 )
self.text.requestFocusInWindow()
self.value = value
self.setting = 0
#---------------------------------------------------------------------------
# Name: working()
# Role: Invoked by WSASTask background method
#---------------------------------------------------------------------------
def working( self ) :
for obj in [
self.r00, self.r15, self.r30,
self.r60, self.rot, self.text
] :
obj.setEnabled( 0 )
#---------------------------------------------------------------------------
# Name: finished()
# Role: Invoked by WSASTask done method
#---------------------------------------------------------------------------
def finished( self ) :
for obj in [
self.r00, self.r15, self.r30,
self.r60, self.rot
] :
obj.setEnabled( 1 )
self.text.setEnabled( self.rot.isSelected() )
#---------------------------------------------------------------------------
# Name: stateChange()
# Role: Used to handle RadioButton itemStateChanged events
#---------------------------------------------------------------------------
def stateChange( self, event ) :
item = event.getItem()
if not self.setting :
if item.getText() == 'Other' :
self.text.setEnabled( item.isSelected() )
else :
self.text.setEnabled( 0 )
self.text.setText( '' )
value = self.getValue()
if value :
self.outer.update( event )
def registerExtenderCallbacks(self, callbacks):
# keep a reference to our callbacks object
self._callbacks = callbacks
# obtain an extension helpers object
self._helpers = callbacks.getHelpers()
# set our extension name
callbacks.setExtensionName("Burp Scope Monitor Experimental")
self.GLOBAL_HANDLER_ANALYZED = False
self.GLOBAL_HANDLER = False
self.STATUS = False
self.AUTOSAVE_REQUESTS = 10
self.AUTOSAVE_TIMEOUT = 600 # 10 minutes should be fine
self.CONFIG_INSCOPE = True
self.BAD_EXTENSIONS_DEFAULT = [
'.gif', '.png', '.js', '.woff', '.woff2', '.jpeg', '.jpg', '.css',
'.ico', '.m3u8', '.ts', '.svg'
]
self.BAD_MIMES_DEFAULT = [
'gif', 'script', 'jpeg', 'jpg', 'png', 'video', 'mp2t'
]
self.BAD_EXTENSIONS = self.BAD_EXTENSIONS_DEFAULT
self.BAD_MIMES = self.BAD_MIMES_DEFAULT
# create the log and a lock on which to synchronize when adding log entries
self._currentlyDisplayedItem = None
self.SELECTED_MODEL_ROW = 0
self.SELECTED_VIEW_ROW = 0
self._log = ArrayList()
self._fullLog = ArrayList()
self._lock = Lock()
self._lockFile = Lock()
# main split pane
self._parentPane = JTabbedPane()
self._splitpane = JSplitPane(JSplitPane.VERTICAL_SPLIT)
##### config pane
self._config = JTabbedPane()
config = JPanel()
iexport = JPanel()
#config.setLayout(BorderLayout())
config.setLayout(None)
iexport.setLayout(None)
# config radio button
X_BASE = 40
Y_OFFSET = 5
Y_OPTION = 200
Y_OPTION_SPACING = 20
Y_CHECKMARK_SPACING = 20
self.showAllButton = JRadioButton(SHOW_ALL_BUTTON_LABEL, True)
self.showNewButton = JRadioButton(SHOW_NEW_BUTTON_LABEL, False)
self.showTestedButton = JRadioButton(SHOW_TEST_BUTTON_LABEL, False)
self.showAllButton.setBounds(40, 60 + Y_OFFSET, 400, 30)
self.showNewButton.setBounds(40, 80 + Y_OFFSET, 400, 30)
self.showTestedButton.setBounds(40, 100 + Y_OFFSET, 400, 30)
#self.showNewButton = JRadioButton(SHOW_NEW_BUTTON_LABEL, False)
#self.showTestedButton = JRadioButton(SHOW_TEST_BUTTON_LABEL, False)
self.showAllButton.addActionListener(self.handleRadioConfig)
self.showNewButton.addActionListener(self.handleRadioConfig)
self.showTestedButton.addActionListener(self.handleRadioConfig)
self.clearButton = JButton("Clear")
self.clearButton.addActionListener(self.handleClearButton)
self.clearButton.setBounds(40, 20, 100, 30)
self.startButton = JButton(MONITOR_ON_LABEL)
self.startButton.addActionListener(self.handleStartButton)
self.startButton.setBounds(150, 20, 200, 30)
self.badExtensionsLabel = JLabel("Ignore extensions:")
self.badExtensionsLabel.setBounds(X_BASE, 150, 200, 30)
self.badExtensionsText = JTextArea("")
self.loadBadExtensions()
self.badExtensionsText.setBounds(X_BASE, 175, 310, 30)
self.badExtensionsButton = JButton("Save")
self.badExtensionsButton.addActionListener(
self.handleBadExtensionsButton)
self.badExtensionsButton.setBounds(355, 175, 70, 30)
self.badExtensionsDefaultButton = JButton("Load Defaults")
self.badExtensionsDefaultButton.addActionListener(
self.handleBadExtensionsDefaultButton)
self.badExtensionsDefaultButton.setBounds(430, 175, 120, 30)
self.badMimesLabel = JLabel("Ignore mime types:")
self.badMimesLabel.setBounds(X_BASE, 220, 200, 30)
self.badMimesText = JTextArea("")
self.loadBadMimes()
self.badMimesText.setBounds(X_BASE, 245, 310, 30)
self.badMimesButton = JButton("Save")
self.badMimesButton.addActionListener(self.handleBadMimesButton)
self.badMimesButton.setBounds(355, 245, 70, 30)
self.badMimesDefaultButton = JButton("Load Defaults")
self.badMimesDefaultButton.addActionListener(
self.handleBadMimesDefaultButton)
self.badMimesDefaultButton.setBounds(430, 245, 120, 30)
self.otherLabel = JLabel("Other:")
self.otherLabel.setBounds(40, 300, 120, 30)
self.otherLabel2 = JLabel("Other:")
self.otherLabel2.setBounds(X_BASE, Y_OPTION, 120, 30)
self.autoSaveOption = JCheckBox("Auto save periodically")
self.autoSaveOption.setSelected(True)
self.autoSaveOption.addActionListener(self.handleAutoSaveOption)
self.autoSaveOption.setBounds(X_BASE, Y_OPTION + Y_CHECKMARK_SPACING,
420, 30)
self.repeaterOptionButton = JCheckBox(
"Repeater request automatically marks as analyzed")
self.repeaterOptionButton.setSelected(True)
self.repeaterOptionButton.addActionListener(
self.handleRepeaterOptionButton)
self.repeaterOptionButton.setBounds(50, 330, 420, 30)
self.scopeOptionButton = JCheckBox("Follow Burp Target In Scope rules")
self.scopeOptionButton.setSelected(True)
self.scopeOptionButton.addActionListener(self.handleScopeOptionButton)
self.scopeOptionButton.setBounds(50, 350, 420, 30)
self.startOptionButton = JCheckBox("Autostart Scope Monitor")
self.startOptionButton.setSelected(True)
self.startOptionButton.addActionListener(self.handleStartOption)
self.startOptionButton.setBounds(50, 350 + Y_OPTION_SPACING, 420, 30)
self.markTestedRequestsProxy = JCheckBox(
"Color request in Proxy tab if analyzed")
self.markTestedRequestsProxy.setSelected(True)
self.markTestedRequestsProxy.addActionListener(
self.handleTestedRequestsProxy)
self.markTestedRequestsProxy.setBounds(50, 350 + Y_OPTION_SPACING * 2,
420, 30)
self.markNotTestedRequestsProxy = JCheckBox(
"Color request in Proxy tab if NOT analyzed")
self.markNotTestedRequestsProxy.setSelected(True)
self.markNotTestedRequestsProxy.addActionListener(
self.handleNotTestedRequestsProxy)
self.markNotTestedRequestsProxy.setBounds(50,
350 + Y_OPTION_SPACING * 3,
420, 30)
self.saveButton = JButton("Save now")
self.saveButton.addActionListener(self.handleSaveButton)
self.saveButton.setBounds(X_BASE + 320, 95, 90, 30)
self.loadButton = JButton("Load now")
self.loadButton.addActionListener(self.handleLoadButton)
self.loadButton.setBounds(X_BASE + 420, 95, 90, 30)
self.selectPath = JButton("Select path")
self.selectPath.addActionListener(self.selectExportFile)
self.selectPath.setBounds(X_BASE + 530, 60, 120, 30)
self.selectPathText = JTextArea("")
self.selectPathText.setBounds(X_BASE, 60, 510, 30)
self.selectPathLabel = JLabel("State file:")
self.selectPathLabel.setBounds(X_BASE, 30, 200, 30)
bGroup = ButtonGroup()
bGroup.add(self.showAllButton)
bGroup.add(self.showNewButton)
bGroup.add(self.showTestedButton)
config.add(self.clearButton)
config.add(self.startButton)
config.add(self.startOptionButton)
config.add(self.showAllButton)
config.add(self.showNewButton)
config.add(self.showTestedButton)
config.add(self.badExtensionsButton)
config.add(self.badExtensionsText)
config.add(self.badExtensionsLabel)
config.add(self.badMimesButton)
config.add(self.badMimesText)
config.add(self.badMimesLabel)
config.add(self.badExtensionsDefaultButton)
config.add(self.badMimesDefaultButton)
config.add(self.otherLabel)
config.add(self.repeaterOptionButton)
config.add(self.scopeOptionButton)
config.add(self.markTestedRequestsProxy)
config.add(self.markNotTestedRequestsProxy)
iexport.add(self.saveButton)
iexport.add(self.loadButton)
iexport.add(self.selectPath)
iexport.add(self.selectPathText)
iexport.add(self.selectPathLabel)
iexport.add(self.otherLabel2)
iexport.add(self.autoSaveOption)
self._config.addTab("General", config)
self._config.addTab("Import/Export", iexport)
##### end config pane
self._parentPane.addTab("Monitor", self._splitpane)
self._parentPane.addTab("Config", self._config)
# table of log entries
self.logTable = Table(self)
#self.logTable.setDefaultRenderer(self.logTable.getColumnClass(0), ColoredTableCellRenderer(self))
self.logTable.setAutoCreateRowSorter(True)
self.logTable.setRowSelectionAllowed(True)
renderer = ColoredTableCellRenderer(self)
#column = TableColumn(0, 190, renderer, None)
print 'Initiating... '
# this could be improved by fetching initial dimensions
self.logTable.getColumn("URL").setPreferredWidth(720) # noscope
self.logTable.getColumn("URL").setResizable(True)
self.logTable.getColumn("Checked").setCellRenderer(renderer)
self.logTable.getColumn("Checked").setPreferredWidth(80)
self.logTable.getColumn("Checked").setMaxWidth(80)
self.logTable.getColumn("Method").setPreferredWidth(120)
#self.logTable.getColumn("Method").setMaxWidth(120)
self.logTable.getColumn("Method").setResizable(True)
self.logTable.getColumn("Time").setPreferredWidth(120) # noscope
self.logTable.getColumn("Time").setResizable(True)
scrollPane = JScrollPane(self.logTable)
self._splitpane.setLeftComponent(scrollPane)
# tabs with request/response viewers
tabs = JTabbedPane()
self._requestViewer = callbacks.createMessageEditor(self, False)
self._responseViewer = callbacks.createMessageEditor(self, False)
tabs.addTab("Request", self._requestViewer.getComponent())
tabs.addTab("Response", self._responseViewer.getComponent())
self._splitpane.setRightComponent(tabs)
## Row sorter shit
#self._tableRowSorterAutoProxyAutoAction = CustomTableRowSorter(self.logTable.getModel())
#self.logTable.setRowSorter(self._tableRowSorterAutoProxyAutoAction)
markAnalyzedButton = JMenuItem("Mark Requests as Analyzed")
markAnalyzedButton.addActionListener(markRequestsHandler(self, True))
markNotAnalyzedButton = JMenuItem("Mark Requests as NOT Analyzed")
markNotAnalyzedButton.addActionListener(
markRequestsHandler(self, False))
sendRequestMenu = JMenuItem("Send Request to Repeater")
sendRequestMenu.addActionListener(sendRequestRepeater(self))
deleteRequestMenu = JMenuItem("Delete request")
deleteRequestMenu.addActionListener(deleteRequestHandler(self))
self.menu = JPopupMenu("Popup")
self.menu.add(markAnalyzedButton)
self.menu.add(markNotAnalyzedButton)
self.menu.add(sendRequestMenu)
self.menu.add(deleteRequestMenu)
# customize our UI components
callbacks.customizeUiComponent(self._parentPane)
callbacks.customizeUiComponent(self._splitpane)
callbacks.customizeUiComponent(self._config)
callbacks.customizeUiComponent(config)
callbacks.customizeUiComponent(self.logTable)
callbacks.customizeUiComponent(scrollPane)
callbacks.customizeUiComponent(tabs)
callbacks.registerContextMenuFactory(self)
callbacks.registerExtensionStateListener(self)
callbacks.registerScannerCheck(passiveScanner(self))
# add the custom tab to Burp's UI
callbacks.addSuiteTab(self)
# register ourselves as an HTTP listener
callbacks.registerHttpListener(self)
self.loadConfigs()
print "Loaded!"
print "Experimental import state.. "
self.importState("")
self.SC = sched.scheduler(time.time, time.sleep)
self.SCC = self.SC.enter(10, 1, self.autoSave, (self.SC, ))
self.SC.run()
return
def __init__(self):
''' Configuration Panel '''
# pconfig = JPanel(GridBagLayout())
# pconfig.setSize(Dimension(500,300))
self.setLayout(GridBagLayout())
# super(self,GridBagLayout())
self.setSize(Dimension(500,300))
''' fila 1 '''
label = JLabel('Configuration panel')
c1 = GridBagConstraints()
c1.fill = GridBagConstraints.HORIZONTAL
c1.weightx = 0.5
c1.gridwidth = 4
c1.gridx = 0
c1.gridy = 0
self.add(label, c1)
''' fila 2 '''
self.radioBtnOMC = JRadioButton('OpenModelica')
c2 = GridBagConstraints()
c2.fill = GridBagConstraints.HORIZONTAL
c2.weightx = 0.5
c2.gridx = 0
c2.gridy = 1
self.add(self.radioBtnOMC, c2)
self.radioBtnJM = JRadioButton('JModelica')
c3 = GridBagConstraints()
c3.fill = GridBagConstraints.HORIZONTAL
c3.weightx = 0.5
c3.gridx = 1
c3.gridy = 1
self.add(self.radioBtnJM, c3)
self.radioBtnDY = JRadioButton('Dymola')
c4 = GridBagConstraints()
c4.fill = GridBagConstraints.HORIZONTAL
c4.weightx = 0.5
c4.gridx = 2
c4.gridy = 1
self.add(self.radioBtnDY, c4)
rbBtnGroup = ButtonGroup()
rbBtnGroup.add(self.radioBtnOMC)
rbBtnGroup.add(self.radioBtnJM)
rbBtnGroup.add(self.radioBtnDY)
''' fila 2 '''
label = JLabel('Start time')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 0
c.gridy = 2
self.add(label, c)
self.txtstart= JTextField('0')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 1
c.gridy = 2
self.add(self.txtstart, c)
label = JLabel('Stop time')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 2
c.gridy = 2
self.add(label, c)
self.txtstop= JTextField('0')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 3
c.gridy = 2
self.add(self.txtstop, c)
''' fila 3 '''
label = JLabel('Solver')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 0
c.gridy = 3
self.add(label, c)
self.cbsolver= JComboBox(['dassl','rkfix2'])
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 1
c.gridy = 3
self.add(self.cbsolver, c)
label = JLabel('Algorithm (JM)')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 2
c.gridy = 3
self.add(label, c)
self.cbalgorithm= JComboBox(['AssimuloAlg'])
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 3
c.gridy = 3
self.add(self.cbalgorithm, c)
''' fila 4 '''
label = JLabel('Interval')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 0
c.gridy = 4
self.add(label, c)
self.txtinterval= JTextField('0')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 1
c.gridy = 4
self.add(self.txtinterval, c)
''' fila 5 '''
label = JLabel('Tolerance')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 0
c.gridy = 5
self.add(label, c)
self.txttolerance= JTextField('0')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 1
c.gridy = 5
self.add(self.txttolerance, c)
''' fila 6 '''
label = JLabel('Output format')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 0
c.gridy = 6
self.add(label, c)
self.cboutformat= JComboBox(['.mat','.h5','.csv'])
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 1
c.gridy = 6
self.add(self.cboutformat, c)
label = JLabel('Initialize (JM)')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 2
c.gridy = 6
self.add(label, c)
self.cbinitialize= JComboBox(['True','False'])
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridx = 3
c.gridy = 6
self.add(self.cbinitialize, c)
''' fila 7 '''
bSaveCfg= JButton('Save Configuration', actionPerformed= self.saveConfiguration)
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridwidth = 2
c.gridx = 0
c.gridy = 7
self.add(bSaveCfg, c)
self.bSimulation= JButton('Load Configuration', actionPerformed= self.loadConfiguration)
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 0.5
c.gridwidth = 2
c.gridx = 2
c.gridy = 7
self.add(self.bSimulation, c)
''' fila 8 '''
self.bSimulation= JButton('Simulate', actionPerformed= self.startSimlation)
self.bSimulation.enabled= 0
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 1
c.gridwidth = 4
c.gridx = 0
c.gridy = 8
self.add(self.bSimulation, c)
''' file 9 '''
simProgress= JProgressBar(0, self.getWidth(), value=0, stringPainted=True)
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 1
c.gridwidth = 4
c.gridx = 0
c.gridy = 9
self.add(simProgress, c)
''' fila 10 '''
self.lblResult= JLabel('Simulation information')
c = GridBagConstraints()
c.fill = GridBagConstraints.HORIZONTAL
c.weightx = 1
c.gridwidth = 4
c.gridx = 0
c.gridy = 10
self.add(self.lblResult, c)
class AutopsyImageClassificationModuleWithUISettingsPanel(
IngestModuleIngestJobSettingsPanel):
_logger = Logger.getLogger(
AutopsyImageClassificationModuleFactory.moduleName)
def log(self, level, msg):
self._logger.logp(level, self.__class__.__name__,
inspect.stack()[1][3], msg)
def __init__(self, settings):
self.local_settings = settings
self.config_location = os.path.join(
os.path.dirname(os.path.abspath(__file__)), 'configs.json')
self.init_components()
self.customize_components()
self.check_server_connection(None)
self.classes_of_interest_changes_list = list()
self.classes_of_interest_checkboxes = list()
# Return the settings used
def getSettings(self):
if not os.path.isfile(self.config_location):
self.log(
Level.INFO,
"Configuration file not found, loading the default configuration"
)
self.local_settings.setServerHost(DEFAULT_HOST)
self.local_settings.setServerPort(DEFAULT_PORT)
self.local_settings.setImageFormats(DEFAULT_IMAGES_FORMAT)
self.local_settings.setMinFileSize(DEFAULT_MIN_FILE_SIZE)
self.local_settings.setMinProbability(DEFAULT_MIN_PROBABILITY)
self.local_settings.setClassesOfInterest(
json.loads(DEFAULT_CLASSES_OF_INTEREST))
# self.saveSettings(None)
return self.local_settings
else:
if not os.access(self.config_location, os.R_OK):
err_string = "Cannot access configuration file, please review the file permissions"
raise IngestModuleException(err_string)
with io.open(self.config_location, 'r', encoding='utf-8') as f:
self.log(Level.INFO, "Configuration file read")
json_configs = json.load(f)
self.local_settings.setServerHost(json_configs['server']['host'])
self.local_settings.setServerPort(json_configs['server']['port'])
image_formats = json_configs['imageFormats']
if not isinstance(image_formats, list) or len(image_formats) == 0:
err_string = "Invalid list of image formats given"
raise IngestModuleException(err_string)
self.local_settings.setImageFormats(image_formats)
self.local_settings.setMinFileSize(json_configs['minFileSize'])
self.local_settings.setMinProbability(
json_configs['minProbability'])
self.local_settings.setClassesOfInterest(
json_configs['classesOfInterest'])
return self.local_settings
def check_server_connection(self, e):
message_string = "Testing connection with server..."
self.log(Level.INFO, message_string)
self.message.setText(message_string)
self.error_message.setText("")
new_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
new_socket.settimeout(1)
try:
new_socket.connect(
(self.host_TF.getText(), int(self.port_TF.getText())))
self.local_settings.setIsServerOnline(True)
message_string = "Server is up!"
self.log(Level.INFO, message_string)
self.message.setText(message_string)
except (socket.timeout, socket.error) as e:
self.local_settings.setIsServerOnline(False)
err_string = "Server is down"
self.error_message.setText(err_string)
self.message.setText("")
self.log(Level.INFO, err_string)
finally:
new_socket.close()
def save_settings(self, e):
self.message.setText("")
self.log(Level.INFO, "Settings save button clicked!")
host = self.host_TF.getText()
if not host.strip():
err_string = "Invalid host"
self.error_message.setText(err_string)
return
# raise IngestModuleException(err_string)
port = self.port_TF.getText()
if not host.strip():
err_string = "Invalid port number"
self.error_message.setText(err_string)
return
# raise IngestModuleException(err_string)
image_formats_array = self.image_formats_TF.getText().strip().split(
';')
if len(image_formats_array) == 0 or not image_formats_array[0]:
err_string = "Invalid image formats"
self.error_message.setText(err_string)
return
# raise IngestModuleException(err_string)
min_probability_string = self.min_probability_TF.getText().strip()
if not min_probability_string:
err_string = "Invalid minimum confidence value"
self.error_message.setText(err_string)
return
# raise IngestModuleException(err_string)
min_probability = int(float(min_probability_string))
min_file_size_string = self.min_file_size_TF.getText().strip()
if not min_file_size_string:
err_string = "Invalid minimum file size"
self.error_message.setText(err_string)
return
# raise IngestModuleException(err_string)
min_file_size = int(float(min_file_size_string))
configs = {
'server': {
'host': host,
'port': port
},
'imageFormats': image_formats_array,
'minProbability': min_probability,
'minFileSize': min_file_size,
'classesOfInterest': self.local_settings.getClassesOfInterest()
}
with io.open(self.config_location, 'w', encoding='utf-8') as f:
f.write(json.dumps(configs, ensure_ascii=False))
self.error_message.setText("")
message = "Settings saved "
self.message.setText(message)
self.log(Level.INFO, message + " in " + self.config_location)
def open_text_editor(self, e):
self.log(Level.INFO, "Lauching external text editor ")
if sys.platform == "win32":
subprocess.call(["notepad", self.config_location])
else:
opener = "open" if sys.platform == "darwin" else "xdg-open"
subprocess.call([opener, self.config_location])
def on_save_classes_of_interest_click(self, e):
self.detectable_obejcts_dialog.setVisible(False)
for item in self.classes_of_interest_changes_list:
for class_oi in self.local_settings.getClassesOfInterest():
if item.getText() == class_oi['name']:
self.log(
Level.INFO, "match found " + class_oi['name'] +
" and is selected " + str(item.isSelected()))
class_oi['enabled'] = item.isSelected()
break
self.classes_of_interest_checkboxes = list()
def on_cancel_classes_of_interest_click(self, e):
self.detectable_obejcts_dialog.setVisible(False)
self.classes_of_interest_changes_list = list()
self.classes_of_interest_checkboxes = list()
def on_class_checkbox_clicked(self, e):
self.classes_of_interest_changes_list.append(e.getSource())
def on_select_all_clicked(self, e):
for checkbox in self.classes_of_interest_checkboxes:
checkbox.setSelected(True)
def on_deselect_all_clicked(self, e):
for checkbox in self.classes_of_interest_checkboxes:
checkbox.setSelected(False)
def show_detectable_objects_dialog(self, e):
parentComponent = SwingUtilities.windowForComponent(self.panel0)
self.detectable_obejcts_dialog = JDialog(
parentComponent, "List of Objects to Detect",
ModalityType.APPLICATION_MODAL)
panel = JPanel()
self.detectable_obejcts_dialog.add(panel)
gbPanel = GridBagLayout()
gbcPanel = GridBagConstraints()
panel.setLayout(gbPanel)
y = 0
x = 0
for line in self.local_settings.getClassesOfInterest():
if y > 15:
y = 0
x = x + 1
class_check_box = JCheckBox(line['name'])
self.classes_of_interest_checkboxes.append(class_check_box)
class_check_box.setEnabled(True)
class_check_box.setSelected(line['enabled'])
class_check_box.addItemListener(self.on_class_checkbox_clicked)
gbcPanel.gridx = x
gbcPanel.gridy = y
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(class_check_box, gbcPanel)
panel.add(class_check_box)
y = y + 1
blank_1_L = JLabel(" ")
blank_1_L.setEnabled(True)
gbcPanel.gridx = 0
gbcPanel.gridy = y + 1
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 0
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(blank_1_L, gbcPanel)
panel.add(blank_1_L)
deselect_all_button = JButton("Deselect all")
deselect_all_button.setEnabled(True)
deselect_all_button.addActionListener(self.on_deselect_all_clicked)
gbcPanel.gridx = 1
gbcPanel.gridy = y + 2
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(deselect_all_button, gbcPanel)
panel.add(deselect_all_button)
select_all_button = JButton("Select all")
select_all_button.setEnabled(True)
select_all_button.addActionListener(self.on_select_all_clicked)
gbcPanel.gridx = 3
gbcPanel.gridy = y + 2
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(select_all_button, gbcPanel)
panel.add(select_all_button)
blank_2_L = JLabel(" ")
blank_2_L.setEnabled(True)
gbcPanel.gridx = 0
gbcPanel.gridy = y + 3
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 0
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(blank_2_L, gbcPanel)
panel.add(blank_2_L)
cancel_button = JButton("Cancel")
cancel_button.setEnabled(True)
cancel_button.addActionListener(
self.on_cancel_classes_of_interest_click)
gbcPanel.gridx = 1
gbcPanel.gridy = y + 4
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(cancel_button, gbcPanel)
panel.add(cancel_button)
save_button = JButton("Save")
save_button.setEnabled(True)
save_button.addActionListener(self.on_save_classes_of_interest_click)
gbcPanel.gridx = 3
gbcPanel.gridy = y + 4
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 2
gbcPanel.weighty = 1
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(save_button, gbcPanel)
panel.add(save_button)
blank_3_L = JLabel(" ")
blank_3_L.setEnabled(True)
gbcPanel.gridx = 0
gbcPanel.gridy = y + 5
gbcPanel.gridwidth = 1
gbcPanel.gridheight = 1
gbcPanel.fill = GridBagConstraints.BOTH
gbcPanel.weightx = 1
gbcPanel.weighty = 0
gbcPanel.anchor = GridBagConstraints.NORTH
gbPanel.setConstraints(blank_3_L, gbcPanel)
panel.add(blank_3_L)
self.detectable_obejcts_dialog.pack()
screenSize = Toolkit.getDefaultToolkit().getScreenSize()
self.detectable_obejcts_dialog.setLocation(
int((screenSize.getWidth() / 2) -
(self.detectable_obejcts_dialog.getWidth() / 2)),
int((screenSize.getHeight() / 2) -
(self.detectable_obejcts_dialog.getHeight() / 2)))
self.detectable_obejcts_dialog.setVisible(True)
def customize_components(self):
settings = self.getSettings()
self.host_TF.setText(settings.getServerHost())
self.port_TF.setText(str(settings.getServerPort()))
self.log(Level.INFO, "[customizeComponents]")
self.log(Level.INFO, settings.getImageFormats()[0])
self.image_formats_TF.setText(';'.join(settings.getImageFormats()))
self.min_probability_TF.setText(str(settings.getMinProbability()))
self.min_file_size_TF.setText(str(settings.getMinFileSize()))
def init_components(self):
self.panel0 = JPanel()
self.rbgPanel0 = ButtonGroup()
self.gbPanel0 = GridBagLayout()
self.gbcPanel0 = GridBagConstraints()
self.panel0.setLayout(self.gbPanel0)
self.host_L = JLabel("Host:")
self.host_L.setEnabled(True)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.host_L, self.gbcPanel0)
self.panel0.add(self.host_L)
self.port_L = JLabel("Port:")
self.port_L.setEnabled(True)
self.gbcPanel0.gridx = 1
self.gbcPanel0.gridy = 1
self.gbcPanel0.gridwidth = 2
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.port_L, self.gbcPanel0)
self.panel0.add(self.port_L)
self.host_TF = JTextField(10)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 2
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.host_TF, self.gbcPanel0)
self.panel0.add(self.host_TF)
format = NumberFormat.getInstance()
format.setGroupingUsed(False)
port_formatter = NumberFormatter(format)
port_formatter.setValueClass(Integer)
port_formatter.setAllowsInvalid(False)
port_formatter.setMinimum(Integer(0))
port_formatter.setMaximum(Integer(65535))
self.port_TF = JFormattedTextField(port_formatter)
self.gbcPanel0.gridx = 1
self.gbcPanel0.gridy = 2
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.port_TF, self.gbcPanel0)
self.panel0.add(self.port_TF)
self.blank_1_L = JLabel(" ")
self.blank_1_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 3
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_1_L, self.gbcPanel0)
self.panel0.add(self.blank_1_L)
self.image_formats_L = JLabel("Format of images (separator \";\"):")
self.port_L.setEnabled(True)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 4
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.image_formats_L, self.gbcPanel0)
self.panel0.add(self.image_formats_L)
self.image_formats_TF = JTextField(10)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 5
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.image_formats_TF, self.gbcPanel0)
self.panel0.add(self.image_formats_TF)
self.blank_2_L = JLabel(" ")
self.blank_2_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 6
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_2_L, self.gbcPanel0)
self.panel0.add(self.blank_2_L)
self.min_probability_L = JLabel("Confidence minimum (%):")
self.port_L.setEnabled(True)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 7
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.min_probability_L, self.gbcPanel0)
self.panel0.add(self.min_probability_L)
min_probabilty_formatter = NumberFormatter(format)
min_probabilty_formatter.setValueClass(Integer)
min_probabilty_formatter.setAllowsInvalid(False)
min_probabilty_formatter.setMinimum(Integer(0))
min_probabilty_formatter.setMaximum(Integer(100))
self.min_probability_TF = JFormattedTextField(min_probabilty_formatter)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 8
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.min_probability_TF, self.gbcPanel0)
self.panel0.add(self.min_probability_TF)
self.blank_3_L = JLabel(" ")
self.blank_3_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 9
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_3_L, self.gbcPanel0)
self.panel0.add(self.blank_3_L)
self.min_file_size_L = JLabel("Minimum file size (KB):")
self.min_file_size_L.setEnabled(True)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 10
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.min_file_size_L, self.gbcPanel0)
self.panel0.add(self.min_file_size_L)
min_file_size_formatter = NumberFormatter(format)
min_file_size_formatter.setValueClass(Integer)
min_file_size_formatter.setAllowsInvalid(False)
min_file_size_formatter.setMinimum(Integer(0))
self.min_file_size_TF = JFormattedTextField(min_file_size_formatter)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 11
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.min_file_size_TF, self.gbcPanel0)
self.panel0.add(self.min_file_size_TF)
self.blank_4_L = JLabel(" ")
self.blank_4_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 12
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_4_L, self.gbcPanel0)
self.panel0.add(self.blank_4_L)
self.detectable_obejcts_BTN = \
JButton("List of Objects to Detect", actionPerformed=self.show_detectable_objects_dialog)
# self.save_Settings_BTN.setPreferredSize(Dimension(1, 20))
self.rbgPanel0.add(self.detectable_obejcts_BTN)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 13
self.gbPanel0.setConstraints(self.detectable_obejcts_BTN,
self.gbcPanel0)
self.panel0.add(self.detectable_obejcts_BTN)
self.blank_5_L = JLabel(" ")
self.blank_5_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 14
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_5_L, self.gbcPanel0)
self.panel0.add(self.blank_5_L)
self.error_message = JLabel("", JLabel.CENTER)
self.error_message.setForeground(Color.red)
self.error_message.setEnabled(True)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 15
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.error_message, self.gbcPanel0)
self.panel0.add(self.error_message)
self.message = JLabel("", JLabel.CENTER)
self.message.setEnabled(True)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 16
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.message, self.gbcPanel0)
self.panel0.add(self.message)
self.save_settings_BTN = JButton("Save Settings",
actionPerformed=self.save_settings)
# self.save_Settings_BTN.setPreferredSize(Dimension(1, 20))
self.rbgPanel0.add(self.save_settings_BTN)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 17
self.gbPanel0.setConstraints(self.save_settings_BTN, self.gbcPanel0)
self.panel0.add(self.save_settings_BTN)
self.blank_6_L = JLabel(" ")
self.blank_6_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 18
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_6_L, self.gbcPanel0)
self.panel0.add(self.blank_6_L)
self.blank_7_L = JLabel(" ")
self.blank_7_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 19
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_7_L, self.gbcPanel0)
self.panel0.add(self.blank_7_L)
self.check_server_connection_BTN = \
JButton("Check server connection", actionPerformed=self.check_server_connection)
# self.save_Settings_BTN.setPreferredSize(Dimension(1, 20))
self.rbgPanel0.add(self.check_server_connection_BTN)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 20
self.gbPanel0.setConstraints(self.check_server_connection_BTN,
self.gbcPanel0)
self.panel0.add(self.check_server_connection_BTN)
self.blank_8_L = JLabel(" ")
self.blank_8_L.setEnabled(True)
self.gbcPanel0.gridx = 2
self.gbcPanel0.gridy = 21
self.gbcPanel0.gridwidth = 1
self.gbcPanel0.gridheight = 1
self.gbcPanel0.fill = GridBagConstraints.BOTH
self.gbcPanel0.weightx = 1
self.gbcPanel0.weighty = 0
self.gbcPanel0.anchor = GridBagConstraints.NORTH
self.gbPanel0.setConstraints(self.blank_8_L, self.gbcPanel0)
self.panel0.add(self.blank_8_L)
self.text_editor_BTN = \
JButton("Open config file", actionPerformed=self.open_text_editor)
# self.save_Settings_BTN.setPreferredSize(Dimension(1, 20))
self.rbgPanel0.add(self.text_editor_BTN)
self.gbcPanel0.gridx = 0
self.gbcPanel0.gridy = 22
self.gbPanel0.setConstraints(self.text_editor_BTN, self.gbcPanel0)
self.panel0.add(self.text_editor_BTN)
self.add(self.panel0)
def getUiComponent(self):
self.panel = JPanel()
self.main = JPanel()
self.main.setLayout(BoxLayout(self.main, BoxLayout.Y_AXIS))
self.access_key_panel = JPanel()
self.main.add(self.access_key_panel)
self.access_key_panel.setLayout(
BoxLayout(self.access_key_panel, BoxLayout.X_AXIS))
self.access_key_panel.add(JLabel('Access Key: '))
self.access_key = JTextField('', 25)
self.access_key_panel.add(self.access_key)
self.secret_key_panel = JPanel()
self.main.add(self.secret_key_panel)
self.secret_key_panel.setLayout(
BoxLayout(self.secret_key_panel, BoxLayout.X_AXIS))
self.secret_key_panel.add(JLabel('Secret Key: '))
self.secret_key = JPasswordField('', 25)
self.secret_key_panel.add(self.secret_key)
self.target_host_panel = JPanel()
self.main.add(self.target_host_panel)
self.target_host_panel.setLayout(
BoxLayout(self.target_host_panel, BoxLayout.X_AXIS))
self.target_host_panel.add(JLabel('Target host: '))
self.target_host = JTextField('example.com', 25)
self.target_host_panel.add(self.target_host)
self.buttons_panel = JPanel()
self.main.add(self.buttons_panel)
self.buttons_panel.setLayout(
BoxLayout(self.buttons_panel, BoxLayout.X_AXIS))
#self.save_button = JButton('Save', actionPerformed = self.saveKeys) #not implemented yet
#self.buttons_panel.add(self.save_button)
self.enable_button = JButton('Enable',
actionPerformed=self.enableGateway)
self.buttons_panel.add(self.enable_button)
self.disable_button = JButton('Disable',
actionPerformed=self.disableGateway)
self.buttons_panel.add(self.disable_button)
self.disable_button.setEnabled(False)
self.protocol_panel = JPanel()
self.main.add(self.protocol_panel)
self.protocol_panel.setLayout(
BoxLayout(self.protocol_panel, BoxLayout.Y_AXIS))
self.protocol_panel.add(JLabel("Target Protocol:"))
self.https_button = JRadioButton("HTTPS", True)
self.http_button = JRadioButton("HTTP", False)
self.protocol_panel.add(self.http_button)
self.protocol_panel.add(self.https_button)
buttongroup = ButtonGroup()
buttongroup.add(self.https_button)
buttongroup.add(self.http_button)
self.regions_title = JPanel()
self.main.add(self.regions_title)
self.regions_title.add(JLabel("Regions to launch API Gateways in:"))
self.regions_panel = JPanel()
self.main.add(self.regions_panel)
glayout = GridLayout(4, 3)
self.regions_panel.setLayout(glayout)
for region in AVAIL_REGIONS:
cur_region = region.replace('-', '_')
cur_region = cur_region + '_status'
setattr(self, cur_region, JCheckBox(region, True))
attr = getattr(self, cur_region)
self.regions_panel.add(attr)
self.status = JPanel()
self.main.add(self.status)
self.status.setLayout(BoxLayout(self.status, BoxLayout.X_AXIS))
self.status_indicator = JLabel(DISABLED, JLabel.CENTER)
self.status.add(self.status_indicator)
self.panel.add(self.main)
return self.panel
class ImagesPanelOptions(FormPanel, ExportPanel):
def __init__(self, processPanel, parameters):
ExportPanel.__init__(self)
FormPanel.__init__(self, gvsig.getResource(__file__,"exportImagesPanels.xml"))
# Translate
toolsSwingManager = ToolsSwingLocator.getToolsSwingManager()
toolsSwingManager.translate(self.lblFieldImages)
toolsSwingManager.translate(self.lblImageFormat)
toolsSwingManager.translate(self.lblOutputPathParameters)
toolsSwingManager.translate(self.rdbOption1)
toolsSwingManager.translate(self.rdbOption2)
toolsSwingManager.translate(self.rdbOption3)
# hay un picker para folder
self.pickerFolder = None
self.processPanel = processPanel
self.params = parameters
self.store = parameters.getSourceFeatureStore()
self.initComponents()
def initComponents(self):
dialogTitle = "_Select_folder"
fileChooserID = "idPathGMLSelector"
seticon = True
# Init
## Button group
self.btgOptions = ButtonGroup()
self.btgOptions.add(self.rdbOption1)
self.btgOptions.add(self.rdbOption2)
self.btgOptions.add(self.rdbOption3)
## Image format
self.cmbImageFormat.addItem("PNG")
self.cmbImageFormat.addItem("JPEG")
self.cmbImageFormat.addItem("GIF")
## Pickers
### Field with image data
swm = DALSwingLocator.getSwingManager()
self.pickerFieldsImageData = swm.createAttributeDescriptorPickerController(self.cmbImageField)
self.pickerFieldsPath = swm.createAttributeDescriptorPickerController(self.cmbPathField)
self.pickerFieldsName = swm.createAttributeDescriptorPickerController(self.cmbPathFieldName)
self.pickerFolder = ToolsSwingLocator.getToolsSwingManager().createFolderPickerController(
self.txtPath,
self.btnPath)
eesmanager = ExpressionEvaluatorSwingLocator.getManager()
self.pickerExp = eesmanager.createExpressionPickerController(self.txtExp, self.btnExp)
self.pickerExpStore= swm.createFeatureStoreElement(self.store)
self.pickerExp.addElement(self.pickerExpStore)
# Input
## Layer
allDocuments = self.getAllValidDocuments()
if len(allDocuments)==0:
logger("Not able to find 2 tables to execute the tool", LOGGER_INFO)
return
else:
self._updateAllUI()
def _updateAllUI(self):
# Input params dependant of layer
swm = DALSwingLocator.getSwingManager()
## Field with image data
ft1 = self.store.getDefaultFeatureType()
self.pickerFieldsImageData.setFeatureType(ft1)
# Output params
## Use absolute path from ield
self.pickerFieldsPath.setFeatureType(ft1)
## Export to folder using name
self.pickerFieldsName.setFeatureType(ft1)
## Expression for absolute path
self.pickerExpStore.setFeatureStore(self.store)
### Preview expression
## Sample feature
sampleFeature = None
try:
sampleFeature = store.getFeatureSelection().first()
if sampleFeature == None:
sampleFeature = store.first()
except:
logger("Not able to create Sample Feature for FieldCalculatorTool", LOGGER_WARN)
if sampleFeature!=None:
dataManager = DALLocator.getDataManager()
featureSymbolTable = dataManager.createFeatureSymbolTable()
featureSymbolTable.setFeature(sampleFeature)
self.pickerExp.setPreviewSymbolTable(featureSymbolTable.createParent())
def getIdPanel(self):
return "ExportImagesPanelOptions"
def getTitlePanel(self):
pass
def validatePanel(self):
return True
def enterPanel(self):
initialPath = File(Utilities.TEMPDIRECTORYPATH)
self.pickerFolder.set(initialPath)
self.pickerFolder.setFileFilter(MyFileFilter())
self._updateAllUI()
imageOutput = None
if self.params.getImageField()!=None:
self.pickerFieldsImageData.set(self.params.getImageField())
if self.params.getImageFormat()!=None:
self.cmbImageFormat.setSelectedItem(self.params.getImageFormat())
if self.params.getImageOutputOption()!= None:
imageOutput = self.params.getImageOutputOption()
if imageOutput!=None:
option = imageOutput["option"]
oparams = imageOutput["params"]
if option==1:
self.rdbOption1.setSelected(True)
op = oparams["path"]
self.pickerFieldsPath.set(op)
elif option==2:
self.rdbOption2.setSelected(True)
op = oparams["path"]
self.pickerFolder.set(op)
of = oparams["field"]
self.pickerFieldsName.set(of)
elif option==3:
self.rdbOption3.setSelected(True)
oe = oparams["expression"]
self.pickerExp.set(oe)
def previousPanel(self):
pass
def nextPanel(self):
#recoger valores del formulario y guardarlos en parameters
self.params.setImageField(self.pickerFieldsImageData.get())
self.params.setImageFormat(self.cmbImageFormat.getSelectedItem())
if self.rdbOption1.isSelected():
outparams={"option":1,"params":{"path",self.pickerFieldsPath.getName()}} #DefaultFeatureAttributeDescriptor
elif self.rdbOption2.isSelected():
outparams={"option":2,"params":{"field":self.pickerFieldsName.getName(),"path":self.pickerFolder.get()}}
elif self.rdbOption3.isSelected():
outparams={"option":3,"params":{"expression":self.pickerExp.get()}}
else:
pass
self.params.setImageOutputOption(outparams)
#print "params:", self.params.getImageOutputOption()
def getAllValidDocuments(self):
application = ApplicationLocator.getManager()
project = application.getCurrentProject()
views = project.getDocuments()
all = []
for view in views:
#print view, type(view)
if isinstance(view, ViewDocument):
#print view, type(view)
for layer in view.getMapContext().getLayers():
#print "--", layer==mlayer, layer.getName()
all.append(layer)
elif isinstance(view, TableDocument):
#print "--", view
all.append(view)
return all
class BurpExtender(IBurpExtender, IHttpListener, IContextMenuFactory, ITab):
def registerExtenderCallbacks(self, callbacks):
self.lastTimestamp = None
self.callbacks = callbacks
self.helpers = callbacks.getHelpers()
sys.stdout = callbacks.getStdout()
l = self.callbacks.loadExtensionSetting
self.es_host = l("elasticburp.host") or 'localhost'
self.es_index = l("elasticburp.index") or 'burp'
self.whitelist = l("elasticburp.whitelist") or ''
tools = l("elasticburp.tools")
self.tools = int(tools) if tools is not None else 511
self.log_level = l("elasticburp.logLevel") or 'INFO'
logger.setLevel(getattr(logging, self.log_level))
self.callbacks.setExtensionName("ElasticBurp")
self.callbacks.registerHttpListener(self)
self.callbacks.registerContextMenuFactory(self)
self.callbacks.addSuiteTab(self)
def processHttpMessage(self, tool, isRequest, msg):
if (tool & self.tools):
logger.warning('Indexing single document')
doc = self.create_document(msg)
if doc is not None:
doc.save()
def create_document(self, msg):
http_service = msg.getHttpService()
host = http_service.getHost()
if self.whitelist and self.whitelist not in host:
logger.warning('Skipping {} because it\'s not in '
'the whitelist'.format(host))
return
doc = ElasticBurpDocument(
self.es_index, self.es_host, http_service.getProtocol(),
host, http_service.getPort())
request = msg.getRequest()
response = msg.getResponse()
if request:
req = self.helpers.analyzeRequest(msg)
doc.request['method'] = req.getMethod()
doc.request['url'] = req.getUrl().toString()
logger.info('Indexing {} request from {}'.format(
doc.request['method'], doc.request['url']))
doc.add_request_headers(req.getHeaders())
parameters = req.getParameters()
for parameter in parameters:
doc.add_request_parameter(
PARAMETER_TYPES[parameter.getType()],
parameter.getName(), parameter.getValue())
ct = CONTENT_TYPES[req.getContentType()]
doc.request['content_type'] = ct
bodyOffset = req.getBodyOffset()
doc.request['body'] = request[bodyOffset:].tostring().decode(
"ascii", "replace")
if response:
iResponse = self.helpers.analyzeResponse(response)
doc.response['status'] = iResponse.getStatusCode()
doc.response['content_type'] = iResponse.getStatedMimeType()
doc.response['inferred_content_type'] = \
iResponse.getInferredMimeType()
logger.info('Indexing {} response'.format(doc.response['status']))
doc.add_response_headers(iResponse.getHeaders())
cookies = iResponse.getCookies()
for cookie in cookies:
expCookie = cookie.getExpiration()
expiration = datetime.fromtimestamp(expCookie.time / 1000) if \
expCookie else None
doc.add_response_cookie(
cookie.getName(), cookie.getValue(), cookie.getDomain(),
cookie.getPath(), expiration)
bodyOffset = iResponse.getBodyOffset()
doc.response['body'] = response[bodyOffset:].tostring().decode(
"ascii", "replace")
return doc
def getTabCaption(self):
return "ElasticBurp"
def getUiComponent(self):
ui_panel = JPanel()
ui_panel.setLayout(BoxLayout(ui_panel, BoxLayout.PAGE_AXIS))
ui_host_line = JPanel()
ui_host_line.setLayout(BoxLayout(ui_host_line, BoxLayout.LINE_AXIS))
ui_host_line.setAlignmentX(JPanel.LEFT_ALIGNMENT)
ui_host_line.add(JLabel("ElasticSearch Host: "))
self.ui_es_host = JTextField(40)
self.ui_es_host.setMaximumSize(self.ui_es_host.getPreferredSize())
self.ui_es_host.setText(self.es_host)
ui_host_line.add(self.ui_es_host)
ui_panel.add(ui_host_line)
ui_index_line = JPanel()
ui_index_line.setLayout(BoxLayout(ui_index_line, BoxLayout.LINE_AXIS))
ui_index_line.setAlignmentX(JPanel.LEFT_ALIGNMENT)
ui_index_line.add(JLabel("ElasticSearch Index: "))
self.ui_es_index = JTextField(40)
self.ui_es_index.setText(self.es_index)
self.ui_es_index.setMaximumSize(self.ui_es_index.getPreferredSize())
ui_index_line.add(self.ui_es_index)
ui_panel.add(ui_index_line)
ui_whitelist_line = JPanel()
ui_whitelist_line.setLayout(
BoxLayout(ui_whitelist_line, BoxLayout.LINE_AXIS))
ui_whitelist_line.setAlignmentX(JPanel.LEFT_ALIGNMENT)
ui_whitelist_line.add(JLabel("Host whitelist: "))
self.ui_whitelist = JTextField(40)
self.ui_whitelist.setText(self.whitelist)
self.ui_whitelist.setMaximumSize(self.ui_whitelist.getPreferredSize())
ui_whitelist_line.add(self.ui_whitelist)
ui_panel.add(ui_whitelist_line)
ui_tools_panel = JPanel()
ui_tools_panel.setLayout(
BoxLayout(ui_tools_panel, BoxLayout.LINE_AXIS))
ui_tools_panel.setAlignmentX(JPanel.LEFT_ALIGNMENT)
self.ui_tool_suite = JCheckBox(
"Suite", self.tools & ECallbacks.TOOL_SUITE != 0)
ui_tools_panel.add(self.ui_tool_suite)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_target = JCheckBox(
"Target", self.tools & ECallbacks.TOOL_TARGET != 0)
ui_tools_panel.add(self.ui_tool_target)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_proxy = JCheckBox(
"Proxy", self.tools & ECallbacks.TOOL_PROXY != 0)
ui_tools_panel.add(self.ui_tool_proxy)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_spider = JCheckBox(
"Spider", self.tools & ECallbacks.TOOL_SPIDER != 0)
ui_tools_panel.add(self.ui_tool_spider)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_scanner = JCheckBox(
"Scanner", self.tools & ECallbacks.TOOL_SCANNER != 0)
ui_tools_panel.add(self.ui_tool_scanner)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_intruder = JCheckBox(
"Intruder", self.tools & ECallbacks.TOOL_INTRUDER != 0)
ui_tools_panel.add(self.ui_tool_intruder)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_repeater = JCheckBox(
"Repeater", self.tools & ECallbacks.TOOL_REPEATER != 0)
ui_tools_panel.add(self.ui_tool_repeater)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_sequencer = JCheckBox(
"Sequencer", self.tools & ECallbacks.TOOL_SEQUENCER != 0)
ui_tools_panel.add(self.ui_tool_sequencer)
ui_tools_panel.add(Box.createRigidArea(Dimension(10, 0)))
self.ui_tool_extender = JCheckBox(
"Extender", self.tools & ECallbacks.TOOL_EXTENDER != 0)
ui_tools_panel.add(self.ui_tool_extender)
ui_panel.add(ui_tools_panel)
ui_panel.add(Box.createRigidArea(Dimension(0, 10)))
ui_log_line = JPanel()
ui_log_line.setLayout(BoxLayout(ui_log_line, BoxLayout.LINE_AXIS))
ui_log_line.setAlignmentX(JPanel.LEFT_ALIGNMENT)
ui_debug = JRadioButton("DEBUG", self.log_level == 'DEBUG')
ui_log_line.add(ui_debug)
ui_log_line.add(Box.createRigidArea(Dimension(10, 0)))
ui_info = JRadioButton("INFO", self.log_level == 'INFO')
ui_log_line.add(ui_info)
ui_log_line.add(Box.createRigidArea(Dimension(10, 0)))
ui_warning = JRadioButton("WARNING", self.log_level == 'WARNING')
ui_log_line.add(ui_warning)
ui_log_line.add(Box.createRigidArea(Dimension(10, 0)))
ui_error = JRadioButton("ERROR", self.log_level == 'ERROR')
ui_log_line.add(ui_error)
ui_log_line.add(Box.createRigidArea(Dimension(10, 0)))
ui_critical = JRadioButton(
"CRITICAL", self.log_level == 'CRITICAL')
ui_log_line.add(ui_critical)
ui_log_line.add(Box.createRigidArea(Dimension(10, 0)))
ui_panel.add(ui_log_line)
ui_panel.add(Box.createRigidArea(Dimension(0, 10)))
self.ui_log_level = ButtonGroup()
self.ui_log_level.add(ui_debug)
self.ui_log_level.add(ui_info)
self.ui_log_level.add(ui_warning)
self.ui_log_level.add(ui_error)
self.ui_log_level.add(ui_critical)
ui_buttons_line = JPanel()
ui_buttons_line.setLayout(
BoxLayout(ui_buttons_line, BoxLayout.LINE_AXIS))
ui_buttons_line.setAlignmentX(JPanel.LEFT_ALIGNMENT)
ui_buttons_line.add(
JButton("Save config", actionPerformed=self.save_config))
ui_panel.add(ui_buttons_line)
return ui_panel
def save_config(self, event):
logger.warning('Saving changes in config')
self.es_host = self.ui_es_host.getText()
self.es_index = self.ui_es_index.getText()
self.whitelist = self.ui_whitelist.getText()
self.tools = int(
(self.ui_tool_suite.isSelected() and ECallbacks.TOOL_SUITE) |
(self.ui_tool_target.isSelected() and ECallbacks.TOOL_TARGET) |
(self.ui_tool_proxy.isSelected() and ECallbacks.TOOL_PROXY) |
(self.ui_tool_spider.isSelected() and ECallbacks.TOOL_SPIDER) |
(self.ui_tool_scanner.isSelected() and ECallbacks.TOOL_SCANNER) |
(self.ui_tool_intruder.isSelected() and ECallbacks.TOOL_INTRUDER) |
(self.ui_tool_repeater.isSelected() and ECallbacks.TOOL_REPEATER) |
(self.ui_tool_sequencer.isSelected() and
ECallbacks.TOOL_SEQUENCER) |
(self.ui_tool_extender.isSelected() and ECallbacks.TOOL_EXTENDER))
logger.debug('Changing tools in config to %d' % self.tools)
for button in self.ui_log_level.getElements():
if button.isSelected():
logger.debug('Changing logLevel to %s' % button.getText())
self.log_level = button.getText()
s = self.callbacks.saveExtensionSetting
s("elasticburp.host", self.es_host)
s("elasticburp.index", self.es_index)
s("elasticburp.tools", str(self.tools))
s("elasticburp.logLevel", self.log_level)
s("elasticburp.whitelist", self.whitelist)
logger.setLevel(getattr(logging, self.log_level))
class ProjectIngestSettingsPanel(IngestModuleIngestJobSettingsPanel):
def __init__(self, settings):
self.local_settings = settings
self.initComponents()
self.customizeComponents()
def initComponents(self):
self.apps_checkboxes_list = []
self.setLayout(BoxLayout(self, BoxLayout.PAGE_AXIS))
self.setPreferredSize(Dimension(300, 0))
# title
self.p_title = SettingsUtils.createPanel()
self.lb_title = JLabel("Forensic Analysis for Mobile Apps")
self.lb_title.setFont(self.lb_title.getFont().deriveFont(
Font.BOLD, 15))
self.p_title.add(self.lb_title)
self.add(self.p_title)
# end of title
# info menu
self.p_info = SettingsUtils.createPanel()
self.p_info.setPreferredSize(Dimension(300, 20))
self.lb_info = SettingsUtils.createInfoLabel("")
self.lb_info2 = SettingsUtils.createInfoLabel("")
self.sp2 = SettingsUtils.createSeparators(1)
self.p_method = SettingsUtils.createPanel()
self.bg_method = ButtonGroup()
autopsy_version = PsyUtils.get_autopsy_version()
if ((autopsy_version["major"] == 4 and autopsy_version["minor"] <= 17)
or autopsy_version["major"] < 4):
self.p_info.add(self.lb_info)
self.p_info.add(self.lb_info2, BorderLayout.SOUTH)
self.rb_selectedDatasource = SettingsUtils.createRadioButton(
"Analyze selected datasource", "method_datasource",
self.onMethodChange)
self.bg_method.add(self.rb_selectedDatasource)
# self.rb_importReportFile = SettingsUtils.createRadioButton("Import previous generated report file","method_importfile" ,self.onMethodChange)
self.rb_liveExtraction = SettingsUtils.createRadioButton(
"Live extraction with ADB", "method_adb", self.onMethodChange)
self.rb_selectedDatasource.setSelected(True)
#self.bg_method.add(self.rb_importReportFile)
self.bg_method.add(self.rb_liveExtraction)
self.p_method.add(JLabel("Analysis method"))
self.p_method.add(self.rb_selectedDatasource)
self.p_method.add(self.rb_liveExtraction)
else:
self.p_info.add(
SettingsUtils.createInfoLabel(
"It will analyze the data source with previously selected method and index the forensic artifacts."
))
self.add(self.p_method)
self.p_apps = SettingsUtils.createPanel(True)
sorted_items = OrderedDict(sorted(Utils.get_all_packages().items()))
for app, app_id in sorted_items.iteritems():
#(app, app_id)
checkbox = SettingsUtils.addApplicationCheckbox(
app, app_id, self.getSelectedApps)
self.add(checkbox)
self.apps_checkboxes_list.append(checkbox)
self.p_apps.add(checkbox)
self.add(self.p_apps)
self.add(self.p_info)
# end of checkboxes menu
def customizeComponents(self):
self.onMethodChange("") #initialize method option
self.getSelectedApps("") #initialize selected apps
def onMethodChange(self, event):
self.method = self.getMethod()
self.local_settings.setSetting("method", self.method)
if self.method == "method_adb":
self.lb_info.setText(
"This method is used when there is no data source but you have the device."
)
self.lb_info2.setText(
"It will extract the content of the selected applications from the device, analyze and index the forensic artifacts."
)
self.toggleCheckboxes(True)
else:
self.lb_info.setText(
"This method is used when the application data has already been collected."
)
self.lb_info2.setText(
"It will analyze the data source previously added to the data source and index the forensic artifacts."
)
self.toggleCheckboxes(False)
def getSettings(self):
return self.local_settings
def getMethod(self):
selection = self.bg_method.getSelection()
if not selection:
return None
return selection.getActionCommand()
def getSelectedApps(self, event):
selected_apps = []
for cb_app in self.apps_checkboxes_list:
if cb_app.isSelected():
selected_apps.append(cb_app.getActionCommand())
self.local_settings.setSetting("apps", json.dumps(selected_apps))
def toggleCheckboxes(self, visible):
for cb_app in self.apps_checkboxes_list:
cb_app.setVisible(visible)
def openGUI(self, invocation):
try:
# Get values from request or response the extension is invoked from and prepopulate GUI values
invMessage = invocation.getSelectedMessages()
message = invMessage[0]
originalHttpService = message.getHttpService()
self.originalMsgProtocol = originalHttpService.getProtocol()
self.originalMsgHost = originalHttpService.getHost()
self.originalMsgPort = originalHttpService.getPort()
except:
self.originalMsgProtocol = ''
self.originalMsgHost = ''
self.originalMsgPort = ''
try:
self.cookies = self._callbacks.getCookieJarContents()
self.cookie = ''
except:
pass
self.SSL = 'http://'
self.listType = ''
self.parsedList = []
# Set up main window (JFrame)
self.window = JFrame("Directory Listing Parser for Burp Suite", preferredSize=(600, 475), windowClosing=self.closeUI)
self.window.setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE)
emptyBorder = BorderFactory.createEmptyBorder(10, 10, 10, 10)
self.window.contentPane.setBorder(emptyBorder)
self.window.contentPane.layout = BorderLayout()
# Main window title placed at the top of the main window with an invisible bottom border
titlePanel = JPanel()
titleBorder = BorderFactory.createEmptyBorder(0, 0, 10, 0)
title = JLabel("Directory Listing Parser for Burp Suite", JLabel.CENTER)
title.setBorder(titleBorder)
title.setFont(Font("Default", Font.PLAIN, 18))
titlePanel.add(title)
self.window.contentPane.add("North", titlePanel)
# Left panel for user input, consisting of hostname, directory prefix, ssl, port, type of listing, and file
self.leftPanel = JPanel()
self.leftPanel.layout = GridLayout(14, 1, 3, 3)
hostnameLabel = JLabel("Hostname:")
if self.originalMsgHost:
self.hostnameTextField = JTextField(self.originalMsgHost.rstrip())
else:
self.hostnameTextField = JTextField('Hostname')
dirPrefixLabel = JLabel("Full Directory Prefix (Windows):")
self.dirPrefixField = JTextField('C:\\var\www\\')
sslLabel = JLabel("SSL:")
self.radioBtnSslEnabled = JRadioButton('Enabled (https)', actionPerformed=self.radioSsl)
self.radioBtnSslDisabled = JRadioButton('Disabled (http)', actionPerformed=self.radioSsl)
sslButtonGroup = ButtonGroup()
sslButtonGroup.add(self.radioBtnSslEnabled)
sslButtonGroup.add(self.radioBtnSslDisabled)
if self.originalMsgProtocol == "https":
self.radioBtnSslEnabled.setSelected(True)
else:
self.radioBtnSslDisabled.setSelected(True)
portLabel = JLabel("Port:")
if self.originalMsgPort:
self.portTextField = JTextField(str(self.originalMsgPort).rstrip())
else:
self.portTextField = JTextField('80')
osLabel = JLabel("Type of File Listing:")
self.types = ('Windows \'dir /s\'', 'Linux \'ls -lR\'', 'Linux \'ls -R\'')
self.comboListingType = JComboBox(self.types)
uploadLabel = JLabel("Directory Listing File:")
self.uploadTextField = JTextField('')
uploadButton = JButton('Choose File', actionPerformed=self.chooseFile)
self.leftPanel.add(hostnameLabel)
self.leftPanel.add(self.hostnameTextField)
self.leftPanel.add(dirPrefixLabel)
self.leftPanel.add(self.dirPrefixField)
self.leftPanel.add(sslLabel)
self.leftPanel.add(self.radioBtnSslEnabled)
self.leftPanel.add(self.radioBtnSslDisabled)
self.leftPanel.add(portLabel)
self.leftPanel.add(self.portTextField)
self.leftPanel.add(osLabel)
self.leftPanel.add(self.comboListingType)
self.leftPanel.add(uploadLabel)
self.leftPanel.add(self.uploadTextField)
self.leftPanel.add(uploadButton)
# Right panel consisting of a text area for the URL list
self.UrlPanelLabel = JLabel("URL List:")
self.textArea = JTextArea()
self.textArea.setEditable(True)
self.textArea.setFont(Font("Default", Font.PLAIN, 14))
if self.cookies:
self.textArea.append('Cookies Found:\n')
for cookie in self.cookies:
if cookie.getDomain() in self.originalMsgHost:
self.cookie += cookie.getName() + '=' + cookie.getValue() + '; '
self.textArea.append(cookie.getName() + '=' + cookie.getValue() + '\n')
scrollArea = JScrollPane(self.textArea)
scrollArea.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_ALWAYS)
scrollArea.setPreferredSize(Dimension(400, 200))
self.rightPanel = JPanel()
self.rightPanel.setLayout(BorderLayout(3, 3))
self.rightPanel.add(self.UrlPanelLabel, BorderLayout.NORTH)
self.rightPanel.add(scrollArea, BorderLayout.CENTER)
# Panel for the generate URL list and import URL list buttons
generatePanel = JPanel()
generatePanel.layout = BorderLayout(3, 3)
generateButton = JButton('Generate URL List', actionPerformed=self.generateUrlList)
importButton = JButton('Import URL List to Burp Site Map', actionPerformed=self.confirmImport)
generatePanel.add("North", generateButton)
generatePanel.add("South", importButton)
self.rightPanel.add("South", generatePanel)
# Add the two main panels to the left and right sides
self.window.contentPane.add("East", self.rightPanel)
self.window.contentPane.add("West", self.leftPanel)
# Create a panel to be used for the file chooser window
self.uploadPanel = JPanel()
self.window.pack()
self.window.show()
def getUiComponent(self):
aws_access_key_id = self.callbacks.loadExtensionSetting(
"aws_access_key_id")
aws_secret_accesskey = self.callbacks.loadExtensionSetting(
"aws_secret_access_key")
if aws_access_key_id:
self.aws_access_key_id = aws_access_key_id
if aws_secret_accesskey:
self.aws_secret_accesskey = aws_secret_accesskey
self.panel = JPanel()
self.main = JPanel()
self.main.setLayout(BoxLayout(self.main, BoxLayout.Y_AXIS))
self.access_key_panel = JPanel()
self.main.add(self.access_key_panel)
self.access_key_panel.setLayout(
BoxLayout(self.access_key_panel, BoxLayout.X_AXIS))
self.access_key_panel.add(JLabel('Access Key: '))
self.access_key = JTextField(self.aws_access_key_id, 25)
self.access_key_panel.add(self.access_key)
self.secret_key_panel = JPanel()
self.main.add(self.secret_key_panel)
self.secret_key_panel.setLayout(
BoxLayout(self.secret_key_panel, BoxLayout.X_AXIS))
self.secret_key_panel.add(JLabel('Secret Key: '))
self.secret_key = JPasswordField(self.aws_secret_accesskey, 25)
self.secret_key_panel.add(self.secret_key)
self.target_host_panel = JPanel()
self.main.add(self.target_host_panel)
self.target_host_panel.setLayout(
BoxLayout(self.target_host_panel, BoxLayout.X_AXIS))
self.target_host_panel.add(JLabel('Target host: '))
self.target_host = JTextField('ifconfig.io', 25)
self.target_host_panel.add(self.target_host)
self.buttons_panel = JPanel()
self.main.add(self.buttons_panel)
self.buttons_panel.setLayout(
BoxLayout(self.buttons_panel, BoxLayout.X_AXIS))
self.save_button = JButton('Save Keys', actionPerformed=self.saveKeys)
self.buttons_panel.add(self.save_button)
self.enable_button = JButton('Enable',
actionPerformed=self.enableGateway)
self.buttons_panel.add(self.enable_button)
self.disable_button = JButton('Disable',
actionPerformed=self.disableGateway)
self.buttons_panel.add(self.disable_button)
self.disable_button.setEnabled(False)
self.protocol_panel = JPanel()
self.main.add(self.protocol_panel)
self.protocol_panel.setLayout(
BoxLayout(self.protocol_panel, BoxLayout.Y_AXIS))
self.protocol_panel.add(JLabel("Target Protocol:"))
self.https_button = JRadioButton("HTTPS", True)
self.http_button = JRadioButton("HTTP", False)
self.protocol_panel.add(self.http_button)
self.protocol_panel.add(self.https_button)
buttongroup = ButtonGroup()
buttongroup.add(self.https_button)
buttongroup.add(self.http_button)
self.regions_title = JPanel()
self.main.add(self.regions_title)
self.regions_title.add(JLabel("Regions to launch API Gateways in:"))
self.regions_panel = JPanel()
self.main.add(self.regions_panel)
glayout = GridLayout(4, 3)
self.regions_panel.setLayout(glayout)
for region in AVAIL_REGIONS:
cur_region = region.replace('-', '_')
cur_region = cur_region + '_status'
if cur_region.startswith(
"ap") and cur_region != 'ap_east_1_status':
setattr(self, cur_region, JCheckBox(region, True))
else:
setattr(self, cur_region, JCheckBox(region, False))
attr = getattr(self, cur_region)
self.regions_panel.add(attr)
self.status = JPanel()
self.main.add(self.status)
self.status.setLayout(BoxLayout(self.status, BoxLayout.X_AXIS))
self.status_indicator = JLabel(DISABLED, JLabel.CENTER)
self.status.add(self.status_indicator)
self.panel.add(self.main)
return self.panel
class InductionApplet(JApplet):
def init(self):
global exampleList
self.thinFont = Font("Dialog", 0, 10)
self.pane = self.getContentPane()
self.examples = exampleList.keys()
self.examples.sort()
self.exampleSelector = JList(self.examples, valueChanged=self.valueChanged)
self.exampleSelector.setSelectionMode(ListSelectionModel.SINGLE_SELECTION)
self.exampleSelector.setLayoutOrientation(JList.VERTICAL)
self.exampleSelector.setPreferredSize(Dimension(150,500))
self.exampleSelector.setBackground(Color(0.95, 0.95, 0.98))
self.exampleSelector.setFont(self.thinFont)
self.centerPanel = JPanel(BorderLayout())
self.canvas = GraphCanvas()
self.canvas.setApplet(self)
self.buttonRow = JPanel(FlowLayout())
self.backButton = JButton("<", actionPerformed = self.backAction)
self.backButton.setFont(self.thinFont)
self.continueButton = JButton("continue >",
actionPerformed=self.continueAction)
self.continueButton.setFont(self.thinFont)
self.scaleGroup = ButtonGroup()
self.linearButton = JRadioButton("linear scale",
actionPerformed=self.linearAction)
self.linearButton.setSelected(True)
self.linearButton.setFont(self.thinFont)
self.logarithmicButton = JRadioButton("logarithmic scale",
actionPerformed=self.logarithmicAction)
self.logarithmicButton.setFont(self.thinFont)
self.aboutButton = JButton("About...",
actionPerformed=self.aboutAction)
self.aboutButton.setFont(self.thinFont)
self.scaleGroup.add(self.linearButton)
self.scaleGroup.add(self.logarithmicButton)
self.buttonRow.add(self.backButton)
self.buttonRow.add(self.continueButton)
self.buttonRow.add(JLabel(" "*5))
self.buttonRow.add(self.linearButton)
self.buttonRow.add(self.logarithmicButton)
self.buttonRow.add(JLabel(" "*20));
self.buttonRow.add(self.aboutButton)
self.centerPanel.add(self.canvas, BorderLayout.CENTER)
self.centerPanel.add(self.buttonRow, BorderLayout.PAGE_END)
self.helpText = JTextPane()
self.helpText.setBackground(Color(1.0, 1.0, 0.5))
self.helpText.setPreferredSize(Dimension(800,80))
self.helpText.setText(re_sub("[ \\n]+", " ", """
Please select one of the examples in the list on the left!
"""))
self.pane.add(self.exampleSelector, BorderLayout.LINE_START)
self.pane.add(self.centerPanel, BorderLayout.CENTER)
self.pane.add(self.helpText, BorderLayout.PAGE_END)
self.graph = None
self.simulation = None
self.touched = ""
self.selected = ""
self.gfxDriver = None
def start(self):
self.gfxDriver = awtGfx.Driver(self.canvas)
#self.gfxDriver.setAntialias(True)
if self.gfxDriver.getSize()[0] < 200: # konqueror java bug work around
self.gfxDriver.w = 650
self.gfxDriver.h = 380
self.graph = Graph.Cartesian(self.gfxDriver, 1, 0.0, 1000, 1.0,
title="Results",
xaxis="Rounds", yaxis="Success Rate")
def stop(self):
pass
def destroy(self):
pass
def refresh(self):
if self.graph != None: self.graph.redraw()
def valueChanged(self, e):
global exampleList
newSelection = self.examples[self.exampleSelector.getSelectedIndex()]
if newSelection != self.touched:
self.touched = newSelection
text = re_sub("[ \\n]+", " ", exampleList[self.touched][-1])
self.helpText.setText(text)
if not e.getValueIsAdjusting() and newSelection != self.selected:
self.selected = newSelection
smallFontPen = copy.copy(Gfx.BLACK_PEN)
smallFontPen.fontSize = Gfx.SMALL
ex = exampleList[self.selected]
myStyleFlags = self.graph.styleFlags
if self.simulation != None: self.simulation.stop()
self.gfxDriver.resizedGfx() # konqueror 3.5.5 java bug workaround
self.graph = Graph.Cartesian(self.gfxDriver, 1, 0.0, ex[3], 1.0,
title=ex[0],
xaxis="Rounds", yaxis="Success Rate",
styleFlags = myStyleFlags,
axisPen = smallFontPen,
captionPen = smallFontPen)
self.zoomFrame = [(1, 0.0, ex[3], 1.0)]
self.simulation = Simulation(self.graph, ex[1], ex[2], ex[3], ex[4])
RunAsThread(self.simulation.simulation).start()
def determineCurrentZoomFrame(self):
i = 0
for zf in self.zoomFrame:
if self.graph.x2 <= zf[2]: break
i += 1
return i
def backAction(self, e):
if self.simulation == None: return
wasRunning = self.simulation.isRunning
self.simulation.stop()
if wasRunning or len(self.zoomFrame) <= 1: return
zi = self.determineCurrentZoomFrame()
if zi > 0 and zi < len(self.zoomFrame):
x1, y1, x2, y2 = self.zoomFrame[zi-1]
self.graph.adjustRange(x1, y1, x2, y2)
def continueAction(self, e):
if self.simulation == None: return
wasRunning = self.simulation.isRunning
self.simulation.stop()
zi = self.determineCurrentZoomFrame()
if zi == len(self.zoomFrame)-1:
if wasRunning or self.simulation.world.round == self.zoomFrame[zi][2]:
if self.graph.styleFlags & Graph.LOG_X == 0:
self.simulation.rounds *= 2
else:
self.simulation.rounds *= 10
self.zoomFrame.append((1, 0.0, self.simulation.rounds, 1.0))
self.graph.adjustRange(1, 0.0, self.simulation.rounds, 1.0)
RunAsThread(self.simulation.simulation).start()
else:
x1, y1, x2, y2 = self.zoomFrame[zi+1]
self.graph.adjustRange(x1, y1, x2, y2)
def linearAction(self, e):
if self.graph != None and (self.graph.styleFlags & Graph.LOG_X) != 0:
if self.simulation != None: self.simulation.stop()
self.graph.setStyle(self.graph.styleFlags & ~Graph.LOG_X, redraw=True)
if self.simulation != None:
RunAsThread(self.simulation.simulation).start()
def logarithmicAction(self, e):
if self.graph != None and (self.graph.styleFlags & Graph.LOG_X) == 0:
if self.simulation != None: self.simulation.stop()
self.graph.setStyle(self.graph.styleFlags | Graph.LOG_X, redraw=True)
if self.simulation != None:
RunAsThread(self.simulation.simulation).start()
def aboutAction(self, e):
aboutText = """Induction Applet v. 0.1
(c) 2007 University of Düsseldorf
Authors: Gerhard Schurz, Eckhart Arnold
"""
aboutText = re_sub(" +", " ", aboutText)
JOptionPane.showMessageDialog(self.getContentPane(), aboutText)
