Changeset 5745

Timestamp:

07/26/10 22:41:55 (3 years ago)

Author:

diogo

Message:

NSEFacilitator: create script from template; save as template

Location:

network-scanner/branches/NSEFacilitator/umitNSEFacilitator

Files:

• ScriptEditor.py (modified) (13 diffs)
• luaParser.py (modified) (3 diffs)

network-scanner/branches/NSEFacilitator/umitNSEFacilitator/ScriptEditor.py

@@ -36 +36 @@
 from umitCore.Paths import Path, check_access
 from umitGUI.BugReport import BugReport
+from luaParser import LuaParser
 
 # try to use gtksourceview if available
@@ -77 +78 @@
 DEFAULT_SELECTED_TEXT_COLOR = gtk.gdk.color_parse("#ffffffffffff")
 DEFAULT_SELECTION_COLOR = gtk.gdk.color_parse("#000000009c9c")
+DEFAULT_TEMPLATE_PATH = os.path.expanduser("~/.umit/nsetemplates")
+
+def get_templates():
+    """
+    Return the list of user templates in nsetemplates folder.
+    """
+    templates = []
+    for template in os.listdir(DEFAULT_TEMPLATE_PATH):
+        ext = template.split(".")[-1]
+        if ext == 'nse':
+            templates.append(template)
+    return templates
 
 class ScriptEditorPageManager(object):
@@ -482 +495 @@
             f.close()
             self.text_buffer.set_modified(False)
+            
+    def save_as_template(self):
+        """
+        Save a script as a template.
+        """
+        file_chooser = SaveNSEFileChooserDialog(_("Save As Template..."))
+        file_chooser.set_current_folder(DEFAULT_TEMPLATE_PATH)
+        if not self.is_untitled():
+            file_chooser.set_filename(self.scriptname)
+        else:
+            file_chooser.set_current_name(self.untitled_name)
+        response = file_chooser.run()
+        filename = file_chooser.get_filename()
+        file_chooser.destroy()
+        if not response == gtk.RESPONSE_OK:
+            return
+        self.set_name(filename)
+        self.save(True)            
         
 class ScriptEditorWindow(ScriptWindow):
@@ -550 +581 @@
              _('Save _As...'), "<Shift><Control>S",
              _('Save the current script file with the different name'), self._save_as_cb),
+            
+            ('Save As Template', gtk.STOCK_SAVE_AS,
+             _('Save As Template'), None,
+             _('Save the current script file as template'), self._save_as_template_cb),
 
             ('Close', gtk.STOCK_CLOSE,
@@ -650 +685 @@
             <menuitem action='Save'/>
             <menuitem action='Save As'/>
+            <menuitem action='Save As Template'/>
             <separator/>
             <menuitem action='Close'/>
@@ -811 +847 @@
         page.save_as()
         self.update_title()
+        
+    def _save_as_template_cb(self, p):
+        page = self.get_current_page()
+        if not page:
+            return
+        page.save_as_template()
+        self.update_title()
 
     def _quit_cb(self, p):
@@ -999 +1042 @@
             return boxes
             
-        self.vbox.pack_start(HIGSectionLabel(_("General")), False, False)
+        scrolled_window = gtk.ScrolledWindow()
+        scrolled_window.set_border_width(5)
+        scrolled_window.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_ALWAYS)
+        self.vbox.pack_start(scrolled_window, True, True, 0)
+        scrolled_window.show()
+        
+        self.vbox1 = gtk.VBox()
+        self.vbox1.pack_start(HIGSectionLabel(_("General")), False, False)
         table = HIGTable(5, 2)
         self.id_entry = create_entry(table, 0, "<b>ID:</b>")
@@ -1013 +1063 @@
         hbox.pack_start(self.host_rule)
         table.attach(hbox, 1, 2, 4, 5)
-        self.vbox.pack_start(HIGSpacer(table), False, False)
-
-        self.vbox.pack_start(HIGSectionLabel(_("Categories")), False, False)
+        self.vbox1.pack_start(HIGSpacer(table), False, False)
+        self.vbox1.pack_start(HIGSpacer())
+
+        self.vbox1.pack_start(HIGSectionLabel(_("Categories")), False, False)
         table = HIGTable(4, 2)
         self.categories = create_checkboxes(table,
@@ -1026 +1077 @@
                                              "version",
                                              "vulnerability"])
-        self.vbox.pack_start(HIGSpacer(table), False, False)
-
-        self.vbox.pack_start(HIGSectionLabel(_("Description")), False, False)
+        self.vbox1.pack_start(HIGSpacer(table), False, False)
+        self.vbox1.pack_start(HIGSpacer())
+
+        self.vbox1.pack_start(HIGSectionLabel(_("User Template")), False, False)
+        table = HIGTable(1, 2)
+        self.templates = gtk.combo_box_new_text()
+        self.templates.append_text("")
+        for template in get_templates():
+            self.templates.append_text(template)
+        self.templates.connect('changed', self._template_changed_cb)
+        table.attach(self.templates, 0, 1, 0, 1)
+        lbl = HIGEntryLabel("")
+        table.attach(lbl, 1, 2, 0, 1)
+        self.vbox1.pack_start(table, False, False)
+        self.vbox1.pack_start(HIGSpacer(table), False, False)
+
+        self.vbox1.pack_start(HIGSectionLabel(_("Description")), False, False)
         self.desc_view = HIGTextView()
