|
@@ -0,0 +1,62 @@
|
|
|
+#!/usr/bin/python3
|
|
|
+
|
|
|
+"""
|
|
|
+tdc_batch.py - a script to generate TC batch file
|
|
|
+
|
|
|
+Copyright (C) 2017 Chris Mi <chrism@mellanox.com>
|
|
|
+"""
|
|
|
+
|
|
|
+import argparse
|
|
|
+
|
|
|
+parser = argparse.ArgumentParser(description='TC batch file generator')
|
|
|
+parser.add_argument("device", help="device name")
|
|
|
+parser.add_argument("file", help="batch file name")
|
|
|
+parser.add_argument("-n", "--number", type=int,
|
|
|
+ help="how many lines in batch file")
|
|
|
+parser.add_argument("-o", "--skip_sw",
|
|
|
+ help="skip_sw (offload), by default skip_hw",
|
|
|
+ action="store_true")
|
|
|
+parser.add_argument("-s", "--share_action",
|
|
|
+ help="all filters share the same action",
|
|
|
+ action="store_true")
|
|
|
+parser.add_argument("-p", "--prio",
|
|
|
+ help="all filters have different prio",
|
|
|
+ action="store_true")
|
|
|
+args = parser.parse_args()
|
|
|
+
|
|
|
+device = args.device
|
|
|
+file = open(args.file, 'w')
|
|
|
+
|
|
|
+number = 1
|
|
|
+if args.number:
|
|
|
+ number = args.number
|
|
|
+
|
|
|
+skip = "skip_hw"
|
|
|
+if args.skip_sw:
|
|
|
+ skip = "skip_sw"
|
|
|
+
|
|
|
+share_action = ""
|
|
|
+if args.share_action:
|
|
|
+ share_action = "index 1"
|
|
|
+
|
|
|
+prio = "prio 1"
|
|
|
+if args.prio:
|
|
|
+ prio = ""
|
|
|
+ if number > 0x4000:
|
|
|
+ number = 0x4000
|
|
|
+
|
|
|
+index = 0
|
|
|
+for i in range(0x100):
|
|
|
+ for j in range(0x100):
|
|
|
+ for k in range(0x100):
|
|
|
+ mac = ("%02x:%02x:%02x" % (i, j, k))
|
|
|
+ src_mac = "e4:11:00:" + mac
|
|
|
+ dst_mac = "e4:12:00:" + mac
|
|
|
+ cmd = ("filter add dev %s %s protocol ip parent ffff: flower %s "
|
|
|
+ "src_mac %s dst_mac %s action drop %s" %
|
|
|
+ (device, prio, skip, src_mac, dst_mac, share_action))
|
|
|
+ file.write("%s\n" % cmd)
|
|
|
+ index += 1
|
|
|
+ if index >= number:
|
|
|
+ file.close()
|
|
|
+ exit(0)
|