#!/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(    "-a",    "--handle_start",    type=int,    default=1,    help="start handle range from (default: 1)")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")parser.add_argument(    "-e",    "--operation",    choices=['add', 'del', 'replace'],    default='add',    help="operation to perform on filters"    "(default: add filter)")parser.add_argument(    "-m",    "--mac_prefix",    type=int,    default=0,    choices=range(0, 256),    help="third byte of source MAC address of flower filter"    "(default: 0)")args = parser.parse_args() device = args.devicefile = open(args.file, 'w') number = 1if args.number:    number = args.number handle_start = args.handle_start 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 mac_prefix = args.mac_prefix def format_add_filter(device, prio, handle, skip, src_mac, dst_mac,                      share_action):    return ("filter add dev {} {} protocol ip ingress handle {} "            " flower {} src_mac {} dst_mac {} action drop {}".format(                device, prio, handle, skip, src_mac, dst_mac, share_action))  def format_rep_filter(device, prio, handle, skip, src_mac, dst_mac,                      share_action):    return ("filter replace dev {} {} protocol ip ingress handle {} "            " flower {} src_mac {} dst_mac {} action drop {}".format(                device, prio, handle, skip, src_mac, dst_mac, share_action))  def format_del_filter(device, prio, handle, skip, src_mac, dst_mac,                      share_action):    return ("filter del dev {} {} protocol ip ingress handle {} "            "flower".format(device, prio, handle))  formatter = format_add_filterif args.operation == "del":    formatter = format_del_filterelif args.operation == "replace":    formatter = format_rep_filter index = 0for 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:{:02x}:{}".format(mac_prefix, mac)            dst_mac = "e4:12:00:" + mac            cmd = formatter(device, prio, handle_start + index, skip, src_mac,                            dst_mac, share_action)            file.write("{}\n".format(cmd))            index += 1            if index >= number:                file.close()                exit(0)