<commit_before>"""
sdn.py defines services to start Open vSwitch and the Ryu SDN Controller.
"""

import re

from core.service import CoreService


class SdnService(CoreService):
    """
    Parent class for SDN services.
    """
    _name = None
    _group = "SDN"
    _depends = ()
    _dirs = ()
    _configs = ()
    _startindex = 50
    _startup = ()
    _shutdown = ()

    @classmethod
    def generateconfig(cls, node, filename, services):
        return ""


class OvsService(SdnService):
    _name = "OvsService"
    _group = "SDN"
    _depends = ()
    _dirs = ("/etc/openvswitch", "/var/run/openvswitch", "/var/log/openvswitch")
    _configs = ('OvsService.sh',)
    _startindex = 50
    _startup = ('sh OvsService.sh',)
    _shutdown = ('killall ovs-vswitchd', 'killall ovsdb-server')

    @classmethod
    def generateconfig(cls, node, filename, services):
        # Check whether the node is running zebra
        has_zebra = 0
        for s in services:
            if s._name == "zebra":
                has_zebra = 1

        # Check whether the node is running an SDN controller
        has_sdn_ctrlr = 0
        for s in services:
            if s._name == "ryuService":
                has_sdn_ctrlr = 1

        cfg = "#!/bin/sh\n"
        cfg += "# auto-generated by OvsService (OvsService.py)\n"
        cfg += "/etc/init.d/openvswitch-switch start < /dev/null\n"
        cfg += "ovs-vsctl add-br ovsbr0\n"
        cfg += "ifconfig ovsbr0 up\n"

        for ifc in node.netifs():
            if hasattr(ifc, 'control') and ifc.control is True:
                continue
            ifnumstr = re.findall(r"\d+", ifc.name)
            ifnum = ifnumstr[0]

            # create virtual interfaces
            cfg += "ip link add rtr%s type veth peer name sw%s\n" % (ifnum, ifnum)
            cfg += "ifconfig rtr%s up\n" % ifnum
            cfg += "ifconfig sw%s up\n" % ifnum

            # remove ip address of eths because quagga/zebra will assign same IPs to rtr interfaces
            # or assign them manually to rtr interfaces if zebra is not running
            for ifcaddr in ifc.addrlist:
                if ifcaddr.find(".") >= 0:
                    cfg += "ip addr del %s dev %s\n" % (ifcaddr, ifc.name)
                    if has_zebra == 0:
                        cfg += "ip addr add %s dev rtr%s\n" % (ifcaddr, ifnum)
                elif ifcaddr.find(":") >= 0:
                    cfg += "ip -6 addr del %s dev %s\n" % (ifcaddr, ifc.name)
                    if has_zebra == 0:
                        cfg += "ip -6 addr add %s dev rtr%s\n" % (ifcaddr, ifnum)
                else:
                    raise ValueError("invalid address: %s" % ifcaddr)

            # add interfaces to bridge
            cfg += "ovs-vsctl add-port ovsbr0 eth%s\n" % ifnum
            cfg += "ovs-vsctl add-port ovsbr0 sw%s\n" % ifnum

        # Add rule for default controller if there is one local (even if the controller is not local, it finds it)
        cfg += "ovs-vsctl set-controller ovsbr0 tcp:127.0.0.1:6633\n"

        # Setup default flows
        portnum = 1
        for ifc in node.netifs():
            if hasattr(ifc, 'control') and ifc.control is True:
                continue
            cfg += "ovs-ofctl add-flow ovsbr0 priority=1000,in_port=%d,action=output:%d\n" % (portnum, portnum + 1)
            cfg += "ovs-ofctl add-flow ovsbr0 priority=1000,in_port=%d,action=output:%d\n" % (portnum + 1, portnum)
            portnum += 2

        return cfg


class RyuService(SdnService):
    _name = "ryuService"
    _group = "SDN"
    _depends = ()
    _dirs = ()
    _configs = ('ryuService.sh',)
    _startindex = 50
    _startup = ('sh ryuService.sh',)
    _shutdown = ('killall ryu-manager',)<commit_msg>add comments to the OVS service<commit_after>"""
sdn.py defines services to start Open vSwitch and the Ryu SDN Controller.
"""

import re

from core.service import CoreService


class SdnService(CoreService):
    """
    Parent class for SDN services.
    """
    _name = None
    _group = "SDN"
    _depends = ()
    _dirs = ()
    _configs = ()
    _startindex = 50
    _startup = ()
    _shutdown = ()

    @classmethod
    def generateconfig(cls, node, filename, services):
        return ""


