import libxml2
-import libxslt
+# allow to run sfa2wsdl if this is missing (for mac)
+import sys
+try:import libxslt
+except: print >>sys.stderr, "WARNING, could not import libxslt"
+
from sfatables.globals import *
class XMLRule:
- rule_number = None
- chain = None
- xmldoc = None
- terminal = 0
- final_processor = '__sfatables_wrap_up__.xsl'
-
- arguments = {'match':None,'target':None}
- processors = {'match':None,'target':None}
- context = {'match':None,'target':None}
-
def apply_processor(self, type, doc, output_xpath_filter=None):
processor = self.processors[type]
filepath = os.path.join(sfatables_config, 'processors', self.final_processor)
if not os.path.exists(filepath):
- # TODO: final_processor is not there yet
- return doc#rspec
+ raise Exception('Could not find final rule filter')
styledoc = libxml2.parseFile(filepath)
style = libxslt.parseStylesheetDoc(styledoc)
- #doc = libxml2.parseDoc(rspec)
result = style.applyStylesheet(doc, None)
- stylesheet_result = result #style.saveResultToString(result)
+ stylesheet_result = result#style.saveResultToString(result)
style.freeStylesheet()
#doc.freeDoc()
#result.freeDoc()
target_result = self.apply_processor('target',rspec,None)
return target_result
+ def add_rule_context_to_rspec(self, doc):
+ p = doc.xpathNewContext()
+ context = p.xpathEval("//RSpec")
+ if (not context):
+ raise Exception('Request is not an rspec')
+ else:
+ # Add the request context
+ matchNode = libxml2.newNode('match-context')
+ for match_argument in self.arguments['match']:
+ matchNode.addChild(match_argument)
+
+ targetNode = libxml2.newNode('target-context')
+ for target_argument in self.arguments['target']:
+ targetNode.addChild(target_argument)
+
+ context[0].addChild(matchNode)
+ context[0].addChild(targetNode)
+ p.xpathFreeContext()
+
+ return doc
+
def apply_interpreted(self, rspec):
+ rspec = self.add_rule_context_to_rspec(rspec)
# Interpreted
#
# output =
# if (match(match_args, rspec)
# then target(target_args, rspec)
# else rspec
-
+
+ import pdb
if (self.match(rspec)):
- return self.target(rspec)
+ return (True,self.wrap_up(self.target(rspec)))
else:
- return rspec
+ return (False,self.wrap_up(rspec))
def apply_compiled(rspec):
self.terminal = 1
self.processors[type] = processor[0].content
- self.arguments[type] = p.xpathEval('//rule')
+ self.arguments[type] = p.xpathEval('//rule//argument[value!=""]')
p.xpathFreeContext()
- def wrap_rspec (self, type, rspec):
- argument = self.arguments[type]
- p = rspec.xmldoc.xpathNewContext()
- root_node = p.xpathEval('/RSpec')
- 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.')
+ 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'
- root_node.addChild(arguments[type])
- return rspec
+ self.arguments = {'match':None,'target':None}
+ self.processors = {'match':None,'target':None}
+ self.context = {'match':None,'target':None}
- def __init__(self, chain=None, rule_number=None):
if (chain and rule_number):
self.load_xml_extension('match', chain, rule_number)
self.load_xml_extension('target',chain, rule_number)