-        self.vbox.pack_start(HIGSpacer(scroll_wrap(self.desc_view)))
+        self.vbox1.pack_start(HIGSpacer(scroll_wrap(self.desc_view)))
+        
+        scrolled_window.add_with_viewport(self.vbox1)
+        self.vbox1.show()
+        
         self.connect("response", self._response_cb)        
         self.show_all()
@@ -1049 +1118 @@
 %(rule)s
 
+%(content)s
+
+'''
+    
+    default_content = '''\
 action = function(host, port)
 \t-- main testing actions here
 \treturn
 end
-
 '''
 
@@ -1061 +1134 @@
         version = self.version_entry.get_text()
         license = self.license_entry.get_text()
-
-        if self.port_rule.get_active():
-            rule = '''portrule = function(host, port)
-\tdecision = true
-\t-- port choosing here
-\treturn decision
-end'''
+        template = self.templates.get_active_text()
+        
+        if template == "" or template==None:
+            content = self.default_content
+            if self.port_rule.get_active():
+                rule = '''portrule = function(host, port)
+    \tdecision = true
+    \t-- port choosing here
+    \treturn decision
+    end'''
+            else:
+                rule = '''hostrule = function(host, port)
+    \tdecision = true
+    \t-- host choosing here
+    \treturn decision
+    end'''
         else:
-            rule = '''hostrule = function(host, port)
-\tdecision = true
-\t-- host choosing here
-\treturn decision
-end'''
+            rule = ''
+            content = self.content
+            pass
+            
         categories = []
         for name, check in self.categories.items():
@@ -1101 +1182 @@
                         os.path.join(Path.docs_dir, "script_editor.html#wizard"), new=2)
         self.stop_emission("response")
+        
+    def _template_changed_cb(self, combobox):
+        """
+        Update fields with template content
+        """
+        if combobox.get_active_text()!='':
+            template = DEFAULT_TEMPLATE_PATH + '/' + combobox.get_active_text()
+            parser = LuaParser(template)
+            d = {}
+            d.update(parser.attr)
+            
+            id = d.get('id', '')
+            author = d.get('author', '')
+            version = d.get('version', '')
+            license = d.get('license', '')
+            description = d.get('description', '')
+            categories = d.get('categories', '')
+            self.content = d.get('content', '')
+            categories = categories.split(',')
+            
+            for name, check in self.categories.items():
+                check.set_active(name in categories)
+                
+            # update entries
+            self.id_entry.set_text(id)
+            self.author_entry.set_text(author)
+            self.version_entry.set_text(version)
+            self.license_entry.set_text(license)
+            self.desc_view.get_buffer().set_text(description)
+        else:
+            self._clear_all()
+            
+    def _clear_all(self):
+        """
+        Clear all text fields.
+        """
+        self.id_entry.set_text('')
+        self.author_entry.set_text('')
+        self.version_entry.set_text('')
+        self.license_entry.set_text('')
+        self.desc_view.get_buffer().set_text('')
+        self.content = ''
+        
+        for name, check in self.categories.items():
+            check.set_active(False)
+        
     
 class ScriptEditorPreferencesDialog(HIGDialog):
@@ -1317 +1444 @@
                         os.path.join(Path.docs_dir, "script_editor.html#preferences"), new=2)
         self.stop_emission("response")
-
+        
 if __name__ == "__main__":
     editor = ScriptEditorWindow()

network-scanner/branches/NSEFacilitator/umitNSEFacilitator/luaParser.py

@@ -62 +62 @@
         ScriptParser.__init__(self, filename)
         
+        self.keywords = ['id', 'description', 'author', 'license', 'version', 'categories']
+        
         self._get_attr('id')
         self._get_attr('description')
         self._get_attr('author')
         self._get_attr('license')
+        self._get_attr('version')
         self._get_attr_list('categories')
+        self.attr['content'] = self._get_content()
         if self._get_function('portrule'):
             self.attr['rule'] = 'port'
@@ -73 +77 @@
         else:
             self.attr['rule'] = None
-        self.attr['version'] = self._get_attr('version')
+        #self.attr['version'] = self._get_attr('version')
         
     def _get_attr_list(self, listname):
@@ -89 +93 @@
         r = re.findall(funcname + '\s*=\s*function', self.data)
         return bool(r)
+    
+    def _get_content(self):
+        content = ""
+        lines = self.data.split('\n')
+        for line in lines:
+            aux = True
+            for key in self.keywords:
+                if line.startswith(key):
+                    aux = False
+            if aux:
+                content = content + line + '\n'
+        first_line = re.search('[^\n]', content).start()
+        return content[first_line:]
 
     def _normilize(self, string):