1import argparse
2
3PROGRAM_TEMPLATE = "750010{}08060001080006040002AA300E3CAA0FBA06AA09BA07AA08BA086A01BA09120C84006F08066A0EA30206000108000604032B12147A27017A020203301A1C820200032D68A30206FFFFFFFFFFFF020E1A267E00000002{}032C020B1A267E00000002{}032CAB24003CCA0606CB0306CB090ACB0306C6{}CA0606CA1C04AA0A3A12AA1AAA25FFFF032F020D120C84001708000A1782100612149C00091FFFAB0D2A10820207032A02117C000E86DD68A30206FFFFFFFFFFFF021603190A1482020002187A023A02120A36820285031F8216886A26A2020FFF020000000000000000000000000003200214"
4
5def generate_apf_program(mac_raw, ip_raw):
6    """
7    Generates an APF program that support ARP offload
8
9    Args:
10        mac_raw (str): The MAC address (colon-separated hexadecimal values).
11        ip_raw (str): The IPv4 address (dot-separated decimal values).
12
13    Returns:
14        str: The generated APF program hex string.
15    """
16    mac_list = mac_raw.split(":")
17    ip_list = ip_raw.split(".")
18
19    ip_addr = "".join(["{:02x}".format(int(i)) for i in ip_list])
20    mac_addr = "".join(mac_list)
21
22    return PROGRAM_TEMPLATE.format(mac_addr, ip_addr, ip_addr, ip_addr)
23
24
25def main():
26    parser = argparse.ArgumentParser(description="Generate an ARP offload APF program.")
27    parser.add_argument("mac", help="The DUT's MAC address (e.g., '00:11:22:33:44:55')")
28    parser.add_argument("ip", help="The DUT's IPv4 address (e.g., '192.168.1.100')")
29    args = parser.parse_args()
30
31    out_program = generate_apf_program(args.mac, args.ip)
32    print("APF Program:\n", out_program)
33
34
35if __name__ == '__main__':
36    main()
37