sfatables/xmlrule.py

   1 import libxml2
   2 from sfatables.globals import *
   3
   4 class XMLRule:
   5     rule_number = None
   6     chain = None
   7     xmldoc = None
   8
   9     arguments = {'match':None,'target':None}
  10     processors = {'match':None,'target':None}
  11     context = {'match':None,'target':None}
  12
  13     def load_xml_extension (self, type, chain, rule_number):
  14         filename = sfatables_config+"/"+chain+"/"+"sfatables-%d-%s"%(rule_number,type)
  15
  16         self.xmldoc = libxml2.parseFile(filename)
  17         p = self.xmldoc.xpathNewContext()
  18
  19         context = p.xpathEval('//context/@select')
  20         self.context[type] = context[0].content
  21
  22         processor = p.xpathEval('//processor/@filename')
  23
  24         self.processors[type] = processor[0].content
  25         self.arguments[type] = p.xpathEval('//rule')
  26
  27         p.xpathFreeContext()
  28
  29
  30     def wrap_rspec (self, type, rspec):
  31         argument = self.arguments[type]
  32         p = rspec.xmldoc.xpathNewContext()
  33         root_node = p.xpathEval('/RSpec')
  34         if (not root_node or not root_node):
  35             raise Exception('An evil aggregate manager sent me a malformed RSpec. Please see the stack trace to identify it.')
  36
  37         root_node.addChild(arguments[type])
  38         return rspec
  39
  40     def __init__(self, chain, rule_number):
  41
  42         self.load_xml_extension('match', chain, rule_number)
  43         self.load_xml_extension('target',chain, rule_number)
  44         self.rule_number = rule_number
  45         self.chain = chain
  46
  47         return
  48
  49     def free(self):
  50         self.xmldoc.freeDoc()