X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfatables%2Fxmlrule.py;h=92a06e24d237ba4261a7c3a518dd226b164929ea;hb=483772a0716028a34cb98012fbe252ffd445fedf;hp=bce804a0e370580f4e6eeacdcaed5985e923fc0d;hpb=c36efa8afed156cc5c25bc61dfd8987fed0010d1;p=sfa.git

diff --git a/sfatables/xmlrule.py b/sfatables/xmlrule.py
index bce804a0..92a06e24 100644
--- a/sfatables/xmlrule.py
+++ b/sfatables/xmlrule.py
@@ -1,17 +1,73 @@
 import libxml2
+import libxslt
 from sfatables.globals import *
 
-class Xmlrule:
+class XMLRule:
     rule_number = None
     chain = None
     xmldoc = None
+    terminal = 0
 
     arguments = {'match':None,'target':None}
     processors = {'match':None,'target':None}
     context = {'match':None,'target':None}
 
+    def apply_processor(self, type, rspec, output_xpath_filter=None):
+        processor = self.processors[type]
+
+        # XXX TO CLEAN UP
+        filepath = 'processors/' + processor
+        # XXX
+
+        styledoc = libxml2.parseFile(filepath)
+        style = libxslt.parseStylesheetDoc(styledoc)
+        doc = libxml2.parseDoc(rspec)
+        result = style.applyStylesheet(doc, None)
+        if (output_xpath_filter):
+            p = result.xpathNewContext()
+            xpath_result = p.xpathEval(output_xpath_filter)
+            if (xpath_result == []):
+                raise Exception("Could not apply processor %s."%processor)
+
+            stylesheet_result = xpath_result[0].content
+            p.xpathFreeContext()
+        else:
+            stylesheet_result = style.saveResultToString(result)
+        style.freeStylesheet()
+        doc.freeDoc()
+        result.freeDoc()
+
+
+        return stylesheet_result
+
+    def match(self, rspec):
+        match_result = self.apply_processor('match',rspec,"//result/@verdict") 
+        return (match_result=='True')
+
+    def target(self, rspec):
+        target_result = self.apply_processor('target',rspec,None)
+        return target_result
+
+    def apply_interpreted(self, rspec):
+        # Interpreted
+        #
+        # output =
+        #    if (match(match_args, rspec)
+        #       then target(target_args, rspec)
+        #       else rspec
+
+        if (self.match(rspec)):
+            return self.target(rspec)
+        else:
+            return rspec
+
+
+    def apply_compiled(rspec):
+        # Not supported yet
+        return None
+
     def load_xml_extension (self, type, chain, rule_number):
-        filename = sfatables_config+"/"+chain+"/"+"sfatables-%d-%s.xml"%(rule_number,type)
+        filename = sfatables_config+"/"+chain+"/"+"sfatables-%d-%s"%(rule_number,type)
 
         self.xmldoc = libxml2.parseFile(filename)
         p = self.xmldoc.xpathNewContext()
@@ -21,18 +77,21 @@ class Xmlrule:
 
         processor = p.xpathEval('//processor/@filename')
 
-        self.processor[type] = processor[0].content
+        context = p.xpathEval('//attributes/attribute[@terminal="yes"]')
+        if (context != []):
+            self.terminal = 1
+        
+        self.processors[type] = processor[0].content
         self.arguments[type] = p.xpathEval('//rule')
 
         p.xpathFreeContext()
 
-        return 
 
     def wrap_rspec (self, type, rspec):
         argument = self.arguments[type]
         p = rspec.xmldoc.xpathNewContext()
         root_node = p.xpathEval('/RSpec')
-        if (!root_node or !len(root_node)):
+        if (not root_node or not root_node):
             raise Exception('An evil aggregate manager sent me a malformed RSpec. Please see the stack trace to identify it.')
 
         root_node.addChild(arguments[type])