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