#!/usr/bin/env python

# Copyright (c) 2015 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


import ipaddress
import argparse
import sys

# map add domain ip4-pfx <pfx> ip6-pfx ::/0 ip6-src <ip6-src> ea-bits-len 0 psid-offset 6 psid-len 6
# map add rule index <0> psid <psid> ip6-dst <ip6-dst>

parser = argparse.ArgumentParser(description='MAP VPP configuration generator')
parser.add_argument('-t', action="store", dest="mapmode")
args = parser.parse_args()

#
# 1:1 Shared IPv4 address, shared BR
#
def shared11br():
    ip4_pfx = ipaddress.ip_network('20.0.0.0/16')
    ip6_dst = ipaddress.ip_network('bbbb::/32')
    psid_len = 6
    for i in range(ip4_pfx.num_addresses):
        print("map add domain ip4-pfx " + str(ip4_pfx[i]) +  "/32 ip6-pfx ::/0 ip6-shared-src cccc:bbbb::1",
              "ea-bits-len 0 psid-offset 6 psid-len", psid_len)
        for psid in range(0x1 << psid_len):
            print("map add rule index", i, "psid", psid, "ip6-dst", ip6_dst[(i * (0x1<<psid_len)) + psid])


#
# 1:1 Shared IPv4 address
#
def shared11():
    ip4_pfx = ipaddress.ip_network('20.0.0.0/16')
    ip6_src = ipaddress.ip_network('cccc:bbbb::/64')
    ip6_dst = ipaddress.ip_network('bbbb::/32')
    psid_len = 6
    for i in range(ip4_pfx.num_addresses):
        print("map add domain ip4-pfx " + str(ip4_pfx[i]) +  "/32 ip6-pfx ::/0 ip6-src", ip6_src[i],
              "ea-bits-len 0 psid-offset 6 psid-len", psid_len)
        for psid in range(0x1 << psid_len):
            print("map add rule index", i, "psid", psid, "ip6-dst", ip6_dst[(i * (0x1<<psid_len)) + psid])

#
# 1:1 Shared IPv4 address small
#
def smallshared11():
    ip4_pfx = ipaddress.ip_network('20.0.0.0/24')
    ip6_src = ipaddress.ip_network('cccc:bbbb::/64')
    ip6_dst = ipaddress.ip_network('bbbb::/32')
    psid_len = 6
    for i in range(ip4_pfx.num_addresses):
        print("map add domain ip4-pfx " + str(ip4_pfx[i]) +  "/32 ip6-pfx ::/0 ip6-src", ip6_src[i],
              "ea-bits-len 0 psid-offset 6 psid-len", psid_len)
        for psid in range(0x1 << psid_len):
            print("map add rule index", i, "psid", psid, "ip6-dst", ip6_dst[(i * (0x1<<psid_len)) + psid])

#
# 1:1 Full IPv4 address
#
def full11():
    ip4_pfx = ipaddress.ip_network('20.0.0.0/16')
    ip6_src = ipaddress.ip_network('cccc:bbbb::/64')
    ip6_dst = ipaddress.ip_network('bbbb::/32')
    psid_len = 0
    for i in range(ip4_pfx.num_addresses):
        print("map add domain ip4-pfx " + str(ip4_pfx[i]) +  "/32 ip6-pfx " + str(ip6_dst[i]) + "/128 ip6-src", ip6_src[i],
              "ea-bits-len 0 psid-offset 0 psid-len 0")
def full11br():
    ip4_pfx = ipaddress.ip_network('20.0.0.0/16')
    ip6_dst = ipaddress.ip_network('bbbb::/32')
    psid_len = 0
    for i in range(ip4_pfx.num_addresses):
        print("map add domain ip4-pfx " + str(ip4_pfx[i]) +  "/32 ip6-pfx " + str(ip6_dst[i]) + "/128 ip6-shared-src cccc:bbbb::1",
              "ea-bits-len 0 psid-offset 0 psid-len 0")

#
# Algorithmic mapping Shared IPv4 address
#
def algo():
    print("map add domain ip4-pfx 20.0.0.0/24 ip6-pfx bbbb::/32 ip6-src cccc:bbbb::1 ea-bits-len 16 psid-offset 6 psid-len 8")
    print("map add domain ip4-pfx 20.0.1.0/24 ip6-pfx bbbb:1::/32 ip6-src cccc:bbbb::2 ea-bits-len 8 psid-offset 0 psid-len 0")

#
# IP4 forwarding
#
def ip4():
    ip4_pfx = ipaddress.ip_network('20.0.0.0/16')
    for i in range(ip4_pfx.num_addresses):
        print("ip route add " + str(ip4_pfx[i]) +  "/32 via 172.16.0.2")


globals()[args.mapmode]()