sfatables/commands/List.py

   1 import os, time
   2 import libxml2
   3 import pdb
   4
   5 from globals import *
   6 from 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
  20     def get_info(self, xmlextension_path):
  21         xmldoc = libxml2.parseFile(xmlextension_path)
  22         p = xmldoc.xpathNewContext()
  23
  24         ext_name_node = p.xpathEval("/match/@name")
  25         ext_name = ext_name_node[0].content
  26
  27         name_nodes = p.xpathEval("//rule/argument[value!='']/name")
  28         value_nodes = p.xpathEval("//rule/argument[value!='']/value")
  29
  30         names = [n.content for n in name_nodes]
  31         values = [v.content for v in value_nodes]
  32
  33         name_values = zip(names,values)
  34         name_value_pairs = map(lambda (n,v):n+'='+v, name_values)
  35
  36         argument_str = ",".join(name_value_pairs)
  37
  38         p.xpathFreeContext()
  39         xmldoc.freeDoc()
  40
  41         return {'name':ext_name, 'arguments':argument_str}
  42
  43     def get_rule_list(self, chain_dir_path):
  44         broken_semantics = os.walk(chain_dir_path)
  45         rule_numbers = {}
  46
  47         for (root, dirs, files) in broken_semantics:
  48             for file in files:
  49                 if (file.startswith('sfatables')):
  50                     (magic,number,type) = file.split('-')
  51                     rule_numbers[int(number)]=1
  52
  53         rule_list = rule_numbers.keys()
  54         rule_list.sort()
  55         return rule_list
  56
  57     def call(self, command_options, match_options, target_options):
  58         chain = command_options.args[0]
  59         chain_dir = sfatables_config + "/" + chain
  60
  61         rule_list = self.get_rule_list(chain_dir)
  62
  63         pretty = Pretty(['Rule','Match','Arguments','Target','Arguments'])
  64
  65         for number in rule_list:
  66             match_file = "sfatables-%d-%s"%(number,'match')
  67             target_file = "sfatables-%d-%s"%(number,'target')
  68
  69             match_path = sfatables_config + '/' + chain + '/' + match_file
  70             target_path = sfatables_config + '/' + chain + '/' + target_file
  71
  72             match_info = self.get_info (match_path)
  73             target_info = self.get_info (target_path)
  74
  75             pretty.push_row(["%d"%number,  match_info['name'], match_info['arguments'], target_info['name'], target_info['arguments']])
  76
  77
  78         pretty.pprint()
  79
  80
  81
  82
  83
  84
  85
  86