3 from sfatables.globals import *
6 def apply_processor(self, type, doc, output_xpath_filter=None):
7 processor = self.processors[type]
10 filepath = os.path.join(sfatables_config, 'processors', processor)
14 styledoc = libxml2.parseFile(filepath)
15 style = libxslt.parseStylesheetDoc(styledoc)
16 result = style.applyStylesheet(doc, None)
17 if (output_xpath_filter):
18 p = result.xpathNewContext()
19 xpath_result = p.xpathEval(output_xpath_filter)
20 if (xpath_result == []):
21 raise Exception("Could not apply processor %s."%processor)
23 stylesheet_result = xpath_result
26 stylesheet_result = result #style.saveResultToString(result)
28 style.freeStylesheet()
32 return stylesheet_result
34 def wrap_up(self, doc):
35 filepath = os.path.join(sfatables_config, 'processors', self.final_processor)
37 if not os.path.exists(filepath):
38 raise Exception('Could not find final rule filter')
40 styledoc = libxml2.parseFile(filepath)
41 style = libxslt.parseStylesheetDoc(styledoc)
42 result = style.applyStylesheet(doc, None)
43 stylesheet_result = result#style.saveResultToString(result)
44 style.freeStylesheet()
48 return stylesheet_result
50 def match(self, rspec):
51 match_result = self.apply_processor('match',rspec,"//result/@verdict")
52 return (match_result[0].content=='True')
54 def target(self, rspec):
55 target_result = self.apply_processor('target',rspec,None)
58 def add_rule_context_to_rspec(self, doc):
59 p = doc.xpathNewContext()
60 context = p.xpathEval("//RSpec")
62 raise Exception('Request is not an rspec')
64 # Add the request context
65 matchNode = libxml2.newNode('match-context')
66 for match_argument in self.arguments['match']:
67 matchNode.addChild(match_argument)
69 targetNode = libxml2.newNode('target-context')
70 for target_argument in self.arguments['target']:
71 targetNode.addChild(target_argument)
73 context[0].addChild(matchNode)
74 context[0].addChild(targetNode)
79 def apply_interpreted(self, rspec):
80 rspec = self.add_rule_context_to_rspec(rspec)
84 # if (match(match_args, rspec)
85 # then target(target_args, rspec)
88 if (self.match(rspec)):
89 return self.wrap_up(self.target(rspec))
91 return self.wrap_up(rspec)
94 def apply_compiled(rspec):
98 def load_xml_extension (self, type, chain, rule_number):
99 filename = sfatables_config+"/"+chain+"/"+"sfatables-%d-%s"%(rule_number,type)
101 self.xmldoc = libxml2.parseFile(filename)
102 p = self.xmldoc.xpathNewContext()
104 context = p.xpathEval('//context/@select')
105 self.context[type] = context[0].content
107 processor = p.xpathEval('//processor/@filename')
109 context = p.xpathEval('//attributes/attribute[@terminal="yes"]')
113 self.processors[type] = processor[0].content
114 self.arguments[type] = p.xpathEval('//rule//argument[value!=""]')
119 def __init__(self, chain=None, rule_number=None):
120 self.rule_number = None
124 self.final_processor = '__sfatables_rule_wrap_up__.xsl'
126 self.arguments = {'match':None,'target':None}
127 self.processors = {'match':None,'target':None}
128 self.context = {'match':None,'target':None}
130 if (chain and rule_number):
131 self.load_xml_extension('match', chain, rule_number)
132 self.load_xml_extension('target',chain, rule_number)
133 self.rule_number = rule_number
138 self.xmldoc.freeDoc()