1 files changed, 194 insertions, 0 deletions
diff --git a/lib/Vyatta/Qos/TrafficLimiter.pm b/lib/Vyatta/Qos/TrafficLimiter.pm
new file mode 100644
index 0000000..8564e87
--- /dev/null
+++ b/lib/Vyatta/Qos/TrafficLimiter.pm
@@ -0,0 +1,194 @@
+# Traffic limiter
+# This is a rate limiter based on ingress qdisc
+#
+# **** 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 LimiterClass;
+    use strict;
+    require VyattaConfig;
+    use Vyatta::Qos::Match;
+    use Vyatta::Qos::Util qw/getRate/;
+
+    my %fields = (
+	id	 => undef,
+        priority => undef,
+        rate     => undef,
+        _match   => undef,
+    );
+
+    sub new {
+        my ( $that, $config, $id ) = @_;
+        my $class = ref($that) || $that;
+        my $self = {%fields};
+
+        $self->{id} = $id;
+
+        bless $self, $class;
+        $self->_define($config);
+
+        return $self;
+    }
+
+    sub _define {
+        my ( $self, $config ) = @_;
+        my $level   = $config->setLevel();
+        my @matches = ();
+	my $rate = $config->returnValue("bandwidth");
+	
+	defined $rate or die "bandwidth must be defined for $level\n";
+        $self->{rate} = getRate($rate);
+
+        $self->{priority} = $config->returnValue("priority");
+
+        foreach my $match ( $config->listNodes("match") ) {
+            $config->setLevel("$level match $match");
+            push @matches, new Vyatta::Qos::Match($config);
+        }
+        $self->{_match} = \@matches;
+    }
+
+    sub matchRules {
+        my ($self) = @_;
+        my $matches = $self->{_match};
+        return @$matches;
+    }
+
+}
+
+package Vyatta::Qos::TrafficLimiter;
+use strict;
+require VyattaConfig;
+
+my %fields = (
+    _level   => undef,
+    _classes => undef,
+);
+
+# Create a new instance based on config information
+sub new {
+    my ( $that, $config, $name ) = @_;
+    my $self = {%fields};
+    my $class = ref($that) || $that;
+
+    bless $self, $class;
+    $self->_define($config);
+
+    return $self;
+}
+
+# Setup new instance.
+# Assumes caller has done $config->setLevel to "traffic-limiter $name"
+sub _define {
+    my ( $self, $config ) = @_;
+    my $level   = $config->setLevel();
+    my @classes = ();
+
+    $self->{_level} = $level;
+
+    # make sure no clash of different types of tc filters
+    my %matchTypes = ();
+    foreach my $class ( $config->listNodes("class") ) {
+        foreach my $match ( $config->listNodes("class $class match") ) {
+            foreach my $type ( $config->listNodes("class $class match $match") )
+            {
+                $matchTypes{$type} = "$class match $match";
+            }
+        }
+    }
+
+    if ( scalar keys %matchTypes > 1 && $matchTypes{ip} ) {
+        print "Match type conflict:\n";
+        while ( my ( $type, $usage ) = each(%matchTypes) ) {
+            print "   class $usage $type\n";
+        }
+        die "$level can not match on both ip and other types\n";
+    }
+
+    foreach my $id ( $config->listNodes("class") ) {
+        $config->setLevel("$level class $id");
+        push @classes, new LimiterClass( $config, $id );
+    }
+    $self->{_classes} = \@classes;
+}
+
+sub commands {
+    my ( $self, $out, $dev ) = @_;
+    my $classes = $self->{_classes};
+    my $parent  = 0xffff;
+
+    printf {$out} "qdisc add dev %s handle %x: ingress\n", $dev, $parent;
+    foreach my $class (@$classes) {
+        my $id       = $class->{id};
+        my $rate     = $class->{rate};
+        my $priority = $class->{priority};
+
+        foreach my $match ( $class->matchRules() ) {
+            $match->filter( $out, $dev, $parent, $priority );
+            printf {$out} " police avrate %s drop flowid :%x\n", $rate, $id;
+        }
+    }
+}
+
+# Walk configuration tree and look for changed nodes
+# The configuration system should do this but doesn't do it right
+sub isChanged {
+    my ( $self, $name ) = @_;
+    my $config = new VyattaConfig;
+
+    $config->setLevel("qos-policy traffic-limiter $name");
+    my %classNodes = $config->listNodeStatus('class');
+    while ( my ( $class, $status ) = each %classNodes ) {
+        if ( $status ne 'static' ) {
+            return "class $class";
+        }
+
+        foreach my $attr ( 'bandwidth', 'burst', 'priority' ) {
+            if ( $config->isChanged("class $class $attr") ) {
+                return "class $class $attr";
+            }
+        }
+
+        my %matchNodes = $config->listNodeStatus("class $class match");
+        while ( my ( $match, $status ) = each %matchNodes ) {
+            my $level = "class $class match $match";
+            if ( $status ne 'static' ) {
+                return $level;
+            }
+
+            foreach my $parm (
+                'vif',
+                'interface',
+                'ip dscp',
+                'ip protocol',
+                'ip source address',
+                'ip destination address',
+                'ip source port',
+                'ip destination port'
+              )
+            {
+                if ( $config->isChanged("$level $parm") ) {
+                    return "$level $parm";
+                }
+            }
+        }
+    }
+
+    return undef;    # false
+}
+
+1;