You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
3.5 KiB
96 lines
3.5 KiB
# ADN DMR Peer Server - tests hbp hbp rate limit
|
|
#
|
|
# Copyright (C) 2026 Rodrigo Pérez, CE5RPY <ce5rpy@qmd.cl>
|
|
#
|
|
###############################################################################
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# 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.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
###############################################################################
|
|
|
|
"""HBP ingress rate limit."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
from tests.harness.assertions import assert_capture_unchanged
|
|
from tests.harness.deterministic import DeterministicScenario, PacketSpec, active_routing_table
|
|
|
|
import adn_server.application.routing_use_cases as buc
|
|
|
|
|
|
@pytest.mark.behavior
|
|
def test_hbp_rate_limit_drops_excessive_ingress_rate() -> None:
|
|
"""Regression: excessive ingress rate triggers drop; post-drop forwards stall."""
|
|
bridges = active_routing_table(91, (("MASTER-A", 2), ("MASTER-B", 2)))
|
|
scenario = DeterministicScenario(routing_table=bridges)
|
|
base = PacketSpec(dst_id=91, stream_id=0x55667788)
|
|
t0 = scenario.clock.time()
|
|
|
|
scenario.inject_hbp(
|
|
"MASTER-A",
|
|
DeterministicScenario.voice_head_spec(base),
|
|
ingress_pkt_time=t0,
|
|
)
|
|
dropped = False
|
|
dropped_at = 0
|
|
for seq in range(1, 30):
|
|
ok = scenario.inject_hbp(
|
|
"MASTER-A",
|
|
DeterministicScenario.voice_burst_spec(base, seq=seq, dtype_vseq=min(seq, 4)),
|
|
ingress_pkt_time=t0 + seq * 0.01,
|
|
)
|
|
if not ok:
|
|
dropped = True
|
|
dropped_at = seq
|
|
break
|
|
assert dropped
|
|
|
|
forwarded_before = len(scenario.capture.for_system("MASTER-B"))
|
|
for seq in range(dropped_at, dropped_at + 5):
|
|
scenario.inject_hbp(
|
|
"MASTER-A",
|
|
DeterministicScenario.voice_burst_spec(base, seq=seq, dtype_vseq=min(seq, 4)),
|
|
ingress_pkt_time=t0 + seq * 0.01,
|
|
)
|
|
assert_capture_unchanged(scenario, "MASTER-B", forwarded_before)
|
|
|
|
|
|
def test_hbp_rate_limit_uses_ingress_time_not_wall_clock() -> None:
|
|
"""Rate control must use ingress_pkt_time even when wall clock is frozen."""
|
|
bridges = active_routing_table(91, (("MASTER-A", 2), ("MASTER-B", 2)))
|
|
scenario = DeterministicScenario(routing_table=bridges)
|
|
base = PacketSpec(dst_id=91, stream_id=0x34343434)
|
|
t0 = 1_700_000_200.0
|
|
|
|
original_time = buc.time.time
|
|
buc.time.time = lambda: t0
|
|
try:
|
|
scenario.inject_hbp(
|
|
"MASTER-A",
|
|
DeterministicScenario.voice_head_spec(base),
|
|
ingress_pkt_time=t0,
|
|
)
|
|
accepted = 0
|
|
for seq in range(1, 24):
|
|
ok = scenario.inject_hbp(
|
|
"MASTER-A",
|
|
DeterministicScenario.voice_burst_spec(base, seq=seq, dtype_vseq=min(seq, 4)),
|
|
ingress_pkt_time=t0 + seq * 0.06,
|
|
)
|
|
if ok is not False:
|
|
accepted += 1
|
|
assert accepted >= 20
|
|
finally:
|
|
buc.time.time = original_time
|