def _cmd_add(self, params): if len(params) == 0: om.out.console('Parameter "type" is missing, see the help:') self._cmd_help(['add']) return if len(params) > 1: om.out.console('Only one parameter is accepted, see the help:') self._cmd_help(['add']) return template_name = params[0] if template_name not in get_template_names(): om.out.console('Type %s is unknown' % template_name) return # Now we use the fact that templates are configurable just like # plugins, misc-settings, etc. template_inst = get_template_by_name(template_name) template_menu = StoreOnBackConfigMenu(template_name, self._console, self._w3af, self, template_inst) # Note: The data is stored in the KB when the user does a "back" # see the StoreOnBackConfigMenu implementation return template_menu
def _cmd_add(self, params): if len(params) == 0: om.out.console('Parameter "type" is missing, see the help:') self._cmd_help(['add']) return if len(params) > 1: om.out.console('Only one parameter is accepted, see the help:') self._cmd_help(['add']) return template_name = params[0] if template_name not in get_template_names(): om.out.console('Type %s is unknown' % template_name) return # Now we use the fact that templates are configurable just like # plugins, misc-settings, etc. template_inst = get_template_by_name(template_name) template_menu = StoreOnBackConfigMenu(template_name, self._console, self._w3af, self, template_inst) # Note: The data is stored in the KB when the user does a "back" # see the StoreOnBackConfigMenu implementation return template_menu
def __init__(self, w3af): super(VulnAddDialog, self).__init__("Add new vulnerability", None, gtk.MESSAGE_QUESTION, (gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_ADD, gtk.RESPONSE_OK)) self.set_icon_from_file(W3AF_ICON) self.w3af = w3af # Main hbox hbox = gtk.HBox() hbox.show() image = gtk.Image() image.set_from_stock(gtk.STOCK_DIALOG_QUESTION, gtk.ICON_SIZE_DIALOG) image.set_padding(20, 20) image.show() hbox.pack_start(image) # Vbox with all the information for the right section of the dialog vbox = gtk.VBox() # Add a label align = gtk.Alignment() align.set_padding(10, 0, 0, 10) label = gtk.Label('Choose the vulnerability template to use:') align.add(label) vbox.pack_start(align) template_long_names = get_template_long_names() template_names = get_template_names() self.vuln_template = None # A list store with the following columns: # * Long template name (show) # * Template name (code internals) liststore = gtk.ListStore(str, str) self.combobox = gtk.ComboBox(liststore) cell = gtk.CellRendererText() self.combobox.pack_start(cell, True) self.combobox.add_attribute(cell, 'text', 0) self.combobox.connect("changed", self._changed_combo) for i, long_name in enumerate(template_long_names): liststore.append((long_name, template_names[i])) vbox.pack_start(self.combobox, False, False) vbox.pack_start(gtk.Label()) # the Cancel button but = self.action_area.get_children()[1] but.connect("clicked", lambda x: self.destroy()) self.connect("delete-event", lambda x, y: self.destroy()) # the Ok button self.ok_but = self.action_area.get_children()[0] self.ok_but.connect("clicked", self._ok) self.ok_but.set_sensitive(False) hbox.pack_start(vbox) self.vbox.pack_start(hbox) self.show_all()
def __init__(self, w3af): super(VulnAddDialog, self).__init__("Add new vulnerability", None, gtk.MESSAGE_QUESTION, (gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_ADD, gtk.RESPONSE_OK)) self.set_icon_from_file(W3AF_ICON) self.w3af = w3af # Main hbox hbox = gtk.HBox() hbox.show() image = gtk.Image() image.set_from_stock(gtk.STOCK_DIALOG_QUESTION, gtk.ICON_SIZE_DIALOG) image.set_padding(20, 20) image.show() hbox.pack_start(image) # Vbox with all the information for the right section of the dialog vbox = gtk.VBox() # Add a label align = gtk.Alignment() align.set_padding(10, 0, 0, 10) label = gtk.Label('Choose the vulnerability template to use:') align.add(label) vbox.pack_start(align) template_long_names = get_template_long_names() template_names = get_template_names() self.vuln_template = None # A list store with the following columns: # * Long template name (show) # * Template name (code internals) liststore = gtk.ListStore(str, str) self.combobox = gtk.ComboBox(liststore) cell = gtk.CellRendererText() self.combobox.pack_start(cell, True) self.combobox.add_attribute(cell, 'text', 0) self.combobox.connect("changed", self._changed_combo) for i, long_name in enumerate(template_long_names): liststore.append((long_name, template_names[i])) vbox.pack_start(self.combobox, False, False) vbox.pack_start(gtk.Label()) # the Cancel button but = self.action_area.get_children()[1] but.connect("clicked", lambda x: self.destroy()) self.connect("delete-event", lambda x, y: self.destroy()) # the Ok button self.ok_but = self.action_area.get_children()[0] self.ok_but.connect("clicked", self._ok) self.ok_but.set_sensitive(False) hbox.pack_start(vbox) self.vbox.pack_start(hbox) self.show_all()
def test_get_template_names(self): self.assertIn('dav', get_template_names())
def _para_add(self, params, part): if len(params): return [] return suggest(get_template_names(), part)
def test_get_template_names(self): self.assertIn('dav', get_template_names())
def _para_add(self, params, part): if len(params): return [] return suggest(get_template_names(), part)