ratelimiter.py
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

ratelimiter.py
View Options

	# Copyright 2022 VyOS maintainers and contributors <maintainers@vyos.io>
	#
	# This library is free software; you can redistribute it and/or
	# modify it under the terms of the GNU Lesser General Public
	# License as published by the Free Software Foundation; either
	# version 2.1 of the License, or (at your option) any later version.
	#
	# This library 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
	# Lesser General Public License for more details.
	#
	# You should have received a copy of the GNU Lesser General Public
	# License along with this library. If not, see <http://www.gnu.org/licenses/>.

	from vyos.qos.base import QoSBase

	class RateLimiter(QoSBase):
	# https://man7.org/linux/man-pages/man8/tc-tbf.8.html
	def update(self, config, direction):
	# call base class
	super().update(config, direction)

	tmp = f'tc qdisc add dev {self._interface} root tbf'
	if 'bandwidth' in config:
	rate = self._rate_convert(config['bandwidth'])
	tmp += f' rate {rate}'

	if 'burst' in config:
	burst = config['burst']
	tmp += f' burst {burst}'

	if 'latency' in config:
	latency = config['latency']
	tmp += f' latency {latency}ms'

	self._cmd(tmp)