1 files changed, 58 insertions, 0 deletions
diff --git a/scripts/VyattaQosRateLimiter.pm b/scripts/VyattaQosRateLimiter.pm
new file mode 100644
index 0000000..548526b
--- /dev/null
+++ b/scripts/VyattaQosRateLimiter.pm
@@ -0,0 +1,58 @@
+# 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 VyattaQosRateLimiter;
+
+use strict;
+
+require VyattaConfig;
+use VyattaQosUtil;
+
+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}     = VyattaQosUtil::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}     = VyattaQosUtil::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};
+}
+
+1;