1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
# 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 Vyatta::Qos::TrafficLimiter;
use strict;
use warnings;
require Vyatta::Config;
require Vyatta::Qos::LimiterClass;
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 "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") ) {
next if ($type eq 'description');
$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";
}
if ($config->exists('default')) {
$config->setLevel("$level default");
push @classes, new Vyatta::Qos::LimiterClass( $config, 0 );
}
foreach my $id ( $config->listNodes('class') ) {
$config->setLevel("$level class $id");
push @classes, new Vyatta::Qos::LimiterClass( $config, $id );
}
$self->{_classes} = \@classes;
}
sub commands {
my ( $self, $dev, $direction ) = @_;
my $classes = $self->{_classes};
my $parent;
if ($direction eq 'in') {
$parent = 0xffff;
printf "qdisc add dev %s handle %x: ingress\n", $dev, $parent;
} else {
$parent = 1;
printf "qdisc add dev %s handle %x: prio\n", $dev, $parent;
}
# find largest class id (to use for default)
my $maxid = 0;
foreach my $class (@$classes) {
my $id = $class->{id};
$maxid = $id if ($id > $maxid);
}
foreach my $class (@$classes) {
foreach my $match ( $class->matchRules() ) {
my $id = $class->{id};
$id = $maxid + 1 if ($id == 0);
my $police = " police rate " . $class->{rate}
. " action drop burst " . $class->{burst};
$match->filter( $dev, $parent, $id, $class->{priority}, undef, $police );
}
}
}
1;
|
