tools/testing/selftests/tc-testing/tdc_batch.py

   1 #!/usr/bin/env python3
   2
   3 """
   4 tdc_batch.py - a script to generate TC batch file
   5
   6 Copyright (C) 2017 Chris Mi <chrism@mellanox.com>
   7 """
   8
   9 import argparse
  10
  11 parser = argparse.ArgumentParser(description='TC batch file generator')
  12 parser.add_argument("device", help="device name")
  13 parser.add_argument("file", help="batch file name")
  14 parser.add_argument("-n", "--number", type=int,
  15                     help="how many lines in batch file")
  16 parser.add_argument("-o", "--skip_sw",
  17                     help="skip_sw (offload), by default skip_hw",
  18                     action="store_true")
  19 parser.add_argument("-s", "--share_action",
  20                     help="all filters share the same action",
  21                     action="store_true")
  22 parser.add_argument("-p", "--prio",
  23                     help="all filters have different prio",
  24                     action="store_true")
  25 args = parser.parse_args()
  26
  27 device = args.device
  28 file = open(args.file, 'w')
  29
  30 number = 1
  31 if args.number:
  32     number = args.number
  33
  34 skip = "skip_hw"
  35 if args.skip_sw:
  36     skip = "skip_sw"
  37
  38 share_action = ""
  39 if args.share_action:
  40     share_action = "index 1"
  41
  42 prio = "prio 1"
  43 if args.prio:
  44     prio = ""
  45     if number > 0x4000:
  46         number = 0x4000
  47
  48 index = 0
  49 for i in range(0x100):
  50     for j in range(0x100):
  51         for k in range(0x100):
  52             mac = ("{:02x}:{:02x}:{:02x}".format(i, j, k))
  53             src_mac = "e4:11:00:" + mac
  54             dst_mac = "e4:12:00:" + mac
  55             cmd = ("filter add dev {} {} protocol ip parent ffff: flower {} "
  56                    "src_mac {} dst_mac {} action drop {}".format
  57                    (device, prio, skip, src_mac, dst_mac, share_action))
  58             file.write("{}\n".format(cmd))
  59             index += 1
  60             if index >= number:
  61                 file.close()
  62                 exit(0)