1#!/usr/bin/python3
2
3"""
4tdc_batch.py - a script to generate TC batch file
5
6Copyright (C) 2017 Chris Mi <chrism@mellanox.com>
7"""
8
9import argparse
10
11parser = argparse.ArgumentParser(description='TC batch file generator')
12parser.add_argument("device", help="device name")
13parser.add_argument("file", help="batch file name")
14parser.add_argument("-n", "--number", type=int,
15                    help="how many lines in batch file")
16parser.add_argument("-o", "--skip_sw",
17                    help="skip_sw (offload), by default skip_hw",
18                    action="store_true")
19parser.add_argument("-s", "--share_action",
20                    help="all filters share the same action",
21                    action="store_true")
22parser.add_argument("-p", "--prio",
23                    help="all filters have different prio",
24                    action="store_true")
25args = parser.parse_args()
26
27device = args.device
28file = open(args.file, 'w')
29
30number = 1
31if args.number:
32    number = args.number
33
34skip = "skip_hw"
35if args.skip_sw:
36    skip = "skip_sw"
37
38share_action = ""
39if args.share_action:
40    share_action = "index 1"
41
42prio = "prio 1"
43if args.prio:
44    prio = ""
45    if number > 0x4000:
46        number = 0x4000
47
48index = 0
49for 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