コード例 #1
0
ファイル: rulegen.py プロジェクト: constanze/rulegen
def generateRules(opts, tcpRules, udpRules, icmpRules):
    print "Generating TCP-Ruleset"
    tcpRuleG = RuleGenerator(tcpRules, "tcp", opts.style, opts.numChecksP, opts.numChecksIP, \
                             opts.numAnyP, opts.numAnyIP, \
                             opts.distanceRange, opts.slashSize, opts.ipsPerSlash)
    if opts.verbose:
        print "Original ruleset:"
        tcpRuleG.printRules()

    tcpRuleG.generateRules()

    print "Generating UDP-Ruleset"
    udpRuleG = RuleGenerator(udpRules, "udp", opts.style, opts.numChecksP, opts.numChecksIP, \
                             opts.numAnyP, opts.numAnyIP, opts.distanceRange, \
                             opts.slashSize, opts.ipsPerSlash)
    if opts.verbose:
        print "Original ruleset:"
        udpRuleG.printRules()

    udpRuleG.generateRules()
    
    print "Generating ICMP-Ruleset"
    icmpRuleG = RuleGenerator(icmpRules, "icmp", opts.style, opts.numChecksP, opts.numChecksIP, \
                             opts.numAnyP, opts.numAnyIP, opts.distanceRange, \
                             opts.slashSize, opts.ipsPerSlash)
    if opts.verbose:
        print "Original ruleset:"
        icmpRuleG.printRules()

    icmpRuleG.generateRules()

    tcpRuleG.printRules(opts.elements)
    print
    udpRuleG.printRules(opts.elements)
    print
    icmpRuleG.printRules(opts.elements)
    print
    return tcpRuleG, udpRuleG, icmpRuleG