FM / tcmsg

/opt/imh-python/lib/python3.9/site-packages/pyroute2/netlink/rtnl/tcmsg/

/opt/imh-python/lib/python3.9/site-packages/pyroute2/netlink/rtnl/tcmsg

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
act_bpf.py	979	0644	edit dl rm
act_connmark.py	1234	0644	edit dl rm
act_gact.py	709	0644	edit dl rm
act_mirred.py	1725	0644	edit dl rm
act_police.py	1866	0644	edit dl rm
act_skbedit.py	3337	0644	edit dl rm
act_vlan.py	1465	0644	edit dl rm
cls_basic.py	8478	0644	edit dl rm
cls_flow.py	4613	0644	edit dl rm
cls_fw.py	1427	0644	edit dl rm
cls_matchall.py	986	0644	edit dl rm
cls_u32.py	8034	0644	edit dl rm
common.py	10728	0644	edit dl rm
common_act.py	2063	0644	edit dl rm
common_ematch.py	3148	0644	edit dl rm
em_cmp.py	2776	0644	edit dl rm
em_ipset.py	1741	0644	edit dl rm
em_meta.py	5933	0644	edit dl rm
sched_bpf.py	2571	0644	edit dl rm
sched_cake.py	12833	0644	edit dl rm
sched_choke.py	3432	0644	edit dl rm
sched_clsact.py	1011	0644	edit dl rm
sched_codel.py	1607	0644	edit dl rm
sched_drr.py	755	0644	edit dl rm
sched_fq_codel.py	2658	0644	edit dl rm
sched_hfsc.py	2353	0644	edit dl rm
sched_htb.py	5739	0644	edit dl rm
sched_ingress.py	214	0644	edit dl rm
sched_netem.py	5142	0644	edit dl rm
sched_pfifo.py	196	0644	edit dl rm
sched_pfifo_fast.py	215	0644	edit dl rm
sched_plug.py	446	0644	edit dl rm
sched_sfq.py	2717	0644	edit dl rm
sched_tbf.py	1087	0644	edit dl rm
sched_template.py	1250	0644	edit dl rm
__init__.py	2743	0644	edit dl rm

Edit: /opt/imh-python/lib/python3.9/site-packages/pyroute2/netlink/rtnl/tcmsg/sched_choke.py (3432B)

'''
choke
+++++

Parameters:

* `limit` (required) -- int
    * `bandwith` (required) -- str/int
    * `min` -- int
    * `max` -- int
    * `avpkt` -- str/int, packet size
    * `burst` -- int
    * `probability` -- float
    * `ecn` -- bool

Example::

ip.tc('add', 'choke', interface,
          limit=5500,
          bandwith="10mbit",
          ecn=True)

'''

import logging
import struct

from pyroute2.netlink import nla, nla_string
from pyroute2.netlink.rtnl import TC_H_ROOT
from pyroute2.netlink.rtnl.tcmsg.common import (
    get_rate,
    get_size,
    red_eval_ewma,
    red_eval_idle_damping,
    red_eval_P,
)
from pyroute2.netlink.rtnl.tcmsg.common import stats2 as c_stats2

log = logging.getLogger(__name__)
parent = TC_H_ROOT

def get_parameters(kwarg):
    # The code is ported from iproute2
    avpkt = 1000
    probability = 0.02
    opt = {
        'limit': kwarg['limit'],  # required
        'qth_min': kwarg.get('min', 0),
        'qth_max': kwarg.get('max', 0),
        'Wlog': 0,
        'Plog': 0,
        'Scell_log': 0,
        'flags': 1 if kwarg.get('ecn') else 0,
    }

rate = get_rate(kwarg['bandwith'])  # required
    burst = kwarg.get('burst', 0)
    avpkt = get_size(kwarg.get('avpkt', 1000))
    probability = kwarg.get('probability', 0.02)

if not opt['qth_max']:
        opt['qth_max'] = opt['limit'] // 4
    if not opt['qth_min']:
        opt['qth_min'] = opt['qth_max'] // 3
    if not burst:
        burst = (2 * opt['qth_min'] + opt['qth_max']) // 3

if opt['qth_max'] > opt['limit']:
        raise Exception('max is larger than limit')
    if opt['qth_min'] >= opt['qth_max']:
        raise Exception('min is not smaller than max')

# Wlog
    opt['Wlog'] = red_eval_ewma(opt['qth_min'] * avpkt, burst, avpkt)
    if opt['Wlog'] < 0:
        raise Exception('failed to calculate EWMA')
    elif opt['Wlog'] > 10:
        log.warning('choke: burst %s seems to be too large' % burst)
    # Plog
    opt['Plog'] = red_eval_P(
        opt['qth_min'] * avpkt, opt['qth_max'] * avpkt, probability
    )
    if opt['Plog'] < 0:
        raise Exception('choke: failed to calculate probability')
    # Scell_log, stab
    opt['Scell_log'], stab = red_eval_idle_damping(opt['Wlog'], avpkt, rate)
    if opt['Scell_log'] < 0:
        raise Exception('choke: failed to calculate idle damping table')

return {
        'attrs': [
            ['TCA_CHOKE_PARMS', opt],
            ['TCA_CHOKE_STAB', stab],
            ['TCA_CHOKE_MAX_P', int(probability * pow(2, 32))],
        ]
    }

class options(nla):
    nla_map = (
        ('TCA_CHOKE_UNSPEC', 'none'),
        ('TCA_CHOKE_PARMS', 'qopt'),
        ('TCA_CHOKE_STAB', 'stab'),
        ('TCA_CHOKE_MAX_P', 'uint32'),
    )

class qopt(nla):
        fields = (
            ('limit', 'I'),
            ('qth_min', 'I'),
            ('qth_max', 'I'),
            ('Wlog', 'B'),
            ('Plog', 'B'),
            ('Scell_log', 'B'),
            ('flags', 'B'),
        )

class stab(nla_string):
        def encode(self):
            self['value'] = struct.pack(
                'B' * 256, *(int(x) for x in self.value)
            )
            nla_string.encode(self)

class stats(nla):
    fields = (
        ('early', 'I'),
        ('pdrop', 'I'),
        ('other', 'I'),
        ('marked', 'I'),
        ('matched', 'I'),
    )

class stats2(c_stats2):
    class stats_app(stats):
        pass

cmd: