sfatables/commands/List.py

   1 import os, time
   2 import libxml2
   3 import pdb
   4
   5 from sfatables.globals import *
   6 from sfatables.pretty import Pretty
   7 from sfatables.command import Command
   8
   9 class List(Command):
  10     options = [('-L','--list')]
  11     help = 'List a chain'
  12     key='list_rule'
  13     matches = False
  14     targets = False
  15
  16     def __init__(self):
  17         return
  18
  19     def get_info(self, type, xmlextension_path):
  20         xmldoc = libxml2.parseFile(xmlextension_path)
  21         p = xmldoc.xpathNewContext()
  22
  23         ext_name_node = p.xpathEval("/%s/@name"%type)
  24         ext_name = ext_name_node[0].content
  25
  26         name_nodes = p.xpathEval("//rule/argument[value!='']/name")
  27         value_nodes = p.xpathEval("//rule/argument[value!='']/value")
  28
  29         names = [n.content for n in name_nodes]
  30         values = [v.content for v in value_nodes]
  31
  32         name_values = zip(names,values)
  33         name_value_pairs = map(lambda (n,v):n+'='+v, name_values)
  34
  35         argument_str = ",".join(name_value_pairs)
  36
  37         p.xpathFreeContext()
  38         xmldoc.freeDoc()
  39
  40         return {'name':ext_name, 'arguments':argument_str}
  41
  42     def get_rule_list(self, chain_dir_path):
  43         broken_semantics = os.walk(chain_dir_path)
  44         rule_numbers = {}
  45
  46         for (root, dirs, files) in broken_semantics:
  47             for file in files:
  48                 if (file.startswith('sfatables')):
  49                     (magic,number,type) = file.split('-')
  50                     rule_numbers[int(number)]=1
  51
  52         rule_list = rule_numbers.keys()
  53         rule_list.sort()
  54         return rule_list
  55
  56     def call(self, command_options, match_options, target_options):
  57         if (len(command_options.args) < 1):
  58             print "Please specify the name of the chain you would like to list, e.g. sfatables -L INCOMING."
  59             return
  60
  61         chain = command_options.args[0]
  62         chain_dir = sfatables_config + "/" + chain
  63
  64         rule_list = self.get_rule_list(chain_dir)
  65
  66         pretty = Pretty(['Rule','Match','Arguments','Target','Arguments'])
  67
  68         for number in rule_list:
  69             match_file = "sfatables-%d-%s"%(number,'match')
  70             target_file = "sfatables-%d-%s"%(number,'target')
  71
  72             match_path = sfatables_config + '/' + chain + '/' + match_file
  73             target_path = sfatables_config + '/' + chain + '/' + target_file
  74
  75             match_info = self.get_info ('match',match_path)
  76             target_info = self.get_info ('target',target_path)
  77
  78             pretty.push_row(["%d"%number,  match_info['name'], match_info['arguments'], target_info['name'], target_info['arguments']])
  79
  80
  81         pretty.pprint()
  82
  83
  84
  85
  86
  87
  88
  89