summaryrefslogtreecommitdiff
path: root/lib/Vyatta/Qos/HFSCShaper.pm
diff options
context:
space:
mode:
Diffstat (limited to 'lib/Vyatta/Qos/HFSCShaper.pm')
-rw-r--r--lib/Vyatta/Qos/HFSCShaper.pm183
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;