summaryrefslogtreecommitdiffstats
path: root/deploy/adapters/ansible/roles/odl_cluster/files/recover_network.py
blob: 2b15f4a2e6f300b68263f41f020dfb7ffc8fd1a2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
##############################################################################
# Copyright (c) 2016 HUAWEI TECHNOLOGIES CO.,LTD and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

import yaml
import netaddr
import os
import log as logging

LOG = logging.getLogger("net-recover")
config_path = os.path.join(os.path.dirname(__file__), "network.cfg")


def setup_bondings(bond_mappings):
    print bond_mappings


def add_ovs_port(ovs_br, ifname, uplink, vlan_id=None):
    LOG.info("add_ovs_port enter")
    cmd = "ovs-vsctl --may-exist add-port %s %s" % (ovs_br, ifname)
    if vlan_id:
        cmd += " tag=%s" % vlan_id
    cmd += " -- set Interface %s type=internal;" % ifname
    cmd += "ip link set dev %s address `ip link show %s |awk '/link\/ether/{print $2}'`;" % (ifname, uplink)   # noqa
    cmd += "ip link set %s up;" % ifname
    LOG.info("add_ovs_port: cmd=%s" % cmd)
    os.system(cmd)


def setup_ips(ip_settings, sys_intf_mappings):
    LOG.info("setup_ips enter")
    for intf_info in ip_settings.values():
        network = netaddr.IPNetwork(intf_info["cidr"])
        if sys_intf_mappings[intf_info["name"]]["type"] == "ovs":
            intf_name = intf_info["name"]
        else:
            intf_name = intf_info["alias"]
        if "gw" in intf_info:
            cmd = "ip addr add %s/%s brd %s dev %s;" \
                  % (intf_info["ip"], intf_info["netmask"], str(network.broadcast), intf_name)  # noqa
            cmd += "route del default;"
            cmd += "ip route add default via %s dev %s" % (
                intf_info["gw"], intf_name)
        LOG.info("setup_ips: cmd=%s" % cmd)
        os.system(cmd)


def setup_intfs(sys_intf_mappings, uplink_map):
    LOG.info("setup_intfs enter")
    for intf_name, intf_info in sys_intf_mappings.items():
        if intf_info["type"] == "ovs":
            add_ovs_port(
                intf_info["interface"],
                intf_name,
                uplink_map[intf_info["interface"]],
                vlan_id=intf_info.get("vlan_tag"))
        else:
            pass


def main(config):
    uplink_map = {}
    setup_bondings(config["bond_mappings"])
    for provider_net in config["provider_net_mappings"]:
        uplink_map[provider_net['name']] = provider_net['interface']

    setup_intfs(config["sys_intf_mappings"], uplink_map)
    setup_ips(config["ip_settings"], config["sys_intf_mappings"])


if __name__ == "__main__":
    os.system("service openvswitch-switch status|| service openvswitch-switch start")  # noqa
    config = yaml.load(open(config_path))
    main(config)