aboutsummaryrefslogtreecommitdiff
path: root/rumba/prototypes/rlite.py
blob: e1e3729d707ab258b084ca5b9d9c6e41b287a229 (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
#
# Commands to setup and instruct rlite
#
#    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 time

# An experiment over the RLITE implementation
class Experiment(mod.Experiment):
    def __init__(self, testbed, nodes=None):
        mod.Experiment.__init__(self, testbed, nodes)

    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:
            ssh.execute_commands(self.testbed, node.ssh_config,
                                 cmds, time_out=None)

    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:
                    print("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:
                    cmds.append("rlite-ctl ipcp-config %s netdev %s" \
                                    % (ipcp.name, ipcp.ifname))

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

    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))

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

    def run_prototype(self):
        print("rlite: setting up")
        self.init()
        print("rlite: software initialized on all nodes")
        self.create_ipcps()
        print("rlite: IPCPs created on all nodes")
        self.register_ipcps()
        print("rlite: IPCPs registered to their lower DIFs on all nodes")