| #!/usr/bin/env 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}".format(i, j, k)) |
| src_mac = "e4:11:00:" + mac |
| dst_mac = "e4:12:00:" + mac |
| cmd = ("filter add dev {} {} protocol ip parent ffff: flower {} " |
| "src_mac {} dst_mac {} action drop {}".format |
| (device, prio, skip, src_mac, dst_mac, share_action)) |
| file.write("{}\n".format(cmd)) |
| index += 1 |
| if index >= number: |
| file.close() |
| exit(0) |
