aboutsummaryrefslogtreecommitdiff
path: root/rumba/prototypes/rlite.py
blob: feca5a496e420ec8f0199afaa817b733bc7bbe98 (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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#
# Commands to setup and instruct rlite
#
#    Author: Vincenzo Maffione <v.maffione@nextworks.it>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA  02110-1301  USA

import rumba.ssh_support as ssh
import rumba.model as mod
import rumba.log as log


logger = log.get_logger(__name__)


# An experiment over the rlite implementation
class Experiment(mod.Experiment):

    def __init__(self, testbed, nodes=None):
        mod.Experiment.__init__(self, testbed, nodes)

    def execute_commands(self, node, cmds):
        ssh.execute_commands(self.testbed, node.ssh_config,
                             cmds, time_out=None)

    def init(self):
        cmds = []

        if False:  # ubuntu
            cmds.append("apt-get update")
            cmds.append("apt-get install g++ gcc cmake "
                        "linux-headers-$(uname -r) "
                        "protobuf-compiler libprotobuf-dev git --yes")
            cmds.append("rm -rf ~/rlite")
            cmds.append("cd ~; git clone https://github.com/vmaffione/rlite")
            cmds.append("cd ~/rlite && ./configure && make "
                        "&& sudo make install")

        # Load kernel modules
        cmds.append("modprobe rlite")
        cmds.append("modprobe rlite-normal")
        cmds.append("modprobe rlite-shim-eth")
        cmds.append("modprobe rlite-shim-udp4")
        cmds.append("modprobe rlite-shim-loopback")

        # Start the uipcps daemon
        cmds.append("rlite-uipcps -v DBG -k 0 &> uipcp.log &")

        for node in self.nodes:
            self.execute_commands(node, cmds)

    def create_ipcps(self):
        for node in self.nodes:
            cmds = []

            for ipcp in node.ipcps:
                # Generate the command to create the IPCP
                if type(ipcp.dif) is mod.NormalDIF:
                    ipcp_type = 'normal'
                elif type(ipcp.dif) is mod.ShimEthDIF:
                    ipcp_type = 'shim-eth'
                elif type(ipcp.dif) is mod.ShimUDPDIF:
                    ipcp_type = 'shim-udp4'
                else:
                    logger.warning(
                        "unknown type for DIF %s, default to loopback",
                        ipcp.dif.name)
                    ipcp_type = 'shim-loopback'

                cmds.append("rlite-ctl ipcp-create %s %s %s" %
                            (ipcp.name, ipcp_type, ipcp.dif.name))

                # Generate the command to configure the interface
                # name for the shim-eth
                if type(ipcp.dif) is mod.ShimEthDIF:
                    ipcp.ifname = 'eth1'
                    cmds.append("rlite-ctl ipcp-config %s netdev %s"
                                % (ipcp.name, ipcp.ifname))

            self.execute_commands(node, cmds)

    def register_ipcps(self):
        for node in self.nodes:
            cmds = []

            for ipcp in node.ipcps:
                for lower in ipcp.registrations:
                    cmds.append("rlite-ctl ipcp-register %s %s"
                                % (ipcp.name, lower.name))

            self.execute_commands(node, cmds)

    def enroll_ipcps(self):
        for el in self.enrollments:
            for e in el:
                d = {'enrollee': e['enrollee'].name,
                     'dif': e['dif'].name,
                     'lower_dif': e['lower_dif'].name,
                     'enroller': e['enroller'].name
                     }
                cmd = "rlite-ctl ipcp-enroll %(enrollee)s %(dif)s "\
                      "%(lower_dif)s %(enroller)s" % d
                self.execute_commands(e['enrollee'], [cmd])

    def run_prototype(self):
        logger.info("rlite: setting up")
        self.init()
        logger.info("rlite: software initialized on all nodes")
        self.create_ipcps()
        logger.info("rlite: IPCPs created on all nodes")
        self.register_ipcps()
        logger.info("rlite: IPCPs registered to their lower DIFs on all nodes")
        self.enroll_ipcps()
        logger.info("rlite: enrollment completed in all DIFs")