196 lines
8.7 KiB
Java
196 lines
8.7 KiB
Java
package burp.ui;
|
|
|
|
import burp.yaml.SetRuleConfig;
|
|
|
|
import javax.swing.*;
|
|
import javax.swing.event.TableModelEvent;
|
|
import javax.swing.event.TableModelListener;
|
|
import javax.swing.table.DefaultTableModel;
|
|
import javax.swing.table.TableRowSorter;
|
|
import java.awt.*;
|
|
import java.awt.event.MouseAdapter;
|
|
import java.awt.event.MouseEvent;
|
|
import java.util.Vector;
|
|
|
|
/*
|
|
* @author LinChen
|
|
*/
|
|
|
|
public class RulePane extends JPanel {
|
|
public RulePane(Object[][] data,JTabbedPane pane) {
|
|
initComponents(data,pane);
|
|
}
|
|
private SetRuleConfig setruleconfig = new SetRuleConfig();
|
|
private Boolean isEdit = false;
|
|
private void RuleAddMouseClicked(MouseEvent e, JTabbedPane pane) {
|
|
// TODO add your code here
|
|
RuleSetting add = new RuleSetting();
|
|
int isOk = JOptionPane.showConfirmDialog(null,add,"RuleSetting - Add Rule",JOptionPane.OK_OPTION);
|
|
if(isOk == 0){
|
|
model.addRow(new Object[0]);
|
|
model.setValueAt(false,(model.getRowCount()-1),0);
|
|
model.setValueAt(add.Name.getText(),(model.getRowCount()-1),1);
|
|
model.setValueAt(add.Regex.getText(),(model.getRowCount()-1),2);
|
|
model.setValueAt(add.ColorSelect.getSelectedItem().toString(),(model.getRowCount()-1),3);
|
|
model.setValueAt(add.ScopeSelect.getSelectedItem().toString(),(model.getRowCount()-1),4);
|
|
model.setValueAt(add.EngineSelect.getSelectedItem().toString(),(model.getRowCount()-1),5);
|
|
model = (DefaultTableModel) table.getModel();
|
|
int select = table.convertRowIndexToModel(table.getSelectedRow());
|
|
setruleconfig.add((Vector) model.getDataVector().get(select),pane.getTitleAt(pane.getSelectedIndex()));
|
|
}
|
|
}
|
|
private void RuleEditMouseClicked(MouseEvent e,JTabbedPane pane){
|
|
if (table.getSelectedRowCount()>=1){
|
|
RuleSetting edit = new RuleSetting();
|
|
edit.Name.setText(table.getValueAt(table.getSelectedRow(),1).toString());
|
|
edit.Regex.setText(table.getValueAt(table.getSelectedRow(),2).toString());
|
|
edit.ColorSelect.setSelectedItem(table.getValueAt(table.getSelectedRow(),3).toString());
|
|
edit.ScopeSelect.setSelectedItem(table.getValueAt(table.getSelectedRow(),4).toString());
|
|
edit.EngineSelect.setSelectedItem(table.getValueAt(table.getSelectedRow(),5).toString());
|
|
int isOk = JOptionPane.showConfirmDialog(null,edit,"RuleSetting - Edit Rule",JOptionPane.OK_OPTION);
|
|
if (isOk ==0){
|
|
int select = table.convertRowIndexToModel(table.getSelectedRow());
|
|
model.setValueAt(edit.Name.getText(),select,1);
|
|
model.setValueAt(edit.Regex.getText(),select,2);
|
|
model.setValueAt(edit.ColorSelect.getSelectedItem().toString(),select,3);
|
|
model.setValueAt(edit.ScopeSelect.getSelectedItem().toString(),select,4);
|
|
model.setValueAt(edit.EngineSelect.getSelectedItem().toString(),select,5);
|
|
model = (DefaultTableModel) table.getModel();
|
|
setruleconfig.edit((Vector) model.getDataVector().get(select),select,pane.getTitleAt(pane.getSelectedIndex()));
|
|
}
|
|
}
|
|
}
|
|
private void RuleRemoveMouseClicked(MouseEvent e,JTabbedPane pane){
|
|
if (table.getSelectedRowCount()>=1){
|
|
int isOk = JOptionPane.showConfirmDialog(null,"Are your sure?","RuleSetting - Delete Rule",JOptionPane.OK_OPTION);
|
|
if (isOk==0){
|
|
model.removeRow(table.convertRowIndexToModel(table.getSelectedRow()));
|
|
table.remove(table.getSelectedRow());
|
|
model = (DefaultTableModel) table.getModel();
|
|
int select = table.convertRowIndexToModel(table.getSelectedRow());
|
|
setruleconfig.remove(select,pane.getTitleAt(pane.getSelectedIndex()));
|
|
}
|
|
}
|
|
}
|
|
private void RuleTableChange(TableModelEvent e,JTabbedPane pane) {
|
|
if (e.getColumn()==0&&table.getSelectedRow()!=-1&&!isEdit){
|
|
model = (DefaultTableModel) table.getModel();
|
|
int select = table.convertRowIndexToModel(table.getSelectedRow());
|
|
setruleconfig.edit((Vector) model.getDataVector().get(select),select,pane.getTitleAt(pane.getSelectedIndex()));
|
|
}
|
|
}
|
|
|
|
private void initComponents(Object[][] data,JTabbedPane pane) {
|
|
// JFormDesigner - Component initialization - DO NOT MODIFY //GEN-BEGIN:initComponents
|
|
RuleAdd = new JButton();
|
|
RuleEdit = new JButton();
|
|
scrollPane = new JScrollPane();
|
|
table = new JTable();
|
|
Remove = new JButton();
|
|
|
|
//======== this ========
|
|
setLayout(new GridBagLayout());
|
|
((GridBagLayout)getLayout()).columnWidths = new int[] {0, 0, 0};
|
|
((GridBagLayout)getLayout()).rowHeights = new int[] {0, 0, 0, 0, 0};
|
|
((GridBagLayout)getLayout()).columnWeights = new double[] {0.0, 1.0, 1.0E-4};
|
|
((GridBagLayout)getLayout()).rowWeights = new double[] {0.0, 0.0, 0.0, 1.0, 1.0E-4};
|
|
|
|
//---- RuleAdd ----
|
|
RuleAdd.setText("Add");
|
|
RuleAdd.addMouseListener(new MouseAdapter() {
|
|
@Override
|
|
public void mouseClicked(MouseEvent e) {
|
|
isEdit = true;
|
|
RuleAddMouseClicked(e,pane);
|
|
model = (DefaultTableModel) table.getModel();
|
|
isEdit = false;
|
|
}
|
|
});
|
|
add(RuleAdd, new GridBagConstraints(0, 0, 1, 1, 0.0, 0.0,
|
|
GridBagConstraints.CENTER, GridBagConstraints.BOTH,
|
|
new Insets(15, 5, 3, 2), 0, 0));
|
|
|
|
//---- RuleEdit ----
|
|
RuleEdit.setText("Edit");
|
|
RuleEdit.addMouseListener(new MouseAdapter() {
|
|
@Override
|
|
public void mouseClicked(MouseEvent e) {
|
|
isEdit = true;
|
|
RuleEditMouseClicked(e,pane);
|
|
model = (DefaultTableModel) table.getModel();
|
|
isEdit = false;
|
|
}
|
|
});
|
|
add(RuleEdit, new GridBagConstraints(0, 1, 1, 1, 0.0, 0.0,
|
|
GridBagConstraints.CENTER, GridBagConstraints.BOTH,
|
|
new Insets(0, 5, 3, 2), 0, 0));
|
|
|
|
//======== scrollPane ========
|
|
{
|
|
|
|
//---- table ----
|
|
table.setShowVerticalLines(false);
|
|
table.setVerifyInputWhenFocusTarget(false);
|
|
table.setUpdateSelectionOnSort(false);
|
|
table.setShowHorizontalLines(false);
|
|
table.setModel(new DefaultTableModel());
|
|
table.setSurrendersFocusOnKeystroke(true);
|
|
scrollPane.setViewportView(table);
|
|
}
|
|
add(scrollPane, new GridBagConstraints(1, 0, 1, 4, 0.0, 0.0,
|
|
GridBagConstraints.CENTER, GridBagConstraints.BOTH,
|
|
new Insets(15, 5, 5, 5), 0, 0));
|
|
|
|
//---- Remove ----
|
|
Remove.setText("Remove");
|
|
Remove.addMouseListener(new MouseAdapter() {
|
|
@Override
|
|
public void mouseClicked(MouseEvent e) {
|
|
isEdit = true;
|
|
RuleRemoveMouseClicked(e,pane);
|
|
model = (DefaultTableModel) table.getModel();
|
|
isEdit = false;
|
|
}
|
|
});
|
|
add(Remove, new GridBagConstraints(0, 2, 1, 1, 0.0, 0.0,
|
|
GridBagConstraints.CENTER, GridBagConstraints.BOTH,
|
|
new Insets(0, 5, 3, 2), 0, 0));
|
|
// JFormDesigner - End of component initialization //GEN-END:initComponents
|
|
table.setModel(model);
|
|
model.setDataVector(data,title);
|
|
model.addTableModelListener(new TableModelListener() {
|
|
@Override
|
|
public void tableChanged(TableModelEvent e) {
|
|
RuleTableChange(e,pane);
|
|
}
|
|
});
|
|
table.setRowSorter(new TableRowSorter(model));
|
|
}
|
|
|
|
// JFormDesigner - Variables declaration - DO NOT MODIFY //GEN-BEGIN:variables
|
|
public JButton RuleAdd;
|
|
public JButton RuleEdit;
|
|
public JScrollPane scrollPane;
|
|
public JTable table;
|
|
public JButton Remove;
|
|
// JFormDesigner - End of variables declaration //GEN-END:variables
|
|
private final String[] title = new String[]{"Loaded", "Name", "Regex", "Color", "Scope", "Engine"};
|
|
private DefaultTableModel model = new DefaultTableModel() {
|
|
public Class<?> getColumnClass ( int column){
|
|
if (column == 0) {
|
|
return Boolean.class;
|
|
}else{
|
|
return String.class;
|
|
}
|
|
}
|
|
public boolean isCellEditable(int row,int column){
|
|
if (column ==0){
|
|
return true;
|
|
}else {
|
|
return false;
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|