class OvsService(SdnService):
    _name = "OvsService"
    _group = "SDN"
    _depends = ()
    _dirs = ("/etc/openvswitch", "/var/run/openvswitch", "/var/log/openvswitch")
    _configs = ('OvsService.sh',)
    _startindex = 50
    _startup = ('sh OvsService.sh',)
    _shutdown = ('killall ovs-vswitchd', 'killall ovsdb-server')

    @classmethod
    def generateconfig(cls, node, filename, services):
        # Check whether the node is running zebra
        has_zebra = 0
        for s in services:
            if s._name == "zebra":
                has_zebra = 1

        # Check whether the node is running an SDN controller
        has_sdn_ctrlr = 0
        for s in services:
            if s._name == "ryuService":
                has_sdn_ctrlr = 1

        cfg = "#!/bin/sh\n"
        cfg += "# auto-generated by OvsService (OvsService.py)\n"
        cfg += "## First make sure that the ovs services are up and running\n"
        cfg += "/etc/init.d/openvswitch-switch start < /dev/null\n\n"
        cfg += "## create the switch itself, set the fail mode to secure, \n"
        cfg += "## this stops it from routing traffic without defined flows.\n"
        cfg += "## remove the -- and everything after if you want it to act as a regular switch\n"
        cfg += "ovs-vsctl add-br ovsbr0 -- set Bridge ovsbr0 fail-mode=secure\n"

        cfg += "\n## Now add all our interfaces as ports to the switch\n"
        portnum = 1
        for ifc in node.netifs():
            if hasattr(ifc, 'control') and ifc.control is True:
                continue
            ifnumstr = re.findall(r"\d+", ifc.name)
            ifnum = ifnumstr[0]

            # create virtual interfaces
            cfg += "## Create a veth pair to send the data to\n"
            cfg += "ip link add rtr%s type veth peer name sw%s\n" % (ifnum, ifnum)
#            cfg += "ifconfig rtr%s up\n" % ifnum
#            cfg += "ifconfig sw%s up\n" % ifnum

            # remove ip address of eths because quagga/zebra will assign same IPs to rtr interfaces
            # or assign them manually to rtr interfaces if zebra is not running
            for ifcaddr in ifc.addrlist:
                if ifcaddr.find(".") >= 0:
                    cfg += "ip addr del %s dev %s\n" % (ifcaddr, ifc.name)
                    if has_zebra == 0:
                        cfg += "ip addr add %s dev rtr%s\n" % (ifcaddr, ifnum)
                elif ifcaddr.find(":") >= 0:
                    cfg += "ip -6 addr del %s dev %s\n" % (ifcaddr, ifc.name)
                    if has_zebra == 0:
                        cfg += "ip -6 addr add %s dev rtr%s\n" % (ifcaddr, ifnum)
                else:
                    raise ValueError("invalid address: %s" % ifcaddr)

            # add interfaces to bridge
            # Make port numbers explicit so they're easier to follow in reading the script
            cfg += "## Add the CORE interface to the switch\n"
            cfg += "ovs-vsctl add-port ovsbr0 eth%s -- set Interface eth%s ofport_request=%d\n" % (ifnum, ifnum, portnum)
            cfg += "## And then add its sibling veth interface\n"
            cfg += "ovs-vsctl add-port ovsbr0 sw%s -- set Interface sw%s ofport_request=%d\n" % (ifnum, ifnum, portnum+1)
            cfg += "## start them up so we can send/receive data\n"
            cfg += "ovs-ofctl mod-port ovsbr0 eth%s up\n" % ifnum
            cfg += "ovs-ofctl mod-port ovsbr0 sw%s up\n" % ifnum
            cfg += "## Bring up the lower part of the veth pair\n"
            cfg += "ip link set dev rtr%s up\n" % ifnum
            portnum += 2

        # Add rule for default controller if there is one local (even if the controller is not local, it finds it)
        cfg += "\n## We assume there will be an SDN controller on the other end of this, \n"
        cfg += "## but it will still function if there's not\n"
        cfg += "ovs-vsctl set-controller ovsbr0 tcp:127.0.0.1:6633\n"

        cfg += "\n## Now to create some default flows, \n"
        cfg += "## if the above controller will be present then you probably want to delete them\n"
        # Setup default flows
        portnum = 1
        for ifc in node.netifs():
            if hasattr(ifc, 'control') and ifc.control is True:
                continue
            cfg += "## Take the data from the CORE interface and put it on the veth and vice versa\n"
            cfg += "ovs-ofctl add-flow ovsbr0 priority=1000,in_port=%d,action=output:%d\n" % (portnum, portnum + 1)
            cfg += "ovs-ofctl add-flow ovsbr0 priority=1000,in_port=%d,action=output:%d\n" % (portnum + 1, portnum)
            portnum += 2

        return cfg


class RyuService(SdnService):
    _name = "ryuService"
    _group = "SDN"
    _depends = ()
    _dirs = ()
    _configs = ('ryuService.sh',)
    _startindex = 50
    _startup = ('sh ryuService.sh',)
    _shutdown = ('killall ryu-manager',)
{'fraction_non_alphanumeric': 0.11475589687328579, 'fraction_numerical': 0.010641799232035107, 'mean_word_length': 1.9180537772087067, 'pattern_counts': {'":': 6, '<': 5, '<?xml version=': 0, '>': 7, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 2, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '1433301', 'n_tokens_mistral': 3049, 'n_tokens_neox': 2896, 'n_words': 937}