- if len(sys.argv) > 1 and sys.argv[1] == "parse_8021q":
- out.write("#define PARSE_8021Q\n")
- if len(sys.argv) > 1 and sys.argv[1].startswith("req_8021q="):
- out.write("#define PARSE_8021Q\n")
- out.write(f"#define REQ_8021Q {sys.argv[1][10:]}\n")
-
-
- out.write("#define RULES \\\n")
-
- def write_rule(r):
- out.write("\t\t" + r.replace("\n", " \\\n\t\t") + " \\\n")
+ parse = argparse.ArgumentParser()
+ parse.add_argument("--ihl", dest="ihl", required=True, choices=["drop-options","accept-options","parse-options"])
+ parse.add_argument("--v6frag", dest="v6frag", required=True, choices=["drop-frags","ignore","parse-frags","ignore-parse-if-rule"])
+ parse.add_argument("--8021q", dest="vlan", required=True, choices=["drop-vlan","accept-vlan","parse-vlan"])
+ parse.add_argument("--require-8021q", dest="vlan_tag")
+ args = parse.parse_args(sys.argv[1:])
+
+ if args.ihl == "drop-options":
+ out.write("#define PARSE_IHL XDP_DROP\n")
+ elif args.ihl == "accept-options":
+ out.write("#define PARSE_IHL XDP_PASS\n")
+ elif args.ihl == "parse-options":
+ out.write("#define PARSE_IHL PARSE\n")
+
+ if args.v6frag == "drop-frags":
+ out.write("#define PARSE_V6_FRAG XDP_DROP\n")
+ elif args.v6frag == "ignore":
+ pass
+ elif args.v6frag == "parse-frags":
+ out.write("#define PARSE_V6_FRAG PARSE\n")
+
+ if args.vlan == "drop-vlan":
+ out.write("#define PARSE_8021Q XDP_DROP\n")
+ elif args.vlan == "accept-vlan":
+ out.write("#define PARSE_8021Q XDP_PASS\n")
+ elif args.vlan == "parse-vlan":
+ out.write("#define PARSE_8021Q PARSE\n")
+
+ if args.vlan_tag is not None:
+ if args.vlan != "parse-vlan":
+ assert False
+ out.write("#define REQ_8021Q " + args.vlan_tag + "\n")
+
+ rules6 = ""
+ rules4 = ""
+ use_v6_frags = False
+ rulecnt = 0