diff options
Diffstat (limited to 'lib/Vyatta/Qos/HFSCShaper.pm')
-rw-r--r-- | lib/Vyatta/Qos/HFSCShaper.pm | 183 |
1 files changed, 183 insertions, 0 deletions
diff --git a/lib/Vyatta/Qos/HFSCShaper.pm b/lib/Vyatta/Qos/HFSCShaper.pm new file mode 100644 index 0000000..389cd8c --- /dev/null +++ b/lib/Vyatta/Qos/HFSCShaper.pm @@ -0,0 +1,183 @@ +# Traffic shaper +# This is a extended form of Hierarchal Token Bucket with +# more admin friendly features. Similar in spirt to other shaper scripts +# such as wondershaper. +# +# **** License **** +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License version 2 as +# published by the Free Software Foundation. +# +# This program 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 +# General Public License for more details. +# +# This code was originally developed by Vyatta, Inc. +# Portions created by Vyatta are Copyright (C) 2008 Vyatta, Inc. +# All Rights Reserved. +# **** End License **** + +package Vyatta::Qos::HFSCShaper; +use strict; +use warnings; + +require Vyatta::Config; +require Vyatta::Qos::HFSCClass; +use Vyatta::Qos::Util qw/getRate getAutoRate/; +use POSIX; + +# Create a new instance based on config information +sub new { + my ( $that, $config, $name ) = @_; + my $rate = $config->returnValue("bandwidth"); + my $level = $config->setLevel(); + my @classes = _getClasses($level); + + _checkClasses( $level, $rate, @classes ); + + my $self = {}; + my $class = ref($that) || $that; + bless $self, $class; + + $self->{_rate} = $rate; + $self->{_level} = $level; + $self->{_classes} = \@classes; + return $self; +} + +sub _getClasses { + my $level = shift; + my @classes; + my $config = new Vyatta::Config; + + $config->setLevel($level); + $config->exists("default") + or die "$level configuration not complete: missing default class\n"; + + $config->setLevel("$level default"); + push @classes, new Vyatta::Qos::HFSCClass($config); + $config->setLevel($level); + + foreach my $id ( $config->listNodes("class") ) { + $config->setLevel("$level class $id"); + push @classes, new Vyatta::Qos::HFSCClass( $config, $id ); + } + + return @classes; +} + +# Check constraints on class bandwidth values +sub _checkClasses { + my $level = shift; + my $rate = shift; + + # if auto, can't check for constraints until later + $rate = ( $rate eq "auto" ) ? undef : getRate($rate); + + # handle default class + my $default = shift; + $default->rateCheck( $rate, "$level default" ) if $rate; + + foreach my $class (@_) { + # Check rates for each class + $class->rateCheck( $rate, "$level class $class->{id}" ) if $rate; + } +} + +sub _minRate { + my ($speed, $classes) = @_; + my $min = $speed / 8; + + foreach my $class (@$classes) { + my $bps = $class->get_rate($speed) / 8; # bytes per second + + $min = $bps if $bps < $min; + } + + return $min; +} + +sub commands { + my ( $self, $dev ) = @_; + my $rate = getAutoRate( $self->{_rate}, $dev ); + my $classes = $self->{_classes}; + my %dsmark = (); + my $default = shift @$classes; + my $maxid = 1; + + $default->rateCheck( $rate, "$self->{_level} default" ); + + foreach my $class (@$classes) { + my $level = "$self->{_level} class $class->{id}"; + $class->rateCheck( $rate, $level ); + + # find largest class id + if ( defined $class->{id} && $class->{id} > $maxid ) { + $maxid = $class->{id}; + } + } + + # fill in id of default + $default->{id} = ++$maxid; + unshift @$classes, $default; + + # Check if we need dsmrk + my $usedsmark; + foreach my $class (@$classes) { + if ( defined $class->{dsmark} ) { + $usedsmark = 1; + last; + } + } + + my $parent = 1; + my $root = "root"; + +=pod + # if we need to change dsfield values, then put dsmark in front + if ($usedsmark) { + + # dsmark max index must be power of 2 + my $indices = $maxid + 1; + while ( ( $indices & ( $indices - 1 ) ) != 0 ) { + ++$indices; + } + + print "qdisc add dev $dev handle 1:0 root dsmark" + . " indices $indices default_index $default->{id} set_tc_index\n"; + + foreach my $class (@$classes) { + $class->dsmarkClass( 1, $dev ); + foreach my $match ( $class->matchRules() ) { + $match->filter( $dev, $parent, $class->{id}, 1 ); + } + } + + $parent = $indices + 1; + $root = "parent 1:1"; + } +=cut + # Add root qdisc + printf "qdisc add dev %s %s handle %x: hfsc default %x\n", + $dev, $root, $parent, $default->{id}; + # Add root class + # Note : sc = realtime + linkshare + printf "class add dev %s parent %x: classid %x:1 hfsc sc rate %s ul rate %s\n", + $dev, $parent, $parent, $rate, $rate; + + printf "qdisc add dev %s parent %x:1 handle f1: sfq perturb 10\n", + $dev, $parent; + + my $prio = 1; + foreach my $class (@$classes) { + $class->gen_class( $dev, 'hfsc', $parent, $rate ); + + foreach my $match ( $class->matchRules() ) { + $match->filter( $dev, $parent, $class->{id}, $prio++, + $class->{dsmark} ); + } + } +} + +1; |