# 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)