sfatables/sfatables

   1 #!/usr/bin/python
   2
   3 # This file parses an sfatables command and generates XML files that parameterize
   4 # matches and targets. Each such XML file defines a rule. Rules are dropped in directories
   5 # that represent 'chains.' SFA loads rules from various chains and invokes them at certain
   6 # 'hook points.' For example, it invokes rules in the 'OUTGOING' chain before returning
   7 # the output of 'get_resources.'
   8
   9 import sys
  10 import os
  11 import pdb
  12 import libxml2
  13
  14 from optparse import OptionParser
  15 from sfatables import commands, matches, targets
  16 from sfatables.xmlextension import Xmlextension
  17
  18 def load_commands(module, list):
  19     command_dict={}
  20
  21     for command_name in list:
  22         command_module = __import__(".".join([module,command_name]),fromlist=[module])
  23         command = getattr(command_module, command_name)
  24         command_dict[command_name]=command()
  25
  26     return command_dict
  27
  28 def load_xml_extensions(module, list):
  29     ext_dict={}
  30
  31     for ext_name in list:
  32         module = Xmlextension(ext_name)
  33         ext_dict[ext_name]=module
  34
  35     return ext_dict
  36
  37
  38 def create_parser(command_dict):
  39     parser = OptionParser(usage="sfatables [command] [chain] [match] [target]",
  40                              description='See "man sfatables" for more detail.')
  41
  42     for k in command_dict.keys():
  43         command = command_dict[k]
  44         for (short_option,long_option) in command.options:
  45             parser.add_option(short_option,long_option,dest=command.type,action=command.action,const=k,help=command.help,metavar="CHAIN")
  46
  47     return parser
  48
  49 def create_parser_xml_ext(ext_dict):
  50     parser = OptionParser(usage="sfatables [command] [chain] [match] [target]",
  51                              description='See "man sfatables" for more detail.')
  52
  53     for k in ext_dict.keys():
  54         command = ext_dict[k]
  55         pdb.set_trace()
  56         for arg in command.arguments:
  57             parser.add_option(None,"--"+arg,dest=arg,help=command.help,metavar=command.operand)
  58
  59     return parser
  60
  61
  62 def partition(sep, lst):
  63     ret = []
  64     curpart = []
  65     for item in lst:
  66         if (item==sep):
  67             ret.append(curpart)
  68             curpart=[]
  69         else:
  70             curpart.append(item)
  71     ret.append(curpart)
  72
  73     return ret
  74
  75
  76 def main():
  77     # sfatables <command> -- <match> -- <target>
  78     pargs = partition('--', sys.argv[1:])
  79
  80     command_dict = load_commands("sfatables.commands",commands.all)
  81     command_parser = create_parser(command_dict)
  82     (options, args) = command_parser.parse_args()
  83
  84     command = command_dict[options.command]
  85
  86     if (command.matches):
  87         if (len(pargs)<2):
  88             raise Exception("Must specify match for this command")
  89         match_dict = load_xml_extensions("sfatables.matches",matches.all)
  90         match_parser = create_parser_xml_ext(match_dict)
  91         matches_str = ",".join(match_dict.keys())
  92         match_parser.add_option('-m','--match',dest='match_name',help='Match name (one of %s)'%matches_str, metavar = 'MATCH')
  93
  94         (match_options, args) = match_parser.parse_args(pargs[1])
  95     else:
  96         match_options=None
  97
  98     if (command.targets):
  99         if (len(pargs)<3):
 100             raise Exception("Must specify a target for this command")
 101         target_dict = load_xml_extensions("sfatables.targets",targets.all)
 102         target_parser = create_parser_xml_ext(target_dict)
 103         targets = ",".join(target_dict.keys())
 104         (target_options, args) = target_parser.parse_args(pargs[2])
 105         target_parser.add_option('-j','--jump',dest='target_name',help='Target name (one of %s)'%targets, metavar = 'TARGET')
 106     else:
 107         target_options = None
 108
 109     command(options, match_options, target_options)
 110
 111 if __name__=='__main__':
 112     main()