Exemplo n.º 1
0
    def pathloadSend(cls):
        """
        None -> None
        
        Client tests bandwidth with pathload and then adds callback to call itself
        """
        shell = fi.throttle.shell.Shell()
        shell.add(os.path.join(fi.throttle.PATHLOAD_DIRECTORY,
                               "pathload/pathload_snd -i"),
                  callback=lambda data: cls.pathloadSend())

        shell.execute()
Exemplo n.º 2
0
    def pathloadReceive(cls):
        """
        None -> None
        
        Calls pathload binary to start measuring bandwidth
        """

        shell = fi.throttle.shell.Shell()
        shell.add(os.path.join(
            fi.throttle.PATHLOAD_DIRECTORY,
            "pathload/pathload_rcv -s %s" % fi.throttle.PATHLOAD_CLIENT),
                  callback=cls.onPathloadReceive)
        shell.execute()
Exemplo n.º 3
0
 def pathloadReceive(cls):
     """
     None -> None
     
     Calls pathload binary to start measuring bandwidth
     """
     
     shell = fi.throttle.shell.Shell()
     shell.add(
         os.path.join(
             fi.throttle.PATHLOAD_DIRECTORY,
             "pathload/pathload_rcv -s %s" % fi.throttle.PATHLOAD_CLIENT
         ),
         callback=cls.onPathloadReceive
     )
     shell.execute()
Exemplo n.º 4
0
 def pathloadSend(cls):
     """
     None -> None
     
     Client tests bandwidth with pathload and then adds callback to call itself
     """
     shell = fi.throttle.shell.Shell()
     shell.add(
         os.path.join(
             fi.throttle.PATHLOAD_DIRECTORY,
             "pathload/pathload_snd -i"
         ),
         callback=lambda data: cls.pathloadSend()
     )
     
     shell.execute()
Exemplo n.º 5
0
    def throttle(cls, allocations):
        """
        allocations:[(str, int)] -> shell:fi.shell.Shell
        
        """

        fi.logmsg(cls, "Making allocations")
        
        shell = fi.throttle.shell.Shell()
        interface = fi.throttle.VPN_INTERFACE
        
        toRun = (
            # Drop current rules
            "/sbin/tc qdisc del dev %s root" % interface,
            "/sbin/iptables -t mangle -F",

            # Create queueing discipline on device root
            "/sbin/tc qdisc add dev %s root handle 1:0 htb" % interface,
            )

        # Run above commands
        for command in toRun:
            shell.add(command)

        # Create node, filter, and /sbin/iptables mark rule for each client
        for i, (ip, allocation) in enumerate(allocations):
            bandwidth = float(allocation) * fi.throttle.BANDWIDTH_HEURISTIC
            # Create classes off of root qdisc
            shell.add("/sbin/tc class add dev %s parent 1: classid 1:%d htb rate %sbps prio %d" % (
                interface, i + 1, bandwidth, i + 1
                )
            )

            # Mark traffic 
            shell.add("/sbin/iptables -t mangle -A POSTROUTING -d %s -j MARK --set-mark %d" % (
                ip, i + 1
                )
            )

            # Filter traffic
            shell.add("/sbin/tc filter add dev %s parent 1:0 protocol ip prio %d handle %d fw flowid 1:%d" % (
                interface, i + 1, i + 1, i + 1
                )
            )
            
        shell.execute()
Exemplo n.º 6
0
    def throttle(cls, allocations):
        """
        allocations:[(str, int)] -> shell:fi.shell.Shell
        
        """

        fi.logmsg(cls, "Making allocations")

        shell = fi.throttle.shell.Shell()
        interface = fi.throttle.VPN_INTERFACE

        toRun = (
            # Drop current rules
            "/sbin/tc qdisc del dev %s root" % interface,
            "/sbin/iptables -t mangle -F",

            # Create queueing discipline on device root
            "/sbin/tc qdisc add dev %s root handle 1:0 htb" % interface,
        )

        # Run above commands
        for command in toRun:
            shell.add(command)

        # Create node, filter, and /sbin/iptables mark rule for each client
        for i, (ip, allocation) in enumerate(allocations):
            bandwidth = float(allocation) * fi.throttle.BANDWIDTH_HEURISTIC
            # Create classes off of root qdisc
            shell.add(
                "/sbin/tc class add dev %s parent 1: classid 1:%d htb rate %sbps prio %d"
                % (interface, i + 1, bandwidth, i + 1))

            # Mark traffic
            shell.add(
                "/sbin/iptables -t mangle -A POSTROUTING -d %s -j MARK --set-mark %d"
                % (ip, i + 1))

            # Filter traffic
            shell.add(
                "/sbin/tc filter add dev %s parent 1:0 protocol ip prio %d handle %d fw flowid 1:%d"
                % (interface, i + 1, i + 1, i + 1))

        shell.execute()