1 files changed, 72 insertions, 0 deletions
diff --git a/lib/Vyatta/Qos/RateLimiter.pm b/lib/Vyatta/Qos/RateLimiter.pm
new file mode 100644
index 0000000..625db62
--- /dev/null
+++ b/lib/Vyatta/Qos/RateLimiter.pm
@@ -0,0 +1,72 @@
+# This is a wrapper around Token Bucket Filter (TBF) queue discipline
+#
+#
+# **** 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::RateLimiter;
+
+use strict;
+use warnings;
+
+use VyattaConfig;
+use Vyatta::Qos::Util qw/getRate getTime/;
+
+my %fields = (
+    _rate	=> undef,
+    _burst	=> undef,
+    _latency	=> undef,
+);
+
+sub new {
+    my ( $that, $config ) = @_;
+    my $level = $config->setLevel();
+    my $class = ref($that) || $that;
+    my $self = {%fields};
+
+    $self->{_rate} = getRate($config->returnValue("bandwidth"));
+    defined $self->{_rate}  or die "$level bandwidth not defined\n";
+
+    $self->{_burst}     = $config->returnValue("burst");
+    defined $self->{_burst}  or die "$level burst not defined\n";
+
+    $self->{_latency} = getTime($config->returnValue("latency"));
+    defined $self->{_latency}  or die "$level latency not defined\n";
+
+    return bless $self, $class;
+}
+
+sub commands {
+    my ( $self, $out, $dev ) = @_;
+
+    
+    printf {$out} "qdisc add dev %s root tbf rate %s latency %s burst %s\n",
+	    $dev, $self->{_rate}, $self->{_latency}, $self->{_burst};
+}
+
+sub isChanged {
+    my ($self, $name) = @_;
+    my $config = new VyattaConfig;
+
+    $config->setLevel("qos-policy rate-limit $name");
+    foreach my $attr ('bandwidth', 'burst', 'latency') {
+	if ($config->isChanged($attr)) {
+	    return $attr
+	}
+    }
+    return undef; # false
+}
+
+1;