X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfatables%2Fxmlrule.py;h=310c84392132150ad67ed2faeec2a09a330392c9;hb=805a8364a6ce7295edb2bad37bd26e12e3dbecac;hp=d787c15ebe391fb6a46a99a6937b8b8e8e64601c;hpb=a4a914c48938c95b54449ba151b93da369453167;p=sfa.git

diff --git a/sfatables/xmlrule.py b/sfatables/xmlrule.py
index d787c15e..310c8439 100644
--- a/sfatables/xmlrule.py
+++ b/sfatables/xmlrule.py
@@ -1,15 +1,77 @@
 import libxml2
+import libxslt
 from sfatables.globals import *
 
 class XMLRule:
-    rule_number = None
-    chain = None
-    xmldoc = None
-    terminal = 0
+    def apply_processor(self, type, doc, output_xpath_filter=None):
+        processor = self.processors[type]
 
-    arguments = {'match':None,'target':None}
-    processors = {'match':None,'target':None}
-    context = {'match':None,'target':None}
+        # XXX TO CLEAN UP
+        filepath = os.path.join(sfatables_config, 'processors', processor)
+        # XXX
+
+        print filepath
+        styledoc = libxml2.parseFile(filepath)
+        style = libxslt.parseStylesheetDoc(styledoc)
+        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
+            p.xpathFreeContext()
+        else:
+            stylesheet_result = result #style.saveResultToString(result)
+
+        style.freeStylesheet()
+        #doc.freeDoc()
+        #result.freeDoc()
+
+        return stylesheet_result
+
+    def wrap_up(self, doc):
+        filepath = os.path.join(sfatables_config, 'processors', self.final_processor)
+
+        if not os.path.exists(filepath):
+            raise Exception('Could not find final rule filter')
+
+        styledoc = libxml2.parseFile(filepath)
+        style = libxslt.parseStylesheetDoc(styledoc)
+        result = style.applyStylesheet(doc, None)
+        stylesheet_result = 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[0].content=='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"%(rule_number,type)
@@ -42,13 +104,22 @@ class XMLRule:
         root_node.addChild(arguments[type])
         return rspec
 
-    def __init__(self, chain, rule_number):
+    def __init__(self, chain=None, rule_number=None):
+        self.rule_number = None
+        self.chain = None
+        self.xmldoc = None
+        self.terminal = 0
+        self.final_processor = '__sfatables_rule_wrap_up__.xsl'
 
-        self.load_xml_extension('match', chain, rule_number)
-        self.load_xml_extension('target',chain, rule_number)
-        self.rule_number = rule_number
-        self.chain = chain
+        self.arguments = {'match':None,'target':None}
+        self.processors = {'match':None,'target':None}
+        self.context = {'match':None,'target':None}
 
+        if (chain and rule_number):
+            self.load_xml_extension('match', chain, rule_number)
+            self.load_xml_extension('target',chain, rule_number)
+            self.rule_number = rule_number
+            self.chain = chain
         return
         
     def free(self):