1 #!/usr/bin/python3 2 3 """ 4 tdc_batch.py - a script to generate TC batch file 5 6 Copyright (C) 2017 Chris Mi <chrism (at] 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) 63